apisec-code-bolt 0.1.0__py3-none-any.whl

apisec_code_bolt/analysis/call_graph_types.py

@@ -0,0 +1,332 @@
+"""
+Call graph type definitions: enums, data models, and built-in constants.
+
+This module contains all pure data types used across the call graph subsystem.
+It has zero dependencies on other analysis modules.
+"""
+
+from __future__ import annotations
+
+from dataclasses import dataclass, field
+from enum import Enum, auto
+from pathlib import Path
+
+# =============================================================================
+# Call Graph Node and Edge Types
+# =============================================================================
+
+
+class NodeType(Enum):
+    """Types of nodes in the call graph."""
+
+    FUNCTION = auto()  # Top-level function
+    METHOD = auto()  # Class method
+    CONSTRUCTOR = auto()  # __init__ or constructor
+    STATIC_METHOD = auto()  # @staticmethod
+    CLASS_METHOD = auto()  # @classmethod
+    PROPERTY = auto()  # @property
+    LAMBDA = auto()  # Lambda expression
+    CLOSURE = auto()  # Nested function
+    EXTERNAL = auto()  # External library function
+    BUILTIN = auto()  # Built-in function
+    UNKNOWN = auto()  # Could not determine
+
+
+class EdgeType(Enum):
+    """Types of edges (call relationships) in the call graph."""
+
+    DIRECT_CALL = auto()  # foo()
+    METHOD_CALL = auto()  # obj.foo()
+    STATIC_CALL = auto()  # Class.foo()
+    SUPER_CALL = auto()  # super().foo()
+    CONSTRUCTOR_CALL = auto()  # MyClass()
+    CALLBACK = auto()  # foo(callback=bar)
+    DECORATOR = auto()  # @decorator
+    COMPREHENSION = auto()  # [f(x) for x in ...]
+    CONDITIONAL = auto()  # foo() if cond else bar()
+    LAMBDA_CALL = auto()  # (lambda x: x)()
+    RETURN_CALL = auto()  # return func()
+    UNKNOWN = auto()
+
+
+class ResolutionConfidence(Enum):
+    """Confidence level in call resolution."""
+
+    EXACT = auto()  # Statically determined, single target
+    HIGH = auto()  # Single likely target with good evidence
+    MEDIUM = auto()  # Few possible targets
+    LOW = auto()  # Many possible targets or weak evidence
+    UNRESOLVED = auto()  # Could not resolve at all
+
+
+class CallContext(Enum):
+    """Syntactic context where a call occurs."""
+
+    NORMAL = auto()  # Regular statement
+    LOOP = auto()  # Inside for/while
+    CONDITIONAL = auto()  # Inside if/elif/else
+    TRY_BLOCK = auto()  # Inside try
+    EXCEPT_HANDLER = auto()  # Inside except
+    FINALLY_BLOCK = auto()  # Inside finally
+    WITH_BLOCK = auto()  # Inside with
+    COMPREHENSION = auto()  # Inside list/dict/set comp or genexp
+    LAMBDA = auto()  # Inside lambda body
+    DEFAULT_ARG = auto()  # As default argument value
+    DECORATOR = auto()  # As decorator
+
+
+@dataclass
+class CallGraphNode:
+    """
+    A node in the call graph (a callable entity).
+
+    Represents a function, method, constructor, lambda, or external symbol.
+    """
+
+    # Identity
+    qualified_name: str  # Fully qualified name
+    name: str  # Simple name
+    node_type: NodeType
+
+    # Location (None for external/builtin)
+    file_path: Path | None = None
+    line: int = 0
+    end_line: int = 0
+
+    # For methods
+    class_name: str | None = None
+    class_qualified_name: str | None = None
+
+    # Metadata
+    is_async: bool = False
+    is_generator: bool = False
+    is_abstract: bool = False
+
+    # Parameters (for signature matching)
+    parameters: list[str] = field(default_factory=list)
+    return_type: str | None = None
+
+    # Return value tracking (for closure/lambda returns)
+    returns_callable: bool = False
+    returned_callables: list[str] = field(default_factory=list)
+
+    # Resolution info
+    is_external: bool = False
+    external_module: str | None = None
+
+    # Analysis metadata
+    docstring: str | None = None
+    decorators: list[str] = field(default_factory=list)
+
+
+@dataclass
+class CallGraphEdge:
+    """
+    An edge in the call graph (a call relationship).
+
+    Represents a call from one function to another.
+    """
+
+    # Identity (required fields first)
+    caller: str  # Qualified name of caller
+    callee: str  # Qualified name of callee
+    edge_type: EdgeType
+
+    # Call site location
+    file_path: Path
+    line: int
+
+    # Resolution
+    confidence: ResolutionConfidence
+
+    # Optional fields with defaults
+    column: int = 0
+
+    # If multiple possible targets
+    possible_callees: list[str] = field(default_factory=list)
+
+    # Arguments at call site
+    argument_count: int = 0
+    keyword_arguments: list[str] = field(default_factory=list)
+    has_spread: bool = False
+    has_keyword_spread: bool = False
+
+    # Context (now properly populated)
+    context: CallContext = CallContext.NORMAL
+    in_loop: bool = False
+    in_conditional: bool = False
+    in_try_block: bool = False
+    in_except_handler: bool = False
+    in_comprehension: bool = False
+
+    # For callbacks/higher-order
+    argument_position: int | None = None  # If callee is passed as argument
+    callback_parameter: str | None = None  # Parameter name if callback
+
+    # Provenance (why we think this is the target)
+    resolution_reason: str = ""
+
+
+# =============================================================================
+# Variable Binding Data Model
+# =============================================================================
+
+
+@dataclass
+class TypeBinding:
+    """
+    A binding between a variable and possible types.
+
+    Used for resolving dynamic dispatch: if we see x.foo(), we need to know
+    what types x could be to resolve which foo() is called.
+    """
+
+    variable: str
+    possible_types: set[str] = field(default_factory=set)
+
+    # Where this binding was established
+    assignment_locations: list[tuple[Path, int]] = field(default_factory=list)
+
+    # Is this a parameter? (could be any type that matches signature)
+    is_parameter: bool = False
+    parameter_type_annotation: str | None = None
+
+    # Is this from an import?
+    is_import: bool = False
+    import_source: str | None = None
+
+    # For transitive propagation: variables this was assigned from
+    source_variables: set[str] = field(default_factory=set)
+
+    # Does this hold a callable (function/lambda)?
+    holds_callable: bool = False
+    callable_targets: set[str] = field(default_factory=set)
+
+    # Is this self/cls?
+    is_self: bool = False
+    is_cls: bool = False
+    enclosing_class: str | None = None
+
+
+# =============================================================================
+# Python Builtins
+# =============================================================================
+
+
+PYTHON_BUILTINS = {
+    "print",
+    "len",
+    "range",
+    "str",
+    "int",
+    "float",
+    "list",
+    "dict",
+    "set",
+    "tuple",
+    "bool",
+    "type",
+    "isinstance",
+    "issubclass",
+    "hasattr",
+    "getattr",
+    "setattr",
+    "delattr",
+    "callable",
+    "iter",
+    "next",
+    "enumerate",
+    "zip",
+    "map",
+    "filter",
+    "sorted",
+    "reversed",
+    "min",
+    "max",
+    "sum",
+    "any",
+    "all",
+    "abs",
+    "round",
+    "divmod",
+    "pow",
+    "open",
+    "input",
+    "repr",
+    "ascii",
+    "format",
+    "hash",
+    "id",
+    "hex",
+    "oct",
+    "bin",
+    "chr",
+    "ord",
+    "bytes",
+    "bytearray",
+    "memoryview",
+    "object",
+    "super",
+    "staticmethod",
+    "classmethod",
+    "property",
+    "vars",
+    "dir",
+    "locals",
+    "globals",
+    "eval",
+    "exec",
+    "compile",
+    "slice",
+    "frozenset",
+    "complex",
+    # Common exceptions
+    "Exception",
+    "BaseException",
+    "ValueError",
+    "TypeError",
+    "KeyError",
+    "IndexError",
+    "AttributeError",
+    "RuntimeError",
+    "StopIteration",
+    "GeneratorExit",
+    "AssertionError",
+    "ImportError",
+    "ModuleNotFoundError",
+    "FileNotFoundError",
+    "OSError",
+    "IOError",
+    "PermissionError",
+    "TimeoutError",
+    "ConnectionError",
+    "BrokenPipeError",
+    "ChildProcessError",
+    "NotImplementedError",
+    "OverflowError",
+    "RecursionError",
+    "ZeroDivisionError",
+    "FloatingPointError",
+    "ArithmeticError",
+    "LookupError",
+    "SyntaxError",
+    "IndentationError",
+    "TabError",
+    "SystemError",
+    "SystemExit",
+    "KeyboardInterrupt",
+    "MemoryError",
+    "BufferError",
+    "Warning",
+    "UserWarning",
+    "DeprecationWarning",
+    "PendingDeprecationWarning",
+    "RuntimeWarning",
+    "SyntaxWarning",
+    "ResourceWarning",
+    "FutureWarning",
+    "ImportWarning",
+    "UnicodeWarning",
+    "BytesWarning",
+    "EncodingWarning",
+}