janito 0.3.0__tar.gz

janito-0.3.0/.gitignore

@@ -0,0 +1,88 @@
+# Python
+
+__pycache__/
+*.py[cod]
+*$py.class
+*.so
+.Python
+build/
+develop-eggs/
+dist/
+downloads/
+eggs/
+.eggs/
+lib/
+lib64/
+parts/
+sdist/
+var/
+wheels/
+*.egg-info/
+.installed.cfg
+*.egg
+
+# Virtual Environment
+venv/
+env/
+ENV/
+
+# IDE
+.idea/
+.vscode/
+*.swp
+*.swo
+
+# Testing
+.coverage
+.pytest_cache/
+htmlcov/
+
+# Documentation
+docs/_build/
+site/
+
+# Misc
+.DS_Store
+# Janito
+.janito/history/
+
+# Poetry
+poetry.lock
+.poetry/
+
+# Pipenv
+Pipfile.lock
+.venv
+
+# Tox
+.tox/
+.coverage.*
+nosetests.xml
+coverage.xml
+*.cover
+
+# Additional IDE
+*.sublime-workspace
+*.sublime-project
+.settings/
+.project
+.pydevproject
+.spyproject/
+.ropeproject/
+
+# Jupyter
+.ipynb_checkpoints
+*.ipynb
+
+# mypy
+.mypy_cache/
+.dmypy.json
+dmypy.json
+
+# Environments
+.env
+.venv
+env.bak/
+venv.bak/
+
+.janito/

janito-0.3.0/CHANGELOG.md

@@ -0,0 +1,26 @@
+# Changelog
+
+All notable changes to this project will be documented in this file.
+
+## [0.3.0] - 2024-01-24
+
+### Added
+- Initial changelog
+- Version tracking
+
+### Changed
+- Updated version from 0.1.0 to 0.3.0
+
+## [0.1.0] - Initial Release
+
+### Added
+- Initial release of Janito CLI
+- AI-powered code analysis and modifications
+- Interactive console mode
+- Support for multiple file types
+- Syntax validation for Python files
+- Interactive change preview and confirmation
+- History tracking of changes
+- Debug and verbose output modes
+- Question-answering about codebase
+- File scanning preview

janito-0.3.0/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2024 João Pinto
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

janito-0.3.0/PKG-INFO

