optave-codegraph 3.13.0

package/dist/ast-analysis/visitors/ast-store-visitor.js

@@ -0,0 +1,288 @@
+const TEXT_MAX = 200;
+// ── Cross-language node-type constants (mirror Rust `helpers.rs`) ────────
+const IDENT_TYPES = new Set([
+    'identifier',
+    'type_identifier',
+    'name',
+    'qualified_name',
+    'scoped_identifier',
+    'qualified_identifier',
+    'member_expression',
+    'member_access_expression',
+    'field_expression',
+    'attribute',
+    'scoped_type_identifier',
+]);
+const CALL_TYPES = new Set([
+    'call_expression',
+    'call',
+    'invocation_expression',
+    'method_invocation',
+    'function_call_expression',
+    'member_call_expression',
+    'scoped_call_expression',
+]);
+const DEFAULT_STRING_CONFIG = { quoteChars: '\'"`', stringPrefixes: '' };
+// Keyword tokens skipped when extracting the inner expression text of a
+// throw/raise/await/new node. Module-level constant avoids reallocating on
+// every call (can be hot in large files).
+const CHILD_EXPR_SKIP_KEYWORDS = new Set(['throw', 'raise', 'await', 'new']);
+function truncate(s, max = TEXT_MAX) {
+    if (!s)
+        return null;
+    return s.length <= max ? s : `${s.slice(0, max - 1)}…`;
+}
+function trimLeadingChars(s, chars) {
+    if (!chars)
+        return s;
+    let i = 0;
+    while (i < s.length && chars.includes(s[i]))
+        i++;
+    return i === 0 ? s : s.slice(i);
+}
+function trimTrailingChars(s, chars) {
+    if (!chars)
+        return s;
+    let i = s.length;
+    while (i > 0 && chars.includes(s[i - 1]))
+        i--;
+    return i === s.length ? s : s.slice(0, i);
+}
+/** Extract constructor name from a `new_expression` / `object_creation_expression`. */
+function extractConstructorName(node) {
+    for (const field of ['type', 'class', 'constructor']) {
+        const f = node.childForFieldName(field);
+        if (f?.text)
+            return f.text;
+    }
+    for (let i = 0; i < node.childCount; i++) {
+        const child = node.child(i);
+        if (!child)
+            continue;
+        if (IDENT_TYPES.has(child.type))
+            return child.text;
+    }
+    const raw = node.text || '';
+    const beforeParen = raw.split('(')[0] || raw;
+    return beforeParen.replace(/^new\s+/, '').trim() || '?';
+}
+/** Extract function name from a call node. */
+function extractCallName(node) {
+    for (const field of ['function', 'method', 'name']) {
+        const f = node.childForFieldName(field);
+        if (f?.text)
+            return f.text;
+    }
+    const text = node.text || '';
+    return text.split('(')[0] || '?';
+}
+/** Extract name from a throw/raise statement — matches native `extract_throw_target`. */
+function extractThrowName(node, newTypes) {
+    for (let i = 0; i < node.childCount; i++) {
+        const child = node.child(i);
+        if (!child)
+            continue;
+        const ck = child.type;
+        if (newTypes.has(ck))
+            return extractConstructorName(child);
+        if (CALL_TYPES.has(ck))
+            return extractCallName(child);
+        if (IDENT_TYPES.has(ck))
+            return child.text;
+    }
+    return truncate(node.text) ?? node.text ?? '';
+}
+/** Extract name from an await expression — matches native `extract_awaited_name`. */
+function extractAwaitName(node) {
+    for (let i = 0; i < node.childCount; i++) {
+        const child = node.child(i);
+        if (!child)
+            continue;
+        const ck = child.type;
+        if (CALL_TYPES.has(ck))
+            return extractCallName(child);
+        if (IDENT_TYPES.has(ck))
+            return child.text;
+    }
+    return truncate(node.text) ?? node.text ?? '';
+}
+/** Extract text of the expression inside a throw/await, skipping the keyword. */
+function extractChildExpressionText(node) {
+    for (let i = 0; i < node.childCount; i++) {
+        const child = node.child(i);
+        if (!child)
+            continue;
+        if (!CHILD_EXPR_SKIP_KEYWORDS.has(child.type))
+            return truncate(child.text);
+    }
+    return truncate(node.text);
+}
+/**
+ * Count code points cheaply: skip the `[...s]` spread when `s.length` already
+ * decides the answer. Each code point is 1 or 2 UTF-16 units, so `.length < 2`
+ * implies `< 2` code points and `.length >= 3` already guarantees `>= 2` code
+ * points (worst case: one surrogate pair + one BMP char = 2 code points).
+ * Only `.length === 2` is genuinely ambiguous (could be a single surrogate
+ * pair = 1 code point, or two BMP chars = 2 code points) and needs the spread.
+ */
+function codePointCountAtLeast2(s) {
+    const len = s.length;
+    if (len < 2)
+        return false;
+    if (len >= 3)
+        return true;
+    return [...s].length >= 2;
+}
+/**
+ * Extract string content from a string-literal node, mirroring the native
+ * engine's `build_string_node` (`helpers.rs`). Returns `null` when the
+ * content is shorter than 2 Unicode code points.
+ */
+function extractStringContent(node, cfg) {
+    const raw = node.text ?? '';
+    const isRawString = node.type.includes('raw_string');
+    let s = raw;
+    s = trimLeadingChars(s, '@');
+    if (cfg.stringPrefixes)
+        s = trimLeadingChars(s, cfg.stringPrefixes);
+    if (isRawString)
+        s = trimLeadingChars(s, 'r#');
+    s = trimLeadingChars(s, cfg.quoteChars);
+    if (isRawString)
+        s = trimTrailingChars(s, '#');
+    s = trimTrailingChars(s, cfg.quoteChars);
+    return codePointCountAtLeast2(s) ? s : null;
+}
+// Per-astTypeMap cache for the set of node-types that map to kind 'new'.
+// Computed once per unique astTypeMap reference (one per language) instead
+// of once per file.
+const _newTypesCache = new WeakMap();
+function newTypesFor(astTypeMap) {
+    let s = _newTypesCache.get(astTypeMap);
+    if (s)
+        return s;
+    s = new Set();
+    for (const type in astTypeMap) {
+        if (astTypeMap[type] === 'new')
+            s.add(type);
+    }
+    _newTypesCache.set(astTypeMap, s);
+    return s;
+}
+const DEFAULT_NAME_TEXT_RESULT = { name: undefined, text: null };
+/** Build the per-kind resolver map for name/text extraction. */
+function buildKindHandlers(newTypes, stringConfig) {
+    return {
+        new: (node) => ({ name: extractConstructorName(node), text: truncate(node.text) }),
+        throw: (node) => ({
+            name: extractThrowName(node, newTypes),
+            text: extractChildExpressionText(node),
+        }),
+        await: (node) => ({ name: extractAwaitName(node), text: extractChildExpressionText(node) }),
+        string: (node) => {
+            const content = extractStringContent(node, stringConfig);
+            if (content == null)
+                return { name: null, text: null, skip: true };
+            return { name: truncate(content, 100), text: truncate(node.text) };
+        },
+        regex: (node) => ({ name: node.text || '?', text: truncate(node.text) }),
+    };
+}
+/** Find the innermost definition whose line range contains `line`. */
+function findParentDef(line, defs) {
+    let best = null;
+    for (const def of defs) {
+        if (def.line <= line && (def.endLine == null || def.endLine >= line)) {
+            if (!best || (def.endLine ?? 0) - def.line < (best.endLine ?? 0) - best.line) {
+                best = def;
+            }
+        }
+    }
+    return best;
+}
+/** Resolve the parent definition's node id for a given source line. */
+function resolveParentNodeId(line, defs, nodeIdMap, skipParentLookup) {
+    if (skipParentLookup)
+        return null;
+    const parentDef = findParentDef(line, defs);
+    if (!parentDef)
+        return null;
+    return nodeIdMap.get(`${parentDef.name}|${parentDef.kind}|${parentDef.line}`) || null;
+}
+function collectNode(ctx, node, kind) {
+    if (ctx.matched.has(node.id))
+        return;
+    const handler = ctx.kindHandlers[kind];
+    const resolved = handler ? handler(node) : DEFAULT_NAME_TEXT_RESULT;
+    if (resolved.skip)
+        return;
+    const line = node.startPosition.row + 1;
+    ctx.rows.push({
+        file: ctx.relPath,
+        line,
+        kind,
+        name: resolved.name,
+        text: resolved.text,
+        receiver: null,
+        parentNodeId: resolveParentNodeId(line, ctx.defs, ctx.nodeIdMap, ctx.skipParentLookup),
+    });
+    ctx.matched.add(node.id);
+}
+/**
+ * Resolve the kind for a tree-sitter node, or `null` if the node should be ignored.
+ *
+ * Gate with `hasOwn` because plain-object lookup walks Object.prototype:
+ * tree-sitter node types like `constructor` (Haskell sum-types: Left,
+ * Right) would otherwise resolve to `Object.prototype.constructor` (the
+ * Object() function), which then crashes the worker boundary with
+ * "function Object() { [native code] } could not be cloned" when the
+ * resulting astNodes row is structured-cloned back to the main thread.
+ */
+function resolveAstKind(node, astTypeMap) {
+    if (!Object.hasOwn(astTypeMap, node.type))
+        return null;
+    return astTypeMap[node.type] || null;
+}
+export function createAstStoreVisitor(astTypeMap, defs, relPath, nodeIdMap, stringConfig = DEFAULT_STRING_CONFIG, stopRecurseKinds = new Set()) {
+    const newTypes = newTypesFor(astTypeMap);
+    // When nodeIdMap is empty, parentNodeId resolution is wasted work — the
+    // worker passes an empty map and the main thread re-resolves against its
+    // own DB-populated map in features/ast.ts::collectFileAstRows. Skip the
+    // findParentDef linear scan in that case.
+    const ctx = {
+        rows: [],
+        matched: new Set(),
+        relPath,
+        defs,
+        nodeIdMap,
+        skipParentLookup: nodeIdMap.size === 0,
+        kindHandlers: buildKindHandlers(newTypes, stringConfig),
+    };
+    return {
+        name: 'ast-store',
+        enterNode(node, _context) {
+            // Guard: skip re-collection but do NOT skipChildren — node.id (memory address)
+            // can be reused by tree-sitter, so a collision would incorrectly suppress an
+            // unrelated subtree. The parent call's skipChildren handles the intended case.
+            if (ctx.matched.has(node.id))
+                return;
+            const kind = resolveAstKind(node, astTypeMap);
+            if (!kind)
+                return;
+            collectNode(ctx, node, kind);
+            // Mirror the native walker's recursion policy. In JS/TS, the native
+            // javascript.rs walker returns after collecting `new` or `throw` to
+            // avoid double-counting the wrapped expression (e.g. `throw new
+            // Error('x')` emits one `throw` row, not throw+new+string). Other
+            // languages go through helpers.rs::walk_ast_nodes_with_config_depth
+            // which always recurses — so `stopRecurseKinds` is empty for them.
+            if (stopRecurseKinds.has(kind)) {
+                return { skipChildren: true };
+            }
+        },
+        finish() {
+            return ctx.rows;
+        },
+    };
+}
+//# sourceMappingURL=ast-store-visitor.js.map

