remem-ai 0.5.11

//! Raw archive layer — captures every user/assistant turn regardless of
//! whether summarize/promote choose to keep it.
//!
//! Spec: SPEC-raw-archive-vs-curated-memory-2026-04-22.md

use anyhow::Result;
use rusqlite::{params, Connection};

pub const ROLE_USER: &str = "user";
pub const ROLE_ASSISTANT: &str = "assistant";

pub const SOURCE_TRANSCRIPT: &str = "transcript";
pub const SOURCE_HOOK: &str = "hook";
pub const SOURCE_MANUAL: &str = "manual";

#[derive(Debug, Clone)]
pub struct RawMessage {
    pub id: i64,
    pub session_id: String,
    pub project: String,
    pub role: String,
    pub content: String,
    pub source: String,
    pub branch: Option<String>,
    pub cwd: Option<String>,
    pub created_at_epoch: i64,
}

/// Exact byte-for-byte hash of the raw message content. Distinct from
/// `memory::promote::slug::content_hash`, which normalizes whitespace/case for
/// semantic dedup of curated memories.
fn exact_content_hash(content: &str) -> String {
    format!("{:016x}", crate::db::deterministic_hash(content.as_bytes()))
}

/// Insert one raw message. UNIQUE(project, session_id, role, content_hash)
/// makes this idempotent across repeated Stop-hook drains of the same
/// transcript while still preserving identical text spoken in different
/// sessions (issue #237).
/// Returns the row id of the existing or newly inserted message, or None
/// when the content is empty.
/// Outcome of a raw-message insert attempt.
///
/// `inserted == false` means a row with the same `(project, session_id, role,
/// content_hash)` already existed and `id` points at the pre-existing row
/// rather than a newly created one.
#[derive(Debug, Clone, Copy)]
pub struct RawInsertOutcome {
    pub id: i64,
    pub inserted: bool,
}

#[derive(Debug, Clone, Default, PartialEq, Eq)]
pub struct RawIngestReport {
    pub inserted: usize,
    pub duplicates: usize,
    pub empty_messages: usize,
    pub skipped_messages: usize,
    pub parse_errors: usize,
    pub insert_errors: usize,
    pub read_error: Option<String>,
}

impl RawIngestReport {
    pub fn has_failures(&self) -> bool {
        self.read_error.is_some() || self.parse_errors > 0 || self.insert_errors > 0
    }

    pub fn failure_kind(&self) -> Option<&'static str> {
        match (
            self.read_error.is_some(),
            self.parse_errors > 0,
            self.insert_errors > 0,
        ) {
            (true, false, false) => Some("read_error"),
            (false, true, false) => Some("parse_errors"),
            (false, false, true) => Some("insert_errors"),
            (true, _, _) | (_, true, true) => Some("mixed_errors"),
            (false, false, false) => None,
        }
    }

    fn failure_message(&self) -> String {
        if let Some(error) = &self.read_error {
            return error.clone();
        }
        format!(
            "parse_errors={} insert_errors={}",
            self.parse_errors, self.insert_errors
        )
    }
}

pub fn insert_raw_message(
    conn: &Connection,
    session_id: &str,
    project: &str,
    role: &str,
    content: &str,
    source: &str,
    branch: Option<&str>,
    cwd: Option<&str>,
) -> Result<Option<RawInsertOutcome>> {
    let trimmed = content.trim();
    if trimmed.is_empty() {
        return Ok(None);
    }
    let hash = exact_content_hash(trimmed);
    let now = chrono::Utc::now().timestamp();

    let inserted = conn.execute(
        "INSERT INTO raw_messages \
         (session_id, project, role, content, content_hash, source, branch, cwd, created_at_epoch) \
         VALUES (?1, ?2, ?3, ?4, ?5, ?6, ?7, ?8, ?9) \
         ON CONFLICT(project, session_id, role, content_hash) DO NOTHING",
        params![session_id, project, role, trimmed, hash, source, branch, cwd, now],
    )?;

    if inserted > 0 {
        Ok(Some(RawInsertOutcome {
            id: conn.last_insert_rowid(),
            inserted: true,
        }))
    } else {
        let existing: i64 = conn.query_row(
            "SELECT id FROM raw_messages \
             WHERE project = ?1 AND session_id = ?2 AND role = ?3 AND content_hash = ?4",
            params![project, session_id, role, hash],
            |row| row.get(0),
        )?;
        Ok(Some(RawInsertOutcome {
            id: existing,
            inserted: false,
        }))
    }
}

