zoi_core/

utils.rs

use std::collections::HashMap;
use std::fs;
use std::io::{Write, stdin, stdout};
use std::path::{Path, PathBuf};
use std::process::Command;
use std::sync::OnceLock;
use std::time::Duration;

use anyhow::{Result, anyhow};
use clap_complete::Shell;
use colored::Colorize;
use crossterm::tty::IsTty;
#[cfg(unix)]
use nix;
use sha2::{Digest, Sha512};

/// Creates an HTTP client with Zoi's default configuration.
///
/// # Errors
///
/// Returns an error if Zoi is in offline mode or if the HTTP client cannot be
/// built.
pub fn get_http_client() -> Result<&'static reqwest::blocking::Client> {
    static HTTP_CLIENT: OnceLock<reqwest::blocking::Client> = OnceLock::new();
    if crate::offline::is_offline() {
        return Err(anyhow!(
            "Cannot create HTTP client: Zoi is in offline mode."
        ));
    }
    if let Some(client) = HTTP_CLIENT.get() {
        return Ok(client);
    }
    let client = reqwest::blocking::Client::builder()
        .user_agent("zoi")
        .timeout(Duration::from_mins(1))
        .use_rustls_tls()
        .build()
        .map_err(|e| anyhow!("Failed to build HTTP client: {e}"))?;
    let _ = HTTP_CLIENT.set(client);
    HTTP_CLIENT
        .get()
        .ok_or_else(|| anyhow!("HTTP_CLIENT should be set but was missing"))
}

/// Builds a blocking HTTP client with a custom timeout.
///
/// # Errors
///
/// Returns an error if Zoi is in offline mode or if the HTTP client cannot be
/// built.
pub fn build_blocking_http_client(
    timeout_secs: u64
) -> Result<reqwest::blocking::Client> {
    if crate::offline::is_offline() {
        return Err(anyhow!(
            "Cannot create HTTP client: Zoi is in offline mode."
        ));
    }
    let client = reqwest::blocking::Client::builder()
        .user_agent("zoi")
        .timeout(Duration::from_secs(timeout_secs))
        .use_rustls_tls()
        .build()?;
    Ok(client)
}

/// Creates a symbolic link for a directory, handling platform-specific
/// requirements.
///
/// # Errors
///
/// Returns an error if the symlink or directory operations fail.
pub fn symlink_dir(target: &Path, link: &Path) -> std::io::Result<()> {
    if link.exists() || link.is_symlink() {
        if link.is_dir() && !link.is_symlink() {
            fs::remove_dir_all(link)?;
        } else {
            fs::remove_file(link)?;
        }
    }
    #[cfg(unix)]
    {
        std::os::unix::fs::symlink(target, link)?;
    }
    #[cfg(windows)]
    {
        if std::os::windows::fs::symlink_dir(target, link).is_err() {
            if junction::create(target, link).is_err() {
                copy_dir_all(target, link)?;
            }
        }
    }
    Ok(())
}

/// Checks if a command exists in the system's PATH.
pub fn command_exists(command: &str) -> bool {
    if cfg!(target_os = "windows") {
        Command::new("where")
            .arg(command)
            .stdout(std::process::Stdio::null())
            .stderr(std::process::Stdio::null())
            .status()
            .is_ok_and(|status| status.success())
    } else {
        Command::new("bash")
            .arg("-c")
            // Pass the command as a positional argument so metadata supplied
            // names cannot alter the shell program being executed.
            .arg("command -v -- \"$1\"")
            .arg("zoi-command-exists")
            .arg(command)
            .stdout(std::process::Stdio::null())
            .stderr(std::process::Stdio::null())
            .status()
            .is_ok_and(|status| status.success())
    }
}

/// Returns a standard Zoi platform identifier (e.g. "linux-amd64",
/// "windows-arm64").
///
/// This string is used extensively in registries and package definitions to
/// handle platform-specific dependencies and build artifacts.
///
/// # Errors
///
/// Returns an error if the current operating system or architecture is
/// unsupported.
pub fn get_platform() -> Result<String> {
    let os = match std::env::consts::OS {
        "linux" => "linux",
        "macos" | "darwin" => "macos",
        "windows" => "windows",
        unsupported_os => {
            return Err(anyhow!(
                "Unsupported operating system: {unsupported_os}"
            ));
        }
    };
    let arch = match std::env::consts::ARCH {
        "x86_64" | "amd64" => "amd64",
        "aarch64" | "arm64" => "arm64",
        "x86" | "i386" | "i686" => "386",
        unsupported_arch => {
            return Err(anyhow!(
                "Unsupported architecture: {unsupported_arch}"
            ));
        }
    };
    Ok(format!("{os}-{arch}"))
}

/// Returns the home directory of the current user, or the original user if run
/// via sudo or doas.
pub fn get_user_home() -> Option<PathBuf> {
    if let Ok(user_var) =
        std::env::var("SUDO_USER").or_else(|_| std::env::var("DOAS_USER"))
    {
        #[cfg(unix)]
        {
            use nix::unistd::User;
            if let Ok(Some(user)) = User::from_name(&user_var) {
                return Some(user.dir);
            }
        }
        #[cfg(not(unix))]
        let _ = user_var;
    }
    home::home_dir()
}

