Merge pull request #127 from FSoft-AI4Code/feat/rust-support

analyzers: add tree-sitter based Rust analyzer

Author: anhnh2002

CHANGELOG.md

@@ -4,6 +4,15 @@ All notable changes to CodeWiki. The format follows
 [Keep a Changelog](https://keepachangelog.com/en/1.1.0/), and the project
 uses [Semantic Versioning](https://semver.org/).
 
+## [Unreleased]
+
+### Added
+
+- **Rust analyzer.** Structs, unions, enums, traits, free functions, and
+  impl/trait methods become components, with trait-impl, field-type,
+  struct-literal, and call edges. `#[cfg(test)]` modules are skipped, and
+  `build.rs` is classified as a build artifact.
+
 ## [2.0.0] - 2026-09-18
 
 The first release since 1.0.1. Everything that landed on `main` in between is

README.md

@@ -119,7 +119,7 @@ Full list in the [CHANGELOG](./CHANGELOG.md).
 ## Features
 
 **Languages.** Python, Java, JavaScript, TypeScript, C, C++, C#, Kotlin,
-PHP, Ruby, Scala.
+PHP, Ruby, Scala, Rust.
 
 **Hierarchical decomposition.** The dependency graph is clustered into a
 module tree, recursively, so a 1.4M-line repository gets the same treatment

codewiki/cli/main.py

@@ -20,7 +20,7 @@ def cli(ctx):
 
     AI agents write the documentation from your dependency graph. Supports
     Python, Java, JavaScript, TypeScript, C, C++, C#, Kotlin, PHP, Ruby,
-    and Scala.
+    Scala, and Rust.
     """
     # Ensure context object exists
     ctx.ensure_object(dict)

codewiki/cli/utils/repo_validator.py

@@ -33,6 +33,7 @@
     ".rb",  # Ruby
     ".scala",  # Scala
     ".sc",  # Scala Scripts
+    ".rs",  # Rust
 }
 
 
@@ -64,7 +65,7 @@ def validate_repository(repo_path: Path) -> tuple[Path, list[tuple[str, int]]]:
         raise RepositoryError(
             f"No supported code files found in {repo_path}\n\n"
             "CodeWiki supports: Python, Java, JavaScript, TypeScript, C, C++, C#, Kotlin, PHP, "
-            "Ruby, Scala\n\n"
+            "Ruby, Scala, Rust\n\n"
             "Please navigate to a code repository or specify a custom directory:\n"
             "  cd /path/to/your/project\n"
             "  codewiki generate"

codewiki/cli/utils/validation.py

@@ -167,6 +167,7 @@ def detect_supported_languages(directory: Path) -> list[tuple[str, int]]:
         "Kotlin": [".kt", ".kts"],
         "Ruby": [".rb"],
         "Scala": [".scala", ".sc"],
+        "Rust": [".rs"],
     }
 
     # Directories to exclude from counting

codewiki/mcp/tools/analysis.py

@@ -228,6 +228,7 @@ def _detect_via_mtime(
         ".rb",
         ".scala",
         ".sc",
+        ".rs",
     }
 
     changed: list[str] = []

codewiki/src/be/dependency_analyzer/analysis/analysis_service.py

@@ -339,7 +339,8 @@ def _filter_supported_languages(self, code_files: list[dict]) -> list[dict]:
         """
         Filter code files to only include supported languages.
 
-        Supports Python, JavaScript, TypeScript, Java, C#, C, C++, PHP, Ruby, Go, and Rust.
+        Supports Python, JavaScript, TypeScript, Java, C#, C, C++, PHP, Ruby, Kotlin,
+        Scala, and Rust.
         """
         supported_languages = {
             "python",
@@ -377,6 +378,7 @@ def _get_supported_languages(self) -> list[str]:
             "ruby",
             "kotlin",
             "scala",
+            "rust",
         ]
 
     def _cleanup_repository(self, temp_dir: str):

codewiki/src/be/dependency_analyzer/analysis/call_graph_analyzer.py

@@ -259,6 +259,8 @@ def _analyze_code_file(self, repo_dir: str, file_info: dict):
                     self._analyze_ruby_file(file_path, content, repo_dir)
                 elif language == "scala":
                     self._analyze_scala_file(file_path, content, repo_dir)
+                elif language == "rust":
+                    self._analyze_rust_file(file_path, content, repo_dir)
                 # else:
                 #     logger.warning(
                 #         f"Unsupported language for call graph analysis: {language} for file {file_path}"
@@ -536,6 +538,28 @@ def _analyze_scala_file(self, file_path: str, content: str, repo_dir: str):
         except Exception:
             logger.exception(f"Failed to analyze Scala file {file_path}")
 
+    def _analyze_rust_file(self, file_path: str, content: str, repo_dir: str):
+        """
+        Analyze Rust file using tree-sitter based analyzer.
+
+        Args:
+            file_path: Relative path to the Rust file
+            content: File content string
+            repo_dir: Repository base directory
+        """
+        from codewiki.src.be.dependency_analyzer.analyzers.rust import analyze_rust_file
+
+        try:
+            functions, relationships = analyze_rust_file(file_path, content, repo_path=repo_dir)
+
+            for func in functions:
+                func_id = func.id if func.id else f"{file_path}:{func.name}"
+                self.functions[func_id] = func
+
+            self.call_relationships.extend(relationships)
+        except Exception:
+            logger.exception(f"Failed to analyze Rust file {file_path}")
+
     def _resolve_call_relationships(self):
         """
         Resolve function call relationships across all languages.
@@ -834,6 +858,8 @@ def _generate_visualization_data(self) -> dict:
                 node_classes.append("lang-ruby")
             elif file_ext in [".scala", ".sc"]:
                 node_classes.append("lang-scala")
+            elif file_ext == ".rs":
+                node_classes.append("lang-rust")
 
             cytoscape_elements.append(
                 {

codewiki/src/be/dependency_analyzer/analyzers/artifact.py

@@ -278,6 +278,7 @@ def is_artifact_file_node(node: Any) -> bool:
 _BUILD_NAMES = {
     "Makefile",
     "GNUmakefile",
+    "build.rs",
     "makefile",
     "CMakeLists.txt",
     "Rakefile",