feat: Implement Type-aware purity analysis and integrate boundary and enum extractor

pyproject.toml

@@ -18,7 +18,9 @@ safe-ds-stubgen = "safeds_stubgen.main:main"
 [tool.poetry.dependencies]
 python = "^3.11"
 mypy = "^1.6.1"
+astroid = "^3.0.0"
 griffe = ">=0.47.0,<0.49"
+spacy = "^3.5.1"
 
 [tool.poetry.group.dev.dependencies]
 pytest = ">=7.4.3,<9.0.0"

src/safeds_stubgen/api_analyzer/__init__.py

@@ -17,6 +17,8 @@
     VarianceKind,
     WildcardImport,
 )
+from safeds_stubgen.api_analyzer._extract_boundary_values import extract_boundary
+from safeds_stubgen.api_analyzer._extract_valid_values import extract_valid_literals
 from ._ast_visitor import result_name_generator
 from ._get_api import get_api
 from ._mypy_helpers import get_classdef_definitions, get_funcdef_definitions, get_mypyfile_definitions
@@ -36,6 +38,7 @@
     TypeVarType,
     UnionType,
     UnknownType,
+    EnumType
 )
 
 __all__ = [
@@ -46,6 +49,9 @@
     "Class",
     "DictType",
     "Enum",
+    "EnumType",
+    "extract_boundary",
+    "extract_valid_literals",
     "FinalType",
     "Function",
     "ListType",

src/safeds_stubgen/api_analyzer/_api.py

@@ -6,6 +6,8 @@
 from enum import Enum as PythonEnum
 from typing import TYPE_CHECKING, Any
 
+from safeds_stubgen.api_analyzer._types import NamedType
+
 if TYPE_CHECKING:
     from pathlib import Path
 
@@ -36,7 +38,7 @@ def ensure_file_exists(file: Path) -> None:
 
 
 class API:
-    def __init__(self, distribution: str, package: str, version: str) -> None:
+    def __init__(self, distribution: str, package: str, version: str, path_to_package: str) -> None:
         self.distribution: str = distribution
         self.package: str = package
         self.version: str = version
@@ -49,15 +51,30 @@ def __init__(self, distribution: str, package: str, version: str) -> None:
         self.attributes_: dict[str, Attribute] = {}
         self.parameters_: dict[str, Parameter] = {}
         self.reexport_map: dict[str, set[Module]] = defaultdict(set)
+        self.path_to_package: str = path_to_package
 
     def add_module(self, module: Module) -> None:
         self.modules[module.id] = module
 
     def add_class(self, class_: Class) -> None:
         self.classes[class_.id] = class_
 
+    def create_astroid_module_path(self, path_str: str) -> str:
+        package_name = self.package 
+        module_path_list = path_str.split(".")
+        index_to_split = module_path_list.index(package_name)
+        module_path = module_path_list[index_to_split:]
+        correct_module_path = ".".join(module_path)
+        return correct_module_path
+
+    def create_astroid_id(self, function: Function) -> str:
+        correct_module_path = self.create_astroid_module_path(function.module_id_which_contains_def)
+        full_id = function.id.split("/")
+        id = full_id[-1]
+        return f"{correct_module_path}.{id}.{function.line}.{function.column}"
+
     def add_function(self, function: Function) -> None:
-        self.functions[function.id] = function
+        self.functions[self.create_astroid_id(function)] = function
 
     def add_enum(self, enum: Enum) -> None:
         self.enums[enum.id] = enum
@@ -168,6 +185,7 @@ class Class:
     id: str
     name: str
     superclasses: list[str]
+    subclasses: list[str]
     is_public: bool
     docstring: ClassDocstring
     constructor: Function | None = None
@@ -235,8 +253,12 @@ def to_dict(self) -> dict[str, Any]:
 @dataclass
 class Function:
     id: str
+    module_id_which_contains_def: str
+    line: int
+    column: int
     name: str
     docstring: FunctionDocstring
+    body: Body
     is_public: bool
     is_static: bool
     is_class_method: bool
@@ -246,6 +268,7 @@ class Function:
     results: list[Result] = field(default_factory=list)
     reexported_by: list[Module] = field(default_factory=list)
     parameters: list[Parameter] = field(default_factory=list)
+    closures: dict[str, Function] = field(default_factory=dict)
 
     def to_dict(self) -> dict[str, Any]:
         return {
@@ -259,8 +282,42 @@ def to_dict(self) -> dict[str, Any]:
             "results": [result.id for result in self.results],
             "reexported_by": [module.id for module in self.reexported_by],
             "parameters": [parameter.id for parameter in self.parameters],
+            # "closures": [closure.to_dict for closure in self.closures],
         }
 
+@dataclass
+class Body:
+    line: int
+    column: int
+    end_line: int | None
+    end_column: int | None
+    call_references: dict[str, CallReference] = field(default_factory=dict)
+
+@dataclass
+class CallReference:
+    receiver: CallReceiver
+    function_name: str
+    line: int
+    column: int
+    possibly_referenced_functions: list[Function] = field(default_factory=list)
+    isSuperCallRef: bool = False
+    reason_for_no_found_functions: str = ""
+    fallbackToSignatureCheck: bool = True
+
+@dataclass
+class CallReceiver:
+    type: Any | NamedType
+    full_name: str
+    path_to_call_reference: list[str]
+    found_classes: list[Class]
+    isFromParameter: bool = False
+    typeThroughTypeHint: bool = False
+    typeThroughDocString: bool = False
+    typeThroughInference: bool = False
+    typeOutsideOfPackage: bool = False
+    decrease: int = 0
+    increase: int = 0
+    missingTypesWhileFindingFunction: bool = False
 
 class UnknownValue:
     pass