package/dist/ast-analysis/visitors/ast-store-visitor.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"ast-store-visitor.js","sourceRoot":"","sources":["../../../src/ast-analysis/visitors/ast-store-visitor.ts"],"names":[],"mappings":"AASA,MAAM,QAAQ,GAAG,GAAG,CAAC;AAErB,4EAA4E;AAC5E,MAAM,WAAW,GAAG,IAAI,GAAG,CAAS;IAClC,YAAY;IACZ,iBAAiB;IACjB,MAAM;IACN,gBAAgB;IAChB,mBAAmB;IACnB,sBAAsB;IACtB,mBAAmB;IACnB,0BAA0B;IAC1B,kBAAkB;IAClB,WAAW;IACX,wBAAwB;CACzB,CAAC,CAAC;AAEH,MAAM,UAAU,GAAG,IAAI,GAAG,CAAS;IACjC,iBAAiB;IACjB,MAAM;IACN,uBAAuB;IACvB,mBAAmB;IACnB,0BAA0B;IAC1B,wBAAwB;IACxB,wBAAwB;CACzB,CAAC,CAAC;AAEH,MAAM,qBAAqB,GAAoB,EAAE,UAAU,EAAE,MAAM,EAAE,cAAc,EAAE,EAAE,EAAE,CAAC;AAE1F,wEAAwE;AACxE,2EAA2E;AAC3E,0CAA0C;AAC1C,MAAM,wBAAwB,GAAG,IAAI,GAAG,CAAS,CAAC,OAAO,EAAE,OAAO,EAAE,OAAO,EAAE,KAAK,CAAC,CAAC,CAAC;AAYrF,SAAS,QAAQ,CAAC,CAA4B,EAAE,MAAc,QAAQ;IACpE,IAAI,CAAC,CAAC;QAAE,OAAO,IAAI,CAAC;IACpB,OAAO,CAAC,CAAC,MAAM,IAAI,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,GAAG,CAAC,CAAC,GAAG,CAAC;AACzD,CAAC;AAED,SAAS,gBAAgB,CAAC,CAAS,EAAE,KAAa;IAChD,IAAI,CAAC,KAAK;QAAE,OAAO,CAAC,CAAC;IACrB,IAAI,CAAC,GAAG,CAAC,CAAC;IACV,OAAO,CAAC,GAAG,CAAC,CAAC,MAAM,IAAI,KAAK,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAE,CAAC;QAAE,CAAC,EAAE,CAAC;IAClD,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;AAClC,CAAC;AAED,SAAS,iBAAiB,CAAC,CAAS,EAAE,KAAa;IACjD,IAAI,CAAC,KAAK;QAAE,OAAO,CAAC,CAAC;IACrB,IAAI,CAAC,GAAG,CAAC,CAAC,MAAM,CAAC;IACjB,OAAO,CAAC,GAAG,CAAC,IAAI,KAAK,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAE,CAAC;QAAE,CAAC,EAAE,CAAC;IAC/C,OAAO,CAAC,KAAK,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;AAC5C,CAAC;AAED,uFAAuF;AACvF,SAAS,sBAAsB,CAAC,IAAoB;IAClD,KAAK,MAAM,KAAK,IAAI,CAAC,MAAM,EAAE,OAAO,EAAE,aAAa,CAAC,EAAE,CAAC;QACrD,MAAM,CAAC,GAAG,IAAI,CAAC,iBAAiB,CAAC,KAAK,CAAC,CAAC;QACxC,IAAI,CAAC,EAAE,IAAI;YAAE,OAAO,CAAC,CAAC,IAAI,CAAC;IAC7B,CAAC;IACD,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,UAAU,EAAE,CAAC,EAAE,EAAE,CAAC;QACzC,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;QAC5B,IAAI,CAAC,KAAK;YAAE,SAAS;QACrB,IAAI,WAAW,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC;YAAE,OAAO,KAAK,CAAC,IAAI,CAAC;IACrD,CAAC;IACD,MAAM,GAAG,GAAG,IAAI,CAAC,IAAI,IAAI,EAAE,CAAC;IAC5B,MAAM,WAAW,GAAG,GAAG,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,IAAI,GAAG,CAAC;IAC7C,OAAO,WAAW,CAAC,OAAO,CAAC,SAAS,EAAE,EAAE,CAAC,CAAC,IAAI,EAAE,IAAI,GAAG,CAAC;AAC1D,CAAC;AAED,8CAA8C;AAC9C,SAAS,eAAe,CAAC,IAAoB;IAC3C,KAAK,MAAM,KAAK,IAAI,CAAC,UAAU,EAAE,QAAQ,EAAE,MAAM,CAAC,EAAE,CAAC;QACnD,MAAM,CAAC,GAAG,IAAI,CAAC,iBAAiB,CAAC,KAAK,CAAC,CAAC;QACxC,IAAI,CAAC,EAAE,IAAI;YAAE,OAAO,CAAC,CAAC,IAAI,CAAC;IAC7B,CAAC;IACD,MAAM,IAAI,GAAG,IAAI,CAAC,IAAI,IAAI,EAAE,CAAC;IAC7B,OAAO,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,IAAI,GAAG,CAAC;AACnC,CAAC;AAED,yFAAyF;AACzF,SAAS,gBAAgB,CAAC,IAAoB,EAAE,QAAqB;IACnE,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,UAAU,EAAE,CAAC,EAAE,EAAE,CAAC;QACzC,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;QAC5B,IAAI,CAAC,KAAK;YAAE,SAAS;QACrB,MAAM,EAAE,GAAG,KAAK,CAAC,IAAI,CAAC;QACtB,IAAI,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC;YAAE,OAAO,sBAAsB,CAAC,KAAK,CAAC,CAAC;QAC3D,IAAI,UAAU,CAAC,GAAG,CAAC,EAAE,CAAC;YAAE,OAAO,eAAe,CAAC,KAAK,CAAC,CAAC;QACtD,IAAI,WAAW,CAAC,GAAG,CAAC,EAAE,CAAC;YAAE,OAAO,KAAK,CAAC,IAAI,CAAC;IAC7C,CAAC;IACD,OAAO,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,IAAI,CAAC,IAAI,IAAI,EAAE,CAAC;AAChD,CAAC;AAED,qFAAqF;AACrF,SAAS,gBAAgB,CAAC,IAAoB;IAC5C,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,UAAU,EAAE,CAAC,EAAE,EAAE,CAAC;QACzC,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;QAC5B,IAAI,CAAC,KAAK;YAAE,SAAS;QACrB,MAAM,EAAE,GAAG,KAAK,CAAC,IAAI,CAAC;QACtB,IAAI,UAAU,CAAC,GAAG,CAAC,EAAE,CAAC;YAAE,OAAO,eAAe,CAAC,KAAK,CAAC,CAAC;QACtD,IAAI,WAAW,CAAC,GAAG,CAAC,EAAE,CAAC;YAAE,OAAO,KAAK,CAAC,IAAI,CAAC;IAC7C,CAAC;IACD,OAAO,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,IAAI,CAAC,IAAI,IAAI,EAAE,CAAC;AAChD,CAAC;AAED,iFAAiF;AACjF,SAAS,0BAA0B,CAAC,IAAoB;IACtD,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,UAAU,EAAE,CAAC,EAAE,EAAE,CAAC;QACzC,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;QAC5B,IAAI,CAAC,KAAK;YAAE,SAAS;QACrB,IAAI,CAAC,wBAAwB,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC;YAAE,OAAO,QAAQ,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;IAC7E,CAAC;IACD,OAAO,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AAC7B,CAAC;AAED;;;;;;;GAOG;AACH,SAAS,sBAAsB,CAAC,CAAS;IACvC,MAAM,GAAG,GAAG,CAAC,CAAC,MAAM,CAAC;IACrB,IAAI,GAAG,GAAG,CAAC;QAAE,OAAO,KAAK,CAAC;IAC1B,IAAI,GAAG,IAAI,CAAC;QAAE,OAAO,IAAI,CAAC;IAC1B,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,MAAM,IAAI,CAAC,CAAC;AAC5B,CAAC;AAED;;;;GAIG;AACH,SAAS,oBAAoB,CAAC,IAAoB,EAAE,GAAoB;IACtE,MAAM,GAAG,GAAG,IAAI,CAAC,IAAI,IAAI,EAAE,CAAC;IAC5B,MAAM,WAAW,GAAG,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC;IAErD,IAAI,CAAC,GAAG,GAAG,CAAC;IACZ,CAAC,GAAG,gBAAgB,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC;IAC7B,IAAI,GAAG,CAAC,cAAc;QAAE,CAAC,GAAG,gBAAgB,CAAC,CAAC,EAAE,GAAG,CAAC,cAAc,CAAC,CAAC;IACpE,IAAI,WAAW;QAAE,CAAC,GAAG,gBAAgB,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC;IAC/C,CAAC,GAAG,gBAAgB,CAAC,CAAC,EAAE,GAAG,CAAC,UAAU,CAAC,CAAC;IACxC,IAAI,WAAW;QAAE,CAAC,GAAG,iBAAiB,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC;IAC/C,CAAC,GAAG,iBAAiB,CAAC,CAAC,EAAE,GAAG,CAAC,UAAU,CAAC,CAAC;IAEzC,OAAO,sBAAsB,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC;AAC9C,CAAC;AAED,yEAAyE;AACzE,2EAA2E;AAC3E,oBAAoB;AACpB,MAAM,cAAc,GAAG,IAAI,OAAO,EAAuC,CAAC;AAC1E,SAAS,WAAW,CAAC,UAAkC;IACrD,IAAI,CAAC,GAAG,cAAc,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;IACvC,IAAI,CAAC;QAAE,OAAO,CAAC,CAAC;IAChB,CAAC,GAAG,IAAI,GAAG,EAAU,CAAC;IACtB,KAAK,MAAM,IAAI,IAAI,UAAU,EAAE,CAAC;QAC9B,IAAI,UAAU,CAAC,IAAI,CAAC,KAAK,KAAK;YAAE,CAAC,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;IAC9C,CAAC;IACD,cAAc,CAAC,GAAG,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC;IAClC,OAAO,CAAC,CAAC;AACX,CAAC;AAKD,MAAM,wBAAwB,GAAmB,EAAE,IAAI,EAAE,SAAS,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC;AAEjF,gEAAgE;AAChE,SAAS,iBAAiB,CACxB,QAAqB,EACrB,YAA6B;IAE7B,OAAO;QACL,GAAG,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,sBAAsB,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC;QAClF,KAAK,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;YAChB,IAAI,EAAE,gBAAgB,CAAC,IAAI,EAAE,QAAQ,CAAC;YACtC,IAAI,EAAE,0BAA0B,CAAC,IAAI,CAAC;SACvC,CAAC;QACF,KAAK,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,gBAAgB,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,0BAA0B,CAAC,IAAI,CAAC,EAAE,CAAC;QAC3F,MAAM,EAAE,CAAC,IAAI,EAAE,EAAE;YACf,MAAM,OAAO,GAAG,oBAAoB,CAAC,IAAI,EAAE,YAAY,CAAC,CAAC;YACzD,IAAI,OAAO,IAAI,IAAI;gBAAE,OAAO,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC;YACnE,OAAO,EAAE,IAAI,EAAE,QAAQ,CAAC,OAAO,EAAE,GAAG,CAAC,EAAE,IAAI,EAAE,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC;QACrE,CAAC;QACD,KAAK,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,IAAI,GAAG,EAAE,IAAI,EAAE,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC;KACzE,CAAC;AACJ,CAAC;AAED,sEAAsE;AACtE,SAAS,aAAa,CAAC,IAAY,EAAE,IAAkB;IACrD,IAAI,IAAI,GAAsB,IAAI,CAAC;IACnC,KAAK,MAAM,GAAG,IAAI,IAAI,EAAE,CAAC;QACvB,IAAI,GAAG,CAAC,IAAI,IAAI,IAAI,IAAI,CAAC,GAAG,CAAC,OAAO,IAAI,IAAI,IAAI,GAAG,CAAC,OAAO,IAAI,IAAI,CAAC,EAAE,CAAC;YACrE,IAAI,CAAC,IAAI,IAAI,CAAC,GAAG,CAAC,OAAO,IAAI,CAAC,CAAC,GAAG,GAAG,CAAC,IAAI,GAAG,CAAC,IAAI,CAAC,OAAO,IAAI,CAAC,CAAC,GAAG,IAAI,CAAC,IAAI,EAAE,CAAC;gBAC7E,IAAI,GAAG,GAAG,CAAC;YACb,CAAC;QACH,CAAC;IACH,CAAC;IACD,OAAO,IAAI,CAAC;AACd,CAAC;AAED,uEAAuE;AACvE,SAAS,mBAAmB,CAC1B,IAAY,EACZ,IAAkB,EAClB,SAA8B,EAC9B,gBAAyB;IAEzB,IAAI,gBAAgB;QAAE,OAAO,IAAI,CAAC;IAClC,MAAM,SAAS,GAAG,aAAa,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;IAC5C,IAAI,CAAC,SAAS;QAAE,OAAO,IAAI,CAAC;IAC5B,OAAO,SAAS,CAAC,GAAG,CAAC,GAAG,SAAS,CAAC,IAAI,IAAI,SAAS,CAAC,IAAI,IAAI,SAAS,CAAC,IAAI,EAAE,CAAC,IAAI,IAAI,CAAC;AACxF,CAAC;AAYD,SAAS,WAAW,CAAC,GAAe,EAAE,IAAoB,EAAE,IAAY;IACtE,IAAI,GAAG,CAAC,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC;QAAE,OAAO;IAErC,MAAM,OAAO,GAAG,GAAG,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC;IACvC,MAAM,QAAQ,GAAG,OAAO,CAAC,CAAC,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,wBAAwB,CAAC;IACpE,IAAI,QAAQ,CAAC,IAAI;QAAE,OAAO;IAE1B,MAAM,IAAI,GAAG,IAAI,CAAC,aAAa,CAAC,GAAG,GAAG,CAAC,CAAC;IACxC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC;QACZ,IAAI,EAAE,GAAG,CAAC,OAAO;QACjB,IAAI;QACJ,IAAI;QACJ,IAAI,EAAE,QAAQ,CAAC,IAAI;QACnB,IAAI,EAAE,QAAQ,CAAC,IAAI;QACnB,QAAQ,EAAE,IAAI;QACd,YAAY,EAAE,mBAAmB,CAAC,IAAI,EAAE,GAAG,CAAC,IAAI,EAAE,GAAG,CAAC,SAAS,EAAE,GAAG,CAAC,gBAAgB,CAAC;KACvF,CAAC,CAAC;IAEH,GAAG,CAAC,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;AAC3B,CAAC;AAED;;;;;;;;;GASG;AACH,SAAS,cAAc,CAAC,IAAoB,EAAE,UAAkC;IAC9E,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,UAAU,EAAE,IAAI,CAAC,IAAI,CAAC;QAAE,OAAO,IAAI,CAAC;IACvD,OAAO,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,IAAI,CAAC;AACvC,CAAC;AAED,MAAM,UAAU,qBAAqB,CACnC,UAAkC,EAClC,IAAkB,EAClB,OAAe,EACf,SAA8B,EAC9B,eAAgC,qBAAqB,EACrD,mBAAwC,IAAI,GAAG,EAAE;IAEjD,MAAM,QAAQ,GAAG,WAAW,CAAC,UAAU,CAAC,CAAC;IACzC,wEAAwE;IACxE,yEAAyE;IACzE,wEAAwE;IACxE,0CAA0C;IAC1C,MAAM,GAAG,GAAe;QACtB,IAAI,EAAE,EAAE;QACR,OAAO,EAAE,IAAI,GAAG,EAAU;QAC1B,OAAO;QACP,IAAI;QACJ,SAAS;QACT,gBAAgB,EAAE,SAAS,CAAC,IAAI,KAAK,CAAC;QACtC,YAAY,EAAE,iBAAiB,CAAC,QAAQ,EAAE,YAAY,CAAC;KACxD,CAAC;IAEF,OAAO;QACL,IAAI,EAAE,WAAW;QAEjB,SAAS,CAAC,IAAoB,EAAE,QAAwB;YACtD,+EAA+E;YAC/E,6EAA6E;YAC7E,+EAA+E;YAC/E,IAAI,GAAG,CAAC,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC;gBAAE,OAAO;YAErC,MAAM,IAAI,GAAG,cAAc,CAAC,IAAI,EAAE,UAAU,CAAC,CAAC;YAC9C,IAAI,CAAC,IAAI;gBAAE,OAAO;YAElB,WAAW,CAAC,GAAG,EAAE,IAAI,EAAE,IAAI,CAAC,CAAC;YAE7B,oEAAoE;YACpE,oEAAoE;YACpE,gEAAgE;YAChE,kEAAkE;YAClE,oEAAoE;YACpE,mEAAmE;YACnE,IAAI,gBAAgB,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC;gBAC/B,OAAO,EAAE,YAAY,EAAE,IAAI,EAAE,CAAC;YAChC,CAAC;QACH,CAAC;QAED,MAAM;YACJ,OAAO,GAAG,CAAC,IAAI,CAAC;QAClB,CAAC;KACF,CAAC;AACJ,CAAC"}

