kopipasta 0.29.0__py3-none-any.whl → 0.31.0__py3-none-any.whl

kopipasta/cache.py

@@ -0,0 +1,37 @@
+import json
+import os
+from pathlib import Path
+from typing import List, Tuple
+
+# Define FileTuple for type hinting
+FileTuple = Tuple[str, bool, List[str] | None, str]
+
+def get_cache_file_path() -> Path:
+    """Gets the cross-platform path to the cache file for the last selection."""
+    cache_dir = Path.home() / ".cache" / "kopipasta"
+    cache_dir.mkdir(parents=True, exist_ok=True)
+    return cache_dir / "last_selection.json"
+
+def save_selection_to_cache(files_to_include: List[FileTuple]):
+    """Saves the list of selected file relative paths to the cache."""
+    cache_file = get_cache_file_path()
+    relative_paths = sorted([os.path.relpath(f[0]) for f in files_to_include])
+    try:
+        with open(cache_file, 'w', encoding='utf-8') as f:
+            json.dump(relative_paths, f, indent=2)
+    except IOError as e:
+        print(f"\nWarning: Could not save selection to cache: {e}")
+
+def load_selection_from_cache() -> List[str]:
+    """Loads the list of selected files from the cache file."""
+    cache_file = get_cache_file_path()
+    if not cache_file.exists():
+        return []
+    try:
+        with open(cache_file, 'r', encoding='utf-8') as f:
+            paths = json.load(f)
+            # Filter out paths that no longer exist
+            return [p for p in paths if os.path.exists(p)]
+    except (IOError, json.JSONDecodeError) as e:
+        print(f"\nWarning: Could not load previous selection from cache: {e}")
+        return []

kopipasta/main.py

@@ -18,6 +18,7 @@ from kopipasta.file import FileTuple, get_human_readable_size, is_binary, is_ign
 import kopipasta.import_parser as import_parser
 from kopipasta.tree_selector import TreeSelector
 from kopipasta.prompt import generate_prompt_template, get_file_snippet, get_language_for_file
