zesdex/src/controller/command.rs

//! Slash-command parser that maps TUI `/foo` input lines into `Command`
//! variants for the action dispatch system.

use crate::app::mode::ModeKind;

/// A parsed slash command from the TUI input buffer.
#[derive(Debug, Clone, PartialEq)]
pub enum Command {
    Help,
    Quit,
    LessonCreate(String),
    LessonExport(String),
    LessonImport(String),
    LessonAccept(String),
    LessonReject(String),
    LessonList,
    Mode(ModeKind),
    Clear,
    ClearConfirm,
    Login { provider: String },
    Edit(String),
    McpAdd {
        name: String,
        command: String,
    },
    ModelList,
    Compact,
    WorkflowOpen,
    WorkflowRun {
        script: String,
    },
    Unknown(String),
}

/// Parse a slash-prefixed input line into a `Command` value.
///
/// Flow: trim -> check for leading `/` -> split on space (max 3 parts) ->
/// match the first token against known commands -> extract arguments from
/// the remaining parts.
///
/// Why: early return `Unknown` for non-slash lines so the caller can treat
/// them as regular chat input.
pub fn parse_command(text: &str) -> Command {
    let text = text.trim();
    if !text.starts_with('/') {
        return Command::Unknown(text.to_string());
    }
    let parts: Vec<&str> = text.splitn(3, ' ').collect();
    let cmd = parts[0];
    let arg1 = parts.get(1).copied().unwrap_or("");
    let arg2 = parts.get(2).copied().unwrap_or("");
    match cmd {
        "/help" => Command::Help,
        "/quit" => Command::Quit,
        "/clear" if arg1.is_empty() => Command::ClearConfirm,
        "/clear" => Command::Clear,
        "/mode" if arg1.is_empty() => Command::Unknown("/mode requires a subcommand: chat, bash, help, settings".to_string()),
        "/mode" => {
            let mode = match arg1 {
                "chat" | "c" => ModeKind::Chat,
                "bash" | "b" => ModeKind::Bash,
                "workflow" | "w" => ModeKind::Workflow,
                "help" | "h" => ModeKind::Help,
                "settings" | "s" => ModeKind::Settings,
                _ => return Command::Unknown(format!("unknown mode: {}", arg1)),
            };
            Command::Mode(mode)
        }
        "/lesson" if arg1 == "export" && !arg2.is_empty() => Command::LessonExport(arg2.to_string()),
        "/lesson" if arg1 == "import" && !arg2.is_empty() => Command::LessonImport(arg2.to_string()),
        "/lesson" if arg1 == "list" || arg1 == "ls" => Command::LessonList,
        "/lesson" if arg1 == "accept" && !arg2.is_empty() => Command::LessonAccept(arg2.to_string()),
        "/lesson" if arg1 == "reject" && !arg2.is_empty() => Command::LessonReject(arg2.to_string()),
        "/lesson" if !arg1.is_empty() => Command::LessonCreate(arg1.to_string()),
        "/lesson" => Command::LessonList,
        "/login" if arg1.is_empty() => Command::Login { provider: String::new() },
        "/login" if !arg1.is_empty() => Command::Login { provider: arg1.to_string() },
        "/edit" if !arg1.is_empty() => Command::Edit(arg1.to_string()),
        "/edit" => Command::Edit(".".to_string()),
        "/mcp" if arg1.is_empty() => {
            Command::Mode(ModeKind::Mcp)
        }
        "/mcp" if arg1 == "add" && !arg2.is_empty() => {
            let rest = arg2.trim();
            if let Some(space) = rest.find(' ') {
                let name = rest[..space].to_string();
                let command = rest[space + 1..].trim().to_string();
                Command::McpAdd { name, command }
            } else {
                Command::McpAdd { name: rest.to_string(), command: String::new() }
            }
        }
        "/model" => Command::ModelList,
        "/compact" => Command::Compact,
        "/workflow" if arg1.is_empty() => Command::WorkflowOpen,
        "/workflow" if arg1 == "run" && !arg2.is_empty() => Command::WorkflowRun {
            script: arg2.to_string(),
        },
        "/workflow" if arg1 == "run" => Command::WorkflowOpen,
        "/workflow" => Command::WorkflowRun {
            script: arg1.to_string(),
        },
        _ => Command::Unknown(cmd.to_string()),
    }
}