package/dist/ast-analysis/visitors/cfg-conditionals.d.ts

@@ -0,0 +1,5 @@
+import type { TreeSitterNode } from '../../types.js';
+import type { AnyRules, CfgBlockInternal, FuncState, ProcessStatementsFn } from './cfg-shared.js';
+export declare function processIf(ifStmt: TreeSitterNode, currentBlock: CfgBlockInternal, S: FuncState, cfgRules: AnyRules, processStatements: ProcessStatementsFn): CfgBlockInternal;
+export declare function processSwitch(switchStmt: TreeSitterNode, currentBlock: CfgBlockInternal, S: FuncState, cfgRules: AnyRules, processStatements: ProcessStatementsFn): CfgBlockInternal;
+//# sourceMappingURL=cfg-conditionals.d.ts.map

package/dist/ast-analysis/visitors/cfg-conditionals.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"cfg-conditionals.d.ts","sourceRoot":"","sources":["../../../src/ast-analysis/visitors/cfg-conditionals.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,gBAAgB,CAAC;AACrD,OAAO,KAAK,EACV,QAAQ,EACR,gBAAgB,EAChB,SAAS,EAET,mBAAmB,EACpB,MAAM,iBAAiB,CAAC;AAGzB,wBAAgB,SAAS,CACvB,MAAM,EAAE,cAAc,EACtB,YAAY,EAAE,gBAAgB,EAC9B,CAAC,EAAE,SAAS,EACZ,QAAQ,EAAE,QAAQ,EAClB,iBAAiB,EAAE,mBAAmB,GACrC,gBAAgB,CA8BlB;AA2GD,wBAAgB,aAAa,CAC3B,UAAU,EAAE,cAAc,EAC1B,YAAY,EAAE,gBAAgB,EAC9B,CAAC,EAAE,SAAS,EACZ,QAAQ,EAAE,QAAQ,EAClB,iBAAiB,EAAE,mBAAmB,GACrC,gBAAgB,CA0ClB"}

