@zzzen/pyright-internal 1.2.0-dev.20260913 → 1.2.0-dev.20260920

package/dist/trace/cli.js

@@ -0,0 +1,196 @@
+"use strict";
+/*
+ * cli.ts
+ * Copyright (c) Microsoft Corporation.
+ * Licensed under the MIT license.
+ */
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.main = main;
+const fs_1 = require("fs");
+const capture_1 = require("./capture");
+const reader_1 = require("./reader");
+const viewerServer_1 = require("./viewerServer");
+const help = `Pyright evaluation traces (developer-only, local unbundled builds)
+  capture --target <repo> --input <python> --output <new.ndjson>
+          [--runtime <unbundled src>] [--experimental]
+          [--mode trace|audit|clean] [--file <python> --start <offset> --end <offset>]
+          [--operation <native operation id>] [--events <n>] [--bytes <n>] [--depth <n>]
+  compare <same capture options>  (writes clean/audit JSON alongside the trace)
+  query <artifact.ndjson> timeline [--kind <kind>] [--span <id>]
+  query <artifact.ndjson> history|latest --entity <id> [--at <sequence>]
+  query <artifact.ndjson> neighbors --entity <id> [--relation <relation>]
+  query <artifact.ndjson> pairs
+  query <artifact.ndjson> diff --baseline <span> --replay <span>
+  query <artifact.ndjson> summary|coverage
+  view <artifact.ndjson> [--port <0..65535>]  (loopback-only; Ctrl+C stops)
+All offsets are zero-based UTF-16 offsets. Output paths must not already exist.
+An absent event is not proof of no effect; inspect coverage and the footer.`;
+async function main(argv = process.argv.slice(2)) {
+    if (!argv.length || argv.includes('--help')) {
+        console.log(help);
+        return;
+    }
+    const positionals = [];
+    const flags = new Map();
+    const allowed = new Set([
+        'target',
+        'input',
+        'output',
+        'runtime',
+        'experimental',
+        'mode',
+        'file',
+        'start',
+        'end',
+        'operation',
+        'events',
+        'bytes',
+        'depth',
+        'kind',
+        'span',
+        'entity',
+        'at',
+        'relation',
+        'baseline',
+        'replay',
+        'port',
+    ]);
+    for (let i = 0; i < argv.length; i++) {
+        const token = argv[i];
+        if (!token.startsWith('--')) {
+            positionals.push(token);
+            continue;
+        }
+        const name = token.slice(2);
+        if (!allowed.has(name) || flags.has(name)) {
+            throw new Error(`Unknown or repeated option ${token}`);
+        }
+        if (name === 'experimental') {
+            flags.set(name, 'true');
+        }
+        else {
+            const value = argv[++i];
+            if (!value || value.startsWith('--')) {
+                throw new Error(`Missing value for ${token}`);
+            }
+            flags.set(name, value);
+        }
+    }
+    const required = (name) => {
+        const value = flags.get(name);
+        if (!value) {
+            throw new Error(`--${name} is required`);
+        }
+        return value;
+    };
+    const number = (name) => {
+        const value = flags.get(name);
+        if (value === undefined) {
+            return undefined;
+        }
+        const parsed = Number(value);
+        if (!Number.isSafeInteger(parsed) || parsed < 0) {
+            throw new Error(`Invalid --${name}: ${value}`);
+        }
+        return parsed;
+    };
+    const [command, path, query] = positionals;
+    if (command === 'view') {
+        if (!path || positionals.length !== 2 || [...flags.keys()].some((flag) => flag !== 'port')) {
+            throw new Error('Usage: view <artifact.ndjson> [--port <0..65535>]');
+        }
+        const viewer = await (0, viewerServer_1.startViewer)((0, viewerServer_1.loadViewerModel)(path), number('port'));
+        console.log(`Pyright trace viewer: ${viewer.url}\nPID ${process.pid}. Local read-only artifact; Ctrl+C to stop.`);
+        let closing = false;
+        const stop = () => {
+            if (!closing) {
+                closing = true;
+                viewer
+                    .close()
+                    .catch((error) => {
+                    console.error('Viewer shutdown failed:', error);
+                    process.exitCode = 1;
+                })
+                    .finally(() => {
+                    process.removeListener('SIGINT', stop);
+                    process.removeListener('SIGTERM', stop);
+                });
+            }
+        };
+        process.on('SIGINT', stop);
+        process.on('SIGTERM', stop);
+        return;
+    }
+    if (command === 'capture' || command === 'compare') {
+        const mode = flags.get('mode') ?? 'trace';
+        if (!['trace', 'audit', 'clean'].includes(mode)) {
+            throw new Error(`Invalid mode ${mode}`);
+        }
+        const options = {
+            target: required('target'),
+            input: required('input'),
+            output: required('output'),
+            runtime: flags.get('runtime'),
+            mode: mode,
+            experimental: flags.has('experimental'),
+            selection: {
+                file: flags.get('file') ?? required('input'),
+                start: number('start'),
+                end: number('end'),
+                operation: number('operation'),
+            },
+            limits: Object.fromEntries(['events', 'bytes', 'depth'].filter((name) => flags.has(name)).map((name) => [name, number(name)])),
+        };
+        if (command === 'compare') {
+            console.log(JSON.stringify((0, capture_1.compareCaptures)(options), null, 2));
+            return;
+        }
+        const result = (0, capture_1.capture)(options);
+        console.log(JSON.stringify({ mode, output: options.output, audit: result.audit, diagnostics: result.diagnostics }));
+        return;
+    }
+    if (command !== 'query' || !path) {
+        throw new Error(help);
+    }
+    if ((0, fs_1.statSync)(path).size > 128 * 1024 * 1024) {
+        throw new Error('Artifact exceeds reader byte limit');
+    }
+    const reader = new reader_1.TraceReader((0, reader_1.readTrace)((0, fs_1.readFileSync)(path, 'utf8')));
+    let result;
+    switch (query) {
+        case 'timeline':
+            result = reader.timeline(flags.get('kind'), flags.get('span'));
+            break;
+        case 'history':
+            result = reader.history(required('entity'), number('at'));
+            break;
+        case 'latest':
+            result = reader.latest(required('entity'), number('at'));
+            break;
+        case 'neighbors':
+            result = reader.neighborhood(required('entity'), flags.get('relation'));
+            break;
+        case 'pairs':
+            result = reader.pairs();
+            break;
+        case 'diff':
+            result = reader.diff(required('baseline'), required('replay'));
+            break;
+        case 'coverage':
+            result = [reader.records[0], reader.records[reader.records.length - 1]];
+            break;
+        case 'summary':
+            result = reader.records.filter((r) => r.kind === 'summary');
+            break;
+        default:
+            throw new Error(`Unknown query ${query}`);
+    }
+    console.log(JSON.stringify(result, null, 2));
+}
+if (require.main === module) {
+    main().catch((error) => {
+        console.error(error instanceof Error ? error.message : error);
+        process.exitCode = 1;
+    });
+}
+//# sourceMappingURL=cli.js.map

