zesdex/src/controller/command.rs

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

#[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,
    Unknown(String),
}

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,
                "agents" | "a" => ModeKind::Agents,
                "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::LessonList,
        "/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,
        _ => Command::Unknown(cmd.to_string()),
    }
}