codebase-graph 1.4.6

use crate::error::NativeError;
use std::path::{Path, PathBuf};

// Both the prebuilt and source-built Ladybug 0.18.3 libraries resolve
// extensions from this ABI-compatible cache path. Seed it so packaged
// binaries never download extensions at runtime.
const LADYBUG_EXTENSION_CACHE_VERSIONS: [&str; 1] = ["0.18.1"];

pub fn preseed_ladybug_extensions(include_fts: bool) -> Result<(), NativeError> {
    let home = ladybug_home_dir()?;
    let Some(platform) = ladybug_platform() else {
        return Ok(());
    };
    let mut extensions = vec!["json"];
    if include_fts {
        extensions.push("fts");
    }
    for extension in extensions {
        let Some(bytes) = bundled_extension_bytes(extension) else {
            continue;
        };
        for cache_version in LADYBUG_EXTENSION_CACHE_VERSIONS {
            let extension_dir = extension_dir(&home, cache_version, platform, extension);
            let extension_path = extension_dir.join(format!("lib{extension}.lbug_extension"));
            if extension_path.exists() {
                continue;
            }
            std::fs::create_dir_all(&extension_dir)?;
            std::fs::write(extension_path, bytes)?;
        }
    }
    Ok(())
}

fn extension_dir(home: &Path, cache_version: &str, platform: &str, extension: &str) -> PathBuf {
    home.join(".lbdb")
        .join("extension")
        .join(cache_version)
        .join(platform)
        .join(extension)
}

fn ladybug_home_dir() -> Result<PathBuf, NativeError> {
    let variable = if cfg!(windows) { "USERPROFILE" } else { "HOME" };
    std::env::var_os(variable)
        .map(PathBuf::from)
        .filter(|path| !path.as_os_str().is_empty())
        .ok_or_else(|| {
            NativeError::Database(format!(
                "LadyBug extension cache cannot be seeded because {variable} is not set"
            ))
        })
}

fn ladybug_platform() -> Option<&'static str> {
    if cfg!(all(target_os = "linux", target_arch = "x86_64")) {
        Some("linux_amd64")
    } else if cfg!(all(target_os = "linux", target_arch = "aarch64")) {
        Some("linux_arm64")
    } else if cfg!(all(target_os = "macos", target_arch = "x86_64")) {
        Some("osx_amd64")
    } else if cfg!(all(target_os = "macos", target_arch = "aarch64")) {
        Some("osx_arm64")
    } else if cfg!(all(target_os = "windows", target_arch = "x86_64")) {
        Some("win_amd64")
    } else {
        None
    }
}

#[cfg(all(target_os = "linux", target_arch = "x86_64"))]
fn bundled_extension_bytes(extension: &str) -> Option<&'static [u8]> {
    match extension {
        "json" => Some(include_bytes!(
            "../../assets/ladybug-extensions/0.18.1/linux_amd64/json/libjson.lbug_extension"
        )),
        "fts" => Some(include_bytes!(
            "../../assets/ladybug-extensions/0.18.1/linux_amd64/fts/libfts.lbug_extension"
        )),
        _ => None,
    }
}

#[cfg(all(target_os = "linux", target_arch = "aarch64"))]
fn bundled_extension_bytes(extension: &str) -> Option<&'static [u8]> {
    match extension {
        "json" => Some(include_bytes!(
            "../../assets/ladybug-extensions/0.18.1/linux_arm64/json/libjson.lbug_extension"
        )),
        "fts" => Some(include_bytes!(
            "../../assets/ladybug-extensions/0.18.1/linux_arm64/fts/libfts.lbug_extension"
        )),
        _ => None,
    }
}

#[cfg(all(target_os = "macos", target_arch = "x86_64"))]
fn bundled_extension_bytes(extension: &str) -> Option<&'static [u8]> {
    match extension {
        "json" => Some(include_bytes!(
            "../../assets/ladybug-extensions/0.18.1/osx_amd64/json/libjson.lbug_extension"
        )),
        "fts" => Some(include_bytes!(
            "../../assets/ladybug-extensions/0.18.1/osx_amd64/fts/libfts.lbug_extension"
        )),
        _ => None,
    }
}

#[cfg(all(target_os = "macos", target_arch = "aarch64"))]
fn bundled_extension_bytes(extension: &str) -> Option<&'static [u8]> {
    match extension {
        "json" => Some(include_bytes!(
            "../../assets/ladybug-extensions/0.18.1/osx_arm64/json/libjson.lbug_extension"
        )),
        "fts" => Some(include_bytes!(
            "../../assets/ladybug-extensions/0.18.1/osx_arm64/fts/libfts.lbug_extension"
        )),
        _ => None,
    }
}

#[cfg(all(
    target_os = "windows",
    target_arch = "x86_64",
    feature = "bundled-windows-extensions"
))]
fn bundled_extension_bytes(extension: &str) -> Option<&'static [u8]> {
    match extension {
        "json" => Some(include_bytes!(
            "../../assets/ladybug-extensions/0.18.1/win_amd64/json/libjson.lbug_extension"
        )),
        "fts" => Some(include_bytes!(
            "../../assets/ladybug-extensions/0.18.1/win_amd64/fts/libfts.lbug_extension"
        )),
        _ => None,
    }
}

#[cfg(not(any(
    all(target_os = "linux", target_arch = "x86_64"),
    all(target_os = "linux", target_arch = "aarch64"),
    all(target_os = "macos", target_arch = "x86_64"),
    all(target_os = "macos", target_arch = "aarch64"),
    all(
        target_os = "windows",
        target_arch = "x86_64",
        feature = "bundled-windows-extensions"
    )
)))]
fn bundled_extension_bytes(_extension: &str) -> Option<&'static [u8]> {
    None
}

#[cfg(test)]
mod tests {
    use super::extension_dir;
    use std::path::{Path, PathBuf};

    #[test]
    fn supports_prebuilt_and_source_built_ladybug_extension_cache_versions() {
        assert_eq!(super::LADYBUG_EXTENSION_CACHE_VERSIONS, ["0.18.1"]);
        assert_eq!(
            extension_dir(Path::new("cache"), "0.18.1", "osx_arm64", "json"),
            PathBuf::from("cache/.lbdb/extension/0.18.1/osx_arm64/json")
        );
    }
}