cryochamber 0.1.2

// src/bin/cryo.rs
use anyhow::{Context, Result};
use clap::{Parser, Subcommand};
use std::path::Path;

use cryochamber::config;
use cryochamber::message;
use cryochamber::protocol;
use cryochamber::state::{self, CryoState};

#[derive(Parser)]
#[command(name = "cryo", about = "Long-term AI agent task scheduler")]
struct Cli {
    #[command(subcommand)]
    command: Commands,
}

#[derive(Subcommand)]
enum Commands {
    /// Initialize a working directory with protocol file and template plan
    Init {
        /// Agent command to target (determines CLAUDE.md vs AGENTS.md)
        #[arg(long, default_value = "opencode")]
        agent: String,
    },
    /// Begin a new plan: initialize and run the first task
    Start {
        /// Agent command to use (overrides cryo.toml)
        #[arg(long)]
        agent: Option<String>,
        /// Max retry attempts on agent spawn failure (overrides cryo.toml)
        #[arg(long)]
        max_retries: Option<u32>,
        /// Maximum session duration in seconds (overrides cryo.toml)
        #[arg(long)]
        max_session_duration: Option<u64>,
    },
    /// Show current status: next wake time, last result
    Status,
    /// List all running cryo daemon processes on this machine
    Ps {
        /// Kill all listed daemons
        #[arg(long)]
        kill_all: bool,
    },
    /// Kill the running daemon and restart it
    Restart,
    /// Stop the daemon and remove state
    Cancel,
    /// Stop the daemon and remove all runtime files (confirms first)
    Clean {
        /// Skip confirmation prompt
        #[arg(long)]
        force: bool,
    },
    /// Print the session log
    Log,
    /// Watch the session log in real-time
    Watch {
        /// Show full log from the beginning (default: start from current position)
        #[arg(long)]
        all: bool,
        /// Which log to follow: "cryo" for structured events, "agent" for raw agent output
        #[arg(long, default_value = "cryo")]
        viewpoint: String,
    },
    /// Send a message to the agent's inbox
    Send {
        /// Message body
        body: String,
        /// Sender name (default: "human")
        #[arg(long, default_value = "human")]
        from: String,
        /// Message subject (default: derived from body)
        #[arg(long)]
        subject: Option<String>,
        /// Wake the agent immediately after sending
        #[arg(long)]
        wake: bool,
    },
    /// Read messages from the agent's outbox
    Receive,
    /// Send a wake message to the daemon's inbox
    Wake {
        /// Message to include in the agent's prompt
        message: Option<String>,
    },
    /// Execute a fallback action (internal — used by timers)
    #[command(hide = true)]
    FallbackExec {
        action: String,
        target: String,
        message: String,
    },
    /// Open a web chat UI for messaging and waking the agent
    Web {
        /// Host to listen on (overrides cryo.toml web_host)
        #[arg(long)]
        host: Option<String>,
        /// Port to listen on (overrides cryo.toml web_port)
        #[arg(long)]
        port: Option<u16>,
        /// Run in foreground instead of installing a service
        #[arg(long, conflicts_with = "stop")]
        foreground: bool,
        /// Stop the web service
        #[arg(long)]
        stop: bool,
    },
    /// Run the persistent daemon (internal — use `cryo start` instead)
    #[command(hide = true)]
    Daemon,
    /// Internal: run the web server (called by OS service)
    #[command(hide = true)]
    WebDaemon {
        #[arg(long)]
        host: String,
        #[arg(long)]
        port: u16,
    },
}

