jax_daemon/fuse/

cache.rs

//! LRU cache with TTL for FUSE file contents
//!
//! This cache stores file content and metadata to reduce HTTP API calls.
//! It supports time-based expiration and size-based eviction.

use std::sync::Arc;
use std::time::Duration;

use moka::sync::Cache;
use serde::{Deserialize, Serialize};

/// Cached file content
#[derive(Debug, Clone)]
pub struct CachedContent {
    /// File content bytes
    pub data: Arc<Vec<u8>>,
    /// MIME type
    pub mime_type: String,
}

/// Cached file/directory attributes
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct CachedAttr {
    /// Size in bytes
    pub size: u64,
    /// Is this a directory?
    pub is_dir: bool,
    /// MIME type for files
    pub mime_type: Option<String>,
    /// Modification time (Unix timestamp)
    pub mtime: i64,
}

/// Cached directory listing
#[derive(Debug, Clone)]
pub struct CachedDirEntry {
    pub name: String,
    pub is_dir: bool,
}

/// Configuration for the file cache
#[derive(Debug, Clone)]
pub struct FileCacheConfig {
    /// Maximum cache size in megabytes
    pub max_size_mb: u32,
    /// TTL for cached entries in seconds
    pub ttl_secs: u32,
}

impl Default for FileCacheConfig {
    fn default() -> Self {
        Self {
            max_size_mb: 100,
            ttl_secs: 60,
        }
    }
}

/// LRU cache for FUSE filesystem
#[derive(Clone)]
pub struct FileCache {
    /// Content cache: path → content
    content: Cache<String, CachedContent>,
    /// Attribute cache: path → attributes
    attrs: Cache<String, CachedAttr>,
    /// Directory listing cache: path → entries
    dirs: Cache<String, Vec<CachedDirEntry>>,
    /// Configuration
    config: FileCacheConfig,
}

impl FileCache {
    /// Create a new file cache with the given configuration
    pub fn new(config: FileCacheConfig) -> Self {
        let ttl = Duration::from_secs(config.ttl_secs as u64);
        // Estimate ~1KB average per entry for size calculation
        let max_capacity = (config.max_size_mb as u64) * 1024;

        Self {
            content: Cache::builder()
                .time_to_live(ttl)
                .max_capacity(max_capacity)
                .build(),
            attrs: Cache::builder()
                .time_to_live(ttl)
                .max_capacity(max_capacity * 10) // Attrs are smaller
                .build(),
            dirs: Cache::builder()
                .time_to_live(ttl)
                .max_capacity(max_capacity)
                .build(),
            config,
        }
    }

    /// Get cached content for a path
    pub fn get_content(&self, path: &str) -> Option<CachedContent> {
        self.content.get(&Self::normalize_key(path))
    }

    /// Cache content for a path
    pub fn put_content(&self, path: &str, content: CachedContent) {
        self.content.insert(Self::normalize_key(path), content);
    }

    /// Get cached attributes for a path
    pub fn get_attr(&self, path: &str) -> Option<CachedAttr> {
        self.attrs.get(&Self::normalize_key(path))
    }

    /// Cache attributes for a path
    pub fn put_attr(&self, path: &str, attr: CachedAttr) {
        self.attrs.insert(Self::normalize_key(path), attr);
    }

    /// Get cached directory listing
    pub fn get_dir(&self, path: &str) -> Option<Vec<CachedDirEntry>> {
        self.dirs.get(&Self::normalize_key(path))
    }

    /// Cache directory listing
    pub fn put_dir(&self, path: &str, entries: Vec<CachedDirEntry>) {
        self.dirs.insert(Self::normalize_key(path), entries);
    }

    /// Invalidate a specific path from all caches
    pub fn invalidate(&self, path: &str) {
        let key = Self::normalize_key(path);
        self.content.invalidate(&key);
        self.attrs.invalidate(&key);
        self.dirs.invalidate(&key);
    }

    /// Invalidate all entries under a path prefix
    pub fn invalidate_prefix(&self, prefix: &str) {
        let prefix = Self::normalize_key(prefix);

        // Moka doesn't have prefix invalidation, so we need to iterate
        // For content cache
        self.content.run_pending_tasks();
        // Note: We can't efficiently iterate moka caches, so we just invalidate_all
        // in practice for prefix invalidations
        if prefix == "/" {
            self.invalidate_all();
        }
    }

