89 lines
3.0 KiB
Go
89 lines
3.0 KiB
Go
package cli
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"github.com/spf13/cobra"
|
|
|
|
"github.com/aoagents/agent-orchestrator/backend/internal/adapters/runtime/zellij"
|
|
)
|
|
|
|
type spawnOptions struct {
|
|
project string
|
|
harness string
|
|
branch string
|
|
prompt string
|
|
issue string
|
|
rules string
|
|
}
|
|
|
|
// spawnRequest mirrors the daemon's SpawnSessionRequest body for
|
|
// POST /api/v1/sessions. The CLI keeps its own copy so it need not import httpd.
|
|
type spawnRequest struct {
|
|
ProjectID string `json:"projectId"`
|
|
IssueID string `json:"issueId,omitempty"`
|
|
Harness string `json:"harness,omitempty"`
|
|
Branch string `json:"branch,omitempty"`
|
|
Prompt string `json:"prompt,omitempty"`
|
|
AgentRules string `json:"agentRules,omitempty"`
|
|
}
|
|
|
|
type spawnResult struct {
|
|
Session struct {
|
|
ID string `json:"id"`
|
|
Status string `json:"status"`
|
|
} `json:"session"`
|
|
}
|
|
|
|
func newSpawnCommand(ctx *commandContext) *cobra.Command {
|
|
var opts spawnOptions
|
|
cmd := &cobra.Command{
|
|
Use: "spawn",
|
|
Short: "Spawn a worker agent session in a registered project",
|
|
Long: "Spawn a worker agent session in a registered project.\n\n" +
|
|
"The session runs the chosen agent (default: the daemon's AO_AGENT) in a\n" +
|
|
"fresh git worktree. Register the project first with `ao project add`.",
|
|
Args: noArgs,
|
|
RunE: func(cmd *cobra.Command, args []string) error {
|
|
if opts.project == "" {
|
|
return usageError{fmt.Errorf("--project is required")}
|
|
}
|
|
req := spawnRequest{
|
|
ProjectID: opts.project,
|
|
IssueID: opts.issue,
|
|
Harness: opts.harness,
|
|
Branch: opts.branch,
|
|
Prompt: opts.prompt,
|
|
AgentRules: opts.rules,
|
|
}
|
|
var res spawnResult
|
|
if err := ctx.postJSON(cmd.Context(), "sessions", req, &res); err != nil {
|
|
return err
|
|
}
|
|
out := cmd.OutOrStdout()
|
|
if _, err := fmt.Fprintf(out, "spawned session %s (%s)\n", res.Session.ID, res.Session.Status); err != nil {
|
|
return err
|
|
}
|
|
// The daemon runs zellij under a short, non-default socket dir (see
|
|
// zellij.DefaultSocketDir), so a plain `zellij attach` wouldn't find
|
|
// the session — prefix the env so the hint is copy-pasteable. Use the
|
|
// sanitised name zellij actually registers (zellij.SessionName): a long
|
|
// session id maps to a different name than the raw id.
|
|
attach := fmt.Sprintf("zellij attach %s", zellij.SessionName(res.Session.ID))
|
|
if dir := zellij.DefaultSocketDir(); dir != "" {
|
|
attach = fmt.Sprintf("ZELLIJ_SOCKET_DIR=%s %s", dir, attach)
|
|
}
|
|
_, err := fmt.Fprintf(out, "attach with: %s\n", attach)
|
|
return err
|
|
},
|
|
}
|
|
f := cmd.Flags()
|
|
f.StringVar(&opts.project, "project", "", "Project id to spawn the session in (required)")
|
|
f.StringVar(&opts.harness, "harness", "", "Agent harness: claude-code, codex, … (default: the daemon's AO_AGENT)")
|
|
f.StringVar(&opts.branch, "branch", "", "Branch for the session worktree (default: ao/<session-id>)")
|
|
f.StringVar(&opts.prompt, "prompt", "", "Initial prompt for the agent")
|
|
f.StringVar(&opts.issue, "issue", "", "Issue id to associate with the session")
|
|
f.StringVar(&opts.rules, "rules", "", "Agent rules appended to the prompt")
|
|
return cmd
|
|
}
|