//! ref: composer/src/Composer/Repository/CompositeRepository.php

use std::any::Any;

use indexmap::IndexMap;
use shirabe_semver::constraint::constraint_interface::ConstraintInterface;

use crate::package::base_package::BasePackage;
use crate::package::package_interface::PackageInterface;
use crate::repository::repository_interface::{
    FindPackageConstraint, LoadPackagesResult, ProviderInfo, RepositoryInterface, SearchResult,
};

#[derive(Debug)]
pub struct CompositeRepository {
    repositories: Vec<Box<dyn RepositoryInterface>>,
}

impl CompositeRepository {
    pub fn new(repositories: Vec<Box<dyn RepositoryInterface>>) -> Self {
        let mut this = Self {
            repositories: vec![],
        };
        for repo in repositories {
            this.add_repository(repo);
        }
        this
    }

    pub fn get_repositories(&self) -> &Vec<Box<dyn RepositoryInterface>> {
        &self.repositories
    }

    pub fn remove_package(&mut self, package: &dyn PackageInterface) {
        for repository in &mut self.repositories {
            // TODO(phase-b): only call remove_package on WritableRepositoryInterface implementors
            let _ = repository.remove_package(package);
        }
    }

    pub fn add_repository(&mut self, repository: Box<dyn RepositoryInterface>) {
        if let Some(composite) =
            (repository.as_any() as &dyn Any).downcast_ref::<CompositeRepository>()
        {
            for repo in composite.get_repositories() {
                self.repositories.push(repo.clone_box());
            }
        } else {
            self.repositories.push(repository);
        }
    }
}

impl shirabe_php_shim::Countable for CompositeRepository {
    fn count(&self) -> i64 {
        self.repositories.iter().map(|r| r.count()).sum()
    }
}

impl RepositoryInterface for CompositeRepository {
    fn get_repo_name(&self) -> String {
        let names: Vec<String> = self
            .repositories
            .iter()
            .map(|r| r.get_repo_name())
            .collect();
        format!("composite repo ({})", names.join(", "))
    }

    fn has_package(&self, package: &dyn PackageInterface) -> bool {
        for repository in &self.repositories {
            if repository.has_package(package) {
                return true;
            }
        }
        false
    }

    fn find_package(
        &self,
        name: String,
        constraint: FindPackageConstraint,
    ) -> Option<Box<BasePackage>> {
        for repository in &self.repositories {
            let package = repository.find_package(name.clone(), constraint.clone());
            if package.is_some() {
                return package;
            }
        }
        None
    }

    fn find_packages(
        &self,
        name: String,
        constraint: Option<FindPackageConstraint>,
    ) -> Vec<Box<BasePackage>> {
        let mut packages = vec![];
        for repository in &self.repositories {
            packages.extend(repository.find_packages(name.clone(), constraint.clone()));
        }
        packages
    }

    fn get_packages(&self) -> Vec<Box<BasePackage>> {
        let mut packages = vec![];
        for repository in &self.repositories {
            packages.extend(repository.get_packages());
        }
        packages
    }

    fn load_packages(
        &self,
        package_name_map: IndexMap<String, Option<Box<dyn ConstraintInterface>>>,
        acceptable_stabilities: IndexMap<String, i64>,
        stability_flags: IndexMap<String, i64>,
        already_loaded: IndexMap<String, IndexMap<String, Box<dyn PackageInterface>>>,
    ) -> LoadPackagesResult {
        let mut all_packages = vec![];
        let mut all_names_found = vec![];

        for repository in &self.repositories {
            let result = repository.load_packages(
                package_name_map.clone(),
                acceptable_stabilities.clone(),
                stability_flags.clone(),
                already_loaded.clone(),
            );
            all_packages.extend(result.packages);
            all_names_found.extend(result.names_found);
        }

        let mut seen = std::collections::HashSet::new();
        let unique_names: Vec<String> = all_names_found
            .into_iter()
            .filter(|s| seen.insert(s.clone()))
            .collect();

        LoadPackagesResult {
            packages: all_packages,
            names_found: unique_names,
        }
    }

    fn search(&self, query: String, mode: i64, r#type: Option<String>) -> Vec<SearchResult> {
        let mut matches = vec![];
        for repository in &self.repositories {
            matches.extend(repository.search(query.clone(), mode, r#type.clone()));
        }
        matches
    }

    fn get_providers(&self, package_name: String) -> IndexMap<String, ProviderInfo> {
        let mut results = IndexMap::new();
        for repository in &self.repositories {
            results.extend(repository.get_providers(package_name.clone()));
        }
        results
    }
}