package/dist/ast-analysis/visitors/cfg-conditionals.js

@@ -0,0 +1,166 @@
+import { getBodyStatements, isCaseNode, isIfNode, nn } from './cfg-shared.js';
+export function processIf(ifStmt, currentBlock, S, cfgRules, processStatements) {
+    currentBlock.endLine = ifStmt.startPosition.row + 1;
+    const condBlock = S.makeBlock('condition', ifStmt.startPosition.row + 1, ifStmt.startPosition.row + 1, 'if');
+    S.addEdge(currentBlock, condBlock, 'fallthrough');
+    const joinBlock = S.makeBlock('body');
+    const consequentField = cfgRules.ifConsequentField || 'consequence';
+    const consequent = ifStmt.childForFieldName(consequentField);
+    const trueBlock = S.makeBlock('branch_true', null, null, 'then');
+    S.addEdge(condBlock, trueBlock, 'branch_true');
+    const trueStmts = getBodyStatements(consequent, cfgRules);
+    const trueEnd = processStatements(trueStmts, trueBlock, S, cfgRules);
+    if (trueEnd) {
+        S.addEdge(trueEnd, joinBlock, 'fallthrough');
+    }
+    if (cfgRules.elifNode) {
+        processElifSiblings(ifStmt, condBlock, joinBlock, S, cfgRules, processStatements);
+    }
+    else {
+        processAlternative(ifStmt, condBlock, joinBlock, S, cfgRules, processStatements);
+    }
+    return joinBlock;
+}
+function processAlternative(ifStmt, condBlock, joinBlock, S, cfgRules, processStatements) {
+    const alternative = ifStmt.childForFieldName('alternative');
+    if (!alternative) {
+        S.addEdge(condBlock, joinBlock, 'branch_false');
+        return;
+    }
+    if (cfgRules.elseViaAlternative && alternative.type !== cfgRules.elseClause) {
+        if (isIfNode(alternative.type, cfgRules)) {
+            const falseBlock = S.makeBlock('branch_false', null, null, 'else-if');
+            S.addEdge(condBlock, falseBlock, 'branch_false');
+            const elseIfEnd = processIf(alternative, falseBlock, S, cfgRules, processStatements);
+            if (elseIfEnd)
+                S.addEdge(elseIfEnd, joinBlock, 'fallthrough');
+        }
+        else {
+            const falseBlock = S.makeBlock('branch_false', null, null, 'else');
+            S.addEdge(condBlock, falseBlock, 'branch_false');
+            const falseStmts = getBodyStatements(alternative, cfgRules);
+            const falseEnd = processStatements(falseStmts, falseBlock, S, cfgRules);
+            if (falseEnd)
+                S.addEdge(falseEnd, joinBlock, 'fallthrough');
+        }
+    }
+    else if (alternative.type === cfgRules.elseClause) {
+        const elseChildren = [];
+        for (let i = 0; i < alternative.namedChildCount; i++) {
+            elseChildren.push(nn(alternative.namedChild(i)));
+        }
+        const firstChild = elseChildren[0];
+        if (elseChildren.length === 1 && firstChild && isIfNode(firstChild.type, cfgRules)) {
+            const falseBlock = S.makeBlock('branch_false', null, null, 'else-if');
+            S.addEdge(condBlock, falseBlock, 'branch_false');
+            const elseIfEnd = processIf(firstChild, falseBlock, S, cfgRules, processStatements);
+            if (elseIfEnd)
+                S.addEdge(elseIfEnd, joinBlock, 'fallthrough');
+        }
+        else {
+            const falseBlock = S.makeBlock('branch_false', null, null, 'else');
+            S.addEdge(condBlock, falseBlock, 'branch_false');
+            const falseEnd = processStatements(elseChildren, falseBlock, S, cfgRules);
+            if (falseEnd)
+                S.addEdge(falseEnd, joinBlock, 'fallthrough');
+        }
+    }
+}
+function processElifSiblings(ifStmt, firstCondBlock, joinBlock, S, cfgRules, processStatements) {
+    let lastCondBlock = firstCondBlock;
+    let foundElse = false;
+    for (let i = 0; i < ifStmt.namedChildCount; i++) {
+        const child = nn(ifStmt.namedChild(i));
+        if (child.type === cfgRules.elifNode) {
+            const elifCondBlock = S.makeBlock('condition', child.startPosition.row + 1, child.startPosition.row + 1, 'else-if');
+            S.addEdge(lastCondBlock, elifCondBlock, 'branch_false');
+            const elifConsequentField = cfgRules.ifConsequentField || 'consequence';
+            const elifConsequent = child.childForFieldName(elifConsequentField);
+            const elifTrueBlock = S.makeBlock('branch_true', null, null, 'then');
+            S.addEdge(elifCondBlock, elifTrueBlock, 'branch_true');
+            const elifTrueStmts = getBodyStatements(elifConsequent, cfgRules);
+            const elifTrueEnd = processStatements(elifTrueStmts, elifTrueBlock, S, cfgRules);
+            if (elifTrueEnd)
+                S.addEdge(elifTrueEnd, joinBlock, 'fallthrough');
+            lastCondBlock = elifCondBlock;
+        }
+        else if (child.type === cfgRules.elseClause) {
+            const elseBlock = S.makeBlock('branch_false', null, null, 'else');
+            S.addEdge(lastCondBlock, elseBlock, 'branch_false');
+            const elseBody = child.childForFieldName('body');
+            let elseStmts;
+            if (elseBody) {
+                elseStmts = getBodyStatements(elseBody, cfgRules);
+            }
+            else {
+                elseStmts = [];
+                for (let j = 0; j < child.namedChildCount; j++) {
+                    elseStmts.push(nn(child.namedChild(j)));
+                }
+            }
+            const elseEnd = processStatements(elseStmts, elseBlock, S, cfgRules);
+            if (elseEnd)
+                S.addEdge(elseEnd, joinBlock, 'fallthrough');
+            foundElse = true;
+        }
+    }
+    if (!foundElse) {
+        S.addEdge(lastCondBlock, joinBlock, 'branch_false');
+    }
+}
+export function processSwitch(switchStmt, currentBlock, S, cfgRules, processStatements) {
+    currentBlock.endLine = switchStmt.startPosition.row + 1;
+    const switchHeader = S.makeBlock('condition', switchStmt.startPosition.row + 1, switchStmt.startPosition.row + 1, 'switch');
+    S.addEdge(currentBlock, switchHeader, 'fallthrough');
+    const joinBlock = S.makeBlock('body');
+    const switchCtx = { headerBlock: switchHeader, exitBlock: joinBlock };
+    S.loopStack.push(switchCtx);
+    const switchBody = switchStmt.childForFieldName('body');
+    const container = switchBody || switchStmt;
+    let hasDefault = false;
+    for (let i = 0; i < container.namedChildCount; i++) {
+        const caseClause = nn(container.namedChild(i));
+        const isDefault = caseClause.type === cfgRules.defaultNode;
+        const isCase = isDefault || isCaseNode(caseClause.type, cfgRules);
+        if (!isCase)
+            continue;
+        const caseLabel = isDefault ? 'default' : 'case';
+        const caseBlock = S.makeBlock('case', caseClause.startPosition.row + 1, null, caseLabel);
+        S.addEdge(switchHeader, caseBlock, isDefault ? 'branch_false' : 'branch_true');
+        if (isDefault)
+            hasDefault = true;
+        const caseStmts = extractCaseBody(caseClause, cfgRules);
+        const caseEnd = processStatements(caseStmts, caseBlock, S, cfgRules);
+        if (caseEnd)
+            S.addEdge(caseEnd, joinBlock, 'fallthrough');
+    }
+    if (!hasDefault) {
+        S.addEdge(switchHeader, joinBlock, 'branch_false');
+    }
+    S.loopStack.pop();
+    return joinBlock;
+}
+function extractCaseBody(caseClause, cfgRules) {
+    const caseBodyNode = caseClause.childForFieldName('body') || caseClause.childForFieldName('consequence');
+    if (caseBodyNode) {
+        return getBodyStatements(caseBodyNode, cfgRules);
+    }
+    const stmts = [];
+    const valueNode = caseClause.childForFieldName('value');
+    const patternNode = caseClause.childForFieldName('pattern');
+    for (let j = 0; j < caseClause.namedChildCount; j++) {
+        const child = nn(caseClause.namedChild(j));
+        if (child !== valueNode && child !== patternNode && child.type !== 'switch_label') {
+            if (child.type === 'statement_list') {
+                for (let k = 0; k < child.namedChildCount; k++) {
+                    stmts.push(nn(child.namedChild(k)));
+                }
+            }
+            else {
+                stmts.push(child);
+            }
+        }
+    }
+    return stmts;
+}
+//# sourceMappingURL=cfg-conditionals.js.map