/// Drain a Claude Code transcript JSONL file into raw_messages.
pub fn drain_transcript(
    conn: &Connection,
    transcript_path: &str,
    session_id: &str,
    project: &str,
    branch: Option<&str>,
    cwd: Option<&str>,
) -> Result<RawIngestReport> {
    let content = match std::fs::read_to_string(transcript_path) {
        Ok(content) => content,
        Err(error) => {
            let report = RawIngestReport {
                read_error: Some(format!(
                    "read transcript {} failed: {}",
                    transcript_path, error
                )),
                ..RawIngestReport::default()
            };
            crate::log::warn(
                "raw-archive",
                report
                    .read_error
                    .as_deref()
                    .unwrap_or("read transcript failed"),
            );
            record_raw_ingest_failure(
                conn,
                session_id,
                project,
                SOURCE_TRANSCRIPT,
                Some(transcript_path),
                &report,
            )?;
            return Ok(report);
        }
    };

    let mut report = RawIngestReport::default();
    for line in content.lines() {
        let Ok(value) = serde_json::from_str::<serde_json::Value>(line) else {
            report.parse_errors += 1;
            continue;
        };
        let role = match value["type"].as_str() {
            Some("user") => ROLE_USER,
            Some("assistant") => ROLE_ASSISTANT,
            _ => {
                report.skipped_messages += 1;
                continue;
            }
        };
        let text = extract_message_text(&value);
        if text.trim().is_empty() {
            report.empty_messages += 1;
            continue;
        }

        match insert_raw_message(
            conn,
            session_id,
            project,
            role,
            &text,
            SOURCE_TRANSCRIPT,
            branch,
            cwd,
        ) {
            Ok(Some(outcome)) if outcome.inserted => report.inserted += 1,
            Ok(Some(_)) => report.duplicates += 1,
            Ok(None) => report.empty_messages += 1,
            Err(error) => {
                report.insert_errors += 1;
                crate::log::warn(
                    "raw-archive",
                    &format!("insert raw message failed: {}", error),
                );
            }
        }
    }
    if report.has_failures() {
        record_raw_ingest_failure(
            conn,
            session_id,
            project,
            SOURCE_TRANSCRIPT,
            Some(transcript_path),
            &report,
        )?;
    }
    Ok(report)
}

pub fn record_raw_ingest_failure(
    conn: &Connection,
    session_id: &str,
    project: &str,
    source: &str,
    transcript_path: Option<&str>,
    report: &RawIngestReport,
) -> Result<()> {
    let Some(kind) = report.failure_kind() else {
        return Ok(());
    };
    conn.execute(
        "INSERT INTO raw_ingest_failures
         (project, session_id, source, transcript_path, error_kind, error_message,
          inserted, duplicates, parse_errors, insert_errors, created_at_epoch)
         VALUES (?1, ?2, ?3, ?4, ?5, ?6, ?7, ?8, ?9, ?10, ?11)",
        params![
            project,
            session_id,
            source,
            transcript_path,
            kind,
            crate::db::truncate_str(&report.failure_message(), 1000),
            report.inserted as i64,
            report.duplicates as i64,
            report.parse_errors as i64,
            report.insert_errors as i64,
            chrono::Utc::now().timestamp()
        ],
    )?;
    Ok(())
}

fn extract_message_text(value: &serde_json::Value) -> String {
    let content = &value["message"]["content"];
    if let Some(array) = content.as_array() {
        let parts: Vec<String> = array
            .iter()
            .filter_map(|entry| {
                match entry["type"].as_str() {
                    Some("text") => entry["text"].as_str().map(|s| s.to_string()),
                    // user messages can carry plain strings under `content` too
                    _ => None,
                }
            })
            .collect();
        return parts.join("\n");
    }
    if let Some(text) = content.as_str() {
        return text.to_string();
    }
    String::new()
}

#[derive(Debug, Clone)]
pub struct RawSearchRequest {
    pub query: String,
    pub project: Option<String>,
    pub branch: Option<String>,
    pub role: Option<String>,
    pub limit: i64,
    pub offset: i64,
}