fn main() -> Result<()> {
    let cli = Cli::parse();

    match cli.command {
        Commands::Init { agent } => cmd_init(&agent),
        Commands::Start {
            agent,
            max_retries,
            max_session_duration,
        } => cmd_start(agent, max_retries, max_session_duration),
        Commands::Status => cmd_status(),
        Commands::Ps { kill_all } => cmd_ps(kill_all),
        Commands::Restart => cmd_restart(),
        Commands::Cancel => cmd_cancel(),
        Commands::Clean { force } => cmd_clean(force),
        Commands::Log => cmd_log(),
        Commands::Watch { all, viewpoint } => cmd_watch(all, &viewpoint),
        Commands::Send {
            body,
            from,
            subject,
            wake,
        } => cmd_send(&body, &from, subject.as_deref(), wake),
        Commands::Wake { message } => cmd_wake(message.as_deref()),
        Commands::Web {
            host,
            port,
            foreground,
            stop,
        } => cmd_web(host, port, foreground, stop),
        Commands::Daemon => cmd_daemon(),
        Commands::WebDaemon { host, port } => cmd_web_daemon(host, port),
        Commands::Receive => cmd_receive(),
        Commands::FallbackExec {
            action,
            target,
            message,
        } => {
            let dir = cryochamber::work_dir()?;
            let fb = cryochamber::fallback::FallbackAction {
                action,
                target,
                message,
            };
            let config = cryochamber::config::load_config(&cryochamber::config::config_path(&dir))?
                .unwrap_or_default();
            fb.execute(&dir, &config.fallback_alert)
        }
    }
}

/// Check that this directory is a valid cryo project (cryo.toml must exist).
fn require_valid_project(dir: &Path) -> Result<()> {
    if !config::config_path(dir).exists() {
        anyhow::bail!("No cryochamber project in this directory. Run `cryo init` first.");
    }
    Ok(())
}

/// Check that a live daemon is running in the current directory.
fn require_live_daemon(dir: &Path) -> Result<CryoState> {
    require_valid_project(dir)?;
    let cryo_state = state::load_state(&state::state_path(dir))?
        .context("No daemon state found. Run `cryo start` first.")?;
    if !state::is_locked(&cryo_state) {
        anyhow::bail!(
            "No live daemon in this directory (stale state from a previous run). \
             Run `cryo start` to start a new one, or `cryo cancel` to clean up stale state."
        );
    }
    Ok(cryo_state)
}

fn cmd_init(agent_cmd: &str) -> Result<()> {
    let dir = cryochamber::work_dir()?;

    // Write cryo.toml first (project config)
    if protocol::write_config_file(&dir, agent_cmd)? {
        println!("  cryo.toml (created)");
    } else {
        println!("  cryo.toml (exists, kept)");
    }

    let filename = protocol::protocol_filename(agent_cmd);
    if protocol::write_protocol_file(&dir, filename)? {
        println!("  {filename} (created)");
    } else {
        println!("  {filename} (exists, kept)");
    }

    if protocol::write_template_plan(&dir)? {
        println!("  plan.md (created)");
    } else {
        println!("  plan.md (exists, kept)");
    }

    if protocol::write_readme(&dir)? {
        println!("  README.md (created)");
    } else {
        println!("  README.md (exists, kept)");
    }

    message::ensure_dirs(&dir)?;

    println!("\nCryochamber initialized. Next steps:");
    println!("  1. Edit plan.md with your task plan");
    println!("  2. Run: cryo start");

    Ok(())
}

/// Check that the agent command is supported and the binary exists on PATH.
fn validate_agent_command(agent_cmd: &str) -> Result<()> {
    let program = cryochamber::agent::agent_program(agent_cmd)?;
    let status = std::process::Command::new("which")
        .arg(&program)
        .stdout(std::process::Stdio::null())
        .stderr(std::process::Stdio::null())
        .status();
    match status {
        Ok(s) if s.success() => Ok(()),
        _ => anyhow::bail!(
            "Agent command '{}' not found. Verify it is installed and on your PATH.",
            program
        ),
    }
}

