crmux 0.19.1

pub fn cwd_to_project_dir(cwd: &str) -> String {
    cwd.trim_end_matches('/').replace(['/', '.'], "-")
}

pub fn parse_summary_from_index(json: &str, session_id: &str) -> Option<String> {
    let parsed: serde_json::Value = serde_json::from_str(json).ok()?;
    let sessions = parsed.as_array()?;
    for entry in sessions {
        if entry.get("sessionId")?.as_str()? == session_id {
            return entry.get("summary")?.as_str().map(ToString::to_string);
        }
    }
    None
}

pub fn read_plan_title(plans_dir: &std::path::Path, slug: &str) -> Option<String> {
    let path = plans_dir.join(format!("{slug}.md"));
    let content = std::fs::read_to_string(path).ok()?;
    let first_line = content.lines().next()?.trim();
    if first_line.is_empty() {
        return None;
    }
    Some(first_line.strip_prefix("# ").unwrap_or(first_line).to_string())
}

pub fn extract_slug_from_jsonl(reader: impl std::io::BufRead) -> Option<String> {
    for line in reader.lines() {
        let line = line.ok()?;
        if line.trim().is_empty() {
            continue;
        }
        let parsed: serde_json::Value = serde_json::from_str(&line).ok()?;
        if let Some(slug) = parsed.get("slug").and_then(|s| s.as_str()) {
            return Some(slug.to_string());
        }
    }
    None
}

fn extract_slash_command(text: &str) -> Option<String> {
    let start_tag = "<command-name>";
    let end_tag = "</command-name>";
    let start = text.find(start_tag)? + start_tag.len();
    let end = text[start..].find(end_tag)? + start;
    let cmd = text[start..end].trim();
    if cmd.is_empty() { None } else { Some(cmd.to_string()) }
}

pub fn extract_last_prompt_from_jsonl(file: &std::fs::File) -> Option<String> {
    use std::io::{Read, Seek, SeekFrom};

    const TAIL_SIZE: u64 = 64 * 1024;
    let mut file = file;
    let file_len = file.metadata().ok()?.len();
    let start = file_len.saturating_sub(TAIL_SIZE);
    file.seek(SeekFrom::Start(start)).ok()?;

    let mut buf = String::new();
    file.read_to_string(&mut buf).ok()?;

    for line in buf.lines().rev() {
        let line = line.trim();
        if line.is_empty() {
            continue;
        }
        let parsed: serde_json::Value = match serde_json::from_str(line) {
            Ok(v) => v,
            Err(_) => continue,
        };
        let is_user = parsed
            .get("type")
            .and_then(|t| t.as_str())
            .is_some_and(|t| t == "user");
        if !is_user {
            continue;
        }
        let text = (|| {
            let content = parsed.get("message")?.get("content")?;
            if let Some(s) = content.as_str() {
                Some(s.to_string())
            } else if let Some(arr) = content.as_array() {
                Some(
                    arr.iter()
                        .find_map(|block| block.get("text")?.as_str())?
                        .to_string(),
                )
            } else {
                None
            }
        })();
        if let Some(text) = text {
            let text = extract_slash_command(&text).unwrap_or(text);
            return Some(text);
        }
    }
    None
}

#[derive(Debug, Clone, PartialEq, Eq)]
pub struct PlanInfo {
    pub slug: String,
    pub title: String,
    pub path: String,
    pub project_name: String,
    pub session_id: String,
}

/// Resolve plan info from a session's JSONL file.
/// Returns `PlanInfo` if the session has a slug and the corresponding plan file exists.
pub fn resolve_plan_info(cwd: &str, session_id: &str, project_name: &str) -> Option<PlanInfo> {
    let home = std::env::var("HOME").ok()?;
    resolve_plan_info_with_home(&home, cwd, session_id, project_name)
}

fn resolve_plan_info_with_home(
    home: &str,
    cwd: &str,
    session_id: &str,
    project_name: &str,
) -> Option<PlanInfo> {
    let project_dir = cwd_to_project_dir(cwd);
    let base = format!("{home}/.claude/projects/{project_dir}");
    let jsonl_path = format!("{base}/{session_id}.jsonl");

    let file = std::fs::File::open(&jsonl_path).ok()?;
    let reader = std::io::BufReader::new(file);
    let slug = extract_slug_from_jsonl(reader)?;

    let plans_dir = std::path::PathBuf::from(format!("{home}/.claude/plans"));
    let plan_path = plans_dir.join(format!("{slug}.md"));
    let title = read_plan_title(&plans_dir, &slug)?;

    Some(PlanInfo {
        slug,
        title,
        path: plan_path.to_string_lossy().into_owned(),
        project_name: project_name.to_string(),
        session_id: session_id.to_string(),
    })
}

