bamboo-agent 2026.4.5

//! Global prompt template defaults used by new sessions.
//!
//! Phase 3 semantics:
//! - `/v1/agent/system-prompt` represents a global default template.
//! - New sessions use this template when no explicit base prompt is provided.
//! - Existing sessions stay stable via persisted `metadata.base_system_prompt`.

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

const CLAUDE_DIR_NAME: &str = ".claude";
const SYSTEM_PROMPT_FILE_NAME: &str = "system-prompt.md";

#[derive(Debug, Clone, Copy, PartialEq, Eq)]
pub enum GlobalSystemPromptSource {
    BambooFile,
    ClaudeLegacyFile,
    BuiltinDefault,
}

#[derive(Debug, Clone, PartialEq, Eq)]
pub struct GlobalSystemPromptTemplate {
    pub content: String,
    pub source: GlobalSystemPromptSource,
    pub path: Option<PathBuf>,
}

fn bamboo_system_prompt_file_path() -> PathBuf {
    crate::core::paths::bamboo_dir().join(SYSTEM_PROMPT_FILE_NAME)
}

fn claude_system_prompt_file_path() -> Option<PathBuf> {
    dirs::home_dir().map(|dir| dir.join(CLAUDE_DIR_NAME).join(SYSTEM_PROMPT_FILE_NAME))
}

fn read_non_empty_prompt_file(path: &Path) -> Option<String> {
    std::fs::read_to_string(path).ok().and_then(|content| {
        let trimmed = content.trim();
        if trimmed.is_empty() {
            None
        } else {
            Some(trimmed.to_string())
        }
    })
}

fn builtin_default_prompt() -> String {
    crate::server::app_state::DEFAULT_BASE_PROMPT.to_string()
}

fn resolve_with_paths(
    bamboo_path: &Path,
    claude_legacy_path: Option<&Path>,
) -> GlobalSystemPromptTemplate {
    if let Some(content) = read_non_empty_prompt_file(bamboo_path) {
        return GlobalSystemPromptTemplate {
            content,
            source: GlobalSystemPromptSource::BambooFile,
            path: Some(bamboo_path.to_path_buf()),
        };
    }

    if let Some(path) = claude_legacy_path {
        if let Some(content) = read_non_empty_prompt_file(path) {
            return GlobalSystemPromptTemplate {
                content,
                source: GlobalSystemPromptSource::ClaudeLegacyFile,
                path: Some(path.to_path_buf()),
            };
        }
    }

    GlobalSystemPromptTemplate {
        content: builtin_default_prompt(),
        source: GlobalSystemPromptSource::BuiltinDefault,
        path: None,
    }
}

/// Read the global default prompt template used for new session initialization.
///
/// Fallback order:
/// 1. `${BAMBOO_DATA_DIR}/system-prompt.md` (typically `~/.bamboo/system-prompt.md`) if present and non-empty
/// 2. `~/.claude/system-prompt.md` if present and non-empty (legacy compatibility)
/// 3. Builtin [`crate::server::app_state::DEFAULT_BASE_PROMPT`]
pub fn read_global_default_system_prompt_template() -> String {
    resolve_global_default_system_prompt_template().content
}

/// Resolve the global default prompt template with source metadata.
pub fn resolve_global_default_system_prompt_template() -> GlobalSystemPromptTemplate {
    let bamboo_path = bamboo_system_prompt_file_path();
    let claude_legacy_path = claude_system_prompt_file_path();

    resolve_with_paths(&bamboo_path, claude_legacy_path.as_deref())
}

#[cfg(test)]
mod tests {
    use super::{
        read_global_default_system_prompt_template, resolve_with_paths, GlobalSystemPromptSource,
    };
    use std::fs;
    use tempfile::tempdir;

    #[test]
    fn global_default_template_is_never_empty() {
        assert!(!read_global_default_system_prompt_template()
            .trim()
            .is_empty());
    }

    #[test]
    fn resolution_prefers_bamboo_over_legacy_claude() {
        let temp = tempdir().expect("temp dir");
        let bamboo_prompt = temp.path().join("bamboo-system-prompt.md");
        let claude_prompt = temp.path().join("claude-system-prompt.md");

        fs::write(&bamboo_prompt, "from bamboo").expect("write bamboo prompt");
        fs::write(&claude_prompt, "from claude").expect("write claude prompt");

        let resolved = resolve_with_paths(&bamboo_prompt, Some(&claude_prompt));
        assert_eq!(resolved.content, "from bamboo");
        assert_eq!(resolved.source, GlobalSystemPromptSource::BambooFile);
        assert_eq!(resolved.path.as_deref(), Some(bamboo_prompt.as_path()));
    }

    #[test]
    fn resolution_falls_back_to_legacy_claude_when_bamboo_missing() {
        let temp = tempdir().expect("temp dir");
        let bamboo_prompt = temp.path().join("missing-bamboo-system-prompt.md");
        let claude_prompt = temp.path().join("claude-system-prompt.md");

        fs::write(&claude_prompt, "from claude").expect("write claude prompt");

        let resolved = resolve_with_paths(&bamboo_prompt, Some(&claude_prompt));
        assert_eq!(resolved.content, "from claude");
        assert_eq!(resolved.source, GlobalSystemPromptSource::ClaudeLegacyFile);
        assert_eq!(resolved.path.as_deref(), Some(claude_prompt.as_path()));
    }
}