package/dist/trace/cli.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"cli.js","sourceRoot":"","sources":["../../../../../src/trace/cli.ts"],"names":[],"mappings":";AAAA;;;;GAIG;;AAwBH,oBAsKC;AA5LD,2BAA4C;AAE5C,uCAAqE;AACrE,qCAAkD;AAClD,iDAA8D;AAE9D,MAAM,IAAI,GAAG;;;;;;;;;;;;;;4EAc+D,CAAC;AAEtE,KAAK,UAAU,IAAI,CAAC,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC;IACnD,IAAI,CAAC,IAAI,CAAC,MAAM,IAAI,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAE,CAAC;QAC1C,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;QAClB,OAAO;IACX,CAAC;IACD,MAAM,WAAW,GAAa,EAAE,CAAC;IACjC,MAAM,KAAK,GAAG,IAAI,GAAG,EAAkB,CAAC;IACxC,MAAM,OAAO,GAAG,IAAI,GAAG,CAAC;QACpB,QAAQ;QACR,OAAO;QACP,QAAQ;QACR,SAAS;QACT,cAAc;QACd,MAAM;QACN,MAAM;QACN,OAAO;QACP,KAAK;QACL,WAAW;QACX,QAAQ;QACR,OAAO;QACP,OAAO;QACP,MAAM;QACN,MAAM;QACN,QAAQ;QACR,IAAI;QACJ,UAAU;QACV,UAAU;QACV,QAAQ;QACR,MAAM;KACT,CAAC,CAAC;IACH,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;QACnC,MAAM,KAAK,GAAG,IAAI,CAAC,CAAC,CAAC,CAAC;QACtB,IAAI,CAAC,KAAK,CAAC,UAAU,CAAC,IAAI,CAAC,EAAE,CAAC;YAC1B,WAAW,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YACxB,SAAS;QACb,CAAC;QACD,MAAM,IAAI,GAAG,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;QAC5B,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC;YACxC,MAAM,IAAI,KAAK,CAAC,8BAA8B,KAAK,EAAE,CAAC,CAAC;QAC3D,CAAC;QACD,IAAI,IAAI,KAAK,cAAc,EAAE,CAAC;YAC1B,KAAK,CAAC,GAAG,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;QAC5B,CAAC;aAAM,CAAC;YACJ,MAAM,KAAK,GAAG,IAAI,CAAC,EAAE,CAAC,CAAC,CAAC;YACxB,IAAI,CAAC,KAAK,IAAI,KAAK,CAAC,UAAU,CAAC,IAAI,CAAC,EAAE,CAAC;gBACnC,MAAM,IAAI,KAAK,CAAC,qBAAqB,KAAK,EAAE,CAAC,CAAC;YAClD,CAAC;YACD,KAAK,CAAC,GAAG,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;QAC3B,CAAC;IACL,CAAC;IACD,MAAM,QAAQ,GAAG,CAAC,IAAY,EAAE,EAAE;QAC9B,MAAM,KAAK,GAAG,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;QAC9B,IAAI,CAAC,KAAK,EAAE,CAAC;YACT,MAAM,IAAI,KAAK,CAAC,KAAK,IAAI,cAAc,CAAC,CAAC;QAC7C,CAAC;QACD,OAAO,KAAK,CAAC;IACjB,CAAC,CAAC;IACF,MAAM,MAAM,GAAG,CAAC,IAAY,EAAsB,EAAE;QAChD,MAAM,KAAK,GAAG,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;QAC9B,IAAI,KAAK,KAAK,SAAS,EAAE,CAAC;YACtB,OAAO,SAAS,CAAC;QACrB,CAAC;QACD,MAAM,MAAM,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC;QAC7B,IAAI,CAAC,MAAM,CAAC,aAAa,CAAC,MAAM,CAAC,IAAI,MAAM,GAAG,CAAC,EAAE,CAAC;YAC9C,MAAM,IAAI,KAAK,CAAC,aAAa,IAAI,KAAK,KAAK,EAAE,CAAC,CAAC;QACnD,CAAC;QACD,OAAO,MAAM,CAAC;IAClB,CAAC,CAAC;IACF,MAAM,CAAC,OAAO,EAAE,IAAI,EAAE,KAAK,CAAC,GAAG,WAAW,CAAC;IAC3C,IAAI,OAAO,KAAK,MAAM,EAAE,CAAC;QACrB,IAAI,CAAC,IAAI,IAAI,WAAW,CAAC,MAAM,KAAK,CAAC,IAAI,CAAC,GAAG,KAAK,CAAC,IAAI,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,KAAK,MAAM,CAAC,EAAE,CAAC;YACzF,MAAM,IAAI,KAAK,CAAC,mDAAmD,CAAC,CAAC;QACzE,CAAC;QACD,MAAM,MAAM,GAAG,MAAM,IAAA,0BAAW,EAAC,IAAA,8BAAe,EAAC,IAAI,CAAC,EAAE,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC;QACxE,OAAO,CAAC,GAAG,CACP,yBAAyB,MAAM,CAAC,GAAG,SAAS,OAAO,CAAC,GAAG,6CAA6C,CACvG,CAAC;QACF,IAAI,OAAO,GAAG,KAAK,CAAC;QACpB,MAAM,IAAI,GAAG,GAAG,EAAE;YACd,IAAI,CAAC,OAAO,EAAE,CAAC;gBACX,OAAO,GAAG,IAAI,CAAC;gBACf,MAAM;qBACD,KAAK,EAAE;qBACP,KAAK,CAAC,CAAC,KAAK,EAAE,EAAE;oBACb,OAAO,CAAC,KAAK,CAAC,yBAAyB,EAAE,KAAK,CAAC,CAAC;oBAChD,OAAO,CAAC,QAAQ,GAAG,CAAC,CAAC;gBACzB,CAAC,CAAC;qBACD,OAAO,CAAC,GAAG,EAAE;oBACV,OAAO,CAAC,cAAc,CAAC,QAAQ,EAAE,IAAI,CAAC,CAAC;oBACvC,OAAO,CAAC,cAAc,CAAC,SAAS,EAAE,IAAI,CAAC,CAAC;gBAC5C,CAAC,CAAC,CAAC;YACX,CAAC;QACL,CAAC,CAAC;QACF,OAAO,CAAC,EAAE,CAAC,QAAQ,EAAE,IAAI,CAAC,CAAC;QAC3B,OAAO,CAAC,EAAE,CAAC,SAAS,EAAE,IAAI,CAAC,CAAC;QAC5B,OAAO;IACX,CAAC;IACD,IAAI,OAAO,KAAK,SAAS,IAAI,OAAO,KAAK,SAAS,EAAE,CAAC;QACjD,MAAM,IAAI,GAAG,KAAK,CAAC,GAAG,CAAC,MAAM,CAAC,IAAI,OAAO,CAAC;QAC1C,IAAI,CAAC,CAAC,OAAO,EAAE,OAAO,EAAE,OAAO,CAAC,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE,CAAC;YAC9C,MAAM,IAAI,KAAK,CAAC,gBAAgB,IAAI,EAAE,CAAC,CAAC;QAC5C,CAAC;QACD,MAAM,OAAO,GAAmB;YAC5B,MAAM,EAAE,QAAQ,CAAC,QAAQ,CAAC;YAC1B,KAAK,EAAE,QAAQ,CAAC,OAAO,CAAC;YACxB,MAAM,EAAE,QAAQ,CAAC,QAAQ,CAAC;YAC1B,OAAO,EAAE,KAAK,CAAC,GAAG,CAAC,SAAS,CAAC;YAC7B,IAAI,EAAE,IAA8B;YACpC,YAAY,EAAE,KAAK,CAAC,GAAG,CAAC,cAAc,CAAC;YACvC,SAAS,EAAE;gBACP,IAAI,EAAE,KAAK,CAAC,GAAG,CAAC,MAAM,CAAC,IAAI,QAAQ,CAAC,OAAO,CAAC;gBAC5C,KAAK,EAAE,MAAM,CAAC,OAAO,CAAC;gBACtB,GAAG,EAAE,MAAM,CAAC,KAAK,CAAC;gBAClB,SAAS,EAAE,MAAM,CAAC,WAAW,CAAC;aACjC;YACD,MAAM,EAAE,MAAM,CAAC,WAAW,CACtB,CAAC,QAAQ,EAAE,OAAO,EAAE,OAAO,CAAC,CAAC,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,IAAI,EAAE,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CACrG;SACJ,CAAC;QACF,IAAI,OAAO,KAAK,SAAS,EAAE,CAAC;YACxB,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,SAAS,CAAC,IAAA,yBAAe,EAAC,OAAO,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC;YAC/D,OAAO;QACX,CAAC;QACD,MAAM,MAAM,GAAG,IAAA,iBAAO,EAAC,OAAO,CAAC,CAAC;QAChC,OAAO,CAAC,GAAG,CACP,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,CAAC,MAAM,EAAE,KAAK,EAAE,MAAM,CAAC,KAAK,EAAE,WAAW,EAAE,MAAM,CAAC,WAAW,EAAE,CAAC,CACzG,CAAC;QACF,OAAO;IACX,CAAC;IACD,IAAI,OAAO,KAAK,OAAO,IAAI,CAAC,IAAI,EAAE,CAAC;QAC/B,MAAM,IAAI,KAAK,CAAC,IAAI,CAAC,CAAC;IAC1B,CAAC;IACD,IAAI,IAAA,aAAQ,EAAC,IAAI,CAAC,CAAC,IAAI,GAAG,GAAG,GAAG,IAAI,GAAG,IAAI,EAAE,CAAC;QAC1C,MAAM,IAAI,KAAK,CAAC,oCAAoC,CAAC,CAAC;IAC1D,CAAC;IACD,MAAM,MAAM,GAAG,IAAI,oBAAW,CAAC,IAAA,kBAAS,EAAC,IAAA,iBAAY,EAAC,IAAI,EAAE,MAAM,CAAC,CAAC,CAAC,CAAC;IACtE,IAAI,MAAe,CAAC;IACpB,QAAQ,KAAK,EAAE,CAAC;QACZ,KAAK,UAAU;YACX,MAAM,GAAG,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,KAAK,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC;YAC/D,MAAM;QACV,KAAK,SAAS;YACV,MAAM,GAAG,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAE,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC;YAC1D,MAAM;QACV,KAAK,QAAQ;YACT,MAAM,GAAG,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAE,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC;YACzD,MAAM;QACV,KAAK,WAAW;YACZ,MAAM,GAAG,MAAM,CAAC,YAAY,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAE,KAAK,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC,CAAC;YACxE,MAAM;QACV,KAAK,OAAO;YACR,MAAM,GAAG,MAAM,CAAC,KAAK,EAAE,CAAC;YACxB,MAAM;QACV,KAAK,MAAM;YACP,MAAM,GAAG,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,UAAU,CAAC,EAAE,QAAQ,CAAC,QAAQ,CAAC,CAAC,CAAC;YAC/D,MAAM;QACV,KAAK,UAAU;YACX,MAAM,GAAG,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC;YACxE,MAAM;QACV,KAAK,SAAS;YACV,MAAM,GAAG,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,SAAS,CAAC,CAAC;YAC5D,MAAM;QACV;YACI,MAAM,IAAI,KAAK,CAAC,iBAAiB,KAAK,EAAE,CAAC,CAAC;IAClD,CAAC;IACD,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,SAAS,CAAC,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC;AACjD,CAAC;AAED,IAAI,OAAO,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;IAC1B,IAAI,EAAE,CAAC,KAAK,CAAC,CAAC,KAAK,EAAE,EAAE;QACnB,OAAO,CAAC,KAAK,CAAC,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC;QAC9D,OAAO,CAAC,QAAQ,GAAG,CAAC,CAAC;IACzB,CAAC,CAAC,CAAC;AACP,CAAC"}