fn cmd_start(
    agent_override: Option<String>,
    max_retries_override: Option<u32>,
    max_session_duration_override: Option<u64>,
) -> Result<()> {
    let dir = cryochamber::work_dir()?;

    // Require init: protocol file or cryo.toml must exist
    require_valid_project(&dir)?;

    // Require plan.md in the working directory
    if !dir.join("plan.md").exists() {
        anyhow::bail!("No plan.md found in the working directory. Create one or run `cryo init`.");
    }

    // Guard: refuse to start if an instance is already active
    if let Some(existing) = state::load_state(&state::state_path(&dir))? {
        if state::is_locked(&existing) {
            anyhow::bail!(
                "A cryochamber session is already running (PID: {:?}). Use `cryo cancel` to stop it first.",
                existing.pid
            );
        }
    }

    // Load config from cryo.toml (fall back to defaults for legacy projects)
    let cfg = config::load_config(&config::config_path(&dir))?.unwrap_or_default();

    // Resolve effective values: CLI override > cryo.toml > hardcoded default
    let effective_agent = agent_override.as_deref().unwrap_or(&cfg.agent);

    // Validate agent command using effective agent value
    validate_agent_command(effective_agent)?;

    // Ensure message dirs exist (needed for inbox watching)
    message::ensure_dirs(&dir)?;

    // Build slim CryoState with override fields only when CLI flags were explicitly provided
    let cryo_state = CryoState {
        session_number: 0, // daemon will increment to 1
        pid: None,         // no PID lock — daemon will set its own
        retry_count: 0,
        agent_override,
        max_retries_override,
        max_session_duration_override,
        next_wake: None,
        last_report_time: None,
        provider_index: None,
    };
    state::save_state(&state::state_path(&dir), &cryo_state)?;

    // CRYO_NO_SERVICE=1 disables OS service installation (useful for tests / debugging)
    if std::env::var("CRYO_NO_SERVICE").is_ok() {
        cryochamber::process::spawn_daemon(&dir)?;
        println!("Cryochamber started (background process).");
    } else {
        let exe = std::env::current_exe().context("Failed to resolve cryo executable path")?;
        let log_path = cryochamber::log::log_path(&dir);
        cryochamber::service::install("daemon", &dir, &exe, &["daemon"], &log_path, false)?;
        println!("Cryochamber started (service installed, survives reboot).");
    }

    // Wait for the daemon to write its PID before returning
    let deadline = std::time::Instant::now() + std::time::Duration::from_secs(10);
    loop {
        std::thread::sleep(std::time::Duration::from_millis(100));
        if let Some(st) = state::load_state(&state::state_path(&dir))? {
            if state::is_locked(&st) {
                break;
            }
        }
        if std::time::Instant::now() > deadline {
            anyhow::bail!("Daemon did not start within 10 seconds. Check cryo.log for errors.");
        }
    }

    println!("Use `cryo watch` or `cryo web` to follow progress.");
    println!("Use `cryo status` to check state.");

    Ok(())
}

fn cmd_daemon() -> Result<()> {
    let dir = cryochamber::work_dir()?;
    let daemon = cryochamber::daemon::Daemon::new(dir);
    daemon.run()
}

fn cmd_web(host: Option<String>, port: Option<u16>, foreground: bool, stop: bool) -> Result<()> {
    let dir = cryochamber::work_dir()?;
    require_valid_project(&dir)?;

    let cfg = config::load_config(&config::config_path(&dir))?.unwrap_or_default();
    let host = host.unwrap_or(cfg.web_host);
    let port = port.unwrap_or(cfg.web_port);

    if stop {
        if cryochamber::service::uninstall("web", &dir)? {
            println!("Web service stopped and removed.");
        } else {
            println!("No web service installed for this directory.");
        }
        return Ok(());
    }

    if foreground {
        let rt = tokio::runtime::Runtime::new()?;
        rt.block_on(cryochamber::web::serve(dir, &host, port))
    } else {
        let exe = std::env::current_exe().context("Failed to resolve cryo executable path")?;
        let port_str = port.to_string();
        let log_path = dir.join("cryo-web.log");
        cryochamber::service::install(
            "web",
            &dir,
            &exe,
            &["web-daemon", "--host", &host, "--port", &port_str],
            &log_path,
            true,
        )?;
        println!("Web UI service installed: http://{}:{}", host, port);
        println!("Log: cryo-web.log");
        println!("Survives reboot. Stop with: cryo web --stop");
        Ok(())
    }
}

fn cmd_web_daemon(host: String, port: u16) -> Result<()> {
    let dir = cryochamber::work_dir()?;
    let rt = tokio::runtime::Runtime::new()?;
    rt.block_on(cryochamber::web::serve(dir, &host, port))
}

