rename: archdoc → wtismycode (WTIsMyCode)

wtismycode-core/src/model.rs

@@ -0,0 +1,174 @@
+//! Intermediate Representation (IR) for WTIsMyCode
+//!
+//! This module defines the data structures that represent the analyzed Python project
+//! and are used for generating documentation.
+
+use serde::{Deserialize, Serialize};
+use std::collections::HashMap;
+
+#[derive(Debug, Clone, Serialize, Deserialize)]
+pub struct ProjectModel {
+    pub modules: HashMap<String, Module>,
+    pub files: HashMap<String, FileDoc>,
+    pub symbols: HashMap<String, Symbol>,
+    pub edges: Edges,
+}
+
+impl ProjectModel {
+    pub fn new() -> Self {
+        Self {
+            modules: HashMap::new(),
+            files: HashMap::new(),
+            symbols: HashMap::new(),
+            edges: Edges::new(),
+        }
+    }
+}
+
+impl Default for ProjectModel {
+    fn default() -> Self {
+        Self::new()
+    }
+}
+
+#[derive(Debug, Clone, Serialize, Deserialize)]
+pub struct Module {
+    pub id: String,
+    pub path: String,
+    pub files: Vec<String>,
+    pub doc_summary: Option<String>,
+    pub outbound_modules: Vec<String>,
+    pub inbound_modules: Vec<String>,
+    pub symbols: Vec<String>,
+}
+
+#[derive(Debug, Clone, Serialize, Deserialize)]
+pub struct FileDoc {
+    pub id: String,
+    pub path: String,
+    pub module_id: String,
+    pub imports: Vec<String>, // normalized import strings
+    pub outbound_modules: Vec<String>,
+    pub inbound_files: Vec<String>,
+    pub symbols: Vec<String>,
+    pub file_purpose: Option<String>,
+}
+
+#[derive(Debug, Clone, Serialize, Deserialize)]
+pub struct Symbol {
+    pub id: String,
+    pub kind: SymbolKind,
+    pub module_id: String,
+    pub file_id: String,
+    pub qualname: String,
+    pub signature: String,
+    pub annotations: Option<HashMap<String, String>>,
+    pub docstring_first_line: Option<String>,
+    pub purpose: String, // docstring or heuristic
+    pub outbound_calls: Vec<String>,
+    pub inbound_calls: Vec<String>,
+    pub integrations_flags: IntegrationFlags,
+    pub metrics: SymbolMetrics,
+}
+
+#[derive(Debug, Clone, Serialize, Deserialize, PartialEq)]
+pub enum SymbolKind {
+    Function,
+    AsyncFunction,
+    Class,
+    Method,
+}
+
+#[derive(Debug, Clone, Serialize, Deserialize)]
+pub struct IntegrationFlags {
+    pub http: bool,
+    pub db: bool,
+    pub queue: bool,
+    #[serde(default)]
+    pub storage: bool,
+    #[serde(default)]
+    pub ai: bool,
+}
+
+#[derive(Debug, Clone, Serialize, Deserialize)]
+pub struct SymbolMetrics {
+    pub fan_in: usize,
+    pub fan_out: usize,
+    pub is_critical: bool,
+    pub cycle_participant: bool,
+}
+
+#[derive(Debug, Clone, Serialize, Deserialize)]
+pub struct Edges {
+    pub module_import_edges: Vec<Edge>,
+    pub file_import_edges: Vec<Edge>,
+    pub symbol_call_edges: Vec<Edge>,
+}
+
+impl Edges {
+    pub fn new() -> Self {
+        Self {
+            module_import_edges: Vec::new(),
+            file_import_edges: Vec::new(),
+            symbol_call_edges: Vec::new(),
+        }
+    }
+}
+
+impl Default for Edges {
+    fn default() -> Self {
+        Self::new()
+    }
+}
+
+#[derive(Debug, Clone, Serialize, Deserialize)]
+pub struct Edge {
+    pub from_id: String,
+    pub to_id: String,
+    pub edge_type: EdgeType,
+    pub meta: Option<HashMap<String, String>>,
+}
+
+#[derive(Debug, Clone, Serialize, Deserialize)]
+pub enum EdgeType {
+    ModuleImport,
+    FileImport,
+    SymbolCall,
+    ExternalCall,
+    UnresolvedCall,
+}
+
+// Additional structures for Python analysis
+
+#[derive(Debug, Clone, serde::Serialize, serde::Deserialize)]
+pub struct ParsedModule {
+    pub path: std::path::PathBuf,
+    pub module_path: String,
+    pub imports: Vec<Import>,
+    pub symbols: Vec<Symbol>,
+    pub calls: Vec<Call>,
+    pub file_docstring: Option<String>,
+}
+
+#[derive(Debug, Clone, serde::Serialize, serde::Deserialize)]
+pub struct Import {
+    pub module_name: String,
+    pub alias: Option<String>,
+    pub line_number: usize,
+}
+
+#[derive(Debug, Clone, serde::Serialize, serde::Deserialize)]
+pub struct Call {
+    pub caller_symbol: String,
+    pub callee_expr: String,
+    pub line_number: usize,
+    pub call_type: CallType,
+}
+
+#[derive(Debug, Clone, serde::Serialize, serde::Deserialize)]
+pub enum CallType {
+    Local,
+    Imported,
+    External,
+    Unresolved,
+}