vtcode-core 0.99.2

Core library for VT Code - a Rust-based terminal coding agent
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

use std::time::Duration;

use crate::cache::{CacheKey, EvictionPolicy, UnifiedCache};

/// Cache key for validation results.
#[derive(Debug, Clone, Hash, PartialEq, Eq)]
struct ValidationKey {
    tool: String,
    args_hash: u64,
}

impl CacheKey for ValidationKey {
    fn to_cache_key(&self) -> String {
        format!("{}:{}", self.tool, self.args_hash)
    }
}

/// Cache for validation results to avoid repetitive checks for the same tool/args combination.
pub struct ValidationCache {
    cache: UnifiedCache<ValidationKey, bool>,
}

impl ValidationCache {
    /// Create a new validation cache with the specified TTL.
    pub fn new(ttl: Duration) -> Self {
        Self {
            cache: UnifiedCache::new(1000, ttl, EvictionPolicy::TtlOnly),
        }
    }

    /// Check if a validation result is cached and valid.
    pub fn check(&self, tool: &str, args_hash: u64) -> Option<bool> {
        let key = ValidationKey {
            tool: tool.to_string(),
            args_hash,
        };
        self.cache.get_owned(&key)
    }

    /// Insert a validation result into the cache.
    pub fn insert(&self, tool: &str, args_hash: u64, result: bool) {
        let key = ValidationKey {
            tool: tool.to_string(),
            args_hash,
        };
        self.cache.insert(key, result, 1); // bool is tiny
    }
}

impl Default for ValidationCache {
    fn default() -> Self {
        Self::new(Duration::from_secs(300)) // 5 minutes default
    }
}