kopipasta 0.17.0__py3-none-any.whl → 0.19.0__py3-none-any.whl

kopipasta/main.py

@@ -5,14 +5,27 @@ import json
 import os
 import argparse
 import re
+import subprocess
+import tempfile
 from typing import Dict, List, Optional, Set, Tuple
 import pyperclip
 import fnmatch
+from pygments import highlight
+from pygments.lexers import get_lexer_for_filename, TextLexer
+from pygments.formatters import TerminalFormatter
+import pygments.util
 
 import requests
 
 FileTuple = Tuple[str, bool, Optional[List[str]], str]
 
+def get_colored_code(file_path, code):
+     try:
+         lexer = get_lexer_for_filename(file_path)
+     except pygments.util.ClassNotFound:
+         lexer = TextLexer()
+     return highlight(code, lexer, TerminalFormatter())
+
 def read_gitignore():
     default_ignore_patterns = [
         '.git', 'node_modules', 'venv', '.venv', 'dist', '.idea', '__pycache__',
@@ -375,7 +388,8 @@ def select_file_patches(file_path):
     print(f"\nSelecting patches for {file_path}")
     for index, (chunk_code, start_line, end_line) in enumerate(code_chunks):
         print(f"\nChunk {index + 1} (Lines {start_line + 1}-{end_line}):")
-        print(f"```{language}\n{chunk_code}\n```")
+        colored_chunk = get_colored_code(file_path, chunk_code)
+        print(colored_chunk)
         while True:
             choice = input("(y)es include / (n)o skip / (q)uit rest of file? ").lower()
             if choice == 'y':
@@ -422,6 +436,10 @@ def get_file_snippet(file_path, max_lines=50, max_bytes=4096):
             byte_count += len(line.encode('utf-8'))
     return snippet
 
+def get_colored_file_snippet(file_path, max_lines=50, max_bytes=4096):
+    snippet = get_file_snippet(file_path, max_lines, max_bytes)
+    return get_colored_code(file_path, snippet)
+
 def print_char_count(count):
     token_estimate = count // 4
     print(f"\rCurrent prompt size: {count} characters (~ {token_estimate} tokens)", flush=True)
@@ -671,7 +689,7 @@ def handle_env_variables(content, env_vars):
 
     return content
 
-def generate_prompt(files_to_include: List[FileTuple], ignore_patterns: List[str], web_contents: Dict[str, Tuple[FileTuple, str]], env_vars: Dict[str, str]) -> str:
+def generate_prompt_template(files_to_include: List[FileTuple], ignore_patterns: List[str], web_contents: Dict[str, Tuple[FileTuple, str]], env_vars: Dict[str, str]) -> Tuple[str, int]:
     prompt = "# Project Overview\n\n"
     prompt += "## Project Structure\n\n"
     prompt += "```\n"
@@ -704,8 +722,8 @@ def generate_prompt(files_to_include: List[FileTuple], ignore_patterns: List[str
             prompt += f"### {url}{' (snippet)' if is_snippet else ''}\n\n```{language}\n{content}\n```\n\n"
     
     prompt += "## Task Instructions\n\n"
-    task_instructions = input("Enter the task instructions: ")
-    prompt += f"{task_instructions}\n\n"
+    cursor_position = len(prompt)
+    prompt += "\n\n"
     prompt += "## Instructions for Achieving the Task\n\n"
     analysis_text = (
         "1. **Confirm and Understand the Task**:\n"
@@ -733,12 +751,38 @@ def generate_prompt(files_to_include: List[FileTuple], ignore_patterns: List[str
         "   - Request the user to share any error messages or outputs from debugging to assist further.\n"
     )
     prompt += analysis_text
-    return prompt
+    return prompt, cursor_position
+
+def open_editor_for_input(template: str, cursor_position: int) -> str:
+    editor = os.environ.get('EDITOR', 'vim')
+    with tempfile.NamedTemporaryFile(mode='w+', suffix='.md', delete=False) as temp_file:
+        temp_file.write(template)
+        temp_file.flush()
+        temp_file_path = temp_file.name
+
+    try:
+        cursor_line = template[:cursor_position].count('\n') + 1
+        cursor_column = cursor_position - template.rfind('\n', 0, cursor_position)
+
+        if 'vim' in editor or 'nvim' in editor:
+            subprocess.call([editor, f'+call cursor({cursor_line}, {cursor_column})', '+startinsert', temp_file_path])
+        elif 'emacs' in editor:
+            subprocess.call([editor, f'+{cursor_line}:{cursor_column}', temp_file_path])
+        elif 'nano' in editor:
+            subprocess.call([editor, f'+{cursor_line},{cursor_column}', temp_file_path])
+        else:
+            subprocess.call([editor, temp_file_path])
 
+        with open(temp_file_path, 'r') as file:
+            content = file.read()
+        return content
+    finally:
+        os.unlink(temp_file_path)
 
 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('-t', '--task', help='Task description for the AI prompt')
     args = parser.parse_args()
 
     ignore_patterns = read_gitignore()
@@ -791,7 +835,9 @@ def main():
                         use_snippet = is_large_file(input_path)
                         files_to_include.append((input_path, use_snippet, None, get_language_for_file(input_path)))
                         if use_snippet:
-                            current_char_count += len(get_file_snippet(input_path))
+                            snippet = get_file_snippet(input_path)
+                            current_char_count += len(snippet)
+                            print(get_colored_code(input_path, snippet))
                         else:
                             current_char_count += os.path.getsize(input_path)
                         print(f"Added file: {input_path}{' (snippet)' if use_snippet else ''}")
@@ -826,16 +872,23 @@ def main():
     print_char_count(current_char_count)
     print(f"Summary: Added {len(files_to_include)} files from {len(processed_dirs)} directories and {len(web_contents)} web sources.")
 
-    prompt = generate_prompt(files_to_include, ignore_patterns, web_contents, env_vars)
+    prompt_template, cursor_position = generate_prompt_template(files_to_include, ignore_patterns, web_contents, env_vars)
+
+    if args.task:
+        task_description = args.task
+        final_prompt = prompt_template[:cursor_position] + task_description + prompt_template[cursor_position:]
+    else:
+        final_prompt = open_editor_for_input(prompt_template, cursor_position)
+
     print("\n\nGenerated prompt:")
-    print(prompt)
+    print(final_prompt)
 
     # Copy the prompt to clipboard
     try:
-        pyperclip.copy(prompt)
+        pyperclip.copy(final_prompt)
         separator = "\n" + "=" * 40 + "\n☕🍝       Kopipasta Complete!       🍝☕\n" + "=" * 40 + "\n"
         print(separator)
-        final_char_count = len(prompt)
+        final_char_count = len(final_prompt)
         final_token_estimate = final_char_count // 4
         print(f"Prompt has been copied to clipboard. Final size: {final_char_count} characters (~ {final_token_estimate} tokens)")
     except pyperclip.PyperclipException as e:

kopipasta-0.19.0.dist-info/METADATA

@@ -0,0 +1,78 @@
+Metadata-Version: 2.1
+Name: kopipasta
+Version: 0.19.0
+Summary: A CLI tool to generate prompts with project structure and file contents
+Home-page: https://github.com/mkorpela/kopipasta
+Author: Mikko Korpela
+Author-email: mikko.korpela@gmail.com
+License: MIT
+Classifier: Development Status :: 3 - Alpha
+Classifier: Intended Audience :: Developers
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Operating System :: OS Independent
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.8
+Classifier: Programming Language :: Python :: 3.9
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Requires-Python: >=3.8
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: pyperclip==1.9.0
+Requires-Dist: requests==2.32.3
+
+# kopipasta
+
+[![Version](https://img.shields.io/pypi/v/kopipasta.svg)](https://pypi.python.org/pypi/kopipasta)
+[![Downloads](http://pepy.tech/badge/kopipasta)](http://pepy.tech/project/kopipasta)
+
+A CLI tool for generating AI-assisted code prompts with project structure and file contents, using an interactive editor-based workflow.
+
+<img src="kopipasta.jpg" alt="kopipasta" width="300">
+
+- An LLM told me that kopi means Coffee in some languages.. and a Diffusion model then made this delicious soup.
+
+## Installation
+
+You can install kopipasta using pipx (or pip):
+
+```bash
+pipx install kopipasta
+```
+
+## Usage
+
+To use kopipasta, run the following command in your terminal:
+
+```bash
+kopipasta [files_or_directories_or_urls]
+```
+
+Replace `[files_or_directories_or_urls]` with the paths to the files or directories you want to include in the prompt, as well as any web URLs you want to fetch content from.
+
+Example:
+```bash
+kopipasta src/ config.json https://example.com/api-docs
+```
+
+This will guide you through an interactive process to:
+1. Select files and directories to include in the prompt
+2. Choose between full content, snippets, or patches for large files
+3. Fetch and include content from provided URLs
+4. Open an editor for you to input the specific task or code generation instructions
+5. Generate a comprehensive prompt that includes project structure, selected file contents, and your task instructions
+
+The generated prompt will be displayed in the console and automatically copied to your clipboard, ready to be used with an AI code generation tool.
+
+## Features
+
+- Generates structured prompts with project overview, file contents, web content, and task instructions
+- Interactive file selection process with options for full content, snippets, or specific patches
+- Fetches and includes content from web URLs
+- Opens your preferred editor (configurable via EDITOR environment variable) for task input
+- Detects and securely handles environment variables from a `.env` file
+- Ignores files and directories based on common .gitignore patterns
+- Allows interactive selection of files to include
+- Supports various file types with syntax highlighting in the selection process
+- Automatically copies the generated prompt to the clipboard

kopipasta-0.19.0.dist-info/RECORD

@@ -0,0 +1,8 @@
+kopipasta/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+kopipasta/main.py,sha256=zHCgL8dIZ4qMDzaN4EgUltH7wRT8UVvLn0Fq5ZgEZ8s,36904
+kopipasta-0.19.0.dist-info/LICENSE,sha256=xw4C9TAU7LFu4r_MwSbky90uzkzNtRwAo3c51IWR8lk,1091
+kopipasta-0.19.0.dist-info/METADATA,sha256=ipQ4jhKhnSgb2YKffbZZdBcx7FBZYodmS_g95Dx3aAY,3147
+kopipasta-0.19.0.dist-info/WHEEL,sha256=GV9aMThwP_4oNCtvEC2ec3qUYutgWeAzklro_0m4WJQ,91
+kopipasta-0.19.0.dist-info/entry_points.txt,sha256=but54qDNz1-F8fVvGstq_QID5tHjczP7bO7rWLFkc6Y,50
+kopipasta-0.19.0.dist-info/top_level.txt,sha256=iXohixMuCdw8UjGDUp0ouICLYBDrx207sgZIJ9lxn0o,10
+kopipasta-0.19.0.dist-info/RECORD,,

kopipasta-0.17.0.dist-info/METADATA

@@ -1,200 +0,0 @@
-Metadata-Version: 2.1
-Name: kopipasta
-Version: 0.17.0
-Summary: A CLI tool to generate prompts with project structure and file contents
-Home-page: https://github.com/mkorpela/kopipasta
-Author: Mikko Korpela
-Author-email: mikko.korpela@gmail.com
-License: MIT
-Classifier: Development Status :: 3 - Alpha
-Classifier: Intended Audience :: Developers
-Classifier: License :: OSI Approved :: MIT License
-Classifier: Operating System :: OS Independent
-Classifier: Programming Language :: Python :: 3
-Classifier: Programming Language :: Python :: 3.8
-Classifier: Programming Language :: Python :: 3.9
-Classifier: Programming Language :: Python :: 3.10
-Classifier: Programming Language :: Python :: 3.11
-Classifier: Programming Language :: Python :: 3.12
-Requires-Python: >=3.8
-Description-Content-Type: text/markdown
-License-File: LICENSE
-Requires-Dist: pyperclip==1.9.0
-Requires-Dist: requests==2.32.3
-
-# kopipasta
-
-A CLI tool to generate prompts with project structure, file contents, and web content, while handling environment variables securely and offering snippets for large files also interactive patch selection, allowing you to include only the most relevant parts of your code.
-
-<img src="kopipasta.jpg" alt="kopipasta" width="300">
-
-- An LLM told me that kopi means Coffee in some languages.. and a Diffusion model then made this delicious soup.
-
-## Installation
-
-You can install kopipasta using pipx (or pip):
-
-```
-pipx install kopipasta
-```
-
-## Usage
-
-To use kopipasta, run the following command in your terminal:
-
-```
-kopipasta [files_or_directories_or_urls]
-```
-
-Replace `[files_or_directories_or_urls]` with the paths to the files or directories you want to include in the prompt, as well as any web URLs you want to fetch content from.
-
-Example:
-```
-kopipasta src/ config.json https://example.com/api-docs
-```
-
-This will generate a prompt including:
-- The project structure
-- Contents of the specified files and directories (with snippet options for large files)
-- Content fetched from the provided URLs (with snippet options for large content)
-- Handling of environment variables found in a `.env` file (if present)
-
-Files and directories typically excluded in version control (based on common .gitignore patterns) are ignored.
-
-The generated prompt will be displayed in the console and automatically copied to your clipboard.
-
-## Features
-
-- Generates a structured prompt with project overview, file contents, web content, and task instructions
-- Offers snippet options for large files (>100 KB) and web content (>10,000 characters)
-- Fetches and includes content from web URLs
-- Detects and securely handles environment variables from a `.env` file
-- Ignores files and directories based on common .gitignore patterns
-- Allows interactive selection of files to include
-- Automatically copies the generated prompt to the clipboard
-
-## Environment Variable Handling
-
-If a `.env` file is present in the current directory, kopipasta will:
-1. Read and store the environment variables
-2. Detect these variables in file contents and web content
-3. Prompt you to choose how to handle each detected variable:
-   - (m)ask: Replace the value with asterisks
-   - (s)kip: Replace the value with "[REDACTED]"
-   - (k)eep: Leave the value as-is
-
-This ensures sensitive information is handled securely in the generated prompt.
-
-## Snippet Functionality
-
-For large files (>100 KB) and web content (>10,000 characters), kopipasta offers a snippet option:
-
-- For files: The first 50 lines or 4 KB (4,096 bytes), whichever comes first
-- For web content: The first 1,000 characters
-
-This helps manage the overall prompt size while still providing useful information about the content structure.
-
-## Example output
-
-```bash
-❯ kopipasta . https://example.com/api-docs
-
-Directory: .
-Files:
-- __init__.py
-- main.py (120 KB, ~120000 chars, ~30000 tokens)
-- large_data.csv (5 MB, ~5000000 chars, ~1250000 tokens)
-- .env
-
-(y)es add all / (n)o ignore all / (s)elect individually / (q)uit? y
-main.py is large. Use (f)ull content or (s)nippet? s
-large_data.csv is large. Use (f)ull content or (s)nippet? s
-Added all files from .
-Added web content from: https://example.com/api-docs
-
-File and web content selection complete.
-Current prompt size: 10500 characters (~ 2625 tokens)
-Summary: Added 3 files from 1 directory and 1 web source.
-
-Detected environment variables:
-- API_KEY=12345abcde
-
-How would you like to handle API_KEY? (m)ask / (k)eep: m
-
-Enter the task instructions: Implement new API endpoint
-
-Generated prompt:
-# Project Overview
-
-## Project Structure
-
-```
-|-- ./
-    |-- __init__.py
-    |-- main.py
-    |-- large_data.csv
-    |-- .env
-```
-
-## File Contents
-
-### __init__.py
-
-```python
-# Initialize package
-```
-
-### main.py (snippet)
-
-```python
-import os
-import pandas as pd
-
-API_KEY = os.getenv('API_KEY')
-
-def process_data(file_path):
-    df = pd.read_csv(file_path)
-    # Rest of the function...
-
-# More code...
-```
-
-### large_data.csv (snippet)
-
-```
-id,name,value
-1,John,100
-2,Jane,200
-3,Bob,150
-4,Alice,300
-# ... (first 50 lines or 4 KB)
-```
-
-## Web Content
-
-### https://example.com/api-docs (snippet)
-
-```
-API Documentation
-Endpoint: /api/v1/data
-Method: GET
-Authorization: Bearer **********
-...
-```
-
-## Task Instructions
-
-Implement new API endpoint
-
-## Task Analysis and Planning
-
-Before starting, explain the task back to me in your own words. Ask for any clarifications if needed. Once you're clear, ask to proceed.
-
-Then, outline a plan for the task. Finally, use your plan to complete the task.
-
-Prompt has been copied to clipboard. Final size: 1500 characters (~ 375 tokens)
-```
-
-## License
-
-This project is licensed under the MIT License.

kopipasta-0.17.0.dist-info/RECORD

@@ -1,8 +0,0 @@
-kopipasta/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-kopipasta/main.py,sha256=jDcsdCYE3u2EuMKZbcpkyayxgj5soLLm7XWbhG46Poo,34657
-kopipasta-0.17.0.dist-info/LICENSE,sha256=xw4C9TAU7LFu4r_MwSbky90uzkzNtRwAo3c51IWR8lk,1091
-kopipasta-0.17.0.dist-info/METADATA,sha256=4JjrjvjRucGOIlIEP3PvPDLefJ-Dq7JTDGd1HIbiraA,5646
-kopipasta-0.17.0.dist-info/WHEEL,sha256=GV9aMThwP_4oNCtvEC2ec3qUYutgWeAzklro_0m4WJQ,91
-kopipasta-0.17.0.dist-info/entry_points.txt,sha256=but54qDNz1-F8fVvGstq_QID5tHjczP7bO7rWLFkc6Y,50
-kopipasta-0.17.0.dist-info/top_level.txt,sha256=iXohixMuCdw8UjGDUp0ouICLYBDrx207sgZIJ9lxn0o,10
-kopipasta-0.17.0.dist-info/RECORD,,