package/dist/trace/nativeAdapter.d.ts

@@ -0,0 +1,16 @@
+export interface HookSite {
+    id: string;
+    file: string;
+    label: string;
+    line: number;
+    start: number;
+    length: number;
+    text: string;
+}
+export declare const adapterVersion = "native-commonjs-v1";
+export declare function adapterFiles(experimental: boolean): string[];
+export declare function instrument(file: string, code: string, experimental?: boolean): {
+    code: string;
+    sites: HookSite[];
+    counts: Record<string, number>;
+};

package/dist/trace/nativeAdapter.js

@@ -0,0 +1,422 @@
+"use strict";
+/*
+ * nativeAdapter.ts
+ * Copyright (c) Microsoft Corporation.
+ * Licensed under the MIT license.
+ *
+ * AST instrumentation for supported unbundled CommonJS builds. Hook locations
+ * are derived from syntax, never byte/line replacements. No files are changed.
+ */
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __importStar = (this && this.__importStar) || (function () {
+    var ownKeys = function(o) {
+        ownKeys = Object.getOwnPropertyNames || function (o) {
+            var ar = [];
+            for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
+            return ar;
+        };
+        return ownKeys(o);
+    };
+    return function (mod) {
+        if (mod && mod.__esModule) return mod;
+        var result = {};
+        if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
+        __setModuleDefault(result, mod);
+        return result;
+    };
+})();
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.adapterVersion = void 0;
+exports.adapterFiles = adapterFiles;
+exports.instrument = instrument;
+const ts = __importStar(require("typescript"));
+exports.adapterVersion = 'native-commonjs-v1';
+const modules = {
+    'analyzer/typeEvaluator.js': {
+        factory: 'createTypeEvaluator',
+        state: [
+            'typeCache',
+            'expectedTypeCache',
+            'typeFormTypeCache',
+            'incompleteGenCount',
+            'codeFlowAnalyzerCache',
+            'returnTypeInferenceTypeCache',
+            'returnTypeInferenceContextStack',
+            'suppressedNodeStack',
+            'signatureTrackerStack',
+            'symbolResolutionStack',
+            'deferredClassCompletions',
+            'speculativeTypeTracker',
+        ],
+        functions: [
+            'getTypeOfExpression',
+            'readTypeCacheEntry',
+            'readTypeFormTypeCacheEntry',
+            'isTypeCacheEntryValid',
+            'readTypeCache',
+            'writeTypeCache',
+            'addExpectedTypeCacheEntry',
+            'validateCallArgs',
+            'validateArgType',
+            'inferFunctionReturnType',
+            'disposeEvaluator',
+        ],
+        required: ['collection', 'branch', 'generation', 'field-write'],
+    },
+    'analyzer/codeFlowEngine.js': {
+        factory: 'getCodeFlowEngine',
+        state: ['flowIncompleteGeneration', 'reachabilityCache', 'callIsNoReturnCache', 'isReachableRecursionSet'],
+        functions: [
+            'createCodeFlowAnalyzer',
+            'getTypeFromCodeFlow',
+            'getTypeFromFlowNode',
+            'setCacheEntry',
+            'setIncompleteSubtype',
+            'getCacheEntry',
+            'preventRecursion',
+            'getFlowNodeReachability',
+            'isCallNoReturn',
+        ],
+        required: ['collection', 'branch', 'generation', 'field-write'],
+    },
+    'analyzer/typeCacheUtils.js': {
+        optionalFunctions: ['findContextualTypeCacheEntry'],
+        functions: [
+            'contextualTypeCacheEntryMatches',
+            'addContextualTypeCacheEntry',
+            'enterSpeculativeContext',
+            'leaveSpeculativeContext',
+            'trackEntry',
+            'disableSpeculativeMode',
+            'enableSpeculativeMode',
+            'getSpeculativeType',
+            'addSpeculativeType',
+            '_dependentTypesMatch',
+        ],
+        required: ['collection', 'branch'],
+    },
+    'analyzer/constraintTracker.js': {
+        functions: ['ConstraintSet.setBounds', 'ConstraintSet.getTypeVar', 'ConstraintTracker.setBounds'],
+        required: ['collection'],
+    },
+    'common/diagnosticSink.js': {
+        functions: ['fetchAndClear', 'addDiagnostic', 'addDiagnostics'],
+        required: ['collection'],
+    },
+    'analyzer/analyzerNodeInfo.js': {
+        functions: ['AnalyzerNodeInfoStore.setFileInfo'],
+        required: [],
+    },
+    'analyzer/overloadResultController.js': {
+        functions: ['_trial', 'evaluate', 'evaluateAndContinue', 'beforeCall'],
+        required: ['collection', 'branch', 'evaluator-link'],
+    },
+};
+function adapterFiles(experimental) {
+    return Object.keys(modules).filter((file) => experimental || !file.includes('overloadResultController'));
+}
+const bridgeName = '__pyrightTrace';
+function parseExpression(text) {
+    const file = ts.createSourceFile('hook.js', `(${text});`, ts.ScriptTarget.ES2020, false, ts.ScriptKind.JS);
+    return synthesize(file.statements[0].expression);
+}
+function parseStatements(text) {
+    return ts
+        .createSourceFile('hook.js', text, ts.ScriptTarget.ES2020, false, ts.ScriptKind.JS)
+        .statements.map(synthesize);
+}
+function synthesize(node) {
+    ts.setTextRange(node, { pos: -1, end: -1 });
+    ts.forEachChild(node, (child) => {
+        synthesize(child);
+    });
+    return node;
+}
+function instrument(file, code, experimental = false) {
+    let spec = modules[file];
+    if (!spec) {
+        throw new Error(`No adapter for ${file}`);
+    }
+    if (experimental && file === 'analyzer/typeCacheUtils.js') {
+        spec = {
+            ...spec,
+            functions: [
+                ...spec.functions,
+                'useTestOnlyCacheIsolation',
+                'SpeculativeTypeTracker.useTestOnlyCacheIsolation',
+            ],
+        };
+    }
+    const source = ts.createSourceFile(file, code, ts.ScriptTarget.ES2020, true, ts.ScriptKind.JS);
+    const sites = [];
+    const counts = {};
+    const matches = new Map([...spec.functions, ...(spec.optionalFunctions ?? [])].map((name) => [name, 0]));
+    const declarations = new Set();
+    let collisions = false;
+    const scan = (node) => {
+        if (ts.isIdentifier(node)) {
+            if (node.text.startsWith('__ptr') || node.text === bridgeName) {
+                collisions = true;
+            }
+        }
+        if (ts.isVariableDeclaration(node) && ts.isIdentifier(node.name)) {
+            declarations.add(node.name.text);
+        }
+        ts.forEachChild(node, scan);
+    };
+    scan(source);
+    if (collisions) {
+        throw new Error(`Adapter identifier collision in ${file}`);
+    }
+    for (const name of spec.state ?? []) {
+        if (!declarations.has(name)) {
+            throw new Error(`Adapter drift: missing state ${file}:${name}`);
+        }
+    }
+    const site = (node, label, kind) => {
+        counts[kind] = (counts[kind] ?? 0) + 1;
+        const result = {
+            id: `${file}:${sites.length + 1}`,
+            file,
+            label,
+            start: node.getStart(source),
+            length: node.getWidth(source),
+            line: source.getLineAndCharacterOfPosition(node.getStart(source)).line + 1,
+            text: node.getText(source).slice(0, 1000),
+        };
+        sites.push(result);
+        return ts.factory.createStringLiteral(result.id);
+    };
+    const call = (name, args) => ts.factory.createCallExpression(ts.factory.createPropertyAccessExpression(ts.factory.createIdentifier(bridgeName), name), undefined, args);
+    let currentFunction = '';
+    let currentClass = '';
+    let selected = false;
+    let inFactory = false;
+    let currentOwner = 'undefined';
+    let parameters = [];
+    let factoryCount = 0;
+    const transform = (context) => {
+        const visit = (node) => {
+            if (ts.isClassDeclaration(node) || ts.isClassExpression(node)) {
+                const saved = currentClass;
+                currentClass = node.name?.text ?? '';
+                const updated = ts.visitEachChild(node, visit, context);
+                currentClass = saved;
+                return updated;
+            }
+            if (ts.isFunctionDeclaration(node) ||
+                ts.isFunctionExpression(node) ||
+                ts.isArrowFunction(node) ||
+                ts.isMethodDeclaration(node)) {
+                const name = node.name && ts.isIdentifier(node.name)
+                    ? node.name.text
+                    : ts.isVariableDeclaration(node.parent) && ts.isIdentifier(node.parent.name)
+                        ? node.parent.name.text
+                        : ts.isBinaryExpression(node.parent) && ts.isPropertyAccessExpression(node.parent.left)
+                            ? node.parent.left.name.text
+                            : '';
+                const qualified = `${currentClass}.${name}`;
+                const key = matches.has(qualified) ? qualified : name;
+                const isSelected = matches.has(key);
+                const isFlowAnalyzer = file === 'analyzer/codeFlowEngine.js' && name === 'createCodeFlowAnalyzer';
+                const isFactory = name === spec.factory || isFlowAnalyzer;
+                if (isSelected) {
+                    matches.set(key, matches.get(key) + 1);
+                }
+                if (name === spec.factory) {
+                    factoryCount++;
+                }
+                if ((isSelected || isFactory) && (!node.body || !ts.isBlock(node.body))) {
+                    throw new Error(`Adapter drift: expected block body ${file}:${name}`);
+                }
+                if (isFactory && node.body && ts.isBlock(node.body)) {
+                    const declared = new Set(node.body.statements.flatMap((s) => ts.isVariableStatement(s)
+                        ? s.declarationList.declarations.flatMap((d) => ts.isIdentifier(d.name) ? [d.name.text] : [])
+                        : []));
+                    for (const state of isFlowAnalyzer ? ['flowNodeTypeCacheSet'] : spec.state) {
+                        if (!declared.has(state)) {
+                            throw new Error(`Adapter drift: ${name} does not directly declare state ${state}`);
+                        }
+                    }
+                }
+                if (isSelected &&
+                    (node.parameters.some((p) => !ts.isIdentifier(p.name) || !!p.dotDotDotToken) ||
+                        node.modifiers?.some((m) => m.kind === ts.SyntaxKind.AsyncKeyword) ||
+                        ('asteriskToken' in node && node.asteriskToken))) {
+                    throw new Error(`Unsupported asynchronous/generator/destructured hook ${file}:${name}`);
+                }
+                const previous = { currentFunction, selected, inFactory, currentOwner, parameters };
+                currentFunction = name;
+                selected = isSelected;
+                inFactory = isFactory || inFactory;
+                parameters = node.parameters.map((p) => p.name.getText(source));
+                if (isFactory) {
+                    currentOwner = `__ptrOwner_${name}`;
+                }
+                let updated = ts.visitEachChild(node, visit, context);
+                if ((isSelected || isFactory) && updated.body && ts.isBlock(updated.body)) {
+                    let statements = [...updated.body.statements];
+                    if (isFactory) {
+                        const state = isFlowAnalyzer ? ['flowNodeTypeCacheSet'] : spec.state;
+                        const parent = name === 'getCodeFlowEngine' ? 'evaluator' : previous.currentOwner;
+                        statements.unshift(...parseStatements(`${bridgeName}.owner(${currentOwner}, ${JSON.stringify(name)}, () => ({${state.join(',')}}), ${parent});`));
+                    }
+                    if (isSelected) {
+                        const hook = site(node, key, 'function');
+                        const args = parameters;
+                        const owner = currentOwner;
+                        const receiver = ts.isMethodDeclaration(node) ? 'this' : 'undefined';
+                        const prefix = parseStatements(`const __ptrSpan = ${bridgeName}.enter(${JSON.stringify(hook.text)}, ${owner}, ` +
+                            `{${args.join(',')}}, ${receiver});`);
+                        const originalBody = ts.factory.createBlock(statements, true);
+                        const caught = ts.factory.createCatchClause('__ptrError', ts.factory.createBlock([
+                            ts.factory.createExpressionStatement(call('thrown', [
+                                ts.factory.createIdentifier('__ptrSpan'),
+                                ts.factory.createIdentifier('__ptrError'),
+                            ])),
+                            ts.factory.createThrowStatement(ts.factory.createIdentifier('__ptrError')),
+                        ], true));
+                        const final = ts.factory.createBlock([
+                            ts.factory.createExpressionStatement(call('exit', [ts.factory.createIdentifier('__ptrSpan')])),
+                        ], true);
+                        statements = [...prefix, ts.factory.createTryStatement(originalBody, caught, final)];
+                    }
+                    if (isFactory) {
+                        statements.unshift(...parseStatements(`const ${currentOwner} = {};`));
+                    }
+                    const body = ts.factory.updateBlock(updated.body, statements);
+                    if (ts.isFunctionDeclaration(updated)) {
+                        updated = ts.factory.updateFunctionDeclaration(updated, updated.modifiers, updated.asteriskToken, updated.name, updated.typeParameters, updated.parameters, updated.type, body);
+                    }
+                    else if (ts.isMethodDeclaration(updated)) {
+                        updated = ts.factory.updateMethodDeclaration(updated, updated.modifiers, updated.asteriskToken, updated.name, updated.questionToken, updated.typeParameters, updated.parameters, updated.type, body);
+                    }
+                    else if (ts.isArrowFunction(updated)) {
+                        updated = ts.factory.updateArrowFunction(updated, updated.modifiers, updated.typeParameters, updated.parameters, updated.type, updated.equalsGreaterThanToken, body);
+                    }
+                    else if (ts.isFunctionExpression(updated)) {
+                        updated = ts.factory.updateFunctionExpression(updated, updated.modifiers, updated.asteriskToken, updated.name, updated.typeParameters, updated.parameters, updated.type, body);
+                    }
+                }
+                ({ currentFunction, selected, inFactory, currentOwner, parameters } = previous);
+                return updated;
+            }
+            if (ts.isReturnStatement(node)) {
+                const value = node.expression
+                    ? ts.visitNode(node.expression, visit)
+                    : ts.factory.createIdentifier('undefined');
+                if (selected) {
+                    const result = currentFunction === 'createCodeFlowAnalyzer'
+                        ? call('bind', [ts.factory.createIdentifier(currentOwner), value])
+                        : value;
+                    return ts.factory.updateReturnStatement(node, call('returned', [
+                        ts.factory.createIdentifier('__ptrSpan'),
+                        result,
+                        parseExpression(`{${parameters.join(',')}}`),
+                        site(node, currentFunction, 'return'),
+                    ]));
+                }
+                if (currentFunction === spec.factory) {
+                    return ts.factory.updateReturnStatement(node, call('bind', [ts.factory.createIdentifier(currentOwner), value]));
+                }
+            }
+            if (currentFunction === '_trial' &&
+                ts.isVariableStatement(node) &&
+                node.declarationList.declarations.some((d) => ts.isIdentifier(d.name) && d.name.text === 'evaluator')) {
+                counts['evaluator-link'] = (counts['evaluator-link'] ?? 0) + 1;
+                return [
+                    ts.visitEachChild(node, visit, context),
+                    ...parseStatements(`${bridgeName}.linkCurrent(evaluator);`),
+                ];
+            }
+            if (selected && ts.isIfStatement(node)) {
+                return ts.factory.updateIfStatement(node, call('branch', [
+                    site(node.expression, currentFunction, 'branch'),
+                    ts.visitNode(node.expression, visit),
+                ]), ts.visitNode(node.thenStatement, visit), ts.visitNode(node.elseStatement, visit));
+            }
+            if (ts.isCallExpression(node) && ts.isPropertyAccessExpression(node.expression) && !node.questionDotToken) {
+                const receiver = node.expression.expression;
+                const method = node.expression.name.text;
+                const receiverText = receiver.getText(source);
+                // These names identify native state containers, not Python fixtures or
+                // selected source functions. The manifest reports every matched site.
+                if (/cache|stack|pendingNodes|closedFinallyGateNodes|incompleteEntries|candidates|_typeVarMap|_diagnostic|entriesToUndo|_activeDependentTypes|RecursionSet/i.test(receiverText) &&
+                    ['get', 'has', 'set', 'delete', 'clear', 'add', 'push', 'pop'].includes(method) &&
+                    !node.expression.questionDotToken) {
+                    return ts.factory.updateCallExpression(node, call('method', [
+                        site(node, currentFunction, 'collection'),
+                        ts.visitNode(receiver, visit),
+                        ts.factory.createStringLiteral(method),
+                    ]), undefined, node.arguments.map((arg) => ts.visitNode(arg, visit)));
+                }
+            }
+            if ((ts.isPostfixUnaryExpression(node) || ts.isPrefixUnaryExpression(node)) &&
+                ts.isIdentifier(node.operand) &&
+                ['incompleteGenCount', 'flowIncompleteGeneration'].includes(node.operand.text)) {
+                return call('effect', [
+                    site(node, node.operand.text, 'generation'),
+                    node,
+                    parseExpression(`{value:${node.operand.text}, domain:${JSON.stringify(node.operand.text)}}`),
+                ]);
+            }
+            if (ts.isBinaryExpression(node) &&
+                node.operatorToken.kind === ts.SyntaxKind.EqualsToken &&
+                ts.isPropertyAccessExpression(node.left) &&
+                (selected || node.left.name.text === 'inferredReturnType')) {
+                const receiver = node.left.expression;
+                if (ts.isIdentifier(receiver) ||
+                    receiver.kind === ts.SyntaxKind.ThisKeyword ||
+                    (ts.isPropertyAccessExpression(receiver) && ts.isIdentifier(receiver.expression))) {
+                    return call('effect', [
+                        site(node, currentFunction, 'field-write'),
+                        ts.visitEachChild(node, visit, context),
+                        ts.factory.createObjectLiteralExpression([
+                            ts.factory.createPropertyAssignment('target', receiver),
+                            ts.factory.createPropertyAssignment('field', ts.factory.createStringLiteral(node.left.name.text)),
+                        ]),
+                    ]);
+                }
+            }
+            return ts.visitEachChild(node, visit, context);
+        };
+        return (root) => ts.visitNode(root, visit);
+    };
+    const result = ts.transform(source, [transform]);
+    try {
+        for (const [name, count] of matches) {
+            counts[`function:${name}`] = count;
+            if (count !== 1 && !(count === 0 && spec.optionalFunctions?.includes(name))) {
+                throw new Error(`Adapter drift: expected one ${file}:${name}, found ${count}`);
+            }
+        }
+        if (spec.factory && factoryCount !== 1) {
+            throw new Error(`Adapter drift: expected one factory ${file}:${spec.factory}, found ${factoryCount}`);
+        }
+        for (const kind of spec.required) {
+            if (!counts[kind]) {
+                throw new Error(`Adapter drift: missing ${kind} in ${file}`);
+            }
+        }
+        return { code: ts.createPrinter().printFile(result.transformed[0]), sites, counts };
+    }
+    finally {
+        result.dispose();
+    }
+}
+//# sourceMappingURL=nativeAdapter.js.map