/// Returns Zoi's user configuration directory according to the XDG Base
/// Directory specification.
///
/// # Errors
///
/// Returns an error if no suitable user configuration directory can be found.
pub fn get_user_config_dir() -> Result<PathBuf> {
    let root =
        xdg_or_platform_dir("XDG_CONFIG_HOME", dirs::config_dir, |home| {
            home.join(".config")
        })?;
    Ok(crate::sysroot::apply_sysroot(root.join("zoi")))
}

/// Returns Zoi's user data directory according to the XDG Base Directory
/// specification.
///
/// # Errors
///
/// Returns an error if no suitable user data directory can be found.
pub fn get_user_data_dir() -> Result<PathBuf> {
    let root =
        xdg_or_platform_dir("XDG_DATA_HOME", dirs::data_local_dir, |home| {
            home.join(".local/share")
        })?;
    Ok(crate::sysroot::apply_sysroot(root.join("zoi")))
}

/// Returns Zoi's user cache directory according to the XDG Base Directory
/// specification.
///
/// # Errors
///
/// Returns an error if no suitable user cache directory can be found.
pub fn get_user_cache_dir() -> Result<PathBuf> {
    let root =
        xdg_or_platform_dir("XDG_CACHE_HOME", dirs::cache_dir, |home| {
            home.join(".cache")
        })?;
    Ok(crate::sysroot::apply_sysroot(root.join("zoi")))
}

/// Returns Zoi's user state directory, honoring `XDG_STATE_HOME` first.
///
/// # Errors
///
/// Returns an error if no suitable user state directory can be found.
pub fn get_user_state_dir() -> Result<PathBuf> {
    let root =
        xdg_or_platform_dir("XDG_STATE_HOME", dirs::data_local_dir, |home| {
            home.join(".local/state")
        })?;
    Ok(crate::sysroot::apply_sysroot(root.join("zoi")))
}

/// Resolves an XDG directory, falling back to the native platform convention.
fn xdg_or_platform_dir(
    variable: &str,
    platform_dir: impl FnOnce() -> Option<PathBuf>,
    unix_fallback: impl FnOnce(PathBuf) -> PathBuf
) -> Result<PathBuf> {
    if let Some(path) =
        std::env::var_os(variable).filter(|path| !path.is_empty())
    {
        let path = PathBuf::from(path);
        if path.is_absolute() {
            return Ok(path);
        }
    }
    if cfg!(unix) && !cfg!(target_os = "macos") {
        return get_user_home()
            .map(unix_fallback)
            .ok_or_else(|| anyhow!("Could not find home directory."));
    }
    platform_dir()
        .ok_or_else(|| anyhow!("Could not determine platform directory."))
}

/// Returns the directory used for user-installed command shims.
///
/// # Errors
///
/// Returns an error if the user data directory cannot be determined.
pub fn get_user_bin_dir() -> Result<PathBuf> {
    Ok(get_user_data_dir()?.join("pkgs").join("bin"))
}

/// Returns the directory used for user shell-completion files.
///
/// # Errors
///
/// Returns an error if the user data directory cannot be determined.
pub fn get_user_completions_dir(shell: &str) -> Result<PathBuf> {
    Ok(get_user_data_dir()?.join("pkgs").join("shell").join(shell))
}

/// Returns the platform-native directory for system-wide Zoi data.
pub fn get_system_data_dir() -> PathBuf {
    let path = if cfg!(target_os = "windows") {
        PathBuf::from("C:\\ProgramData\\zoi")
    } else if cfg!(target_os = "macos") {
        PathBuf::from("/Library/Application Support/zoi")
    } else {
        PathBuf::from("/var/lib/zoi")
    };
    crate::sysroot::apply_sysroot(path)
}

/// Returns the platform-native directory for system-wide Zoi configuration.
pub fn get_system_config_dir() -> PathBuf {
    let path = if cfg!(target_os = "windows") {
        PathBuf::from("C:\\ProgramData\\zoi")
    } else if cfg!(target_os = "macos") {
        PathBuf::from("/Library/Application Support/zoi")
    } else {
        PathBuf::from("/etc/zoi")
    };
    crate::sysroot::apply_sysroot(path)
}

/// Returns the platform-native directory for system-wide Zoi cache files.
pub fn get_system_cache_dir() -> PathBuf {
    let path = if cfg!(target_os = "windows") {
        PathBuf::from("C:\\ProgramData\\zoi\\cache")
    } else if cfg!(target_os = "macos") {
        PathBuf::from("/Library/Caches/zoi")
    } else {
        PathBuf::from("/var/cache/zoi")
    };
    crate::sysroot::apply_sysroot(path)
}

/// Returns the root directory for the package database.
///
/// # Errors
///
/// Returns an error if the home directory cannot be determined.
pub fn get_db_root() -> Result<std::path::PathBuf> {
    if let Ok(path) = std::env::var("ZOI_DB_DIR") {
        return Ok(std::path::PathBuf::from(path));
    }
    Ok(get_user_data_dir()?.join("pkgs").join("db"))
}

