@mmnto/totem 0.21.0 → 0.22.0

package/dist/adversarial.test.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=adversarial.test.d.ts.map

package/dist/adversarial.test.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"adversarial.test.d.ts","sourceRoot":"","sources":["../src/adversarial.test.ts"],"names":[],"mappings":""}

package/dist/adversarial.test.js

@@ -0,0 +1,255 @@
+import { describe, expect, it } from 'vitest';
+import { applyRules, applyRulesToAdditions, } from './compiler.js';
+// ─── Adversarial Evaluation Harness ─────────────────
+//
+// Tests that compiled Totem rules catch planted bugs in realistic diffs.
+// This is the deterministic subset of #196 — no LLM calls, pure regex.
+// Each test simulates a developer making a known-bad change and verifies
+// the compiled rules catch it before it reaches production.
+const makeRule = (pattern, message, heading) => ({
+    lessonHash: 'adversarial',
+    lessonHeading: heading,
+    pattern,
+    message,
+    engine: 'regex',
+    compiledAt: new Date().toISOString(),
+});
+// ─── Rule set: common project traps ──────────────────
+const ADVERSARIAL_RULES = [
+    makeRule('catch\\s*\\(\\s*error\\s*[\\):]', 'Use err, not error, in catch blocks (project convention)', 'Use err not error'),
+    makeRule('\\bnpm\\s+(install|run|exec|ci)\\b', 'Use pnpm instead of npm (monorepo convention)', 'Never use npm'),
+    makeRule('\\bdebugger\\b', 'Remove debugger statements before commit', 'No debugger in production'),
+    makeRule('process\\.exit\\(0\\)', 'Do not call process.exit(0) — let the process end naturally', 'Avoid process.exit(0)'),
+    makeRule('\\.catch\\(\\(\\)\\s*=>\\s*\\{\\s*\\}\\)', 'Empty catch blocks swallow errors silently', 'No empty catch blocks'),
+    makeRule('TODO|FIXME|HACK|XXX', 'Resolve TODO/FIXME comments before merging', 'No TODO in production'),
+    makeRule('password\\s*[:=]\\s*["\'][^"\']+["\']', 'Hardcoded passwords detected — use environment variables', 'No hardcoded secrets'),
+    makeRule('\\bany\\b', 'Avoid TypeScript any — use unknown or proper types', 'No any types'),
+];
+// ─── Adversarial diffs: planted bugs ─────────────────
+describe('adversarial evaluation harness', () => {
+    it('catches npm usage in a CI script change', () => {
+        const diff = `diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml
+--- a/.github/workflows/ci.yml
++++ b/.github/workflows/ci.yml
+@@ -10,3 +10,4 @@ jobs:
+       - uses: actions/setup-node@v6
+       - run: pnpm install
++      - run: npm run test
+       - run: pnpm build
+`;
+        const violations = applyRules(ADVERSARIAL_RULES, diff);
+        expect(violations.length).toBeGreaterThan(0); // totem-ignore
+        expect(violations.some((v) => v.rule.lessonHeading === 'Never use npm')).toBe(true);
+    });
+    it('catches error instead of err in catch block', () => {
+        const diff = `diff --git a/src/handler.ts b/src/handler.ts
+--- a/src/handler.ts
++++ b/src/handler.ts
+@@ -5,3 +5,5 @@ export function handler() {
+   try {
+     doWork();
++  } catch (error) {
++    console.error(error);
+   }
+`;
+        const violations = applyRules(ADVERSARIAL_RULES, diff);
+        expect(violations.some((v) => v.rule.lessonHeading === 'Use err not error')).toBe(true);
+    });
+    it('catches debugger statements left in production code', () => {
+        const diff = `diff --git a/src/api.ts b/src/api.ts
+--- a/src/api.ts
++++ b/src/api.ts
+@@ -1,3 +1,5 @@
+ export async function fetchData() {
++  debugger;
+   const res = await fetch('/api/data');
++  debugger;
+   return res.json();
+`;
+        const violations = applyRules(ADVERSARIAL_RULES, diff);
+        const debugViolations = violations.filter((v) => v.rule.lessonHeading === 'No debugger in production');
+        expect(debugViolations).toHaveLength(2);
+    });
+    it('catches empty catch block', () => {
+        const diff = `diff --git a/src/utils.ts b/src/utils.ts
+--- a/src/utils.ts
++++ b/src/utils.ts
+@@ -1,3 +1,4 @@
+ export function tryParse(json: string) {
+-  return JSON.parse(json);
++  return fetchData().catch(() => {})
+ }
+`;
+        const violations = applyRules(ADVERSARIAL_RULES, diff);
+        expect(violations.some((v) => v.rule.lessonHeading === 'No empty catch blocks')).toBe(true);
+    });
+    it('catches TODO comments in new code', () => {
+        const diff = `diff --git a/src/auth.ts b/src/auth.ts
+--- a/src/auth.ts
++++ b/src/auth.ts
+@@ -1,3 +1,5 @@
+ export function authenticate(token: string) {
++  // TODO: add rate limiting
++  // FIXME: this is a temporary hack
+   return validateToken(token);
+`;
+        const violations = applyRules(ADVERSARIAL_RULES, diff);
+        const todoViolations = violations.filter((v) => v.rule.lessonHeading === 'No TODO in production');
+        expect(todoViolations).toHaveLength(2);
+    });
+    it('catches hardcoded password', () => {
+        const diff = `diff --git a/src/config.ts b/src/config.ts
+--- a/src/config.ts
++++ b/src/config.ts
+@@ -1,3 +1,4 @@
+ export const config = {
+   host: 'localhost',
++  password: 'hunter2',
+ };
+`;
+        const violations = applyRules(ADVERSARIAL_RULES, diff);
+        expect(violations.some((v) => v.rule.lessonHeading === 'No hardcoded secrets')).toBe(true);
+    });
+    it('catches TypeScript any type', () => {
+        const diff = `diff --git a/src/types.ts b/src/types.ts
+--- a/src/types.ts
++++ b/src/types.ts
+@@ -1,3 +1,4 @@
+ export interface Config {
+   host: string;
++  metadata: any;
+ }
+`;
+        const violations = applyRules(ADVERSARIAL_RULES, diff);
+        expect(violations.some((v) => v.rule.lessonHeading === 'No any types')).toBe(true);
+    });
+    it('passes a clean diff with no violations', () => {
+        const diff = `diff --git a/src/clean.ts b/src/clean.ts
+--- a/src/clean.ts
++++ b/src/clean.ts
+@@ -1,3 +1,5 @@
+ export function greet(name: string): string {
+-  return 'hello';
++  return \`Hello, \${name}!\`;
+ }
+`;
+        const violations = applyRules(ADVERSARIAL_RULES, diff);
+        expect(violations).toHaveLength(0);
+    });
+    it('respects inline suppression (totem-ignore)', () => {
+        const diff = `diff --git a/src/cli.ts b/src/cli.ts
+--- a/src/cli.ts
++++ b/src/cli.ts
+@@ -1,3 +1,4 @@
+ export function main() {
++  debugger; // totem-ignore
+   run();
+`;
+        const violations = applyRules(ADVERSARIAL_RULES, diff);
+        const debugViolations = violations.filter((v) => v.rule.lessonHeading === 'No debugger in production');
+        expect(debugViolations).toHaveLength(0);
+    });
+    it('catches multiple violations in a single diff', () => {
+        const diff = `diff --git a/src/bad.ts b/src/bad.ts
+--- a/src/bad.ts
++++ b/src/bad.ts
+@@ -1,5 +1,10 @@
+ export function bad() {
++  debugger;
++  // TODO: remove this
+   try {
+     doWork();
++  } catch (error) {
++    // swallow
+   }
++  const data: any = fetch('/api');
+ }
+`;
+        const violations = applyRules(ADVERSARIAL_RULES, diff);
+        // Should catch: debugger, TODO, error (not err), any
+        expect(violations.length).toBeGreaterThanOrEqual(4); // totem-ignore
+        const headings = new Set(violations.map((v) => v.rule.lessonHeading));
+        expect(headings.has('No debugger in production')).toBe(true);
+        expect(headings.has('No TODO in production')).toBe(true);
+        expect(headings.has('Use err not error')).toBe(true);
+        expect(headings.has('No any types')).toBe(true);
+    });
+});
+// ─── AST gating: false-positive suppression ─────────
+describe('AST gating suppresses false positives', () => {
+    it('skips violations in string-context additions', () => {
+        const rules = ADVERSARIAL_RULES;
+        // Simulate additions where AST gate has classified lines as string
+        // (e.g., inside a template literal test fixture)
+        const additions = [
+            {
+                file: 'src/test.ts',
+                line: '  debugger;',
+                lineNumber: 3,
+                precedingLine: null,
+                astContext: 'string', // inside template literal
+            },
+            {
+                file: 'src/test.ts',
+                line: '  // TODO: remove this',
+                lineNumber: 4,
+                precedingLine: '  debugger;',
+                astContext: 'string',
+            },
+            {
+                file: 'src/test.ts',
+                line: '  password: "hunter2",',
+                lineNumber: 5,
+                precedingLine: '  // TODO: remove this',
+                astContext: 'string',
+            },
+        ];
+        const violations = applyRulesToAdditions(rules, additions);
+        expect(violations).toHaveLength(0);
+    });
+    it('skips violations in comment-context additions', () => {
+        const rules = ADVERSARIAL_RULES;
+        const additions = [
+            {
+                file: 'src/app.ts',
+                line: '// npm install is required for setup',
+                lineNumber: 1,
+                precedingLine: null,
+                astContext: 'comment',
+            },
+        ];
+        const violations = applyRulesToAdditions(rules, additions);
+        expect(violations).toHaveLength(0);
+    });
+    it('still catches violations in code-context additions', () => {
+        const rules = ADVERSARIAL_RULES;
+        const additions = [
+            {
+                file: 'src/app.ts',
+                line: '  debugger;',
+                lineNumber: 5,
+                precedingLine: null,
+                astContext: 'code',
+            },
+        ];
+        const violations = applyRulesToAdditions(rules, additions);
+        expect(violations.length).toBeGreaterThan(0); // totem-ignore
+        expect(violations.some((v) => v.rule.lessonHeading === 'No debugger in production')).toBe(true);
+    });
+    it('still catches violations when astContext is undefined (fail-open)', () => {
+        const rules = ADVERSARIAL_RULES;
+        // No astContext = not classified = treated as code (fail-open)
+        const additions = [
+            {
+                file: 'src/app.ts',
+                line: '  debugger;',
+                lineNumber: 5,
+                precedingLine: null,
+                // astContext is undefined
+            },
+        ];
+        const violations = applyRulesToAdditions(rules, additions);
+        expect(violations.length).toBeGreaterThan(0); // totem-ignore
+    });
+});
+//# sourceMappingURL=adversarial.test.js.map

