Skip to main content

aube_resolver/
peer_context.rs

1//! Peer-dependency post-processing over an already-resolved graph.
2//!
3//! Two user-visible passes live here:
4//!
5//! * [`hoist_auto_installed_peers`] — promotes peers declared by direct
6//!   dependencies up to importer direct deps, matching pnpm's
7//!   `auto-install-peers=true` behavior. Idempotent on graphs that already
8//!   ship with those hoists (npm v7+ output, lockfile-driven installs).
9//! * [`apply_peer_contexts`] — computes pnpm-style `(peer@ver)` suffixes
10//!   on contextualized `dep_path`s. Drives the sibling-symlink wiring in
11//!   `aube-linker` so each subtree that pins different peer versions gets
12//!   its own virtual-store entry.
13//!
14//! [`detect_unmet_peers`] reports what the two passes above couldn't wire
15//! up, so the CLI can surface warnings.
16//!
17//! Call order from `Resolver::resolve`: `hoist_auto_installed_peers`
18//! (fresh resolves only) → `apply_peer_contexts` → `detect_unmet_peers`.
19
20use crate::version_satisfies;
21use aube_lockfile::{DepType, DirectDep, LockedPackage, LockfileGraph};
22use rustc_hash::{FxHashMap, FxHashSet};
23use std::collections::{BTreeMap, BTreeSet};
24
25/// A peer dependency whose declared range doesn't match the version the
26/// tree actually ends up providing. Emitted as a warning by `aube install`.
27#[derive(Debug, Clone, PartialEq, Eq)]
28pub struct UnmetPeer {
29    /// dep_path of the package that declared the peer.
30    pub from_dep_path: String,
31    /// Human-friendly package name (pre-context) for display.
32    pub from_name: String,
33    /// Name of the peer being declared (e.g. `"react"`).
34    pub peer_name: String,
35    /// The declared peer range from the package's packument
36    /// (e.g. `"^16.8.0 || ^17.0.0 || ^18.0.0"`).
37    pub declared: String,
38    /// What the tree actually provides, if anything. `None` means the
39    /// peer is completely missing — rare in practice because the BFS
40    /// auto-install path usually drags *some* version in, but it can
41    /// happen for corner cases.
42    pub found: Option<String>,
43}
44
45/// Scan the resolved graph and return every declared required peer whose
46/// resolved version doesn't satisfy its declared range. Optional peers
47/// (`peerDependenciesMeta.optional = true`) are skipped — pnpm treats
48/// those as "warn suppressed" with `auto-install-peers=true`. The result
49/// is purely informational; aube never fails an install on unmet peers,
50/// matching pnpm.
51///
52/// The "found" version for each package comes from its own
53/// `dependencies` map — the peer-context pass writes the resolved peer
54/// tail there, so we don't have to re-walk ancestors. Any peer suffix on
55/// the stored tail is stripped before the semver check so `18.2.0(foo@1)`
56/// is treated as `18.2.0`.
57pub fn detect_unmet_peers(graph: &LockfileGraph) -> Vec<UnmetPeer> {
58    let mut unmet = Vec::new();
59    for pkg in graph.packages.values() {
60        for (peer_name, declared_range) in &pkg.peer_dependencies {
61            let optional = pkg
62                .peer_dependencies_meta
63                .get(peer_name)
64                .map(|m| m.optional)
65                .unwrap_or(false);
66            if optional {
67                continue;
68            }
69
70            let found_tail = pkg.dependencies.get(peer_name);
71            let found_version = found_tail.map(|t| canonical_tail(t).to_string());
72
73            let satisfied = match &found_version {
74                Some(v) => version_satisfies(v, declared_range),
75                None => false,
76            };
77            if satisfied {
78                continue;
79            }
80
81            unmet.push(UnmetPeer {
82                from_dep_path: pkg.dep_path.clone(),
83                from_name: pkg.name.clone(),
84                peer_name: peer_name.clone(),
85                declared: declared_range.clone(),
86                found: found_version,
87            });
88        }
89    }
90    // Stable order for deterministic test output and readable warnings.
91    unmet.sort_by(|a, b| {
92        (a.from_dep_path.as_str(), a.peer_name.as_str())
93            .cmp(&(b.from_dep_path.as_str(), b.peer_name.as_str()))
94    });
95    unmet
96}
97
98/// Promote direct dependencies' unmet peers to importer direct deps.
99///
100/// Walks each importer's direct dependencies and hoists any peer they
101/// declare that isn't already a direct dep of the importer up to the
102/// importer's `dependencies` list — what pnpm's
103/// `auto-install-peers=true` produces in its v9 lockfile. Peers declared by
104/// transitive dependencies stay in the resolved graph for peer-context
105/// sibling wiring, but they are not surfaced as top-level
106/// `node_modules/<peer>` entries.
107///
108/// Public so lockfile-driven installs that need to re-derive peer
109/// wiring (npm/yarn/bun formats, which don't record peer contexts)
110/// can run this before [`apply_peer_contexts`] to match fresh-resolve
111/// behavior. Idempotent in the npm case: npm v7+ already hoists
112/// auto-installed peers into root's `dependencies`, so they arrive
113/// pre-`satisfied` and no additions are emitted.
114///
115/// Algorithm:
116///   1. For each importer, collect the set of names already in its
117///      direct deps. Those are "satisfied" and need no hoist.
118///   2. Visit only those direct dependency packages and examine their
119///      `peer_dependencies` declarations. For each declared peer not
120///      already satisfied by the importer, find a resolved version somewhere
121///      in the graph and synthesize a `DirectDep` entry. Mark it as
122///      satisfied so a second direct dep doesn't add a duplicate.
123///   3. Stable: we walk in-order and take the first declared peer range
124///      encountered per name as the specifier. Conflicting ranges across
125///      the tree are not reconciled — first one wins. This matches pnpm
126///      for the simple case; the complex case is deferred.
127///
128/// Leaves everything else about the graph untouched — no packages are
129/// added or removed, only importer entries grow.
130pub fn hoist_auto_installed_peers(mut graph: LockfileGraph) -> LockfileGraph {
131    let importer_paths: Vec<String> = graph.importers.keys().cloned().collect();
132    for importer_path in importer_paths {
133        let Some(direct_deps) = graph.importers.get(&importer_path) else {
134            continue;
135        };
136        let mut satisfied: FxHashSet<String> = direct_deps.iter().map(|d| d.name.clone()).collect();
137
138        // Additions are gathered into a separate vec so we don't mutate
139        // the importer's direct-dep list while still borrowing from it.
140        let mut additions: Vec<DirectDep> = Vec::new();
141
142        for dep_path in direct_deps.iter().map(|d| &d.dep_path) {
143            let Some(pkg) = graph.packages.get(dep_path) else {
144                continue;
145            };
146
147            // Collect unmet peer declarations from this package.
148            for (peer_name, peer_range) in &pkg.peer_dependencies {
149                if satisfied.contains(peer_name) {
150                    continue;
151                }
152                // Find any resolved version in the graph for this peer.
153                // Prefer the one the package already wired via its own
154                // dependencies map (the BFS auto-install result), and
155                // fall back to scanning `graph.packages` for a name
156                // match. If nothing matches, we quietly drop the peer —
157                // that's the only path where aube stays stricter than
158                // pnpm today; a future PR will emit an unmet warning.
159                //
160                // Fallback takes the semver-max version rather than
161                // whatever `BTreeMap` iteration order surfaces first —
162                // otherwise two resolved `react` entries like `18.0.0`
163                // and `18.3.1` would pick the lexicographically-earlier
164                // (older) one.
165                let resolved_version = pkg.dependencies.get(peer_name).cloned().or_else(|| {
166                    // Filter to parseable semver versions *before* the
167                    // max_by — returning `Equal` on parse failure makes
168                    // the comparator non-transitive, so an unparseable
169                    // entry sitting between two valid ones would cause
170                    // `max_by` to pick an iteration-order-dependent
171                    // result instead of the true maximum.
172                    graph
173                        .packages
174                        .values()
175                        .filter(|p| p.name == *peer_name)
176                        .filter_map(|p| {
177                            node_semver::Version::parse(&p.version)
178                                .ok()
179                                .map(|v| (v, p.version.clone()))
180                        })
181                        .max_by(|a, b| a.0.cmp(&b.0))
182                        .map(|(_, s)| s)
183                });
184                let Some(version) = resolved_version else {
185                    continue;
186                };
187                let canonical_version = canonical_tail(&version).to_string();
188                let synth_dep_path = format!("{peer_name}@{canonical_version}");
189                if !graph.packages.contains_key(&synth_dep_path) {
190                    // The peer version the package wired didn't match an
191                    // actual package entry — bail out for this peer
192                    // rather than writing a dangling DirectDep.
193                    continue;
194                }
195                satisfied.insert(peer_name.clone());
196                additions.push(DirectDep {
197                    name: peer_name.clone(),
198                    dep_path: synth_dep_path,
199                    // Peers auto-hoisted to the root are in the prod
200                    // graph by convention — matches what pnpm writes.
201                    dep_type: DepType::Production,
202                    specifier: Some(peer_range.clone()),
203                });
204            }
205        }
206
207        if !additions.is_empty() {
208            tracing::debug!(
209                "hoisted {} auto-installed peer(s) into importer {}",
210                additions.len(),
211                importer_path
212            );
213            if let Some(deps) = graph.importers.get_mut(&importer_path) {
214                deps.extend(additions);
215                deps.sort_by(|a, b| a.name.cmp(&b.name));
216            }
217        }
218    }
219    graph
220}
221
222/// Walk the resolved graph top-down from each importer and compute a
223/// peer-dependency context for every package, producing a new graph whose
224/// dep_paths carry pnpm-style `(peer@ver)` suffixes.
225///
226/// The goal is parity with pnpm's v9 lockfile output: the same
227/// `name@version` can appear multiple times — once per distinct set of peer
228/// resolutions — so different subtrees that pin incompatible peers get
229/// isolated virtual-store entries and truly different sibling-symlink
230/// neighborhoods.
231///
232/// Algorithm per visited package P, reached at some point in a DFS from an
233/// importer with `ancestor_scope: name -> dep_path_tail`:
234///
235///  1. For each peer name declared by P, look it up in `ancestor_scope`
236///     (nearest-ancestor-wins, since the scope is rebuilt per recursion).
237///     If missing, fall back to P's own entry in `dependencies` — the BFS
238///     enqueue above auto-installed it as a transitive, which matches
239///     pnpm's `auto-install-peers=true` default.
240///  2. Sort the (peer_name, resolution) pairs and serialize as
241///     `(n1@v1)(n2@v2)…` for the suffix.
242///  3. Produce a contextualized dep_path `name@version{suffix}`. If that
243///     key is already in `out_packages` (or currently on the DFS stack via
244///     `visiting`), short-circuit — we've already emitted this variant.
245///  4. Build a new scope for P's children by merging the ancestor scope
246///     with P's own `dependencies` (rewritten to point at contextualized
247///     children) and the resolved peer map. Recurse.
248///  5. Emit the contextualized LockedPackage.
249///
250/// Cycles: protected by `visiting` — if a package is re-entered via a
251/// dependency cycle, we return the already-computed dep_path without
252/// recursing again. The peer context is fixed at first visit; any cycle
253/// traversal uses whatever context was live at that first visit.
254///
255/// Nested peer suffixes: pnpm writes `(react-dom@18.2.0(react@18.2.0))`
256/// when a declared peer has its own resolved peers. A single top-down
257/// DFS pass can't produce that form, because when a parent P records
258/// a peer version in its children's scope, it only knows the canonical
259/// tail — the peer's OWN suffix is computed later when the peer itself
260/// gets visited. We solve this by running `apply_peer_contexts_once` in
261/// a fixed-point loop: the second iteration's input has Pass 1's
262/// contextualized tails in every `pkg.dependencies` map, so when a
263/// descendant looks a peer up in ancestor scope it sees the full
264/// nested tail and serializes it as such. Most peer chains converge in
265/// 2–3 iterations; we cap at 16 as a safety belt.
266///
267/// Limitations (documented as follow-ups in the README):
268///   - No per-peer range satisfaction — we take whatever the ancestor has,
269///     even if it technically doesn't match P's declared peer range.
270///
271/// Knobs controlling the peer-context pass. Plumbed from four
272/// pnpm-compatible settings (`dedupe-peer-dependents`, `dedupe-peers`,
273/// `resolve-peers-from-workspace-root`, `peers-suffix-max-length`)
274/// through the `Resolver`'s `with_*` setters.
275#[derive(Debug, Clone, Copy)]
276pub struct PeerContextOptions {
277    /// When true, run the cross-subtree peer-variant collapse pass
278    /// after every iteration of the fixed-point loop. Matches pnpm's
279    /// default.
280    pub dedupe_peer_dependents: bool,
281    /// When true, emit suffixes as `(version)` instead of
282    /// `(name@version)`. Affects both the package key, the reference
283    /// tails stored in `dependencies`, and the cycle-break form of
284    /// `contains_canonical_back_ref`.
285    pub dedupe_peers: bool,
286    /// When true, unresolved peers can be satisfied by a dep declared
287    /// at the root importer (`"."`) even if no ancestor scope carries
288    /// the peer. Runs between own-deps and graph-wide scan in the
289    /// peer-context visitor — see `visit_peer_context` in this
290    /// module for the owning implementation (intentionally crate-
291    /// private; the public API here is the option flag itself).
292    pub resolve_from_workspace_root: bool,
293    /// Byte cap on the peer-ID suffix after which the entire suffix
294    /// is hashed to `_<10-char-sha256-hex>`. pnpm's default is 1000.
295    pub peers_suffix_max_length: usize,
296}
297
298impl Default for PeerContextOptions {
299    fn default() -> Self {
300        Self {
301            dedupe_peer_dependents: true,
302            dedupe_peers: false,
303            resolve_from_workspace_root: true,
304            peers_suffix_max_length: 1000,
305        }
306    }
307}
308
309/// Compute peer-context suffixes over an already-resolved graph.
310///
311/// Takes a *canonical* graph — one `LockedPackage` per `(name,
312/// version)` with `peer_dependencies` populated — and produces a
313/// *contextualized* graph whose keys and transitive references carry
314/// `(peer@ver)` suffixes when packages resolve peers differently in
315/// different subtrees. Drives the sibling-symlink wiring in
316/// `aube-linker` for peers, so every fetch/materialize site sees a
317/// per-context identity for any package whose peers disambiguate.
318///
319/// Public so lockfile-driven installs can run the pass over graphs
320/// parsed from npm/yarn/bun lockfiles (which emit canonical form —
321/// no peer suffixes — and would otherwise leave peer-dependent
322/// packages without their peers as `.aube/<pkg>/node_modules/<peer>`
323/// siblings). Fresh resolves call it internally from
324/// `Resolver::resolve`.
325pub fn apply_peer_contexts(
326    canonical: LockfileGraph,
327    options: &PeerContextOptions,
328) -> LockfileGraph {
329    const MAX_ITERATIONS: usize = 16;
330    let mut current = canonical;
331    let mut converged = false;
332    // Hash both keys and dependency tails. A peer-context iteration can
333    // rewrite a dependency value to point at an existing key without
334    // adding a new key, so a key-only convergence test ships partially
335    // rewritten tails. Linker reads tails directly to locate sibling
336    // symlink targets, stale tails produce broken `node_modules`.
337    let graph_hash = |g: &LockfileGraph| -> u64 {
338        let total_deps: usize = g.packages.values().map(|p| p.dependencies.len()).sum();
339        let mut tokens: Vec<&str> = Vec::with_capacity(g.packages.len() * 3 + total_deps * 2);
340        for (k, pkg) in &g.packages {
341            tokens.push(k.as_str());
342            tokens.push("\x1f");
343            for (name, tail) in &pkg.dependencies {
344                tokens.push(name.as_str());
345                tokens.push(tail.as_str());
346            }
347            tokens.push("\x1e");
348        }
349        aube_util::hash::ordered_seq_hash(tokens.iter().copied())
350    };
351    // Carry the post-iteration hash forward as the next iteration's
352    // pre-hash. Saves one full graph walk per iteration (the loop runs
353    // up to 16 times; each `graph_hash` allocates a Vec<&str> sized
354    // to `pkgs * 3 + deps * 2` tokens — ~25k entries on a 1000-pkg
355    // graph). One hash per iter instead of two.
356    let mut before = graph_hash(&current);
357    for i in 0..MAX_ITERATIONS {
358        let after_once = apply_peer_contexts_once(current, options);
359        let next = if options.dedupe_peer_dependents {
360            dedupe_peer_variants(after_once)
361        } else {
362            after_once
363        };
364        let after = graph_hash(&next);
365        if before == after {
366            tracing::debug!("peer-context pass converged after {i} iteration(s)");
367            current = next;
368            converged = true;
369            break;
370        }
371        current = next;
372        before = after;
373    }
374    if !converged {
375        // Hit iteration cap. Means mutually recursive peers or
376        // genuine cycle. Lockfile now has partial nested suffixes.
377        // Linker downstream will wire symlinks against incomplete
378        // graph. Returning this silently ships broken node_modules.
379        // Old code used warn!, warn gets swallowed in CI. Bump to
380        // error! so ops see it. Proper fix is returning a Result
381        // from apply_peer_contexts but that cascades up through
382        // Resolver::resolve signature, do that separately.
383        tracing::error!(
384            code = aube_codes::errors::ERR_AUBE_PEER_CONTEXT_NOT_CONVERGED,
385            max_iterations = MAX_ITERATIONS,
386            "peer-context hit MAX_ITERATIONS={MAX_ITERATIONS} without convergence. \
387             mutually recursive peers likely. lockfile incomplete, linker output will be wrong"
388        );
389    }
390    // `dedupe-peers=true` rewrites the parenthesized peer suffix to
391    // drop the `name@` prefix. Done as a post-pass rather than inline
392    // so cycle detection during the fixed-point loop keeps the full
393    // `name@version` form (otherwise unrelated same-version packages
394    // would false-positive as back-references).
395    if options.dedupe_peers {
396        dedupe_peer_suffixes(current)
397    } else {
398        current
399    }
400}
401
402/// Cross-subtree peer-variant dedupe. When `dedupe-peer-dependents` is
403/// on, packages that landed at different contextualized dep_paths but
404/// resolved every declared peer to the *same* version (ignoring the
405/// nested peer suffix on each peer tail) collapse into a single
406/// canonical variant — chosen as the lexicographically smallest key in
407/// the equivalence class. References in every surviving
408/// `LockedPackage.dependencies` map and every `importers[*]` direct
409/// dep get rewritten through the old→canonical map, and the
410/// non-canonical entries are dropped from `packages`.
411///
412/// Packages whose `peer_dependencies` map is empty — i.e. the canonical
413/// base already has only one variant — are skipped.
414pub(crate) fn dedupe_peer_variants(graph: LockfileGraph) -> LockfileGraph {
415    let canonical_base = |key: &str| -> String { canonical_tail(key).to_string() };
416    // Only the peer-bearing part of the resolved peer tail is
417    // comparable across subtrees — the nested suffix could differ even
418    // for peer-equivalent variants on mid-iterations of the outer
419    // fixed-point loop.
420    let peer_base = |tail: &str| -> String { canonical_tail(tail).to_string() };
421
422    // Group dep_paths by their peer-free base name.
423    let mut groups: BTreeMap<String, Vec<String>> = BTreeMap::new();
424    for key in graph.packages.keys() {
425        groups
426            .entry(canonical_base(key))
427            .or_default()
428            .push(key.clone());
429    }
430
431    let mut rewrite: BTreeMap<String, String> = BTreeMap::new();
432    for (_base, mut keys) in groups {
433        if keys.len() < 2 {
434            continue;
435        }
436        // Deterministic order for canonical selection + stable hashing.
437        keys.sort();
438        // Union-find over equivalence classes. Two variants are
439        // equivalent when each declared peer name resolves to the same
440        // peer base in both (or is missing from both).
441        let mut parent: Vec<usize> = (0..keys.len()).collect();
442        fn find(parent: &mut [usize], i: usize) -> usize {
443            if parent[i] == i {
444                i
445            } else {
446                let r = find(parent, parent[i]);
447                parent[i] = r;
448                r
449            }
450        }
451        for i in 0..keys.len() {
452            for j in (i + 1)..keys.len() {
453                let pa = &graph.packages[&keys[i]];
454                let pb = &graph.packages[&keys[j]];
455                // Same canonical version is required — packages with
456                // different versions but the same name would share no
457                // canonical_base only if the name-without-version
458                // collided, which doesn't happen (version is in the
459                // base). Still, belt-and-suspenders.
460                if pa.version != pb.version {
461                    continue;
462                }
463                let peer_names: BTreeSet<&String> = pa
464                    .peer_dependencies
465                    .keys()
466                    .chain(pb.peer_dependencies.keys())
467                    .collect();
468                let equivalent = peer_names.iter().all(|name| {
469                    match (
470                        pa.dependencies.get(name.as_str()),
471                        pb.dependencies.get(name.as_str()),
472                    ) {
473                        (Some(va), Some(vb)) => peer_base(va) == peer_base(vb),
474                        (None, None) => true,
475                        _ => false,
476                    }
477                });
478                if equivalent {
479                    let ri = find(&mut parent, i);
480                    let rj = find(&mut parent, j);
481                    if ri != rj {
482                        parent[ri] = rj;
483                    }
484                }
485            }
486        }
487        // Build class → canonical (smallest key) mapping. Using
488        // index-based iteration here because `find` takes a mutable
489        // reference into `parent`, so holding an immutable borrow
490        // from `keys.iter()` at the same time would double-borrow.
491        #[allow(clippy::needless_range_loop)]
492        {
493            let mut class_rep: BTreeMap<usize, String> = BTreeMap::new();
494            for i in 0..keys.len() {
495                let root = find(&mut parent, i);
496                class_rep
497                    .entry(root)
498                    .and_modify(|cur| {
499                        if keys[i] < *cur {
500                            *cur = keys[i].clone();
501                        }
502                    })
503                    .or_insert_with(|| keys[i].clone());
504            }
505            for i in 0..keys.len() {
506                let root = find(&mut parent, i);
507                let canonical = class_rep[&root].clone();
508                if keys[i] != canonical {
509                    rewrite.insert(keys[i].clone(), canonical);
510                }
511            }
512        }
513    }
514
515    if rewrite.is_empty() {
516        return graph;
517    }
518
519    // Rewrite package dependency tails and keep only canonicals.
520    let LockfileGraph {
521        importers,
522        packages,
523        settings,
524        overrides,
525        ignored_optional_dependencies,
526        times,
527        skipped_optional_dependencies,
528        catalogs,
529        bun_config_version,
530        patched_dependencies,
531        trusted_dependencies,
532        extra_fields,
533        workspace_extra_fields,
534    } = graph;
535
536    let mut new_packages: BTreeMap<String, LockedPackage> = BTreeMap::new();
537    for (key, mut pkg) in packages {
538        if rewrite.contains_key(&key) {
539            continue;
540        }
541        for (dep_name, dep_tail) in pkg.dependencies.iter_mut() {
542            let dep_key = format!("{dep_name}@{dep_tail}");
543            if let Some(canonical) = rewrite.get(&dep_key) {
544                let new_tail = canonical
545                    .strip_prefix(&format!("{dep_name}@"))
546                    .map(|s| s.to_string())
547                    .unwrap_or_else(|| canonical.clone());
548                *dep_tail = new_tail;
549            }
550        }
551        new_packages.insert(key, pkg);
552    }
553
554    let mut new_importers: BTreeMap<String, Vec<DirectDep>> = BTreeMap::new();
555    for (importer_path, deps) in importers {
556        let mut new_deps = Vec::with_capacity(deps.len());
557        for mut dep in deps {
558            if let Some(canonical) = rewrite.get(&dep.dep_path) {
559                dep.dep_path = canonical.clone();
560            }
561            new_deps.push(dep);
562        }
563        new_importers.insert(importer_path, new_deps);
564    }
565
566    LockfileGraph {
567        importers: new_importers,
568        packages: new_packages,
569        settings,
570        overrides,
571        ignored_optional_dependencies,
572        times,
573        skipped_optional_dependencies,
574        catalogs,
575        bun_config_version,
576        patched_dependencies,
577        trusted_dependencies,
578        extra_fields,
579        workspace_extra_fields,
580    }
581}
582
583/// Single pass of the peer-context computation. See `apply_peer_contexts`
584/// for the wrapping fixed-point loop.
585///
586/// Algorithm per visited package P, reached at some point in a DFS from an
587/// importer with `ancestor_scope: name -> dep_path_tail`:
588///
589///  1. For each peer name declared by P, look it up in `ancestor_scope`
590///     (nearest-ancestor-wins, since the scope is rebuilt per recursion).
591///     If missing, fall back to P's own entry in `dependencies` — the BFS
592///     enqueue auto-installed it as a transitive, matching pnpm's
593///     `auto-install-peers=true` default.
594///  2. Sort the (peer_name, resolution) pairs and serialize as
595///     `(n1@v1)(n2@v2)…` for the suffix.
596///  3. Produce a contextualized dep_path `name@version{suffix}`. If that
597///     key is already in `out_packages` (or currently on the DFS stack via
598///     `visiting`), short-circuit — we've already emitted this variant.
599///  4. Build a new scope for P's children by merging the ancestor scope
600///     with P's own `dependencies` and the resolved peer map. Recurse.
601///  5. Emit the contextualized LockedPackage.
602///
603/// Cycles: protected by `visiting` — if a package is re-entered via a
604/// dependency cycle, we return the already-computed dep_path without
605/// recursing again. The peer context is fixed at first visit; any cycle
606/// traversal uses whatever context was live at that first visit.
607fn apply_peer_contexts_once(
608    canonical: LockfileGraph,
609    options: &PeerContextOptions,
610) -> LockfileGraph {
611    let mut out_packages: BTreeMap<String, LockedPackage> = BTreeMap::new();
612    let mut new_importers: BTreeMap<String, Vec<DirectDep>> = BTreeMap::new();
613
614    // Name-indexed view of the canonical graph, shared across
615    // every `visit_peer_context` call in this pass. Peer-resolution
616    // scan-by-name is the resolver's hottest inner loop. Without
617    // this, each peer runs `O(|graph|)` per package per fixed-point
618    // iter. Prebuilt index drops the scan to O(1) average.
619    let mut name_index: FxHashMap<&str, Vec<&LockedPackage>> = FxHashMap::default();
620    for pkg in canonical.packages.values() {
621        name_index.entry(pkg.name.as_str()).or_default().push(pkg);
622    }
623
624    // Root-importer scope used by `resolve-peers-from-workspace-root`.
625    // Computed once from the canonical input so it reflects the
626    // contextualized state of every root dep on fixed-point iterations
627    // 2+ — same logic as per-importer `importer_scope` below.
628    let root_scope: FxHashMap<String, String> = canonical
629        .importers
630        .get(".")
631        .map(|deps| scope_map_from_deps(deps))
632        .unwrap_or_default();
633
634    for (importer_path, direct_deps) in &canonical.importers {
635        // An importer's own direct deps are in scope for its children's
636        // peer resolution — this is how pnpm's "auto-install at the root"
637        // path gets peer links that point at root-level packages.
638        //
639        // Use the *full contextualized tail* off each DirectDep rather
640        // than the package's plain version. On Pass 1 of the fixed-point
641        // loop the tail is canonical and equal to `p.version`; on Pass 2+
642        // it's already contextualized, and passing the plain version
643        // would make descendants look up keys that don't exist in the
644        // (now-nested) graph.
645        let importer_scope = scope_map_from_deps(direct_deps);
646
647        let mut new_deps = Vec::with_capacity(direct_deps.len());
648        for dep in direct_deps {
649            // `visiting` is the DFS stack guard for this particular descent
650            // — reset per direct dep so we don't incorrectly flag a package
651            // as a cycle when it's reached again from a sibling subtree.
652            // The shared `out_packages` still dedupes across siblings since
653            // the second visit hits the `contains_key` short-circuit below.
654            //
655            // Invariant (see `visit_peer_context` for the detailed handling):
656            // a dep_path returned from the cycle-break branch may not yet
657            // be present in `out_packages` at the moment of return, because
658            // the package is still being assembled up the call stack. The
659            // parent that records the returned tail will complete its own
660            // insertion before the recursion unwinds, so by the time
661            // anything reads the graph, every referenced dep_path exists.
662            let mut visiting: FxHashSet<String> = FxHashSet::default();
663            let new_dep_path = visit_peer_context(
664                &dep.dep_path,
665                &canonical,
666                &name_index,
667                &importer_scope,
668                &root_scope,
669                &mut out_packages,
670                &mut visiting,
671                options,
672            )
673            .unwrap_or_else(|| dep.dep_path.clone());
674            new_deps.push(DirectDep {
675                name: dep.name.clone(),
676                dep_path: new_dep_path,
677                dep_type: dep.dep_type,
678                specifier: dep.specifier.clone(),
679            });
680        }
681        new_importers.insert(importer_path.clone(), new_deps);
682    }
683
684    // Any canonical package that was never reached by the DFS (orphaned
685    // from every importer) is dropped — that matches the filter_deps
686    // semantics and avoids emitting dead entries into the lockfile.
687
688    LockfileGraph {
689        importers: new_importers,
690        packages: out_packages,
691        // The post-pass is pure — settings + overrides carry through
692        // from the input graph untouched.
693        settings: canonical.settings,
694        overrides: canonical.overrides,
695        ignored_optional_dependencies: canonical.ignored_optional_dependencies,
696        times: canonical.times,
697        skipped_optional_dependencies: canonical.skipped_optional_dependencies,
698        catalogs: canonical.catalogs,
699        bun_config_version: canonical.bun_config_version,
700        patched_dependencies: canonical.patched_dependencies,
701        trusted_dependencies: canonical.trusted_dependencies,
702        extra_fields: canonical.extra_fields,
703        workspace_extra_fields: canonical.workspace_extra_fields,
704    }
705}
706
707/// DFS helper for `apply_peer_contexts`. Returns the peer-contextualized
708/// dep_path of the visited package, or `None` if the canonical package is
709/// missing (shouldn't happen in practice but we degrade gracefully).
710/// Does `value` contain a peer-suffix reference to `canonical` as a
711/// proper name@version boundary (i.e. preceded by `(` and followed by
712/// `(` / `)` / end-of-string)? Used by the peer-context pass to detect
713/// when a nested tail loops back to the current package so it can
714/// short-circuit the chain instead of growing the suffix forever.
715/// If `s` ends with `_<10 lowercase hex>` (the marker written by
716/// `hash_peer_suffix`), strip it and return the prefix. Otherwise
717/// return `s` unchanged.
718///
719/// Safe against false positives: `s` here is always a post-split
720/// `name@version` base, and semver forbids `_` inside a version, so
721/// an underscore 10 chars from the end of `name@version` can only be
722/// our marker.
723/// Everything before the first `(` — i.e. the canonical `name@version`
724/// part of a dep-path with the peer-context suffix stripped. Returns
725/// the original string when no `(` is present. Borrowed; callers that
726/// need owned bump with `.to_string()`.
727fn canonical_tail(s: &str) -> &str {
728    s.split('(').next().unwrap_or(s)
729}
730
731/// Build a `name → contextualized tail` map from a direct-dep slice.
732/// The tail is the dep_path with the `{name}@` prefix stripped, which
733/// on pass 1 is equal to `pkg.version` and on pass 2+ carries the
734/// nested peer-context suffix. Used both for the root scope and for
735/// each importer's own scope inside `apply_peer_contexts_once`.
736fn scope_map_from_deps(deps: &[DirectDep]) -> FxHashMap<String, String> {
737    let mut out = FxHashMap::with_capacity_and_hasher(deps.len(), Default::default());
738    for d in deps {
739        let prefix_len = d.name.len() + 1;
740        let tail = if d.dep_path.len() > prefix_len
741            && d.dep_path.as_bytes().get(d.name.len()) == Some(&b'@')
742            && d.dep_path.as_bytes().starts_with(d.name.as_bytes())
743        {
744            d.dep_path[prefix_len..].to_string()
745        } else {
746            d.dep_path.clone()
747        };
748        out.insert(d.name.clone(), tail);
749    }
750    out
751}
752
753fn strip_hashed_peer_suffix(s: &str) -> &str {
754    const MARKER_LEN: usize = 11; // `_` + 10 hex chars
755    if s.len() < MARKER_LEN {
756        return s;
757    }
758    let tail = &s[s.len() - MARKER_LEN..];
759    if !tail.starts_with('_') {
760        return s;
761    }
762    if tail[1..]
763        .chars()
764        .all(|c| c.is_ascii_digit() || ('a'..='f').contains(&c))
765    {
766        &s[..s.len() - MARKER_LEN]
767    } else {
768        s
769    }
770}
771
772/// Hash a peer-ID suffix with SHA-256 and return `_<10-char-hex>`.
773/// Used by the peer-context pass when the raw suffix length exceeds
774/// `peersSuffixMaxLength`. Matches pnpm's format so lockfile dep_path
775/// keys stay portable.
776pub(crate) fn hash_peer_suffix(suffix: &str) -> String {
777    use sha2::{Digest, Sha256};
778    let digest = Sha256::digest(suffix.as_bytes());
779    let mut out = String::with_capacity(11);
780    out.push('_');
781    for byte in digest.iter().take(5) {
782        use std::fmt::Write;
783        let _ = write!(out, "{byte:02x}");
784    }
785    out
786}
787
788pub(crate) fn contains_canonical_back_ref(value: &str, canonical: &str) -> bool {
789    let bytes = value.as_bytes();
790    let target = canonical.as_bytes();
791    if target.is_empty() || target.len() > bytes.len() {
792        return false;
793    }
794    let mut i = 0;
795    while i + target.len() <= bytes.len() {
796        if &bytes[i..i + target.len()] == target {
797            let before = if i == 0 { b'\0' } else { bytes[i - 1] };
798            let after = bytes.get(i + target.len()).copied().unwrap_or(b'\0');
799            let before_ok = before == b'(';
800            let after_ok = after == b'(' || after == b')' || after == b'\0';
801            if before_ok && after_ok {
802                return true;
803            }
804        }
805        i += 1;
806    }
807    false
808}
809
810/// Dedupe-peers post-pass: strip the `name@` prefix from every
811/// parenthesized peer segment in every dep_path key and reference,
812/// turning `react-dom@18.2.0(react@18.2.0)` into
813/// `react-dom@18.2.0(18.2.0)`. Nested segments get the same treatment
814/// so `a@1(b@2(c@3))` becomes `a@1(2(3))`.
815///
816/// Running this as a final post-pass (instead of inline during suffix
817/// assembly in `visit_peer_context`) keeps cycle detection correct:
818/// the detection path works against the full `name@version` form
819/// throughout the fixed-point loop, and only the serialized output
820/// gets the shorter form. A version-only inline approach would
821/// false-positive on unrelated packages that coincidentally share a
822/// version with the current package's canonical base.
823///
824/// Pure: no-op when `dedupe_peers` is off (caller gates the call);
825/// otherwise rewrites every package key, every `LockedPackage.dep_path`
826/// and `LockedPackage.dependencies` value, and every `importers[*]`
827/// DirectDep `dep_path` through the same `apply_dedupe_peers_to_tail`
828/// helper. Package bodies (integrity, metadata, etc.) are cloned
829/// verbatim.
830pub(crate) fn dedupe_peer_suffixes(graph: LockfileGraph) -> LockfileGraph {
831    // Pass 1: compute the intended deduped key for each package and
832    // tally how many distinct full-form keys map to it. Stripping
833    // `name@` from suffix segments is lossy — two variants whose peer
834    // *names* differ but whose peer *versions* coincide would collapse
835    // onto the same deduped key (e.g. `consumer@1.0.0(foo@1.0.0)` and
836    // `consumer@1.0.0(bar@1.0.0)` both → `consumer@1.0.0(1.0.0)`).
837    // `dedupe_peer_variants` already merged the peer-equivalent
838    // duplicates, so any remaining collision here represents genuinely
839    // distinct variants — losing one would silently drop its
840    // dependency wiring. We detect those collisions and keep both
841    // sides in full form.
842    let mut target_counts: BTreeMap<String, usize> = BTreeMap::new();
843    let mut intended: BTreeMap<String, String> = BTreeMap::new();
844    for key in graph.packages.keys() {
845        let new_key = apply_dedupe_peers_to_key(key);
846        *target_counts.entry(new_key.clone()).or_insert(0) += 1;
847        intended.insert(key.clone(), new_key);
848    }
849    let rewrite: BTreeMap<String, String> = intended
850        .into_iter()
851        .map(|(old, new)| {
852            if target_counts.get(&new).copied().unwrap_or(0) > 1 {
853                tracing::warn!(
854                    code = aube_codes::warnings::WARN_AUBE_PEER_DEDUPE_COLLISION,
855                    "dedupe-peers: collision on {new} — keeping {old} in full form to avoid \
856                     dropping a distinct peer-variant"
857                );
858                (old.clone(), old)
859            } else {
860                (old, new)
861            }
862        })
863        .collect();
864
865    // Rewrite a `(child_name, tail)` reference by reconstructing the
866    // target's full-form key, looking up its effective rewrite, and
867    // stripping `child_name@` off the result to recover the tail.
868    // Tails always follow their target package's rewrite decision,
869    // so references stay consistent when a collision forces a target
870    // back to full form.
871    let rewrite_tail = |child_name: &str, tail: &str| -> String {
872        let old_key = format!("{child_name}@{tail}");
873        match rewrite.get(&old_key) {
874            Some(new_key) => new_key
875                .strip_prefix(&format!("{child_name}@"))
876                .map(|s| s.to_string())
877                .unwrap_or_else(|| tail.to_string()),
878            None => apply_dedupe_peers_to_tail(tail),
879        }
880    };
881
882    let mut new_packages: BTreeMap<String, LockedPackage> = BTreeMap::new();
883    for (old_key, pkg) in graph.packages {
884        let new_key = rewrite
885            .get(&old_key)
886            .cloned()
887            .unwrap_or_else(|| old_key.clone());
888        let new_dependencies: BTreeMap<String, String> = pkg
889            .dependencies
890            .into_iter()
891            .map(|(n, v)| {
892                let new_v = rewrite_tail(&n, &v);
893                (n, new_v)
894            })
895            .collect();
896        let new_optional_dependencies: BTreeMap<String, String> = pkg
897            .optional_dependencies
898            .into_iter()
899            .map(|(n, v)| {
900                let new_v = rewrite_tail(&n, &v);
901                (n, new_v)
902            })
903            .collect();
904        new_packages.insert(
905            new_key.clone(),
906            LockedPackage {
907                name: pkg.name,
908                version: pkg.version,
909                integrity: pkg.integrity,
910                dependencies: new_dependencies,
911                optional_dependencies: new_optional_dependencies,
912                peer_dependencies: pkg.peer_dependencies,
913                peer_dependencies_meta: pkg.peer_dependencies_meta,
914                dep_path: new_key,
915                local_source: pkg.local_source,
916                os: pkg.os,
917                cpu: pkg.cpu,
918                libc: pkg.libc,
919                bundled_dependencies: pkg.bundled_dependencies,
920                optional: pkg.optional,
921                transitive_peer_dependencies: pkg.transitive_peer_dependencies,
922                tarball_url: pkg.tarball_url,
923                alias_of: pkg.alias_of,
924                yarn_checksum: pkg.yarn_checksum,
925                engines: pkg.engines,
926                bin: pkg.bin,
927                declared_dependencies: pkg.declared_dependencies,
928                license: pkg.license,
929                funding_url: pkg.funding_url,
930                extra_meta: pkg.extra_meta,
931            },
932        );
933    }
934
935    let new_importers: BTreeMap<String, Vec<DirectDep>> = graph
936        .importers
937        .into_iter()
938        .map(|(path, deps)| {
939            let rewritten = deps
940                .into_iter()
941                .map(|d| {
942                    let new_dep_path = rewrite
943                        .get(&d.dep_path)
944                        .cloned()
945                        .unwrap_or_else(|| apply_dedupe_peers_to_key(&d.dep_path));
946                    DirectDep {
947                        name: d.name,
948                        dep_path: new_dep_path,
949                        dep_type: d.dep_type,
950                        specifier: d.specifier,
951                    }
952                })
953                .collect();
954            (path, rewritten)
955        })
956        .collect();
957
958    LockfileGraph {
959        importers: new_importers,
960        packages: new_packages,
961        settings: graph.settings,
962        overrides: graph.overrides,
963        ignored_optional_dependencies: graph.ignored_optional_dependencies,
964        times: graph.times,
965        skipped_optional_dependencies: graph.skipped_optional_dependencies,
966        catalogs: graph.catalogs,
967        bun_config_version: graph.bun_config_version,
968        patched_dependencies: graph.patched_dependencies,
969        trusted_dependencies: graph.trusted_dependencies,
970        extra_fields: graph.extra_fields,
971        workspace_extra_fields: graph.workspace_extra_fields,
972    }
973}
974
975/// Strip `name@` from inside every parenthesized segment of a full
976/// dep_path key (e.g. `react-dom@18.2.0(react@18.2.0)` →
977/// `react-dom@18.2.0(18.2.0)`). The first `name@version` outside any
978/// parens is preserved verbatim — that's the canonical head of the
979/// dep_path and `dedupe-peers` only affects the peer suffix.
980pub(crate) fn apply_dedupe_peers_to_key(key: &str) -> String {
981    let mut parts = key.split('(');
982    let Some(first) = parts.next() else {
983        return key.to_string();
984    };
985    let mut out = String::with_capacity(key.len());
986    out.push_str(first);
987    for part in parts {
988        out.push('(');
989        // In a well-formed key, `part` looks like `name@version)` /
990        // `name@version` / `version)` / ... We strip everything up to
991        // and including the LAST `@` (scoped packages like
992        // `@types/react@18.2.0` contain two `@`s; the separator is the
993        // rightmost one). We only strip if that `@` comes before the
994        // first `)` or `(` (i.e. the segment actually starts with
995        // `name@`, not the outer parens closing with no name inside).
996        if let Some(at_idx) = part.rfind('@') {
997            let close_idx = part.find([')', '(']).unwrap_or(usize::MAX);
998            if at_idx < close_idx {
999                out.push_str(&part[at_idx + 1..]);
1000                continue;
1001            }
1002        }
1003        out.push_str(part);
1004    }
1005    out
1006}
1007
1008/// Same as [`apply_dedupe_peers_to_key`] but for dep-tail values
1009/// stored in `LockedPackage.dependencies` (e.g. `18.2.0(react@18.2.0)`
1010/// → `18.2.0(18.2.0)`). Tails differ from keys only by lacking the
1011/// leading `name@` prefix — both use the same parens-based suffix
1012/// shape, so the algorithm is identical.
1013fn apply_dedupe_peers_to_tail(tail: &str) -> String {
1014    apply_dedupe_peers_to_key(tail)
1015}
1016
1017#[allow(clippy::too_many_arguments)]
1018fn visit_peer_context<'g>(
1019    input_dep_path: &str,
1020    graph: &'g LockfileGraph,
1021    name_index: &FxHashMap<&'g str, Vec<&'g LockedPackage>>,
1022    ancestor_scope: &FxHashMap<String, String>,
1023    root_scope: &FxHashMap<String, String>,
1024    out_packages: &mut BTreeMap<String, LockedPackage>,
1025    visiting: &mut FxHashSet<String>,
1026    options: &PeerContextOptions,
1027) -> Option<String> {
1028    let pkg = graph.packages.get(input_dep_path)?;
1029
1030    // The input key may already carry a peer suffix (fixed-point loop
1031    // Pass 2+). Drop it before we build a new one — otherwise we'd
1032    // append the new suffix on top of the old and grow unboundedly
1033    // across iterations (classic mutual-peer-cycle blow-up).
1034    //
1035    // Two suffix forms can be present from a prior pass:
1036    //   1. `(name@version)(…)` — the normal nested peer suffix. Stripped
1037    //      by splitting on the first `(`.
1038    //   2. `_<10-char-sha256-hex>` — the hashed form produced when the
1039    //      normal suffix exceeded `peersSuffixMaxLength`. Must also be
1040    //      stripped; otherwise each pass re-hashes the already-hashed
1041    //      key and appends another marker (exposed by the
1042    //      `peer_suffix_is_hashed_when_exceeding_cap` unit test).
1043    let canonical_base = canonical_tail(input_dep_path);
1044    let canonical_base = strip_hashed_peer_suffix(canonical_base).to_string();
1045
1046    // Compute peer context: walk declared peers, resolve from ancestors
1047    // (nearest wins — the scope is rebuilt as we recurse) or from the
1048    // package's own dependency map as the auto-install fallback. Both
1049    // sides may produce nested tails on the second and later iterations
1050    // of the fixed-point loop.
1051    // Resolution source priority for each declared peer:
1052    //   1. Ancestor scope — if the ancestor's version actually
1053    //      satisfies the declared peer range. Different subtrees
1054    //      naturally see different ancestors (lib-a in subtree-A
1055    //      and lib-b in subtree-B keep their own peer pins), so
1056    //      preferring the closest ancestor here doesn't conflate
1057    //      cross-subtree variants.
1058    //   2. The current package's own `pkg.dependencies` entry — the
1059    //      BFS peer-walk enqueued this peer with the declared range,
1060    //      so whatever got picked there is guaranteed to satisfy.
1061    //      Captures the case where a single subtree holds two
1062    //      consumers with conflicting peer ranges (lib-a@^17 next to
1063    //      a parent that pins react@18): the BFS auto-installs the
1064    //      satisfying version into lib-a's own deps, which beats the
1065    //      ancestor's incompatible version.
1066    //   3. Ancestor scope — even when the version doesn't satisfy
1067    //      the declared range. This mirrors what Node's module
1068    //      resolution would surface (`require('peer')` from the
1069    //      package would walk up node_modules and find the parent's
1070    //      version). pnpm and bun do the same and emit an unmet-peer
1071    //      warning rather than picking a more-distant matching
1072    //      version. `detect_unmet_peers` flags the mismatch after
1073    //      the pass.
1074    //   4. The current package's own `pkg.dependencies` entry,
1075    //      ignoring range satisfaction — symmetric to (3) for the
1076    //      BFS-installed case.
1077    //   5. Workspace root scope (compatible) — `resolve-peers-from-
1078    //      workspace-root` fallback for monorepos that pin shared
1079    //      peers at the root.
1080    //   6. A graph-wide scan: any package whose name matches and
1081    //      whose version satisfies the declared range. Last resort
1082    //      for nested-context callers when nothing closer has it.
1083    //   7. Workspace root scope, ignoring range satisfaction.
1084    //
1085    // If nothing in the graph holds a version of this peer at all,
1086    // it's left out of the context entirely — `detect_unmet_peers`
1087    // will surface it as a warning after the pass.
1088    let mut peer_context: Vec<(String, String)> = Vec::new();
1089    for (peer_name, declared_range) in &pkg.peer_dependencies {
1090        let satisfies_declared = |v: &str| -> bool {
1091            // The tail may carry a nested peer suffix on fixed-point
1092            // iterations 2+; strip it before checking the semver.
1093            let canonical = canonical_tail(v);
1094            version_satisfies(canonical, declared_range)
1095        };
1096
1097        let from_ancestor = ancestor_scope
1098            .get(peer_name)
1099            .filter(|v| satisfies_declared(v))
1100            .cloned();
1101        let from_ancestor_incompatible = ancestor_scope.get(peer_name).cloned();
1102
1103        let from_pkg_deps = pkg
1104            .dependencies
1105            .get(peer_name)
1106            .filter(|v| satisfies_declared(v))
1107            .cloned();
1108        let from_pkg_deps_incompatible = pkg.dependencies.get(peer_name).cloned();
1109
1110        // `resolve-peers-from-workspace-root`: fall back to the root
1111        // importer's direct deps before the graph-wide scan. Common in
1112        // monorepos where the workspace root pins shared peers (e.g.
1113        // `react`) that leaf packages peer on without declaring them
1114        // in their own subtree. Skipped when the setting is off —
1115        // matches pnpm's `resolve-peers-from-workspace-root=false`.
1116        let from_root = if options.resolve_from_workspace_root {
1117            root_scope
1118                .get(peer_name)
1119                .filter(|v| satisfies_declared(v))
1120                .cloned()
1121        } else {
1122            None
1123        };
1124        let from_root_incompatible = if options.resolve_from_workspace_root {
1125            root_scope.get(peer_name).cloned()
1126        } else {
1127            None
1128        };
1129
1130        // Return the full dep_path TAIL (the part after `name@`), not
1131        // just `p.version`. On fixed-point iteration 2+, the input
1132        // graph's keys are contextualized — e.g. `react-dom` lives at
1133        // `react-dom@18.2.0(react@18.2.0)`. Downstream code
1134        // reconstructs the child lookup key with
1135        // `format!("{child_name}@{tail}")` and needs the tail to
1136        // match whatever the graph has keyed it under, otherwise the
1137        // lookup returns None and the peer gets silently dropped
1138        // from `new_dependencies`. The semver check is against the
1139        // package's canonical `version` field, not the tail, because
1140        // the tail may carry a peer suffix that isn't valid semver.
1141        let from_graph_scan = || {
1142            name_index
1143                .get(peer_name.as_str())
1144                .into_iter()
1145                .flat_map(|bucket| bucket.iter().copied())
1146                .filter(|p| version_satisfies(&p.version, declared_range))
1147                .filter_map(|p| {
1148                    let tail = p
1149                        .dep_path
1150                        .strip_prefix(&format!("{}@", p.name))
1151                        .map(|s| s.to_string())
1152                        .unwrap_or_else(|| p.version.clone());
1153                    node_semver::Version::parse(&p.version)
1154                        .ok()
1155                        .map(|ver| (ver, tail))
1156                })
1157                .max_by(|a, b| a.0.cmp(&b.0))
1158                .map(|(_, tail)| tail)
1159        };
1160
1161        if let Some(version) = from_ancestor
1162            .or(from_pkg_deps)
1163            .or(from_ancestor_incompatible)
1164            .or(from_pkg_deps_incompatible)
1165            .or(from_root)
1166            .or_else(from_graph_scan)
1167            .or(from_root_incompatible)
1168        {
1169            peer_context.push((peer_name.clone(), version));
1170        }
1171    }
1172    peer_context.sort_by(|a, b| a.0.cmp(&b.0));
1173
1174    // For the SUFFIX we build a cycle-broken copy: any peer value that
1175    // nests a reference back to the current package's canonical base
1176    // gets stripped to its plain version. Without this, mutual peer
1177    // cycles (a peers on b, b peers on a) grow the suffix one level
1178    // per iteration of the fixed-point loop and never converge.
1179    //
1180    // The non-cycle paths are untouched, so a regular nested chain
1181    // like `(react-dom@18.2.0(react@18.2.0))` still serializes fully.
1182    // We deliberately keep the full nested tails in `peer_context` for
1183    // downstream scope propagation and child lookups — suffix cycle-
1184    // breaking is cosmetic and should not change what packages exist
1185    // or which snapshot entries reference each other.
1186    //
1187    // Cycle detection is always done against the full `name@version`
1188    // canonical base — even when `dedupe-peers=true` is on, because
1189    // the version-only form is ambiguous (two unrelated packages at
1190    // the same version would false-positive). `dedupe-peers` is
1191    // applied as a post-pass over the final graph in
1192    // `dedupe_peer_suffixes` after cycle detection is done.
1193    let suffix: String = peer_context
1194        .iter()
1195        .map(|(n, v)| {
1196            let cycles_back = contains_canonical_back_ref(v, &canonical_base);
1197            let display_v = if cycles_back {
1198                canonical_tail(v).to_string()
1199            } else {
1200                v.clone()
1201            };
1202            format!("({n}@{display_v})")
1203        })
1204        .collect();
1205    // pnpm's `peersSuffixMaxLength`: when the built suffix exceeds the
1206    // cap, replace the entire suffix with `_<10-char-sha256-hex>` so the
1207    // lockfile key stays bounded. Matches pnpm's lockfile format, so
1208    // lockfiles shared between aube and pnpm stay comparable.
1209    let effective_suffix = if suffix.len() > options.peers_suffix_max_length {
1210        hash_peer_suffix(&suffix)
1211    } else {
1212        suffix
1213    };
1214    let contextualized = format!("{canonical_base}{effective_suffix}");
1215
1216    if out_packages.contains_key(&contextualized) || visiting.contains(&contextualized) {
1217        return Some(contextualized);
1218    }
1219    visiting.insert(contextualized.clone());
1220
1221    // Build the scope for P's children. This is ancestor_scope, overlaid
1222    // with P's own dependencies and its resolved peer map. Children see
1223    // their grandparents too — this mirrors pnpm's all-the-way-up peer
1224    // walk.
1225    //
1226    // We deliberately do NOT strip any existing peer-context suffix
1227    // off the tails we put into the scope. On the first pass the
1228    // values are plain (BFS output has no suffixes), so preserving
1229    // them is a no-op; on subsequent passes (see the fixed-point loop
1230    // in `apply_peer_contexts`) the input graph already carries
1231    // contextualized tails, and keeping them in scope is exactly how
1232    // nested peer suffixes propagate down to consumers — a package
1233    // that peers on `react-dom` and reaches it through a parent whose
1234    // `react-dom` entry is already `18.2.0(react@18.2.0)` will see
1235    // that nested tail in its own scope, and its own suffix will
1236    // serialize as `(react-dom@18.2.0(react@18.2.0))`. That's the
1237    // nested form pnpm writes.
1238    let mut child_scope = ancestor_scope.clone();
1239    for (name, version) in &pkg.dependencies {
1240        child_scope.insert(name.clone(), version.clone());
1241    }
1242    for (name, version) in &peer_context {
1243        child_scope.insert(name.clone(), version.clone());
1244    }
1245
1246    // Recurse into each child, rewriting its dependency map entry to
1247    // point at the contextualized dep_path's tail. A child whose visit
1248    // fails (orphaned / missing) keeps its own tail.
1249    //
1250    // For declared peer names, the peer context (filled from the
1251    // ancestor scope) is authoritative — we override whatever the BFS
1252    // peer walk auto-installed. Otherwise the snapshot suffix and the
1253    // actual wired `dependencies[peer]` could disagree, which made the
1254    // sibling symlink target inconsistent with the peer-context claim.
1255    // When the ancestor's version doesn't satisfy the declared range,
1256    // `detect_unmet_peers` will flag it as a warning after the pass.
1257    let peer_context_versions: FxHashMap<String, String> = peer_context.iter().cloned().collect();
1258
1259    let mut new_dependencies: BTreeMap<String, String> = BTreeMap::new();
1260    let mut visited_dep_names: FxHashSet<String> = FxHashSet::default();
1261
1262    for (child_name, child_version_tail) in &pkg.dependencies {
1263        // If this child is a declared peer, its tail comes from the
1264        // peer context (which may be nested). Otherwise we use the
1265        // tail we already have — also possibly nested on a 2nd pass.
1266        let lookup_tail = match peer_context_versions.get(child_name) {
1267            Some(v) => v.clone(),
1268            None => child_version_tail.clone(),
1269        };
1270        let child_canonical_dep_path = format!("{child_name}@{lookup_tail}");
1271        let child_new = visit_peer_context(
1272            &child_canonical_dep_path,
1273            graph,
1274            name_index,
1275            &child_scope,
1276            root_scope,
1277            out_packages,
1278            visiting,
1279            options,
1280        );
1281        let new_tail = match child_new {
1282            Some(new_dep_path) => new_dep_path
1283                .strip_prefix(&format!("{child_name}@"))
1284                .map(|s| s.to_string())
1285                .unwrap_or_else(|| lookup_tail.clone()),
1286            None => lookup_tail.clone(),
1287        };
1288        new_dependencies.insert(child_name.clone(), new_tail);
1289        visited_dep_names.insert(child_name.clone());
1290    }
1291
1292    // Peers that were satisfied purely from the ancestor scope may not
1293    // have been in `pkg.dependencies` at all (no auto-install needed).
1294    // Wire them as deps now so the linker creates the sibling symlink
1295    // and the lockfile snapshot records them.
1296    for (peer_name, peer_version) in &peer_context {
1297        if visited_dep_names.contains(peer_name) {
1298            continue;
1299        }
1300        let child_canonical_dep_path = format!("{peer_name}@{peer_version}");
1301        let child_new = visit_peer_context(
1302            &child_canonical_dep_path,
1303            graph,
1304            name_index,
1305            &child_scope,
1306            root_scope,
1307            out_packages,
1308            visiting,
1309            options,
1310        );
1311        if let Some(new_dep_path) = child_new {
1312            let new_tail = new_dep_path
1313                .strip_prefix(&format!("{peer_name}@"))
1314                .map(|s| s.to_string())
1315                .unwrap_or_else(|| peer_version.clone());
1316            new_dependencies.insert(peer_name.clone(), new_tail);
1317        }
1318    }
1319
1320    visiting.remove(&contextualized);
1321    let new_optional_dependencies: BTreeMap<String, String> = pkg
1322        .optional_dependencies
1323        .keys()
1324        .filter_map(|name| {
1325            new_dependencies
1326                .get(name)
1327                .map(|tail| (name.clone(), tail.clone()))
1328        })
1329        .collect();
1330
1331    out_packages.insert(
1332        contextualized.clone(),
1333        LockedPackage {
1334            name: pkg.name.clone(),
1335            version: pkg.version.clone(),
1336            integrity: pkg.integrity.clone(),
1337            dependencies: new_dependencies,
1338            optional_dependencies: new_optional_dependencies,
1339            peer_dependencies: pkg.peer_dependencies.clone(),
1340            peer_dependencies_meta: pkg.peer_dependencies_meta.clone(),
1341            dep_path: contextualized.clone(),
1342            local_source: pkg.local_source.clone(),
1343            os: pkg.os.clone(),
1344            cpu: pkg.cpu.clone(),
1345            libc: pkg.libc.clone(),
1346            bundled_dependencies: pkg.bundled_dependencies.clone(),
1347            optional: pkg.optional,
1348            transitive_peer_dependencies: pkg.transitive_peer_dependencies.clone(),
1349            tarball_url: pkg.tarball_url.clone(),
1350            alias_of: pkg.alias_of.clone(),
1351            yarn_checksum: pkg.yarn_checksum.clone(),
1352            engines: pkg.engines.clone(),
1353            bin: pkg.bin.clone(),
1354            declared_dependencies: pkg.declared_dependencies.clone(),
1355            license: pkg.license.clone(),
1356            funding_url: pkg.funding_url.clone(),
1357            extra_meta: pkg.extra_meta.clone(),
1358        },
1359    );
1360    Some(contextualized)
1361}