    /// Invalidate all cached entries
    pub fn invalidate_all(&self) {
        self.content.invalidate_all();
        self.attrs.invalidate_all();
        self.dirs.invalidate_all();
    }

    /// Get current cache statistics
    pub fn stats(&self) -> CacheStats {
        CacheStats {
            content_count: self.content.entry_count(),
            attr_count: self.attrs.entry_count(),
            dir_count: self.dirs.entry_count(),
            max_size_mb: self.config.max_size_mb,
            ttl_secs: self.config.ttl_secs,
        }
    }

    /// Normalize a path to a consistent cache key
    fn normalize_key(path: &str) -> String {
        let path = path.trim();
        if path.is_empty() || path == "/" {
            return "/".to_string();
        }

        let mut key = if path.starts_with('/') {
            path.to_string()
        } else {
            format!("/{}", path)
        };

        if key.len() > 1 && key.ends_with('/') {
            key.pop();
        }

        key
    }
}

impl std::fmt::Debug for FileCache {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        f.debug_struct("FileCache")
            .field("config", &self.config)
            .field("content_count", &self.content.entry_count())
            .field("attr_count", &self.attrs.entry_count())
            .field("dir_count", &self.dirs.entry_count())
            .finish()
    }
}

/// Cache statistics
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct CacheStats {
    pub content_count: u64,
    pub attr_count: u64,
    pub dir_count: u64,
    pub max_size_mb: u32,
    pub ttl_secs: u32,
}

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

    #[test]
    fn test_content_cache() {
        let cache = FileCache::new(FileCacheConfig::default());

        let content = CachedContent {
            data: Arc::new(vec![1, 2, 3]),
            mime_type: "text/plain".to_string(),
        };

        cache.put_content("/foo.txt", content.clone());

        let cached = cache.get_content("/foo.txt").unwrap();
        assert_eq!(cached.data.as_ref(), &[1, 2, 3]);
        assert_eq!(cached.mime_type, "text/plain");
    }

    #[test]
    fn test_attr_cache() {
        let cache = FileCache::new(FileCacheConfig::default());

        let attr = CachedAttr {
            size: 100,
            is_dir: false,
            mime_type: Some("text/plain".to_string()),
            mtime: 1234567890,
        };

        cache.put_attr("/foo.txt", attr.clone());

        let cached = cache.get_attr("/foo.txt").unwrap();
        assert_eq!(cached.size, 100);
        assert!(!cached.is_dir);
    }

    #[test]
    fn test_dir_cache() {
        let cache = FileCache::new(FileCacheConfig::default());

        let entries = vec![
            CachedDirEntry {
                name: "file.txt".to_string(),
                is_dir: false,
            },
            CachedDirEntry {
                name: "subdir".to_string(),
                is_dir: true,
            },
        ];

        cache.put_dir("/", entries.clone());

        let cached = cache.get_dir("/").unwrap();
        assert_eq!(cached.len(), 2);
        assert_eq!(cached[0].name, "file.txt");
    }

    #[test]
    fn test_invalidate() {
        let cache = FileCache::new(FileCacheConfig::default());

        let content = CachedContent {
            data: Arc::new(vec![1, 2, 3]),
            mime_type: "text/plain".to_string(),
        };

        cache.put_content("/foo.txt", content);
        assert!(cache.get_content("/foo.txt").is_some());

        cache.invalidate("/foo.txt");
        assert!(cache.get_content("/foo.txt").is_none());
    }

    #[test]
    fn test_invalidate_all() {
        let cache = FileCache::new(FileCacheConfig::default());

        cache.put_content(
            "/a.txt",
            CachedContent {
                data: Arc::new(vec![1]),
                mime_type: "text/plain".to_string(),
            },
        );
        cache.put_content(
            "/b.txt",
            CachedContent {
                data: Arc::new(vec![2]),
                mime_type: "text/plain".to_string(),
            },
        );

        cache.invalidate_all();

        assert!(cache.get_content("/a.txt").is_none());
        assert!(cache.get_content("/b.txt").is_none());
    }

    #[test]
    fn test_normalize_key() {
        assert_eq!(FileCache::normalize_key(""), "/");
        assert_eq!(FileCache::normalize_key("/"), "/");
        assert_eq!(FileCache::normalize_key("foo"), "/foo");
        assert_eq!(FileCache::normalize_key("/foo"), "/foo");
        assert_eq!(FileCache::normalize_key("/foo/"), "/foo");
    }
}