package/dist/adversarial.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"adversarial.test.js","sourceRoot":"","sources":["../src/adversarial.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAE9C,OAAO,EACL,UAAU,EACV,qBAAqB,GAGtB,MAAM,eAAe,CAAC;AAEvB,uDAAuD;AACvD,EAAE;AACF,yEAAyE;AACzE,uEAAuE;AACvE,yEAAyE;AACzE,4DAA4D;AAE5D,MAAM,QAAQ,GAAG,CAAC,OAAe,EAAE,OAAe,EAAE,OAAe,EAAgB,EAAE,CAAC,CAAC;IACrF,UAAU,EAAE,aAAa;IACzB,aAAa,EAAE,OAAO;IACtB,OAAO;IACP,OAAO;IACP,MAAM,EAAE,OAAO;IACf,UAAU,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;CACrC,CAAC,CAAC;AAEH,wDAAwD;AAExD,MAAM,iBAAiB,GAAmB;IACxC,QAAQ,CACN,iCAAiC,EACjC,0DAA0D,EAC1D,mBAAmB,CACpB;IACD,QAAQ,CACN,oCAAoC,EACpC,+CAA+C,EAC/C,eAAe,CAChB;IACD,QAAQ,CACN,gBAAgB,EAChB,0CAA0C,EAC1C,2BAA2B,CAC5B;IACD,QAAQ,CACN,uBAAuB,EACvB,6DAA6D,EAC7D,uBAAuB,CACxB;IACD,QAAQ,CACN,0CAA0C,EAC1C,4CAA4C,EAC5C,uBAAuB,CACxB;IACD,QAAQ,CACN,qBAAqB,EACrB,4CAA4C,EAC5C,uBAAuB,CACxB;IACD,QAAQ,CACN,uCAAuC,EACvC,0DAA0D,EAC1D,sBAAsB,CACvB;IACD,QAAQ,CAAC,WAAW,EAAE,oDAAoD,EAAE,cAAc,CAAC;CAC5F,CAAC;AAEF,wDAAwD;AAExD,QAAQ,CAAC,gCAAgC,EAAE,GAAG,EAAE;IAC9C,EAAE,CAAC,yCAAyC,EAAE,GAAG,EAAE;QACjD,MAAM,IAAI,GAAG;;;;;;;;CAQhB,CAAC;QACE,MAAM,UAAU,GAAG,UAAU,CAAC,iBAAiB,EAAE,IAAI,CAAC,CAAC;QACvD,MAAM,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC,CAAC,eAAe;QAC7D,MAAM,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,aAAa,KAAK,eAAe,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACtF,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6CAA6C,EAAE,GAAG,EAAE;QACrD,MAAM,IAAI,GAAG;;;;;;;;;CAShB,CAAC;QACE,MAAM,UAAU,GAAG,UAAU,CAAC,iBAAiB,EAAE,IAAI,CAAC,CAAC;QACvD,MAAM,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,aAAa,KAAK,mBAAmB,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAC1F,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qDAAqD,EAAE,GAAG,EAAE;QAC7D,MAAM,IAAI,GAAG;;;;;;;;;CAShB,CAAC;QACE,MAAM,UAAU,GAAG,UAAU,CAAC,iBAAiB,EAAE,IAAI,CAAC,CAAC;QACvD,MAAM,eAAe,GAAG,UAAU,CAAC,MAAM,CACvC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,aAAa,KAAK,2BAA2B,CAC5D,CAAC;QACF,MAAM,CAAC,eAAe,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;IAC1C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2BAA2B,EAAE,GAAG,EAAE;QACnC,MAAM,IAAI,GAAG;;;;;;;;CAQhB,CAAC;QACE,MAAM,UAAU,GAAG,UAAU,CAAC,iBAAiB,EAAE,IAAI,CAAC,CAAC;QACvD,MAAM,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,aAAa,KAAK,uBAAuB,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAC9F,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,mCAAmC,EAAE,GAAG,EAAE;QAC3C,MAAM,IAAI,GAAG;;;;;;;;CAQhB,CAAC;QACE,MAAM,UAAU,GAAG,UAAU,CAAC,iBAAiB,EAAE,IAAI,CAAC,CAAC;QACvD,MAAM,cAAc,GAAG,UAAU,CAAC,MAAM,CACtC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,aAAa,KAAK,uBAAuB,CACxD,CAAC;QACF,MAAM,CAAC,cAAc,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;IACzC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,4BAA4B,EAAE,GAAG,EAAE;QACpC,MAAM,IAAI,GAAG;;;;;;;;CAQhB,CAAC;QACE,MAAM,UAAU,GAAG,UAAU,CAAC,iBAAiB,EAAE,IAAI,CAAC,CAAC;QACvD,MAAM,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,aAAa,KAAK,sBAAsB,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAC7F,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6BAA6B,EAAE,GAAG,EAAE;QACrC,MAAM,IAAI,GAAG;;;;;;;;CAQhB,CAAC;QACE,MAAM,UAAU,GAAG,UAAU,CAAC,iBAAiB,EAAE,IAAI,CAAC,CAAC;QACvD,MAAM,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,aAAa,KAAK,cAAc,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACrF,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,wCAAwC,EAAE,GAAG,EAAE;QAChD,MAAM,IAAI,GAAG;;;;;;;;CAQhB,CAAC;QACE,MAAM,UAAU,GAAG,UAAU,CAAC,iBAAiB,EAAE,IAAI,CAAC,CAAC;QACvD,MAAM,CAAC,UAAU,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;IACrC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,4CAA4C,EAAE,GAAG,EAAE;QACpD,MAAM,IAAI,GAAG;;;;;;;CAOhB,CAAC;QACE,MAAM,UAAU,GAAG,UAAU,CAAC,iBAAiB,EAAE,IAAI,CAAC,CAAC;QACvD,MAAM,eAAe,GAAG,UAAU,CAAC,MAAM,CACvC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,aAAa,KAAK,2BAA2B,CAC5D,CAAC;QACF,MAAM,CAAC,eAAe,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;IAC1C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8CAA8C,EAAE,GAAG,EAAE;QACtD,MAAM,IAAI,GAAG;;;;;;;;;;;;;;CAchB,CAAC;QACE,MAAM,UAAU,GAAG,UAAU,CAAC,iBAAiB,EAAE,IAAI,CAAC,CAAC;QACvD,qDAAqD;QACrD,MAAM,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC,sBAAsB,CAAC,CAAC,CAAC,CAAC,CAAC,eAAe;QAEpE,MAAM,QAAQ,GAAG,IAAI,GAAG,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC,CAAC;QACtE,MAAM,CAAC,QAAQ,CAAC,GAAG,CAAC,2BAA2B,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC7D,MAAM,CAAC,QAAQ,CAAC,GAAG,CAAC,uBAAuB,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACzD,MAAM,CAAC,QAAQ,CAAC,GAAG,CAAC,mBAAmB,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACrD,MAAM,CAAC,QAAQ,CAAC,GAAG,CAAC,cAAc,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAClD,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,uDAAuD;AAEvD,QAAQ,CAAC,uCAAuC,EAAE,GAAG,EAAE;IACrD,EAAE,CAAC,8CAA8C,EAAE,GAAG,EAAE;QACtD,MAAM,KAAK,GAAG,iBAAiB,CAAC;QAEhC,mEAAmE;QACnE,iDAAiD;QACjD,MAAM,SAAS,GAAmB;YAChC;gBACE,IAAI,EAAE,aAAa;gBACnB,IAAI,EAAE,aAAa;gBACnB,UAAU,EAAE,CAAC;gBACb,aAAa,EAAE,IAAI;gBACnB,UAAU,EAAE,QAAQ,EAAE,0BAA0B;aACjD;YACD;gBACE,IAAI,EAAE,aAAa;gBACnB,IAAI,EAAE,wBAAwB;gBAC9B,UAAU,EAAE,CAAC;gBACb,aAAa,EAAE,aAAa;gBAC5B,UAAU,EAAE,QAAQ;aACrB;YACD;gBACE,IAAI,EAAE,aAAa;gBACnB,IAAI,EAAE,wBAAwB;gBAC9B,UAAU,EAAE,CAAC;gBACb,aAAa,EAAE,wBAAwB;gBACvC,UAAU,EAAE,QAAQ;aACrB;SACF,CAAC;QAEF,MAAM,UAAU,GAAG,qBAAqB,CAAC,KAAK,EAAE,SAAS,CAAC,CAAC;QAC3D,MAAM,CAAC,UAAU,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;IACrC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+CAA+C,EAAE,GAAG,EAAE;QACvD,MAAM,KAAK,GAAG,iBAAiB,CAAC;QAEhC,MAAM,SAAS,GAAmB;YAChC;gBACE,IAAI,EAAE,YAAY;gBAClB,IAAI,EAAE,sCAAsC;gBAC5C,UAAU,EAAE,CAAC;gBACb,aAAa,EAAE,IAAI;gBACnB,UAAU,EAAE,SAAS;aACtB;SACF,CAAC;QAEF,MAAM,UAAU,GAAG,qBAAqB,CAAC,KAAK,EAAE,SAAS,CAAC,CAAC;QAC3D,MAAM,CAAC,UAAU,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;IACrC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,oDAAoD,EAAE,GAAG,EAAE;QAC5D,MAAM,KAAK,GAAG,iBAAiB,CAAC;QAEhC,MAAM,SAAS,GAAmB;YAChC;gBACE,IAAI,EAAE,YAAY;gBAClB,IAAI,EAAE,aAAa;gBACnB,UAAU,EAAE,CAAC;gBACb,aAAa,EAAE,IAAI;gBACnB,UAAU,EAAE,MAAM;aACnB;SACF,CAAC;QAEF,MAAM,UAAU,GAAG,qBAAqB,CAAC,KAAK,EAAE,SAAS,CAAC,CAAC;QAC3D,MAAM,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC,CAAC,eAAe;QAC7D,MAAM,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,aAAa,KAAK,2BAA2B,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAClG,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,mEAAmE,EAAE,GAAG,EAAE;QAC3E,MAAM,KAAK,GAAG,iBAAiB,CAAC;QAEhC,+DAA+D;QAC/D,MAAM,SAAS,GAAmB;YAChC;gBACE,IAAI,EAAE,YAAY;gBAClB,IAAI,EAAE,aAAa;gBACnB,UAAU,EAAE,CAAC;gBACb,aAAa,EAAE,IAAI;gBACnB,0BAA0B;aAC3B;SACF,CAAC;QAEF,MAAM,UAAU,GAAG,qBAAqB,CAAC,KAAK,EAAE,SAAS,CAAC,CAAC;QAC3D,MAAM,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC,CAAC,eAAe;IAC/D,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/ast-classifier.d.ts

@@ -0,0 +1,17 @@
+import type { AstContext } from './compiler.js';
+export type SupportedLanguage = 'typescript' | 'tsx' | 'javascript';
+/**
+ * Classify specific lines of source code by their AST context.
+ *
+ * @param content - Full file content (used for parsing)
+ * @param lineNumbers - 1-based line numbers to classify
+ * @param language - Language to parse as
+ * @returns Map from line number to AstContext
+ */
+export declare function classifyLines(content: string, lineNumbers: number[], language: SupportedLanguage): Promise<Map<number, AstContext>>;
+/**
+ * Map file extension to a supported Tree-sitter language.
+ * Returns undefined for unsupported extensions.
+ */
+export declare function extensionToLanguage(ext: string): SupportedLanguage | undefined;
+//# sourceMappingURL=ast-classifier.d.ts.map

package/dist/ast-classifier.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"ast-classifier.d.ts","sourceRoot":"","sources":["../src/ast-classifier.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,eAAe,CAAC;AAIhD,MAAM,MAAM,iBAAiB,GAAG,YAAY,GAAG,KAAK,GAAG,YAAY,CAAC;AAgGpE;;;;;;;GAOG;AACH,wBAAsB,aAAa,CACjC,OAAO,EAAE,MAAM,EACf,WAAW,EAAE,MAAM,EAAE,EACrB,QAAQ,EAAE,iBAAiB,GAC1B,OAAO,CAAC,GAAG,CAAC,MAAM,EAAE,UAAU,CAAC,CAAC,CA+ClC;AAED;;;GAGG;AACH,wBAAgB,mBAAmB,CAAC,GAAG,EAAE,MAAM,GAAG,iBAAiB,GAAG,SAAS,CAe9E"}

package/dist/ast-classifier.js

@@ -0,0 +1,155 @@
+import { createRequire } from 'node:module';
+// ─── Lazy-loaded Tree-sitter state ──────────────────
+let Parser = null;
+let initPromise = null;
+const grammarCache = new Map();
+/**
+ * Initialize web-tree-sitter WASM engine. Idempotent — safe to call multiple times.
+ */
+async function ensureInit() {
+    if (Parser)
+        return;
+    if (initPromise)
+        return initPromise;
+    initPromise = (async () => {
+        const TreeSitter = await import('web-tree-sitter');
+        const ParserClass = TreeSitter.default?.Parser ?? TreeSitter.Parser;
+        const require = createRequire(import.meta.url);
+        // Locate the WASM engine file
+        const wasmPath = require.resolve('web-tree-sitter/web-tree-sitter.wasm');
+        await ParserClass.init({ locateFile: () => wasmPath });
+        Parser = ParserClass;
+    })();
+    return initPromise;
+}
+/**
+ * Load a Tree-sitter grammar WASM file for the given language.
+ */
+async function loadGrammar(lang) {
+    const cached = grammarCache.get(lang);
+    if (cached)
+        return cached;
+    await ensureInit();
+    const TreeSitter = await import('web-tree-sitter');
+    const LanguageClass = TreeSitter.default?.Language ?? TreeSitter.Language;
+    const require = createRequire(import.meta.url);
+    let wasmFile;
+    switch (lang) {
+        case 'typescript':
+            wasmFile = require.resolve('tree-sitter-typescript/tree-sitter-typescript.wasm');
+            break;
+        case 'tsx':
+            wasmFile = require.resolve('tree-sitter-typescript/tree-sitter-tsx.wasm');
+            break;
+        case 'javascript':
+            wasmFile = require.resolve('tree-sitter-javascript/tree-sitter-javascript.wasm');
+            break;
+    }
+    const grammar = await LanguageClass.load(wasmFile);
+    grammarCache.set(lang, grammar);
+    return grammar;
+}
+// ─── Node type classification ───────────────────────
+/** Tree-sitter node types that represent string literals */
+const STRING_NODE_TYPES = new Set([
+    'string',
+    'template_string',
+    'string_fragment',
+    'template_literal_type',
+]);
+/** Tree-sitter node types that represent comments */
+const COMMENT_NODE_TYPES = new Set(['comment']);
+/** Tree-sitter node types that represent regex literals */
+const REGEX_NODE_TYPES = new Set(['regex', 'regex_pattern']);
+/**
+ * Walk the AST ancestry from a node to determine its syntactic context.
+ * Returns the deepest enclosing string/comment/regex context, or 'code'.
+ */
+function classifyNode(node) {
+    let current = node;
+    while (current) {
+        if (STRING_NODE_TYPES.has(current.type))
+            return 'string';
+        if (COMMENT_NODE_TYPES.has(current.type))
+            return 'comment';
+        if (REGEX_NODE_TYPES.has(current.type))
+            return 'regex';
+        current = current.parent;
+    }
+    return 'code';
+}
+// ─── Public API ─────────────────────────────────────
+/**
+ * Classify specific lines of source code by their AST context.
+ *
+ * @param content - Full file content (used for parsing)
+ * @param lineNumbers - 1-based line numbers to classify
+ * @param language - Language to parse as
+ * @returns Map from line number to AstContext
+ */
+export async function classifyLines(content, lineNumbers, language) {
+    const result = new Map();
+    if (lineNumbers.length === 0)
+        return result;
+    await ensureInit();
+    const grammar = await loadGrammar(language);
+    const parser = new Parser();
+    try {
+        parser.setLanguage(grammar);
+        const tree = parser.parse(content);
+        if (!tree) {
+            // Parse failed — fail-open, leave all lines unclassified
+            return result;
+        }
+        try {
+            const rootNode = tree.rootNode;
+            const lines = content.split('\n');
+            for (const lineNum of lineNumbers) {
+                // Tree-sitter uses 0-based rows
+                const row = lineNum - 1;
+                // Get the named node at the start of this line (skip leading whitespace)
+                const lineText = lines[row];
+                if (lineText === undefined)
+                    continue;
+                const col = lineText.length - lineText.trimStart().length;
+                const node = rootNode.descendantForPosition({ row, column: col });
+                if (!node)
+                    continue;
+                // If the node is an ERROR node, classify as 'code' (fail-open)
+                if (node.type === 'ERROR' || node.hasError) {
+                    result.set(lineNum, 'code');
+                    continue;
+                }
+                result.set(lineNum, classifyNode(node));
+            }
+        }
+        finally {
+            tree.delete();
+        }
+    }
+    finally {
+        parser.delete();
+    }
+    return result;
+}
+/**
+ * Map file extension to a supported Tree-sitter language.
+ * Returns undefined for unsupported extensions.
+ */
+export function extensionToLanguage(ext) {
+    switch (ext.toLowerCase()) {
+        case '.ts':
+            return 'typescript';
+        case '.tsx':
+            return 'tsx';
+        case '.js':
+        case '.mjs':
+        case '.cjs':
+            return 'javascript';
+        case '.jsx':
+            return 'tsx'; // TSX grammar handles JSX
+        default:
+            return undefined;
+    }
+}
+//# sourceMappingURL=ast-classifier.js.map

package/dist/ast-classifier.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"ast-classifier.js","sourceRoot":"","sources":["../src/ast-classifier.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,aAAa,EAAE,MAAM,aAAa,CAAC;AAQ5C,uDAAuD;AAEvD,IAAI,MAAM,GAAmD,IAAI,CAAC;AAClE,IAAI,WAAW,GAAyB,IAAI,CAAC;AAE7C,MAAM,YAAY,GAAG,IAAI,GAAG,EAAyD,CAAC;AAEtF;;GAEG;AACH,KAAK,UAAU,UAAU;IACvB,IAAI,MAAM;QAAE,OAAO;IACnB,IAAI,WAAW;QAAE,OAAO,WAAW,CAAC;IAEpC,WAAW,GAAG,CAAC,KAAK,IAAI,EAAE;QACxB,MAAM,UAAU,GAAG,MAAM,MAAM,CAAC,iBAAiB,CAAC,CAAC;QACnD,MAAM,WAAW,GAAG,UAAU,CAAC,OAAO,EAAE,MAAM,IAAI,UAAU,CAAC,MAAM,CAAC;QACpE,MAAM,OAAO,GAAG,aAAa,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QAE/C,8BAA8B;QAC9B,MAAM,QAAQ,GAAG,OAAO,CAAC,OAAO,CAAC,sCAAsC,CAAC,CAAC;QACzE,MAAM,WAAW,CAAC,IAAI,CAAC,EAAE,UAAU,EAAE,GAAG,EAAE,CAAC,QAAQ,EAAE,CAAC,CAAC;QACvD,MAAM,GAAG,WAAW,CAAC;IACvB,CAAC,CAAC,EAAE,CAAC;IAEL,OAAO,WAAW,CAAC;AACrB,CAAC;AAED;;GAEG;AACH,KAAK,UAAU,WAAW,CAAC,IAAuB;IAChD,MAAM,MAAM,GAAG,YAAY,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;IACtC,IAAI,MAAM;QAAE,OAAO,MAAM,CAAC;IAE1B,MAAM,UAAU,EAAE,CAAC;IACnB,MAAM,UAAU,GAAG,MAAM,MAAM,CAAC,iBAAiB,CAAC,CAAC;IACnD,MAAM,aAAa,GAAG,UAAU,CAAC,OAAO,EAAE,QAAQ,IAAI,UAAU,CAAC,QAAQ,CAAC;IAE1E,MAAM,OAAO,GAAG,aAAa,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;IAC/C,IAAI,QAAgB,CAAC;IAErB,QAAQ,IAAI,EAAE,CAAC;QACb,KAAK,YAAY;YACf,QAAQ,GAAG,OAAO,CAAC,OAAO,CAAC,oDAAoD,CAAC,CAAC;YACjF,MAAM;QACR,KAAK,KAAK;YACR,QAAQ,GAAG,OAAO,CAAC,OAAO,CAAC,6CAA6C,CAAC,CAAC;YAC1E,MAAM;QACR,KAAK,YAAY;YACf,QAAQ,GAAG,OAAO,CAAC,OAAO,CAAC,oDAAoD,CAAC,CAAC;YACjF,MAAM;IACV,CAAC;IAED,MAAM,OAAO,GAAG,MAAM,aAAa,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;IACnD,YAAY,CAAC,GAAG,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC;IAChC,OAAO,OAAO,CAAC;AACjB,CAAC;AAED,uDAAuD;AAEvD,4DAA4D;AAC5D,MAAM,iBAAiB,GAAG,IAAI,GAAG,CAAC;IAChC,QAAQ;IACR,iBAAiB;IACjB,iBAAiB;IACjB,uBAAuB;CACxB,CAAC,CAAC;AAEH,qDAAqD;AACrD,MAAM,kBAAkB,GAAG,IAAI,GAAG,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC;AAEhD,2DAA2D;AAC3D,MAAM,gBAAgB,GAAG,IAAI,GAAG,CAAC,CAAC,OAAO,EAAE,eAAe,CAAC,CAAC,CAAC;AAE7D;;;GAGG;AACH,SAAS,YAAY,CAAC,IAAoC;IACxD,IAAI,OAAO,GAA0C,IAAI,CAAC;IAE1D,OAAO,OAAO,EAAE,CAAC;QACf,IAAI,iBAAiB,CAAC,GAAG,CAAC,OAAO,CAAC,IAAI,CAAC;YAAE,OAAO,QAAQ,CAAC;QACzD,IAAI,kBAAkB,CAAC,GAAG,CAAC,OAAO,CAAC,IAAI,CAAC;YAAE,OAAO,SAAS,CAAC;QAC3D,IAAI,gBAAgB,CAAC,GAAG,CAAC,OAAO,CAAC,IAAI,CAAC;YAAE,OAAO,OAAO,CAAC;QACvD,OAAO,GAAG,OAAO,CAAC,MAAM,CAAC;IAC3B,CAAC;IAED,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,uDAAuD;AAEvD;;;;;;;GAOG;AACH,MAAM,CAAC,KAAK,UAAU,aAAa,CACjC,OAAe,EACf,WAAqB,EACrB,QAA2B;IAE3B,MAAM,MAAM,GAAG,IAAI,GAAG,EAAsB,CAAC;IAC7C,IAAI,WAAW,CAAC,MAAM,KAAK,CAAC;QAAE,OAAO,MAAM,CAAC;IAE5C,MAAM,UAAU,EAAE,CAAC;IAEnB,MAAM,OAAO,GAAG,MAAM,WAAW,CAAC,QAAQ,CAAC,CAAC;IAC5C,MAAM,MAAM,GAAG,IAAI,MAAO,EAAE,CAAC;IAC7B,IAAI,CAAC;QACH,MAAM,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC;QAC5B,MAAM,IAAI,GAAG,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;QACnC,IAAI,CAAC,IAAI,EAAE,CAAC;YACV,yDAAyD;YACzD,OAAO,MAAM,CAAC;QAChB,CAAC;QAED,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,IAAI,CAAC,QAAQ,CAAC;YAC/B,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;YAElC,KAAK,MAAM,OAAO,IAAI,WAAW,EAAE,CAAC;gBAClC,gCAAgC;gBAChC,MAAM,GAAG,GAAG,OAAO,GAAG,CAAC,CAAC;gBACxB,yEAAyE;gBACzE,MAAM,QAAQ,GAAG,KAAK,CAAC,GAAG,CAAC,CAAC;gBAC5B,IAAI,QAAQ,KAAK,SAAS;oBAAE,SAAS;gBAErC,MAAM,GAAG,GAAG,QAAQ,CAAC,MAAM,GAAG,QAAQ,CAAC,SAAS,EAAE,CAAC,MAAM,CAAC;gBAC1D,MAAM,IAAI,GAAG,QAAQ,CAAC,qBAAqB,CAAC,EAAE,GAAG,EAAE,MAAM,EAAE,GAAG,EAAE,CAAC,CAAC;gBAClE,IAAI,CAAC,IAAI;oBAAE,SAAS;gBAEpB,+DAA+D;gBAC/D,IAAI,IAAI,CAAC,IAAI,KAAK,OAAO,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAC;oBAC3C,MAAM,CAAC,GAAG,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;oBAC5B,SAAS;gBACX,CAAC;gBAED,MAAM,CAAC,GAAG,CAAC,OAAO,EAAE,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC;YAC1C,CAAC;QACH,CAAC;gBAAS,CAAC;YACT,IAAI,CAAC,MAAM,EAAE,CAAC;QAChB,CAAC;IACH,CAAC;YAAS,CAAC;QACT,MAAM,CAAC,MAAM,EAAE,CAAC;IAClB,CAAC;IAED,OAAO,MAAM,CAAC;AAChB,CAAC;AAED;;;GAGG;AACH,MAAM,UAAU,mBAAmB,CAAC,GAAW;IAC7C,QAAQ,GAAG,CAAC,WAAW,EAAE,EAAE,CAAC;QAC1B,KAAK,KAAK;YACR,OAAO,YAAY,CAAC;QACtB,KAAK,MAAM;YACT,OAAO,KAAK,CAAC;QACf,KAAK,KAAK,CAAC;QACX,KAAK,MAAM,CAAC;QACZ,KAAK,MAAM;YACT,OAAO,YAAY,CAAC;QACtB,KAAK,MAAM;YACT,OAAO,KAAK,CAAC,CAAC,0BAA0B;QAC1C;YACE,OAAO,SAAS,CAAC;IACrB,CAAC;AACH,CAAC"}

package/dist/ast-classifier.test.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=ast-classifier.test.d.ts.map

package/dist/ast-classifier.test.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"ast-classifier.test.d.ts","sourceRoot":"","sources":["../src/ast-classifier.test.ts"],"names":[],"mappings":""}