pub fn search_raw_messages(conn: &Connection, req: &RawSearchRequest) -> Result<Vec<RawMessage>> {
    let limit = req.limit.max(1);
    let offset = req.offset.max(0);
    let query = req.query.trim();
    if query.is_empty() {
        return Ok(vec![]);
    }

    let mut sql = String::from(
        "SELECT r.id, r.session_id, r.project, r.role, r.content, r.source, \
                r.branch, r.cwd, r.created_at_epoch \
         FROM raw_messages r \
         JOIN raw_messages_fts f ON f.rowid = r.id \
         WHERE raw_messages_fts MATCH ?1",
    );
    let mut binds: Vec<Box<dyn rusqlite::types::ToSql>> = vec![Box::new(fts_query(query))];

    if let Some(project) = req.project.as_deref() {
        sql.push_str(" AND r.project = ?");
        sql.push_str(&(binds.len() + 1).to_string());
        binds.push(Box::new(project.to_string()));
    }
    if let Some(branch) = req.branch.as_deref() {
        let idx = binds.len() + 1;
        sql.push_str(&format!(" AND (r.branch = ?{idx} OR r.branch IS NULL)"));
        binds.push(Box::new(branch.to_string()));
    }
    if let Some(role) = req.role.as_deref() {
        sql.push_str(" AND r.role = ?");
        sql.push_str(&(binds.len() + 1).to_string());
        binds.push(Box::new(role.to_string()));
    }

    sql.push_str(&format!(
        " ORDER BY r.created_at_epoch DESC LIMIT {} OFFSET {}",
        limit, offset
    ));

    let mut stmt = conn.prepare(&sql)?;
    let rows = stmt.query_map(
        rusqlite::params_from_iter(crate::db::to_sql_refs(&binds)),
        |row| {
            Ok(RawMessage {
                id: row.get(0)?,
                session_id: row.get(1)?,
                project: row.get(2)?,
                role: row.get(3)?,
                content: row.get(4)?,
                source: row.get(5)?,
                branch: row.get(6)?,
                cwd: row.get(7)?,
                created_at_epoch: row.get(8)?,
            })
        },
    )?;

    let mut out = Vec::new();
    for row in rows {
        out.push(row?);
    }
    Ok(out)
}