/// Returns the root directory of the package store for a given scope.
///
/// Store Locations:
/// - `User`: `$XDG_DATA_HOME/zoi/pkgs/store/`, or the native fallback.
/// - `System`: `/var/lib/zoi/pkgs/store/` (Linux) or
///   `C:\ProgramData\zoi\pkgs\store` (Windows)
/// - `Project`: `./.zoi/pkgs/store/` (Relative to current project root)
///
/// # Errors
///
/// Returns an error if the home directory or current directory cannot be
/// determined.
pub fn get_store_base_dir(scope: crate::types::Scope) -> Result<PathBuf> {
    match scope {
        crate::types::Scope::User => {
            Ok(get_user_data_dir()?.join("pkgs").join("store"))
        }
        crate::types::Scope::System => {
            Ok(get_system_data_dir().join("pkgs").join("store"))
        }
        crate::types::Scope::Project => {
            let current_dir = std::env::current_dir()?;
            Ok(current_dir.join(".zoi").join("pkgs").join("store"))
        }
    }
}

/// Returns the root directory of the package database for a given scope.
///
/// # Errors
///
/// Returns an error if the home directory or current directory cannot be
/// determined.
pub fn get_db_base_dir(scope: crate::types::Scope) -> Result<PathBuf> {
    match scope {
        crate::types::Scope::User => {
            Ok(get_user_data_dir()?.join("pkgs").join("db"))
        }
        crate::types::Scope::System => {
            Ok(get_system_data_dir().join("pkgs").join("db"))
        }
        crate::types::Scope::Project => {
            let current_dir = std::env::current_dir()?;
            Ok(current_dir.join(".zoi").join("pkgs").join("db"))
        }
    }
}

/// Returns the root directory for Git repositories for a given scope.
///
/// # Errors
///
/// Returns an error if the home directory or current directory cannot be
/// determined.
pub fn get_git_base_dir(scope: crate::types::Scope) -> Result<PathBuf> {
    match scope {
        crate::types::Scope::User => {
            Ok(get_user_data_dir()?.join("pkgs").join("git"))
        }
        crate::types::Scope::System => {
            Ok(get_system_data_dir().join("pkgs").join("git"))
        }
        crate::types::Scope::Project => {
            let current_dir = std::env::current_dir()?;
            Ok(current_dir.join(".zoi").join("pkgs").join("git"))
        }
    }
}

/// Generates a unique, origin-aware ID for a package.
///
/// This ID prevents collisions between packages with the same name that reside
/// in different registries or repository tiers.
///
/// ID Format: `#{registry-handle}@{repo-path}/{package-name}`
/// Hashed Result: First 32 characters of the SHA-512 hash of the ID string.
pub fn generate_package_id(
    registry_handle: &str,
    repo_path: &str,
    package_name: &str
) -> String {
    let format_string =
        format!("#{registry_handle}@{repo_path}/{package_name}");
    let mut hasher = Sha512::new();
    hasher.update(format_string.as_bytes());
    let result = hasher.finalize();
    let hex_string = hex::encode(result);
    hex_string[..32].to_string()
}

/// Generates a unique ID for a package including its version.
pub fn generate_versioned_package_id(
    registry_handle: &str,
    repo_path: &str,
    package_name: &str,
    version: &str
) -> String {
    let format_string =
        format!("#{registry_handle}@{repo_path}/{package_name}@{version}");
    let mut hasher = Sha512::new();
    hasher.update(format_string.as_bytes());
    let result = hasher.finalize();
    let hex_string = hex::encode(result);
    hex_string[..32].to_string()
}

/// Creates the directory name for the package in the store.
/// Format: `{hash}-{name}`
pub fn get_package_dir_name(package_id: &str, package_name: &str) -> String {
    format!("{package_id}-{package_name}")
}

/// Recursively copies all files and subdirectories from source to destination.
///
/// # Errors
///
/// Returns an error if the directory creation or file copy operation fails.
pub fn copy_dir_all(src: &Path, dst: &Path) -> std::io::Result<()> {
    let src = if src.as_os_str().is_empty() {
        Path::new(".")
    } else {
        src
    };
    fs::create_dir_all(dst)?;
    for entry in fs::read_dir(src)? {
        let entry = entry?;
        let ty = entry.file_type()?;
        if ty.is_dir() {
            copy_dir_all(&entry.path(), &dst.join(entry.file_name()))?;
        } else if ty.is_symlink() {
            let target = fs::read_link(entry.path())?;
            let destination = dst.join(entry.file_name());
            #[cfg(unix)]
            std::os::unix::fs::symlink(target, destination)?;
            #[cfg(windows)]
            {
                if target.is_dir() {
                    std::os::windows::fs::symlink_dir(target, destination)?;
                } else {
                    std::os::windows::fs::symlink_file(target, destination)?;
                }
            }
        } else {
            fs::copy(entry.path(), dst.join(entry.file_name()))?;
        }
    }
    Ok(())
}

/// Performs a jittered exponential backoff sleep.
///
/// Used during network retries to prevent thundering herd problems and
/// improve reliability on unstable connections.
pub fn retry_backoff_sleep(attempt: u32) {
    let base_ms = 500u64.saturating_mul(1u64 << (attempt.saturating_sub(1)));
    let jitter = u64::from(
        std::time::SystemTime::now()
            .duration_since(std::time::UNIX_EPOCH)
            .unwrap_or(Duration::from_secs(0))
            .subsec_millis()
            % 200
    );
    let sleep_ms = (base_ms + jitter).min(8000);
    std::thread::sleep(Duration::from_millis(sleep_ms));
}