package/dist/ast-classifier.test.js

@@ -0,0 +1,109 @@
+import { describe, expect, it } from 'vitest';
+import { classifyLines, extensionToLanguage } from './ast-classifier.js';
+// ─── extensionToLanguage ────────────────────────────
+describe('extensionToLanguage', () => {
+    it('maps .ts to typescript', () => {
+        expect(extensionToLanguage('.ts')).toBe('typescript');
+    });
+    it('maps .tsx to tsx', () => {
+        expect(extensionToLanguage('.tsx')).toBe('tsx');
+    });
+    it('maps .js/.mjs/.cjs to javascript', () => {
+        expect(extensionToLanguage('.js')).toBe('javascript');
+        expect(extensionToLanguage('.mjs')).toBe('javascript');
+        expect(extensionToLanguage('.cjs')).toBe('javascript');
+    });
+    it('maps .jsx to tsx', () => {
+        expect(extensionToLanguage('.jsx')).toBe('tsx');
+    });
+    it('returns undefined for unsupported extensions', () => {
+        expect(extensionToLanguage('.py')).toBeUndefined();
+        expect(extensionToLanguage('.rs')).toBeUndefined();
+        expect(extensionToLanguage('.md')).toBeUndefined();
+    });
+});
+// ─── classifyLines ──────────────────────────────────
+describe('classifyLines', () => {
+    it('classifies a regular code line as code', async () => {
+        const content = `const x = 1;\nconsole.log(x);\n`; // totem-ignore
+        const result = await classifyLines(content, [1, 2], 'typescript');
+        expect(result.get(1)).toBe('code');
+        expect(result.get(2)).toBe('code');
+    });
+    it('classifies a single-line comment as comment', async () => {
+        const content = `// this is a comment\nconst x = 1;\n`;
+        const result = await classifyLines(content, [1], 'typescript');
+        expect(result.get(1)).toBe('comment');
+    });
+    it('classifies a multi-line comment as comment', async () => {
+        const content = `/*\n * block comment\n */\nconst x = 1;\n`;
+        const result = await classifyLines(content, [2], 'typescript');
+        expect(result.get(2)).toBe('comment');
+    });
+    it('classifies a string literal as string', async () => {
+        const content = `const msg = "hello world";\n`;
+        // The string token is on line 1 — but the whole line is an assignment (code)
+        // The string node is a child of the variable declaration
+        // classifyLines checks the node at the start of the trimmed line
+        const result = await classifyLines(content, [1], 'typescript');
+        // The leftmost token is `const`, which is code
+        expect(result.get(1)).toBe('code');
+    });
+    it('classifies content inside a template literal as string', async () => {
+        const content = [
+            'const fixture = `',
+            '  this is inside a template literal',
+            '  console.log("fake code")',
+            '`;',
+        ].join('\n');
+        const result = await classifyLines(content, [2, 3], 'typescript');
+        expect(result.get(2)).toBe('string');
+        expect(result.get(3)).toBe('string');
+    });
+    it('classifies a multi-line template literal body as string', async () => {
+        const content = [
+            'const diff = `diff --git a/foo.ts b/foo.ts',
+            '--- a/foo.ts',
+            '+++ b/foo.ts',
+            '@@ -1,3 +1,4 @@',
+            ' const a = 1;',
+            '+const b = 2;',
+            '`;',
+        ].join('\n');
+        // Lines 2-6 are inside the template literal
+        const result = await classifyLines(content, [2, 3, 4, 5, 6], 'typescript');
+        for (const line of [2, 3, 4, 5, 6]) {
+            expect(result.get(line)).toBe('string');
+        }
+    });
+    it('classifies a regex literal as regex', async () => {
+        const content = `const re = /foo\\.bar/;\n`;
+        const result = await classifyLines(content, [1], 'typescript');
+        // The leftmost token is `const` → code
+        expect(result.get(1)).toBe('code');
+    });
+    it('returns empty map for empty line numbers', async () => {
+        const result = await classifyLines('const x = 1;', [], 'typescript');
+        expect(result.size).toBe(0);
+    });
+    it('handles JavaScript files', async () => {
+        const content = `// JS comment\nconst x = 1;\n`;
+        const result = await classifyLines(content, [1, 2], 'javascript');
+        expect(result.get(1)).toBe('comment');
+        expect(result.get(2)).toBe('code');
+    });
+    it('handles TSX files', async () => {
+        const content = `// TSX comment\nconst x = <div>hello</div>;\n`;
+        const result = await classifyLines(content, [1, 2], 'tsx');
+        expect(result.get(1)).toBe('comment');
+        expect(result.get(2)).toBe('code');
+    });
+    it('classifies ERROR nodes as code (fail-open)', async () => {
+        // Intentionally broken syntax
+        const content = `const x = {{\n`;
+        const result = await classifyLines(content, [1], 'typescript');
+        // Should not throw, and should classify as code (fail-open)
+        expect(result.get(1)).toBe('code');
+    });
+});
+//# sourceMappingURL=ast-classifier.test.js.map