fn cmd_status() -> Result<()> {
    let dir = cryochamber::work_dir()?;
    require_valid_project(&dir)?;

    let cfg = config::load_config(&config::config_path(&dir))?.unwrap_or_default();

    match state::load_state(&state::state_path(&dir))? {
        None => {
            println!("No daemon has been started yet. Run `cryo start` to begin.");
            println!("\nConfig (cryo.toml):");
            println!("  Agent: {}", cfg.agent);
        }
        Some(st) => {
            // Runtime state first
            println!(
                "Daemon: {}",
                if state::is_locked(&st) {
                    "running"
                } else {
                    "stopped"
                }
            );
            println!("Session: {}", st.session_number);
            if let Some(pid) = st.pid {
                println!("PID: {pid}");
            }

            // Config
            let effective_agent = st.agent_override.as_deref().unwrap_or(&cfg.agent);
            println!("Agent: {effective_agent}");
            if st.agent_override.is_some() {
                println!("  (override; cryo.toml has \"{}\")", cfg.agent);
            }
            if !cfg.providers.is_empty() {
                let idx = st.provider_index.unwrap_or(0);
                if let Some(provider) = cfg.providers.get(idx) {
                    println!(
                        "Provider: {} ({}/{})",
                        provider.name,
                        idx + 1,
                        cfg.providers.len()
                    );
                }
            }
            let effective_timeout = st
                .max_session_duration_override
                .unwrap_or(cfg.max_session_duration);
            if effective_timeout > 0 {
                println!("Session timeout: {effective_timeout}s");
            }

            let log = cryochamber::log::log_path(&dir);
            if let Some(latest) = cryochamber::log::read_latest_session(&log)? {
                println!("\n--- Latest session ---");
                let lines: Vec<&str> = latest.lines().collect();
                let start = lines.len().saturating_sub(10);
                for line in &lines[start..] {
                    println!("{line}");
                }
            }
        }
    }

    Ok(())
}

fn cmd_restart() -> Result<()> {
    let dir = cryochamber::work_dir()?;
    let cryo_state = require_live_daemon(&dir)?;

    // Uninstall old service (systemd/launchd stop may already kill the process)
    let _ = cryochamber::service::uninstall("daemon", &dir);

    // Kill existing daemon process only if still alive after service removal
    if state::is_locked(&cryo_state) {
        if let Some(pid) = cryo_state.pid {
            cryochamber::process::terminate_pid(pid)?;
        }
    }

    // Clear PID, keep session_number and overrides
    let updated = CryoState {
        pid: None,
        ..cryo_state
    };
    state::save_state(&state::state_path(&dir), &updated)?;

    let exe = std::env::current_exe().context("Failed to resolve cryo executable path")?;
    let log_path = cryochamber::log::log_path(&dir);
    cryochamber::service::install("daemon", &dir, &exe, &["daemon"], &log_path, false)?;

    println!("Restarted (service reinstalled).");
    println!("Use `cryo watch` or `cryo web` to follow progress.");
    Ok(())
}

fn cmd_ps(kill_all: bool) -> Result<()> {
    // list() auto-cleans dead PIDs from the registry
    let entries = cryochamber::registry::list()?;

    if entries.is_empty() {
        println!("No cryo daemons running.");
        return Ok(());
    }

    for entry in &entries {
        if kill_all {
            cryochamber::process::terminate_pid(entry.pid)?;
            println!("Killed PID {:>6}  {}", entry.pid, entry.dir);
        } else {
            println!("PID {:>6}  {}", entry.pid, entry.dir);
        }
    }

    Ok(())
}

fn cmd_cancel() -> Result<()> {
    let dir = cryochamber::work_dir()?;
    require_valid_project(&dir)?;

    // Uninstall system service (launchd/systemd) if installed
    let service_removed = cryochamber::service::uninstall("daemon", &dir)?;
    if service_removed {
        println!("Service removed.");
    }

    let sp = state::state_path(&dir);
    match state::load_state(&sp)? {
        None => {
            if !service_removed {
                anyhow::bail!("Nothing to cancel. No daemon state or service found.");
            }
        }
        Some(cryo_state) => {
            // Kill daemon process if still alive
            if state::is_locked(&cryo_state) {
                if let Some(pid) = cryo_state.pid {
                    cryochamber::process::terminate_pid(pid)?;
                    println!("Killed daemon (PID {pid}).");
                }
            }
            // Always clean up state file
            std::fs::remove_file(sp)?;
            println!("Removed timer.json.");
        }
    }

    println!("Cryochamber cancelled.");
    Ok(())
}

