81 lines
1.6 KiB
Go
81 lines
1.6 KiB
Go
package agent
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"encoding/json"
|
|
"fmt"
|
|
"os"
|
|
"os/exec"
|
|
"path/filepath"
|
|
|
|
"chat-agent-system/internal/config"
|
|
)
|
|
|
|
type CursorCLIBackend struct {
|
|
cfg *config.Config
|
|
}
|
|
|
|
func (c *CursorCLIBackend) Name() string { return "cursor-cli" }
|
|
|
|
func (c *CursorCLIBackend) Execute(ctx context.Context, req Request) (*Result, error) {
|
|
if len(req.Targets) == 0 {
|
|
req.Targets = []string{"output.md"}
|
|
}
|
|
|
|
payload, _ := json.Marshal(map[string]any{
|
|
"workspace": req.WorkDir,
|
|
"api_key": c.cfg.CursorAPIKey,
|
|
"model": c.cfg.CursorModel,
|
|
"prompt": req.Prompt,
|
|
"targets": req.Targets,
|
|
})
|
|
|
|
bridgeBin := c.cfg.CursorBridgeBin
|
|
if !filepath.IsAbs(bridgeBin) {
|
|
wd, _ := os.Getwd()
|
|
bridgeBin = filepath.Join(wd, bridgeBin)
|
|
}
|
|
|
|
timeout := req.Timeout
|
|
if timeout <= 0 {
|
|
timeout = c.cfg.AgentTimeout
|
|
}
|
|
runCtx, cancel := context.WithTimeout(ctx, timeout)
|
|
defer cancel()
|
|
|
|
cmd := exec.CommandContext(runCtx, bridgeBin)
|
|
cmd.Dir = req.WorkDir
|
|
cmd.Stdin = bytes.NewReader(payload)
|
|
cmd.Env = append(os.Environ(),
|
|
"CURSOR_API_KEY="+c.cfg.CursorAPIKey,
|
|
"CURSOR_MODEL="+c.cfg.CursorModel,
|
|
)
|
|
|
|
var stdout, stderr bytes.Buffer
|
|
cmd.Stdout = &stdout
|
|
cmd.Stderr = &stderr
|
|
|
|
err := cmd.Run()
|
|
if err != nil {
|
|
return &Result{
|
|
Success: false,
|
|
ErrorReason: fmt.Sprintf("cursor execution error: %v, stderr: %s", err, stderr.String()),
|
|
}, nil
|
|
}
|
|
|
|
patches := make(map[string][]byte)
|
|
for _, t := range req.Targets {
|
|
p := filepath.Join(req.WorkDir, t)
|
|
if b, err := os.ReadFile(p); err == nil {
|
|
patches[t] = b
|
|
}
|
|
}
|
|
|
|
return &Result{
|
|
Success: true,
|
|
Patches: patches,
|
|
Logs: stdout.String(),
|
|
}, nil
|
|
}
|