remem-ai 0.5.96

use std::sync::LazyLock;

pub mod claude;
pub mod codex;
pub(crate) mod common;
pub(crate) mod redaction;

/// Normalized event parsed from a hook's raw JSON input.
#[derive(Debug, Clone, serde::Serialize, serde::Deserialize)]
pub struct ParsedHookEvent {
    pub session_id: String,
    pub cwd: Option<String>,
    pub project: String,
    pub reference_time_epoch: Option<i64>,
    pub tool_name: String,
    pub tool_input: Option<serde_json::Value>,
    pub tool_response: Option<serde_json::Value>,
}

/// Structured event summary extracted from a tool call.
#[derive(Debug, Clone, serde::Serialize, serde::Deserialize)]
pub struct EventSummary {
    pub event_type: String,
    pub summary: String,
    pub detail: Option<String>,
    pub files_json: Option<String>,
    pub exit_code: Option<i32>,
}

/// Adapter trait for AI coding tool integrations.
/// Each supported tool (Claude Code, Codex, Cursor, ...) implements this.
pub trait ToolAdapter: Send + Sync {
    /// Adapter identifier (e.g. "claude-code", "codex-cli")
    fn name(&self) -> &str;

    /// Try to parse raw hook JSON into a normalized event.
    /// Returns None if this adapter cannot handle the input format.
    fn parse_hook(&self, raw_json: &str) -> Option<ParsedHookEvent>;

    /// Whether this event should be skipped entirely.
    fn should_skip(&self, event: &ParsedHookEvent) -> bool;

    /// Whether a Bash/shell command should be skipped (read-only, routine).
    fn should_skip_bash(&self, command: &str) -> bool;

    /// Extract a structured event summary from the tool call.
    fn classify_event(&self, event: &ParsedHookEvent) -> Option<EventSummary>;
}

static ADAPTERS: LazyLock<Vec<Box<dyn ToolAdapter>>> = LazyLock::new(|| {
    vec![
        Box::new(crate::adapter::claude::ClaudeCodeAdapter),
        Box::new(crate::adapter::codex::CodexAdapter),
    ]
});

/// Auto-detect adapter from raw hook JSON and parse the event.
pub fn detect_adapter(raw_json: &str) -> Option<(&'static dyn ToolAdapter, ParsedHookEvent)> {
    if let Ok(name) = std::env::var("REMEM_HOOK_ADAPTER") {
        if !name.trim().is_empty() {
            return detect_adapter_by_name(raw_json, &name);
        }
    }

    for adapter in ADAPTERS.iter() {
        if let Some(event) = adapter.parse_hook(raw_json) {
            return Some((adapter.as_ref(), event));
        }
    }
    None
}

pub(crate) fn detect_adapter_by_name(
    raw_json: &str,
    name: &str,
) -> Option<(&'static dyn ToolAdapter, ParsedHookEvent)> {
    for adapter in ADAPTERS.iter() {
        if adapter.name() == name {
            return adapter
                .parse_hook(raw_json)
                .map(|event| (adapter.as_ref(), event));
        }
    }
    None
}

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn detect_claude_code_input() {
        let json = r#"{"session_id":"s1","cwd":"/tmp","tool_name":"Edit","tool_input":{"file_path":"x.rs"}}"#;
        let result = detect_adapter(json);
        assert!(result.is_some());
        let (adapter, event) = result.unwrap();
        assert_eq!(adapter.name(), "claude-code");
        assert_eq!(event.session_id, "s1");
        assert_eq!(event.tool_name, "Edit");
        assert_eq!(event.reference_time_epoch, None);
    }

    #[test]
    fn detect_codex_input_when_strategy_is_forced() {
        let json = r#"{"session_id":"s1","cwd":"/tmp","hook_event_name":"PostToolUse","tool_name":"Bash","tool_input":{"command":"python test.py"},"tool_result":{"exitCode":0}}"#;
        let result = detect_adapter_by_name(json, "codex-cli");
        assert!(result.is_some());
        let (adapter, event) = result.unwrap();
        assert_eq!(adapter.name(), "codex-cli");
        assert_eq!(event.session_id, "s1");
        assert_eq!(event.tool_name, "Bash");
        assert_eq!(event.tool_response.unwrap()["exitCode"], 0);
    }

    #[test]
    fn forced_strategy_keeps_adapter_selection_isolated() {
        let json = r#"{"session_id":"s1","cwd":"/tmp","tool_name":"Edit","tool_input":{"file_path":"x.rs"}}"#;

        let (adapter, _) = detect_adapter_by_name(json, "codex-cli").unwrap();
        assert_eq!(adapter.name(), "codex-cli");
        let (adapter, _) = detect_adapter_by_name(json, "claude-code").unwrap();
        assert_eq!(adapter.name(), "claude-code");
        assert!(detect_adapter_by_name(json, "missing-adapter").is_none());
    }

    #[test]
    fn detect_unknown_input_returns_none() {
        let json = r#"{"unknown_field": true}"#;
        assert!(detect_adapter(json).is_none());
    }
}