resolution

func ExtractCallSites ¶

func ExtractCallSites(filePath string, sourceCode []byte, importMap *core.ImportMap) ([]*core.CallSite, error)

ExtractCallSites extracts all function/method call sites from a Python file. It traverses the AST to find call expressions and builds CallSite objects with caller context, callee information, and arguments.

Algorithm:

1. Parse source code with tree-sitter Python parser
2. Traverse AST to find call expressions
3. For each call, extract: - Caller function/method (containing context) - Callee name (function/method being called) - Arguments (positional and keyword) - Source location (file, line, column)
4. Build CallSite objects for each call

Parameters:

• filePath: absolute path to the Python file being analyzed
• sourceCode: contents of the Python file as byte array
• importMap: import mappings for resolving qualified names

Returns:

• []CallSite: list of all call sites found in the file
• error: if parsing fails or source is invalid

Example:

Source code:
  def process_data():
      result = sanitize(data)
      db.query(result)

Extracts CallSites:
  [
    {Caller: "process_data", Callee: "sanitize", Args: ["data"]},
    {Caller: "process_data", Callee: "db.query", Args: ["result"]}
  ]

func ExtractImports ¶

func ExtractImports(filePath string, sourceCode []byte, registry *core.ModuleRegistry) (*core.ImportMap, error)

ExtractImports extracts all import statements from a Python file and builds an ImportMap. It handles four main import styles:

1. Simple imports: import module
2. From imports: from module import name
3. Aliased imports: from module import name as alias
4. Relative imports: from . import module, from .. import module

The resulting ImportMap maps local names (aliases or imported names) to their fully qualified module paths, enabling later resolution of function calls.

Algorithm:

1. Parse source code with tree-sitter Python parser
2. Traverse AST to find all import statements
3. Process each import to extract module paths and aliases
4. Resolve relative imports using module registry
5. Build ImportMap with resolved fully qualified names

Parameters:

• filePath: absolute path to the Python file being analyzed
• sourceCode: contents of the Python file as byte array
• registry: module registry for resolving module paths and relative imports

Returns:

• ImportMap: map of local names to fully qualified module paths
• error: if parsing fails or source is invalid

Example:

Source code:
  import os
  from myapp.utils import sanitize
  from myapp.db import query as db_query
  from . import helper
  from ..config import settings

Result ImportMap:
  {
    "os": "os",
    "sanitize": "myapp.utils.sanitize",
    "db_query": "myapp.db.query",
    "helper": "myapp.submodule.helper",
    "settings": "myapp.config.settings"
  }

func MergeReturnTypes ¶

func MergeReturnTypes(statements []*ReturnStatement) map[string]*core.TypeInfo

MergeReturnTypes combines multiple return statements for same function. Takes the highest confidence return type.

func ResolveClassInstantiation ¶

func ResolveClassInstantiation(
	callNode *sitter.Node,
	sourceCode []byte,
	modulePath string,
	importMap *core.ImportMap,
	registry *core.ModuleRegistry,
) *core.TypeInfo

ResolveClassInstantiation attempts to resolve class instantiation patterns.