vivacity_resolver/

optimizer.rs

//! Port of `Composer\DependencyResolver\PoolOptimizer`: removes from the
//! pool the versions whose dependencies are identical to a preferred
//! version, and those a locked package makes impossible.

use crate::constraint::{Constraint, Op};
use crate::intervals;
use crate::package::Package;
use crate::policy::DefaultPolicy;
use crate::pool::{OrderedMap, Pool, Request};
use std::collections::{BTreeMap, HashMap, HashSet};

/// String-keyed PHP array with an index: insertion order and O(1) access.
#[derive(Debug, Clone)]
struct IndexedMap<V> {
    entries: Vec<(String, V)>,
    index: HashMap<String, usize>,
}

impl<V> Default for IndexedMap<V> {
    fn default() -> Self {
        IndexedMap {
            entries: Vec::new(),
            index: HashMap::new(),
        }
    }
}

impl<V> IndexedMap<V> {
    fn contains(&self, key: &str) -> bool {
        self.index.contains_key(key)
    }
    /// `$map[$key] ??= $default` then a mutable reference.
    fn entry_or_insert_with(&mut self, key: &str, default: impl FnOnce() -> V) -> &mut V {
        let i = match self.index.get(key) {
            Some(&i) => i,
            None => {
                self.entries.push((key.to_owned(), default()));
                let i = self.entries.len() - 1;
                self.index.insert(key.to_owned(), i);
                i
            }
        };
        &mut self.entries[i].1
    }
    fn iter(&self) -> impl Iterator<Item = (&String, &V)> {
        self.entries.iter().map(|(k, v)| (k, v))
    }
}

pub struct PoolOptimizer {
    /// Textual form of a constraint -> its disjunctive pieces
    /// (`expandDisjunctiveMultiConstraints`), memoized: the same link text
    /// shows up thousands of times in a pool.
    expansion_cache: HashMap<String, Vec<(String, Constraint)>>,
    irremovable: HashSet<usize>,
    /// name -> constraints (deduplicated by textual form, insertion order).
    require_constraints: HashMap<String, IndexedMap<Constraint>>,
    conflict_constraints: HashMap<String, IndexedMap<Constraint>>,
    to_remove: HashSet<usize>,
    /// base id -> alias ids.
    aliases_per_package: HashMap<usize, Vec<usize>>,
    /// `removedVersionsByPackage`, keyed by arena index (PHP: object id).
    removed_versions_by_package: BTreeMap<usize, BTreeMap<String, String>>,
}

impl PoolOptimizer {
    pub fn new() -> PoolOptimizer {
        PoolOptimizer {
            expansion_cache: HashMap::new(),
            irremovable: HashSet::new(),
            require_constraints: HashMap::new(),
            conflict_constraints: HashMap::new(),
            to_remove: HashSet::new(),
            aliases_per_package: HashMap::new(),
            removed_versions_by_package: BTreeMap::new(),
        }
    }

    /// `optimize($request, $pool)`: the reduced pool (same arena indices,
    /// new ids).
    pub fn optimize(
        mut self,
        request: &Request,
        pool: &Pool,
        arena: &[Package],
        policy: &mut DefaultPolicy,
    ) -> Pool {
        self.prepare(request, pool, arena);
        self.optimize_by_identical_dependencies(pool, arena, policy);
        self.optimize_impossible_packages_away(request, pool, arena);
        // `applyRemovalsToPool`: the dropped versions are remembered for the
        // messages, per name and per kept package.
        let mut kept: Vec<usize> = Vec::new();
        let mut removed_versions: BTreeMap<String, BTreeMap<String, String>> = BTreeMap::new();
        for id in 1..=pool.len() {
            let idx = pool.package_by_id(id);
            if self.to_remove.contains(&id) {
                let p = &arena[idx];
                removed_versions
                    .entry(p.name.clone())
                    .or_default()
                    .insert(p.version.clone(), p.pretty_version.clone());
            } else {
                kept.push(idx);
            }
        }
        let mut optimized = pool.with_packages(kept, arena);
        optimized.removed_versions = removed_versions;
        optimized.removed_versions_by_package = self.removed_versions_by_package;
        optimized
    }