/// Prompt the user for y/n confirmation. Returns true if confirmed.
fn confirm(prompt: &str) -> bool {
    eprint!("{prompt} [y/N] ");
    let mut input = String::new();
    if std::io::stdin().read_line(&mut input).is_err() {
        return false;
    }
    matches!(input.trim(), "y" | "Y" | "yes" | "Yes")
}

fn cmd_clean(force: bool) -> Result<()> {
    let dir = cryochamber::work_dir()?;
    require_valid_project(&dir)?;

    if !force && !confirm("Stop daemon and remove all runtime files?") {
        println!("Aborted.");
        return Ok(());
    }

    // Uninstall services (daemon + gh-sync)
    if cryochamber::service::uninstall("daemon", &dir)? {
        println!("Removed daemon service.");
    }
    if cryochamber::service::uninstall("gh-sync", &dir)? {
        println!("Removed gh-sync service.");
    }
    if cryochamber::service::uninstall("zulip-sync", &dir)? {
        println!("Removed zulip-sync service.");
    }
    if cryochamber::service::uninstall("web", &dir)? {
        println!("Removed web service.");
    }

    // Kill daemon process if still running
    let sp = state::state_path(&dir);
    if let Some(cryo_state) = state::load_state(&sp)? {
        if state::is_locked(&cryo_state) {
            if let Some(pid) = cryo_state.pid {
                cryochamber::process::terminate_pid(pid)?;
                println!("Killed daemon (PID {pid}).");
            }
        }
    }

    // Remove runtime files
    let runtime_files = [
        "timer.json",
        "cryo.log",
        "cryo-agent.log",
        "cryo-gh-sync.log",
        "gh-sync.json",
        "cryo-zulip-sync.log",
        "zulip-sync.json",
        "cryo-web.log",
    ];
    for name in &runtime_files {
        let path = dir.join(name);
        if path.exists() {
            std::fs::remove_file(&path)?;
            println!("Removed {name}");
        }
    }

    // Remove runtime directories
    let runtime_dirs = ["messages", ".cryo"];
    for name in &runtime_dirs {
        let path = dir.join(name);
        if path.exists() {
            std::fs::remove_dir_all(&path)?;
            println!("Removed {name}/");
        }
    }

    println!("Clean.");
    Ok(())
}

fn cmd_log() -> Result<()> {
    let dir = cryochamber::work_dir()?;
    let log = cryochamber::log::log_path(&dir);
    if log.exists() {
        let contents = std::fs::read_to_string(log)?;
        println!("{contents}");
    } else {
        println!("No log file found.");
    }
    Ok(())
}

fn build_inbox_message(from: &str, subject: &str, body: &str) -> message::Message {
    message::Message {
        from: from.to_string(),
        subject: subject.to_string(),
        body: body.to_string(),
        timestamp: chrono::Local::now().naive_local(),
        metadata: std::collections::BTreeMap::new(),
    }
}

/// Check if a daemon is running in the given directory.
fn is_daemon_running(dir: &std::path::Path) -> bool {
    if let Ok(Some(st)) = state::load_state(&state::state_path(dir)) {
        return state::is_locked(&st);
    }
    false
}

/// Send SIGUSR1 to the daemon to force an immediate wake.
/// Returns true if the signal was delivered successfully.
fn signal_daemon_wake(dir: &std::path::Path) -> bool {
    cryochamber::process::signal_daemon_wake(dir)
}

/// After writing an inbox message, notify the daemon and print status.
/// When watch_inbox is true, the inotify watcher handles wake — no signal needed.
/// When watch_inbox is false, send SIGUSR1.
fn notify_daemon_wake(dir: &std::path::Path) -> Result<()> {
    let watch_inbox = config::load_config(&config::config_path(dir))?
        .map(|c| c.watch_inbox)
        .unwrap_or(true);

    if !is_daemon_running(dir) {
        eprintln!("Warning: no daemon is running. Message queued for the next `cryo start`.");
    } else if watch_inbox {
        println!("Daemon will pick it up shortly.");
    } else if signal_daemon_wake(dir) {
        println!("Wake signal sent. Daemon waking now.");
    } else {
        eprintln!("Warning: failed to signal daemon. Message queued for the next session.");
    }
    Ok(())
}

