fix: update dependencies to be firstparty or thirdparty.

Signed-off-by: Rithul Kamesh <[email protected]>

Author: rithulkamesh

cldk/models/rust/models.py

@@ -1,38 +1,42 @@
 from enum import Enum
-from typing import Dict, List, Optional, Set
+from typing import Dict, List, Optional, Set, ForwardRef
 from pydantic import BaseModel, Field
 
 
 class RustVisibility(Enum):
     """Represents Rust visibility modifiers."""
+
     PUBLIC = "pub"
-    PRIVATE = "" 
+    PRIVATE = ""
     CRATE = "pub(crate)"
     SUPER = "pub(super)"
     IN_PATH = "pub(in path)"
 
 
 class SafetyClassification(Enum):
     """Classifies the safety level of Rust code."""
+
     SAFE = "safe"
     UNSAFE = "unsafe"
-    UNSAFE_CONTAINER = "unsafe_container" 
+    UNSAFE_CONTAINER = "unsafe_container"
     FFI = "ffi"
 
 
 class UnsafeReason(Enum):
     """Reasons why code might be unsafe."""
+
     RAW_POINTER_DEREF = "raw_pointer_deref"
     MUTABLE_STATIC = "mutable_static"
     FFI_CALL = "ffi_call"
     UNION_FIELD_ACCESS = "union_field_access"
     INLINE_ASSEMBLY = "inline_assembly"
     UNSAFE_TRAIT_IMPL = "unsafe_trait_impl"
-    CUSTOM = "custom" 
+    CUSTOM = "custom"
 
 
 class UnsafeBlock(BaseModel):
     """Represents an unsafe block within Rust code."""
+
     start_line: int
     end_line: int
     reasons: List[UnsafeReason] = Field(default_factory=list)
@@ -42,6 +46,7 @@ class UnsafeBlock(BaseModel):
 
 class RustType(BaseModel):
     """Represents a Rust type."""
+
     name: str
     is_reference: bool = False
     is_mutable: bool = False
@@ -55,13 +60,15 @@ class RustType(BaseModel):
 
 class RustAttribute(BaseModel):
     """Represents a Rust attribute."""
+
     name: str
     arguments: List[str] = Field(default_factory=list)
     is_inner: bool = False  # #![foo] vs #[foo]
 
 
 class SafetyAnalysis(BaseModel):
     """Analyzes and tracks safety-related information."""
+
     classification: SafetyClassification
     unsafe_blocks: List[UnsafeBlock] = Field(default_factory=list)
     unsafe_fn_calls: List[str] = Field(default_factory=list)
@@ -74,6 +81,7 @@ class SafetyAnalysis(BaseModel):
 
 class RustCallable(BaseModel):
     """Represents a Rust function or method."""
+
     name: str
     visibility: RustVisibility = RustVisibility.PRIVATE
     doc_comment: Optional[str] = None
@@ -97,23 +105,24 @@ class RustCallable(BaseModel):
     variable_declarations: List["RustVariableDeclaration"] = Field(default_factory=list)
     cyclomatic_complexity: Optional[int] = None
     safety_analysis: SafetyAnalysis
-    
+
     def is_fully_safe(self) -> bool:
         """Check if the function is completely safe (no unsafe blocks or calls)."""
         return (
             self.safety_analysis.classification == SafetyClassification.SAFE
             and not self.safety_analysis.unsafe_blocks
             and not self.safety_analysis.unsafe_fn_calls
         )
-    
+
     def contains_unsafe(self) -> bool:
         """Check if the function contains any unsafe code."""
         return (
-            self.safety_analysis.classification in [SafetyClassification.UNSAFE, SafetyClassification.UNSAFE_CONTAINER]
+            self.safety_analysis.classification
+            in [SafetyClassification.UNSAFE, SafetyClassification.UNSAFE_CONTAINER]
             or bool(self.safety_analysis.unsafe_blocks)
             or bool(self.safety_analysis.unsafe_fn_calls)
         )
