normalize-package-index 0.3.2

//! EndeavourOS package index fetcher.
//!
//! EndeavourOS is an Arch-based distro with its own repositories.
//! Has its own repository in addition to Arch repos and AUR.
//!
//! ## API Strategy
//! - **fetch**: EndeavourOS repo + Arch + AUR fallback
//! - **fetch_versions**: Same sources
//! - **search**: Filters cached repo data + Arch/AUR
//! - **fetch_all**: Parses EndeavourOS + Arch repo databases
//!
//! ## Multi-repo Support
//! ```rust,ignore
//! use normalize_packages::index::endeavouros::{EndeavourOs, EndeavourOsRepo};
//!
//! // All repos (default)
//! let all = EndeavourOs::all();
//!
//! // EndeavourOS repos only
//! let eos = EndeavourOs::endeavouros_only();
//!
//! // With Arch repos
//! let with_arch = EndeavourOs::with_arch();
//! ```

use super::arch_common;
use super::{IndexError, PackageIndex, PackageMeta, VersionMeta};
use crate::cache;
use rayon::prelude::*;
use std::time::Duration;

/// Cache TTL for repo databases (1 hour).
const CACHE_TTL: Duration = Duration::from_secs(60 * 60);

/// EndeavourOS mirror URL.
const EOS_MIRROR: &str = "https://mirror.alpix.eu/endeavouros/repo";

/// Available EndeavourOS repositories.
#[derive(Debug, Clone, Copy, PartialEq, Eq, Hash)]
pub enum EndeavourOsRepo {
    // === EndeavourOS Repos ===
    /// EndeavourOS main repository
    EndeavourOs,

    // === Arch Repos ===
    /// Arch core
    Core,
    /// Arch extra
    Extra,
    /// Arch multilib
    Multilib,

    // === AUR ===
    /// Arch User Repository
    Aur,
}

impl EndeavourOsRepo {
    /// Get the database URL for this repository.
    fn db_url(&self) -> Option<String> {
        match self {
            Self::EndeavourOs => Some(format!("{}/endeavouros/x86_64/endeavouros.db", EOS_MIRROR)),
            Self::Core => {
                Some("https://mirror.rackspace.com/archlinux/core/os/x86_64/core.db".to_string())
            }
            Self::Extra => {
                Some("https://mirror.rackspace.com/archlinux/extra/os/x86_64/extra.db".to_string())
            }
            Self::Multilib => Some(
                "https://mirror.rackspace.com/archlinux/multilib/os/x86_64/multilib.db".to_string(),
            ),
            Self::Aur => None,
        }
    }

    /// Get the repository name for tagging.
    pub fn name(&self) -> &'static str {
        match self {
            Self::EndeavourOs => "endeavouros",
            Self::Core => "core",
            Self::Extra => "extra",
            Self::Multilib => "multilib",
            Self::Aur => "aur",
        }
    }

    /// All available repositories.
    pub fn all() -> &'static [EndeavourOsRepo] {
        &[
            Self::EndeavourOs,
            Self::Core,
            Self::Extra,
            Self::Multilib,
            Self::Aur,
        ]
    }

    /// EndeavourOS-specific repos only.
    pub fn endeavouros_only() -> &'static [EndeavourOsRepo] {
        &[Self::EndeavourOs]
    }

    /// Arch-compatible repos.
    pub fn arch() -> &'static [EndeavourOsRepo] {
        &[Self::Core, Self::Extra, Self::Multilib, Self::Aur]
    }

    /// Stable repos (EndeavourOS + Arch main).
    pub fn stable() -> &'static [EndeavourOsRepo] {
        &[
            Self::EndeavourOs,
            Self::Core,
            Self::Extra,
            Self::Multilib,
            Self::Aur,
        ]
    }
}

/// EndeavourOS package index fetcher with configurable repositories.
pub struct EndeavourOs {
    repos: Vec<EndeavourOsRepo>,
}

impl EndeavourOs {
    const ARCH_API: &'static str = "https://archlinux.org/packages/search/json/";
    const AUR_RPC: &'static str = "https://aur.archlinux.org/rpc/";

    /// Create a fetcher with all repositories.
    pub fn all() -> Self {
        Self {
            repos: EndeavourOsRepo::all().to_vec(),
        }
    }

    /// Create a fetcher with EndeavourOS repos only.
    pub fn endeavouros_only() -> Self {
        Self {
            repos: EndeavourOsRepo::endeavouros_only().to_vec(),
        }
    }