@@ -0,0 +1,138 @@
+Metadata-Version: 2.3
+Name: janito
+Version: 0.3.0
+Summary: A CLI tool for software development tasks powered by AI
+Project-URL: Homepage, https://github.com/joaompinto/janito
+Project-URL: Repository, https://github.com/joaompinto/janito.git
+Author-email: João Pinto <lamego.pinto@gmail.com>
+License: MIT
+Classifier: Development Status :: 4 - Beta
+Classifier: Environment :: Console
+Classifier: Intended Audience :: Developers
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Programming Language :: Python :: 3.8
+Classifier: Programming Language :: Python :: 3.9
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Topic :: Software Development
+Requires-Python: >=3.8
+Requires-Dist: anthropic
+Requires-Dist: pathspec
+Requires-Dist: rich
+Requires-Dist: typer
+Description-Content-Type: text/markdown
+
+# 🤖 Janito CLI
+
+A CLI tool for software development tasks powered by AI.
+
+Janito is an AI-powered assistant that helps automate common software development tasks like refactoring, documentation updates, and code optimization.
+
+## 📥 Installation
+
+```bash
+# Install from PyPI
+pip install janito
+
+# Install from source
+git clone https://github.com/joaompinto/janito.git
+cd janito
+pip install -e .
+```
+
+## ⚡ Requirements
+
+- Python 3.8+
+- Anthropic API key
+- Required packages (automatically installed):
+  - typer
+  - pathspec
+  - rich
+
+## ⚙️ Configuration
+
+### 🔑 API Key Setup
+Janito requires an Anthropic API key to function. Set it as an environment variable:
+
+```bash
+export ANTHROPIC_API_KEY='your-api-key-here'
+```
+
+You can also add this to your shell profile (~/.bashrc, ~/.zshrc, etc.) for persistence.
+
+## 📖 Usage
+
+Janito can be used in two modes: Command Line or Interactive Console.
+
+### 💻 Command Line Mode
+
+```bash
+janito REQUEST [OPTIONS]
+```
+
+#### Arguments
+- `REQUEST`: The modification request
+
+#### Options
+- `-w, --workdir PATH`: Working directory (defaults to current directory)
+- `--raw`: Print raw response instead of markdown format
+- `--play PATH`: Replay a saved prompt file
+- `-i, --include PATH`: Additional paths to include in analysis
+- `--debug`: Show debug information
+- `-v, --verbose`: Show verbose output
+- `--ask`: Ask a question about the codebase
+- `--scan`: Preview files that would be analyzed
+
+### 🖥️ Interactive Console Mode
+
+Start the interactive console by running `janito` without arguments:
+
+```bash
+janito
+```
+
+In console mode, you can:
+- Enter requests directly
+- Navigate history with up/down arrows
+- Use special commands starting with /
+
+### 📝 Examples
+
+```bash
+# Command Line Mode Examples
+janito "create docstrings for all functions"
+janito "add error handling" -w ./myproject
+janito "update tests" -i ./tests -i ./lib
+janito --ask "explain the authentication flow"
+janito --scan  # Preview files to be analyzed
+
+# Console Mode
+janito  # Starts interactive session
+```
+
+## ✨ Features
+
+- 🤖 AI-powered code analysis and modifications
+- 💻 Interactive console mode for continuous interaction
+- 📁 Support for multiple file types
+- ✅ Syntax validation for Python files
+- 👀 Interactive change preview and confirmation
+- 📜 History tracking of all changes
+- 🐛 Debug and verbose output modes
+- ❓ Question-answering about codebase
+- 🔍 File scanning preview
+
+## 📚 History and Debugging
+
+Changes are automatically saved in `.janito/history/` with timestamps:
+- `*_analysis.txt`: Initial analysis
+- `*_selected.txt`: Selected implementation
+- `*_changes.txt`: Actual changes
+
+Enable debug mode for detailed logging:
+```bash
+janito "request" --debug
+```
+
+## 📄 License
+
+This project is licensed under the MIT License - see the [LICENSE](LICENSE) file for details.

janito-0.3.0/README.md