/// Retrieves information about the current Linux distribution from
/// /etc/os-release.
pub fn get_linux_distribution_info() -> Option<HashMap<String, String>> {
    let path = crate::sysroot::apply_sysroot("/etc/os-release");
    if let Ok(contents) = fs::read_to_string(path) {
        let info: HashMap<String, String> = contents
            .lines()
            .filter_map(|line| {
                let mut parts = line.splitn(2, '=');
                let key = parts.next()?;
                let value = parts.next()?.trim_matches('"').to_string();
                if key.is_empty() {
                    None
                } else {
                    Some((key.to_string(), value))
                }
            })
            .collect();
        if info.is_empty() { None } else { Some(info) }
    } else {
        None
    }
}

/// Detects the general family of a Linux distribution (e.g. "debian", "arch",
/// "fedora").
///
/// This is used to map specific distributions to their primary package manager
/// and standard filesystem locations.
///
/// Strategy:
/// - `ID_LIKE` Check: We first check the `ID_LIKE` field in `/etc/os-release`.
///   This is the most reliable way to identify derivatives (e.g. Ubuntu is
///   `debian`).
/// - Direct ID Match: If `ID_LIKE` is missing, we fall back to the primary
///   `ID`.
/// - Normalization: We group similar distros under a common "family" key to
///   simplify downstream logic (e.g. Rocky, Alma, and `CentOS` all map to
///   `fedora` because they share the DNF/RPM ecosystem).
pub fn get_linux_distro_family() -> Option<String> {
    if is_zoios() {
        return Some("zoios".to_string());
    }
    if let Some(info) = get_linux_distribution_info() {
        if let Some(id_like) = info.get("ID_LIKE") {
            let families: Vec<&str> = id_like.split_whitespace().collect();
            if families.contains(&"debian") {
                return Some("debian".to_string());
            }
            if families.contains(&"arch") {
                return Some("arch".to_string());
            }
            if families.contains(&"fedora") {
                return Some("fedora".to_string());
            }
            if families.contains(&"rhel") {
                return Some("fedora".to_string());
            }
            if families.contains(&"suse") {
                return Some("suse".to_string());
            }
            if families.contains(&"gentoo") {
                return Some("gentoo".to_string());
            }
        }
        if let Some(id) = info.get("ID") {
            return match id.as_str() {
                "debian" | "ubuntu" | "linuxmint" | "pop" | "kali"
                | "kubuntu" | "lubuntu" | "xubuntu" | "zorin"
                | "elementary" => Some("debian".to_string()),
                "arch" | "manjaro" | "cachyos" | "endeavouros" | "garuda" => {
                    Some("arch".to_string())
                }
                "fedora" | "centos" | "rhel" | "rocky" | "almalinux" => {
                    Some("fedora".to_string())
                }
                "opensuse" | "opensuse-tumbleweed" | "opensuse-leap" => {
                    Some("suse".to_string())
                }
                "gentoo" => Some("gentoo".to_string()),
                "alpine" => Some("alpine".to_string()),
                "void" => Some("void".to_string()),
                "solus" => Some("solus".to_string()),
                "guix" => Some("guix".to_string()),
                _ => None
            };
        }
    }
    None
}

/// Returns the ID of the current Linux distribution (e.g. "debian", "fedora").
pub fn get_linux_distribution() -> Option<String> {
    get_linux_distribution_info().and_then(|info| info.get("ID").cloned())
}

/// Returns true if the current system is a ZoiOS-based distribution (like
/// Parlex).
pub fn is_zoios() -> bool {
    if let Some(info) = get_linux_distribution_info() {
        if let Some(id) = info.get("ID")
            && (id == "zoios" || id == "parlex")
        {
            return true;
        }
        if let Some(id_like) = info.get("ID_LIKE")
            && id_like.split_whitespace().any(|s| s == "zoios")
        {
            return true;
        }
    }
    false
}

/// Resolves the default installation scope based on the current environment.
pub fn resolve_fallback_scope() -> crate::types::Scope {
    if std::path::Path::new("zoi.lua").exists()
        || std::path::Path::new("zoi.yaml").exists()
    {
        crate::types::Scope::Project
    } else if is_zoios() {
        crate::types::Scope::System
    } else {
        crate::types::Scope::User
    }
}

/// Returns the name of the current desktop environment (e.g. "gnome", "kde").
pub fn get_desktop_environment() -> Option<String> {
    if cfg!(target_os = "windows") {
        return Some("windows".to_string());
    }
    if let Ok(de) = std::env::var("XDG_CURRENT_DESKTOP")
        && !de.is_empty()
    {
        return Some(de.to_lowercase());
    }
    if let Ok(ds) = std::env::var("DESKTOP_SESSION")
        && !ds.is_empty()
    {
        return Some(ds.to_lowercase());
    }
    None
}

/// Returns the name of the current display server (e.g. "wayland", "x11").
pub fn get_display_server() -> Option<String> {
    if cfg!(target_os = "windows") {
        return Some("windows".to_string());
    }
    if cfg!(target_os = "macos") {
        return Some("quartz".to_string());
    }
    if let Ok(st) = std::env::var("XDG_SESSION_TYPE")
        && !st.is_empty()
    {
        return Some(st.to_lowercase());
    }
    None
}

