Merge pull request #41 from p0thi/enable-kotlin-support

feat: Add Kotlin Language Support

Author: anhnh2002

README.md

@@ -76,7 +76,7 @@ codewiki generate --github-pages --create-branch
 
 ## What is CodeWiki?
 
-CodeWiki is an open-source framework for **automated repository-level documentation** across seven programming languages. It generates holistic, architecture-aware documentation that captures not only individual functions but also their cross-file, cross-module, and system-level interactions.
+CodeWiki is an open-source framework for **automated repository-level documentation** across eight programming languages. It generates holistic, architecture-aware documentation that captures not only individual functions but also their cross-file, cross-module, and system-level interactions.
 
 ### Key Innovations
 
@@ -88,7 +88,7 @@ CodeWiki is an open-source framework for **automated repository-level documentat
 
 ### Supported Languages
 
-**🐍 Python** • **☕ Java** • **🟨 JavaScript** • **🔷 TypeScript** • **⚙️ C** • **🔧 C++** • **🪟 C#**
+**🐍 Python** • **☕ Java** • **🟨 JavaScript** • **🔷 TypeScript** • **⚙️ C** • **🔧 C++** • **🪟 C#** • **🎯 Kotlin**
 
 ---
 

codewiki/cli/utils/repo_validator.py

@@ -30,6 +30,8 @@
     '.php',     # PHP
     '.phtml',   # PHP templates
     '.inc',     # PHP includes
+    '.kt',      # Kotlin
+    '.kts',     # Kotlin Scripts
 }
 
 

codewiki/cli/utils/validation.py

@@ -172,6 +172,7 @@ def detect_supported_languages(directory: Path) -> List[Tuple[str, int]]:
         'C++': ['.cpp', '.hpp', '.cc', '.hh', '.cxx', '.hxx'],
         'C#': ['.cs'],
         'PHP': ['.php', '.phtml', '.inc'],
+        'Kotlin': ['.kt', '.kts'],
     }
 
     # Directories to exclude from counting

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

@@ -310,6 +310,7 @@ def _filter_supported_languages(self, code_files: List[Dict]) -> List[Dict]:
             "php",
             "go",
             "rust",
+            "kotlin",
         }
 
         return [
@@ -320,7 +321,7 @@ def _filter_supported_languages(self, code_files: List[Dict]) -> List[Dict]:
 
     def _get_supported_languages(self) -> List[str]:
         """Get list of currently supported languages for analysis."""
-        return ["python", "javascript", "typescript", "java", "csharp", "c", "cpp", "php"]
+        return ["python", "javascript", "typescript", "java", "csharp", "c", "cpp", "php", "kotlin"]
 
     def _cleanup_repository(self, temp_dir: str):
         """Clean up cloned repository."""

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

@@ -126,6 +126,8 @@ def _analyze_code_file(self, repo_dir: str, file_info: Dict):
                 self._analyze_typescript_file(file_path, content, repo_dir)
             elif language == "java":
                 self._analyze_java_file(file_path, content, repo_dir)
+            elif language == "kotlin":
+                self._analyze_kotlin_file(file_path, content, repo_dir)
             elif language == "csharp":
                 self._analyze_csharp_file(file_path, content, repo_dir)
             elif language == "c":
@@ -280,6 +282,27 @@ def _analyze_java_file(self, file_path: str, content: str, repo_dir: str):
         except Exception as e:
             logger.error(f"Failed to analyze Java file {file_path}: {e}", exc_info=True)
 
+    def _analyze_kotlin_file(self, file_path: str, content: str, repo_dir: str):
+        """
+        Analyze Kotlin file using tree-sitter based analyzer.
+
+        Args:
+            file_path: Relative path to the Kotlin file
+            content: File content string
+            repo_dir: Repository base directory
+        """
+        from codewiki.src.be.dependency_analyzer.analyzers.kotlin import analyze_kotlin_file
+
+        try:
+            functions, relationships = analyze_kotlin_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 as e:
+            logger.error(f"Failed to analyze Kotlin file {file_path}: {e}", exc_info=True)
+
     def _analyze_csharp_file(self, file_path: str, content: str, repo_dir: str):
         """
         Analyze C# file using tree-sitter based analyzer.
@@ -408,6 +431,8 @@ def _generate_visualization_data(self) -> Dict:
                 node_classes.append("lang-c")
             elif file_ext in [".cpp", ".cc", ".cxx", ".hpp", ".hxx"]:
                 node_classes.append("lang-cpp")
+            elif file_ext in [".kt", ".kts"]:
+                node_classes.append("lang-kotlin")
             elif file_ext in [".php", ".phtml", ".inc"]:
                 node_classes.append("lang-php")