    /// Create a fetcher with Arch repos.
    pub fn arch() -> Self {
        Self {
            repos: EndeavourOsRepo::arch().to_vec(),
        }
    }

    /// Create a fetcher with stable repos.
    pub fn stable() -> Self {
        Self {
            repos: EndeavourOsRepo::stable().to_vec(),
        }
    }

    /// Create a fetcher with custom repository selection.
    pub fn with_repos(repos: &[EndeavourOsRepo]) -> Self {
        Self {
            repos: repos.to_vec(),
        }
    }

    /// Load packages from a single repository database.
    fn load_repo(repo: EndeavourOsRepo) -> Result<Vec<PackageMeta>, IndexError> {
        if repo == EndeavourOsRepo::Aur {
            return arch_common::fetch_all_aur();
        }

        let url = repo
            .db_url()
            .ok_or_else(|| IndexError::NotFound(format!("no db URL for {}", repo.name())))?;

        let (data, _was_cached) = cache::fetch_with_cache(
            "endeavouros",
            &format!("{}-db", repo.name()),
            &url,
            CACHE_TTL,
        )
        .map_err(IndexError::Network)?;

        Self::parse_db(&data, repo)
    }

    /// Parse an Arch-style .db file.
    fn parse_db(data: &[u8], repo: EndeavourOsRepo) -> Result<Vec<PackageMeta>, IndexError> {
        arch_common::parse_arch_db(data, |content| {
            arch_common::parse_arch_db_desc(content, repo.name())
        })
    }

    /// Load packages from all configured repositories in parallel.
    fn load_packages(&self) -> Result<Vec<PackageMeta>, IndexError> {
        let results: Vec<_> = self
            .repos
            .par_iter()
            .map(|&repo| Self::load_repo(repo))
            .collect();

        let mut packages = Vec::new();
        for result in results {
            match result {
                Ok(pkgs) => packages.extend(pkgs),
                Err(e) => {
                    tracing::warn!("failed to load EndeavourOS repo: {}", e);
                }
            }
        }

        Ok(packages)
    }
}

impl PackageIndex for EndeavourOs {
    fn ecosystem(&self) -> &'static str {
        "endeavouros"
    }

    fn display_name(&self) -> &'static str {
        "EndeavourOS"
    }

    fn fetch(&self, name: &str) -> Result<PackageMeta, IndexError> {
        // Try EndeavourOS repos first, then fall back to Arch API + AUR
        let packages = self.load_packages()?;
        if let Some(pkg) = packages.into_iter().find(|p| p.name == name) {
            return Ok(pkg);
        }

        arch_common::fetch_official(Self::ARCH_API, name)
            .or_else(|_| arch_common::fetch_aur(Self::AUR_RPC, name))
    }

    fn fetch_versions(&self, name: &str) -> Result<Vec<VersionMeta>, IndexError> {
        let packages = self.load_packages()?;

        let versions: Vec<_> = packages
            .into_iter()
            .filter(|p| p.name == name)
            .map(|p| VersionMeta {
                version: p.version,
                released: None,
                yanked: false,
            })
            .collect();

        if versions.is_empty() {
            return Err(IndexError::NotFound(name.to_string()));
        }

        Ok(versions)
    }

    fn supports_fetch_all(&self) -> bool {
        true
    }

    fn fetch_all(&self) -> Result<Vec<PackageMeta>, IndexError> {
        self.load_packages()
    }

    fn search(&self, query: &str) -> Result<Vec<PackageMeta>, IndexError> {
        let packages = self.load_packages()?;
        let query_lower = query.to_lowercase();

        let mut results: Vec<_> = packages
            .into_iter()
            .filter(|p| {
                p.name.to_lowercase().contains(&query_lower)
                    || p.description
                        .as_ref()
                        .map(|d| d.to_lowercase().contains(&query_lower))
                        .unwrap_or(false)
            })
            .collect();

        // Also search Arch API if AUR is included
        if self.repos.contains(&EndeavourOsRepo::Aur) {
            if let Ok(arch_packages) = arch_common::search_official(Self::ARCH_API, query) {
                results.extend(arch_packages);
            }
            if let Ok(aur_packages) = arch_common::search_aur(Self::AUR_RPC, query) {
                results.extend(aur_packages);
            }
        }

        Ok(results)
    }
}