huskies/server/src/chat/commands/freeze.rs

//! Handler for the `freeze` and `unfreeze` commands.
//!
//! `freeze <number>` transitions the story to `Stage::Frozen`, halting pipeline
//! advancement and auto-assign until `unfreeze <number>` restores the prior stage.

use super::CommandContext;
use crate::io::story_metadata::parse_front_matter;
use std::path::Path;

/// Handle the `freeze` command.
///
/// Parses `<number>` from `ctx.args`, locates the work item, and sets
/// `frozen: true` in its front matter.
pub(super) fn handle_freeze(ctx: &CommandContext) -> Option<String> {
    let num_str = ctx.args.trim();
    if num_str.is_empty() || !num_str.chars().all(|c| c.is_ascii_digit()) {
        return Some(format!(
            "Usage: `{} freeze <number>` (e.g. `freeze 42`)",
            ctx.services.bot_name
        ));
    }
    Some(freeze_by_number(ctx.effective_root(), num_str))
}

/// Core freeze logic: find story by numeric prefix and set `frozen: true`.
///
/// Returns a Markdown-formatted response string suitable for all transports.
pub(crate) fn freeze_by_number(project_root: &Path, story_number: &str) -> String {
    let (story_id, _, _, _) =
        match crate::chat::lookup::find_story_by_number(project_root, story_number) {
            Some(found) => found,
            None => {
                return format!("No story, bug, or spike with number **{story_number}** found.");
            }
        };

    freeze_by_story_id(&story_id)
}

fn freeze_by_story_id(story_id: &str) -> String {
    let contents = match crate::db::read_content(story_id) {
        Some(c) => c,
        None => return format!("Failed to read story content for **{story_id}**"),
    };

    let meta = match parse_front_matter(&contents) {
        Ok(m) => m,
        Err(e) => return format!("Failed to parse front matter for **{story_id}**: {e}"),
    };

    let story_name = meta.name.as_deref().unwrap_or(story_id).to_string();

    // Check if already frozen via the typed stage.
    if crate::pipeline_state::read_typed(story_id)
        .ok()
        .flatten()
        .map(|i| i.stage.is_frozen())
        .unwrap_or(false)
    {
        return format!("**{story_name}** ({story_id}) is already frozen.");
    }

    match crate::pipeline_state::transition_to_frozen(story_id) {
        Ok(_) => format!(
            "Frozen **{story_name}** ({story_id}). Pipeline advancement and auto-assign suppressed until unfrozen."
        ),
        Err(e) => format!("Failed to freeze **{story_name}** ({story_id}): {e}"),
    }
}

/// Handle the `unfreeze` command.
///
/// Parses `<number>` from `ctx.args`, locates the work item, and clears the
/// `frozen` flag to resume normal pipeline behaviour.
pub(super) fn handle_unfreeze(ctx: &CommandContext) -> Option<String> {
    let num_str = ctx.args.trim();
    if num_str.is_empty() || !num_str.chars().all(|c| c.is_ascii_digit()) {
        return Some(format!(
            "Usage: `{} unfreeze <number>` (e.g. `unfreeze 42`)",
            ctx.services.bot_name
        ));
    }
    Some(unfreeze_by_number(ctx.effective_root(), num_str))
}

/// Core unfreeze logic: find story by numeric prefix and clear `frozen` flag.
pub(crate) fn unfreeze_by_number(project_root: &Path, story_number: &str) -> String {
    let (story_id, _, _, _) =
        match crate::chat::lookup::find_story_by_number(project_root, story_number) {
            Some(found) => found,
            None => {
                return format!("No story, bug, or spike with number **{story_number}** found.");
            }
        };

    unfreeze_by_story_id(&story_id)
}

fn unfreeze_by_story_id(story_id: &str) -> String {
    let contents = match crate::db::read_content(story_id) {
        Some(c) => c,
        None => return format!("Failed to read story content for **{story_id}**"),
    };

    let meta = match parse_front_matter(&contents) {
        Ok(m) => m,
        Err(e) => return format!("Failed to parse front matter for **{story_id}**: {e}"),
    };

    let story_name = meta.name.as_deref().unwrap_or(story_id).to_string();

    // Check frozen via typed stage.
    let is_frozen = crate::pipeline_state::read_typed(story_id)
        .ok()
        .flatten()
        .map(|i| i.stage.is_frozen())
        .unwrap_or(false);

    if !is_frozen {
        return format!("**{story_name}** ({story_id}) is not frozen. Nothing to unfreeze.");
    }

    match crate::pipeline_state::transition_to_unfrozen(story_id) {
        Ok(_) => {
            format!("Unfrozen **{story_name}** ({story_id}). Normal pipeline behaviour resumed.")
        }
        Err(e) => format!("Failed to unfreeze **{story_name}** ({story_id}): {e}"),
    }
}

// ---------------------------------------------------------------------------
// Tests
// ---------------------------------------------------------------------------

#[cfg(test)]
mod tests {
    use crate::chat::test_helpers::write_story_file;

    use super::super::{CommandDispatch, try_handle_command};

    fn freeze_cmd_with_root(root: &std::path::Path, args: &str) -> Option<String> {
        let services = crate::services::Services::new_test(root.to_path_buf(), "Timmy".to_string());
        let room_id = "!test:example.com".to_string();
        let dispatch = CommandDispatch {
            services: &services,
            project_root: &services.project_root,
            bot_user_id: "@timmy:homeserver.local",
            room_id: &room_id,
        };
        try_handle_command(&dispatch, &format!("@timmy freeze {args}"))
    }