package/dist/ast-analysis/visitors/cfg-conditionals.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"cfg-conditionals.js","sourceRoot":"","sources":["../../../src/ast-analysis/visitors/cfg-conditionals.ts"],"names":[],"mappings":"AAQA,OAAO,EAAE,iBAAiB,EAAE,UAAU,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,iBAAiB,CAAC;AAE9E,MAAM,UAAU,SAAS,CACvB,MAAsB,EACtB,YAA8B,EAC9B,CAAY,EACZ,QAAkB,EAClB,iBAAsC;IAEtC,YAAY,CAAC,OAAO,GAAG,MAAM,CAAC,aAAa,CAAC,GAAG,GAAG,CAAC,CAAC;IAEpD,MAAM,SAAS,GAAG,CAAC,CAAC,SAAS,CAC3B,WAAW,EACX,MAAM,CAAC,aAAa,CAAC,GAAG,GAAG,CAAC,EAC5B,MAAM,CAAC,aAAa,CAAC,GAAG,GAAG,CAAC,EAC5B,IAAI,CACL,CAAC;IACF,CAAC,CAAC,OAAO,CAAC,YAAY,EAAE,SAAS,EAAE,aAAa,CAAC,CAAC;IAElD,MAAM,SAAS,GAAG,CAAC,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;IAEtC,MAAM,eAAe,GAAG,QAAQ,CAAC,iBAAiB,IAAI,aAAa,CAAC;IACpE,MAAM,UAAU,GAAG,MAAM,CAAC,iBAAiB,CAAC,eAAe,CAAC,CAAC;IAC7D,MAAM,SAAS,GAAG,CAAC,CAAC,SAAS,CAAC,aAAa,EAAE,IAAI,EAAE,IAAI,EAAE,MAAM,CAAC,CAAC;IACjE,CAAC,CAAC,OAAO,CAAC,SAAS,EAAE,SAAS,EAAE,aAAa,CAAC,CAAC;IAC/C,MAAM,SAAS,GAAG,iBAAiB,CAAC,UAAU,EAAE,QAAQ,CAAC,CAAC;IAC1D,MAAM,OAAO,GAAG,iBAAiB,CAAC,SAAS,EAAE,SAAS,EAAE,CAAC,EAAE,QAAQ,CAAC,CAAC;IACrE,IAAI,OAAO,EAAE,CAAC;QACZ,CAAC,CAAC,OAAO,CAAC,OAAO,EAAE,SAAS,EAAE,aAAa,CAAC,CAAC;IAC/C,CAAC;IAED,IAAI,QAAQ,CAAC,QAAQ,EAAE,CAAC;QACtB,mBAAmB,CAAC,MAAM,EAAE,SAAS,EAAE,SAAS,EAAE,CAAC,EAAE,QAAQ,EAAE,iBAAiB,CAAC,CAAC;IACpF,CAAC;SAAM,CAAC;QACN,kBAAkB,CAAC,MAAM,EAAE,SAAS,EAAE,SAAS,EAAE,CAAC,EAAE,QAAQ,EAAE,iBAAiB,CAAC,CAAC;IACnF,CAAC;IAED,OAAO,SAAS,CAAC;AACnB,CAAC;AAED,SAAS,kBAAkB,CACzB,MAAsB,EACtB,SAA2B,EAC3B,SAA2B,EAC3B,CAAY,EACZ,QAAkB,EAClB,iBAAsC;IAEtC,MAAM,WAAW,GAAG,MAAM,CAAC,iBAAiB,CAAC,aAAa,CAAC,CAAC;IAC5D,IAAI,CAAC,WAAW,EAAE,CAAC;QACjB,CAAC,CAAC,OAAO,CAAC,SAAS,EAAE,SAAS,EAAE,cAAc,CAAC,CAAC;QAChD,OAAO;IACT,CAAC;IAED,IAAI,QAAQ,CAAC,kBAAkB,IAAI,WAAW,CAAC,IAAI,KAAK,QAAQ,CAAC,UAAU,EAAE,CAAC;QAC5E,IAAI,QAAQ,CAAC,WAAW,CAAC,IAAI,EAAE,QAAQ,CAAC,EAAE,CAAC;YACzC,MAAM,UAAU,GAAG,CAAC,CAAC,SAAS,CAAC,cAAc,EAAE,IAAI,EAAE,IAAI,EAAE,SAAS,CAAC,CAAC;YACtE,CAAC,CAAC,OAAO,CAAC,SAAS,EAAE,UAAU,EAAE,cAAc,CAAC,CAAC;YACjD,MAAM,SAAS,GAAG,SAAS,CAAC,WAAW,EAAE,UAAU,EAAE,CAAC,EAAE,QAAQ,EAAE,iBAAiB,CAAC,CAAC;YACrF,IAAI,SAAS;gBAAE,CAAC,CAAC,OAAO,CAAC,SAAS,EAAE,SAAS,EAAE,aAAa,CAAC,CAAC;QAChE,CAAC;aAAM,CAAC;YACN,MAAM,UAAU,GAAG,CAAC,CAAC,SAAS,CAAC,cAAc,EAAE,IAAI,EAAE,IAAI,EAAE,MAAM,CAAC,CAAC;YACnE,CAAC,CAAC,OAAO,CAAC,SAAS,EAAE,UAAU,EAAE,cAAc,CAAC,CAAC;YACjD,MAAM,UAAU,GAAG,iBAAiB,CAAC,WAAW,EAAE,QAAQ,CAAC,CAAC;YAC5D,MAAM,QAAQ,GAAG,iBAAiB,CAAC,UAAU,EAAE,UAAU,EAAE,CAAC,EAAE,QAAQ,CAAC,CAAC;YACxE,IAAI,QAAQ;gBAAE,CAAC,CAAC,OAAO,CAAC,QAAQ,EAAE,SAAS,EAAE,aAAa,CAAC,CAAC;QAC9D,CAAC;IACH,CAAC;SAAM,IAAI,WAAW,CAAC,IAAI,KAAK,QAAQ,CAAC,UAAU,EAAE,CAAC;QACpD,MAAM,YAAY,GAAqB,EAAE,CAAC;QAC1C,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,WAAW,CAAC,eAAe,EAAE,CAAC,EAAE,EAAE,CAAC;YACrD,YAAY,CAAC,IAAI,CAAC,EAAE,CAAC,WAAW,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;QACnD,CAAC;QACD,MAAM,UAAU,GAAG,YAAY,CAAC,CAAC,CAAC,CAAC;QACnC,IAAI,YAAY,CAAC,MAAM,KAAK,CAAC,IAAI,UAAU,IAAI,QAAQ,CAAC,UAAU,CAAC,IAAI,EAAE,QAAQ,CAAC,EAAE,CAAC;YACnF,MAAM,UAAU,GAAG,CAAC,CAAC,SAAS,CAAC,cAAc,EAAE,IAAI,EAAE,IAAI,EAAE,SAAS,CAAC,CAAC;YACtE,CAAC,CAAC,OAAO,CAAC,SAAS,EAAE,UAAU,EAAE,cAAc,CAAC,CAAC;YACjD,MAAM,SAAS,GAAG,SAAS,CAAC,UAAU,EAAE,UAAU,EAAE,CAAC,EAAE,QAAQ,EAAE,iBAAiB,CAAC,CAAC;YACpF,IAAI,SAAS;gBAAE,CAAC,CAAC,OAAO,CAAC,SAAS,EAAE,SAAS,EAAE,aAAa,CAAC,CAAC;QAChE,CAAC;aAAM,CAAC;YACN,MAAM,UAAU,GAAG,CAAC,CAAC,SAAS,CAAC,cAAc,EAAE,IAAI,EAAE,IAAI,EAAE,MAAM,CAAC,CAAC;YACnE,CAAC,CAAC,OAAO,CAAC,SAAS,EAAE,UAAU,EAAE,cAAc,CAAC,CAAC;YACjD,MAAM,QAAQ,GAAG,iBAAiB,CAAC,YAAY,EAAE,UAAU,EAAE,CAAC,EAAE,QAAQ,CAAC,CAAC;YAC1E,IAAI,QAAQ;gBAAE,CAAC,CAAC,OAAO,CAAC,QAAQ,EAAE,SAAS,EAAE,aAAa,CAAC,CAAC;QAC9D,CAAC;IACH,CAAC;AACH,CAAC;AAED,SAAS,mBAAmB,CAC1B,MAAsB,EACtB,cAAgC,EAChC,SAA2B,EAC3B,CAAY,EACZ,QAAkB,EAClB,iBAAsC;IAEtC,IAAI,aAAa,GAAG,cAAc,CAAC;IACnC,IAAI,SAAS,GAAG,KAAK,CAAC;IAEtB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,MAAM,CAAC,eAAe,EAAE,CAAC,EAAE,EAAE,CAAC;QAChD,MAAM,KAAK,GAAG,EAAE,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,CAAC;QAEvC,IAAI,KAAK,CAAC,IAAI,KAAK,QAAQ,CAAC,QAAQ,EAAE,CAAC;YACrC,MAAM,aAAa,GAAG,CAAC,CAAC,SAAS,CAC/B,WAAW,EACX,KAAK,CAAC,aAAa,CAAC,GAAG,GAAG,CAAC,EAC3B,KAAK,CAAC,aAAa,CAAC,GAAG,GAAG,CAAC,EAC3B,SAAS,CACV,CAAC;YACF,CAAC,CAAC,OAAO,CAAC,aAAa,EAAE,aAAa,EAAE,cAAc,CAAC,CAAC;YAExD,MAAM,mBAAmB,GAAG,QAAQ,CAAC,iBAAiB,IAAI,aAAa,CAAC;YACxE,MAAM,cAAc,GAAG,KAAK,CAAC,iBAAiB,CAAC,mBAAmB,CAAC,CAAC;YACpE,MAAM,aAAa,GAAG,CAAC,CAAC,SAAS,CAAC,aAAa,EAAE,IAAI,EAAE,IAAI,EAAE,MAAM,CAAC,CAAC;YACrE,CAAC,CAAC,OAAO,CAAC,aAAa,EAAE,aAAa,EAAE,aAAa,CAAC,CAAC;YACvD,MAAM,aAAa,GAAG,iBAAiB,CAAC,cAAc,EAAE,QAAQ,CAAC,CAAC;YAClE,MAAM,WAAW,GAAG,iBAAiB,CAAC,aAAa,EAAE,aAAa,EAAE,CAAC,EAAE,QAAQ,CAAC,CAAC;YACjF,IAAI,WAAW;gBAAE,CAAC,CAAC,OAAO,CAAC,WAAW,EAAE,SAAS,EAAE,aAAa,CAAC,CAAC;YAElE,aAAa,GAAG,aAAa,CAAC;QAChC,CAAC;aAAM,IAAI,KAAK,CAAC,IAAI,KAAK,QAAQ,CAAC,UAAU,EAAE,CAAC;YAC9C,MAAM,SAAS,GAAG,CAAC,CAAC,SAAS,CAAC,cAAc,EAAE,IAAI,EAAE,IAAI,EAAE,MAAM,CAAC,CAAC;YAClE,CAAC,CAAC,OAAO,CAAC,aAAa,EAAE,SAAS,EAAE,cAAc,CAAC,CAAC;YAEpD,MAAM,QAAQ,GAAG,KAAK,CAAC,iBAAiB,CAAC,MAAM,CAAC,CAAC;YACjD,IAAI,SAA2B,CAAC;YAChC,IAAI,QAAQ,EAAE,CAAC;gBACb,SAAS,GAAG,iBAAiB,CAAC,QAAQ,EAAE,QAAQ,CAAC,CAAC;YACpD,CAAC;iBAAM,CAAC;gBACN,SAAS,GAAG,EAAE,CAAC;gBACf,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,KAAK,CAAC,eAAe,EAAE,CAAC,EAAE,EAAE,CAAC;oBAC/C,SAAS,CAAC,IAAI,CAAC,EAAE,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;gBAC1C,CAAC;YACH,CAAC;YACD,MAAM,OAAO,GAAG,iBAAiB,CAAC,SAAS,EAAE,SAAS,EAAE,CAAC,EAAE,QAAQ,CAAC,CAAC;YACrE,IAAI,OAAO;gBAAE,CAAC,CAAC,OAAO,CAAC,OAAO,EAAE,SAAS,EAAE,aAAa,CAAC,CAAC;YAE1D,SAAS,GAAG,IAAI,CAAC;QACnB,CAAC;IACH,CAAC;IAED,IAAI,CAAC,SAAS,EAAE,CAAC;QACf,CAAC,CAAC,OAAO,CAAC,aAAa,EAAE,SAAS,EAAE,cAAc,CAAC,CAAC;IACtD,CAAC;AACH,CAAC;AAED,MAAM,UAAU,aAAa,CAC3B,UAA0B,EAC1B,YAA8B,EAC9B,CAAY,EACZ,QAAkB,EAClB,iBAAsC;IAEtC,YAAY,CAAC,OAAO,GAAG,UAAU,CAAC,aAAa,CAAC,GAAG,GAAG,CAAC,CAAC;IAExD,MAAM,YAAY,GAAG,CAAC,CAAC,SAAS,CAC9B,WAAW,EACX,UAAU,CAAC,aAAa,CAAC,GAAG,GAAG,CAAC,EAChC,UAAU,CAAC,aAAa,CAAC,GAAG,GAAG,CAAC,EAChC,QAAQ,CACT,CAAC;IACF,CAAC,CAAC,OAAO,CAAC,YAAY,EAAE,YAAY,EAAE,aAAa,CAAC,CAAC;IAErD,MAAM,SAAS,GAAG,CAAC,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;IACtC,MAAM,SAAS,GAAY,EAAE,WAAW,EAAE,YAAY,EAAE,SAAS,EAAE,SAAS,EAAE,CAAC;IAC/E,CAAC,CAAC,SAAS,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;IAE5B,MAAM,UAAU,GAAG,UAAU,CAAC,iBAAiB,CAAC,MAAM,CAAC,CAAC;IACxD,MAAM,SAAS,GAAG,UAAU,IAAI,UAAU,CAAC;IAE3C,IAAI,UAAU,GAAG,KAAK,CAAC;IACvB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,SAAS,CAAC,eAAe,EAAE,CAAC,EAAE,EAAE,CAAC;QACnD,MAAM,UAAU,GAAG,EAAE,CAAC,SAAS,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,CAAC;QAE/C,MAAM,SAAS,GAAG,UAAU,CAAC,IAAI,KAAK,QAAQ,CAAC,WAAW,CAAC;QAC3D,MAAM,MAAM,GAAG,SAAS,IAAI,UAAU,CAAC,UAAU,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC;QAClE,IAAI,CAAC,MAAM;YAAE,SAAS;QAEtB,MAAM,SAAS,GAAG,SAAS,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,MAAM,CAAC;QACjD,MAAM,SAAS,GAAG,CAAC,CAAC,SAAS,CAAC,MAAM,EAAE,UAAU,CAAC,aAAa,CAAC,GAAG,GAAG,CAAC,EAAE,IAAI,EAAE,SAAS,CAAC,CAAC;QACzF,CAAC,CAAC,OAAO,CAAC,YAAY,EAAE,SAAS,EAAE,SAAS,CAAC,CAAC,CAAC,cAAc,CAAC,CAAC,CAAC,aAAa,CAAC,CAAC;QAC/E,IAAI,SAAS;YAAE,UAAU,GAAG,IAAI,CAAC;QAEjC,MAAM,SAAS,GAAG,eAAe,CAAC,UAAU,EAAE,QAAQ,CAAC,CAAC;QACxD,MAAM,OAAO,GAAG,iBAAiB,CAAC,SAAS,EAAE,SAAS,EAAE,CAAC,EAAE,QAAQ,CAAC,CAAC;QACrE,IAAI,OAAO;YAAE,CAAC,CAAC,OAAO,CAAC,OAAO,EAAE,SAAS,EAAE,aAAa,CAAC,CAAC;IAC5D,CAAC;IAED,IAAI,CAAC,UAAU,EAAE,CAAC;QAChB,CAAC,CAAC,OAAO,CAAC,YAAY,EAAE,SAAS,EAAE,cAAc,CAAC,CAAC;IACrD,CAAC;IAED,CAAC,CAAC,SAAS,CAAC,GAAG,EAAE,CAAC;IAClB,OAAO,SAAS,CAAC;AACnB,CAAC;AAED,SAAS,eAAe,CAAC,UAA0B,EAAE,QAAkB;IACrE,MAAM,YAAY,GAChB,UAAU,CAAC,iBAAiB,CAAC,MAAM,CAAC,IAAI,UAAU,CAAC,iBAAiB,CAAC,aAAa,CAAC,CAAC;IACtF,IAAI,YAAY,EAAE,CAAC;QACjB,OAAO,iBAAiB,CAAC,YAAY,EAAE,QAAQ,CAAC,CAAC;IACnD,CAAC;IAED,MAAM,KAAK,GAAqB,EAAE,CAAC;IACnC,MAAM,SAAS,GAAG,UAAU,CAAC,iBAAiB,CAAC,OAAO,CAAC,CAAC;IACxD,MAAM,WAAW,GAAG,UAAU,CAAC,iBAAiB,CAAC,SAAS,CAAC,CAAC;IAC5D,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,UAAU,CAAC,eAAe,EAAE,CAAC,EAAE,EAAE,CAAC;QACpD,MAAM,KAAK,GAAG,EAAE,CAAC,UAAU,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,CAAC;QAC3C,IAAI,KAAK,KAAK,SAAS,IAAI,KAAK,KAAK,WAAW,IAAI,KAAK,CAAC,IAAI,KAAK,cAAc,EAAE,CAAC;YAClF,IAAI,KAAK,CAAC,IAAI,KAAK,gBAAgB,EAAE,CAAC;gBACpC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,KAAK,CAAC,eAAe,EAAE,CAAC,EAAE,EAAE,CAAC;oBAC/C,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;gBACtC,CAAC;YACH,CAAC;iBAAM,CAAC;gBACN,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YACpB,CAAC;QACH,CAAC;IACH,CAAC;IACD,OAAO,KAAK,CAAC;AACf,CAAC"}