/// Returns the version of the operating system kernel.
pub fn get_kernel_version() -> Option<String> {
    if cfg!(unix) {
        let output = Command::new("uname").arg("-r").output().ok()?;
        if output.status.success() {
            return Some(
                String::from_utf8_lossy(&output.stdout).trim().to_string()
            );
        }
    } else if cfg!(target_os = "windows") {
        let output = Command::new("pwsh")
            .arg("-Command")
            .arg("(Get-CimInstance Win32_OperatingSystem).Version")
            .output()
            .ok()?;
        if output.status.success() {
            return Some(
                String::from_utf8_lossy(&output.stdout).trim().to_string()
            );
        }
    }
    None
}

/// Returns the name of the system's init system (e.g. "systemd", "openrc").
pub fn get_init_system() -> Option<String> {
    if let Ok(val) = std::env::var("ZOI_INIT") {
        return Some(val.to_lowercase());
    }

    let run_systemd =
        crate::sysroot::apply_sysroot(PathBuf::from("/run/systemd/system"));
    if run_systemd.exists() {
        return Some("systemd".to_string());
    }

    let run_openrc =
        crate::sysroot::apply_sysroot(PathBuf::from("/run/openrc"));
    if run_openrc.exists() {
        return Some("openrc".to_string());
    }

    let sbin_init = crate::sysroot::apply_sysroot(PathBuf::from("/sbin/init"));
    if let Ok(target) = std::fs::read_link(&sbin_init) {
        let target_str = target.to_string_lossy();
        if target_str.contains("systemd") {
            return Some("systemd".to_string());
        } else if target_str.contains("openrc") {
            return Some("openrc".to_string());
        } else if target_str.contains("busybox") {
            return Some("busybox".to_string());
        }
    }

    None
}

/// Returns the name of the available privilege escalation tool (e.g. "sudo",
/// "doas").
pub fn get_privilege_escalator() -> Option<String> {
    if command_exists("sudo") {
        Some("sudo".to_string())
    } else if command_exists("doas") {
        Some("doas".to_string())
    } else {
        None
    }
}

/// Returns the version of the current distribution.
pub fn get_distro_version() -> Option<String> {
    if let Some(info) = get_linux_distribution_info()
        && let Some(vid) = info.get("VERSION_ID")
    {
        return Some(vid.clone());
    }
    if cfg!(target_os = "macos") {
        let output = Command::new("sw_vers")
            .arg("-productVersion")
            .output()
            .ok()?;
        if output.status.success() {
            return Some(
                String::from_utf8_lossy(&output.stdout).trim().to_string()
            );
        }
    } else if cfg!(target_os = "windows") {
        let output = Command::new("pwsh")
            .arg("-Command")
            .arg("(Get-CimInstance Win32_OperatingSystem).Version")
            .output()
            .ok()?;
        if output.status.success() {
            return Some(
                String::from_utf8_lossy(&output.stdout).trim().to_string()
            );
        }
    }
    None
}

/// Returns a short description of the system's CPU.
pub fn get_cpu_info() -> Option<String> {
    if cfg!(target_os = "linux") {
        if let Ok(cpuinfo) = fs::read_to_string("/proc/cpuinfo") {
            for line in cpuinfo.lines() {
                if line.starts_with("model name")
                    && let Some((_, model)) = line.split_once(':')
                {
                    return Some(model.trim().to_string());
                }
            }
        }
    } else if cfg!(target_os = "macos") {
        let output = Command::new("sysctl")
            .arg("-n")
            .arg("machdep.cpu.brand_string")
            .output()
            .ok()?;
        if output.status.success() {
            return Some(
                String::from_utf8_lossy(&output.stdout).trim().to_string()
            );
        }
    } else if cfg!(target_os = "windows") {
        let output = Command::new("pwsh")
            .arg("-Command")
            .arg("(Get-CimInstance Win32_Processor).Name")
            .output()
            .ok()?;
        if output.status.success() {
            return Some(
                String::from_utf8_lossy(&output.stdout).trim().to_string()
            );
        }
    }
    None
}

/// Returns a short description of the system's GPU.
pub fn get_gpu_info() -> Option<String> {
    if cfg!(target_os = "linux") {
        if let Ok(output) = Command::new("lspci").output()
            && output.status.success()
        {
            let stdout = String::from_utf8_lossy(&output.stdout);
            for line in stdout.lines() {
                if (line.contains("VGA compatible controller")
                    || line.contains("3D controller"))
                    && let Some((_, model)) = line.split_once(": ")
                {
                    return Some(model.trim().to_string());
                }
            }
        }
    } else if cfg!(target_os = "macos") {
        let output = Command::new("system_profiler")
            .arg("SPDisplaysDataType")
            .output()
            .ok()?;
        if output.status.success() {
            let stdout = String::from_utf8_lossy(&output.stdout);
            for line in stdout.lines() {
                if line.trim().starts_with("Chipset Model:")
                    && let Some((_, model)) = line.split_once(':')
                {
                    return Some(model.trim().to_string());
                }
            }
        }
    } else if cfg!(target_os = "windows") {
        let output = Command::new("pwsh")
            .arg("-Command")
            .arg("(Get-CimInstance Win32_VideoController).Name")
            .output()
            .ok()?;
        if output.status.success() {
            return Some(
                String::from_utf8_lossy(&output.stdout).trim().to_string()
            );
        }
    }
    None
}

