code-puppy 0.0.54__tar.gz → 0.0.55__tar.gz

{code_puppy-0.0.54 → code_puppy-0.0.55}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: code-puppy
-Version: 0.0.54
+Version: 0.0.55
 Summary: Code generation agent
 Author: Michael Pfaffenberger
 License: MIT

{code_puppy-0.0.54 → code_puppy-0.0.55}/code_puppy/tools/file_operations.py

@@ -40,29 +40,148 @@ def should_ignore_path(path: str) -> bool:
 def _list_files(
     context: RunContext, directory: str = ".", recursive: bool = True
 ) -> List[Dict[str, Any]]:
-    """Light-weight `list_files` implementation sufficient for unit-tests and agent tooling."""
+    results = []
+    directory = os.path.abspath(directory)
+    console.print("\n[bold white on blue] DIRECTORY LISTING [/bold white on blue]")
     console.print(
-        f"\n[bold white on blue] LIST FILES [/bold white on blue] \U0001f4c2 [bold cyan]{directory}[/bold cyan]"
+        f"\U0001f4c2 [bold cyan]{directory}[/bold cyan] [dim](recursive={recursive})[/dim]"
     )
     console.print("[dim]" + "-" * 60 + "[/dim]")
-    directory = os.path.abspath(directory)
-    results: List[Dict[str, Any]] = []
-    if not os.path.exists(directory) or not os.path.isdir(directory):
+    if not os.path.exists(directory):
         console.print(
-            f"[bold red]Directory '{directory}' does not exist or is not a directory[/bold red]"
+            f"[bold red]Error:[/bold red] Directory '{directory}' does not exist"
         )
-        return [
-            {"error": f"Directory '{directory}' does not exist or is not a directory"}
-        ]
+        console.print("[dim]" + "-" * 60 + "[/dim]\n")
+        return [{"error": f"Directory '{directory}' does not exist"}]
+    if not os.path.isdir(directory):
+        console.print(f"[bold red]Error:[/bold red] '{directory}' is not a directory")
+        console.print("[dim]" + "-" * 60 + "[/dim]\n")
+        return [{"error": f"'{directory}' is not a directory"}]
+    folder_structure = {}
+    file_list = []
     for root, dirs, files in os.walk(directory):
-        rel_root = os.path.relpath(root, directory)
-        if rel_root == ".":
-            rel_root = ""
-        for f in files:
-            fp = os.path.join(rel_root, f) if rel_root else f
-            results.append({"path": fp, "type": "file"})
+        dirs[:] = [d for d in dirs if not should_ignore_path(os.path.join(root, d))]
+        rel_path = os.path.relpath(root, directory)
+        depth = 0 if rel_path == "." else rel_path.count(os.sep) + 1
+        if rel_path == ".":
+            rel_path = ""
+        if rel_path:
+            dir_path = os.path.join(directory, rel_path)
+            results.append(
+                {
+                    "path": rel_path,
+                    "type": "directory",
+                    "size": 0,
+                    "full_path": dir_path,
+                    "depth": depth,
+                }
+            )
+            folder_structure[rel_path] = {
+                "path": rel_path,
+                "depth": depth,
+                "full_path": dir_path,
+            }
+        for file in files:
+            file_path = os.path.join(root, file)
+            if should_ignore_path(file_path):
+                continue
+            rel_file_path = os.path.join(rel_path, file) if rel_path else file
+            try:
+                size = os.path.getsize(file_path)
+                file_info = {
+                    "path": rel_file_path,
+                    "type": "file",
+                    "size": size,
+                    "full_path": file_path,
+                    "depth": depth,
+                }
+                results.append(file_info)
+                file_list.append(file_info)
+            except (FileNotFoundError, PermissionError):
+                continue
         if not recursive:
             break
