orchestrator

type AggregatedTaskContext struct {
	TaskSummaries  map[string]string // taskID -> summary
	AllIssues      []string          // All issues from all tasks
	AllSuggestions []string          // All suggestions from all tasks
	Dependencies   []string          // Deduplicated list of new dependencies
	Notes          []string          // Implementation notes from all tasks
}

type ConflictError struct {
	TaskID       string
	Branch       string
	Files        []string
	WorktreePath string
	Underlying   error
}

type ConflictResolution struct {
	File       string `json:"file"`       // File that had the conflict
	Resolution string `json:"resolution"` // Description of how it was resolved
}

type ConsolidationCompletionFile struct {
	Status           string                   `json:"status"` // "complete", "partial", "failed"
	Mode             string                   `json:"mode"`   // "stacked" or "single"
	GroupResults     []GroupConsolidationInfo `json:"group_results"`
	PRsCreated       []PRInfo                 `json:"prs_created"`
	SynthesisContext *SynthesisCompletionFile `json:"synthesis_context,omitempty"`
	TotalCommits     int                      `json:"total_commits"`
	FilesChanged     []string                 `json:"files_changed"`
}

type ConsolidationConfig struct {
	Mode           ConsolidationMode
	BranchPrefix   string
	CreateDraftPRs bool
	PRLabels       []string
}

type ConsolidationEvent struct {
	Type      ConsolidationEventType `json:"type"`
	GroupIdx  int                    `json:"group_idx,omitempty"`
	TaskID    string                 `json:"task_id,omitempty"`
	Message   string                 `json:"message,omitempty"`
	Timestamp time.Time              `json:"timestamp"`
}

type ConsolidationEventType string

type ConsolidationMode string

type ConsolidationPhase string

type ConsolidationState struct {
	Phase            ConsolidationPhase `json:"phase"`
	CurrentGroup     int                `json:"current_group"`
	TotalGroups      int                `json:"total_groups"`
	CurrentTask      string             `json:"current_task,omitempty"`
	GroupBranches    []string           `json:"group_branches"`
	PRUrls           []string           `json:"pr_urls"`
	ConflictFiles    []string           `json:"conflict_files,omitempty"`
	ConflictTaskID   string             `json:"conflict_task_id,omitempty"`
	ConflictWorktree string             `json:"conflict_worktree,omitempty"`
	Error            string             `json:"error,omitempty"`
	StartedAt        *time.Time         `json:"started_at,omitempty"`
	CompletedAt      *time.Time         `json:"completed_at,omitempty"`
}

type Consolidator struct {
	// contains filtered or unexported fields
}

type Coordinator struct {
	// contains filtered or unexported fields
}

type CoordinatorCallbacks struct {
	// OnPhaseChange is called when the ultra-plan phase changes
	OnPhaseChange func(phase UltraPlanPhase)

	// OnTaskStart is called when a task begins execution
	OnTaskStart func(taskID, instanceID string)

	// OnTaskComplete is called when a task completes successfully
	OnTaskComplete func(taskID string)

	// OnTaskFailed is called when a task fails
	OnTaskFailed func(taskID, reason string)

	// OnGroupComplete is called when an execution group completes
	OnGroupComplete func(groupIndex int)

	// OnPlanReady is called when the plan is ready (after planning phase)
	OnPlanReady func(plan *PlanSpec)

	// OnProgress is called periodically with progress updates
	OnProgress func(completed, total int, phase UltraPlanPhase)

	// OnComplete is called when the entire ultra-plan completes
	OnComplete func(success bool, summary string)
}

type CoordinatorEvent struct {
	Type       CoordinatorEventType `json:"type"`
	TaskID     string               `json:"task_id,omitempty"`
	InstanceID string               `json:"instance_id,omitempty"`
	Message    string               `json:"message,omitempty"`
	Timestamp  time.Time            `json:"timestamp"`
}

type CoordinatorEventType string

type FlexibleString string

type GroupConsolidationCompletionFile struct {
	GroupIndex         int                    `json:"group_index"`
	Status             string                 `json:"status"` // "complete", "failed"
	BranchName         string                 `json:"branch_name"`
	TasksConsolidated  []string               `json:"tasks_consolidated"`
	ConflictsResolved  []ConflictResolution   `json:"conflicts_resolved,omitempty"`
	Verification       VerificationResult     `json:"verification"`
	AggregatedContext  *AggregatedTaskContext `json:"aggregated_context,omitempty"`
	Notes              string                 `json:"notes,omitempty"`                 // Consolidator's observations
	IssuesForNextGroup []string               `json:"issues_for_next_group,omitempty"` // Warnings/concerns to pass forward
}