@@ -0,0 +1,115 @@
+# 🤖 Janito CLI
+
+A CLI tool for software development tasks powered by AI.
+
+Janito is an AI-powered assistant that helps automate common software development tasks like refactoring, documentation updates, and code optimization.
+
+## 📥 Installation
+
+```bash
+# Install from PyPI
+pip install janito
+
+# Install from source
+git clone https://github.com/joaompinto/janito.git
+cd janito
+pip install -e .
+```
+
+## ⚡ Requirements
+
+- Python 3.8+
+- Anthropic API key
+- Required packages (automatically installed):
+  - typer
+  - pathspec
+  - rich
+
+## ⚙️ Configuration
+
+### 🔑 API Key Setup
+Janito requires an Anthropic API key to function. Set it as an environment variable:
+
+```bash
+export ANTHROPIC_API_KEY='your-api-key-here'
+```
+
+You can also add this to your shell profile (~/.bashrc, ~/.zshrc, etc.) for persistence.
+
+## 📖 Usage
+
+Janito can be used in two modes: Command Line or Interactive Console.
+
+### 💻 Command Line Mode
+
+```bash
+janito REQUEST [OPTIONS]
+```
+
+#### Arguments
+- `REQUEST`: The modification request
+
+#### Options
+- `-w, --workdir PATH`: Working directory (defaults to current directory)
+- `--raw`: Print raw response instead of markdown format
+- `--play PATH`: Replay a saved prompt file
+- `-i, --include PATH`: Additional paths to include in analysis
+- `--debug`: Show debug information
+- `-v, --verbose`: Show verbose output
+- `--ask`: Ask a question about the codebase
+- `--scan`: Preview files that would be analyzed
+
+### 🖥️ Interactive Console Mode
+
+Start the interactive console by running `janito` without arguments:
+
+```bash
+janito
+```
+
+In console mode, you can:
+- Enter requests directly
+- Navigate history with up/down arrows
+- Use special commands starting with /
+
+### 📝 Examples
+
+```bash
+# Command Line Mode Examples
+janito "create docstrings for all functions"
+janito "add error handling" -w ./myproject
+janito "update tests" -i ./tests -i ./lib
+janito --ask "explain the authentication flow"
+janito --scan  # Preview files to be analyzed
+
+# Console Mode
+janito  # Starts interactive session
+```
+
+## ✨ Features
+
+- 🤖 AI-powered code analysis and modifications
+- 💻 Interactive console mode for continuous interaction
+- 📁 Support for multiple file types
+- ✅ Syntax validation for Python files
+- 👀 Interactive change preview and confirmation
+- 📜 History tracking of all changes
+- 🐛 Debug and verbose output modes
+- ❓ Question-answering about codebase
+- 🔍 File scanning preview
+
+## 📚 History and Debugging
+
+Changes are automatically saved in `.janito/history/` with timestamps:
+- `*_analysis.txt`: Initial analysis
+- `*_selected.txt`: Selected implementation
+- `*_changes.txt`: Actual changes
+
+Enable debug mode for detailed logging:
+```bash
+janito "request" --debug
+```
+
+## 📄 License
+
+This project is licensed under the MIT License - see the [LICENSE](LICENSE) file for details.

janito-0.3.0/janito/__init__.py

@@ -0,0 +1,2 @@
+
+# Empty file to make the directory a Python package

janito-0.3.0/janito/__main__.py