/// Identifies the primary package manager for the current operating system.
///
/// This is used to resolve `native:` dependencies.
pub fn get_native_package_manager() -> Option<String> {
    let os = std::env::consts::OS;
    match os {
        "linux" => get_linux_distro_family()
            .map(|family| {
                match family.as_str() {
                    "debian" => "apt",
                    "arch" => "pacman",
                    "fedora" => "dnf",
                    "suse" => "zypper",
                    "gentoo" => "portage",
                    "alpine" => "apk",
                    "void" => "xbps-install",
                    "solus" => "eopkg",
                    "guix" => "guix",
                    "zoios" => "zoi",
                    _ => "unknown"
                }
                .to_string()
            })
            .filter(|s| s != "unknown"),
        "macos" => {
            if command_exists("brew") {
                Some("brew".to_string())
            } else if command_exists("port") {
                Some("macports".to_string())
            } else {
                None
            }
        }
        "windows" => {
            if command_exists("scoop") {
                Some("scoop".to_string())
            } else if command_exists("choco") {
                Some("choco".to_string())
            } else if command_exists("winget") {
                Some("winget".to_string())
            } else {
                None
            }
        }
        _ => None
    }
}

/// Scans the system for all supported package managers.
///
/// This provides the list of available managers shown in `zoi info` and
/// used to validate `manager:` prefixes in dependency strings.
pub fn get_all_available_package_managers() -> Vec<String> {
    let mut managers = Vec::new();
    let all_possible_managers = [
        "apt",
        "pacman",
        "yay",
        "paru",
        "pikaur",
        "trizen",
        "dnf",
        "yum",
        "zypper",
        "portage",
        "apk",
        "snap",
        "flatpak",
        "nix",
        "brew",
        "port",
        "scoop",
        "choco",
        "winget",
        "pkg",
        "pkg_add",
        "xbps-install",
        "eopkg",
        "guix",
        "mas"
    ];

    for manager in &all_possible_managers {
        if command_exists(manager) {
            managers.push(manager.to_string());
        }
    }
    managers.sort();
    managers.dedup();
    managers
}

/// Formats a byte count into a human-readable string (e.g. "1.24 MiB").
pub fn format_bytes(bytes: u64) -> String {
    const KIB: u64 = 1024;
    const MIB: u64 = 1024 * KIB;
    const GIB: u64 = 1024 * MIB;
    if bytes >= GIB {
        format!("{:.2} GiB", bytes as f64 / GIB as f64)
    } else if bytes >= MIB {
        format!("{:.2} MiB", bytes as f64 / MIB as f64)
    } else if bytes >= KIB {
        format!("{:.2} KiB", bytes as f64 / KIB as f64)
    } else {
        format!("{bytes} B")
    }
}

/// Formats a size difference into a signed human-readable string (e.g. "+50
/// B").
pub fn format_size_diff(diff: i64) -> String {
    if diff == 0 {
        return "0 B".to_string();
    }
    let sign = if diff > 0 { "+" } else { "-" };
    let bytes = diff.unsigned_abs();
    format!("{}{}", sign, format_bytes(bytes))
}

/// Verifies that a given path is "Safe" and doesn't attempt to escape the base
/// directory.
///
/// This is a critical security check against "Path Traversal" attacks in
/// package archives or Lua scripts.
pub fn is_safe_path(base: &Path, path: &Path) -> bool {
    let base = base.canonicalize().unwrap_or_else(|_| base.to_path_buf());
    let joined = if path.is_absolute() {
        path.to_path_buf()
    } else {
        base.join(path)
    };
    let mut normalized = PathBuf::new();
    for component in joined.components() {
        match component {
            std::path::Component::Prefix(_) | std::path::Component::RootDir => {
                normalized.push(component);
            }
            std::path::Component::CurDir => {}
            std::path::Component::ParentDir => {
                if !normalized.pop() {
                    return false;
                }
            }
            std::path::Component::Normal(p) => normalized.push(p)
        }
    }
    normalized.starts_with(&base)
}

/// Creates a symbolic link for a file, handling platform-specific requirements.
///
/// # Errors
///
/// Returns an error if the symlink operation fails.
pub fn symlink_file(target: &Path, link: &Path) -> std::io::Result<()> {
    if link.exists() || link.is_symlink() {
        fs::remove_file(link)?;
    }
    #[cfg(unix)]
    {
        std::os::unix::fs::symlink(target, link)
    }
    #[cfg(windows)]
    {
        if std::os::windows::fs::symlink_file(target, link).is_err() {
            if fs::hard_link(target, link).is_err() {
                fs::copy(target, link)?;
            }
        }
        Ok(())
    }
}

/// Checks if the current process has administrative privileges.
pub fn is_admin() -> bool {
    #[cfg(unix)]
    {
        nix::unistd::getuid().is_root()
    }
    #[cfg(windows)]
    {
        false
    }
}