pub fn collect_all_plans_for_project(cwd: &str, project_name: &str) -> Vec<PlanInfo> {
    let Some(home) = std::env::var("HOME").ok() else {
        return Vec::new();
    };
    collect_all_plans_for_project_with_home(&home, cwd, project_name)
}

fn collect_all_plans_for_project_with_home(
    home: &str,
    cwd: &str,
    project_name: &str,
) -> Vec<PlanInfo> {
    let project_dir = cwd_to_project_dir(cwd);
    let base = std::path::PathBuf::from(format!("{home}/.claude/projects/{project_dir}"));
    let plans_dir = std::path::PathBuf::from(format!("{home}/.claude/plans"));

    let Ok(entries) = std::fs::read_dir(&base) else {
        return Vec::new();
    };

    let mut plans = Vec::new();
    let mut seen_slugs = std::collections::HashSet::new();
    for entry in entries.flatten() {
        let path = entry.path();
        if path.extension().and_then(|e| e.to_str()) != Some("jsonl") {
            continue;
        }
        let session_id = match path.file_stem().and_then(|s| s.to_str()) {
            Some(s) => s.to_string(),
            None => continue,
        };
        let Ok(file) = std::fs::File::open(&path) else {
            continue;
        };
        let reader = std::io::BufReader::new(file);
        let Some(slug) = extract_slug_from_jsonl(reader) else {
            continue;
        };
        // clone is needed: slug is moved into PlanInfo below, but HashSet needs its own copy.
        // Using HashSet<&str> would avoid the clone but complicates lifetimes.
        if !seen_slugs.insert(slug.clone()) {
            continue;
        }
        let Some(title) = read_plan_title(&plans_dir, &slug) else {
            continue;
        };
        let plan_path = plans_dir.join(format!("{slug}.md"));
        plans.push(PlanInfo {
            slug,
            title,
            path: plan_path.to_string_lossy().into_owned(),
            project_name: project_name.to_string(),
            session_id,
        });
    }
    plans
}

pub fn resolve_auto_title(cwd: &str, session_id: &str) -> Option<String> {
    let home = std::env::var("HOME").ok()?;
    let project_dir = cwd_to_project_dir(cwd);
    let base = format!("{home}/.claude/projects/{project_dir}");

    // 1. Try sessions-index.json summary
    let index_path = format!("{base}/sessions-index.json");
    if let Ok(json) = std::fs::read_to_string(&index_path)
        && let Some(summary) = parse_summary_from_index(&json, session_id)
    {
        return Some(summary);
    }

    let jsonl_path = format!("{base}/{session_id}.jsonl");

    // 2. Try plan title from slug
    if let Ok(file) = std::fs::File::open(&jsonl_path) {
        let reader = std::io::BufReader::new(file);
        if let Some(slug) = extract_slug_from_jsonl(reader) {
            let plans_dir = std::path::PathBuf::from(format!("{home}/.claude/plans"));
            if let Some(title) = read_plan_title(&plans_dir, &slug) {
                return Some(title);
            }
        }
    }

    // 3. Fallback: last user message from session JSONL
    if let Ok(file) = std::fs::File::open(&jsonl_path)
        && let Some(prompt) = extract_last_prompt_from_jsonl(&file)
    {
        return Some(prompt);
    }

    None
}

#[cfg(test)]
mod tests {
    #[test]
    fn test_cwd_to_project_dir_basic() {
        assert_eq!(
            super::cwd_to_project_dir("/home/user/proj"),
            "-home-user-proj"
        );
    }

    #[test]
    fn test_cwd_to_project_dir_with_dot() {
        assert_eq!(
            super::cwd_to_project_dir("/home/user/.config/proj"),
            "-home-user--config-proj"
        );
    }

    #[test]
    fn test_cwd_to_project_dir_trailing_slash() {
        assert_eq!(
            super::cwd_to_project_dir("/home/user/proj/"),
            "-home-user-proj"
        );
    }