fn fts_query(query: &str) -> String {
    // Wrap each token in quotes so we use phrase matching (robust against
    // punctuation that trigram tokenizer would otherwise choke on).
    let cleaned: Vec<String> = query
        .split_whitespace()
        .filter(|token| !token.is_empty())
        .map(|token| format!("\"{}\"", token.replace('\"', "\"\"")))
        .collect();
    if cleaned.is_empty() {
        format!("\"{}\"", query.replace('\"', "\"\""))
    } else {
        cleaned.join(" ")
    }
}

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

    fn setup_conn() -> Connection {
        let conn = Connection::open_in_memory().unwrap();
        crate::migrate::run_migrations(&conn).unwrap();
        conn
    }

    fn write_temp_transcript(name: &str, content: &str) -> Result<std::path::PathBuf> {
        let path = std::env::temp_dir().join(format!(
            "remem-{name}-{}-{}.jsonl",
            std::process::id(),
            chrono::Utc::now().timestamp_nanos_opt().unwrap_or_default()
        ));
        std::fs::write(&path, content)?;
        Ok(path)
    }

    fn raw_ingest_failure_count(conn: &Connection) -> Result<i64> {
        Ok(
            conn.query_row("SELECT COUNT(*) FROM raw_ingest_failures", [], |row| {
                row.get(0)
            })?,
        )
    }

    #[test]
    fn insert_is_idempotent_per_session_role_content() {
        let conn = setup_conn();
        let id1 = insert_raw_message(
            &conn,
            "s1",
            "/proj",
            ROLE_USER,
            "hello world",
            SOURCE_HOOK,
            None,
            None,
        )
        .unwrap()
        .expect("first insert returns Some");
        // Same session + same text => deduped onto the existing row.
        let id2 = insert_raw_message(
            &conn,
            "s1",
            "/proj",
            ROLE_USER,
            "hello world",
            SOURCE_HOOK,
            None,
            None,
        )
        .unwrap()
        .expect("second insert returns Some");
        assert_eq!(id1.id, id2.id);
        assert!(id1.inserted, "first call must mark inserted");
        assert!(!id2.inserted, "second call must mark not-inserted");
        let count: i64 = conn
            .query_row("SELECT COUNT(*) FROM raw_messages", [], |row| row.get(0))
            .unwrap();
        assert_eq!(count, 1);
    }

    /// Regression for #237: the same text spoken in two different sessions must
    /// keep BOTH turns. The old UNIQUE(project, role, content_hash) globally
    /// deduped across sessions and silently dropped the second turn.
    #[test]
    fn identical_text_across_sessions_keeps_both_turns() {
        let conn = setup_conn();
        let id1 = insert_raw_message(
            &conn,
            "s1",
            "/proj",
            ROLE_USER,
            "let's deploy the service",
            SOURCE_HOOK,
            None,
            None,
        )
        .unwrap()
        .expect("first insert returns Some");
        let id2 = insert_raw_message(
            &conn,
            "s2",
            "/proj",
            ROLE_USER,
            "let's deploy the service",
            SOURCE_HOOK,
            None,
            None,
        )
        .unwrap()
        .expect("second insert returns Some");

        assert!(id1.inserted, "first session turn must be inserted");
        assert!(id2.inserted, "second session turn must also be inserted");
        assert_ne!(id1.id, id2.id, "the two sessions must keep distinct rows");

        let count: i64 = conn
            .query_row("SELECT COUNT(*) FROM raw_messages", [], |row| row.get(0))
            .unwrap();
        assert_eq!(count, 2, "both session turns must be preserved");
    }

    #[test]
    fn empty_content_is_skipped() {
        let conn = setup_conn();
        let id = insert_raw_message(
            &conn,
            "s1",
            "/proj",
            ROLE_USER,
            "   \n\t  ",
            SOURCE_HOOK,
            None,
            None,
        )
        .unwrap();
        assert!(id.is_none());
    }

    #[test]
    fn fts_finds_inserted_content() {
        let conn = setup_conn();
        insert_raw_message(
            &conn,
            "s1",
            "/proj",
            ROLE_USER,
            "帮我看看 VPS RackNerd 的价格",
            SOURCE_HOOK,
            None,
            None,
        )
        .unwrap();
        let hits = search_raw_messages(
            &conn,
            &RawSearchRequest {
                query: "RackNerd".to_string(),
                project: Some("/proj".to_string()),
                branch: None,
                role: None,
                limit: 10,
                offset: 0,
            },
        )
        .unwrap();
        assert_eq!(hits.len(), 1);
        assert!(hits[0].content.contains("RackNerd"));
    }

    #[test]
    fn search_branch_filter_keeps_matching_and_branchless_raw_messages() {
        let conn = setup_conn();
        insert_raw_message(
            &conn,
            "s-main",
            "/proj",
            ROLE_USER,
            "shared needle on main",
            SOURCE_HOOK,
            Some("main"),
            None,
        )
        .unwrap();
        insert_raw_message(
            &conn,
            "s-feature",
            "/proj",
            ROLE_USER,
            "shared needle on feature",
            SOURCE_HOOK,
            Some("feature"),
            None,
        )
        .unwrap();
        insert_raw_message(
            &conn,
            "s-branchless",
            "/proj",
            ROLE_USER,
            "shared needle without branch",
            SOURCE_HOOK,
            None,
            None,
        )
        .unwrap();

        let hits = search_raw_messages(
            &conn,
            &RawSearchRequest {
                query: "needle".to_string(),
                project: Some("/proj".to_string()),
                branch: Some("main".to_string()),
                role: None,
                limit: 10,
                offset: 0,
            },
        )
        .unwrap();
        let branches: Vec<Option<String>> = hits.into_iter().map(|hit| hit.branch).collect();

        assert!(branches.contains(&Some("main".to_string())));
        assert!(branches.contains(&None));
        assert!(
            !branches.contains(&Some("feature".to_string())),
            "{branches:?}"
        );
    }

    #[test]
    fn drain_transcript_counts_parse_errors_and_records_failure() -> Result<()> {
        let conn = setup_conn();
        let path = write_temp_transcript(
            "raw-parse-error",
            format!(
                "{}\nnot json\n",
                r#"{"type":"assistant","message":{"content":[{"type":"text","text":"kept message"}]}}"#
            )
            .as_str(),
        )?;

        let report = drain_transcript(
            &conn,
            path.to_string_lossy().as_ref(),
            "session-parse",
            "/proj",
            None,
            None,
        )?;

        assert_eq!(report.inserted, 1);
        assert_eq!(report.parse_errors, 1);
        assert_eq!(raw_ingest_failure_count(&conn)?, 1);
        let (kind, parse_errors): (String, i64) = conn.query_row(
            "SELECT error_kind, parse_errors FROM raw_ingest_failures",
            [],
            |row| Ok((row.get(0)?, row.get(1)?)),
        )?;
        assert_eq!(kind, "parse_errors");
        assert_eq!(parse_errors, 1);
        std::fs::remove_file(path)?;
        Ok(())
    }

    #[test]
    fn drain_transcript_counts_insert_errors_and_records_failure() -> Result<()> {
        let conn = setup_conn();
        conn.execute_batch(
            "CREATE TRIGGER fail_raw_archive_insert
             BEFORE INSERT ON raw_messages
             BEGIN
                 SELECT RAISE(FAIL, 'raw insert failed');
             END;",
        )?;
        let path = write_temp_transcript(
            "raw-insert-error",
            r#"{"type":"assistant","message":{"content":[{"type":"text","text":"cannot insert"}]}}"#,
        )?;

        let report = drain_transcript(
            &conn,
            path.to_string_lossy().as_ref(),
            "session-insert",
            "/proj",
            None,
            None,
        )?;

        assert_eq!(report.inserted, 0);
        assert_eq!(report.insert_errors, 1);
        assert_eq!(raw_ingest_failure_count(&conn)?, 1);
        let (kind, insert_errors): (String, i64) = conn.query_row(
            "SELECT error_kind, insert_errors FROM raw_ingest_failures",
            [],
            |row| Ok((row.get(0)?, row.get(1)?)),
        )?;
        assert_eq!(kind, "insert_errors");
        assert_eq!(insert_errors, 1);
        std::fs::remove_file(path)?;
        Ok(())
    }
}