    fn unfreeze_cmd_with_root(root: &std::path::Path, args: &str) -> Option<String> {
        let services = crate::services::Services::new_test(root.to_path_buf(), "Timmy".to_string());
        let room_id = "!test:example.com".to_string();
        let dispatch = CommandDispatch {
            services: &services,
            project_root: &services.project_root,
            bot_user_id: "@timmy:homeserver.local",
            room_id: &room_id,
        };
        try_handle_command(&dispatch, &format!("@timmy unfreeze {args}"))
    }

    #[test]
    fn freeze_command_is_registered() {
        use super::super::commands;
        assert!(
            commands().iter().any(|c| c.name == "freeze"),
            "freeze command must be in the registry"
        );
    }

    #[test]
    fn unfreeze_command_is_registered() {
        use super::super::commands;
        assert!(
            commands().iter().any(|c| c.name == "unfreeze"),
            "unfreeze command must be in the registry"
        );
    }

    #[test]
    fn freeze_command_no_args_returns_usage() {
        let tmp = tempfile::TempDir::new().unwrap();
        let output = freeze_cmd_with_root(tmp.path(), "").unwrap();
        assert!(
            output.contains("Usage"),
            "no args should show usage: {output}"
        );
    }

    #[test]
    fn unfreeze_command_no_args_returns_usage() {
        let tmp = tempfile::TempDir::new().unwrap();
        let output = unfreeze_cmd_with_root(tmp.path(), "").unwrap();
        assert!(
            output.contains("Usage"),
            "no args should show usage: {output}"
        );
    }

    #[test]
    fn freeze_command_not_found_returns_error() {
        let tmp = tempfile::TempDir::new().unwrap();
        let output = freeze_cmd_with_root(tmp.path(), "9988").unwrap();
        assert!(
            output.contains("9988") && output.contains("found"),
            "not-found message should include number and 'found': {output}"
        );
    }

    #[test]
    fn freeze_command_sets_stage_to_frozen() {
        let tmp = tempfile::TempDir::new().unwrap();
        crate::crdt_state::init_for_test();
        crate::db::ensure_content_store();
        write_story_file(
            tmp.path(),
            "2_current",
            "9940_story_freezeme.md",
            "---\nname: Freeze Me\n---\n# Story\n",
        );
        let output = freeze_cmd_with_root(tmp.path(), "9940").unwrap();
        assert!(
            output.contains("Frozen") && output.contains("Freeze Me"),
            "should confirm freeze with story name: {output}"
        );
        let item = crate::pipeline_state::read_typed("9940_story_freezeme")
            .expect("read_typed should succeed")
            .expect("item should be present");
        assert!(
            item.stage.is_frozen(),
            "stage should be Frozen after freeze: {:?}",
            item.stage
        );
    }

    #[test]
    fn unfreeze_command_restores_prior_stage() {
        let tmp = tempfile::TempDir::new().unwrap();
        crate::crdt_state::init_for_test();
        crate::db::ensure_content_store();
        write_story_file(
            tmp.path(),
            "2_current",
            "9941_story_frozen.md",
            "---\nname: Frozen Story\n---\n# Story\n",
        );
        // Freeze first.
        let freeze_out = freeze_cmd_with_root(tmp.path(), "9941").unwrap();
        assert!(
            freeze_out.contains("Frozen"),
            "should confirm freeze: {freeze_out}"
        );
        // Now unfreeze.
        let output = unfreeze_cmd_with_root(tmp.path(), "9941").unwrap();
        assert!(
            output.contains("Unfrozen") && output.contains("Frozen Story"),
            "should confirm unfreeze with story name: {output}"
        );
        let item = crate::pipeline_state::read_typed("9941_story_frozen")
            .expect("read_typed should succeed")
            .expect("item should be present");
        assert!(
            matches!(item.stage, crate::pipeline_state::Stage::Coding),
            "stage should be restored to Coding: {:?}",
            item.stage
        );
    }

    #[test]
    fn unfreeze_command_not_frozen_returns_error() {
        let tmp = tempfile::TempDir::new().unwrap();
        crate::crdt_state::init_for_test();
        crate::db::ensure_content_store();
        write_story_file(
            tmp.path(),
            "2_current",
            "9942_story_notfrozen.md",
            "---\nname: Not Frozen\n---\n# Story\n",
        );
        let output = unfreeze_cmd_with_root(tmp.path(), "9942").unwrap();
        assert!(
            output.contains("not frozen"),
            "should return not-frozen error: {output}"
        );
    }

    #[test]
    fn freeze_command_already_frozen_returns_message() {
        let tmp = tempfile::TempDir::new().unwrap();
        crate::crdt_state::init_for_test();
        crate::db::ensure_content_store();
        write_story_file(
            tmp.path(),
            "2_current",
            "9943_story_alreadyfrozen.md",
            "---\nname: Already Frozen\n---\n# Story\n",
        );
        // Freeze it first.
        freeze_cmd_with_root(tmp.path(), "9943").unwrap();
        // Try to freeze again.
        let output = freeze_cmd_with_root(tmp.path(), "9943").unwrap();
        assert!(
            output.contains("already frozen"),
            "should say already frozen: {output}"
        );
    }
}