+
+    def format_size(size_bytes):
+        if size_bytes < 1024:
+            return f"{size_bytes} B"
+        elif size_bytes < 1024 * 1024:
+            return f"{size_bytes / 1024:.1f} KB"
+        elif size_bytes < 1024 * 1024 * 1024:
+            return f"{size_bytes / (1024 * 1024):.1f} MB"
+        else:
+            return f"{size_bytes / (1024 * 1024 * 1024):.1f} GB"
+
+    def get_file_icon(file_path):
+        ext = os.path.splitext(file_path)[1].lower()
+        if ext in [".py", ".pyw"]:
+            return "\U0001f40d"
+        elif ext in [".js", ".jsx", ".ts", ".tsx"]:
+            return "\U0001f4dc"
+        elif ext in [".html", ".htm", ".xml"]:
+            return "\U0001f310"
+        elif ext in [".css", ".scss", ".sass"]:
+            return "\U0001f3a8"
+        elif ext in [".md", ".markdown", ".rst"]:
+            return "\U0001f4dd"
+        elif ext in [".json", ".yaml", ".yml", ".toml"]:
+            return "\u2699\ufe0f"
+        elif ext in [".jpg", ".jpeg", ".png", ".gif", ".svg", ".webp"]:
+            return "\U0001f5bc\ufe0f"
+        elif ext in [".mp3", ".wav", ".ogg", ".flac"]:
+            return "\U0001f3b5"
+        elif ext in [".mp4", ".avi", ".mov", ".webm"]:
+            return "\U0001f3ac"
+        elif ext in [".pdf", ".doc", ".docx", ".xls", ".xlsx", ".ppt", ".pptx"]:
+            return "\U0001f4c4"
+        elif ext in [".zip", ".tar", ".gz", ".rar", ".7z"]:
+            return "\U0001f4e6"
+        elif ext in [".exe", ".dll", ".so", ".dylib"]:
+            return "\u26a1"
+        else:
+            return "\U0001f4c4"
+
+    if results:
+        files = sorted(
+            [f for f in results if f["type"] == "file"], key=lambda x: x["path"]
+        )
+        console.print(
+            f"\U0001f4c1 [bold blue]{os.path.basename(directory) or directory}[/bold blue]"
+        )
+    all_items = sorted(results, key=lambda x: x["path"])
+    parent_dirs_with_content = set()
+    for i, item in enumerate(all_items):
+        if item["type"] == "directory" and not item["path"]:
+            continue
+        if os.sep in item["path"]:
+            parent_path = os.path.dirname(item["path"])
+            parent_dirs_with_content.add(parent_path)
+        depth = item["path"].count(os.sep) + 1 if item["path"] else 0
+        prefix = ""
+        for d in range(depth):
+            if d == depth - 1:
+                prefix += "\u2514\u2500\u2500 "
+            else:
+                prefix += "    "
+        name = os.path.basename(item["path"]) or item["path"]
+        if item["type"] == "directory":
+            console.print(f"{prefix}\U0001f4c1 [bold blue]{name}/[/bold blue]")
+        else:
+            icon = get_file_icon(item["path"])
+            size_str = format_size(item["size"])
+            console.print(
+                f"{prefix}{icon} [green]{name}[/green] [dim]({size_str})[/dim]"
+            )
+    else:
+        console.print("[yellow]Directory is empty[/yellow]")
+    dir_count = sum(1 for item in results if item["type"] == "directory")
+    file_count = sum(1 for item in results if item["type"] == "file")
+    total_size = sum(item["size"] for item in results if item["type"] == "file")
+    console.print("\n[bold cyan]Summary:[/bold cyan]")
+    console.print(
+        f"\U0001f4c1 [blue]{dir_count} directories[/blue], \U0001f4c4 [green]{file_count} files[/green] [dim]({format_size(total_size)} total)[/dim]"
+    )
+    console.print("[dim]" + "-" * 60 + "[/dim]\n")
     return results
 
 

{code_puppy-0.0.54 → code_puppy-0.0.55}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "code-puppy"
-version = "0.0.54"
+version = "0.0.55"
 description = "Code generation agent"
 readme = "README.md"
 requires-python = ">=3.10"