    fn expand_disjunctive(constraint: &Constraint) -> Vec<Constraint> {
        let compact = intervals::compact_constraint(constraint);
        match &compact {
            Constraint::Multi {
                constraints,
                conjunctive: false,
            } => constraints.clone(),
            _ => vec![compact],
        }
    }

    /// `extractRequireConstraintsPerPackage` / `...Conflict...`: `pretty` is
    /// the memoization key (same text -> same parsed constraint).
    fn extract(
        cache: &mut HashMap<String, Vec<(String, Constraint)>>,
        map: &mut HashMap<String, IndexedMap<Constraint>>,
        package: &str,
        pretty: &str,
        constraint: &Constraint,
    ) {
        // `self.version`: same text, different constraint per package.
        let pretty = if pretty == "self.version" {
            format!("\u{0}{constraint}")
        } else {
            pretty.to_owned()
        };
        let expansions = cache.entry(pretty).or_insert_with(|| {
            Self::expand_disjunctive(constraint)
                .into_iter()
                .map(|c| (c.to_string(), c))
                .collect()
        });
        let per_name = map.entry(package.to_owned()).or_default();
        for (key, expanded) in expansions {
            // `$map[$package][(string) $expanded] = $expanded`: in-place
            // rewrite, same textual form -> same constraint.
            if !per_name.contains(key) {
                per_name.entry_or_insert_with(key, || expanded.clone());
            }
        }
    }

    /// `prepare`.
    fn prepare(&mut self, request: &Request, pool: &Pool, arena: &[Package]) {
        let mut irremovable_groups: OrderedMap<Vec<Constraint>> = OrderedMap::default();
        for idx in request.fixed_or_locked_packages() {
            let p = &arena[idx];
            let c = Constraint::new(Op::Eq, p.version.clone());
            match irremovable_groups.0.iter_mut().find(|(k, _)| *k == p.name) {
                Some((_, list)) => list.push(c),
                None => irremovable_groups.insert(&p.name, vec![c]),
            }
        }
        for (name, constraint) in request.requires.iter() {
            // Root constraints have no stable text at hand: their display
            // form serves as the key.
            let pretty = format!("\u{0}{constraint}");
            Self::extract(
                &mut self.expansion_cache,
                &mut self.require_constraints,
                name,
                &pretty,
                constraint,
            );
        }
        for id in 1..=pool.len() {
            let p = &arena[pool.package_by_id(id)];
            for link in p.requires.iter() {
                Self::extract(
                    &mut self.expansion_cache,
                    &mut self.require_constraints,
                    &link.target,
                    &link.pretty_constraint,
                    &link.constraint,
                );
            }
            for link in p.conflicts.iter() {
                Self::extract(
                    &mut self.expansion_cache,
                    &mut self.conflict_constraints,
                    &link.target,
                    &link.pretty_constraint,
                    &link.constraint,
                );
            }
            if let Some(base) = p.alias_of.and_then(|b| pool.id_of(b)) {
                self.aliases_per_package.entry(base).or_default().push(id);
            }
        }
        let irremovable: HashMap<String, Constraint> = irremovable_groups
            .0
            .into_iter()
            .map(|(name, constraints)| {
                let c = if constraints.len() == 1 {
                    constraints.into_iter().next().expect("one")
                } else {
                    Constraint::Multi {
                        constraints,
                        conjunctive: false,
                    }
                };
                (name, c)
            })
            .collect();
        for id in 1..=pool.len() {
            let p = &arena[pool.package_by_id(id)];
            let Some(c) = irremovable.get(&p.name) else {
                continue;
            };
            if c.matches_version(&p.version) {
                self.mark_irremovable(id, pool, arena);
            }
        }
    }