type GroupConsolidationInfo struct {
	GroupIndex    int      `json:"group_index"`
	BranchName    string   `json:"branch_name"`
	TasksIncluded []string `json:"tasks_included"`
	CommitCount   int      `json:"commit_count"`
	Success       bool     `json:"success"`
}

type GroupConsolidationResult struct {
	GroupIndex   int      `json:"group_index"`
	TaskIDs      []string `json:"task_ids"`
	BranchName   string   `json:"branch_name"`
	CommitCount  int      `json:"commit_count"`
	FilesChanged []string `json:"files_changed"`
	PRUrl        string   `json:"pr_url,omitempty"`
	Success      bool     `json:"success"`
	Error        string   `json:"error,omitempty"`
}

type GroupDecisionState struct {
	GroupIndex       int      `json:"group_index"`
	SucceededTasks   []string `json:"succeeded_tasks"`   // Tasks with verified commits
	FailedTasks      []string `json:"failed_tasks"`      // Tasks that failed or produced no commits
	AwaitingDecision bool     `json:"awaiting_decision"` // True when paused for user input
}

type Instance struct {
	ID            string         `json:"id"`
	WorktreePath  string         `json:"worktree_path"`
	Branch        string         `json:"branch"`
	Task          string         `json:"task"`
	Status        InstanceStatus `json:"status"`
	PID           int            `json:"pid,omitempty"`
	FilesModified []string       `json:"files_modified,omitempty"`
	Created       time.Time      `json:"created"`
	TmuxSession   string         `json:"tmux_session,omitempty"` // Tmux session name for recovery
	Metrics       *Metrics       `json:"metrics,omitempty"`
	Output        []byte         `json:"-"` // Not persisted, runtime only
}

type InstanceStatus string

type Metrics struct {
	InputTokens  int64      `json:"input_tokens"`
	OutputTokens int64      `json:"output_tokens"`
	CacheRead    int64      `json:"cache_read,omitempty"`
	CacheWrite   int64      `json:"cache_write,omitempty"`
	Cost         float64    `json:"cost"`
	APICalls     int        `json:"api_calls"`
	StartTime    *time.Time `json:"start_time,omitempty"`
	EndTime      *time.Time `json:"end_time,omitempty"`
}

type Orchestrator struct {
	// contains filtered or unexported fields
}

type PRContent struct {
	Title string
	Body  string
}

type PRInfo struct {
	URL        string `json:"url"`
	Title      string `json:"title"`
	GroupIndex int    `json:"group_index"`
}

type PlanSpec struct {
	ID              string              `json:"id"`
	Objective       string              `json:"objective"` // Original user request
	Summary         string              `json:"summary"`   // Executive summary of the plan
	Tasks           []PlannedTask       `json:"tasks"`
	DependencyGraph map[string][]string `json:"dependency_graph"` // task_id -> depends_on[]
	ExecutionOrder  [][]string          `json:"execution_order"`  // Groups of parallelizable tasks
	Insights        []string            `json:"insights"`         // Key findings from exploration
	Constraints     []string            `json:"constraints"`      // Identified constraints/risks
	CreatedAt       time.Time           `json:"created_at"`
}

type PlannedTask struct {
	ID            string         `json:"id"`
	Title         string         `json:"title"`
	Description   string         `json:"description"`     // Detailed task prompt for child session
	Files         []string       `json:"files,omitempty"` // Expected files to be modified
	DependsOn     []string       `json:"depends_on"`      // Task IDs this depends on
	Priority      int            `json:"priority"`        // Execution priority (lower = earlier)
	EstComplexity TaskComplexity `json:"est_complexity"`
}

type RevisionCompletionFile struct {
	TaskID          string   `json:"task_id"`
	RevisionRound   int      `json:"revision_round"`
	IssuesAddressed []string `json:"issues_addressed"` // Issue descriptions that were fixed
	Summary         string   `json:"summary"`          // What was changed
	FilesModified   []string `json:"files_modified"`
	RemainingIssues []string `json:"remaining_issues"` // Issues that couldn't be fixed
}

