73 lines
1.6 KiB
Go
73 lines
1.6 KiB
Go
|
package agent
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"cursor-api-proxy/internal/config"
|
||
|
|
"cursor-api-proxy/internal/process"
|
||
|
|
"os"
|
||
|
|
"path/filepath"
|
||
|
|
)
|
||
|
|
|
||
|
|
func init() {
|
||
|
|
process.MaxModeFn = RunMaxModePreflight
|
||
|
|
}
|
||
|
|
|
||
|
|
func cacheTokenForAccount(configDir string) {
|
||
|
|
if configDir == "" {
|
||
|
|
return
|
||
|
|
}
|
||
|
|
token := ReadKeychainToken()
|
||
|
|
if token != "" {
|
||
|
|
WriteCachedToken(configDir, token)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func AccountsDir() string {
|
||
|
|
home := os.Getenv("HOME")
|
||
|
|
if home == "" {
|
||
|
|
home = os.Getenv("USERPROFILE")
|
||
|
|
}
|
||
|
|
return filepath.Join(home, ".cursor-api-proxy", "accounts")
|
||
|
|
}
|
||
|
|
|
||
|
|
func RunAgentSync(cfg config.BridgeConfig, workspaceDir string, cmdArgs []string, tempDir, configDir string, ctx context.Context) (process.RunResult, error) {
|
||
|
|
opts := process.RunOptions{
|
||
|
|
Cwd: workspaceDir,
|
||
|
|
TimeoutMs: cfg.TimeoutMs,
|
||
|
|
MaxMode: cfg.MaxMode,
|
||
|
|
ConfigDir: configDir,
|
||
|
|
Ctx: ctx,
|
||
|
|
}
|
||
|
|
|
||
|
|
result, err := process.Run(cfg.AgentBin, cmdArgs, opts)
|
||
|
|
|
||
|
|
cacheTokenForAccount(configDir)
|
||
|
|
if tempDir != "" {
|
||
|
|
os.RemoveAll(tempDir)
|
||
|
|
}
|
||
|
|
|
||
|
|
return result, err
|
||
|
|
}
|
||
|
|
|
||
|
|
func RunAgentStreamWithContext(cfg config.BridgeConfig, workspaceDir string, cmdArgs []string, onLine func(string), tempDir, configDir string, ctx context.Context) (process.StreamResult, error) {
|
||
|
|
opts := process.RunStreamingOptions{
|
||
|
|
RunOptions: process.RunOptions{
|
||
|
|
Cwd: workspaceDir,
|
||
|
|
TimeoutMs: cfg.TimeoutMs,
|
||
|
|
MaxMode: cfg.MaxMode,
|
||
|
|
ConfigDir: configDir,
|
||
|
|
Ctx: ctx,
|
||
|
|
},
|
||
|
|
OnLine: onLine,
|
||
|
|
}
|
||
|
|
|
||
|
|
result, err := process.RunStreaming(cfg.AgentBin, cmdArgs, opts)
|
||
|
|
|
||
|
|
cacheTokenForAccount(configDir)
|
||
|
|
if tempDir != "" {
|
||
|
|
os.RemoveAll(tempDir)
|
||
|
|
}
|
||
|
|
|
||
|
|
return result, err
|
||
|
|
}
|