agentvfs 0.1.6

Virtual filesystem CLI backed by embedded databases for AI agents
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79

//! Database compaction command.

use clap::Args;
use serde::Serialize;

use crate::commands::Output;
use crate::error::Result;
use crate::vault::VaultManager;

#[derive(Args)]
pub struct CompactArgs {}

#[derive(Serialize)]
struct CompactOutput {
    size_before: u64,
    size_after: u64,
    bytes_freed: u64,
    size_before_human: String,
    size_after_human: String,
    bytes_freed_human: String,
}

impl std::fmt::Display for CompactOutput {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        write!(
            f,
            "Compacted database: {} -> {} (freed {})",
            self.size_before_human, self.size_after_human, self.bytes_freed_human
        )
    }
}

fn format_size(bytes: u64) -> String {
    const KB: u64 = 1024;
    const MB: u64 = KB * 1024;
    const GB: u64 = MB * 1024;

    if bytes >= GB {
        format!("{:.2} GB", bytes as f64 / GB as f64)
    } else if bytes >= MB {
        format!("{:.2} MB", bytes as f64 / MB as f64)
    } else if bytes >= KB {
        format!("{:.2} KB", bytes as f64 / KB as f64)
    } else {
        format!("{} bytes", bytes)
    }
}

pub fn run(_args: CompactArgs, output: &Output, vault: Option<String>) -> Result<()> {
    let manager = VaultManager::new()?;
    let backend = match vault {
        Some(name) => manager.open(&name)?,
        None => manager.open_current()?,
    };

    // Get size before
    let size_before = backend.get_db_size()?;

    // Run VACUUM
    backend.vacuum()?;

    // Get size after
    let size_after = backend.get_db_size()?;

    let bytes_freed = size_before.saturating_sub(size_after);

    let result = CompactOutput {
        size_before,
        size_after,
        bytes_freed,
        size_before_human: format_size(size_before),
        size_after_human: format_size(size_after),
        bytes_freed_human: format_size(bytes_freed),
    };

    output.print(&result);

    Ok(())
}