type RevisionIssue struct {
	TaskID      string   `json:"task_id"`              // Task ID that needs revision (empty for cross-cutting issues)
	Description string   `json:"description"`          // Description of the issue
	Files       []string `json:"files,omitempty"`      // Files affected by the issue
	Severity    string   `json:"severity,omitempty"`   // "critical", "major", "minor"
	Suggestion  string   `json:"suggestion,omitempty"` // Suggested fix
}

type RevisionState struct {
	Issues          []RevisionIssue   `json:"issues"`                     // Issues identified during synthesis
	RevisionRound   int               `json:"revision_round"`             // Current revision iteration (starts at 1)
	MaxRevisions    int               `json:"max_revisions"`              // Maximum allowed revision rounds
	TasksToRevise   []string          `json:"tasks_to_revise,omitempty"`  // Task IDs that need revision
	RevisedTasks    []string          `json:"revised_tasks,omitempty"`    // Task IDs that have been revised
	RevisionPrompts map[string]string `json:"revision_prompts,omitempty"` // Task ID -> revision prompt
	StartedAt       *time.Time        `json:"started_at,omitempty"`
	CompletedAt     *time.Time        `json:"completed_at,omitempty"`
}

type Session struct {
	ID        string      `json:"id"`
	Name      string      `json:"name"`
	BaseRepo  string      `json:"base_repo"`
	Created   time.Time   `json:"created"`
	Instances []*Instance `json:"instances"`

	// UltraPlan holds the ultra-plan session state (nil for regular sessions)
	UltraPlan *UltraPlanSession `json:"ultra_plan,omitempty"`
}

type SessionMetrics struct {
	TotalInputTokens  int64
	TotalOutputTokens int64
	TotalCacheRead    int64
	TotalCacheWrite   int64
	TotalCost         float64
	TotalAPICalls     int
	TotalDuration     time.Duration
	InstanceCount     int
	ActiveCount       int
}

type SynthesisCompletionFile struct {
	Status           string          `json:"status"`            // "complete", "needs_revision"
	RevisionRound    int             `json:"revision_round"`    // Current round (0 for first synthesis)
	IssuesFound      []RevisionIssue `json:"issues_found"`      // All issues identified
	TasksAffected    []string        `json:"tasks_affected"`    // Task IDs needing revision
	IntegrationNotes string          `json:"integration_notes"` // Free-form observations about integration
	Recommendations  []string        `json:"recommendations"`   // Suggestions for consolidation phase
}

type TaskCompletionFile struct {
	TaskID        string   `json:"task_id"`
	Status        string   `json:"status"` // "complete", "blocked", or "failed"
	Summary       string   `json:"summary"`
	FilesModified []string `json:"files_modified"`
	// Rich context for consolidation
	Notes        FlexibleString `json:"notes,omitempty"`        // Free-form implementation notes (accepts string or array)
	Issues       []string       `json:"issues,omitempty"`       // Blocking issues or concerns found
	Suggestions  []string       `json:"suggestions,omitempty"`  // Integration suggestions for other tasks
	Dependencies []string       `json:"dependencies,omitempty"` // Runtime dependencies added
}

type TaskComplexity string

type TaskRetryState struct {
	TaskID       string `json:"task_id"`
	RetryCount   int    `json:"retry_count"`
	MaxRetries   int    `json:"max_retries"`
	LastError    string `json:"last_error,omitempty"`
	CommitCounts []int  `json:"commit_counts,omitempty"` // Commits per attempt (for debugging)
}

type TaskWorktreeInfo struct {
	TaskID       string `json:"task_id"`
	TaskTitle    string `json:"task_title"`
	WorktreePath string `json:"worktree_path"`
	Branch       string `json:"branch"`
}