/// Executes a shell command and returns an error if it fails.
///
/// # Errors
///
/// Returns an error if the command fails to execute or returns a non-zero exit
/// status.
pub fn run_shell_command(command_str: &str) -> anyhow::Result<()> {
    let status = if cfg!(target_os = "windows") {
        Command::new("pwsh")
            .arg("-Command")
            .arg(command_str)
            .status()?
    } else {
        Command::new("bash").arg("-c").arg(command_str).status()?
    };
    if !status.success() {
        return Err(anyhow!("Command failed: {command_str}"));
    }
    Ok(())
}

/// Executes a shell command quietly (suppressing output) and returns an error
/// if it fails.
///
/// # Errors
///
/// Returns an error if the command fails to execute or returns a non-zero exit
/// status.
pub fn run_shell_command_quietly(command_str: &str) -> anyhow::Result<()> {
    let status = if cfg!(target_os = "windows") {
        Command::new("pwsh")
            .arg("-Command")
            .arg(command_str)
            .stdout(std::process::Stdio::null())
            .stderr(std::process::Stdio::null())
            .status()?
    } else {
        Command::new("bash")
            .arg("-c")
            .arg(command_str)
            .stdout(std::process::Stdio::null())
            .stderr(std::process::Stdio::null())
            .status()?
    };
    if !status.success() {
        return Err(anyhow!("Command failed: {command_str}"));
    }
    Ok(())
}

/// Returns true if Zoi is running in "Mini" mode (lightweight, zero-sync).
pub fn is_mini_mode() -> bool {
    std::env::var("ZOI_MINI_MODE").is_ok_and(|v| v == "1")
}

/// Prompts the user for confirmation (y/N) unless the `yes` flag is set.
pub fn ask_for_confirmation(prompt: &str, yes: bool) -> bool {
    if yes {
        return true;
    }
    if std::env::var("ZOI_TEST").is_ok() || !stdin().is_tty() {
        return false;
    }
    print!("{prompt} [y/N]: ");
    let _ = stdout().flush();
    let mut input = String::new();
    if stdin().read_line(&mut input).is_err() {
        return false;
    }
    input.trim().eq_ignore_ascii_case("y")
}

/// Recursively sets a directory and its contents to be read-only.
///
/// # Errors
///
/// Returns an error if the permission change fails.
pub fn set_path_read_only(path: &Path) -> anyhow::Result<()> {
    if !path.exists() {
        return Ok(());
    }
    for entry in walkdir::WalkDir::new(path) {
        let entry = entry?;
        let mut perms = fs::metadata(entry.path())?.permissions();
        if !perms.readonly() {
            perms.set_readonly(true);
            fs::set_permissions(entry.path(), perms)?;
        }
    }
    Ok(())
}

/// Recursively ensures a directory and its contents are writable.
///
/// # Errors
///
/// Returns an error if the permission change fails.
pub fn set_path_writable(path: &Path) -> anyhow::Result<()> {
    if !path.exists() {
        return Ok(());
    }
    for entry in walkdir::WalkDir::new(path) {
        let entry = entry?;
        let mut perms = fs::metadata(entry.path())?.permissions();
        if perms.readonly() {
            #[cfg(unix)]
            {
                use std::os::unix::fs::PermissionsExt;
                let mode = perms.mode();
                perms.set_mode(mode | 0o200);
            }
            #[cfg(not(unix))]
            {
                perms.set_readonly(false);
            }
            fs::set_permissions(entry.path(), perms)?;
        }
    }
    Ok(())
}

/// Sets the owner and group for a file or directory (Unix only).
///
/// # Errors
///
/// Returns an error if the owner or group lookup fails, or if the chown
/// operation fails.
#[cfg(unix)]
pub fn set_path_owner(
    path: &Path,
    owner: &str,
    group: &str
) -> anyhow::Result<()> {
    use nix::unistd::{Gid, Group, Uid, User, chown};
    let uid = if let Ok(u) = owner.parse::<u32>() {
        Some(Uid::from_raw(u))
    } else if !owner.is_empty() {
        Some(
            User::from_name(owner)
                .map_err(|e| anyhow!("Error looking up user '{owner}': {e}"))?
                .ok_or_else(|| anyhow!("User not found: {owner}"))?
                .uid
        )
    } else {
        None
    };
    let gid = if let Ok(g) = group.parse::<u32>() {
        Some(Gid::from_raw(g))
    } else if !group.is_empty() {
        Some(
            Group::from_name(group)
                .map_err(|e| anyhow!("Error looking up group '{group}': {e}"))?
                .ok_or_else(|| anyhow!("Group not found: {group}"))?
                .gid
        )
    } else {
        None
    };
    chown(path, uid, gid)
        .map_err(|e| anyhow!("Failed to chown '{}': {}", path.display(), e))?;
    Ok(())
}

/// Checks if a target platform is compatible with a list of allowed platforms.
pub fn is_platform_compatible(
    current_platform: &str,
    allowed_platforms: &[String]
) -> bool {
    let os_part = current_platform
        .split('-')
        .next()
        .unwrap_or(current_platform);
    let os = match os_part {
        "darwin" => "macos",
        other => other
    };
    allowed_platforms.iter().any(|p| {
        if let Some(rest) = p.strip_prefix("ci:") {
            let target = rest.split(':').next().unwrap_or_default();
            target == current_platform || target == os
        } else {
            let p_norm = if p == "darwin" { "macos" } else { p };
            p_norm == "all" || p_norm == os || p_norm == current_platform
        }
    })
}