package/dist/ast-analysis/visitors/cfg-loops.d.ts

@@ -0,0 +1,7 @@
+import type { TreeSitterNode } from '../../types.js';
+import type { AnyRules, CfgBlockInternal, FuncState, ProcessStatementsFn } from './cfg-shared.js';
+export declare function processForLoop(forStmt: TreeSitterNode, currentBlock: CfgBlockInternal, S: FuncState, cfgRules: AnyRules, processStatements: ProcessStatementsFn): CfgBlockInternal;
+export declare function processWhileLoop(whileStmt: TreeSitterNode, currentBlock: CfgBlockInternal, S: FuncState, cfgRules: AnyRules, processStatements: ProcessStatementsFn): CfgBlockInternal;
+export declare function processDoWhileLoop(doStmt: TreeSitterNode, currentBlock: CfgBlockInternal, S: FuncState, cfgRules: AnyRules, processStatements: ProcessStatementsFn): CfgBlockInternal;
+export declare function processInfiniteLoop(loopStmt: TreeSitterNode, currentBlock: CfgBlockInternal, S: FuncState, cfgRules: AnyRules, processStatements: ProcessStatementsFn): CfgBlockInternal;
+//# sourceMappingURL=cfg-loops.d.ts.map

package/dist/ast-analysis/visitors/cfg-loops.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"cfg-loops.d.ts","sourceRoot":"","sources":["../../../src/ast-analysis/visitors/cfg-loops.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,gBAAgB,CAAC;AACrD,OAAO,KAAK,EACV,QAAQ,EACR,gBAAgB,EAChB,SAAS,EAET,mBAAmB,EACpB,MAAM,iBAAiB,CAAC;AAGzB,wBAAgB,cAAc,CAC5B,OAAO,EAAE,cAAc,EACvB,YAAY,EAAE,gBAAgB,EAC9B,CAAC,EAAE,SAAS,EACZ,QAAQ,EAAE,QAAQ,EAClB,iBAAiB,EAAE,mBAAmB,GACrC,gBAAgB,CAyBlB;AAED,wBAAgB,gBAAgB,CAC9B,SAAS,EAAE,cAAc,EACzB,YAAY,EAAE,gBAAgB,EAC9B,CAAC,EAAE,SAAS,EACZ,QAAQ,EAAE,QAAQ,EAClB,iBAAiB,EAAE,mBAAmB,GACrC,gBAAgB,CAyBlB;AAED,wBAAgB,kBAAkB,CAChC,MAAM,EAAE,cAAc,EACtB,YAAY,EAAE,gBAAgB,EAC9B,CAAC,EAAE,SAAS,EACZ,QAAQ,EAAE,QAAQ,EAClB,iBAAiB,EAAE,mBAAmB,GACrC,gBAAgB,CAqBlB;AAED,wBAAgB,mBAAmB,CACjC,QAAQ,EAAE,cAAc,EACxB,YAAY,EAAE,gBAAgB,EAC9B,CAAC,EAAE,SAAS,EACZ,QAAQ,EAAE,QAAQ,EAClB,iBAAiB,EAAE,mBAAmB,GACrC,gBAAgB,CAwBlB"}

