arceus-s 1.6.4

package/dist/core/ingestion/type-extractors/jvm.js

@@ -0,0 +1,856 @@
+import { findChild } from '../utils/ast-helpers.js';
+import { extractSimpleTypeName, extractVarName, extractGenericTypeArgs, resolveIterableElementType, methodToTypeArgPosition, extractElementTypeFromString, } from './shared.js';
+// ── Java ──────────────────────────────────────────────────────────────────
+const JAVA_DECLARATION_NODE_TYPES = new Set([
+    'local_variable_declaration',
+    'field_declaration',
+]);
+/** Java: Type x = ...; Type x; */
+const extractJavaDeclaration = (node, env) => {
+    const typeNode = node.childForFieldName('type');
+    if (!typeNode)
+        return;
+    const typeName = extractSimpleTypeName(typeNode);
+    if (!typeName || typeName === 'var')
+        return; // skip Java 10 var — handled by extractInitializer
+    // Find variable_declarator children
+    for (let i = 0; i < node.namedChildCount; i++) {
+        const child = node.namedChild(i);
+        if (child?.type !== 'variable_declarator')
+            continue;
+        const nameNode = child.childForFieldName('name');
+        if (nameNode) {
+            const varName = extractVarName(nameNode);
+            if (varName)
+                env.set(varName, typeName);
+        }
+    }
+};
+/** Java 10+: var x = new User() — infer type from object_creation_expression */
+const extractJavaInitializer = (node, env, _classNames) => {
+    for (let i = 0; i < node.namedChildCount; i++) {
+        const child = node.namedChild(i);
+        if (child?.type !== 'variable_declarator')
+            continue;
+        const nameNode = child.childForFieldName('name');
+        const valueNode = child.childForFieldName('value');
+        if (!nameNode || !valueNode)
+            continue;
+        // Skip declarators that already have a binding from extractDeclaration
+        const varName = extractVarName(nameNode);
+        if (!varName || env.has(varName))
+            continue;
+        if (valueNode.type !== 'object_creation_expression')
+            continue;
+        const ctorType = valueNode.childForFieldName('type');
+        if (!ctorType)
+            continue;
+        const typeName = extractSimpleTypeName(ctorType);
+        if (typeName)
+            env.set(varName, typeName);
+    }
+};
+/** Java: formal_parameter → type name */
+const extractJavaParameter = (node, env) => {
+    let nameNode = null;
+    let typeNode = null;
+    if (node.type === 'formal_parameter') {
+        typeNode = node.childForFieldName('type');
+        nameNode = node.childForFieldName('name');
+    }
+    else {
+        // Generic fallback
+        nameNode = node.childForFieldName('name') ?? node.childForFieldName('pattern');
+        typeNode = node.childForFieldName('type');
+    }
+    if (!nameNode || !typeNode)
+        return;
+    const varName = extractVarName(nameNode);
+    const typeName = extractSimpleTypeName(typeNode);
+    if (varName && typeName)
+        env.set(varName, typeName);
+};
+/** Java: var x = SomeFactory.create() — constructor binding for `var` with method_invocation */
+const scanJavaConstructorBinding = (node) => {
+    if (node.type !== 'local_variable_declaration')
+        return undefined;
+    const typeNode = node.childForFieldName('type');
+    if (!typeNode)
+        return undefined;
+    if (typeNode.text !== 'var')
+        return undefined;
+    const declarator = findChild(node, 'variable_declarator');
+    if (!declarator)
+        return undefined;
+    const nameNode = declarator.childForFieldName('name');
+    const value = declarator.childForFieldName('value');
+    if (!nameNode || !value)
+        return undefined;
+    if (value.type === 'object_creation_expression')
+        return undefined;
+    if (value.type !== 'method_invocation')
+        return undefined;
+    const methodName = value.childForFieldName('name');
+    if (!methodName)
+        return undefined;
+    return { varName: nameNode.text, calleeName: methodName.text };
+};
+const JAVA_FOR_LOOP_NODE_TYPES = new Set(['enhanced_for_statement']);
+/** Extract element type from a Java type annotation AST node.
+ *  Handles generic_type (List<User>), array_type (User[]). */
+const extractJavaElementTypeFromTypeNode = (typeNode, pos = 'last') => {
+    if (typeNode.type === 'generic_type') {
+        const args = extractGenericTypeArgs(typeNode);
+        if (args.length >= 1)
+            return pos === 'first' ? args[0] : args[args.length - 1];
+    }
+    if (typeNode.type === 'array_type') {
+        const elemNode = typeNode.firstNamedChild;
+        if (elemNode)
+            return extractSimpleTypeName(elemNode);
+    }
+    return undefined;
+};
+/** Walk up from a for-each to the enclosing method_declaration and search parameters. */
+const findJavaParamElementType = (iterableName, startNode, pos = 'last') => {
+    let current = startNode.parent;
+    while (current) {
+        if (current.type === 'method_declaration' || current.type === 'constructor_declaration') {
+            const paramsNode = current.childForFieldName('parameters');
+            if (paramsNode) {
+                for (let i = 0; i < paramsNode.namedChildCount; i++) {
+                    const param = paramsNode.namedChild(i);
+                    if (!param || param.type !== 'formal_parameter')
+                        continue;
+                    const nameNode = param.childForFieldName('name');
+                    if (nameNode?.text !== iterableName)
+                        continue;
+                    const typeNode = param.childForFieldName('type');
+                    if (typeNode)
+                        return extractJavaElementTypeFromTypeNode(typeNode, pos);
+                }
+            }
+            break;
+        }
+        current = current.parent;
+    }
+    return undefined;
+};
+/** Java: for (User user : users) — extract loop variable binding.
+ *  Tier 1c: for `for (var user : users)`, resolves element type from iterable. */
+const extractJavaForLoopBinding = (node, { scopeEnv, declarationTypeNodes, scope, returnTypeLookup }) => {
+    const typeNode = node.childForFieldName('type');
+    const nameNode = node.childForFieldName('name');
+    if (!typeNode || !nameNode)
+        return;
+    const varName = extractVarName(nameNode);
+    if (!varName)
+        return;
+    // Explicit type (existing behavior): for (User user : users)
+    const typeName = extractSimpleTypeName(typeNode);
+    if (typeName && typeName !== 'var') {
+        scopeEnv.set(varName, typeName);
+        return;
+    }
+    // Tier 1c: var — resolve from iterable's container type
+    const iterableNode = node.childForFieldName('value');
+    if (!iterableNode)
+        return;
+    let iterableName;
+    let methodName;
+    let callExprElementType;
+    if (iterableNode.type === 'identifier') {
+        iterableName = iterableNode.text;
+    }
+    else if (iterableNode.type === 'field_access') {
+        const field = iterableNode.childForFieldName('field');
+        if (field)
+            iterableName = field.text;
+    }
+    else if (iterableNode.type === 'method_invocation') {
+        // data.keySet() → method_invocation > object: identifier + name: identifier
+        // Also handles this.data.values() → object is field_access, extract inner field name
+        const obj = iterableNode.childForFieldName('object');
+        const name = iterableNode.childForFieldName('name');
+        if (obj?.type === 'identifier') {
+            iterableName = obj.text;
+        }
+        else if (obj?.type === 'field_access') {
+            const innerField = obj.childForFieldName('field');
+            if (innerField)
+                iterableName = innerField.text;
+        }
+        else if (!obj && name) {
+            // Direct function call: for (var u : getUsers()) — no receiver object
+            const rawReturn = returnTypeLookup.lookupRawReturnType(name.text);
+            if (rawReturn)
+                callExprElementType = extractElementTypeFromString(rawReturn);
+        }
+        if (name)
+            methodName = name.text;
+    }
+    if (!iterableName && !callExprElementType)
+        return;
+    let elementType;
+    if (callExprElementType) {
+        elementType = callExprElementType;
+    }
+    else {
+        const containerTypeName = scopeEnv.get(iterableName);
+        const typeArgPos = methodToTypeArgPosition(methodName, containerTypeName);
+        elementType = resolveIterableElementType(iterableName, node, scopeEnv, declarationTypeNodes, scope, extractJavaElementTypeFromTypeNode, findJavaParamElementType, typeArgPos);
+    }
+    if (elementType)
+        scopeEnv.set(varName, elementType);
+};
+/** Java: var alias = u → local_variable_declaration > variable_declarator with name/value */
+const extractJavaPendingAssignment = (node, scopeEnv) => {
+    for (let i = 0; i < node.namedChildCount; i++) {
+        const child = node.namedChild(i);
+        if (!child || child.type !== 'variable_declarator')
+            continue;
+        const nameNode = child.childForFieldName('name');
+        const valueNode = child.childForFieldName('value');
+        if (!nameNode || !valueNode)
+            continue;
+        const lhs = nameNode.text;
+        if (scopeEnv.has(lhs))
+            continue;
+        if (valueNode.type === 'identifier' || valueNode.type === 'simple_identifier')
+            return { kind: 'copy', lhs, rhs: valueNode.text };
+        // field_access RHS → fieldAccess (a.field)
+        if (valueNode.type === 'field_access') {
+            const obj = valueNode.childForFieldName('object');
+            const field = valueNode.childForFieldName('field');
+            if (obj?.type === 'identifier' && field) {
+                return { kind: 'fieldAccess', lhs, receiver: obj.text, field: field.text };
+            }
+        }
+        // method_invocation RHS
+        if (valueNode.type === 'method_invocation') {
+            const objField = valueNode.childForFieldName('object');
+            if (!objField) {
+                // No receiver → callResult
+                const nameField = valueNode.childForFieldName('name');
+                if (nameField?.type === 'identifier') {
+                    return { kind: 'callResult', lhs, callee: nameField.text };
+                }
+            }
+            else if (objField.type === 'identifier') {
+                // With receiver → methodCallResult
+                const nameField = valueNode.childForFieldName('name');
+                if (nameField?.type === 'identifier') {
+                    return { kind: 'methodCallResult', lhs, receiver: objField.text, method: nameField.text };
+                }
+            }
+        }
+    }
+    return undefined;
+};
+/**
+ * Java 16+ `instanceof` pattern variable: `x instanceof User user`
+ *
+ * AST structure:
+ *   instanceof_expression
+ *     left: expression (the variable being tested)
+ *     instanceof keyword
+ *     right: type (the type to test against)
+ *     name: identifier (the pattern variable — optional, Java 16+)
+ *
+ * Conservative: returns undefined when the `name` field is absent (plain instanceof
+ * without pattern variable, e.g. `x instanceof User`) or when the type cannot be
+ * extracted. The source variable's existing type is NOT used — the pattern explicitly
+ * declares the new type, so no scopeEnv lookup is needed.
+ */
+const extractJavaPatternBinding = (node) => {
+    if (node.type === 'type_pattern') {
+        // Java 17+ switch pattern: case User u -> ...
+        // type_pattern has positional children (NO named fields):
+        //   namedChild(0) = type (type_identifier, e.g., User)
+        //   namedChild(1) = identifier (e.g., u)
+        const typeNode = node.namedChild(0);
+        const nameNode = node.namedChild(1);
+        if (!typeNode || !nameNode)
+            return undefined;
+        const typeName = extractSimpleTypeName(typeNode);
+        const varName = extractVarName(nameNode);
+        if (!typeName || !varName)
+            return undefined;
+        return { varName, typeName };
+    }
+    if (node.type !== 'instanceof_expression')
+        return undefined;
+    const nameNode = node.childForFieldName('name');
+    if (!nameNode)
+        return undefined;
+    const typeNode = node.childForFieldName('right');
+    if (!typeNode)
+        return undefined;
+    const typeName = extractSimpleTypeName(typeNode);
+    const varName = extractVarName(nameNode);
+    if (!typeName || !varName)
+        return undefined;
+    return { varName, typeName };
+};
+/** Infer the type of a literal AST node for Java/Kotlin overload disambiguation. */
+const inferJvmLiteralType = (node) => {
+    switch (node.type) {
+        case 'decimal_integer_literal':
+        case 'integer_literal':
+        case 'hex_integer_literal':
+        case 'octal_integer_literal':
+        case 'binary_integer_literal':
+            // Check for long suffix
+            if (node.text.endsWith('L') || node.text.endsWith('l'))
+                return 'long';
+            return 'int';
+        case 'decimal_floating_point_literal':
+        case 'real_literal':
+            if (node.text.endsWith('f') || node.text.endsWith('F'))
+                return 'float';
+            return 'double';
+        case 'string_literal':
+        case 'line_string_literal':
+        case 'multi_line_string_literal':
+            return 'String';
+        case 'character_literal':
+            return 'char';
+        case 'true':
+        case 'false':
+        case 'boolean_literal':
+            return 'boolean';
+        case 'null_literal':
+            return 'null';
+        default:
+            return undefined;
+    }
+};
+export const javaTypeConfig = {
+    declarationNodeTypes: JAVA_DECLARATION_NODE_TYPES,
+    forLoopNodeTypes: JAVA_FOR_LOOP_NODE_TYPES,
+    patternBindingNodeTypes: new Set(['instanceof_expression', 'type_pattern']),
+    extractDeclaration: extractJavaDeclaration,
+    extractParameter: extractJavaParameter,
+    extractInitializer: extractJavaInitializer,
+    scanConstructorBinding: scanJavaConstructorBinding,
+    extractForLoopBinding: extractJavaForLoopBinding,
+    extractPendingAssignment: extractJavaPendingAssignment,
+    extractPatternBinding: extractJavaPatternBinding,
+    inferLiteralType: inferJvmLiteralType,
+};
+// ── Kotlin ────────────────────────────────────────────────────────────────
+const KOTLIN_DECLARATION_NODE_TYPES = new Set([
+    'property_declaration',
+    'variable_declaration',
+]);
+/** Kotlin: val x: Foo = ... */
+const extractKotlinDeclaration = (node, env) => {
+    if (node.type === 'property_declaration') {
+        // Kotlin property_declaration: name/type are inside a variable_declaration child
+        const varDecl = findChild(node, 'variable_declaration');
+        if (varDecl) {
+            const nameNode = findChild(varDecl, 'simple_identifier');
+            const typeNode = findChild(varDecl, 'user_type') ?? findChild(varDecl, 'nullable_type');
+            if (!nameNode || !typeNode)
+                return;
+            const varName = extractVarName(nameNode);
+            const typeName = extractSimpleTypeName(typeNode);
+            if (varName && typeName)
+                env.set(varName, typeName);
+            return;
+        }
+        // Fallback: try direct fields
+        const nameNode = node.childForFieldName('name') ?? findChild(node, 'simple_identifier');
+        const typeNode = node.childForFieldName('type') ?? findChild(node, 'user_type');
+        if (!nameNode || !typeNode)
+            return;
+        const varName = extractVarName(nameNode);
+        const typeName = extractSimpleTypeName(typeNode);
+        if (varName && typeName)
+            env.set(varName, typeName);
+    }
+    else if (node.type === 'variable_declaration') {
+        // variable_declaration directly inside functions
+        const nameNode = findChild(node, 'simple_identifier');
+        const typeNode = findChild(node, 'user_type');
+        if (nameNode && typeNode) {
+            const varName = extractVarName(nameNode);
+            const typeName = extractSimpleTypeName(typeNode);
+            if (varName && typeName)
+                env.set(varName, typeName);
+        }
+    }
+};
+/** Kotlin: parameter / formal_parameter → type name.
+ *  Kotlin's tree-sitter grammar uses positional children (simple_identifier, user_type)
+ *  rather than named fields (name, type) on `parameter` nodes, so we fall back to
+ *  findChild when childForFieldName returns null. */
+const extractKotlinParameter = (node, env) => {
+    let nameNode = null;
+    let typeNode = null;
+    if (node.type === 'formal_parameter') {
+        typeNode = node.childForFieldName('type');
+        nameNode = node.childForFieldName('name');
+    }
+    else {
+        nameNode = node.childForFieldName('name') ?? node.childForFieldName('pattern');
+        typeNode = node.childForFieldName('type');
+    }
+    // Fallback: Kotlin `parameter` nodes use positional children, not named fields
+    if (!nameNode)
+        nameNode = findChild(node, 'simple_identifier');
+    if (!typeNode)
+        typeNode = findChild(node, 'user_type') ?? findChild(node, 'nullable_type');
+    if (!nameNode || !typeNode)
+        return;
+    const varName = extractVarName(nameNode);
+    const typeName = extractSimpleTypeName(typeNode);
+    if (varName && typeName)
+        env.set(varName, typeName);
+};
+/** Find the constructor callee name in a Kotlin property_declaration's initializer.
+ *  Returns the class name if the callee is a verified class constructor, undefined otherwise. */
+const findKotlinConstructorCallee = (node, classNames) => {
+    if (node.type !== 'property_declaration')
+        return undefined;
+    const value = node.childForFieldName('value') ?? findChild(node, 'call_expression');
+    if (!value || value.type !== 'call_expression')
+        return undefined;
+    const callee = value.firstNamedChild;
+    if (!callee || callee.type !== 'simple_identifier')
+        return undefined;
+    const calleeName = callee.text;
+    if (!calleeName || !classNames.has(calleeName))
+        return undefined;
+    return calleeName;
+};
+/** Kotlin: val user = User() — infer type from call_expression when callee is a known class.
+ *  Kotlin constructors are syntactically identical to function calls, so we verify
+ *  against classNames (which may include cross-file SymbolTable lookups). */
+const extractKotlinInitializer = (node, env, classNames) => {
+    // Skip if there's an explicit type annotation — Tier 0 already handled it
+    const varDecl = findChild(node, 'variable_declaration');
+    if (varDecl && findChild(varDecl, 'user_type'))
+        return;
+    const calleeName = findKotlinConstructorCallee(node, classNames);
+    if (!calleeName)
+        return;
+    // Extract the variable name from the variable_declaration inside property_declaration
+    const nameNode = varDecl
+        ? findChild(varDecl, 'simple_identifier')
+        : findChild(node, 'simple_identifier');
+    if (!nameNode)
+        return;
+    const varName = extractVarName(nameNode);
+    if (varName)
+        env.set(varName, calleeName);
+};
+/** Kotlin: detect constructor type from call_expression in typed declarations.
+ *  Unlike extractKotlinInitializer (which SKIPS typed declarations), this detects
+ *  the constructor type EVEN when a type annotation exists, enabling virtual dispatch
+ *  for patterns like `val a: Animal = Dog()`. */
+const detectKotlinConstructorType = (node, classNames) => {
+    return findKotlinConstructorCallee(node, classNames);
+};
+/** Kotlin: val x = User(...) — constructor binding for property_declaration with call_expression */
+const scanKotlinConstructorBinding = (node) => {
+    if (node.type !== 'property_declaration')
+        return undefined;
+    const varDecl = findChild(node, 'variable_declaration');
+    if (!varDecl)
+        return undefined;
+    if (findChild(varDecl, 'user_type'))
+        return undefined;
+    const callExpr = findChild(node, 'call_expression');
+    if (!callExpr)
+        return undefined;
+    const callee = callExpr.firstNamedChild;
+    if (!callee)
+        return undefined;
+    let calleeName;
+    if (callee.type === 'simple_identifier') {
+        calleeName = callee.text;
+    }
+    else if (callee.type === 'navigation_expression') {
+        // Extract method name from qualified call: service.getUser() → getUser
+        const suffix = callee.lastNamedChild;
+        if (suffix?.type === 'navigation_suffix') {
+            const methodName = suffix.lastNamedChild;
+            if (methodName?.type === 'simple_identifier') {
+                calleeName = methodName.text;
+            }
+        }
+    }
+    if (!calleeName)
+        return undefined;
+    const nameNode = findChild(varDecl, 'simple_identifier');
+    if (!nameNode)
+        return undefined;
+    return { varName: nameNode.text, calleeName };
+};
+const KOTLIN_FOR_LOOP_NODE_TYPES = new Set(['for_statement']);
+/** Extract element type from a Kotlin type annotation AST node (user_type wrapping generic).
+ *  Kotlin: user_type → [type_identifier, type_arguments → [type_projection → user_type]]
+ *  Handles the type_projection wrapper that Kotlin uses for generic type arguments. */
+const extractKotlinElementTypeFromTypeNode = (typeNode, pos = 'last') => {
+    if (typeNode.type === 'user_type') {
+        const argsNode = findChild(typeNode, 'type_arguments');
+        if (argsNode && argsNode.namedChildCount >= 1) {
+            const targetArg = pos === 'first'
+                ? argsNode.namedChild(0)
+                : argsNode.namedChild(argsNode.namedChildCount - 1);
+            if (!targetArg)
+                return undefined;
+            // Kotlin wraps type args in type_projection — unwrap to get the inner type
+            const inner = targetArg.type === 'type_projection' ? targetArg.firstNamedChild : targetArg;
+            if (inner)
+                return extractSimpleTypeName(inner);
+        }
+    }
+    return undefined;
+};
+/** Walk up from a for-loop to the enclosing function_declaration and search parameters.
+ *  Kotlin parameters use positional children (simple_identifier, user_type), not named fields. */
+const findKotlinParamElementType = (iterableName, startNode, pos = 'last') => {
+    let current = startNode.parent;
+    while (current) {
+        if (current.type === 'function_declaration') {
+            const paramsNode = findChild(current, 'function_value_parameters');
+            if (paramsNode) {
+                for (let i = 0; i < paramsNode.namedChildCount; i++) {
+                    const param = paramsNode.namedChild(i);
+                    if (!param || param.type !== 'parameter')
+                        continue;
+                    const nameNode = findChild(param, 'simple_identifier');
+                    if (nameNode?.text !== iterableName)
+                        continue;
+                    const typeNode = findChild(param, 'user_type');
+                    if (typeNode)
+                        return extractKotlinElementTypeFromTypeNode(typeNode, pos);
+                }
+            }
+            break;
+        }
+        current = current.parent;
+    }
+    return undefined;
+};
+/** Kotlin: for (user: User in users) — extract loop variable binding.
+ *  Tier 1c: for `for (user in users)` without annotation, resolves from iterable. */
+const extractKotlinForLoopBinding = (node, ctx) => {
+    const { scopeEnv, declarationTypeNodes, scope, returnTypeLookup } = ctx;
+    const varDecl = findChild(node, 'variable_declaration');
+    if (!varDecl)
+        return;
+    const nameNode = findChild(varDecl, 'simple_identifier');
+    if (!nameNode)
+        return;
+    const varName = extractVarName(nameNode);
+    if (!varName)
+        return;
+    // Explicit type annotation (existing behavior): for (user: User in users)
+    const typeNode = findChild(varDecl, 'user_type');
+    if (typeNode) {
+        const typeName = extractSimpleTypeName(typeNode);
+        if (typeName)
+            scopeEnv.set(varName, typeName);
+        return;
+    }
+    // Tier 1c: no annotation — resolve from iterable's container type
+    // Kotlin for-loop children: [variable_declaration, iterable_expr, control_structure_body]
+    // The iterable is the second named child of the for_statement (after variable_declaration)
+    let iterableName;
+    let methodName;
+    let fallbackIterableName;
+    let callExprElementType;
+    let foundVarDecl = false;
+    for (let i = 0; i < node.namedChildCount; i++) {
+        const child = node.namedChild(i);
+        if (child === varDecl) {
+            foundVarDecl = true;
+            continue;
+        }
+        if (!foundVarDecl || !child)
+            continue;
+        if (child.type === 'simple_identifier') {
+            iterableName = child.text;
+            break;
+        }
+        if (child.type === 'navigation_expression') {
+            // data.keys → navigation_expression > simple_identifier(data) + navigation_suffix > simple_identifier(keys)
+            const obj = child.firstNamedChild;
+            const suffix = findChild(child, 'navigation_suffix');
+            const prop = suffix ? findChild(suffix, 'simple_identifier') : null;
+            const hasCallSuffix = suffix ? findChild(suffix, 'call_suffix') !== null : false;
+            // Always try object as iterable + property as method first (handles data.values, data.keys).
+            // For bare property access without call_suffix, also save property as fallback
+            // (handles this.users, repo.items where the property IS the iterable).
+            if (obj?.type === 'simple_identifier')
+                iterableName = obj.text;
+            if (prop)
+                methodName = prop.text;
+            if (!hasCallSuffix && prop) {
+                fallbackIterableName = prop.text;
+            }
+            break;
+        }
+        if (child.type === 'call_expression') {
+            // data.values() → call_expression > navigation_expression > simple_identifier + navigation_suffix
+            const callee = child.firstNamedChild;
+            if (callee?.type === 'navigation_expression') {
+                const obj = callee.firstNamedChild;
+                if (obj?.type === 'simple_identifier')
+                    iterableName = obj.text;
+                const suffix = findChild(callee, 'navigation_suffix');
+                if (suffix) {
+                    const prop = findChild(suffix, 'simple_identifier');
+                    if (prop)
+                        methodName = prop.text;
+                }
+            }
+            else if (callee?.type === 'simple_identifier') {
+                // Direct function call: for (u in getUsers())
+                const rawReturn = returnTypeLookup.lookupRawReturnType(callee.text);
+                if (rawReturn)
+                    callExprElementType = extractElementTypeFromString(rawReturn);
+            }
+            break;
+        }
+    }
+    if (!iterableName && !callExprElementType)
+        return;
+    let elementType;
+    if (callExprElementType) {
+        elementType = callExprElementType;
+    }
+    else {
+        let containerTypeName = scopeEnv.get(iterableName);
+        // Fallback: if object has no type in scope, try the property as the iterable name.
+        // Handles patterns like this.users where the property itself is the iterable variable.
+        if (!containerTypeName && fallbackIterableName) {
+            iterableName = fallbackIterableName;
+            methodName = undefined;
+            containerTypeName = scopeEnv.get(iterableName);
+        }
+        const typeArgPos = methodToTypeArgPosition(methodName, containerTypeName);
+        elementType = resolveIterableElementType(iterableName, node, scopeEnv, declarationTypeNodes, scope, extractKotlinElementTypeFromTypeNode, findKotlinParamElementType, typeArgPos);
+    }
+    if (elementType)
+        scopeEnv.set(varName, elementType);
+};
+/** Kotlin: val alias = u → property_declaration or variable_declaration.
+ *  property_declaration has: binding_pattern_kind("val"), variable_declaration("alias"),
+ *  "=", and the RHS value (simple_identifier "u").
+ *  variable_declaration appears directly inside functions and has simple_identifier children. */
+const extractKotlinPendingAssignment = (node, scopeEnv) => {
+    if (node.type === 'property_declaration') {
+        // Find the variable name from variable_declaration child
+        const varDecl = findChild(node, 'variable_declaration');
+        if (!varDecl)
+            return undefined;
+        const nameNode = varDecl.firstNamedChild;
+        if (!nameNode || nameNode.type !== 'simple_identifier')
+            return undefined;
+        const lhs = nameNode.text;
+        if (scopeEnv.has(lhs))
+            return undefined;
+        // Find the RHS after the "=" token
+        let foundEq = false;
+        for (let i = 0; i < node.childCount; i++) {
+            const child = node.child(i);
+            if (!child)
+                continue;
+            if (child.type === '=') {
+                foundEq = true;
+                continue;
+            }
+            if (foundEq && child.type === 'simple_identifier') {
+                return { kind: 'copy', lhs, rhs: child.text };
+            }
+            // navigation_expression RHS → fieldAccess (a.field)
+            if (foundEq && child.type === 'navigation_expression') {
+                const recv = child.firstNamedChild;
+                const suffix = child.lastNamedChild;
+                const fieldNode = suffix?.type === 'navigation_suffix' ? suffix.lastNamedChild : suffix;
+                if (recv?.type === 'simple_identifier' && fieldNode?.type === 'simple_identifier') {
+                    return { kind: 'fieldAccess', lhs, receiver: recv.text, field: fieldNode.text };
+                }
+            }
+            // call_expression RHS
+            if (foundEq && child.type === 'call_expression') {
+                const calleeNode = child.firstNamedChild;
+                if (calleeNode?.type === 'simple_identifier') {
+                    return { kind: 'callResult', lhs, callee: calleeNode.text };
+                }
+                // navigation_expression callee → methodCallResult (a.method())
+                if (calleeNode?.type === 'navigation_expression') {
+                    const recv = calleeNode.firstNamedChild;
+                    const suffix = calleeNode.lastNamedChild;
+                    const methodNode = suffix?.type === 'navigation_suffix' ? suffix.lastNamedChild : suffix;
+                    if (recv?.type === 'simple_identifier' && methodNode?.type === 'simple_identifier') {
+                        return { kind: 'methodCallResult', lhs, receiver: recv.text, method: methodNode.text };
+                    }
+                }
+            }
+        }
+        return undefined;
+    }
+    if (node.type === 'variable_declaration') {
+        // variable_declaration directly inside functions: simple_identifier children
+        const nameNode = findChild(node, 'simple_identifier');
+        if (!nameNode)
+            return undefined;
+        const lhs = nameNode.text;
+        if (scopeEnv.has(lhs))
+            return undefined;
+        // Look for RHS after "=" in the parent (property_declaration)
+        const parent = node.parent;
+        if (!parent)
+            return undefined;
+        let foundEq = false;
+        for (let i = 0; i < parent.childCount; i++) {
+            const child = parent.child(i);
+            if (!child)
+                continue;
+            if (child.type === '=') {
+                foundEq = true;
+                continue;
+            }
+            if (foundEq && child.type === 'simple_identifier') {
+                return { kind: 'copy', lhs, rhs: child.text };
+            }
+            if (foundEq && child.type === 'navigation_expression') {
+                const recv = child.firstNamedChild;
+                const suffix = child.lastNamedChild;
+                const fieldNode = suffix?.type === 'navigation_suffix' ? suffix.lastNamedChild : suffix;
+                if (recv?.type === 'simple_identifier' && fieldNode?.type === 'simple_identifier') {
+                    return { kind: 'fieldAccess', lhs, receiver: recv.text, field: fieldNode.text };
+                }
+            }
+            if (foundEq && child.type === 'call_expression') {
+                const calleeNode = child.firstNamedChild;
+                if (calleeNode?.type === 'simple_identifier') {
+                    return { kind: 'callResult', lhs, callee: calleeNode.text };
+                }
+                if (calleeNode?.type === 'navigation_expression') {
+                    const recv = calleeNode.firstNamedChild;
+                    const suffix = calleeNode.lastNamedChild;
+                    const methodNode = suffix?.type === 'navigation_suffix' ? suffix.lastNamedChild : suffix;
+                    if (recv?.type === 'simple_identifier' && methodNode?.type === 'simple_identifier') {
+                        return { kind: 'methodCallResult', lhs, receiver: recv.text, method: methodNode.text };
+                    }
+                }
+            }
+        }
+        return undefined;
+    }
+    return undefined;
+};
+/** Walk up from a node to find an ancestor of a given type. */
+const findAncestorByType = (node, type) => {
+    let current = node.parent;
+    while (current) {
+        if (current.type === type)
+            return current;
+        current = current.parent;
+    }
+    return undefined;
+};
+const extractKotlinPatternBinding = (node, scopeEnv, declarationTypeNodes, scope) => {
+    // Kotlin when/is smart casts (existing behavior)
+    if (node.type === 'type_test') {
+        const typeNode = node.lastNamedChild;
+        if (!typeNode)
+            return undefined;
+        const typeName = extractSimpleTypeName(typeNode);
+        if (!typeName)
+            return undefined;
+        const whenExpr = findAncestorByType(node, 'when_expression');
+        if (!whenExpr)
+            return undefined;
+        const whenSubject = whenExpr.namedChild(0);
+        const subject = whenSubject?.firstNamedChild ?? whenSubject;
+        if (!subject)
+            return undefined;
+        const varName = extractVarName(subject);
+        if (!varName)
+            return undefined;
+        return { varName, typeName };
+    }
+    // Null-check narrowing: if (x != null) { ... }
+    // Kotlin AST: equality_expression > simple_identifier, "!=" [anon], "null" [anon]
+    // Note: `null` is an anonymous node in tree-sitter-kotlin, not `null_literal`.
+    if (node.type === 'equality_expression') {
+        const op = node.children.find((c) => !c.isNamed && c.text === '!=');
+        if (!op)
+            return undefined;
+        // `null` is anonymous in Kotlin grammar — use positional child scan
+        let varNode;
+        let hasNull = false;
+        for (let i = 0; i < node.childCount; i++) {
+            const c = node.child(i);
+            if (!c)
+                continue;
+            if (c.type === 'simple_identifier')
+                varNode = c;
+            if (!c.isNamed && c.text === 'null')
+                hasNull = true;
+        }
+        if (!varNode || !hasNull)
+            return undefined;
+        const varName = varNode.text;
+        const resolvedType = scopeEnv.get(varName);
+        if (!resolvedType)
+            return undefined;
+        // Check if the original declaration type was nullable (ends with ?)
+        const declTypeNode = declarationTypeNodes.get(`${scope}\0${varName}`);
+        if (!declTypeNode)
+            return undefined;
+        const declText = declTypeNode.text;
+        if (!declText.includes('?') && !declText.includes('null'))
+            return undefined;
+        // Find the if-body: walk up to if_expression, then find control_structure_body
+        const ifExpr = findAncestorByType(node, 'if_expression');
+        if (!ifExpr)
+            return undefined;
+        // The consequence is the first control_structure_body child
+        for (let i = 0; i < ifExpr.childCount; i++) {
+            const child = ifExpr.child(i);
+            if (child?.type === 'control_structure_body') {
+                return {
+                    varName,
+                    typeName: resolvedType,
+                    narrowingRange: { startIndex: child.startIndex, endIndex: child.endIndex },
+                };
+            }
+        }
+        return undefined;
+    }
+    return undefined;
+};
+export const kotlinTypeConfig = {
+    allowPatternBindingOverwrite: true,
+    declarationNodeTypes: KOTLIN_DECLARATION_NODE_TYPES,
+    getDeclarationTypeNode: (node) => {
+        // Kotlin property_declaration wraps the actual declaration in variable_declaration.
+        // The type is commonly a user_type / nullable_type child (positional, not 'type' field).
+        const varDecl = node.type === 'property_declaration' ? findChild(node, 'variable_declaration') : node;
+        if (varDecl) {
+            return (varDecl.childForFieldName('type') ??
+                findChild(varDecl, 'user_type') ??
+                findChild(varDecl, 'nullable_type'));
+        }
+        return node.childForFieldName('type') ?? findChild(node, 'user_type') ?? null;
+    },
+    forLoopNodeTypes: KOTLIN_FOR_LOOP_NODE_TYPES,
+    patternBindingNodeTypes: new Set(['type_test', 'equality_expression']),
+    extractDeclaration: extractKotlinDeclaration,
+    extractParameter: extractKotlinParameter,
+    extractInitializer: extractKotlinInitializer,
+    scanConstructorBinding: scanKotlinConstructorBinding,
+    extractForLoopBinding: extractKotlinForLoopBinding,
+    extractPendingAssignment: extractKotlinPendingAssignment,
+    extractPatternBinding: extractKotlinPatternBinding,
+    inferLiteralType: inferJvmLiteralType,
+    detectConstructorType: detectKotlinConstructorType,
+};