batty-cli 0.11.63

//! Free helper functions used by the daemon module.

use std::path::{Path, PathBuf};
use std::process::Command;

use anyhow::{Context, Result, bail};

use crate::agent::BackendHealth;
use crate::team::board_cmd;
use crate::team::config::RoleType;
use crate::team::git_cmd;
use crate::team::hierarchy::MemberInstance;
use crate::tmux;

#[derive(Debug, Clone, PartialEq, Eq)]
pub(super) struct MemberWorktreeContext {
    pub path: PathBuf,
    pub branch: Option<String>,
}

pub(super) fn describe_command_failure(
    command: &str,
    args: &[&str],
    output: &std::process::Output,
) -> String {
    let stderr = String::from_utf8_lossy(&output.stderr).trim().to_string();
    let stdout = String::from_utf8_lossy(&output.stdout).trim().to_string();
    let details = if !stderr.is_empty() {
        stderr
    } else if !stdout.is_empty() {
        stdout
    } else {
        format!("process exited with status {}", output.status)
    };

    format!("`{command} {}` failed: {details}", args.join(" "))
}

pub(super) fn default_prompt_file_for_role(role_type: RoleType) -> &'static str {
    match role_type {
        RoleType::Architect => "architect.md",
        RoleType::Manager => "manager.md",
        RoleType::Engineer => "engineer.md",
        RoleType::User => "architect.md",
    }
}

pub(super) fn role_prompt_path(
    team_config_dir: &Path,
    prompt_override: Option<&str>,
    role_type: RoleType,
) -> PathBuf {
    team_config_dir.join(prompt_override.unwrap_or(default_prompt_file_for_role(role_type)))
}

/// Extract the `## Nudge` section from a prompt .md file.
///
/// Returns the text after `## Nudge` up to the next `## ` heading or EOF.
/// Returns `None` if no `## Nudge` section is found.
pub(super) fn extract_nudge_section(prompt_path: &Path) -> Option<String> {
    let content = std::fs::read_to_string(prompt_path).ok()?;
    let mut in_nudge = false;
    let mut lines = Vec::new();

    for line in content.lines() {
        if line.starts_with("## Nudge") {
            in_nudge = true;
            continue;
        }
        if in_nudge {
            // Stop at next heading
            if line.starts_with("## ") {
                break;
            }
            lines.push(line);
        }
    }

    if lines.is_empty() {
        return None;
    }

    let text = lines.join("\n").trim().to_string();
    if text.is_empty() { None } else { Some(text) }
}

pub(super) fn format_stuck_duration(stuck_age_secs: u64) -> String {
    if stuck_age_secs >= 3600 {
        let hours = stuck_age_secs / 3600;
        let mins = (stuck_age_secs % 3600) / 60;
        format!("{hours}h {mins}m")
    } else if stuck_age_secs >= 60 {
        let mins = stuck_age_secs / 60;
        let secs = stuck_age_secs % 60;
        format!("{mins}m {secs}s")
    } else {
        format!("{stuck_age_secs}s")
    }
}

pub(super) fn ensure_tmux_session_ready(session: &str) -> Result<()> {
    if tmux::session_exists(session) {
        Ok(())
    } else {
        bail!("daemon startup pre-flight failed: tmux session '{session}' is missing")
    }
}

pub(super) fn ensure_kanban_available() -> Result<()> {
    let output = Command::new("kanban-md")
        .arg("--help")
        .output()
        .context(
            "daemon startup pre-flight failed while verifying board tooling: could not execute `kanban-md --help`",
        )?;
    if output.status.success() {
        return Ok(());
    }

    let stderr = String::from_utf8_lossy(&output.stderr).trim().to_string();
    let detail = if stderr.is_empty() {
        "unknown error".to_string()
    } else {
        stderr
    };
    bail!("daemon startup pre-flight failed: `kanban-md --help` failed: {detail}");
}

pub(super) fn board_dir(project_root: &Path) -> PathBuf {
    project_root
        .join(".batty")
        .join("team_config")
        .join("board")
}

pub(super) fn ensure_board_initialized(project_root: &Path) -> Result<bool> {
    let board_dir = board_dir(project_root);
    if board_dir.join("tasks").is_dir() {
        return Ok(false);
    }

    board_cmd::init(&board_dir).map_err(|error| {
        anyhow::anyhow!(
            "daemon startup pre-flight failed: unable to initialize board at '{}': {error}",
            board_dir.display()
        )
    })?;
    Ok(true)
}