/// Validates a package license against SPDX and organizational policies.
pub fn check_license(license: &str) {
    if license.is_empty() || license.eq_ignore_ascii_case("None") {
        return;
    }
    if license.eq_ignore_ascii_case("Proprietary")
        || license.eq_ignore_ascii_case("Unknown")
    {
        return;
    }
    match spdx::Expression::parse(license) {
        Ok(expr) => {
            if !expr.evaluate(|req| match req.license {
                spdx::LicenseItem::Spdx { id, .. } => id.is_osi_approved(),
                spdx::LicenseItem::Other { .. } => false
            }) {
                println!(
                    "{} License '{}' is not an OSI approved license.",
                    "Warning:".yellow(),
                    license.yellow().bold()
                );
            }
        }
        Err(_) => {
            println!(
                "{} Could not parse license expression '{}'. It may not be a \
                 valid SPDX identifier.",
                "Warning:".yellow(),
                license.yellow().bold()
            );
        }
    }
}

/// Prompts the user to confirm installation from an untrusted (non-official)
/// source.
///
/// # Errors
///
/// Returns an error if the user aborts the operation.
pub fn confirm_untrusted_source(
    source_type: &crate::types::SourceType,
    yes: bool
) -> anyhow::Result<()> {
    if is_mini_mode() {
        return Ok(());
    }
    if source_type == &crate::types::SourceType::OfficialRepo {
        return Ok(());
    }
    let warning_message = match source_type {
        crate::types::SourceType::UntrustedRepo(repo) => {
            format!(
                "The package from repository '@{repo}' is not an official Zoi \
                 repository."
            )
        }
        crate::types::SourceType::LocalFile => {
            "You are installing from a local file.".to_string()
        }
        crate::types::SourceType::Url => {
            "You are installing from a remote URL. This script will be \
             executed with your user's permissions, which could lead to remote \
             code execution if the source is malicious."
                .to_string()
        }
        crate::types::SourceType::GitRepo(repo) => format!(
            "You are installing from an external git repository '{repo}'. \
             This script will be executed with your user's permissions."
        ),
        crate::types::SourceType::OfficialRepo => return Ok(())
    };
    println!(
        "\n{}: {}",
        "SECURITY WARNING".yellow().bold(),
        warning_message
    );
    if ask_for_confirmation(
        "This source is not trusted. Are you sure you want to continue?",
        yes
    ) {
        Ok(())
    } else {
        Err(anyhow!("Operation aborted by user."))
    }
}

/// Expands standard Zoi path placeholders (e.g. `${pkgstore}`, `${usrhome}`)
/// in a string.
///
/// Note that `${createpkgdir}` is deliberately not expanded here: it depends
/// on the directory Zoi was invoked from during installation, so resolving it
/// at cleanup time could target an unrelated path. Pooled installations
/// record the absolute location for these entries instead.
///
/// # Errors
///
/// Returns an error if the current directory cannot be determined.
pub fn expand_placeholders(
    path: &str,
    version_dir: &Path,
    scope: crate::types::Scope
) -> Result<String> {
    let mut expanded = path.to_string();
    expanded = expanded.replace("${pkgstore}", &version_dir.to_string_lossy());
    expanded = expanded.replace(
        "${usrroot}",
        &crate::sysroot::apply_sysroot(PathBuf::from("/")).to_string_lossy()
    );
    if let Some(home_dir) = get_user_home() {
        expanded = expanded.replace("${usrhome}", &home_dir.to_string_lossy());
    }

    let applications_dir = match scope {
        crate::types::Scope::System => PathBuf::from("/Applications"),
        crate::types::Scope::User => get_user_home().map_or_else(
            || PathBuf::from("/Applications"),
            |h| h.join("Applications")
        ),
        crate::types::Scope::Project => std::env::current_dir()
            .unwrap_or_default()
            .join("Applications")
    };
    expanded = expanded
        .replace("${applications}", &applications_dir.to_string_lossy());

    Ok(expanded)
}

/// Expands the `~` character to the user's home directory.
pub fn expand_tilde<P: AsRef<Path>>(path: P) -> PathBuf {
    let path = path.as_ref();
    if !path.starts_with("~") {
        return path.to_path_buf();
    }
    if let Some(home_dir) = get_user_home() {
        if path == Path::new("~") {
            return home_dir;
        }
        if let Ok(stripped) = path.strip_prefix("~/") {
            return home_dir.join(stripped);
        }
    }
    path.to_path_buf()
}

/// Detects and returns the current shell being used.
pub fn get_current_shell() -> Option<Shell> {
    if cfg!(windows) {
        return Some(Shell::PowerShell);
    }
    if let Ok(shell_path) = std::env::var("SHELL") {
        let shell_name = Path::new(&shell_path).file_name()?.to_str()?;
        match shell_name {
            "bash" => Some(Shell::Bash),
            "zsh" => Some(Shell::Zsh),
            "fish" => Some(Shell::Fish),
            "elvish" => Some(Shell::Elvish),
            "pwsh" => Some(Shell::PowerShell),
            _ => None
        }
    } else {
        None
    }
}