Skip to main content

sqlite_graphrag/commands/
history.rs

1use crate::errors::AppError;
2use crate::i18n::erros;
3use crate::output;
4use crate::paths::AppPaths;
5use crate::storage::connection::open_ro;
6use crate::storage::memories;
7use rusqlite::params;
8use serde::Serialize;
9
10#[derive(clap::Args)]
11pub struct HistoryArgs {
12    #[arg(long)]
13    pub name: String,
14    #[arg(long, default_value = "global")]
15    pub namespace: Option<String>,
16    #[arg(long, hide = true, help = "No-op; JSON is always emitted on stdout")]
17    pub json: bool,
18    #[arg(long, env = "SQLITE_GRAPHRAG_DB_PATH")]
19    pub db: Option<String>,
20}
21
22#[derive(Serialize)]
23struct HistoryVersion {
24    version: i64,
25    name: String,
26    #[serde(rename = "type")]
27    memory_type: String,
28    description: String,
29    body: String,
30    metadata: String,
31    change_reason: String,
32    changed_by: Option<String>,
33    created_at: i64,
34    created_at_iso: String,
35}
36
37#[derive(Serialize)]
38struct HistoryResponse {
39    name: String,
40    namespace: String,
41    versions: Vec<HistoryVersion>,
42    /// Tempo total de execução em milissegundos desde início do handler até serialização.
43    elapsed_ms: u64,
44}
45
46pub fn run(args: HistoryArgs) -> Result<(), AppError> {
47    let inicio = std::time::Instant::now();
48    let namespace = crate::namespace::resolve_namespace(args.namespace.as_deref())?;
49    let paths = AppPaths::resolve(args.db.as_deref())?;
50    let conn = open_ro(&paths.db)?;
51
52    let (memory_id, _, _) = memories::find_by_name(&conn, &namespace, &args.name)?
53        .ok_or_else(|| AppError::NotFound(erros::memoria_nao_encontrada(&args.name, &namespace)))?;
54
55    let mut stmt = conn.prepare(
56        "SELECT version, name, type, description, body, metadata,
57                change_reason, changed_by, created_at
58         FROM memory_versions
59         WHERE memory_id = ?1
60         ORDER BY version ASC",
61    )?;
62
63    let versions = stmt
64        .query_map(params![memory_id], |r| {
65            let created_at: i64 = r.get(8)?;
66            let created_at_iso = crate::tz::epoch_para_iso(created_at);
67            Ok(HistoryVersion {
68                version: r.get(0)?,
69                name: r.get(1)?,
70                memory_type: r.get(2)?,
71                description: r.get(3)?,
72                body: r.get(4)?,
73                metadata: r.get(5)?,
74                change_reason: r.get(6)?,
75                changed_by: r.get(7)?,
76                created_at,
77                created_at_iso,
78            })
79        })?
80        .collect::<Result<Vec<_>, _>>()?;
81
82    output::emit_json(&HistoryResponse {
83        name: args.name,
84        namespace,
85        versions,
86        elapsed_ms: inicio.elapsed().as_millis() as u64,
87    })?;
88
89    Ok(())
90}
91
92#[cfg(test)]
93mod testes {
94    #[test]
95    fn epoch_zero_gera_iso_valido() {
96        // epoch_para_iso usa chrono-tz com offset explícito (+00:00 para UTC)
97        let iso = crate::tz::epoch_para_iso(0);
98        assert!(iso.starts_with("1970-01-01T00:00:00"), "obtido: {iso}");
99        assert!(iso.contains("00:00"), "deve conter offset, obtido: {iso}");
100    }
101
102    #[test]
103    fn epoch_tipico_gera_iso_rfc3339() {
104        let iso = crate::tz::epoch_para_iso(1_745_000_000);
105        assert!(!iso.is_empty(), "created_at_iso não deve ser vazio");
106        assert!(iso.contains('T'), "created_at_iso deve conter separador T");
107        // Com UTC o offset é +00:00; verifica formato geral sem depender do fuso global
108        assert!(
109            iso.contains('+') || iso.contains('-'),
110            "deve conter sinal de offset, obtido: {iso}"
111        );
112    }
113
114    #[test]
115    fn epoch_invalido_retorna_fallback() {
116        let iso = crate::tz::epoch_para_iso(i64::MIN);
117        assert!(
118            !iso.is_empty(),
119            "epoch inválido deve retornar fallback não-vazio"
120        );
121    }
122}