crates/shirabe/src/repository/canonical_packages_trait.rs


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36

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

use crate::package::package_interface::PackageInterface;
use indexmap::IndexMap;

/// Provides get_canonical_packages() to various repository implementations.
pub trait CanonicalPackagesTrait {
    fn get_packages(&self) -> Vec<Box<dyn PackageInterface>>;

    /// Get unique packages (at most one package of each name), with aliases resolved and removed.
    fn get_canonical_packages(&self) -> Vec<Box<dyn PackageInterface>> {
        let packages = self.get_packages();

        // get at most one package of each name, preferring non-aliased ones
        let mut packages_by_name: IndexMap<String, Box<dyn PackageInterface>> = IndexMap::new();
        for package in packages {
            let name = package.get_name();
            if !packages_by_name.contains_key(&name) || packages_by_name[&name].is_alias_package() {
                packages_by_name.insert(name, package);
            }
        }

        let mut canonical_packages = Vec::new();

        // unfold aliased packages
        for mut package in packages_by_name.into_values() {
            while package.is_alias_package() {
                package = package.get_alias_of();
            }

            canonical_packages.push(package);
        }

        canonical_packages
    }
}