From f349711071dc66696fc120af122620de1b9234ee Mon Sep 17 00:00:00 2001 From: nsfisis Date: Sat, 16 May 2026 00:03:24 +0900 Subject: feat(port): port CompositeRepository.php Co-Authored-By: Claude Sonnet 4.6 --- .../shirabe/src/repository/composite_repository.rs | 144 +++++++++++++++++++++ 1 file changed, 144 insertions(+) (limited to 'crates/shirabe/src') diff --git a/crates/shirabe/src/repository/composite_repository.rs b/crates/shirabe/src/repository/composite_repository.rs index 9e4dfde..9787b6a 100644 --- a/crates/shirabe/src/repository/composite_repository.rs +++ b/crates/shirabe/src/repository/composite_repository.rs @@ -1 +1,145 @@ //! 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>, +} + +impl CompositeRepository { + pub fn new(repositories: Vec>) -> Self { + let mut this = Self { repositories: vec![] }; + for repo in repositories { + this.add_repository(repo); + } + this + } + + pub fn get_repositories(&self) -> &Vec> { + &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) { + if let Some(composite) = (repository.as_any() as &dyn Any).downcast_ref::() { + 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 = 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> { + 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) -> Vec> { + let mut packages = vec![]; + for repository in &self.repositories { + packages.extend(repository.find_packages(name.clone(), constraint.clone())); + } + packages + } + + fn get_packages(&self) -> Vec> { + let mut packages = vec![]; + for repository in &self.repositories { + packages.extend(repository.get_packages()); + } + packages + } + + fn load_packages( + &self, + package_name_map: IndexMap>>, + acceptable_stabilities: IndexMap, + stability_flags: IndexMap, + already_loaded: IndexMap>>, + ) -> 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 = 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) -> Vec { + 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 { + let mut results = IndexMap::new(); + for repository in &self.repositories { + results.extend(repository.get_providers(package_name.clone())); + } + results + } +} -- cgit v1.3.1