-    
+
     def get_unsafe_reasons(self) -> Set[UnsafeReason]:
         """Get all reasons for unsafe usage in this function."""
         reasons = set()
@@ -124,6 +133,7 @@ def get_unsafe_reasons(self) -> Set[UnsafeReason]:
 
 class RustModule(BaseModel):
     """Represents a Rust module."""
+
     name: str
     doc_comment: Optional[str] = None
     attributes: List[RustAttribute] = Field(default_factory=list)
@@ -132,36 +142,47 @@ class RustModule(BaseModel):
     functions: Dict[str, RustCallable] = Field(default_factory=dict)
     safe_functions: Dict[str, RustCallable] = Field(default_factory=dict)
     unsafe_functions: Dict[str, RustCallable] = Field(default_factory=dict)
-    submodules: Dict[str, 'RustModule'] = Field(default_factory=dict)
+    submodules: Dict[str, "RustModule"] = Field(default_factory=dict)
     constants: List["RustVariableDeclaration"] = Field(default_factory=list)
     macros: List[str] = Field(default_factory=list)
     use_declarations: List[str] = Field(default_factory=list)
     extern_crates: List[str] = Field(default_factory=list)
     is_unsafe: bool = False
     file_path: Optional[str] = None
     is_mod_rs: bool = False
+    is_root_module: bool = False
 
     def categorize_functions(self):
         """Categorize functions based on their safety analysis."""
         self.safe_functions.clear()
         self.unsafe_functions.clear()
-        
+
         for name, func in self.functions.items():
             if func.is_fully_safe():
                 self.safe_functions[name] = func
             else:
                 self.unsafe_functions[name] = func
 
 
+class RustDependency(BaseModel):
+    """Represents a Rust Dependency, Maybe internal or external"""
+
+    name: str
+    is_external: bool = True
+    crate: Optional[ForwardRef("RustCrate")]
+
+
 class RustCrate(BaseModel):
     """Represents a complete Rust crate."""
+
     name: str
     version: str
-    root_module: RustModule
-    dependencies: List["RustDependencyEdge"] = Field(default_factory=list)
+    is_lib: bool = False
+    modules: List[RustModule]
     edition: str = "2021"
     features: List[str] = Field(default_factory=list)
-    
+    dependencies: List[RustDependency] = Field(default_factory=list)
+
     def analyze_safety(self) -> Dict[str, int]:
         """Analyze safety statistics across the crate."""
         stats = {
@@ -171,7 +192,7 @@ def analyze_safety(self) -> Dict[str, int]:
             "unsafe_blocks": 0,
             "ffi_functions": 0,
         }
-        
+
         def analyze_module(module: RustModule):
             for func in module.functions.values():
                 stats["total_functions"] += 1
@@ -182,26 +203,26 @@ def analyze_module(module: RustModule):
                     if func.safety_analysis.classification == SafetyClassification.FFI:
                         stats["ffi_functions"] += 1
                 stats["unsafe_blocks"] += len(func.safety_analysis.unsafe_blocks)
-            
+
             for submodule in module.submodules.values():
                 analyze_module(submodule)
-        
+
         analyze_module(self.root_module)
         return stats
-    
+
     def get_unsafe_functions(self) -> List[tuple[str, RustCallable]]:
         """Get all unsafe functions in the crate with their module paths."""
         unsafe_fns = []
-        
+
         def collect_unsafe(module: RustModule, path: str):
             for name, func in module.functions.items():
                 if not func.is_fully_safe():
                     full_path = f"{path}::{name}" if path else name
                     unsafe_fns.append((full_path, func))
-            
+
             for submod_name, submod in module.submodules.items():
                 new_path = f"{path}::{submod_name}" if path else submod_name
                 collect_unsafe(submod, new_path)
-        
+
         collect_unsafe(self.root_module, "")
-        return unsafe_fns
+        return unsafe_fns