tmuxido/src/cache.rs

use anyhow::{Context, Result};
use serde::{Deserialize, Serialize};
use std::collections::HashMap;
use std::fs;
use std::path::{Path, PathBuf};
use std::time::{SystemTime, UNIX_EPOCH};

#[derive(Debug, Serialize, Deserialize)]
pub struct ProjectCache {
    pub projects: Vec<PathBuf>,
    pub last_updated: u64,
    /// mtime de cada diretório visitado durante o scan.
    /// Usado para detectar mudanças incrementais sem precisar varrer tudo.
    #[serde(default)]
    pub dir_mtimes: HashMap<PathBuf, u64>,
}

fn now_secs() -> u64 {
    SystemTime::now()
        .duration_since(UNIX_EPOCH)
        .unwrap()
        .as_secs()
}

fn mtime_secs(time: SystemTime) -> u64 {
    time.duration_since(UNIX_EPOCH).unwrap_or_default().as_secs()
}

/// Retorna o subconjunto mínimo de diretórios: aqueles que não têm nenhum
/// ancestral também na lista. Evita rescanear a mesma subárvore duas vezes.
fn minimal_roots(dirs: &[PathBuf]) -> Vec<PathBuf> {
    dirs.iter()
        .filter(|dir| !dirs.iter().any(|other| other != *dir && dir.starts_with(other)))
        .cloned()
        .collect()
}

impl ProjectCache {
    pub fn new(projects: Vec<PathBuf>, dir_mtimes: HashMap<PathBuf, u64>) -> Self {
        Self {
            projects,
            last_updated: now_secs(),
            dir_mtimes,
        }
    }

    pub fn cache_path() -> Result<PathBuf> {
        let cache_dir = dirs::cache_dir()
            .context("Could not determine cache directory")?
            .join("tmuxido");

        fs::create_dir_all(&cache_dir)
            .with_context(|| format!("Failed to create cache directory: {}", cache_dir.display()))?;

        Ok(cache_dir.join("projects.json"))
    }

    pub fn load() -> Result<Option<Self>> {
        let cache_path = Self::cache_path()?;

        if !cache_path.exists() {
            return Ok(None);
        }

        let content = fs::read_to_string(&cache_path)
            .with_context(|| format!("Failed to read cache file: {}", cache_path.display()))?;

        let cache: ProjectCache = serde_json::from_str(&content)
            .with_context(|| format!("Failed to parse cache file: {}", cache_path.display()))?;

        Ok(Some(cache))
    }

    pub fn save(&self) -> Result<()> {
        let cache_path = Self::cache_path()?;

        let content = serde_json::to_string_pretty(self)
            .context("Failed to serialize cache")?;

        fs::write(&cache_path, content)
            .with_context(|| format!("Failed to write cache file: {}", cache_path.display()))?;

        Ok(())
    }

    /// Valida e atualiza o cache de forma incremental.
    ///
    /// 1. Remove projetos cujo `.git` não existe mais.
    /// 2. Detecta diretórios com mtime alterado.
    /// 3. Resscaneia apenas as subárvores mínimas que mudaram.
    ///
    /// Retorna `true` se o cache foi modificado.
    /// Retorna `false` com `dir_mtimes` vazio (cache antigo) — chamador deve fazer rescan completo.
    pub fn validate_and_update(
        &mut self,
        scan_fn: &dyn Fn(&Path) -> Result<(Vec<PathBuf>, HashMap<PathBuf, u64>)>,
    ) -> Result<bool> {
        let mut changed = false;

        // Passo 1: remover projetos cujo .git não existe mais
        let before = self.projects.len();
        self.projects.retain(|p| p.join(".git").exists());
        if self.projects.len() != before {
            changed = true;
        }

        // Sem fingerprints = cache no formato antigo; sinaliza ao chamador
        if self.dir_mtimes.is_empty() {
            return Ok(changed);
        }

        // Passo 2: encontrar diretórios com mtime diferente do armazenado
        let changed_dirs: Vec<PathBuf> = self
            .dir_mtimes
            .iter()
            .filter(|(dir, stored_mtime)| {
                fs::metadata(dir)
                    .and_then(|m| m.modified())
                    .map(|t| mtime_secs(t) != **stored_mtime)
                    .unwrap_or(true) // diretório sumiu = tratar como mudança
            })
            .map(|(dir, _)| dir.clone())
            .collect();

        if changed_dirs.is_empty() {
            return Ok(changed);
        }

        // Passo 3: resscanear apenas as raízes mínimas das subárvores alteradas
        for root in minimal_roots(&changed_dirs) {
            eprintln!("Rescanning: {}", root.display());

            // Remover entradas antigas desta subárvore
            self.projects.retain(|p| !p.starts_with(&root));
            self.dir_mtimes.retain(|d, _| !d.starts_with(&root));

            // Resscanear e mesclar
            let (new_projects, new_fingerprints) = scan_fn(&root)?;
            self.projects.extend(new_projects);
            self.dir_mtimes.extend(new_fingerprints);
            changed = true;
        }

        if changed {
            self.projects.sort();
            self.projects.dedup();
            self.last_updated = now_secs();
        }

        Ok(changed)
    }

    pub fn age_in_seconds(&self) -> u64 {
        now_secs().saturating_sub(self.last_updated)
    }
}