fn cmd_wake(wake_message: Option<&str>) -> Result<()> {
    let dir = cryochamber::work_dir()?;
    require_valid_project(&dir)?;
    message::ensure_dirs(&dir)?;

    let body = wake_message.unwrap_or("Manual wake requested by operator.");
    let msg = build_inbox_message("operator", "Wake", body);
    message::write_message(&dir, "inbox", &msg)?;

    notify_daemon_wake(&dir)
}

fn cmd_send(body: &str, from: &str, subject: Option<&str>, wake: bool) -> Result<()> {
    let dir = cryochamber::work_dir()?;
    require_valid_project(&dir)?;
    message::ensure_dirs(&dir)?;

    let subject = subject.unwrap_or_else(|| {
        // Truncate at a char boundary to avoid panic on non-ASCII input
        let mut end = body.len().min(50);
        while end > 0 && !body.is_char_boundary(end) {
            end -= 1;
        }
        &body[..end]
    });
    let msg = build_inbox_message(from, subject, body);
    let path = message::write_message(&dir, "inbox", &msg)?;
    println!(
        "Message sent to {}",
        path.strip_prefix(&dir).unwrap_or(&path).display()
    );

    if wake {
        notify_daemon_wake(&dir)?;
    }

    Ok(())
}

fn cmd_receive() -> Result<()> {
    let dir = cryochamber::work_dir()?;
    let messages = message::read_outbox(&dir)?;

    if messages.is_empty() {
        println!("No messages in outbox.");
        return Ok(());
    }

    for (filename, msg) in &messages {
        println!("--- {} ---", filename);
        println!("From: {}", msg.from);
        println!("Subject: {}", msg.subject);
        println!("Time: {}", msg.timestamp.format("%Y-%m-%dT%H:%M:%S"));
        println!();
        println!("{}", msg.body);
        println!();
    }

    Ok(())
}

fn cmd_watch(show_all: bool, viewpoint: &str) -> Result<()> {
    use std::io::Read;

    let dir = cryochamber::work_dir()?;
    require_valid_project(&dir)?;
    let log = match viewpoint {
        "agent" => cryochamber::log::agent_log_path(&dir),
        "cryo" => cryochamber::log::log_path(&dir),
        other => anyhow::bail!("Unknown viewpoint '{other}'. Use 'cryo' or 'agent'."),
    };
    let state_file = state::state_path(&dir);

    if !log.exists() {
        println!("Waiting for first session output...");
    }

    // Start from end of file unless --all
    let mut pos = if show_all {
        0
    } else {
        log.metadata().map(|m| m.len()).unwrap_or(0)
    };

    let mut no_state_ticks: u32 = 0;

    loop {
        // Read new content from the log file
        if log.exists() {
            let file_len = log.metadata().map(|m| m.len()).unwrap_or(0);
            if file_len > pos {
                let mut f = std::fs::File::open(&log)?;
                std::io::Seek::seek(&mut f, std::io::SeekFrom::Start(pos))?;
                let mut buf = String::new();
                f.read_to_string(&mut buf)?;
                print!("{buf}");
                pos = file_len;
                no_state_ticks = 0; // reset grace period on new output
            }
        }

        // Check if a daemon is currently running (PID is alive)
        if let Some(st) = state::load_state(&state_file)? {
            no_state_ticks = 0;
            if state::is_locked(&st) {
                // Daemon is running, keep polling
            } else {
                // Daemon has exited — final drain
                if log.exists() {
                    let file_len = log.metadata().map(|m| m.len()).unwrap_or(0);
                    if file_len > pos {
                        let mut f = std::fs::File::open(&log)?;
                        std::io::Seek::seek(&mut f, std::io::SeekFrom::Start(pos))?;
                        let mut buf = String::new();
                        f.read_to_string(&mut buf)?;
                        print!("{buf}");
                    }
                }
                println!("\n(No active session or pending timer. Exiting watch.)");
                break;
            }
        } else {
            no_state_ticks += 1;
            // 500ms * 20 = 10s grace period
            if no_state_ticks >= 20 {
                println!("\n(No cryochamber instance found. Exiting watch.)");
                break;
            }
        }

        std::thread::sleep(std::time::Duration::from_millis(500));
    }

    Ok(())
}