package/dist/ast-classifier.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"ast-classifier.test.js","sourceRoot":"","sources":["../src/ast-classifier.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAE9C,OAAO,EAAE,aAAa,EAAE,mBAAmB,EAAE,MAAM,qBAAqB,CAAC;AAEzE,uDAAuD;AAEvD,QAAQ,CAAC,qBAAqB,EAAE,GAAG,EAAE;IACnC,EAAE,CAAC,wBAAwB,EAAE,GAAG,EAAE;QAChC,MAAM,CAAC,mBAAmB,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;IACxD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kBAAkB,EAAE,GAAG,EAAE;QAC1B,MAAM,CAAC,mBAAmB,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IAClD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kCAAkC,EAAE,GAAG,EAAE;QAC1C,MAAM,CAAC,mBAAmB,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;QACtD,MAAM,CAAC,mBAAmB,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;QACvD,MAAM,CAAC,mBAAmB,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;IACzD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kBAAkB,EAAE,GAAG,EAAE;QAC1B,MAAM,CAAC,mBAAmB,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IAClD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8CAA8C,EAAE,GAAG,EAAE;QACtD,MAAM,CAAC,mBAAmB,CAAC,KAAK,CAAC,CAAC,CAAC,aAAa,EAAE,CAAC;QACnD,MAAM,CAAC,mBAAmB,CAAC,KAAK,CAAC,CAAC,CAAC,aAAa,EAAE,CAAC;QACnD,MAAM,CAAC,mBAAmB,CAAC,KAAK,CAAC,CAAC,CAAC,aAAa,EAAE,CAAC;IACrD,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,uDAAuD;AAEvD,QAAQ,CAAC,eAAe,EAAE,GAAG,EAAE;IAC7B,EAAE,CAAC,wCAAwC,EAAE,KAAK,IAAI,EAAE;QACtD,MAAM,OAAO,GAAG,iCAAiC,CAAC,CAAC,eAAe;QAClE,MAAM,MAAM,GAAG,MAAM,aAAa,CAAC,OAAO,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,YAAY,CAAC,CAAC;QAClE,MAAM,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QACnC,MAAM,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;IACrC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6CAA6C,EAAE,KAAK,IAAI,EAAE;QAC3D,MAAM,OAAO,GAAG,sCAAsC,CAAC;QACvD,MAAM,MAAM,GAAG,MAAM,aAAa,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,YAAY,CAAC,CAAC;QAC/D,MAAM,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;IACxC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,4CAA4C,EAAE,KAAK,IAAI,EAAE;QAC1D,MAAM,OAAO,GAAG,2CAA2C,CAAC;QAC5D,MAAM,MAAM,GAAG,MAAM,aAAa,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,YAAY,CAAC,CAAC;QAC/D,MAAM,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;IACxC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,uCAAuC,EAAE,KAAK,IAAI,EAAE;QACrD,MAAM,OAAO,GAAG,8BAA8B,CAAC;QAC/C,6EAA6E;QAC7E,yDAAyD;QACzD,iEAAiE;QACjE,MAAM,MAAM,GAAG,MAAM,aAAa,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,YAAY,CAAC,CAAC;QAC/D,+CAA+C;QAC/C,MAAM,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;IACrC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,wDAAwD,EAAE,KAAK,IAAI,EAAE;QACtE,MAAM,OAAO,GAAG;YACd,mBAAmB;YACnB,qCAAqC;YACrC,4BAA4B;YAC5B,IAAI;SACL,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACb,MAAM,MAAM,GAAG,MAAM,aAAa,CAAC,OAAO,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,YAAY,CAAC,CAAC;QAClE,MAAM,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QACrC,MAAM,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;IACvC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yDAAyD,EAAE,KAAK,IAAI,EAAE;QACvE,MAAM,OAAO,GAAG;YACd,4CAA4C;YAC5C,cAAc;YACd,cAAc;YACd,iBAAiB;YACjB,eAAe;YACf,eAAe;YACf,IAAI;SACL,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACb,4CAA4C;QAC5C,MAAM,MAAM,GAAG,MAAM,aAAa,CAAC,OAAO,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,EAAE,YAAY,CAAC,CAAC;QAC3E,KAAK,MAAM,IAAI,IAAI,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC;YACnC,MAAM,CAAC,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAC1C,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qCAAqC,EAAE,KAAK,IAAI,EAAE;QACnD,MAAM,OAAO,GAAG,2BAA2B,CAAC;QAC5C,MAAM,MAAM,GAAG,MAAM,aAAa,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,YAAY,CAAC,CAAC;QAC/D,uCAAuC;QACvC,MAAM,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;IACrC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,0CAA0C,EAAE,KAAK,IAAI,EAAE;QACxD,MAAM,MAAM,GAAG,MAAM,aAAa,CAAC,cAAc,EAAE,EAAE,EAAE,YAAY,CAAC,CAAC;QACrE,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAC9B,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,0BAA0B,EAAE,KAAK,IAAI,EAAE;QACxC,MAAM,OAAO,GAAG,+BAA+B,CAAC;QAChD,MAAM,MAAM,GAAG,MAAM,aAAa,CAAC,OAAO,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,YAAY,CAAC,CAAC;QAClE,MAAM,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QACtC,MAAM,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;IACrC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,mBAAmB,EAAE,KAAK,IAAI,EAAE;QACjC,MAAM,OAAO,GAAG,+CAA+C,CAAC;QAChE,MAAM,MAAM,GAAG,MAAM,aAAa,CAAC,OAAO,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC;QAC3D,MAAM,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QACtC,MAAM,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;IACrC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,4CAA4C,EAAE,KAAK,IAAI,EAAE;QAC1D,8BAA8B;QAC9B,MAAM,OAAO,GAAG,gBAAgB,CAAC;QACjC,MAAM,MAAM,GAAG,MAAM,aAAa,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,YAAY,CAAC,CAAC;QAC/D,4DAA4D;QAC5D,MAAM,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;IACrC,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/ast-gate.d.ts

@@ -0,0 +1,18 @@
+import type { DiffAddition } from './compiler.js';
+export interface AstGateOptions {
+    /** Working directory to resolve file paths against */
+    cwd?: string;
+    /** Optional callback for non-fatal warnings (e.g., file not readable) */
+    onWarn?: (msg: string) => void;
+}
+/**
+ * Enrich diff additions with AST context by parsing the actual files.
+ *
+ * Groups additions by file, reads each file from disk, runs Tree-sitter
+ * classification, and sets `astContext` on each addition.
+ *
+ * Additions for unsupported file types or unreadable files are left
+ * with `astContext: undefined` (fail-open — treated as code by the shield).
+ */
+export declare function enrichWithAstContext(additions: DiffAddition[], options?: AstGateOptions): Promise<void>;
+//# sourceMappingURL=ast-gate.d.ts.map

package/dist/ast-gate.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"ast-gate.d.ts","sourceRoot":"","sources":["../src/ast-gate.ts"],"names":[],"mappings":"AAIA,OAAO,KAAK,EAAc,YAAY,EAAE,MAAM,eAAe,CAAC;AAI9D,MAAM,WAAW,cAAc;IAC7B,sDAAsD;IACtD,GAAG,CAAC,EAAE,MAAM,CAAC;IACb,yEAAyE;IACzE,MAAM,CAAC,EAAE,CAAC,GAAG,EAAE,MAAM,KAAK,IAAI,CAAC;CAChC;AAID;;;;;;;;GAQG;AACH,wBAAsB,oBAAoB,CACxC,SAAS,EAAE,YAAY,EAAE,EACzB,OAAO,GAAE,cAAmB,GAC3B,OAAO,CAAC,IAAI,CAAC,CA0Bf"}