update location

kimi_dev/serve/templates.py

@@ -22,7 +22,7 @@ def show_project_structure(structure, spacing=0) -> str:
             pp_string += ' ' * spacing + str(key) + '\n'
         else:
             pp_string += ' ' * spacing + str(key) + '/' + '\n'
-        if 'classes' not in value:
+        if 'text' not in value:
             pp_string += show_project_structure(value, spacing + 4)
 
     return pp_string
@@ -112,35 +112,6 @@ def parse_python_file(file_path, file_content=None):
                 )
     return class_info, function_names, file_content.splitlines()
 
-def create_structure(directory_path):
-    """Create the structure of the repository directory by parsing Python files.
-    :param directory_path: Path to the repository directory.
-    :return: A dictionary representing the structure.
-    """
-    structure = {}
-    for root, _, files in os.walk(directory_path):
-        repo_name = os.path.basename(directory_path)
-        relative_root = os.path.relpath(root, directory_path)
-        if relative_root == ".":
-            relative_root = repo_name
-        curr_struct = structure
-        for part in relative_root.split(os.sep):
-            if part not in curr_struct:
-                curr_struct[part] = {}
-            curr_struct = curr_struct[part]
-        for file_name in files:
-            if file_name.endswith(".py"):
-                file_path = os.path.join(root, file_name)
-                class_info, function_names, file_lines = parse_python_file(file_path)
-                curr_struct[file_name] = {
-                    "classes": class_info,
-                    "functions": function_names,
-                    "text": file_lines,
-                }
-            else:
-                curr_struct[file_name] = {}
-    return structure
-
 # def create_structure(directory_path):
 #     """Create the structure of the repository directory by parsing Python files.
 #     :param directory_path: Path to the repository directory.
@@ -148,15 +119,15 @@ def create_structure(directory_path):
 #     """
 #     structure = {}
 #     for root, _, files in os.walk(directory_path):
+#         repo_name = os.path.basename(directory_path)
 #         relative_root = os.path.relpath(root, directory_path)
 #         if relative_root == ".":
-#             curr_struct = structure
-#         else:
-#             curr_struct = structure
-#             for part in relative_root.split(os.sep):
-#                 if part not in curr_struct:
-#                     curr_struct[part] = {}
-#                 curr_struct = curr_struct[part]
+#             relative_root = repo_name
+#         curr_struct = structure
+#         for part in relative_root.split(os.sep):
+#             if part not in curr_struct:
+#                 curr_struct[part] = {}
+#             curr_struct = curr_struct[part]
 #         for file_name in files:
 #             if file_name.endswith(".py"):
 #                 file_path = os.path.join(root, file_name)
@@ -167,18 +138,70 @@ def create_structure(directory_path):
 #                     "text": file_lines,
 #                 }
 #             else:
-#                 file_path = os.path.join(root, file_name)
-#                 try:
-#                     with open(file_path, 'r', encoding='utf-8') as f:
-#                         file_content = f.read()
-#                     curr_struct[file_name] = {
-#                         "text": file_content.splitlines()
-#                     }
-#                 except Exception as e:
-#                     print(f"fail to read {file_path}: {e}")
-#                     curr_struct[file_name] = {}
+#                 curr_struct[file_name] = {}
 #     return structure
 
+def create_structure(directory_path):
+    """Create the structure of the repository directory by parsing Python files.
+    :param directory_path: Path to the repository directory.
+    :return: A dictionary representing the structure.
+    """
+    structure = {}
+    for root, dirs, files in os.walk(directory_path):
+        relative_root = os.path.relpath(root, directory_path)
+        # Build the current directory position in the structure
+        if relative_root == ".":
+            curr_struct = structure
+        else:
+            curr_struct = structure
+            # Split by path separator and create directory structure layer by layer
+            path_parts = relative_root.split(os.sep)
+            for part in path_parts:
+                if part not in curr_struct:
+                    curr_struct[part] = {}
+                curr_struct = curr_struct[part]
+        # First create empty dictionary structure for all subdirectories
+        for dir_name in dirs:
+            # Skip hidden directories and common ignored directories
+            if not dir_name.startswith('.') and dir_name not in ['__pycache__', 'node_modules']:
+                if dir_name not in curr_struct:
+                    curr_struct[dir_name] = {}
+        # Process all files in the current directory
+        for file_name in files:
+            # Skip hidden files and compiled files
+            if file_name.startswith('.') or file_name.endswith('.pyc'):
+                continue
+            file_path = os.path.join(root, file_name)
+            if file_name.endswith(".py"):
+                # Python files: parse class and function information
+                try:
+                    class_info, function_names, file_lines = parse_python_file(file_path)
+                    curr_struct[file_name] = {
+                        "classes": class_info,
+                        "functions": function_names,
+                        "text": file_lines,
+                    }
+                except Exception as e:
+                    print(f"Failed to parse Python file {file_path}: {e}")
+                    curr_struct[file_name] = {"text": []}
+            else:
+                code_extensions = ['.js', '.ts', '.jsx', '.tsx', '.java', '.cpp', '.c', '.h', '.hpp', 
+                                 '.cs', '.php', '.rb', '.go', '.rs', '.swift', '.kt', '.scala', 
+                                 '.sh', '.bat', '.ps1', '.sql', '.html', '.css', '.scss', '.less',
+                                 '.json', '.xml', '.yaml', '.yml', '.toml', '.ini', '.cfg', '.conf',
+                                 '.md', '.txt', '.rst', '.tex', '.r', '.R', '.m', '.pl', '.lua']
+                if any(file_name.endswith(ext) for ext in code_extensions):
+                    try:
+                        with open(file_path, 'r', encoding='utf-8', errors='ignore') as f:
+                            file_content = f.read()
+                        curr_struct[file_name] = {"text": file_content.splitlines()}
+                    except Exception as e:
+                        print(f"Failed to read file {file_path}: {e}")
+                        curr_struct[file_name] = {"text": []}
+                else:
+                    curr_struct[file_name] = {"text": []}
+    return structure
+
 def build_repo_structure(root_path):
     """Build repository structure using improved parsing method"""
     return create_structure(root_path)