package/dist/ast-analysis/visitors/cfg-loops.js

@@ -0,0 +1,73 @@
+import { getBodyStatements, registerLabelCtx } from './cfg-shared.js';
+export function processForLoop(forStmt, currentBlock, S, cfgRules, processStatements) {
+    const headerBlock = S.makeBlock('loop_header', forStmt.startPosition.row + 1, forStmt.startPosition.row + 1, 'for');
+    S.addEdge(currentBlock, headerBlock, 'fallthrough');
+    const loopExitBlock = S.makeBlock('body');
+    const loopCtx = { headerBlock, exitBlock: loopExitBlock };
+    S.loopStack.push(loopCtx);
+    registerLabelCtx(S, headerBlock, loopExitBlock);
+    const body = forStmt.childForFieldName('body');
+    const bodyBlock = S.makeBlock('loop_body');
+    S.addEdge(headerBlock, bodyBlock, 'branch_true');
+    const bodyStmts = getBodyStatements(body, cfgRules);
+    const bodyEnd = processStatements(bodyStmts, bodyBlock, S, cfgRules);
+    if (bodyEnd)
+        S.addEdge(bodyEnd, headerBlock, 'loop_back');
+    S.addEdge(headerBlock, loopExitBlock, 'loop_exit');
+    S.loopStack.pop();
+    return loopExitBlock;
+}
+export function processWhileLoop(whileStmt, currentBlock, S, cfgRules, processStatements) {
+    const headerBlock = S.makeBlock('loop_header', whileStmt.startPosition.row + 1, whileStmt.startPosition.row + 1, 'while');
+    S.addEdge(currentBlock, headerBlock, 'fallthrough');
+    const loopExitBlock = S.makeBlock('body');
+    const loopCtx = { headerBlock, exitBlock: loopExitBlock };
+    S.loopStack.push(loopCtx);
+    registerLabelCtx(S, headerBlock, loopExitBlock);
+    const body = whileStmt.childForFieldName('body');
+    const bodyBlock = S.makeBlock('loop_body');
+    S.addEdge(headerBlock, bodyBlock, 'branch_true');
+    const bodyStmts = getBodyStatements(body, cfgRules);
+    const bodyEnd = processStatements(bodyStmts, bodyBlock, S, cfgRules);
+    if (bodyEnd)
+        S.addEdge(bodyEnd, headerBlock, 'loop_back');
+    S.addEdge(headerBlock, loopExitBlock, 'loop_exit');
+    S.loopStack.pop();
+    return loopExitBlock;
+}
+export function processDoWhileLoop(doStmt, currentBlock, S, cfgRules, processStatements) {
+    const bodyBlock = S.makeBlock('loop_body', doStmt.startPosition.row + 1, null, 'do');
+    S.addEdge(currentBlock, bodyBlock, 'fallthrough');
+    const condBlock = S.makeBlock('loop_header', null, null, 'do-while');
+    const loopExitBlock = S.makeBlock('body');
+    const loopCtx = { headerBlock: condBlock, exitBlock: loopExitBlock };
+    S.loopStack.push(loopCtx);
+    registerLabelCtx(S, condBlock, loopExitBlock);
+    const body = doStmt.childForFieldName('body');
+    const bodyStmts = getBodyStatements(body, cfgRules);
+    const bodyEnd = processStatements(bodyStmts, bodyBlock, S, cfgRules);
+    if (bodyEnd)
+        S.addEdge(bodyEnd, condBlock, 'fallthrough');
+    S.addEdge(condBlock, bodyBlock, 'loop_back');
+    S.addEdge(condBlock, loopExitBlock, 'loop_exit');
+    S.loopStack.pop();
+    return loopExitBlock;
+}
+export function processInfiniteLoop(loopStmt, currentBlock, S, cfgRules, processStatements) {
+    const headerBlock = S.makeBlock('loop_header', loopStmt.startPosition.row + 1, loopStmt.startPosition.row + 1, 'loop');
+    S.addEdge(currentBlock, headerBlock, 'fallthrough');
+    const loopExitBlock = S.makeBlock('body');
+    const loopCtx = { headerBlock, exitBlock: loopExitBlock };
+    S.loopStack.push(loopCtx);
+    registerLabelCtx(S, headerBlock, loopExitBlock);
+    const body = loopStmt.childForFieldName('body');
+    const bodyBlock = S.makeBlock('loop_body');
+    S.addEdge(headerBlock, bodyBlock, 'branch_true');
+    const bodyStmts = getBodyStatements(body, cfgRules);
+    const bodyEnd = processStatements(bodyStmts, bodyBlock, S, cfgRules);
+    if (bodyEnd)
+        S.addEdge(bodyEnd, headerBlock, 'loop_back');
+    S.loopStack.pop();
+    return loopExitBlock;
+}
+//# sourceMappingURL=cfg-loops.js.map