    fn mark_irremovable(&mut self, id: usize, pool: &Pool, arena: &[Package]) {
        self.irremovable.insert(id);
        if let Some(base) = arena[pool.package_by_id(id)]
            .alias_of
            .and_then(|b| pool.id_of(b))
        {
            self.mark_irremovable(base, pool, arena);
        }
        if let Some(aliases) = self.aliases_per_package.get(&id) {
            for a in aliases.clone() {
                self.irremovable.insert(a);
            }
        }
    }

    /// `calculateDependencyHash`.
    fn dependency_hash(p: &Package) -> String {
        let mut hash = String::new();
        for (key, links) in [
            ("requires", &p.requires),
            ("conflicts", &p.conflicts),
            ("replaces", &p.replaces),
            ("provides", &p.provides),
        ] {
            if links.is_empty() {
                continue;
            }
            hash.push_str(key);
            hash.push(':');
            let mut sub: BTreeMap<Vec<u8>, String> = BTreeMap::new();
            for link in links.iter() {
                sub.insert(link.target.as_bytes().to_vec(), link.constraint.to_string());
            }
            for (target, constraint) in sub {
                hash.push_str(&String::from_utf8_lossy(&target));
                hash.push('@');
                hash.push_str(&constraint);
            }
        }
        hash
    }

    /// `optimizeByIdenticalDependencies`.
    fn optimize_by_identical_dependencies(
        &mut self,
        pool: &Pool,
        arena: &[Package],
        policy: &mut DefaultPolicy,
    ) {
        // name -> groupHash -> dependencyHash -> ids (insertion orders).
        let mut identical: IndexedMap<IndexedMap<IndexedMap<Vec<usize>>>> = IndexedMap::default();
        for id in 1..=pool.len() {
            if self.irremovable.contains(&id) {
                continue;
            }
            self.to_remove.insert(id);
            let p = &arena[pool.package_by_id(id)];
            let dependency_hash = Self::dependency_hash(p);
            // The `replace` pieces do not depend on the constraint under
            // examination: once per package.
            let replace_parts: String = p
                .replaces
                .iter()
                .filter(|l| l.constraint.matches_version(&p.version))
                .map(|l| format!("require:{}", l.constraint))
                .collect();
            for name in p.names(false) {
                let Some(requires) = self.require_constraints.get(&name) else {
                    continue;
                };
                // Same for conflicts: once per name.
                let conflict_parts: String = match self.conflict_constraints.get(&name) {
                    Some(conflicts) => conflicts
                        .iter()
                        .filter(|(_, c)| c.matches_version(&p.version))
                        .map(|(key, _)| format!("conflict:{key}"))
                        .collect(),
                    None => String::new(),
                };
                for (key, require_constraint) in requires.iter() {
                    let matched = require_constraint.matches_version(&p.version);
                    if !matched && replace_parts.is_empty() && conflict_parts.is_empty() {
                        continue;
                    }
                    let mut group_hash = String::with_capacity(
                        key.len() + 8 + replace_parts.len() + conflict_parts.len(),
                    );
                    if matched {
                        group_hash.push_str("require:");
                        group_hash.push_str(key);
                    }
                    group_hash.push_str(&replace_parts);
                    group_hash.push_str(&conflict_parts);
                    identical
                        .entry_or_insert_with(&name, IndexedMap::default)
                        .entry_or_insert_with(&group_hash, IndexedMap::default)
                        .entry_or_insert_with(&dependency_hash, Vec::new)
                        .push(id);
                }
            }
        }
        for (name, groups) in identical.entries {
            for (_, group) in groups.entries {
                for (_, ids) in group.entries {
                    if ids.len() == 1 {
                        self.keep_package_in_group(ids[0], pool, arena, &name, &ids);
                        continue;
                    }
                    let literals: Vec<i64> = ids.iter().map(|&i| i as i64).collect();
                    for preferred in policy.select_preferred_packages(pool, arena, &literals, None)
                    {
                        self.keep_package_in_group(preferred as usize, pool, arena, &name, &ids);
                    }
                }
            }
        }
    }

