//! Supervisor subcommand - contract orchestration commands.
use clap::Args;
use uuid::Uuid;
/// Common arguments for supervisor commands.
#[derive(Args, Debug, Clone)]
pub struct SupervisorArgs {
/// API URL
#[arg(long, env = "MAKIMA_API_URL", default_value = "https://api.makima.jp")]
pub api_url: String,
/// API key for authentication
#[arg(long, env = "MAKIMA_API_KEY")]
pub api_key: String,
/// Current task ID (optional) - the supervisor's own task ID
#[arg(long, env = "MAKIMA_TASK_ID")]
pub self_task_id: Option<Uuid>,
/// Contract ID
#[arg(long, env = "MAKIMA_CONTRACT_ID")]
pub contract_id: Uuid,
}
/// Arguments for spawn command.
#[derive(Args, Debug)]
pub struct SpawnArgs {
#[command(flatten)]
pub common: SupervisorArgs,
/// Name of the task
#[arg(index = 1)]
pub name: String,
/// Plan/description for the task
#[arg(index = 2)]
pub plan: String,
/// Parent task ID to branch from
#[arg(long)]
pub parent: Option<Uuid>,
/// Checkpoint SHA to start from
#[arg(long)]
pub checkpoint: Option<String>,
/// Repository URL (local path or remote URL). If not provided, will try to detect from current directory.
#[arg(long)]
pub repo: Option<String>,
}
/// Arguments for wait command.
#[derive(Args, Debug)]
pub struct WaitArgs {
#[command(flatten)]
pub common: SupervisorArgs,
/// Task ID to wait for
#[arg(index = 1)]
pub task_id: Uuid,
/// Timeout in seconds (total wait time)
#[arg(index = 2, default_value = "300")]
pub timeout: i32,
/// Polling interval in seconds (how often to check task status via client-side polling)
#[arg(long, default_value = "5")]
pub poll_interval: u64,
}
/// Arguments for read-file command.
#[derive(Args, Debug)]
pub struct ReadFileArgs {
#[command(flatten)]
pub common: SupervisorArgs,
/// Task ID to read from
#[arg(index = 1)]
pub task_id: Uuid,
/// File path to read
#[arg(index = 2)]
pub file_path: String,
}
/// Arguments for branch command.
#[derive(Args, Debug)]
pub struct BranchArgs {
#[command(flatten)]
pub common: SupervisorArgs,
/// Branch name to create
#[arg(index = 1)]
pub name: String,
/// Reference (task ID or SHA) to branch from
#[arg(long)]
pub from: Option<String>,
}
/// Arguments for merge command.
#[derive(Args, Debug)]
pub struct MergeArgs {
#[command(flatten)]
pub common: SupervisorArgs,
/// Task ID to merge
#[arg(index = 1)]
pub task_id: Uuid,
/// Target branch to merge into
#[arg(long)]
pub to: Option<String>,
/// Squash commits on merge
#[arg(long)]
pub squash: bool,
}
/// Arguments for pr command.
#[derive(Args, Debug)]
pub struct PrArgs {
#[command(flatten)]
pub common: SupervisorArgs,
/// Task ID to create PR for
#[arg(index = 1)]
pub task_id: Uuid,
/// PR title
#[arg(long)]
pub title: String,
/// PR body/description
#[arg(long)]
pub body: Option<String>,
/// Base branch (default: main)
#[arg(long, default_value = "main")]
pub base: String,
}
/// Arguments for diff command.
#[derive(Args, Debug)]
pub struct DiffArgs {
#[command(flatten)]
pub common: SupervisorArgs,
/// Task ID to get diff for
#[arg(index = 1)]
pub task_id: Uuid,
}
/// Arguments for checkpoint command.
#[derive(Args, Debug)]
pub struct CheckpointArgs {
#[command(flatten)]
pub common: SupervisorArgs,
/// Checkpoint message
#[arg(index = 1)]
pub message: String,
}
/// Arguments for ask command (ask user a question).
#[derive(Args, Debug)]
pub struct AskArgs {
#[command(flatten)]
pub common: SupervisorArgs,
/// The question to ask
#[arg(index = 1)]
pub question: String,
/// Optional choices (comma-separated)
#[arg(long)]
pub choices: Option<String>,
/// Context about what this relates to
#[arg(long)]
pub context: Option<String>,
/// Timeout in seconds (default: 3600 = 1 hour)
#[arg(long, default_value = "3600")]
pub timeout: i32,
/// Block indefinitely until user responds (no timeout)
#[arg(long, default_value = "false")]
pub phaseguard: bool,
/// Allow selecting multiple choices (response will be comma-separated)
#[arg(long, default_value = "false")]
pub multi_select: bool,
/// Non-blocking mode - returns immediately without waiting for response
#[arg(long, default_value = "false")]
pub non_blocking: bool,
/// Question type (general, phase_confirmation, contract_complete)
#[arg(long, default_value = "general")]
pub question_type: String,
}
/// Arguments for status command (get contract status including phase).
#[derive(Args, Debug)]
pub struct StatusArgs {
#[command(flatten)]
pub common: SupervisorArgs,
}
/// Arguments for advance-phase command.
#[derive(Args, Debug)]
pub struct AdvancePhaseArgs {
#[command(flatten)]
pub common: SupervisorArgs,
/// The phase to advance to (specify, plan, execute, review)
#[arg(index = 1)]
pub phase: String,
}
/// Arguments for mark-deliverable command.
#[derive(Args, Debug)]
pub struct MarkDeliverableArgs {
#[command(flatten)]
pub common: SupervisorArgs,
/// The deliverable ID to mark as complete (e.g., 'plan-document', 'pull-request', 'research-notes')
#[arg(index = 1)]
pub deliverable_id: String,
/// Phase the deliverable belongs to. Defaults to current contract phase if not specified.
#[arg(long)]
pub phase: Option<String>,
}
/// Arguments for task command (get individual task details).
#[derive(Args, Debug)]
pub struct GetTaskArgs {
#[command(flatten)]
pub common: SupervisorArgs,
/// Task ID to get details for
#[arg(index = 1, id = "target_task_id")]
pub target_task_id: Uuid,
}
/// Arguments for output command (get task output/claude log).
#[derive(Args, Debug)]
pub struct GetTaskOutputArgs {
#[command(flatten)]
pub common: SupervisorArgs,
/// Task ID to get output for
#[arg(index = 1, id = "target_task_id")]
pub target_task_id: Uuid,
}
// ============================================================================
// History Command Args
// ============================================================================
/// Arguments for task-history command.
#[derive(Args, Debug)]
pub struct TaskHistoryArgs {
#[command(flatten)]
pub common: SupervisorArgs,
/// Task ID to view history for
#[arg(index = 1)]
pub task_id: Uuid,
/// Include tool calls in output
#[arg(long, default_value = "true")]
pub tool_calls: bool,
/// Maximum messages to return
#[arg(long)]
pub limit: Option<i32>,
/// Output format (table, json, chat)
#[arg(long, default_value = "chat")]
pub format: String,
}
/// Arguments for task-checkpoints command (with optional diff).
#[derive(Args, Debug)]
pub struct TaskCheckpointsArgs {
#[command(flatten)]
pub common: SupervisorArgs,
/// Task ID to list checkpoints for
#[arg(index = 1)]
pub task_id: Uuid,
/// Include diff summary
#[arg(long)]
pub with_diff: bool,
}
// ============================================================================
// Resume Command Args
// ============================================================================
/// Arguments for resume command.
#[derive(Args, Debug)]
pub struct ResumeArgs {
#[command(flatten)]
pub common: SupervisorArgs,
/// Resume mode: continue, restart_phase, from_checkpoint
#[arg(long, default_value = "continue")]
pub mode: String,
/// Checkpoint ID (required for from_checkpoint mode)
#[arg(long)]
pub checkpoint: Option<Uuid>,
/// Additional context to inject
#[arg(long)]
pub context: Option<String>,
}
/// Arguments for task-resume-from command.
#[derive(Args, Debug)]
pub struct TaskResumeFromArgs {
#[command(flatten)]
pub common: SupervisorArgs,
/// Source task ID
#[arg(index = 1)]
pub task_id: Uuid,
/// Checkpoint number to resume from
#[arg(long)]
pub checkpoint: i32,
/// Plan for the new task
#[arg(long)]
pub plan: String,
/// Name for the new task
#[arg(long)]
pub name: Option<String>,
}
// ============================================================================
// Rewind Command Args
// ============================================================================
/// Arguments for task-rewind command.
#[derive(Args, Debug)]
pub struct TaskRewindArgs {
#[command(flatten)]
pub common: SupervisorArgs,
/// Task ID to rewind
#[arg(index = 1)]
pub task_id: Uuid,
/// Checkpoint number to rewind to
#[arg(long)]
pub checkpoint: i32,
/// Preserve mode: discard, create_branch, stash
#[arg(long, default_value = "create_branch")]
pub preserve: String,
/// Branch name (for create_branch mode)
#[arg(long)]
pub branch_name: Option<String>,
}
/// Arguments for task-fork command.
#[derive(Args, Debug)]
pub struct TaskForkArgs {
#[command(flatten)]
pub common: SupervisorArgs,
/// Source task ID
#[arg(index = 1)]
pub task_id: Uuid,
/// Checkpoint number to fork from
#[arg(long)]
pub checkpoint: i32,
/// Name for the new task
#[arg(long)]
pub name: String,
/// Plan for the new task
#[arg(long)]
pub plan: String,
/// Include conversation history
#[arg(long, default_value = "true")]
pub include_conversation: bool,
}
/// Arguments for rewind-conversation command.
#[derive(Args, Debug)]
pub struct ConversationRewindArgs {
#[command(flatten)]
pub common: SupervisorArgs,
/// Number of messages to rewind
#[arg(long)]
pub by_messages: Option<i32>,
/// Message ID to rewind to
#[arg(long)]
pub to_message: Option<String>,
/// Also rewind code to matching checkpoint
#[arg(long)]
pub rewind_code: bool,
}
/// Arguments for complete command (mark contract as complete).
#[derive(Args, Debug)]
pub struct CompleteArgs {
#[command(flatten)]
pub common: SupervisorArgs,
}
// ============================================================================
// Resume Contract Command Args
// ============================================================================
/// Arguments for resume-contract command (reactivate a completed contract).
#[derive(Args, Debug)]
pub struct ResumeContractArgs {
/// API URL
#[arg(long, env = "MAKIMA_API_URL", default_value = "https://api.makima.jp")]
pub api_url: String,
/// API key for authentication
#[arg(long, env = "MAKIMA_API_KEY")]
pub api_key: String,
/// Contract ID to resume
#[arg(index = 1)]
pub contract_id: Uuid,
}