    #[test]
    fn test_parse_summary_from_index_found() {
        let json = r#"[
            {"sessionId": "abc-123", "summary": "Fix login bug"},
            {"sessionId": "def-456", "summary": "Add tests"}
        ]"#;
        assert_eq!(
            super::parse_summary_from_index(json, "abc-123"),
            Some("Fix login bug".to_string())
        );
    }

    #[test]
    fn test_parse_summary_from_index_not_found() {
        let json = r#"[{"sessionId": "abc-123", "summary": "Fix bug"}]"#;
        assert_eq!(super::parse_summary_from_index(json, "unknown"), None);
    }

    #[test]
    fn test_parse_summary_from_index_invalid_json() {
        assert_eq!(super::parse_summary_from_index("not json", "abc"), None);
    }

    #[test]
    fn test_read_plan_title_with_heading() {
        let dir = tempfile::tempdir().unwrap();
        let plan_path = dir.path().join("my-slug.md");
        std::fs::write(&plan_path, "# My Cool Plan\nsome details\n").unwrap();
        assert_eq!(
            super::read_plan_title(dir.path(), "my-slug"),
            Some("My Cool Plan".to_string())
        );
    }

    #[test]
    fn test_read_plan_title_no_heading_prefix() {
        let dir = tempfile::tempdir().unwrap();
        let plan_path = dir.path().join("my-slug.md");
        std::fs::write(&plan_path, "No heading here\n").unwrap();
        assert_eq!(
            super::read_plan_title(dir.path(), "my-slug"),
            Some("No heading here".to_string())
        );
    }

    #[test]
    fn test_read_plan_title_file_not_found() {
        let dir = tempfile::tempdir().unwrap();
        assert_eq!(super::read_plan_title(dir.path(), "nonexistent"), None);
    }

    #[test]
    fn test_extract_slug_from_jsonl_found() {
        let jsonl = r#"{"type":"user","message":{"content":"hello"},"slug":"my-cool-plan"}
{"type":"user","message":{"content":"world"}}"#;
        let reader = std::io::BufReader::new(jsonl.as_bytes());
        assert_eq!(
            super::extract_slug_from_jsonl(reader),
            Some("my-cool-plan".to_string())
        );
    }

    #[test]
    fn test_extract_slug_from_jsonl_no_slug() {
        let jsonl = r#"{"type":"user","message":{"content":"hello"}}
{"type":"user","message":{"content":"world"}}"#;
        let reader = std::io::BufReader::new(jsonl.as_bytes());
        assert_eq!(super::extract_slug_from_jsonl(reader), None);
    }

    #[test]
    fn test_extract_slug_from_jsonl_empty() {
        let reader = std::io::BufReader::new("".as_bytes());
        assert_eq!(super::extract_slug_from_jsonl(reader), None);
    }

    fn write_tempfile(content: &str) -> (tempfile::TempDir, std::fs::File) {
        let dir = tempfile::tempdir().unwrap();
        let path = dir.path().join("test.jsonl");
        std::fs::write(&path, content).unwrap();
        let file = std::fs::File::open(&path).unwrap();
        (dir, file)
    }

    #[test]
    fn test_extract_last_prompt_short() {
        let (_dir, file) = write_tempfile(
            r#"{"type":"user","message":{"content":"hello world"}}"#,
        );
        assert_eq!(
            super::extract_last_prompt_from_jsonl(&file),
            Some("hello world".to_string())
        );
    }


    #[test]
    fn test_extract_last_prompt_skips_non_user() {
        let jsonl = "{\"type\":\"system\",\"message\":{\"content\":\"sys\"}}\n\
                     {\"type\":\"user\",\"message\":{\"content\":\"user msg\"}}";
        let (_dir, file) = write_tempfile(jsonl);
        assert_eq!(
            super::extract_last_prompt_from_jsonl(&file),
            Some("user msg".to_string())
        );
    }

    #[test]
    fn test_extract_last_prompt_array_content() {
        let jsonl = r#"{"type":"user","message":{"content":[{"type":"text","text":"block text"}]}}"#;
        let (_dir, file) = write_tempfile(jsonl);
        assert_eq!(
            super::extract_last_prompt_from_jsonl(&file),
            Some("block text".to_string())
        );
    }

    #[test]
    fn test_extract_last_prompt_empty() {
        let (_dir, file) = write_tempfile("");
        assert_eq!(super::extract_last_prompt_from_jsonl(&file), None);
    }

    #[test]
    fn test_extract_last_prompt_picks_last_user() {
        let jsonl = "{\"type\":\"user\",\"message\":{\"content\":\"first msg\"}}\n\
                     {\"type\":\"assistant\",\"message\":{\"content\":\"reply\"}}\n\
                     {\"type\":\"user\",\"message\":{\"content\":\"last msg\"}}";
        let (_dir, file) = write_tempfile(jsonl);
        assert_eq!(
            super::extract_last_prompt_from_jsonl(&file),
            Some("last msg".to_string())
        );
    }

    #[test]
    fn test_resolve_plan_info_found() {
        let home_dir = tempfile::tempdir().unwrap();
        let home = home_dir.path().to_str().unwrap();

        let project_dir = super::cwd_to_project_dir("/work/myproject");
        let session_dir = home_dir.path().join(format!(".claude/projects/{project_dir}"));
        std::fs::create_dir_all(&session_dir).unwrap();
        let jsonl_content = r#"{"slug":"my-plan","type":"user","message":{"content":"hello"}}"#;
        std::fs::write(session_dir.join("sess-001.jsonl"), jsonl_content).unwrap();

        let plans_dir = home_dir.path().join(".claude/plans");
        std::fs::create_dir_all(&plans_dir).unwrap();
        std::fs::write(plans_dir.join("my-plan.md"), "# My Great Plan\ndetails\n").unwrap();

        let result =
            super::resolve_plan_info_with_home(home, "/work/myproject", "sess-001", "myproject");
        assert!(result.is_some());
        let info = result.unwrap();
        assert_eq!(info.slug, "my-plan");
        assert_eq!(info.title, "My Great Plan");
        assert_eq!(info.project_name, "myproject");
        assert_eq!(info.session_id, "sess-001");
        assert!(info.path.ends_with("my-plan.md"));
    }

    #[test]
    fn test_resolve_plan_info_no_slug() {
        let home_dir = tempfile::tempdir().unwrap();
        let home = home_dir.path().to_str().unwrap();

        let project_dir = super::cwd_to_project_dir("/work/myproject");
        let session_dir = home_dir.path().join(format!(".claude/projects/{project_dir}"));
        std::fs::create_dir_all(&session_dir).unwrap();
        let jsonl_content = r#"{"type":"user","message":{"content":"no slug here"}}"#;
        std::fs::write(session_dir.join("sess-002.jsonl"), jsonl_content).unwrap();

        let result =
            super::resolve_plan_info_with_home(home, "/work/myproject", "sess-002", "myproject");
        assert!(result.is_none());
    }

    #[test]
    fn test_resolve_plan_info_no_plan_file() {
        let home_dir = tempfile::tempdir().unwrap();
        let home = home_dir.path().to_str().unwrap();

        let project_dir = super::cwd_to_project_dir("/work/myproject");
        let session_dir = home_dir.path().join(format!(".claude/projects/{project_dir}"));
        std::fs::create_dir_all(&session_dir).unwrap();
        let jsonl_content = r#"{"slug":"missing-plan","type":"user","message":{"content":"hello"}}"#;
        std::fs::write(session_dir.join("sess-003.jsonl"), jsonl_content).unwrap();

        let plans_dir = home_dir.path().join(".claude/plans");
        std::fs::create_dir_all(&plans_dir).unwrap();

        let result =
            super::resolve_plan_info_with_home(home, "/work/myproject", "sess-003", "myproject");
        assert!(result.is_none());
    }

    #[test]
    fn test_collect_all_plans_empty_dir() {
        let home_dir = tempfile::tempdir().unwrap();
        let home = home_dir.path().to_str().unwrap();
        let project_dir = super::cwd_to_project_dir("/work/myproject");

        // Create the project sessions dir but leave it empty
        let session_dir = home_dir.path().join(format!(".claude/projects/{project_dir}"));
        std::fs::create_dir_all(&session_dir).unwrap();
        let plans_dir = home_dir.path().join(".claude/plans");
        std::fs::create_dir_all(&plans_dir).unwrap();

        let result = super::collect_all_plans_for_project_with_home(home, "/work/myproject", "myproject");
        assert!(result.is_empty());
    }

    #[test]
    fn test_collect_all_plans_with_slugs() {
        let home_dir = tempfile::tempdir().unwrap();
        let home = home_dir.path().to_str().unwrap();
        let project_dir = super::cwd_to_project_dir("/work/myproject");

        let session_dir = home_dir.path().join(format!(".claude/projects/{project_dir}"));
        std::fs::create_dir_all(&session_dir).unwrap();

        // Create a JSONL file with a slug
        let jsonl_content = r#"{"slug":"plan-alpha","type":"user","message":{"content":"hello"}}"#;
        std::fs::write(session_dir.join("sess-001.jsonl"), jsonl_content).unwrap();

        // Create the corresponding plan file
        let plans_dir = home_dir.path().join(".claude/plans");
        std::fs::create_dir_all(&plans_dir).unwrap();
        std::fs::write(plans_dir.join("plan-alpha.md"), "# Alpha Plan\ndetails\n").unwrap();

        let result = super::collect_all_plans_for_project_with_home(home, "/work/myproject", "myproject");
        assert_eq!(result.len(), 1);
        assert_eq!(result[0].slug, "plan-alpha");
        assert_eq!(result[0].title, "Alpha Plan");
        assert_eq!(result[0].project_name, "myproject");
        assert_eq!(result[0].session_id, "sess-001");
    }

    #[test]
    fn test_collect_all_plans_deduplicates_by_slug() {
        let home_dir = tempfile::tempdir().unwrap();
        let home = home_dir.path().to_str().unwrap();
        let project_dir = super::cwd_to_project_dir("/work/myproject");

        let session_dir = home_dir.path().join(format!(".claude/projects/{project_dir}"));
        std::fs::create_dir_all(&session_dir).unwrap();

        // Two JSONL files with the same slug
        let jsonl = r#"{"slug":"same-plan","type":"user","message":{"content":"hello"}}"#;
        std::fs::write(session_dir.join("sess-001.jsonl"), jsonl).unwrap();
        std::fs::write(session_dir.join("sess-002.jsonl"), jsonl).unwrap();

        let plans_dir = home_dir.path().join(".claude/plans");
        std::fs::create_dir_all(&plans_dir).unwrap();
        std::fs::write(plans_dir.join("same-plan.md"), "# Same Plan\n").unwrap();

        let result = super::collect_all_plans_for_project_with_home(home, "/work/myproject", "myproject");
        assert_eq!(result.len(), 1);
        assert_eq!(result[0].slug, "same-plan");
    }

    #[test]
    fn test_collect_all_plans_missing_plan_file() {
        let home_dir = tempfile::tempdir().unwrap();
        let home = home_dir.path().to_str().unwrap();
        let project_dir = super::cwd_to_project_dir("/work/myproject");

        let session_dir = home_dir.path().join(format!(".claude/projects/{project_dir}"));
        std::fs::create_dir_all(&session_dir).unwrap();

        // JSONL with slug, but no corresponding plan.md
        let jsonl = r#"{"slug":"ghost-plan","type":"user","message":{"content":"hello"}}"#;
        std::fs::write(session_dir.join("sess-001.jsonl"), jsonl).unwrap();

        let plans_dir = home_dir.path().join(".claude/plans");
        std::fs::create_dir_all(&plans_dir).unwrap();
        // No ghost-plan.md created

        let result = super::collect_all_plans_for_project_with_home(home, "/work/myproject", "myproject");
        assert!(result.is_empty());
    }

    #[test]
    fn test_extract_slash_command_simple() {
        assert_eq!(
            super::extract_slash_command("<command-name>/clear</command-name>"),
            Some("/clear".to_string())
        );
    }

    #[test]
    fn test_extract_slash_command_multiple_tags() {
        assert_eq!(
            super::extract_slash_command("<command-name>/commit</command-name><command-message>commit</command-message><command-args></command-args>"),
            Some("/commit".to_string())
        );
    }

    #[test]
    fn test_extract_slash_command_no_tags() {
        assert_eq!(
            super::extract_slash_command("hello world"),
            None
        );
    }

    #[test]
    fn test_extract_slash_command_empty_name() {
        assert_eq!(
            super::extract_slash_command("<command-name></command-name>"),
            None
        );
    }

    #[test]
    fn test_extract_last_prompt_slash_command() {
        let jsonl = r#"{"type":"user","message":{"content":"<command-name>/clear</command-name>\n            <command-message>clear</command-message>\n            <command-args></command-args>"}}"#;
        let (_dir, file) = write_tempfile(jsonl);
        assert_eq!(
            super::extract_last_prompt_from_jsonl(&file),
            Some("/clear".to_string())
        );
    }

    #[test]
    fn test_extract_last_prompt_normal_text_unchanged() {
        let jsonl = r#"{"type":"user","message":{"content":"hello world"}}"#;
        let (_dir, file) = write_tempfile(jsonl);
        assert_eq!(
            super::extract_last_prompt_from_jsonl(&file),
            Some("hello world".to_string())
        );
    }

    #[test]
    fn test_collect_all_plans_nonexistent_dir() {
        let home_dir = tempfile::tempdir().unwrap();
        let home = home_dir.path().to_str().unwrap();
        // Don't create any directories
        let result = super::collect_all_plans_for_project_with_home(home, "/nonexistent/path", "myproject");
        assert!(result.is_empty());
    }
}