lisette_semantics/cache/

go_stdlib.rs

use rustc_hash::FxHashMap as HashMap;
use std::fs;
use std::path::PathBuf;

use serde::{Deserialize, Serialize};
use stdlib::{Target, get_go_stdlib_typedef};

use super::types::CachedDefinition;
use super::{COMPILER_VERSION_HASH, GO_STDLIB_HASH};
use crate::checker::registration::extract_package_directive;
use crate::store::Store;
use syntax::program::File;

#[derive(Serialize, Deserialize)]
pub struct GoStdlibCache {
    pub content_hash: u64,
    pub compiler_version: u64,
    pub modules: HashMap<String, GoModuleCache>,
}

#[derive(Serialize, Deserialize)]
pub struct GoModuleCache {
    pub definitions: HashMap<String, CachedDefinition>,
    /// Go module imports (e.g., `["go:io", "go:sync"]`).
    pub go_imports: Vec<String>,
}

fn cache_file_name(target: Target) -> String {
    format!("stdlib_defs_{}.bin", target.cache_segment())
}

fn cache_path(target: Target) -> Option<PathBuf> {
    let home = std::env::var("HOME").ok()?;
    Some(
        PathBuf::from(home)
            .join(".lisette")
            .join("cache")
            .join(cache_file_name(target)),
    )
}

pub fn try_load_go_stdlib_cache(target: Target) -> Option<GoStdlibCache> {
    let path = cache_path(target)?;
    let bytes = fs::read(&path).ok()?;
    let cache: GoStdlibCache = match bincode::deserialize(&bytes) {
        Ok(cache) => cache,
        Err(_) => {
            let _ = fs::remove_file(&path);
            return None;
        }
    };

    if cache.content_hash != GO_STDLIB_HASH || cache.compiler_version != COMPILER_VERSION_HASH {
        let _ = fs::remove_file(&path);
        return None;
    }

    Some(cache)
}

pub fn save_go_stdlib_cache(store: &Store, go_module_ids: &[String], target: Target) {
    let Some(path) = cache_path(target) else {
        return;
    };

    let mut modules = HashMap::default();
    // Go definitions don't reference files, so file_id_to_index is always empty.
    let empty_file_map = HashMap::default();
    for module_id in go_module_ids {
        let Some(module) = store.get_module(module_id) else {
            continue;
        };
        let definitions: HashMap<String, CachedDefinition> = module
            .definitions
            .iter()
            .map(|(name, definition)| {
                (
                    name.to_string(),
                    CachedDefinition::from_definition(definition, &empty_file_map),
                )
            })
            .collect();

        let go_imports = get_go_imports_from_source(module_id, target);

        modules.insert(
            module_id.clone(),
            GoModuleCache {
                definitions,
                go_imports,
            },
        );
    }

    let cache = GoStdlibCache {
        content_hash: GO_STDLIB_HASH,
        compiler_version: COMPILER_VERSION_HASH,
        modules,
    };

    let Ok(bytes) = bincode::serialize(&cache) else {
        return;
    };

    let Some(parent) = path.parent() else {
        return;
    };
    let _ = fs::create_dir_all(parent);

    let temp_path = super::global_cache_temp_path(&path);
    if fs::write(&temp_path, &bytes).is_err() {
        return;
    }
    if fs::rename(&temp_path, &path).is_err() {
        let _ = fs::remove_file(&temp_path);
        return;
    }
    super::prune_legacy_global_caches(parent, "stdlib_defs");
}

/// Load a Go module and its transitive deps from cache, recursively.
pub fn load_cached_go_module(
    store: &mut Store,
    module_id: &str,
    cache: &GoStdlibCache,
    target: Target,
) {
    if store.is_visited(module_id) {
        return;
    }

    let Some(cached) = cache.modules.get(module_id) else {
        return;
    };

    // Load transitive deps first
    let imports = cached.go_imports.clone();
    for dep in &imports {
        load_cached_go_module(store, dep, cache, target);
    }

    if store.is_visited(module_id) {
        return; // May have been loaded as a transitive dep of a sibling
    }

    register_cached_go_module(store, module_id, cached, target);
}

fn register_cached_go_module(
    store: &mut Store,
    module_id: &str,
    cached: &GoModuleCache,
    target: Target,
) {
    store.add_module(module_id);
    store.mark_visited(module_id);

    let go_pkg = module_id.strip_prefix("go:");
    let source = go_pkg.and_then(|go_pkg| get_go_stdlib_typedef(go_pkg, target));

    if let Some(source) = source
        && let Some(pkg_name) = extract_package_directive(source)
        && module_id.rsplit('/').next() != Some(pkg_name.as_str())
    {
        store
            .go_package_names
            .insert(module_id.to_string(), pkg_name);
    }

    // Register the typedef File and its on-disk path so go-to-definition can
    // navigate. The files are written by the LSP at startup.
    let owned_file_id;
    let mut file_ids: &[u32] = &[];
    if let (Some(go_pkg), Some(source)) = (go_pkg, source) {
        let file_id = store.new_file_id();
        let filename = format!("{}.d.lis", go_pkg.replace('/', "_"));
        store.store_file(
            module_id,
            File::new_cached(module_id, &filename, &filename, source, file_id),
        );
        if let Some(path) = deps::stdlib_typedef_path(target, go_pkg) {
            store.typedef_paths.insert(file_id, path);
        }
        owned_file_id = [file_id];
        file_ids = &owned_file_id;
    }

    let module = store.get_module_mut(module_id).unwrap();
    for (qualified_name, cached_definition) in &cached.definitions {
        let definition = cached_definition.to_definition(file_ids);
        module
            .definitions
            .insert(qualified_name.clone().into(), definition);
    }
}

/// Extract Go imports from a module's `.d.lis` source without parsing.
fn get_go_imports_from_source(module_id: &str, target: Target) -> Vec<String> {
    let Some(go_pkg) = module_id.strip_prefix("go:") else {
        return vec![];
    };
    let Some(source) = get_go_stdlib_typedef(go_pkg, target) else {
        return vec![];
    };
    source
        .lines()
        .filter_map(|line| {
            let line = line.trim();
            let rest = line.strip_prefix("import \"go:")?;
            let pkg = rest.strip_suffix('"')?;
            Some(format!("go:{pkg}"))
        })
        .collect()
}

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

    #[test]
    fn cache_file_name_includes_target_only() {
        let target = Target::new("darwin", "arm64");
        assert_eq!(cache_file_name(target), "stdlib_defs_darwin_arm64.bin");
    }
}