package/dist/ast-analysis/visitors/cfg-loops.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"cfg-loops.js","sourceRoot":"","sources":["../../../src/ast-analysis/visitors/cfg-loops.ts"],"names":[],"mappings":"AAQA,OAAO,EAAE,iBAAiB,EAAE,gBAAgB,EAAE,MAAM,iBAAiB,CAAC;AAEtE,MAAM,UAAU,cAAc,CAC5B,OAAuB,EACvB,YAA8B,EAC9B,CAAY,EACZ,QAAkB,EAClB,iBAAsC;IAEtC,MAAM,WAAW,GAAG,CAAC,CAAC,SAAS,CAC7B,aAAa,EACb,OAAO,CAAC,aAAa,CAAC,GAAG,GAAG,CAAC,EAC7B,OAAO,CAAC,aAAa,CAAC,GAAG,GAAG,CAAC,EAC7B,KAAK,CACN,CAAC;IACF,CAAC,CAAC,OAAO,CAAC,YAAY,EAAE,WAAW,EAAE,aAAa,CAAC,CAAC;IAEpD,MAAM,aAAa,GAAG,CAAC,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;IAC1C,MAAM,OAAO,GAAY,EAAE,WAAW,EAAE,SAAS,EAAE,aAAa,EAAE,CAAC;IACnE,CAAC,CAAC,SAAS,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IAC1B,gBAAgB,CAAC,CAAC,EAAE,WAAW,EAAE,aAAa,CAAC,CAAC;IAEhD,MAAM,IAAI,GAAG,OAAO,CAAC,iBAAiB,CAAC,MAAM,CAAC,CAAC;IAC/C,MAAM,SAAS,GAAG,CAAC,CAAC,SAAS,CAAC,WAAW,CAAC,CAAC;IAC3C,CAAC,CAAC,OAAO,CAAC,WAAW,EAAE,SAAS,EAAE,aAAa,CAAC,CAAC;IAEjD,MAAM,SAAS,GAAG,iBAAiB,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC;IACpD,MAAM,OAAO,GAAG,iBAAiB,CAAC,SAAS,EAAE,SAAS,EAAE,CAAC,EAAE,QAAQ,CAAC,CAAC;IACrE,IAAI,OAAO;QAAE,CAAC,CAAC,OAAO,CAAC,OAAO,EAAE,WAAW,EAAE,WAAW,CAAC,CAAC;IAE1D,CAAC,CAAC,OAAO,CAAC,WAAW,EAAE,aAAa,EAAE,WAAW,CAAC,CAAC;IACnD,CAAC,CAAC,SAAS,CAAC,GAAG,EAAE,CAAC;IAClB,OAAO,aAAa,CAAC;AACvB,CAAC;AAED,MAAM,UAAU,gBAAgB,CAC9B,SAAyB,EACzB,YAA8B,EAC9B,CAAY,EACZ,QAAkB,EAClB,iBAAsC;IAEtC,MAAM,WAAW,GAAG,CAAC,CAAC,SAAS,CAC7B,aAAa,EACb,SAAS,CAAC,aAAa,CAAC,GAAG,GAAG,CAAC,EAC/B,SAAS,CAAC,aAAa,CAAC,GAAG,GAAG,CAAC,EAC/B,OAAO,CACR,CAAC;IACF,CAAC,CAAC,OAAO,CAAC,YAAY,EAAE,WAAW,EAAE,aAAa,CAAC,CAAC;IAEpD,MAAM,aAAa,GAAG,CAAC,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;IAC1C,MAAM,OAAO,GAAY,EAAE,WAAW,EAAE,SAAS,EAAE,aAAa,EAAE,CAAC;IACnE,CAAC,CAAC,SAAS,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IAC1B,gBAAgB,CAAC,CAAC,EAAE,WAAW,EAAE,aAAa,CAAC,CAAC;IAEhD,MAAM,IAAI,GAAG,SAAS,CAAC,iBAAiB,CAAC,MAAM,CAAC,CAAC;IACjD,MAAM,SAAS,GAAG,CAAC,CAAC,SAAS,CAAC,WAAW,CAAC,CAAC;IAC3C,CAAC,CAAC,OAAO,CAAC,WAAW,EAAE,SAAS,EAAE,aAAa,CAAC,CAAC;IAEjD,MAAM,SAAS,GAAG,iBAAiB,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC;IACpD,MAAM,OAAO,GAAG,iBAAiB,CAAC,SAAS,EAAE,SAAS,EAAE,CAAC,EAAE,QAAQ,CAAC,CAAC;IACrE,IAAI,OAAO;QAAE,CAAC,CAAC,OAAO,CAAC,OAAO,EAAE,WAAW,EAAE,WAAW,CAAC,CAAC;IAE1D,CAAC,CAAC,OAAO,CAAC,WAAW,EAAE,aAAa,EAAE,WAAW,CAAC,CAAC;IACnD,CAAC,CAAC,SAAS,CAAC,GAAG,EAAE,CAAC;IAClB,OAAO,aAAa,CAAC;AACvB,CAAC;AAED,MAAM,UAAU,kBAAkB,CAChC,MAAsB,EACtB,YAA8B,EAC9B,CAAY,EACZ,QAAkB,EAClB,iBAAsC;IAEtC,MAAM,SAAS,GAAG,CAAC,CAAC,SAAS,CAAC,WAAW,EAAE,MAAM,CAAC,aAAa,CAAC,GAAG,GAAG,CAAC,EAAE,IAAI,EAAE,IAAI,CAAC,CAAC;IACrF,CAAC,CAAC,OAAO,CAAC,YAAY,EAAE,SAAS,EAAE,aAAa,CAAC,CAAC;IAElD,MAAM,SAAS,GAAG,CAAC,CAAC,SAAS,CAAC,aAAa,EAAE,IAAI,EAAE,IAAI,EAAE,UAAU,CAAC,CAAC;IACrE,MAAM,aAAa,GAAG,CAAC,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;IAE1C,MAAM,OAAO,GAAY,EAAE,WAAW,EAAE,SAAS,EAAE,SAAS,EAAE,aAAa,EAAE,CAAC;IAC9E,CAAC,CAAC,SAAS,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IAC1B,gBAAgB,CAAC,CAAC,EAAE,SAAS,EAAE,aAAa,CAAC,CAAC;IAE9C,MAAM,IAAI,GAAG,MAAM,CAAC,iBAAiB,CAAC,MAAM,CAAC,CAAC;IAC9C,MAAM,SAAS,GAAG,iBAAiB,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC;IACpD,MAAM,OAAO,GAAG,iBAAiB,CAAC,SAAS,EAAE,SAAS,EAAE,CAAC,EAAE,QAAQ,CAAC,CAAC;IACrE,IAAI,OAAO;QAAE,CAAC,CAAC,OAAO,CAAC,OAAO,EAAE,SAAS,EAAE,aAAa,CAAC,CAAC;IAE1D,CAAC,CAAC,OAAO,CAAC,SAAS,EAAE,SAAS,EAAE,WAAW,CAAC,CAAC;IAC7C,CAAC,CAAC,OAAO,CAAC,SAAS,EAAE,aAAa,EAAE,WAAW,CAAC,CAAC;IAEjD,CAAC,CAAC,SAAS,CAAC,GAAG,EAAE,CAAC;IAClB,OAAO,aAAa,CAAC;AACvB,CAAC;AAED,MAAM,UAAU,mBAAmB,CACjC,QAAwB,EACxB,YAA8B,EAC9B,CAAY,EACZ,QAAkB,EAClB,iBAAsC;IAEtC,MAAM,WAAW,GAAG,CAAC,CAAC,SAAS,CAC7B,aAAa,EACb,QAAQ,CAAC,aAAa,CAAC,GAAG,GAAG,CAAC,EAC9B,QAAQ,CAAC,aAAa,CAAC,GAAG,GAAG,CAAC,EAC9B,MAAM,CACP,CAAC;IACF,CAAC,CAAC,OAAO,CAAC,YAAY,EAAE,WAAW,EAAE,aAAa,CAAC,CAAC;IAEpD,MAAM,aAAa,GAAG,CAAC,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;IAC1C,MAAM,OAAO,GAAY,EAAE,WAAW,EAAE,SAAS,EAAE,aAAa,EAAE,CAAC;IACnE,CAAC,CAAC,SAAS,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IAC1B,gBAAgB,CAAC,CAAC,EAAE,WAAW,EAAE,aAAa,CAAC,CAAC;IAEhD,MAAM,IAAI,GAAG,QAAQ,CAAC,iBAAiB,CAAC,MAAM,CAAC,CAAC;IAChD,MAAM,SAAS,GAAG,CAAC,CAAC,SAAS,CAAC,WAAW,CAAC,CAAC;IAC3C,CAAC,CAAC,OAAO,CAAC,WAAW,EAAE,SAAS,EAAE,aAAa,CAAC,CAAC;IAEjD,MAAM,SAAS,GAAG,iBAAiB,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC;IACpD,MAAM,OAAO,GAAG,iBAAiB,CAAC,SAAS,EAAE,SAAS,EAAE,CAAC,EAAE,QAAQ,CAAC,CAAC;IACrE,IAAI,OAAO;QAAE,CAAC,CAAC,OAAO,CAAC,OAAO,EAAE,WAAW,EAAE,WAAW,CAAC,CAAC;IAE1D,CAAC,CAAC,SAAS,CAAC,GAAG,EAAE,CAAC;IAClB,OAAO,aAAa,CAAC;AACvB,CAAC"}

package/dist/ast-analysis/visitors/cfg-shared.d.ts

@@ -0,0 +1,56 @@
+import type { TreeSitterNode } from '../../types.js';
+export type AnyRules = any;
+/** Callback type for the mutual recursion with processStatements in cfg-visitor. */
+export type ProcessStatementsFn = (stmts: TreeSitterNode[], currentBlock: CfgBlockInternal, S: FuncState, cfgRules: AnyRules) => CfgBlockInternal | null;
+export declare function nn(node: TreeSitterNode | null, context?: string): TreeSitterNode;
+export interface CfgBlockInternal {
+    index: number;
+    type: string;
+    startLine: number | null;
+    endLine: number | null;
+    label: string | null;
+}
+export interface CfgEdgeInternal {
+    sourceIndex: number;
+    targetIndex: number;
+    kind: string;
+}
+export interface LabelCtx {
+    headerBlock: CfgBlockInternal | null;
+    exitBlock: CfgBlockInternal | null;
+}
+export interface LoopCtx {
+    headerBlock: CfgBlockInternal;
+    exitBlock: CfgBlockInternal;
+}
+export interface FuncState {
+    blocks: CfgBlockInternal[];
+    edges: CfgEdgeInternal[];
+    makeBlock(type: string, startLine?: number | null, endLine?: number | null, label?: string | null): CfgBlockInternal;
+    addEdge(source: CfgBlockInternal, target: CfgBlockInternal, kind: string): void;
+    entryBlock: CfgBlockInternal;
+    exitBlock: CfgBlockInternal;
+    currentBlock: CfgBlockInternal | null;
+    loopStack: LoopCtx[];
+    labelMap: Map<string, LabelCtx>;
+    cfgStack: FuncState[];
+    funcNode: TreeSitterNode | null;
+}
+export interface CFGResultInternal {
+    funcNode: TreeSitterNode;
+    blocks: CfgBlockInternal[];
+    edges: CfgEdgeInternal[];
+    cyclomatic: number;
+}
+export declare function isIfNode(type: string, cfgRules: AnyRules): boolean;
+export declare function isForNode(type: string, cfgRules: AnyRules): boolean;
+export declare function isWhileNode(type: string, cfgRules: AnyRules): boolean;
+export declare function isSwitchNode(type: string, cfgRules: AnyRules): boolean;
+export declare function isCaseNode(type: string, cfgRules: AnyRules): boolean;
+export declare function isBlockNode(type: string, cfgRules: AnyRules): boolean;
+export declare function isControlFlow(type: string, cfgRules: AnyRules): boolean;
+export declare function effectiveNode(node: TreeSitterNode, cfgRules: AnyRules): TreeSitterNode;
+export declare function registerLabelCtx(S: FuncState, headerBlock: CfgBlockInternal, exitBlock: CfgBlockInternal): void;
+export declare function getBodyStatements(bodyNode: TreeSitterNode | null, cfgRules: AnyRules): TreeSitterNode[];
+export declare function makeFuncState(): FuncState;
+//# sourceMappingURL=cfg-shared.d.ts.map