vtcode_memory/
migration.rs

1//! One-off migration of the legacy, overlapping history stores into the
2//! unified per-session store.
3//!
4//! Legacy inputs:
5//! - `.vtcode/history/session-*.memory.json` → `<session>/derived/memory.json`
6//! - `.vtcode/logs/trajectory-*.jsonl`  → `<session>/derived/trajectory.jsonl`
7//!
8//! Legacy `checkpoints/` are intentionally *not* migrated here: they require a
9//! lossy `Message` → `ThreadEvent` mapping and `/revert` must be rewired first.
10
11use std::path::Path;
12
13use chrono::Utc;
14
15use crate::error::SessionStoreError;
16use crate::{SessionManifest, session_dir};
17
18/// Outcome of a legacy migration run.
19#[derive(Debug, Clone, Default, PartialEq, Eq)]
20pub struct MigrationReport {
21    /// Number of session directories created.
22    pub sessions_created: usize,
23    /// Number of session memory envelopes imported.
24    pub memory_imported: usize,
25    /// Number of trajectory logs imported.
26    pub trajectory_imported: usize,
27    /// Total bytes copied into the unified store.
28    pub bytes_migrated: u64,
29}
30
31/// Migrate legacy history/trajectory stores into the unified session store.
32///
33/// When `remove_legacy` is true, the now-imported `history/` and `logs/`
34/// directories are deleted (the `checkpoints/` directory is preserved).
35pub fn migrate_legacy(workspace: &Path, remove_legacy: bool) -> Result<MigrationReport, SessionStoreError> {
36    let mut report = MigrationReport::default();
37    let vt = workspace.join(".vtcode");
38
39    let history_dir = vt.join("history");
40    if history_dir.is_dir() {
41        for entry in std::fs::read_dir(&history_dir)
42            .map_err(|e| SessionStoreError::io(history_dir.clone(), e))?
43            .filter_map(Result::ok)
44        {
45            let path = entry.path();
46            if path.extension().and_then(|e| e.to_str()) != Some("json") {
47                continue;
48            }
49            let Some(name) = path.file_stem().and_then(|s| s.to_str()) else {
50                continue;
51            };
52            let session_id = name.to_string();
53            let bytes = std::fs::read(&path).map_err(|e| SessionStoreError::io(path.clone(), e))?;
54            let dir = session_dir(workspace, &session_id);
55            std::fs::create_dir_all(dir.join(crate::DERIVED_DIR))
56                .map_err(|e| SessionStoreError::CreateDir { path: dir.clone(), source: e })?;
57            let dest = dir.join(crate::DERIVED_DIR).join("memory.json");
58            std::fs::write(&dest, &bytes).map_err(|e| SessionStoreError::io(dest, e))?;
59            write_manifest(&dir, &session_id, &path, "completed")?;
60            report.sessions_created += 1;
61            report.memory_imported += 1;
62            report.bytes_migrated += bytes.len() as u64;
63        }
64    }
65
66    let logs_dir = vt.join("logs");
67    if logs_dir.is_dir() {
68        for entry in std::fs::read_dir(&logs_dir)
69            .map_err(|e| SessionStoreError::io(logs_dir.clone(), e))?
70            .filter_map(Result::ok)
71        {
72            let path = entry.path();
73            if path.extension().and_then(|e| e.to_str()) != Some("jsonl") {
74                continue;
75            }
76            let Some(name) = path.file_stem().and_then(|s| s.to_str()) else {
77                continue;
78            };
79            // `trajectory-<ts>` → session id `traj-<ts>` to avoid colliding with
80            // history session ids.
81            let session_id = format!("traj-{}", name.trim_start_matches("trajectory-"));
82            let bytes = std::fs::read(&path).map_err(|e| SessionStoreError::io(path.clone(), e))?;
83            let dir = session_dir(workspace, &session_id);
84            std::fs::create_dir_all(dir.join(crate::DERIVED_DIR))
85                .map_err(|e| SessionStoreError::CreateDir { path: dir.clone(), source: e })?;
86            let dest = dir.join(crate::DERIVED_DIR).join("trajectory.jsonl");
87            std::fs::write(&dest, &bytes).map_err(|e| SessionStoreError::io(dest, e))?;
88            write_manifest(&dir, &session_id, &path, "completed")?;
89            report.sessions_created += 1;
90            report.trajectory_imported += 1;
91            report.bytes_migrated += bytes.len() as u64;
92        }
93    }
94
95    if remove_legacy {
96        let freed = crate::retention::gc_legacy(workspace)?;
97        let _ = freed;
98    }
99
100    Ok(report)
101}
102
103fn write_manifest(dir: &Path, session_id: &str, source: &Path, status: &str) -> Result<(), SessionStoreError> {
104    let ts = source
105        .metadata()
106        .ok()
107        .and_then(|m| m.modified().ok())
108        .map(|t| chrono::DateTime::<Utc>::from(t).to_rfc3339())
109        .unwrap_or_else(|| Utc::now().to_rfc3339());
110    let mut manifest = SessionManifest::new(session_id);
111    manifest.created_at = ts.clone();
112    manifest.updated_at = ts;
113    manifest.status = status.to_string();
114    let path = dir.join("manifest.json");
115    let bytes = serde_json::to_vec(&manifest)?;
116    std::fs::write(&path, bytes).map_err(|e| SessionStoreError::io(path, e))?;
117    Ok(())
118}
vtcode_memory/migration.rs

vtcode_memory/
migration.rs