@@ -0,0 +1,260 @@
+import typer
+from typing import Optional, Dict, Any, List
+from pathlib import Path
+from janito.claude import ClaudeAPIAgent
+import shutil
+from janito.prompts import (
+    build_request_analisys_prompt,
+    build_selected_option_prompt,
+    SYSTEM_PROMPT,
+    parse_options
+)
+from rich.console import Console
+from rich.markdown import Markdown
+import re
+import tempfile
+import json
+from rich.syntax import Syntax
+from janito.contentchange import (
+    handle_changes_file, 
+    get_file_type,
+    parse_block_changes,
+    preview_and_apply_changes,
+    format_parsed_changes,
+)
+from rich.table import Table
+from rich.columns import Columns
+from rich.panel import Panel
+from rich.text import Text
+from rich.rule import Rule
+from rich import box
+from datetime import datetime, timezone
+from itertools import chain
+from janito.scan import collect_files_content, is_dir_empty, preview_scan
+from janito.qa import ask_question, display_answer
+from rich.prompt import Prompt, Confirm
+from janito.config import config
+from importlib.metadata import version
+
+def get_version() -> str:
+    try:
+        return version("janito")
+    except:
+        return "dev"
+
+def format_analysis(analysis: str, raw: bool = False, claude: Optional[ClaudeAPIAgent] = None) -> None:
+    """Format and display the analysis output"""
+    console = Console()
+    if raw and claude:
+        console.print("\n=== Message History ===")
+        for role, content in claude.messages_history:
+            console.print(f"\n[bold cyan]{role.upper()}:[/bold cyan]")
+            console.print(content)
+        console.print("\n=== End Message History ===\n")
+    else:
+        md = Markdown(analysis)
+        console.print(md)
+
+def prompt_user(message: str, choices: List[str] = None) -> str:
+    """Display a prominent user prompt with optional choices"""
+    console = Console()
+    console.print()
+    console.print(Rule(" User Input Required ", style="bold cyan"))
+    
+    if choices:
+        choice_text = f"[cyan]Options: {', '.join(choices)}[/cyan]"
+        console.print(Panel(choice_text, box=box.ROUNDED))
+    
+    return Prompt.ask(f"[bold cyan]> {message}[/bold cyan]")
+
+def get_option_selection() -> int:
+    """Get user input for option selection"""
+    while True:
+        try:
+            option = int(prompt_user("Select option number"))
+            return option
+        except ValueError:
+            console = Console()
+            console.print("[red]Please enter a valid number[/red]")
+
+def get_history_path(workdir: Path) -> Path:
+    """Create and return the history directory path"""
+    history_dir = workdir / '.janito' / 'history'
+    history_dir.mkdir(parents=True, exist_ok=True)
+    return history_dir
+
+def get_timestamp() -> str:
+    """Get current UTC timestamp in YMD_HMS format with leading zeros"""
+    return datetime.now(timezone.utc).strftime('%Y%m%d_%H%M%S')
+
+def save_prompt_to_file(prompt: str) -> Path:
+    """Save prompt to a named temporary file that won't be deleted"""
+    temp_file = tempfile.NamedTemporaryFile(prefix='selected_', suffix='.txt', delete=False)
+    temp_path = Path(temp_file.name)
+    temp_path.write_text(prompt)
+    return temp_path
+
+def save_to_file(content: str, prefix: str, workdir: Path) -> Path:
+    """Save content to a timestamped file in history directory"""
+    history_dir = get_history_path(workdir)
+    timestamp = get_timestamp()
+    filename = f"{timestamp}_{prefix}.txt"
+    file_path = history_dir / filename
+    file_path.write_text(content)
+    return file_path
+
+def handle_option_selection(claude: ClaudeAPIAgent, initial_response: str, request: str, raw: bool = False, workdir: Optional[Path] = None, include: Optional[List[Path]] = None) -> None:
+    """Handle option selection and implementation details"""
+    option = get_option_selection()
+    paths_to_scan = [workdir] if workdir else []
+    if include:
+        paths_to_scan.extend(include)
+    files_content = collect_files_content(paths_to_scan, workdir) if paths_to_scan else ""
+    
+    selected_prompt = build_selected_option_prompt(option, request, initial_response, files_content)
+    prompt_file = save_to_file(selected_prompt, 'selected', workdir)
+    if config.verbose:
+        print(f"\nSelected prompt saved to: {prompt_file}")
+    
+    selected_response = claude.send_message(selected_prompt)
+    changes_file = save_to_file(selected_response, 'changes', workdir)
+    if config.verbose:
+        print(f"\nChanges saved to: {changes_file}")
+    
+    changes = parse_block_changes(selected_response)
+    preview_and_apply_changes(changes, workdir)
+
+def replay_saved_file(filepath: Path, claude: ClaudeAPIAgent, workdir: Path, raw: bool = False) -> None:
+    """Process a saved prompt file and display the response"""
+    if not filepath.exists():
+        raise FileNotFoundError(f"File {filepath} not found")
+    
+    file_type = get_file_type(filepath)
+    content = filepath.read_text()
+    
+    if file_type == 'changes':
+        changes = parse_block_changes(content)
+        preview_and_apply_changes(changes, workdir)
+    elif file_type == 'analysis':
+        format_analysis(content, raw, claude)
+        handle_option_selection(claude, content, content, raw, workdir)
+    elif file_type == 'selected':
+        if raw:
+            console = Console()
+            console.print("\n=== Prompt Content ===")
+            console.print(content)
+            console.print("=== End Prompt Content ===\n")
+        response = claude.send_message(content)
+        changes_file = save_to_file(response, 'changes_', workdir)
+        print(f"\nChanges saved to: {changes_file}")
+        
+        changes = parse_block_changes(response)
+        preview_and_apply_changes(preview_changes, workdir)
+    else:
+        response = claude.send_message(content)
+        format_analysis(response, raw)
+
+def process_question(question: str, workdir: Path, include: List[Path], raw: bool, claude: ClaudeAPIAgent) -> None:
+    """Process a question about the codebase"""
+    paths_to_scan = [workdir] if workdir else []
+    if include:
+        paths_to_scan.extend(include)
+    files_content = collect_files_content(paths_to_scan, workdir)
+
+    answer = ask_question(question, files_content, claude)
+    display_answer(answer, raw)
+
+def ensure_workdir(workdir: Path) -> Path:
+    """Ensure working directory exists, prompt for creation if it doesn't"""
+    if workdir.exists():
+        return workdir
+        
+    console = Console()
+    console.print(f"\n[yellow]Directory does not exist:[/yellow] {workdir}")
+    if Confirm.ask("Create directory?"):
+        workdir.mkdir(parents=True)
+        console.print(f"[green]Created directory:[/green] {workdir}")
+        return workdir
+    raise typer.Exit(1)
+
+def typer_main(
+    request: Optional[str] = typer.Argument(None, help="The modification request"),
+    ask: Optional[str] = typer.Option(None, "--ask", help="Ask a question about the codebase"),
+    workdir: Optional[Path] = typer.Option(None, "-w", "--workdir", 
+                                         help="Working directory (defaults to current directory)", 
+                                         file_okay=False, dir_okay=True),
+    raw: bool = typer.Option(False, "--raw", help="Print raw response instead of markdown format"),
+    play: Optional[Path] = typer.Option(None, "--play", help="Replay a saved prompt file"),
+    include: Optional[List[Path]] = typer.Option(None, "-i", "--include", help="Additional paths to include in analysis", exists=True),
+    debug: bool = typer.Option(False, "--debug", help="Show debug information"),
+    debug_line: Optional[int] = typer.Option(None, "--debug-line", help="Show debug information only for specific line number"),
+    verbose: bool = typer.Option(False, "-v", "--verbose", help="Show verbose output"),
+    scan: bool = typer.Option(False, "--scan", help="Preview files that would be analyzed"),
+    version: bool = typer.Option(False, "--version", help="Show version and exit"),
+) -> None:
+    """
+    Analyze files and provide modification instructions.
+    """
+    if version:
+        console = Console()
+        console.print(f"Janito v{get_version()}")
+        raise typer.Exit()
+
+    config.set_debug(debug)
+    config.set_verbose(verbose)
+    config.set_debug_line(debug_line)
+
+    claude = ClaudeAPIAgent(system_prompt=SYSTEM_PROMPT)
+
+    if not any([request, ask, play, scan]):
+        workdir = workdir or Path.cwd()
+        workdir = ensure_workdir(workdir)
+        from janito.console import start_console_session
+        start_console_session(workdir, include)
+        return
+
+    workdir = workdir or Path.cwd()
+    workdir = ensure_workdir(workdir)
+    
+    if include:
+        include = [
+            path if path.is_absolute() else (workdir / path).resolve()
+            for path in include
+        ]
+    
+    if ask:
+        process_question(ask, workdir, include, raw, claude)
+        return
+
+    if scan:
+        paths_to_scan = include if include else [workdir]
+        preview_scan(paths_to_scan, workdir)
+        return
+
+    if play:
+        replay_saved_file(play, claude, workdir, raw)
+        return
+
+    paths_to_scan = include if include else [workdir]
+    
+    is_empty = is_dir_empty(workdir)
+    if is_empty and not include:
+        console = Console()
+        console.print("\n[bold blue]Empty directory - will create new files as needed[/bold blue]")
+        files_content = ""
+    else:
+        files_content = collect_files_content(paths_to_scan, workdir)
+    
+    initial_prompt = build_request_analisys_prompt(files_content, request)
+    initial_response = claude.send_message(initial_prompt)
+    analysis_file = save_to_file(initial_response, 'analysis', workdir)
+    
+    format_analysis(initial_response, raw, claude)
+    
+    handle_option_selection(claude, initial_response, request, raw, workdir, include)
+
+def main():
+    typer.run(typer_main)
+
+if __name__ == "__main__":
+    main()