mirror of
https://github.com/abhigyanpatwari/GitNexus.git
synced 2026-09-29 01:41:42 +00:00
* feat(ingestion): TypeScript registry-primary scope resolution (Ring 3) - Add TypeScript ScopeResolver stack (query/captures/interpret, import decomposition, hooks, arity, merge, receiver binding) and register in SCOPE_RESOLVERS. - Harden shared compound receiver and receiver-bound CALLS pass for map for-of tuple bindings, dotted typeRef shapes, and callable-alias fallbacks. - Flip TypeScript into MIGRATED_LANGUAGES; refresh AGENTS.md and type-resolution-system.md. - Shared finalize-algorithm updates for cross-file scope parity. - Tests: TS scope-resolution unit suite; legacy call-processor suite forces REGISTRY_PRIMARY_TYPESCRIPT=0; registry-primary flag test opts out TS in override scenario. Made-with: Cursor * fix(ingestion): SCC-ordered cross-file return-type propagation + multi-hop re-export resolution Fix CI failures on PR #1050 (TypeScript registry-primary migration) by making `propagateImportedReturnTypes` deterministic via reverse- topological SCC ordering and updating the multi-hop re-export contract to match `followReexportChain` behavior. Why: the legacy pass mirrored an intermediate ref instead of the terminal type when an importer was processed before its source module had its own typeBindings chain-followed (4-file alias chain regression in `ts-simple` fixture: `models.User -> service.user -> app.user` collapsed to `getUser` instead of `User`). Reverse-topological walk of `indexes.sccs` (leaves first) lets every importer see the source's already-followed terminal type in a single pass. Changes: - `imported-return-types.ts`: rewrite to walk SCCs leaves-first, chain- follow the source module's typeBindings BEFORE mirroring, and chain- follow the importer's typeBindings AFTER mirroring. Cyclic SCCs reach a partial fixpoint (no convergence guarantee, ts-circular only asserts no-throw). - `finalize-algorithm.ts`: docstring update on `FinalizeFile.localDefs` to reflect that `followReexportChain` resolves multi-hop re-exports through barrels even when intermediates do not surface the name - surfacing is now a static optimization, not a correctness requirement. - `contract/scope-resolver.ts` Invariant I3: explicitly document the SCC ordering requirement. - `pipeline/run.ts`: split PROF timer into `finalize` and `propagate` so the pass's cost is observable independently. - `ARCHITECTURE.md` Performance notes: describe SCC-ordered propagation. - `imported-return-types.ts`: expand chain-depth comment (2x effective depth from pre/post follow), add multi-ref break rationale, add `ts-simple` motivating-fixture pointer. Tests: - `finalize-algorithm.test.ts`: add 4 cases (3-hop chain, cyclic re-export visited-set guard, wildcard re-export fall-through, multi-source first-match-wins); fix misleading shared nodeId in the thick variant; rename and update the multi-hop test for the new contract (transitiveVia assertion on the thin variant). - `imported-return-types.test.ts` (NEW): unit tests for the SCC pass pinning topological collapse, local-annotation guard, missing-source skip, and cyclic-SCC no-throw. - `cross-file-binding.test.ts` + `ts-deep-alias-chain` fixture (NEW): 5-file integration regression guard for SCC-ordered propagation through 4 module boundaries. Validation: 865 scope-resolution + cross-file tests pass on Windows; typecheck clean across both packages; only pre-existing Swift overload failures remain (verified on PR base commit, environmental). Made-with: Cursor * fix(ingestion): address PR #1050 review findings — side-effect imports, resolve-cache perf, adapter signature Three independent fixes surfaced by the production-readiness review of the TypeScript registry-primary scope-resolution migration (RFC #909 Ring 3). All three pass under both REGISTRY_PRIMARY_TYPESCRIPT=0 and =1. 1. Side-effect imports were silently dropped (correctness regression). The legacy DAG emitted IMPORTS edges for `import './polyfill'` because its tree-sitter query matches `(import_statement source: (string))` regardless of clause. The new registry-primary path returned `[]` from `splitImportStatement()` for clause-less imports, so no ParsedImport / ImportEdge was ever produced — silent file-level edge loss. Add a generic 'side-effect' variant to `ParsedImport` and `ImportEdge['kind']` in `gitnexus-shared`; finalize resolves the target file and pre-finalizes the edge (no `targetDefId`, no `BindingRef`) so the SCC fixpoint loop skips it. The TypeScript provider now emits + interprets the new kind end-to-end. The variant is intentionally generic so other languages (Rust `use foo as _`, Python module-init) can adopt it. 2. Per-import re-derivation in `resolveImportTarget` (perf regression). The TS adapter built `new Set(allFilePaths)` on every call and let `resolveTsImportTarget` re-derive `allFileList` / `normalizedFileList` and discard the `resolveCache`. For a workspace with N files and M imports that's O(N × M) work per pass. Wrap the adapter in a closure that memoizes all five derived values keyed on the orchestrator's `ReadonlySet` identity; reset only when the set reference changes (start of new pass). New cost: O(N + M). 3. Misleading fake `ParsedImport` in the adapter (architecture). The adapter constructed `{ kind: 'named', localName: '_', importedName: '_', targetRaw }` to call `resolveTsImportTarget`, even though only `targetRaw` and the structural-typed context are read. Extract `resolveTsTarget(targetRaw, ctx)` so the adapter has an honest signature; `resolveTsImportTarget` still works for other callers. Also extract `narrowTsContext` for the type narrowing. Tests: - New 4-file fixture `typescript-side-effect-imports` with two side-effect imports + one named import. - New "TypeScript side-effect imports" describe in `test/integration/resolvers/typescript.test.ts` (parity-gated by `ci-scope-parity.yml` — runs under both flag states). - Updated 2 unit tests to expect 1 side-effect ParsedImport and 4 `@import.statement` matches (was 0 / 3). - 785 / 785 TS scope-resolution tests pass under both REGISTRY_PRIMARY_TYPESCRIPT=0 and =1. Made-with: Cursor * fix(scope): address Codex adversarial review findings on PR #1050 Four findings from the Codex adversarial review broke registry-primary TypeScript resolution for common patterns. All four now have unit and integration regression coverage that pass under both `REGISTRY_PRIMARY_TYPESCRIPT=0` (legacy DAG) and the default registry-primary path. [high] tsconfig path aliases dropped: Threaded `tsconfigPaths` through ScopeResolver via a new opaque `resolutionConfig` parameter and a `loadResolutionConfig(repoPath)` hook. The orchestrator (`scopeResolutionPhase` + `runScopeResolution`) loads it once per workspace pass and forwards into every `resolveImportTarget` call. TypeScript resolver now resolves `@/services/user` style imports through the standard resolver's alias branch. [high] TSX parsed with the wrong grammar: `emitTsScopeCaptures` now picks the parser/query by `filePath` (`.tsx` -> TSX grammar) and validates cached trees against the expected grammar via the new exported `tsCachedTreeMatchesGrammar` helper. Stale TS-grammar trees for `.tsx` files no longer leak through the scope query. [medium] Literal dynamic imports never linked: Added `kind: 'dynamic-resolved'` to `ParsedImport` and `ImportEdge`. The decomposer emits a synthetic `@import.literal` capture for string-literal dynamic imports; the interpreter maps that to `dynamic-resolved`; finalize pre-finalizes it as a file-level terminal (same shape as `side-effect`). `import('./feature')` now produces a real IMPORTS edge under the registry-primary path. Legacy DAG keeps its existing behavior — the new integration assertion is gated behind the flag. [medium] Namespace re-exports invisible from barrels: The decomposer now emits TWO captures for `export * as ns from './m'` — the existing `reexport-namespace` import draft AND a synthetic `@declaration.namespace` capture (via `buildNamespaceDeclarationMatch`). The latter creates a Namespace `SymbolDefinition` in the barrel's `localDefs`, so downstream `import { ns } from './barrel'` resolves through `findExportByName`. Regression fixtures under `gitnexus/test/fixtures/lang-resolution/`: - typescript-tsconfig-aliases (`@/` alias) - typescript-tsx-jsx (Button.tsx + App.tsx with JSX) - typescript-dynamic-import (`await import('./feature')`) - typescript-reexport-namespace (`export * as Models from './base'`) Validation: - gitnexus-shared builds clean - gitnexus typecheck clean - 385/385 TS scope-resolution tests pass under both `REGISTRY_PRIMARY_TYPESCRIPT=0` and default Made-with: Cursor * perf(scope): O(1) defById lookup + bounded re-export depth (PR #1050 round 3) Addresses the round-3 PR #1050 reviews (Claude adversarial + xkonjin): both flagged the existing O(N²) `findDefById` linear scan in `materializeBindings` and the unbounded recursion in `followReexportChain` as production-readiness blockers for TypeScript monorepos. Both fixes land alongside their regression tests under both `REGISTRY_PRIMARY_TYPESCRIPT=0` and the default registry-primary path. [high] materializeBindings O(N_files × N_defs × N_edges) → O(N_defs + N_edges): Build a `nodeId → SymbolDefinition` index map once at the top of `materializeBindings` (one O(N_defs) pass), then replace the per-edge `findDefById(files, edge.targetDefId)` linear scan with an O(1) `defById.get(edge.targetDefId)` lookup. Also drop the now-unused `findDefById` helper. At realistic TypeScript monorepo scale (~5k files × ~50 defs/file × ~100k linked import edges) this is the difference between ~25 s and a few ms inside finalize. Regression test in `finalize-algorithm.test.ts` builds 200 leaf files + 1 consumer importing one symbol from each, asserts every binding materializes correctly. [medium] followReexportChain unbounded recursion: The existing `visited` set caps depth at `O(N_files)` but allows recursion proportional to barrel-chain depth, mismatching the explicit "Iterative DFS to avoid stack overflow" policy in `tarjanSccs`. Added a `MAX_REEXPORT_DEPTH = 100` constant and a `depth` parameter to `followReexportChain` (defaults to 0); each recursive call passes `depth + 1` and the function returns `null` when the cap is exceeded. 100 is comfortably above any realistic hand-authored barrel chain (typical depth 1-5; auto-generated barrels rarely exceed 20) while staying well below JS engine call stack limits. Regression test wires a 200-link reexport chain and verifies the crawl terminates cleanly with `linkStatus: 'unresolved'` (no terminal def reachable within the budget). [low] synthesizeInstanceofNarrowings bare-identifier-only limitation: xkonjin's review #4 noted that the LHS narrowing only handles bare identifiers (`if (x instanceof Foo)`), not member expressions (`if (user.address instanceof Address)`). Added a JSDoc note explaining the constraint and pointing readers at field-type resolution as the workaround for member-chain receivers. Validation: - gitnexus-shared builds clean - gitnexus typecheck clean - 413/413 tests pass under both flag states for finalize-algorithm + TS unit + TS integration suites - 972/972 tests pass across full scope-resolution + Python + C# integration smoke (no cross-language regression) Made-with: Cursor * refactor(finalize): replace recursive followReexportChain with SCC-condensed iterative closure The legacy `followReexportChain` walked re-export drafts via mutual recursion guarded by a per-call visited set + a `MAX_REEXPORT_DEPTH` ceiling. Recursion is fragile (call-stack ceiling, no bound on depth that's actually meaningful), so this replaces it with a structurally better algorithm: a precomputed per-file re-export closure built by running Tarjan SCC over the re-export sub-graph and propagating names in reverse-topological order with a bounded intra-SCC fixpoint. Algorithm (`buildReexportClosures` in finalize-algorithm.ts): 1. Sub-graph: build the directed graph of `reexport` + `wildcard` drafts only (regular/namespace/dynamic imports do not contribute). 2. SCC condensation: run the same iterative `tarjanSccs` already used for the file-level import graph; output is in reverse-topo order so out-of-SCC neighbors are always already-finalized. 3. Per-SCC propagation: - Acyclic singleton: one pass populates from neighbors' closures. - Cyclic SCC: bounded fixpoint capped at |SCC|+1 iterations. With first-wins precedence the closure map is monotone, so each name needs at most |SCC| hops to traverse the cycle. Precedence (preserved from the recursive crawl): - Named re-exports take precedence over wildcards. - Within each kind, declaration order wins. Lookup at finalize time becomes O(1) (`lookupReexportedName`), down from O(chain_depth × drafts) per consult and recursive at that. Properties vs the legacy implementation: - Stack-safe by construction; no `MAX_REEXPORT_DEPTH` guard needed. - 1000-hop barrel chains now resolve in full (legacy capped at 100 and surfaced anything deeper as `unresolved`). - Cycles handled structurally via SCC, not via per-call visited set. - Same observable semantics: every existing test passes unchanged. Tests: - Replace the obsolete `MAX_REEXPORT_DEPTH (200-hop chain stops cleanly without stack overflow)` test (which asserted the OLD bug — that deep chains failed to resolve) with a positive 1000-hop test that asserts full resolution + accurate `transitiveVia`. Proves both the recursion is gone AND the closure correctly inherits the leaf def across all hops. - Update commentary on adjacent re-export tests to reference the closure mechanism. - Update `FinalizeFile.localDefs` JSDoc + import-decomposer.ts inline doc to point at `buildReexportClosures` instead of the removed function name. Validation: - gitnexus-shared builds cleanly. - gitnexus typechecks cleanly. - 28/28 finalize-algorithm.test.ts tests pass (incl. new 1000-hop). - 801/801 TypeScript scope-resolution tests pass under default (registry-primary) AND `REGISTRY_PRIMARY_TYPESCRIPT=0` (legacy DAG). - 404/404 Python + C# integration tests pass — no regression in cross-language consumers of the shared `finalize`. Made-with: Cursor * fix(scope): remove non-null assertions from scope resolution Made-with: Cursor * fix(scope): address TypeScript review follow-ups Made-with: Cursor * fix(scope): address TypeScript import review follow-ups Add regression coverage for non-binding import edges and circular TypeScript bindings so PR #1050 review concerns stay visible without changing runtime semantics. Made-with: Cursor
969 lines
36 KiB
TypeScript
969 lines
36 KiB
TypeScript
/**
|
||
* `finalize` — cross-file finalize algorithm for the SemanticModel
|
||
* (RFC §3.2 Phase 2; Ring 2 SHARED #915).
|
||
*
|
||
* Pure logic that takes per-file parse output (`ParsedImport[]` +
|
||
* `SymbolDefinition[]`) and returns:
|
||
*
|
||
* - Linked `ImportEdge[]` per module scope, with `targetModuleScope` and
|
||
* `targetDefId` filled where resolvable; edges that could not be
|
||
* resolved within the hard fixpoint cap are marked
|
||
* `linkStatus: 'unresolved'`.
|
||
* - Materialized `bindings` per module scope — local defs merged with
|
||
* imported / wildcard-expanded / re-exported names via the provider's
|
||
* `mergeBindings` precedence.
|
||
* - The SCC condensation of the import graph, exposed so disjoint SCCs
|
||
* can be processed in parallel by callers that want that.
|
||
*
|
||
* The algorithm is **SCC-aware**: it runs Tarjan SCC over the file-level
|
||
* import graph, processes SCCs in reverse-topological order (leaves
|
||
* first), and within each SCC runs a bounded fixpoint link pass capped at
|
||
* `N = |edges in SCC|`. Cyclic imports finalize without hanging; malformed
|
||
* inputs are bounded by the cap.
|
||
*
|
||
* **No language-specific logic.** Target resolution, wildcard expansion,
|
||
* and binding precedence all go through caller-supplied hooks
|
||
* (`resolveImportTarget`, `expandsWildcardTo`, `mergeBindings`) that
|
||
* match the LanguageProvider surface from #911.
|
||
*
|
||
* **Non-binding imports rule.** `dynamic-unresolved` passes through with
|
||
* `targetFile: null`; `dynamic-resolved` and `side-effect` resolve to
|
||
* file-level `ImportEdge`s. None of these materialize `BindingRef`s.
|
||
*/
|
||
|
||
import type { SymbolDefinition } from './symbol-definition.js';
|
||
import type { BindingRef, ImportEdge, ParsedImport, ScopeId, WorkspaceIndex } from './types.js';
|
||
|
||
// ─── Public contracts ───────────────────────────────────────────────────────
|
||
|
||
/** Per-file input for the finalize pass. */
|
||
export interface FinalizeFile {
|
||
readonly filePath: string;
|
||
/** The module scope id for this file; owns the finalized imports + bindings. */
|
||
readonly moduleScope: ScopeId;
|
||
readonly parsedImports: readonly ParsedImport[];
|
||
/**
|
||
* Defs exported from this file — the "what other files can import by name"
|
||
* surface. Typically those with `isExported: true` (the module's own
|
||
* declarations); parsers MAY also surface re-exported names here as a
|
||
* shortcut, but it is no longer required for correctness.
|
||
*
|
||
* **Multi-hop re-export contract.** `finalize` resolves an edge
|
||
* `A → B (importedName: 'X')` by first looking up `X` in `B.localDefs`.
|
||
* If `B` only has `export { X } from './C'` and does NOT surface `X` in
|
||
* its own `localDefs`, `finalize` falls back to the precomputed
|
||
* per-file re-export closure (`buildReexportClosures`), which encodes
|
||
* every name reachable through `B`'s named and wildcard re-exports —
|
||
* including transitively through cyclic SCCs. The lookup is O(1) and
|
||
* inherits the upstream `targetDefId`, populating `transitiveVia` with
|
||
* the file paths traversed to reach the leaf def.
|
||
*
|
||
* Surfacing re-exported names in `localDefs` is still a valid (and
|
||
* slightly cheaper) optimization: the direct lookup short-circuits the
|
||
* closure consult. Parsers SHOULD prefer surfacing names they can resolve
|
||
* statically (e.g., `export { X } from './c'` when `c.ts` is parsed in
|
||
* the same workspace), and rely on the closure for the long tail of
|
||
* barrel patterns.
|
||
*
|
||
* The fixpoint does NOT mutate `localDefs` across iterations — it is
|
||
* static input.
|
||
*/
|
||
readonly localDefs: readonly SymbolDefinition[];
|
||
}
|
||
|
||
/** Input to `finalize`. */
|
||
export interface FinalizeInput {
|
||
readonly files: readonly FinalizeFile[];
|
||
/** Opaque workspace context forwarded to provider hooks. */
|
||
readonly workspaceIndex: WorkspaceIndex;
|
||
}
|
||
|
||
/**
|
||
* Provider-supplied hooks. Mirror the optional LanguageProvider scope-
|
||
* resolution hooks declared in #911; `finalize` calls them pure-ly and
|
||
* expects pure answers.
|
||
*/
|
||
export interface FinalizeHooks {
|
||
/**
|
||
* Resolve a raw import target to the concrete file path that owns it.
|
||
* Return `null` when no target file is resolvable (e.g., `np.foo` when
|
||
* `numpy` is external to the workspace).
|
||
*/
|
||
resolveImportTarget(
|
||
targetRaw: string,
|
||
fromFile: string,
|
||
workspaceIndex: WorkspaceIndex,
|
||
): string | null;
|
||
|
||
/**
|
||
* For a wildcard `import * from M`, return the names visible in the
|
||
* exporting module scope `M`. The finalize pass looks each name up in
|
||
* `M`'s local defs to produce a concrete `BindingRef`; names with no
|
||
* matching export are dropped.
|
||
*/
|
||
expandsWildcardTo(targetModuleScope: ScopeId, workspaceIndex: WorkspaceIndex): readonly string[];
|
||
|
||
/**
|
||
* Merge `incoming` bindings into `existing` for a given name. Called
|
||
* once per name at each scope. Typical rules:
|
||
* - Python: local > imported > wildcard (last-write-wins within tier).
|
||
* - Rust: explicit `use` > glob; `pub use` overrides.
|
||
* Return value replaces the bucket entirely — no implicit append.
|
||
*/
|
||
mergeBindings(
|
||
existing: readonly BindingRef[],
|
||
incoming: readonly BindingRef[],
|
||
scope: ScopeId,
|
||
): readonly BindingRef[];
|
||
}
|
||
|
||
/** One SCC in the file-level import graph. */
|
||
export interface FinalizedScc {
|
||
readonly files: readonly string[];
|
||
/** True iff this SCC has ≥ 2 files OR a single file that self-imports. */
|
||
readonly isCycle: boolean;
|
||
}
|
||
|
||
/**
|
||
* Counters reported by `finalize`.
|
||
*
|
||
* **Counting granularity** — all edge counters are **per-`ParsedImport`**,
|
||
* not per-materialized-`ImportEdge`. A single `wildcard` ParsedImport that
|
||
* expands to N exports counts as one linked edge in these stats; the
|
||
* materialized output (`FinalizeOutput.imports`) will have N edges for
|
||
* that input. `dynamic-unresolved` ParsedImports count as linked (they
|
||
* pass through with no `linkStatus`), so `linkedEdges` ≠ "has a
|
||
* BindingRef" — use the `bindings` map for that.
|
||
*
|
||
* In other words: `totalEdges === input.parsedImports.length` summed
|
||
* across files, and `linkedEdges + unresolvedEdges === totalEdges`.
|
||
*/
|
||
export interface FinalizeStats {
|
||
readonly totalFiles: number;
|
||
/** Total `ParsedImport` records seen across all files. */
|
||
readonly totalEdges: number;
|
||
/**
|
||
* `ParsedImport`s whose finalized edge does NOT carry
|
||
* `linkStatus: 'unresolved'`. Includes `dynamic-unresolved` pass-throughs.
|
||
*/
|
||
readonly linkedEdges: number;
|
||
/** `ParsedImport`s whose finalized edge carries `linkStatus: 'unresolved'`. */
|
||
readonly unresolvedEdges: number;
|
||
readonly sccCount: number;
|
||
readonly largestSccSize: number;
|
||
}
|
||
|
||
export interface FinalizeOutput {
|
||
/** Linked `ImportEdge[]` per module scope, in original input order. */
|
||
readonly imports: ReadonlyMap<ScopeId, readonly ImportEdge[]>;
|
||
/** Materialized bindings per module scope. */
|
||
readonly bindings: ReadonlyMap<ScopeId, ReadonlyMap<string, readonly BindingRef[]>>;
|
||
/** SCCs in reverse-topological order (leaves first). */
|
||
readonly sccs: readonly FinalizedScc[];
|
||
readonly stats: FinalizeStats;
|
||
}
|
||
|
||
// ─── Entry point ───────────────────────────────────────────────────────────
|
||
|
||
export function finalize(input: FinalizeInput, hooks: FinalizeHooks): FinalizeOutput {
|
||
const byFilePath = new Map<string, FinalizeFile>();
|
||
for (const f of input.files) byFilePath.set(f.filePath, f);
|
||
|
||
// ── Phase 0: pre-resolve raw import targets (one syscall-equivalent per
|
||
// (file, parsedImport)). Edges with no resolvable target become
|
||
// `linkStatus: 'unresolved'` or, for dynamic-unresolved, pass through
|
||
// with `targetFile: null`.
|
||
const edgeIndex = new Map<string, ImportEdgeDraft[]>(); // filePath → drafts
|
||
let totalEdges = 0;
|
||
|
||
for (const file of input.files) {
|
||
const drafts: ImportEdgeDraft[] = [];
|
||
for (const parsed of file.parsedImports) {
|
||
const draft = makeEdgeDraft(parsed, file, hooks, input.workspaceIndex);
|
||
drafts.push(draft);
|
||
totalEdges++;
|
||
}
|
||
edgeIndex.set(file.filePath, drafts);
|
||
}
|
||
|
||
// ── Phase 1: build file-level import graph (only resolvable edges form
|
||
// graph edges; unresolvable ones are terminal and contribute no
|
||
// fixpoint obligation).
|
||
const graph = new Map<string, Set<string>>();
|
||
for (const file of input.files) {
|
||
graph.set(file.filePath, new Set());
|
||
}
|
||
for (const [fromFile, drafts] of edgeIndex) {
|
||
const edges = graph.get(fromFile);
|
||
if (edges === undefined) continue;
|
||
for (const d of drafts) {
|
||
if (d.targetFile !== null && byFilePath.has(d.targetFile)) {
|
||
edges.add(d.targetFile);
|
||
}
|
||
}
|
||
}
|
||
|
||
// ── Phase 2: Tarjan SCC → reverse-topological list of SCCs.
|
||
const sccs = tarjanSccs(graph);
|
||
|
||
// ── Phase 2.5: precompute the per-file re-export closure (iterative,
|
||
// SCC-condensed). Eliminates the recursive crawl that the per-edge
|
||
// `tryFinalize` call site used to do; lookups are O(1) afterwards.
|
||
// See `buildReexportClosures` for the algorithm.
|
||
const reexportClosures = buildReexportClosures(input.files, byFilePath, edgeIndex);
|
||
|
||
// ── Phase 3: process SCCs in reverse-topological order (leaves first).
|
||
// Within each SCC, run a bounded fixpoint that resolves intra-SCC edges.
|
||
// Edges leaving the SCC are already resolved (their target SCC is
|
||
// already finalized); edges inside the SCC may need multiple passes.
|
||
const linkedByScope = new Map<ScopeId, readonly ImportEdge[]>();
|
||
let linkedEdges = 0;
|
||
|
||
for (const scc of sccs) {
|
||
const sccFiles = new Set(scc.files);
|
||
const capacity = countEdgesWithin(edgeIndex, sccFiles);
|
||
|
||
// Run the fixpoint up to `capacity` iterations. Each iteration tries to
|
||
// resolve every still-unlinked edge in the SCC; stops early if a pass
|
||
// makes no progress.
|
||
let progressed = true;
|
||
let iterations = 0;
|
||
while (progressed && iterations < capacity) {
|
||
progressed = false;
|
||
iterations++;
|
||
for (const filePath of scc.files) {
|
||
const drafts = edgeIndex.get(filePath);
|
||
if (drafts === undefined) continue;
|
||
for (const draft of drafts) {
|
||
if (draft.finalized !== null) continue;
|
||
const finalized = tryFinalize(draft, byFilePath, reexportClosures);
|
||
if (finalized !== null) {
|
||
draft.finalized = finalized;
|
||
progressed = true;
|
||
}
|
||
}
|
||
}
|
||
}
|
||
|
||
// Any drafts still not finalized within this SCC hit the cap → unresolved.
|
||
for (const filePath of scc.files) {
|
||
const drafts = edgeIndex.get(filePath);
|
||
if (drafts === undefined) continue;
|
||
for (const draft of drafts) {
|
||
if (draft.finalized !== null) continue;
|
||
draft.finalized = {
|
||
...draft.base,
|
||
linkStatus: 'unresolved' as const,
|
||
};
|
||
}
|
||
}
|
||
}
|
||
|
||
// ── Phase 4: collect finalized `ImportEdge[]` per module scope, preserving
|
||
// input order within each file, and wildcard-expand where applicable.
|
||
for (const file of input.files) {
|
||
const drafts = edgeIndex.get(file.filePath);
|
||
if (drafts === undefined) continue;
|
||
const finalized: ImportEdge[] = [];
|
||
for (const d of drafts) {
|
||
const edge = d.finalized;
|
||
if (edge === null) {
|
||
throw new Error(`Invariant violated: import edge was not finalized for ${file.filePath}`);
|
||
}
|
||
if (d.source.kind === 'wildcard' && edge.linkStatus !== 'unresolved') {
|
||
// Produce one `wildcard-expanded` ImportEdge per exported name.
|
||
const expanded = expandWildcard(edge, byFilePath, hooks, input.workspaceIndex);
|
||
for (const e of expanded) finalized.push(e);
|
||
} else {
|
||
finalized.push(edge);
|
||
}
|
||
if (edge.linkStatus !== 'unresolved') linkedEdges++;
|
||
}
|
||
linkedByScope.set(file.moduleScope, Object.freeze(finalized));
|
||
}
|
||
|
||
// ── Phase 5: materialize module-scope bindings (local + imports + wildcards),
|
||
// delegating precedence to `provider.mergeBindings`.
|
||
const bindingsByScope = materializeBindings(input.files, linkedByScope, hooks);
|
||
|
||
// ── Stats.
|
||
const sccCount = sccs.length;
|
||
let largestSccSize = 0;
|
||
for (const scc of sccs) {
|
||
if (scc.files.length > largestSccSize) largestSccSize = scc.files.length;
|
||
}
|
||
const stats: FinalizeStats = {
|
||
totalFiles: input.files.length,
|
||
totalEdges,
|
||
linkedEdges,
|
||
unresolvedEdges: totalEdges - linkedEdges,
|
||
sccCount,
|
||
largestSccSize,
|
||
};
|
||
|
||
return Object.freeze({
|
||
imports: linkedByScope,
|
||
bindings: bindingsByScope,
|
||
sccs,
|
||
stats,
|
||
});
|
||
}
|
||
|
||
// ─── Internal: edge drafting (phase 0) ──────────────────────────────────────
|
||
|
||
interface ImportEdgeDraft {
|
||
readonly source: ParsedImport;
|
||
readonly fromFile: string;
|
||
readonly fromScope: ScopeId;
|
||
readonly targetFile: string | null;
|
||
readonly base: ImportEdge;
|
||
finalized: ImportEdge | null;
|
||
}
|
||
|
||
function makeEdgeDraft(
|
||
parsed: ParsedImport,
|
||
file: FinalizeFile,
|
||
hooks: FinalizeHooks,
|
||
workspace: WorkspaceIndex,
|
||
): ImportEdgeDraft {
|
||
// Dynamic-unresolved passes through — no `BindingRef`, no target file.
|
||
if (parsed.kind === 'dynamic-unresolved') {
|
||
const base: ImportEdge = {
|
||
localName: parsed.localName,
|
||
targetFile: null,
|
||
targetExportedName: '',
|
||
kind: 'dynamic-unresolved',
|
||
};
|
||
return {
|
||
source: parsed,
|
||
fromFile: file.filePath,
|
||
fromScope: file.moduleScope,
|
||
targetFile: null,
|
||
base,
|
||
finalized: base, // already fully finalized
|
||
};
|
||
}
|
||
|
||
const targetFile = hooks.resolveImportTarget(parsed.targetRaw ?? '', file.filePath, workspace);
|
||
|
||
// Edge is unresolvable at the file level — mark unresolved now.
|
||
if (targetFile === null) {
|
||
const base: ImportEdge = {
|
||
localName: extractLocalName(parsed),
|
||
targetFile: null,
|
||
targetExportedName: extractExportedName(parsed),
|
||
kind: edgeKindFor(parsed),
|
||
linkStatus: 'unresolved',
|
||
};
|
||
return {
|
||
source: parsed,
|
||
fromFile: file.filePath,
|
||
fromScope: file.moduleScope,
|
||
targetFile: null,
|
||
base,
|
||
finalized: base,
|
||
};
|
||
}
|
||
|
||
// Resolvable at the file level; intra-SCC fixpoint may still fail to fill
|
||
// in `targetDefId` (e.g., symbol not exported from target). Side-effect
|
||
// and resolved-dynamic imports are terminal at the file level — no
|
||
// `targetDefId` needed since they materialize no `BindingRef`. Pre-
|
||
// finalize them here so the fixpoint loop skips them entirely.
|
||
const base: ImportEdge = {
|
||
localName: extractLocalName(parsed),
|
||
targetFile,
|
||
targetExportedName: extractExportedName(parsed),
|
||
kind: edgeKindFor(parsed),
|
||
};
|
||
const isFileLevelTerminal = parsed.kind === 'side-effect' || parsed.kind === 'dynamic-resolved';
|
||
return {
|
||
source: parsed,
|
||
fromFile: file.filePath,
|
||
fromScope: file.moduleScope,
|
||
targetFile,
|
||
base,
|
||
finalized: isFileLevelTerminal ? base : null,
|
||
};
|
||
}
|
||
|
||
function edgeKindFor(parsed: ParsedImport): ImportEdge['kind'] {
|
||
if (parsed.kind === 'wildcard') return 'wildcard-expanded';
|
||
return parsed.kind;
|
||
}
|
||
|
||
function extractLocalName(parsed: ParsedImport): string {
|
||
switch (parsed.kind) {
|
||
case 'wildcard':
|
||
case 'side-effect':
|
||
case 'dynamic-resolved':
|
||
return '';
|
||
default:
|
||
return parsed.localName;
|
||
}
|
||
}
|
||
|
||
function extractExportedName(parsed: ParsedImport): string {
|
||
switch (parsed.kind) {
|
||
case 'named':
|
||
case 'alias':
|
||
case 'namespace':
|
||
case 'reexport':
|
||
return parsed.importedName;
|
||
case 'wildcard':
|
||
case 'dynamic-unresolved':
|
||
case 'dynamic-resolved':
|
||
case 'side-effect':
|
||
return '';
|
||
}
|
||
}
|
||
|
||
// ─── Internal: per-edge finalization (phase 3) ─────────────────────────────
|
||
|
||
function tryFinalize(
|
||
draft: ImportEdgeDraft,
|
||
byFilePath: Map<string, FinalizeFile>,
|
||
reexportClosures: ReadonlyMap<string, FileReexportClosure>,
|
||
): ImportEdge | null {
|
||
const targetFile = draft.targetFile;
|
||
if (targetFile === null) return draft.base; // already terminal
|
||
|
||
const targetModule = byFilePath.get(targetFile);
|
||
if (targetModule === undefined) return draft.base; // external target — leave as-is
|
||
|
||
// Wildcards finalize at the file level; their per-name expansion happens
|
||
// in phase 4. At this stage we just record the target module scope.
|
||
if (draft.source.kind === 'wildcard') {
|
||
return {
|
||
...draft.base,
|
||
targetModuleScope: targetModule.moduleScope,
|
||
};
|
||
}
|
||
|
||
// Namespace imports alias the target *module*; they don't name a
|
||
// specific export. Link the module scope unconditionally. If the target
|
||
// also exposes a def whose simple name matches `importedName` (some
|
||
// languages emit a synthetic module-def), pick it up as the `targetDefId`
|
||
// so consumers can reach the module as a symbol — but its absence is not
|
||
// a failure.
|
||
if (draft.source.kind === 'namespace') {
|
||
const moduleDef = findExportByName(targetModule.localDefs, extractExportedName(draft.source));
|
||
return {
|
||
...draft.base,
|
||
targetModuleScope: targetModule.moduleScope,
|
||
...(moduleDef !== undefined ? { targetDefId: moduleDef.nodeId } : {}),
|
||
};
|
||
}
|
||
|
||
// named / alias / reexport: look up the imported name in the target's
|
||
// local defs. Multi-hop re-export chains settle iteratively — each hop
|
||
// resolves once its prior hop is finalized.
|
||
const importedName = extractExportedName(draft.source);
|
||
const exported = findExportByName(targetModule.localDefs, importedName);
|
||
|
||
if (exported !== undefined) {
|
||
const transitiveVia =
|
||
draft.source.kind === 'reexport' ? Object.freeze([targetFile]) : undefined;
|
||
return {
|
||
...draft.base,
|
||
targetModuleScope: targetModule.moduleScope,
|
||
targetDefId: exported.nodeId,
|
||
...(transitiveVia !== undefined ? { transitiveVia } : {}),
|
||
};
|
||
}
|
||
|
||
// Multi-hop re-export follow. Barrel modules like
|
||
// // models.ts
|
||
// export { User } from './base';
|
||
// emit no local def for `User`; the name surfaces only via their own
|
||
// `reexport` edge. The per-file re-export closure built in phase 2.5
|
||
// already encodes every name reachable through that file's named and
|
||
// wildcard re-exports — including transitively through cyclic SCCs —
|
||
// so the lookup is O(1) and never recurses.
|
||
const followed = lookupReexportedName(reexportClosures, targetFile, importedName);
|
||
if (followed === null) {
|
||
// Target resolvable but the name isn't exported — keep trying in case a
|
||
// re-export inside the target's SCC surfaces it in a later iteration.
|
||
return null;
|
||
}
|
||
|
||
const viaFiles = [targetFile, ...followed.via];
|
||
const transitiveVia =
|
||
draft.source.kind === 'reexport' || viaFiles.length > 1 ? Object.freeze(viaFiles) : undefined;
|
||
|
||
return {
|
||
...draft.base,
|
||
targetModuleScope: targetModule.moduleScope,
|
||
targetDefId: followed.def.nodeId,
|
||
...(transitiveVia !== undefined ? { transitiveVia } : {}),
|
||
};
|
||
}
|
||
|
||
// ─── Internal: re-export closure (phase 2.5) ───────────────────────────────
|
||
|
||
/**
|
||
* Per-file map of `name → terminal def + via path` — i.e. every name
|
||
* importable from this file via its named/wildcard re-export chain
|
||
* (excluding the file's own `localDefs`, which the caller checks first
|
||
* via `findExportByName`). `via` is the ordered list of intermediate
|
||
* files traversed to reach the def.
|
||
*
|
||
* Built once per finalize pass. Lookups are O(1).
|
||
*/
|
||
type ReexportClosureEntry = { readonly def: SymbolDefinition; readonly via: readonly string[] };
|
||
type FileReexportClosure = ReadonlyMap<string, ReexportClosureEntry>;
|
||
|
||
/**
|
||
* Build per-file re-export closures.
|
||
*
|
||
* **Algorithm.** Iterative SCC-condensed reverse-topological propagation,
|
||
* structurally identical to how `finalize` itself processes the file-
|
||
* level import graph. Replaces the legacy recursive
|
||
* `followReexportChain` crawl with a bounded, stack-safe pass:
|
||
*
|
||
* 1. **Sub-graph.** Build a directed graph whose edges are
|
||
* `reexport` and `wildcard` drafts only (regular imports do not
|
||
* contribute to the export surface, and `namespace`/
|
||
* `reexport-namespace` are terminal — their target def lives in
|
||
* `localDefs`).
|
||
* 2. **SCC condensation.** Run the same iterative `tarjanSccs` over
|
||
* the sub-graph. Output is in reverse-topological order (leaves
|
||
* first), so when we process an SCC every out-of-SCC neighbor
|
||
* already has its closure populated.
|
||
* 3. **Per-SCC propagation.**
|
||
* * Acyclic singleton: one pass — read neighbors' (already
|
||
* fully populated) closures.
|
||
* * Cyclic SCC (cycle ≥ 2 files, or self-loop): bounded
|
||
* fixpoint inside the SCC, capped at `|SCC| + 1` iterations
|
||
* (each iteration propagates names one hop further around
|
||
* the cycle; first-wins precedence keeps the map monotone
|
||
* so the fixpoint converges in at most |SCC| hops).
|
||
*
|
||
* **Precedence semantics — preserved from the recursive crawl.**
|
||
* * Named re-exports take precedence over wildcards.
|
||
* * Within each kind, declaration order wins (first match for a
|
||
* given exported name is kept; later drafts skip).
|
||
*
|
||
* **Complexity.**
|
||
* * Pre-pass: O(V + E_re) for SCC, plus O(|SCC| × Σ drafts) per cyclic
|
||
* SCC. For tree-shaped barrel graphs (the common case) it
|
||
* collapses to O(E_re) total.
|
||
* * Per-edge lookup at finalize time: O(1).
|
||
* * `transitiveVia` preserves the exact file path chain for diagnostics
|
||
* and graph provenance. Building those arrays copies the inherited path,
|
||
* which is O(depth²) in a pathological single-name barrel chain; practical
|
||
* TypeScript barrel chains are shallow enough that we keep exact paths
|
||
* instead of capping or summarizing them.
|
||
* * Pathological deep chains that previously needed
|
||
* `MAX_REEXPORT_DEPTH=100` to bound stack growth now resolve
|
||
* in full and are bounded only by available memory — the
|
||
* iterative formulation has no call-stack ceiling.
|
||
*/
|
||
function buildReexportClosures(
|
||
files: readonly FinalizeFile[],
|
||
byFilePath: ReadonlyMap<string, FinalizeFile>,
|
||
edgeIndex: ReadonlyMap<string, ImportEdgeDraft[]>,
|
||
): ReadonlyMap<string, FileReexportClosure> {
|
||
const closures = new Map<string, Map<string, ReexportClosureEntry>>();
|
||
for (const file of files) closures.set(file.filePath, new Map());
|
||
|
||
// ── Step 1: build the re-export sub-graph (only resolvable
|
||
// reexport/wildcard targets contribute edges).
|
||
const subGraph = new Map<string, Set<string>>();
|
||
for (const file of files) {
|
||
const targets = new Set<string>();
|
||
const drafts = edgeIndex.get(file.filePath);
|
||
if (drafts !== undefined) {
|
||
for (const d of drafts) {
|
||
if (d.source.kind !== 'reexport' && d.source.kind !== 'wildcard') continue;
|
||
if (d.targetFile === null) continue;
|
||
if (!byFilePath.has(d.targetFile)) continue;
|
||
targets.add(d.targetFile);
|
||
}
|
||
}
|
||
subGraph.set(file.filePath, targets);
|
||
}
|
||
|
||
// ── Step 2: SCC over the sub-graph. Reuses the same iterative Tarjan
|
||
// implementation that drives the file-level finalize loop, so any
|
||
// call-stack-safety guarantees there transfer here unchanged.
|
||
const subSccs = tarjanSccs(subGraph);
|
||
|
||
// ── Step 3: process SCCs in reverse-topological order. Acyclic
|
||
// singletons settle in one pass; cyclic SCCs run a bounded fixpoint.
|
||
for (const scc of subSccs) {
|
||
if (!scc.isCycle) {
|
||
const filePath = scc.files[0];
|
||
if (filePath !== undefined) {
|
||
populateFileClosure(filePath, byFilePath, edgeIndex, closures);
|
||
}
|
||
continue;
|
||
}
|
||
// Cap = |SCC| + 1. With first-wins precedence each name needs at
|
||
// most |SCC| iterations to propagate fully around the cycle; the
|
||
// extra iteration confirms no progress and breaks the loop.
|
||
const cap = scc.files.length + 1;
|
||
let progressed = true;
|
||
let iter = 0;
|
||
while (progressed && iter < cap) {
|
||
progressed = false;
|
||
iter++;
|
||
for (const filePath of scc.files) {
|
||
if (populateFileClosure(filePath, byFilePath, edgeIndex, closures)) {
|
||
progressed = true;
|
||
}
|
||
}
|
||
}
|
||
}
|
||
|
||
return closures;
|
||
}
|
||
|
||
/**
|
||
* Populate one file's re-export closure for one pass. Returns `true`
|
||
* iff the closure grew (signalling fixpoint progress to the caller).
|
||
*
|
||
* Walks the file's drafts in declaration order, named re-exports first
|
||
* (precedence), then wildcards. For each draft, attempts:
|
||
* 1. **Direct hit** — name exists in the target file's `localDefs`.
|
||
* 2. **Inherited** — name exists in the target file's already-populated
|
||
* closure (which encodes the target's own re-export chain).
|
||
*
|
||
* `closures.get(targetFile)` may itself still be empty for in-SCC
|
||
* targets on the first iteration; the outer fixpoint loop handles
|
||
* that by re-invoking this function.
|
||
*/
|
||
function populateFileClosure(
|
||
filePath: string,
|
||
byFilePath: ReadonlyMap<string, FinalizeFile>,
|
||
edgeIndex: ReadonlyMap<string, ImportEdgeDraft[]>,
|
||
closures: Map<string, Map<string, ReexportClosureEntry>>,
|
||
): boolean {
|
||
const myClosure = closures.get(filePath);
|
||
if (myClosure === undefined) return false;
|
||
const before = myClosure.size;
|
||
const drafts = edgeIndex.get(filePath);
|
||
if (drafts === undefined) return false;
|
||
|
||
// Named re-exports — precedence over wildcards, declaration order
|
||
// first-wins for duplicates of the same exported name.
|
||
for (const draft of drafts) {
|
||
if (draft.source.kind !== 'reexport') continue;
|
||
const targetFile = draft.targetFile;
|
||
if (targetFile === null) continue;
|
||
const targetModule = byFilePath.get(targetFile);
|
||
if (targetModule === undefined) continue;
|
||
|
||
const localName = draft.source.localName;
|
||
if (myClosure.has(localName)) continue;
|
||
|
||
const importedName = draft.source.importedName;
|
||
const direct = findExportByName(targetModule.localDefs, importedName);
|
||
if (direct !== undefined) {
|
||
myClosure.set(localName, { def: direct, via: Object.freeze([targetFile]) });
|
||
continue;
|
||
}
|
||
const inherited = closures.get(targetFile)?.get(importedName);
|
||
if (inherited !== undefined) {
|
||
myClosure.set(localName, {
|
||
def: inherited.def,
|
||
via: Object.freeze([targetFile, ...inherited.via]),
|
||
});
|
||
}
|
||
// Else: target's closure is still empty (in-SCC, awaiting next
|
||
// iteration). Outer loop will revisit.
|
||
}
|
||
|
||
// Wildcard re-exports — fan out the target's own surface (localDefs
|
||
// + transitive closure). `myClosure.has(name)` checks below preserve
|
||
// the named-precedence and first-wins semantics from above.
|
||
for (const draft of drafts) {
|
||
if (draft.source.kind !== 'wildcard') continue;
|
||
const targetFile = draft.targetFile;
|
||
if (targetFile === null) continue;
|
||
const targetModule = byFilePath.get(targetFile);
|
||
if (targetModule === undefined) continue;
|
||
|
||
for (const def of targetModule.localDefs) {
|
||
const name = deriveSimpleName(def);
|
||
if (name === null || myClosure.has(name)) continue;
|
||
myClosure.set(name, { def, via: Object.freeze([targetFile]) });
|
||
}
|
||
const targetClosure = closures.get(targetFile);
|
||
if (targetClosure !== undefined) {
|
||
for (const [name, entry] of targetClosure) {
|
||
if (myClosure.has(name)) continue;
|
||
myClosure.set(name, {
|
||
def: entry.def,
|
||
via: Object.freeze([targetFile, ...entry.via]),
|
||
});
|
||
}
|
||
}
|
||
}
|
||
|
||
return myClosure.size > before;
|
||
}
|
||
|
||
/**
|
||
* O(1) lookup into a precomputed re-export closure. Replaces the legacy
|
||
* recursive `followReexportChain` traversal with a single map indexing.
|
||
*/
|
||
function lookupReexportedName(
|
||
closures: ReadonlyMap<string, FileReexportClosure>,
|
||
filePath: string,
|
||
name: string,
|
||
): { def: SymbolDefinition; via: readonly string[] } | null {
|
||
const closure = closures.get(filePath);
|
||
if (closure === undefined) return null;
|
||
const entry = closure.get(name);
|
||
if (entry === undefined) return null;
|
||
return { def: entry.def, via: entry.via };
|
||
}
|
||
|
||
/**
|
||
* The "simple" (unqualified) name of a def, for import-name matching.
|
||
*
|
||
* Canonical source: `def.qualifiedName` — the tail after the last `.` (or
|
||
* the whole string if no dot). Defs without a qualifiedName can't be
|
||
* resolved by name here and return `null`; callers treat that as "name
|
||
* not exported" and either retry in a later fixpoint iteration or mark
|
||
* the edge unresolved.
|
||
*/
|
||
function deriveSimpleName(def: SymbolDefinition): string | null {
|
||
const q = def.qualifiedName;
|
||
if (q === undefined || q.length === 0) return null;
|
||
const dot = q.lastIndexOf('.');
|
||
return dot === -1 ? q : q.slice(dot + 1);
|
||
}
|
||
|
||
function findExportByName(
|
||
defs: readonly SymbolDefinition[],
|
||
name: string,
|
||
): SymbolDefinition | undefined {
|
||
for (const d of defs) {
|
||
if (deriveSimpleName(d) === name) return d;
|
||
}
|
||
return undefined;
|
||
}
|
||
|
||
function countEdgesWithin(edgeIndex: Map<string, ImportEdgeDraft[]>, files: Set<string>): number {
|
||
let n = 0;
|
||
for (const filePath of files) {
|
||
const drafts = edgeIndex.get(filePath);
|
||
if (drafts === undefined) continue;
|
||
for (const d of drafts) {
|
||
if (d.targetFile !== null && files.has(d.targetFile)) n++;
|
||
}
|
||
}
|
||
// Guarantee at least one pass even for a trivial SCC (ensures deterministic
|
||
// fixpoint termination even when a single-file SCC has zero intra-SCC edges
|
||
// but still needs one settle pass).
|
||
return Math.max(n, 1);
|
||
}
|
||
|
||
// ─── Internal: wildcard expansion (phase 4) ────────────────────────────────
|
||
|
||
function expandWildcard(
|
||
edge: ImportEdge,
|
||
byFilePath: Map<string, FinalizeFile>,
|
||
hooks: FinalizeHooks,
|
||
workspace: WorkspaceIndex,
|
||
): readonly ImportEdge[] {
|
||
if (edge.targetModuleScope === undefined || edge.targetFile === null) {
|
||
return [edge]; // unresolvable wildcard survives as a single unlinked edge
|
||
}
|
||
const target = byFilePath.get(edge.targetFile);
|
||
if (target === undefined) return [edge];
|
||
|
||
const names = hooks.expandsWildcardTo(edge.targetModuleScope, workspace);
|
||
if (names.length === 0) return [];
|
||
|
||
const expanded: ImportEdge[] = [];
|
||
for (const name of names) {
|
||
const def = findExportByName(target.localDefs, name);
|
||
if (def === undefined) continue;
|
||
expanded.push({
|
||
localName: name,
|
||
targetFile: edge.targetFile,
|
||
targetExportedName: name,
|
||
kind: 'wildcard-expanded',
|
||
targetModuleScope: edge.targetModuleScope,
|
||
targetDefId: def.nodeId,
|
||
});
|
||
}
|
||
return expanded;
|
||
}
|
||
|
||
// ─── Internal: bindings materialization (phase 5) ───────────────────────────
|
||
|
||
function materializeBindings(
|
||
files: readonly FinalizeFile[],
|
||
linkedByScope: ReadonlyMap<ScopeId, readonly ImportEdge[]>,
|
||
hooks: FinalizeHooks,
|
||
): ReadonlyMap<ScopeId, ReadonlyMap<string, readonly BindingRef[]>> {
|
||
const out = new Map<ScopeId, ReadonlyMap<string, readonly BindingRef[]>>();
|
||
|
||
// Build a `nodeId → SymbolDefinition` index once across all files
|
||
// (O(N_files × D_defs)) so the per-edge lookup below is O(1) instead
|
||
// of a full linear scan. At realistic TypeScript monorepo scale
|
||
// (~5k files × ~50 defs × ~100k linked import edges) this is the
|
||
// difference between ~25 s and a few ms inside finalize. The map
|
||
// is local to this pass — no cross-pass state leaks.
|
||
const defById = new Map<string, SymbolDefinition>();
|
||
for (const f of files) {
|
||
for (const d of f.localDefs) defById.set(d.nodeId, d);
|
||
}
|
||
|
||
for (const file of files) {
|
||
const scopeBindings = new Map<string, readonly BindingRef[]>();
|
||
|
||
// Start with local defs as `origin: 'local'` bindings.
|
||
for (const def of file.localDefs) {
|
||
const name = deriveSimpleName(def);
|
||
if (name === null) continue;
|
||
const incoming: BindingRef[] = [{ def, origin: 'local' }];
|
||
const existing = scopeBindings.get(name) ?? [];
|
||
scopeBindings.set(name, hooks.mergeBindings(existing, incoming, file.moduleScope));
|
||
}
|
||
|
||
// Layer in finalized imports.
|
||
const imports = linkedByScope.get(file.moduleScope) ?? [];
|
||
for (const edge of imports) {
|
||
if (edge.targetDefId === undefined || edge.linkStatus === 'unresolved') continue;
|
||
const def = defById.get(edge.targetDefId);
|
||
if (def === undefined) continue;
|
||
|
||
const origin: BindingRef['origin'] =
|
||
edge.kind === 'namespace'
|
||
? 'namespace'
|
||
: edge.kind === 'wildcard-expanded'
|
||
? 'wildcard'
|
||
: edge.kind === 'reexport'
|
||
? 'reexport'
|
||
: 'import';
|
||
const fallback = deriveSimpleName(def);
|
||
const name = edge.localName.length > 0 ? edge.localName : fallback;
|
||
if (name === null) continue;
|
||
const incoming: BindingRef[] = [{ def, origin, via: edge }];
|
||
const existing = scopeBindings.get(name) ?? [];
|
||
scopeBindings.set(name, hooks.mergeBindings(existing, incoming, file.moduleScope));
|
||
}
|
||
|
||
// Freeze nested buckets for immutability.
|
||
const frozen = new Map<string, readonly BindingRef[]>();
|
||
for (const [name, refs] of scopeBindings) {
|
||
frozen.set(name, Object.freeze(refs.slice()));
|
||
}
|
||
out.set(file.moduleScope, frozen);
|
||
}
|
||
|
||
return out;
|
||
}
|
||
|
||
// ─── Internal: Tarjan SCC ──────────────────────────────────────────────────
|
||
|
||
/**
|
||
* Iterative Tarjan SCC. Returns SCCs in **reverse-topological** order
|
||
* (leaves first — a property Tarjan gives for free, and the order
|
||
* `finalize` wants so leaves are fully resolved before their dependents).
|
||
*/
|
||
function tarjanSccs(graph: ReadonlyMap<string, ReadonlySet<string>>): FinalizedScc[] {
|
||
const index = new Map<string, number>();
|
||
const lowlink = new Map<string, number>();
|
||
const onStack = new Set<string>();
|
||
const stack: string[] = [];
|
||
const sccs: FinalizedScc[] = [];
|
||
let idx = 0;
|
||
|
||
// Iterative DFS to avoid stack overflow on deep import chains.
|
||
const allNodes = Array.from(graph.keys()).sort(); // deterministic order
|
||
const iterStack: Array<{ node: string; children: Iterator<string>; entered: boolean }> = [];
|
||
|
||
for (const root of allNodes) {
|
||
if (index.has(root)) continue;
|
||
iterStack.push({
|
||
node: root,
|
||
children: (graph.get(root) ?? new Set<string>()).values(),
|
||
entered: false,
|
||
});
|
||
while (iterStack.length > 0) {
|
||
const frame = iterStack[iterStack.length - 1];
|
||
if (frame === undefined) break;
|
||
|
||
if (!frame.entered) {
|
||
frame.entered = true;
|
||
index.set(frame.node, idx);
|
||
lowlink.set(frame.node, idx);
|
||
idx++;
|
||
stack.push(frame.node);
|
||
onStack.add(frame.node);
|
||
}
|
||
|
||
const nextChild = frame.children.next();
|
||
if (nextChild.done) {
|
||
// Post-visit: compute SCC membership if frame.node is a root.
|
||
if (lowlink.get(frame.node) === index.get(frame.node)) {
|
||
const scc: string[] = [];
|
||
let selfInCycle = false;
|
||
while (true) {
|
||
const w = stack.pop();
|
||
if (w === undefined) {
|
||
throw new Error(`Invariant violated: Tarjan stack exhausted at ${frame.node}`);
|
||
}
|
||
onStack.delete(w);
|
||
scc.push(w);
|
||
// A single-file self-loop counts as a cycle.
|
||
if (w === frame.node) {
|
||
selfInCycle = (graph.get(w) ?? new Set()).has(w);
|
||
break;
|
||
}
|
||
}
|
||
const isCycle = scc.length > 1 || selfInCycle;
|
||
sccs.push({ files: Object.freeze(scc), isCycle });
|
||
}
|
||
iterStack.pop();
|
||
// Propagate lowlink to parent.
|
||
if (iterStack.length > 0) {
|
||
const parent = iterStack[iterStack.length - 1];
|
||
if (parent !== undefined) {
|
||
lowlink.set(
|
||
parent.node,
|
||
Math.min(
|
||
requiredNumber(lowlink, parent.node, 'lowlink'),
|
||
requiredNumber(lowlink, frame.node, 'lowlink'),
|
||
),
|
||
);
|
||
}
|
||
}
|
||
continue;
|
||
}
|
||
|
||
const child = nextChild.value;
|
||
if (!index.has(child)) {
|
||
iterStack.push({
|
||
node: child,
|
||
children: (graph.get(child) ?? new Set<string>()).values(),
|
||
entered: false,
|
||
});
|
||
} else if (onStack.has(child)) {
|
||
lowlink.set(
|
||
frame.node,
|
||
Math.min(
|
||
requiredNumber(lowlink, frame.node, 'lowlink'),
|
||
requiredNumber(index, child, 'index'),
|
||
),
|
||
);
|
||
}
|
||
}
|
||
}
|
||
|
||
return sccs;
|
||
}
|
||
|
||
function requiredNumber(map: ReadonlyMap<string, number>, key: string, label: string): number {
|
||
const value = map.get(key);
|
||
if (value === undefined) {
|
||
throw new Error(`Invariant violated: missing Tarjan ${label} for ${key}`);
|
||
}
|
||
return value;
|
||
}
|