@lzehrung/codegraph 2.1.0 → 2.1.1

package/dist/impact/callCompatibility.js

@@ -7,570 +7,12 @@ import { isJsTsLanguage } from "../languages/js-family.js";
 import { sliceText, toRange } from "../util/ast.js";
 import { fileIdentityKey } from "../util/paths.js";
 import { getCallCompatibilityProvider, getCallCompatibilitySupportedLanguages, registerCallCompatibilityExtractors, } from "./call-compatibility/providers/index.js";
-import { directSignatureParameterNode, findAncestorOfTypes, findFirstDescendantOfTypes, } from "./signature-node-utils.js";
-import { canStartReferenceLookup, recordReferenceLookupOmitted, recordReferenceLookupStarted, } from "./budgets.js";
+import { findOpeningParen, findSignatureOpeningParen, findCommentEnd, findCallOpeningParen, findBalancedParentheses, canStartRegexLiteral, findRegexLiteralEnd, splitTopLevelCommaGroups, } from "./call-compatibility/textScanner.js";
 function supportsCallCompatibilityLanguage(languageId) {
     return getCallCompatibilityProvider(languageId) !== null;
 }
-function findOpeningParen(source, startIndex) {
-    if (startIndex < 0 || startIndex >= source.length) {
-        return -1;
-    }
-    return source.indexOf("(", startIndex);
-}
-function findSignatureOpeningParen(source, startIndex) {
-    if (startIndex < 0 || startIndex >= source.length) {
-        return -1;
-    }
-    let quote = null;
-    let escaped = false;
-    let isTypeAnnotation = false;
-    for (let index = startIndex; index < source.length; index += 1) {
-        const char = source[index];
-        if (quote) {
-            if (escaped) {
-                escaped = false;
-                continue;
-            }
-            if (char === "\\") {
-                escaped = true;
-                continue;
-            }
-            if (char === quote) {
-                quote = null;
-            }
-            continue;
-        }
-        const commentEnd = findCommentEnd(source, index);
-        if (commentEnd !== null) {
-            if (commentEnd < 0) {
-                return -1;
-            }
-            index = commentEnd - 1;
-            continue;
-        }
-        if (char === '"' || char === "'" || char === "`") {
-            quote = char;
-            continue;
-        }
-        if (char === "<") {
-            const closeIndex = findBalancedAngleBrackets(source, index);
-            if (closeIndex < 0) {
-                return -1;
-            }
-            index = closeIndex;
-            continue;
-        }
-        if (char === ":") {
-            isTypeAnnotation = true;
-            continue;
-        }
-        if (char === "=") {
-            if (source[index + 1] === ">") {
-                index += 1;
-                continue;
-            }
-            isTypeAnnotation = false;
-            continue;
-        }
-        if (char === "(") {
-            if (!isTypeAnnotation) {
-                return index;
-            }
-            const balanced = findBalancedParentheses(source, index);
-            if (!balanced) {
-                return -1;
-            }
-            index = balanced.end - 1;
-        }
-    }
-    return -1;
-}
-function findCommentEnd(source, index) {
-    if (source[index] !== "/") {
-        return null;
-    }
-    const nextChar = source[index + 1];
-    if (nextChar === "/") {
-        const newlineIndex = source.indexOf("\n", index + 2);
-        if (newlineIndex < 0) {
-            return source.length;
-        }
-        return newlineIndex;
-    }
-    if (nextChar === "*") {
-        const closeIndex = source.indexOf("*/", index + 2);
-        if (closeIndex < 0) {
-            return -1;
-        }
-        return closeIndex + 2;
-    }
-    return null;
-}
-function findCallOpeningParen(source, startIndex, endIndex) {
-    if (endIndex === undefined) {
-        return findOpeningParen(source, startIndex);
-    }
-    if (endIndex < startIndex || endIndex > source.length) {
-        return -1;
-    }
-    let skippedWhitespace = false;
-    for (let index = endIndex; index < source.length; index += 1) {
-        const char = source[index];
-        if (char === undefined) {
-            return -1;
-        }
-        if (/\s/.test(char)) {
-            skippedWhitespace = true;
-            continue;
-        }
-        if (char === "(") {
-            return index;
-        }
-        if (char === "<") {
-            if (skippedWhitespace) {
-                return -1;
-            }
-            const closeIndex = findBalancedAngleBrackets(source, index);
-            if (closeIndex < 0) {
-                return -1;
-            }
-            for (let afterGeneric = closeIndex + 1; afterGeneric < source.length; afterGeneric += 1) {
-                const nextChar = source[afterGeneric];
-                if (nextChar === undefined) {
-                    return -1;
-                }
-                if (/\s/.test(nextChar)) {
-                    continue;
-                }
-                if (nextChar === "(") {
-                    return afterGeneric;
-                }
-                return -1;
-            }
-            return -1;
-        }
-        return -1;
-    }
-    return -1;
-}
-function findBalancedAngleBrackets(source, openIndex) {
-    if (source[openIndex] !== "<") {
-        return -1;
-    }
-    let depth = 0;
-    let parenDepth = 0;
-    let bracketDepth = 0;
-    let braceDepth = 0;
-    let quote = null;
-    let escaped = false;
-    for (let index = openIndex; index < source.length; index += 1) {
-        const char = source[index];
-        if (quote) {
-            if (escaped) {
-                escaped = false;
-                continue;
-            }
-            if (char === "\\") {
-                escaped = true;
-                continue;
-            }
-            if (char === quote) {
-                quote = null;
-            }
-            continue;
-        }
-        const commentEnd = findCommentEnd(source, index);
-        if (commentEnd !== null) {
-            if (commentEnd < 0) {
-                return -1;
-            }
-            index = commentEnd - 1;
-            continue;
-        }
-        if (char === '"' || char === "'" || char === "`") {
-            quote = char;
-            continue;
-        }
-        if (char === "(") {
-            parenDepth += 1;
-            continue;
-        }
-        if (char === ")") {
-            parenDepth -= 1;
-            if (parenDepth < 0) {
-                return -1;
-            }
-            continue;
-        }
-        if (char === "[") {
-            bracketDepth += 1;
-            continue;
-        }
-        if (char === "]") {
-            bracketDepth -= 1;
-            if (bracketDepth < 0) {
-                return -1;
-            }
-            continue;
-        }
-        if (char === "{") {
-            braceDepth += 1;
-            continue;
-        }
-        if (char === "}") {
-            braceDepth -= 1;
-            if (braceDepth < 0) {
-                return -1;
-            }
-            continue;
-        }
-        const atTopLevel = !parenDepth && !bracketDepth && !braceDepth;
-        if (char === "<" && atTopLevel) {
-            depth += 1;
-            continue;
-        }
-        if (char === ">" && source[index - 1] !== "=" && atTopLevel) {
-            depth -= 1;
-            if (!depth) {
-                return index;
-            }
-            if (depth < 0) {
-                return -1;
-            }
-        }
-    }
-    return -1;
-}
-function findBalancedParentheses(source, openIndex) {
-    if (openIndex < 0 || source[openIndex] !== "(") {
-        return null;
-    }
-    let depth = 0;
-    let quote = null;
-    let escaped = false;
-    for (let index = openIndex; index < source.length; index += 1) {
-        const char = source[index];
-        if (quote) {
-            if (escaped) {
-                escaped = false;
-                continue;
-            }
-            if (char === "\\") {
-                escaped = true;
-                continue;
-            }
-            if (char === quote) {
-                quote = null;
-            }
-            continue;
-        }
-        const commentEnd = findCommentEnd(source, index);
-        if (commentEnd !== null) {
-            if (commentEnd < 0) {
-                return null;
-            }
-            index = commentEnd - 1;
-            continue;
-        }
-        if (char === '"' || char === "'" || char === "`") {
-            quote = char;
-            continue;
-        }
-        if (char === "(") {
-            depth += 1;
-            continue;
-        }
-        if (char === ")") {
-            depth -= 1;
-            if (!depth) {
-                return {
-                    start: openIndex,
-                    end: index + 1,
-                    inner: source.slice(openIndex + 1, index),
-                };
-            }
-            if (depth < 0) {
-                return null;
-            }
-        }
-    }
-    return null;
-}
-function hasTypeContextBeforeLessThan(text, index, groupStart) {
-    const left = text.slice(groupStart, index).trimEnd();
-    if (!left) {
-        return false;
-    }
-    const typeKeywordPattern = /(^|[\s([{,:?=<>|&])(?:as|satisfies|new)\s+[A-Za-z_$][\w$.[\]\s]*$/;
-    return typeKeywordPattern.test(left) || hasOpenTypeAnnotationBefore(text, index, groupStart);
-}
-function hasOpenTypeAnnotationBefore(text, index, groupStart) {
-    let parenDepth = 0;
-    let bracketDepth = 0;
-    let braceDepth = 0;
-    let quote = null;
-    let escaped = false;
-    let hasTypeAnnotation = false;
-    for (let current = groupStart; current < index; current += 1) {
-        const char = text[current];
-        if (quote) {
-            if (escaped) {
-                escaped = false;
-                continue;
-            }
-            if (char === "\\") {
-                escaped = true;
-                continue;
-            }
-            if (char === quote) {
-                quote = null;
-            }
-            continue;
-        }
-        const commentEnd = findCommentEnd(text, current);
-        if (commentEnd !== null) {
-            if (commentEnd < 0) {
-                return false;
-            }
-            current = commentEnd - 1;
-            continue;
-        }
-        if (char === '"' || char === "'" || char === "`") {
-            quote = char;
-            continue;
-        }
-        if (char === "(") {
-            parenDepth += 1;
-            continue;
-        }
-        if (char === ")") {
-            parenDepth -= 1;
-            if (parenDepth < 0) {
-                return false;
-            }
-            continue;
-        }
-        if (char === "[") {
-            bracketDepth += 1;
-            continue;
-        }
-        if (char === "]") {
-            bracketDepth -= 1;
-            if (bracketDepth < 0) {
-                return false;
-            }
-            continue;
-        }
-        if (char === "{") {
-            braceDepth += 1;
-            continue;
-        }
-        if (char === "}") {
-            braceDepth -= 1;
-            if (braceDepth < 0) {
-                return false;
-            }
-            continue;
-        }
-        const atTopLevel = !parenDepth && !bracketDepth && !braceDepth;
-        if (!atTopLevel) {
-            continue;
-        }
-        if (char === ":") {
-            hasTypeAnnotation = true;
-            continue;
-        }
-        if (char === "=" && text[current + 1] !== ">") {
-            hasTypeAnnotation = false;
-        }
-    }
-    return hasTypeAnnotation;
-}
-function previousSignificantIndex(text, index, groupStart) {
-    for (let current = index - 1; current >= groupStart; current -= 1) {
-        const char = text[current];
-        if (char !== undefined && !/\s/.test(char)) {
-            return current;
-        }
-    }
-    return -1;
-}
-function previousSignificantChar(text, index, groupStart) {
-    const previousIndex = previousSignificantIndex(text, index, groupStart);
-    if (previousIndex < 0) {
-        return null;
-    }
-    return text[previousIndex] ?? null;
-}
-function previousSignificantCharBefore(text, index, groupStart) {
-    const previousIndex = previousSignificantIndex(text, index, groupStart);
-    if (previousIndex < 0) {
-        return null;
-    }
-    return previousSignificantChar(text, previousIndex, groupStart);
-}
-function canStartRegexLiteral(text, index, groupStart) {
-    if (text[index] !== "/" || text[index + 1] === "/" || text[index + 1] === "*") {
-        return false;
-    }
-    const previous = previousSignificantChar(text, index, groupStart);
-    if (!previous) {
-        return true;
-    }
-    if (previous === ">" && previousSignificantCharBefore(text, index, groupStart) === "=") {
-        return true;
-    }
-    return "([{,=:+-!*?&|;".includes(previous);
-}
-function findRegexLiteralEnd(text, index) {
-    if (text[index] !== "/" || text[index + 1] === "/" || text[index + 1] === "*") {
-        return null;
-    }
-    let escaped = false;
-    let inCharacterClass = false;
-    for (let current = index + 1; current < text.length; current += 1) {
-        const char = text[current];
-        if (char === "\n" || char === "\r") {
-            return -1;
-        }
-        if (escaped) {
-            escaped = false;
-            continue;
-        }
-        if (char === "\\") {
-            escaped = true;
-            continue;
-        }
-        if (char === "[") {
-            inCharacterClass = true;
-            continue;
-        }
-        if (char === "]") {
-            inCharacterClass = false;
-            continue;
-        }
-        if (char === "/" && !inCharacterClass) {
-            let endIndex = current + 1;
-            while (/[A-Za-z]/.test(text[endIndex] ?? "")) {
-                endIndex += 1;
-            }
-            return endIndex;
-        }
-    }
-    return -1;
-}
-function splitTopLevelCommaGroups(text, angleMode, detectRegexLiterals = true) {
-    const trimmed = text.trim();
-    if (!trimmed) {
-        return [];
-    }
-    const groups = [];
-    let groupStart = 0;
-    let parenDepth = 0;
-    let bracketDepth = 0;
-    let braceDepth = 0;
-    let angleDepth = 0;
-    let quote = null;
-    let escaped = false;
-    for (let index = 0; index < text.length; index += 1) {
-        const char = text[index];
-        if (quote) {
-            if (escaped) {
-                escaped = false;
-                continue;
-            }
-            if (char === "\\") {
-                escaped = true;
-                continue;
-            }
-            if (char === quote) {
-                quote = null;
-            }
-            continue;
-        }
-        const commentEnd = findCommentEnd(text, index);
-        if (commentEnd !== null) {
-            if (commentEnd < 0) {
-                return null;
-            }
-            index = commentEnd - 1;
-            continue;
-        }
-        if (detectRegexLiterals && canStartRegexLiteral(text, index, groupStart)) {
-            const regexEnd = findRegexLiteralEnd(text, index);
-            if (regexEnd === null) {
-                return null;
-            }
-            if (regexEnd < 0) {
-                return null;
-            }
-            index = regexEnd - 1;
-            continue;
-        }
-        if (char === '"' || char === "'" || char === "`") {
-            quote = char;
-            continue;
-        }
-        if (char === "(") {
-            parenDepth += 1;
-            continue;
-        }
-        if (char === ")") {
-            parenDepth -= 1;
-            if (parenDepth < 0) {
-                return null;
-            }
-            continue;
-        }
-        if (char === "[") {
-            bracketDepth += 1;
-            continue;
-        }
-        if (char === "]") {
-            bracketDepth -= 1;
-            if (bracketDepth < 0) {
-                return null;
-            }
-            continue;
-        }
-        if (char === "{") {
-            braceDepth += 1;
-            continue;
-        }
-        if (char === "}") {
-            braceDepth -= 1;
-            if (braceDepth < 0) {
-                return null;
-            }
-            continue;
-        }
-        const atDelimiterTopLevel = !parenDepth && !bracketDepth && !braceDepth;
-        if (char === "<" &&
-            atDelimiterTopLevel &&
-            (angleDepth || angleMode === "always" || hasTypeContextBeforeLessThan(text, index, groupStart))) {
-            angleDepth += 1;
-            continue;
-        }
-        if (char === ">" && angleDepth && text[index - 1] !== "=" && atDelimiterTopLevel) {
-            angleDepth -= 1;
-            continue;
-        }
-        const atTopLevel = atDelimiterTopLevel && !angleDepth;
-        if (char === "," && atTopLevel) {
-            groups.push(text.slice(groupStart, index).trim());
-            groupStart = index + 1;
-        }
-    }
-    if (quote || parenDepth || bracketDepth || braceDepth || angleDepth) {
-        return null;
-    }
-    groups.push(text.slice(groupStart).trim());
-    if (groups.length && groups[groups.length - 1] === "") {
-        groups.pop();
-    }
-    return groups;
-}
+import { directSignatureParameterNode, findAncestorOfTypes, findFirstDescendantOfTypes, } from "./signature-node-utils.js";
+import { canStartReferenceLookup, recordReferenceLookupOmitted, recordReferenceLookupStarted, } from "./budgets.js";
 function referenceScanLimitForCallsites(maxRefs) {
     return Math.max(maxRefs + 50, maxRefs * 4);
 }
@@ -1312,7 +754,7 @@ async function collectVerifiedCallsiteReferences(index, changedSymbol, maxRefs,
         if (!support || !supportsCallCompatibilityLanguage(support.id)) {
             continue;
         }
-        const parsed = await tryEnsureParsedContext(file, index.parsed?.get(fileIdentityKey(file)), diagnostics);
+        const parsed = await tryEnsureParsedContext(file, index.parsed?.get(fileIdentityKey(file)), index.languageExtensions, diagnostics);
         if (!parsed) {
             continue;
         }
@@ -1366,14 +808,11 @@ function classifyCompatibility(expected, actual) {
     }
     return { status: "compatible", reason: "compatible_argument_count" };
 }
-async function tryEnsureParsedContext(file, parsedEntry, diagnostics) {
-    try {
-        return await ensureParsedContext(file, parsedEntry);
-    }
-    catch {
+function tryEnsureParsedContext(file, parsedEntry, languageExtensions, diagnostics) {
+    return ensureParsedContext(file, parsedEntry, languageExtensions).catch(() => {
         incrementSkippedReason(diagnostics, "parse-failed");
         return null;
-    }
+    });
 }
 function incrementSkippedReason(diagnostics, reason) {
     if (!diagnostics) {
@@ -1391,7 +830,7 @@ async function buildCallCompatibilityHintForReference(input) {
     if (calleeStartIndex === undefined) {
         return null;
     }
-    const parsedCallsite = await tryEnsureParsedContext(ref.file, index.parsed?.get(fileIdentityKey(ref.file)), diagnostics);
+    const parsedCallsite = await tryEnsureParsedContext(ref.file, index.parsed?.get(fileIdentityKey(ref.file)), index.languageExtensions, diagnostics);
     if (!parsedCallsite) {
         return null;
     }
@@ -1447,7 +886,7 @@ export async function attachCallCompatibilityHints(index, changedSymbols, option
             recordReferenceLookupOmitted(options.workBudget, 1);
             continue;
         }
-        const parsedDefinition = await tryEnsureParsedContext(changedSymbol.file, index.parsed?.get(fileIdentityKey(changedSymbol.file)), diagnostics);
+        const parsedDefinition = await tryEnsureParsedContext(changedSymbol.file, index.parsed?.get(fileIdentityKey(changedSymbol.file)), index.languageExtensions, diagnostics);
         if (!parsedDefinition) {
             continue;
         }