+from kopipasta.cache import save_selection_to_cache
 
 def _propose_and_add_dependencies(
     file_just_added: str,
@@ -993,10 +994,14 @@ def open_editor_for_input(template: str, cursor_position: int) -> str:
 
 def main():
     parser = argparse.ArgumentParser(description="Generate a prompt with project structure, file contents, and web content.")
-    parser.add_argument('inputs', nargs='+', help='Files, directories, or URLs to include in the prompt')
+    parser.add_argument('inputs', nargs='*', help='Files, directories, or URLs to include. Defaults to current directory.')
     parser.add_argument('-t', '--task', help='Task description for the AI prompt')
     args = parser.parse_args()
 
+    # Default to the current directory if no inputs are provided
+    if not args.inputs:
+        args.inputs.append('.')
+
     ignore_patterns = read_gitignore()
     env_vars = read_env_file()
     project_root_abs = os.path.abspath(os.getcwd())
@@ -1054,7 +1059,7 @@ def main():
     # Use tree selector for file/directory selection
     if paths_for_tree:
         print("\nStarting interactive file selection...")
-        print("Use arrow keys to navigate, Space to select, 'q' to finish. Press 'h' for help.\n")
+        print("Use arrow keys to navigate, Space to select, 'q' to finish. See all keys below.\n")
         
         tree_selector = TreeSelector(ignore_patterns, project_root_abs)
         try:
@@ -1069,6 +1074,10 @@ def main():
         print("No files or web content were selected. Exiting.")
         return
 
+    # Save the final selection for the next run
+    if files_to_include:
+        save_selection_to_cache(files_to_include)
+
     print("\nFile and web content selection complete.")
     print_char_count(current_char_count)
 

kopipasta/tree_selector.py

@@ -1,16 +1,15 @@
 import os
+import shutil
 from typing import Dict, List, Optional, Tuple
 from rich.console import Console
 from rich.tree import Tree
 from rich.panel import Panel
-from rich.table import Table
 from rich.text import Text
-from rich.live import Live
-from rich.layout import Layout
 import click
 
 from kopipasta.file import FileTuple, is_binary, is_ignored, get_human_readable_size
 from kopipasta.prompt import get_file_snippet, get_language_for_file
+from kopipasta.cache import load_selection_from_cache
 
 
 class FileNode:
@@ -52,6 +51,7 @@ class TreeSelector:
         self.visible_nodes: List[FileNode] = []
         self.char_count = 0
         self.quit_selection = False
+        self.viewport_offset = 0  # First visible item index
         
     def build_tree(self, paths: List[str]) -> FileNode:
         """Build tree structure from given paths"""
@@ -154,17 +154,49 @@ class TreeSelector:
         return result
     
     def _build_display_tree(self) -> Tree:
-        """Build Rich tree for display"""
-        tree = Tree("📁 Project Files", guide_style="dim")
+        """Build Rich tree for display with viewport"""
+        # Get terminal size
+        term_width, term_height = shutil.get_terminal_size()
         
-        # Flatten tree and rebuild visible nodes list
+        # Reserve space for header, help panel, and status
+        available_height = term_height - 15  # Adjust based on your UI
+        available_height = max(5, available_height)  # Minimum height
+        
+        # Flatten tree to get all visible nodes
         flat_tree = self._flatten_tree(self.root)
         self.visible_nodes = [node for node, _ in flat_tree]
         
-        # Build tree structure - we'll map absolute paths to tree nodes
+        # Calculate viewport
+        if self.visible_nodes:
+            # Ensure current selection is visible
+            if self.current_index < self.viewport_offset:
+                self.viewport_offset = self.current_index
+            elif self.current_index >= self.viewport_offset + available_height:
+                self.viewport_offset = self.current_index - available_height + 1
+                
+            # Clamp viewport to valid range
+            max_offset = max(0, len(self.visible_nodes) - available_height)
+            self.viewport_offset = max(0, min(self.viewport_offset, max_offset))
+        else:
+            self.viewport_offset = 0
+        
+        # Create tree with scroll indicators
+        tree_title = "📁 Project Files"
+        if self.viewport_offset > 0:
+            tree_title += f" ↑ ({self.viewport_offset} more)"
+        
+        tree = Tree(tree_title, guide_style="dim")
+        
+        # Build tree structure - only for visible portion
         node_map = {}
+        viewport_end = min(len(flat_tree), self.viewport_offset + available_height)
+        
+        # Track what level each visible item is at for proper tree structure
+        level_stacks = {}  # level -> stack of tree nodes
         
-        for i, (node, level) in enumerate(flat_tree):
+        for i in range(self.viewport_offset, viewport_end):
+            node, level = flat_tree[i]
+            
             # Determine style and icon
             is_current = i == self.current_index
             style = "bold cyan" if is_current else ""
@@ -193,40 +225,45 @@ class TreeSelector:
             label.append(f"{selection} ", style="dim")
             label.append(f"{icon} {node.name}{size_str}", style=style)
             
-            # Add to tree at correct position
-            # For root-level items, add directly to tree
-            if node.parent and node.parent.path == os.path.abspath("."):
+            # Add to tree at correct level
+            if level == 0:
                 tree_node = tree.add(label)
-                node_map[abs_path] = tree_node
+                level_stacks[0] = tree_node
             else:
-                # Find parent node in map
-                parent_abs_path = os.path.abspath(node.parent.path) if node.parent else None
-                if parent_abs_path and parent_abs_path in node_map:
-                    parent_tree = node_map[parent_abs_path]
+                # Find parent at previous level
+                parent_level = level - 1
+                if parent_level in level_stacks:
+                    parent_tree = level_stacks[parent_level]
                     tree_node = parent_tree.add(label)
-                    node_map[abs_path] = tree_node
+                    level_stacks[level] = tree_node
                 else:
-                    # Fallback - add to root
-                    tree_node = tree.add(label)
-                    node_map[abs_path] = tree_node
+                    # Fallback - add to root with indentation indicator
+                    indent_label = Text()
+                    indent_label.append("  " * level + f"{selection} ", style="dim")
+                    indent_label.append(f"{icon} {node.name}{size_str}", style=style)
+                    tree_node = tree.add(indent_label)
+                    level_stacks[level] = tree_node
+        
+        # Add scroll indicator at bottom if needed
+        if viewport_end < len(self.visible_nodes):
+            remaining = len(self.visible_nodes) - viewport_end
+            tree.add(Text(f"↓ ({remaining} more items)", style="dim italic"))
             
         return tree
-    
+
     def _show_help(self) -> Panel:
         """Create help panel"""
         help_text = """[bold]Navigation:[/bold]
-↑/k: Move up     ↓/j: Move down     →/l/Enter: Expand dir     ←/h: Collapse dir
+↑/k: Up  ↓/j: Down  →/l/Enter: Expand  ←/h: Collapse
 
 [bold]Selection:[/bold]  
 Space: Toggle file/dir     a: Add all in dir     s: Snippet mode
 
 [bold]Actions:[/bold]
-g: Grep in directory     d: Show dependencies     q: Quit selection
-
-[bold]Status:[/bold]
-Selected: [green]● Full[/green]  [yellow]◐ Snippet[/yellow]  ○ Not selected"""
+r: Reuse last selection  g: Grep in directory   d: Show dependencies
+q: Quit and finalize"""
         
-        return Panel(help_text, title="Keyboard Shortcuts", border_style="dim", expand=False)
+        return Panel(help_text, title="Keyboard Controls", border_style="dim", expand=False)
     
     def _get_status_bar(self) -> str:
         """Create status bar with selection info"""
@@ -317,7 +354,7 @@ Selected: [green]● Full[/green]  [yellow]◐ Snippet[/yellow]  ○ Not selecte
                     self.char_count += node.size
     
     def _toggle_directory(self, node: FileNode):
-        """Toggle all files in a directory"""
+        """Toggle all files in a directory, now fully recursive."""
         if not node.is_dir:
             return
             
@@ -330,6 +367,9 @@ Selected: [green]● Full[/green]  [yellow]◐ Snippet[/yellow]  ○ Not selecte
         
         def collect_files(n: FileNode):
             if n.is_dir:
+                # CRITICAL FIX: Ensure sub-directory children are loaded before recursing
+                if not n.children:
+                    self._scan_directory(n.path, n)
                 for child in n.children:
                     collect_files(child)
             else:
@@ -341,16 +381,92 @@ Selected: [green]● Full[/green]  [yellow]◐ Snippet[/yellow]  ○ Not selecte
         any_unselected = any(os.path.abspath(f.path) not in self.selected_files for f in all_files)
         
         if any_unselected:
-            # Select all unselected
+            # Select all unselected files
             for file_node in all_files:
-                if file_node.path not in self.selected_files:
-                    self._toggle_selection(file_node)
+                abs_path = os.path.abspath(file_node.path)
+                if abs_path not in self.selected_files:
+                    self.selected_files[abs_path] = (False, None)
+                    self.char_count += file_node.size
         else:
-            # Unselect all
+            # Unselect all files
             for file_node in all_files:
-                if file_node.path in self.selected_files:
-                    self._toggle_selection(file_node)
-    
+                abs_path = os.path.abspath(file_node.path)
+                if abs_path in self.selected_files:
+                    is_snippet, _ = self.selected_files[abs_path]
+                    del self.selected_files[abs_path]
+                    if is_snippet:
+                        self.char_count -= len(get_file_snippet(file_node.path))
+                    else:
+                        self.char_count -= file_node.size
+
+    def _propose_and_apply_last_selection(self):
+        """Loads paths from cache, shows a confirmation dialog, and applies the selection if confirmed."""
+        cached_paths = load_selection_from_cache()
+
+        if not cached_paths:
+            self.console.print(Panel("[yellow]No cached selection found to reuse.[/yellow]", title="Info", border_style="dim"))
+            click.pause("Press any key to continue...")
+            return
+
+        # Categorize cached paths for the preview
+        files_to_add = []
+        files_already_selected = []
+        files_not_found = []
+
+        for rel_path in cached_paths:
+            abs_path = os.path.abspath(rel_path)
+            if not os.path.isfile(abs_path):
+                files_not_found.append(rel_path)
+                continue
+
+            if abs_path in self.selected_files:
+                files_already_selected.append(rel_path)
+            else:
+                files_to_add.append(rel_path)
+        
+        # Build the rich text for the confirmation panel
+        preview_text = Text()
+        if files_to_add:
+            preview_text.append("The following files will be ADDED:\n", style="bold")
+            for path in sorted(files_to_add):
+                preview_text.append("  ")
+                preview_text.append("+", style="cyan")
+                preview_text.append(f" {path}\n")
+        
+        if files_already_selected:
+            preview_text.append("\nAlready selected (no change):\n", style="bold dim")
+            for path in sorted(files_already_selected):
+                preview_text.append(f"  ✓ {path}\n")
+
+        if files_not_found:
+            preview_text.append("\nNot found on disk (will be skipped):\n", style="bold dim")
+            for path in sorted(files_not_found):
+                preview_text.append("  ")
+                preview_text.append("-", style="red")
+                preview_text.append(f" {path}\n")
+
+        # Display the confirmation panel and prompt
+        self.console.clear()
+        self.console.print(Panel(preview_text, title="[bold cyan]Reuse Last Selection?", border_style="cyan", padding=(1, 2)))
+        
+        if not files_to_add:
+            self.console.print("\n[yellow]No new files to add from the last selection.[/yellow]")
+            click.pause("Press any key to continue...")
+            return
+
+        # Use click.confirm for a simple and effective y/n prompt
+        if not click.confirm(f"\nAdd {len(files_to_add)} file(s) to your current selection?", default=True):
+            return
+
+        # If confirmed, apply the changes
+        for rel_path in files_to_add:
+            abs_path = os.path.abspath(rel_path)
+            if os.path.isfile(abs_path) and abs_path not in self.selected_files:
+                file_size = os.path.getsize(abs_path)
+                self.selected_files[abs_path] = (False, None)
+                self.char_count += file_size
+                self._ensure_path_visible(abs_path)
+
     def _ensure_path_visible(self, file_path: str):
         """Ensure a file path is visible in the tree by expanding parent directories"""
         abs_file_path = os.path.abspath(file_path)
@@ -456,6 +572,20 @@ Selected: [green]● Full[/green]  [yellow]◐ Snippet[/yellow]  ○ Not selecte
                     self.current_index = max(0, self.current_index - 1)
                 elif key in ['\x1b[B', 'j']:  # Down arrow or j  
                     self.current_index = min(len(self.visible_nodes) - 1, self.current_index + 1)
+                elif key == '\x1b[5~':  # Page Up
+                    term_width, term_height = shutil.get_terminal_size()
+                    page_size = max(1, term_height - 15)
+                    self.current_index = max(0, self.current_index - page_size)
+                elif key == '\x1b[6~':  # Page Down
+                    term_width, term_height = shutil.get_terminal_size()
+                    page_size = max(1, term_height - 15)
+                    self.current_index = min(len(self.visible_nodes) - 1, self.current_index + page_size)
+                elif key == '\x1b[H':  # Home - go to top
+                    self.current_index = 0
+                elif key == '\x1b[F':  # End - go to bottom
+                    self.current_index = len(self.visible_nodes) - 1
+                elif key == 'G':  # Shift+G - go to bottom (vim style)
+                    self.current_index = len(self.visible_nodes) - 1
                 elif key in ['\x1b[C', 'l', '\r']:  # Right arrow, l, or Enter
                     if current_node.is_dir:
                         current_node.expanded = True
@@ -480,10 +610,11 @@ Selected: [green]● Full[/green]  [yellow]◐ Snippet[/yellow]  ○ Not selecte
                         self._toggle_directory(current_node)
                         
                 # Handle actions
+                elif key == 'r':  # Reuse last selection
+                    self._propose_and_apply_last_selection()
                 elif key == 'g':  # Grep
                     self.console.print()  # Add some space
                     self._handle_grep(current_node)
-                    click.pause("Press any key to continue...")
                 elif key == 'd':  # Dependencies
                     self.console.print()  # Add some space
                     self._show_dependencies(current_node)

{kopipasta-0.29.0.dist-info → kopipasta-0.31.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: kopipasta
-Version: 0.29.0
+Version: 0.31.0
 Summary: A CLI tool to generate prompts with project structure and file contents
 Home-page: https://github.com/mkorpela/kopipasta
 Author: Mikko Korpela

kopipasta-0.31.0.dist-info/RECORD

@@ -0,0 +1,13 @@
+kopipasta/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+kopipasta/cache.py,sha256=rH52yX3g7p22_1NXmrBDcBEntm3ivMjtV8n8zxTLY1k,1454
+kopipasta/file.py,sha256=2HEoNczbKH3TtLO0zYUcZfUoHqb0-o83xFUOgY9rG-Y,1244
+kopipasta/import_parser.py,sha256=yLzkMlQm2avKjfqcpMY0PxbA_2ihV9gSYJplreWIPEQ,12424
+kopipasta/main.py,sha256=3Sle6pFEko4HSCXClmEb1L4QsN41VEOd0ZuzxeUDFUk,49210
+kopipasta/prompt.py,sha256=fOCuJVTLUfR0fjKf5qIlnl_3pNsKNKsvt3C8f4tsmxk,6889
+kopipasta/tree_selector.py,sha256=Iov8KLFopb7OLvX-xhn0FmFy0Ee5rFpPYzV6JLxCR84,27614
+kopipasta-0.31.0.dist-info/LICENSE,sha256=xw4C9TAU7LFu4r_MwSbky90uzkzNtRwAo3c51IWR8lk,1091
+kopipasta-0.31.0.dist-info/METADATA,sha256=RO8xJpSYGdqzsr-A7a22wTJEu2EwF_xtzJaziMD9XUw,4894
+kopipasta-0.31.0.dist-info/WHEEL,sha256=GV9aMThwP_4oNCtvEC2ec3qUYutgWeAzklro_0m4WJQ,91
+kopipasta-0.31.0.dist-info/entry_points.txt,sha256=but54qDNz1-F8fVvGstq_QID5tHjczP7bO7rWLFkc6Y,50
+kopipasta-0.31.0.dist-info/top_level.txt,sha256=iXohixMuCdw8UjGDUp0ouICLYBDrx207sgZIJ9lxn0o,10
+kopipasta-0.31.0.dist-info/RECORD,,

kopipasta-0.29.0.dist-info/RECORD

@@ -1,12 +0,0 @@
-kopipasta/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-kopipasta/file.py,sha256=2HEoNczbKH3TtLO0zYUcZfUoHqb0-o83xFUOgY9rG-Y,1244
-kopipasta/import_parser.py,sha256=yLzkMlQm2avKjfqcpMY0PxbA_2ihV9gSYJplreWIPEQ,12424
-kopipasta/main.py,sha256=MlfmP_eG0ZyvdhxdGyPsN2TxqnM4hq9pa2ToQEbbNP4,48894
-kopipasta/prompt.py,sha256=fOCuJVTLUfR0fjKf5qIlnl_3pNsKNKsvt3C8f4tsmxk,6889
-kopipasta/tree_selector.py,sha256=Dr7R4d19NzxWMwfQI3GioaD2i1ho84hMRxCzVp4sfCg,21342
-kopipasta-0.29.0.dist-info/LICENSE,sha256=xw4C9TAU7LFu4r_MwSbky90uzkzNtRwAo3c51IWR8lk,1091
-kopipasta-0.29.0.dist-info/METADATA,sha256=4vQpgi-XLKavkyQulJTbEscOUX5uAe1n6n-PJZwKTck,4894
-kopipasta-0.29.0.dist-info/WHEEL,sha256=GV9aMThwP_4oNCtvEC2ec3qUYutgWeAzklro_0m4WJQ,91
-kopipasta-0.29.0.dist-info/entry_points.txt,sha256=but54qDNz1-F8fVvGstq_QID5tHjczP7bO7rWLFkc6Y,50
-kopipasta-0.29.0.dist-info/top_level.txt,sha256=iXohixMuCdw8UjGDUp0ouICLYBDrx207sgZIJ9lxn0o,10
-kopipasta-0.29.0.dist-info/RECORD,,