pub(super) fn ensure_git_ready(project_root: &Path) -> Result<()> {
    let repo_check = Command::new("git")
        .args(["rev-parse", "--is-inside-work-tree"])
        .current_dir(project_root)
        .output()
        .context(
            "daemon startup pre-flight failed while verifying git repository: could not execute `git rev-parse --is-inside-work-tree`",
        )?;
    if !repo_check.status.success() || String::from_utf8_lossy(&repo_check.stdout).trim() != "true"
    {
        bail!("daemon startup pre-flight failed: project root is not a git repository");
    }

    let dirty_lines: Vec<String> = git_cmd::status_porcelain(project_root)
        .context("daemon startup pre-flight failed while checking git dirty state")?
        .lines()
        .map(str::trim)
        .filter(|line| !line.is_empty())
        .filter(|line| {
            let path = line.get(3..).unwrap_or("").trim();
            let path = path.split(" -> ").last().unwrap_or(path).trim_matches('"');
            !path.starts_with(".batty/")
        })
        .map(str::to_string)
        .collect();
    if !dirty_lines.is_empty() {
        bail!(
            "daemon startup pre-flight failed: git worktree is dirty:\n{}",
            dirty_lines.join("\n")
        );
    }

    for key in ["user.name", "user.email"] {
        let output = Command::new("git")
            .args(["config", "--get", key])
            .current_dir(project_root)
            .output()
            .with_context(|| {
                format!(
                    "daemon startup pre-flight failed while verifying git config: could not execute `git config --get {key}`"
                )
            })?;
        if !output.status.success() || String::from_utf8_lossy(&output.stdout).trim().is_empty() {
            bail!("daemon startup pre-flight failed: git config '{key}' is missing");
        }
    }

    Ok(())
}

pub(super) fn ensure_worktree_operations(project_root: &Path) -> Result<()> {
    let probe_root = project_root.join(".batty").join("startup-preflight");
    std::fs::create_dir_all(&probe_root).with_context(|| {
        format!(
            "daemon startup pre-flight failed: unable to create startup preflight directory '{}'",
            probe_root.display()
        )
    })?;
    let worktree_dir = probe_root.join(format!(
        "worktree-probe-{}",
        std::time::SystemTime::now()
            .duration_since(std::time::UNIX_EPOCH)
            .unwrap_or_default()
            .as_nanos()
    ));
    let worktree_path = worktree_dir.to_string_lossy().to_string();

    let add_output = Command::new("git")
        .args(["worktree", "add", "--detach", &worktree_path, "HEAD"])
        .current_dir(project_root)
        .output()
        .context(
            "daemon startup pre-flight failed while verifying git worktree support: could not execute `git worktree add`",
        )?;
    if !add_output.status.success() {
        let detail = describe_command_failure(
            "git",
            &["worktree", "add", "--detach", &worktree_path, "HEAD"],
            &add_output,
        );
        bail!("daemon startup pre-flight failed: {detail}");
    }

    let remove_output = Command::new("git")
        .args(["worktree", "remove", "--force", &worktree_path])
        .current_dir(project_root)
        .output()
        .context(
            "daemon startup pre-flight failed while verifying git worktree cleanup: could not execute `git worktree remove`",
        )?;
    if !remove_output.status.success() {
        let detail = describe_command_failure(
            "git",
            &["worktree", "remove", "--force", &worktree_path],
            &remove_output,
        );
        let _ = std::fs::remove_dir_all(&worktree_dir);
        bail!("daemon startup pre-flight failed: {detail}");
    }

    if worktree_dir.exists() {
        let _ = std::fs::remove_dir_all(&worktree_dir);
        bail!(
            "daemon startup pre-flight failed: temporary worktree '{}' was not removed",
            worktree_dir.display()
        );
    }

    Ok(())
}

pub(super) fn ensure_telemetry_writable(project_root: &Path) -> Result<()> {
    let conn = crate::team::telemetry_db::open(project_root)
        .context("daemon startup pre-flight failed: telemetry.db is not writable")?;
    conn.execute(
        "CREATE TABLE IF NOT EXISTS __batty_preflight_probe (id INTEGER)",
        [],
    )
    .context("daemon startup pre-flight failed: telemetry.db probe write failed")?;
    Ok(())
}

pub(super) fn ensure_agent_binaries_available(members: &[MemberInstance]) -> Result<()> {
    for member in members {
        if member.role_type == RoleType::User {
            continue;
        }

        let agent_name = member.agent.as_deref().unwrap_or("claude");
        let Some(health) = crate::agent::health_check_by_name(agent_name) else {
            bail!(
                "daemon startup pre-flight failed: unknown agent backend '{}' for member '{}'",
                agent_name,
                member.name
            );
        };
        if health != BackendHealth::Healthy {
            bail!(
                "daemon startup pre-flight failed: backend '{}' for member '{}' is {}",
                agent_name,
                member.name,
                health.as_str()
            );
        }
    }

    Ok(())
}