context.ts

/**
 * PipelineContext — shared mutable state threaded through all build stages.
 *
 * Each stage reads what it needs and writes what it produces.
 * This replaces the closure-captured locals in the old monolithic buildGraph().
 */
import type {
  BetterSqlite3Database,
  BuildGraphOpts,
  CodegraphConfig,
  EngineOpts,
  ExtractorOutput,
  FileToParse,
  MetadataUpdate,
  NativeDatabase,
  NodeRow,
  ParseChange,
  PathAliases,
} from '../../../types.js';
import type { BarrelExportResolution } from './stages/resolve-imports.js';

export class PipelineContext {
  // ── Inputs (set during setup) ──────────────────────────────────────
  rootDir!: string;
  db!: BetterSqlite3Database;
  dbPath!: string;
  config!: CodegraphConfig;
  opts!: BuildGraphOpts;
  engineOpts!: EngineOpts;
  engineName!: 'native' | 'wasm';
  engineVersion!: string | null;
  /**
   * The version reported by the native binary itself (CARGO_PKG_VERSION at
   * build time), as opposed to `engineVersion` which prefers the platform
   * package.json. The Rust orchestrator's check_version_mismatch compares
   * `build_meta.engine_version` against CARGO_PKG_VERSION, so build_meta
   * writes must use this value to avoid a perpetual full-rebuild loop when
   * the binary and platform package.json drift apart (e.g., CI hot-swap
   * via ci-install-native.mjs — #1066).
   */
  nativeBinaryVersion!: string | null;
  aliases!: PathAliases;
  incremental!: boolean;
  forceFullRebuild: boolean = false;
  schemaVersion!: number;
  nativeDb?: NativeDatabase;
  /** Whether native engine is available (deferred — DB opened only when needed). */
  nativeAvailable: boolean = false;
  /** True when ctx.db is a NativeDbProxy — single rusqlite connection for the entire pipeline. */
  nativeFirstProxy: boolean = false;

  // ── File collection (set by collectFiles stage) ────────────────────
  allFiles!: string[];
  discoveredDirs!: Set<string>;

  // ── Change detection (set by detectChanges stage) ──────────────────
  isFullBuild!: boolean;
  parseChanges!: ParseChange[];
  metadataUpdates!: MetadataUpdate[];
  removed!: string[];
  /**
   * Forward+reverse import-neighbor files of `removed`, captured before
   * `purgeFilesFromGraph`/`purgeFilesData` deletes those files' edges. Lets
   * `refreshAffectedDirectoryMetrics` still discover a removed file's
   * cross-directory neighbor even though the live edge evidence for it is
   * gone by the time the structure stage runs (#1839).
   */
  removedFileNeighbors: string[] = [];
  earlyExit: boolean = false;

  // ── Parsing (set by parseFiles stage) ──────────────────────────────
  allSymbols!: Map<string, ExtractorOutput>;
  fileSymbols!: Map<string, ExtractorOutput>;
  filesToParse!: FileToParse[];

  // ── Import resolution (set by resolveImports stage) ────────────────
  batchResolved!: Map<string, string> | null;
  reexportMap!: Map<string, unknown[]>;
  barrelOnlyFiles!: Set<string>;
  /** Phase 8.4: cache for resolveBarrelExport results keyed as "barrelPath|symbolName". */
  barrelExportCache: Map<string, BarrelExportResolution | null> = new Map();

  // ── Node lookup (set by insertNodes / buildEdges stages) ───────────
  nodesByName!: Map<string, NodeRow[]>;
  nodesByNameAndFile!: Map<string, NodeRow[]>;

  // ── Reverse-dep edge reconnection (set by detectChanges) ───────────
  /**
   * Edges from reverse-dep files to changed files, saved before purge so they
   * can be reconnected to new node IDs after insertNodes (#932, #933).
   * Eliminates the need to reparse reverse-dep files entirely.
   */
  savedReverseDepEdges: Array<{
    sourceId: number;
    tgtName: string;
    tgtKind: string;
    tgtFile: string;
    tgtLine: number;
    /**
     * 1-based rank of the target (by ascending line) among nodes sharing its
     * (name, kind) within `tgtFile`, computed at save time. Lets
     * `reconnectReverseDepEdges` map an old target to its correct new node
     * even when multiple distinct symbols in the file share the same name
     * and kind (e.g. several object-literal `close() {}` methods) — see #1752.
     */
    tgtOrdinal: number;
    /** Size of that (name, kind) sibling group at save time. */
    tgtSiblingCount: number;
    edgeKind: string;
    confidence: number;
    dynamic: number;
    technique: string | null;
    dynamicKind: string | null;
  }> = [];

  // ── Misc state ─────────────────────────────────────────────────────
  hasEmbeddings: boolean = false;
  lineCountMap!: Map<string, number>;

  // ── Phase timing ───────────────────────────────────────────────────
  timing: {
    setupMs?: number;
    collectMs?: number;
    detectMs?: number;
    parseMs?: number;
    insertMs?: number;
    resolveMs?: number;
    edgesMs?: number;
    structureMs?: number;
    rolesMs?: number;
    astMs?: number;
    complexityMs?: number;
    cfgMs?: number;
    dataflowMs?: number;
    finalizeMs?: number;
    [key: string]: number | undefined;
  } = {};
  buildStart!: number;
}