    /// `keepPackageInGroup`, with `recordRemovedVersionsForPackage` at the
    /// reference's exact points (the group's versions are recorded on the
    /// kept package BEFORE the early return, then on its alias/aliased
    /// packages).
    fn keep_package_in_group(
        &mut self,
        id: usize,
        pool: &Pool,
        arena: &[Package],
        name: &str,
        ids: &[usize],
    ) {
        let mut versions: BTreeMap<String, String> = BTreeMap::new();
        for &gid in ids {
            let mut idx = pool.package_by_id(gid);
            let gp = &arena[idx];
            if let Some(base) = gp.alias_of.filter(|_| gp.pretty_version == "9999999-dev") {
                idx = base;
            }
            versions.insert(
                arena[idx].version.clone(),
                arena[idx].pretty_version.clone(),
            );
        }
        let idx = pool.package_by_id(id);
        self.record_removed_versions(idx, arena, name, &versions);
        if !self.to_remove.contains(&id) {
            return;
        }
        self.to_remove.remove(&id);
        let p = &arena[idx];
        if let Some(base_idx) = p.alias_of {
            if let Some(base) = pool.id_of(base_idx) {
                self.to_remove.remove(&base);
                self.record_removed_versions(base_idx, arena, name, &versions);
                if let Some(aliases) = self.aliases_per_package.get(&base).cloned() {
                    for a in aliases {
                        self.to_remove.remove(&a);
                        self.record_removed_versions(pool.package_by_id(a), arena, name, &versions);
                    }
                }
            }
            return;
        }
        if let Some(aliases) = self.aliases_per_package.get(&id).cloned() {
            for a in aliases {
                self.to_remove.remove(&a);
                self.record_removed_versions(pool.package_by_id(a), arena, name, &versions);
            }
        }
    }

    /// `recordRemovedVersionsForPackage`: only when the package carries
    /// `name` itself (`getNames(false)`: name + replaces).
    fn record_removed_versions(
        &mut self,
        idx: usize,
        arena: &[Package],
        name: &str,
        versions: &BTreeMap<String, String>,
    ) {
        if !arena[idx].names(false).iter().any(|n| n == name) {
            return;
        }
        let entry = self.removed_versions_by_package.entry(idx).or_default();
        for (v, p) in versions {
            entry.insert(v.clone(), p.clone());
        }
    }

    /// `optimizeImpossiblePackagesAway`.
    fn optimize_impossible_packages_away(
        &mut self,
        request: &Request,
        pool: &Pool,
        arena: &[Package],
    ) {
        let locked = request.locked_packages_all();
        if locked.is_empty() {
            return;
        }
        // name -> [(id, arena idx)] (pool order).
        let mut index: HashMap<String, Vec<(usize, usize)>> = HashMap::new();
        for id in 1..=pool.len() {
            if self.irremovable.contains(&id) {
                continue;
            }
            let idx = pool.package_by_id(id);
            let p = &arena[idx];
            if self.aliases_per_package.contains_key(&id) || p.is_alias() {
                continue;
            }
            if request.is_fixed_package(idx) || request.is_locked_package(idx) {
                continue;
            }
            index.entry(p.name.clone()).or_default().push((id, idx));
        }
        for locked_idx in locked {
            let p = &arena[locked_idx];
            let unused = p
                .names(false)
                .iter()
                .all(|n| !self.require_constraints.contains_key(n));
            if unused {
                continue;
            }
            for link in p.requires.iter() {
                let Some(candidates) = index.get_mut(&link.target) else {
                    continue;
                };
                let mut kept = Vec::new();
                for (id, idx) in candidates.iter().copied() {
                    if !link.constraint.matches_version(&arena[idx].version) {
                        self.to_remove.insert(id);
                    } else {
                        kept.push((id, idx));
                    }
                }
                *candidates = kept;
            }
        }
    }
}

impl Default for PoolOptimizer {
    fn default() -> Self {
        Self::new()
    }
}