type UltraPlanConfig struct {
	MaxParallel int  `json:"max_parallel"` // Maximum concurrent child sessions
	DryRun      bool `json:"dry_run"`      // Run planning only, don't execute
	NoSynthesis bool `json:"no_synthesis"` // Skip synthesis phase after execution
	AutoApprove bool `json:"auto_approve"` // Auto-approve spawned tasks without confirmation

	// Consolidation settings
	ConsolidationMode ConsolidationMode `json:"consolidation_mode,omitempty"` // "stacked" or "single"
	CreateDraftPRs    bool              `json:"create_draft_prs"`             // Create PRs as drafts
	PRLabels          []string          `json:"pr_labels,omitempty"`          // Labels to add to PRs
	BranchPrefix      string            `json:"branch_prefix,omitempty"`      // Branch prefix for consolidated branches

	// Task verification settings
	MaxTaskRetries         int  `json:"max_task_retries,omitempty"` // Max retry attempts for tasks with no commits (default: 3)
	RequireVerifiedCommits bool `json:"require_verified_commits"`   // If true, tasks must produce commits to be marked successful (default: true)
}

type UltraPlanManager struct {
	// contains filtered or unexported fields
}

type UltraPlanPhase string

type UltraPlanSession struct {
	ID              string            `json:"id"`
	Objective       string            `json:"objective"`
	Plan            *PlanSpec         `json:"plan,omitempty"`
	Phase           UltraPlanPhase    `json:"phase"`
	Config          UltraPlanConfig   `json:"config"`
	CoordinatorID   string            `json:"coordinator_id,omitempty"`   // Instance ID of the planning coordinator
	SynthesisID     string            `json:"synthesis_id,omitempty"`     // Instance ID of the synthesis reviewer
	RevisionID      string            `json:"revision_id,omitempty"`      // Instance ID of the current revision coordinator
	ConsolidationID string            `json:"consolidation_id,omitempty"` // Instance ID of the consolidation agent
	TaskToInstance  map[string]string `json:"task_to_instance"`           // PlannedTask.ID -> Instance.ID
	CompletedTasks  []string          `json:"completed_tasks"`
	FailedTasks     []string          `json:"failed_tasks"`
	CurrentGroup    int               `json:"current_group"` // Index into ExecutionOrder
	Created         time.Time         `json:"created"`
	StartedAt       *time.Time        `json:"started_at,omitempty"`
	CompletedAt     *time.Time        `json:"completed_at,omitempty"`
	Error           string            `json:"error,omitempty"` // Error message if failed

	// Revision state (persisted for recovery and display)
	Revision *RevisionState `json:"revision,omitempty"`

	// Synthesis completion context (populated from sentinel file)
	SynthesisCompletion *SynthesisCompletionFile `json:"synthesis_completion,omitempty"`

	// Task worktree information for consolidation context
	TaskWorktrees []TaskWorktreeInfo `json:"task_worktrees,omitempty"`

	// Per-group consolidated branches: index -> branch name
	// After each group completes, parallel task branches are merged into one consolidated branch
	// The next group's tasks use this consolidated branch as their base
	GroupConsolidatedBranches []string `json:"group_consolidated_branches,omitempty"`

	// Per-group consolidator instance IDs: index -> instance ID
	// Each group has a dedicated Claude session that consolidates its task branches
	GroupConsolidatorIDs []string `json:"group_consolidator_ids,omitempty"`

	// Per-group consolidation contexts: index -> completion file data
	// Stores the context from each group's consolidator to pass to the next group
	GroupConsolidationContexts []*GroupConsolidationCompletionFile `json:"group_consolidation_contexts,omitempty"`

	// Consolidation results (persisted for recovery and display)
	Consolidation *ConsolidationState `json:"consolidation,omitempty"`
	PRUrls        []string            `json:"pr_urls,omitempty"`

	// Task retry tracking: task ID -> retry state
	TaskRetries map[string]*TaskRetryState `json:"task_retries,omitempty"`

	// Group decision state (set when group has mix of success/failure)
	GroupDecision *GroupDecisionState `json:"group_decision,omitempty"`

	// Verified commit counts per task (populated after task completion)
	TaskCommitCounts map[string]int `json:"task_commit_counts,omitempty"`
}

type VerificationResult struct {
	ProjectType    string             `json:"project_type,omitempty"` // Detected: "go", "node", "ios", "python", etc.
	CommandsRun    []VerificationStep `json:"commands_run"`
	OverallSuccess bool               `json:"overall_success"`
	Summary        string             `json:"summary,omitempty"` // Brief summary of verification outcome
}

type VerificationStep struct {
	Name    string `json:"name"`    // e.g., "build", "lint", "test"
	Command string `json:"command"` // Actual command run
	Success bool   `json:"success"`
	Output  string `json:"output,omitempty"` // Truncated output on failure
}