up-cli 0.1.1__py3-none-any.whl → 0.5.0__py3-none-any.whl

up/commands/sync.py

@@ -0,0 +1,317 @@
+"""up sync - Sync all systems."""
+
+from pathlib import Path
+
+import click
+from rich.console import Console
+from rich.panel import Panel
+
+console = Console()
+
+
+@click.command()
+@click.option("--memory/--no-memory", default=True, help="Sync memory system")
+@click.option("--docs/--no-docs", default=True, help="Update docs")
+@click.option("--full", is_flag=True, help="Full sync including file re-indexing")
+def sync_cmd(memory: bool, docs: bool, full: bool):
+    """Sync all up systems.
+    
+    Updates memory index, refreshes docs, and ensures all systems
+    are in sync with current project state.
+    
+    \b
+    Examples:
+      up sync              # Standard sync
+      up sync --full       # Full sync with file re-indexing
+      up sync --no-docs    # Memory only
+    """
+    cwd = Path.cwd()
+    
+    console.print(Panel.fit(
+        "[bold blue]Syncing Systems[/]",
+        border_style="blue"
+    ))
+    
+    results = {}
+    
+    # Initialize event system
+    from up.events import initialize_event_system
+    initialize_event_system(cwd)
+    
+    steps = []
+    if memory:
+        steps.append(("Memory", _sync_memory))
+    if docs:
+        steps.append(("Docs", _sync_docs))
+    if full:
+        steps.append(("Files", _sync_files))
+    
+    for name, func in steps:
+        console.print(f"[dim]Syncing {name}...[/]", end=" ")
+        try:
+            result = func(cwd, full)
+            results[name.lower()] = result
+            details = ", ".join(f"{k}={v}" for k, v in result.items() if v)
+            console.print(f"[green]✓[/] {details or 'done'}")
+        except Exception as e:
+            results[name.lower()] = {"error": str(e)}
+            console.print(f"[red]✗[/] {e}")
+    
+    # Summary
+    console.print()
+    errors = [k for k, v in results.items() if "error" in v]
+    if errors:
+        console.print(f"[yellow]Completed with {len(errors)} error(s)[/]")
+    else:
+        console.print("[green]✓[/] All systems synced")
+
+
+def _sync_memory(workspace: Path, full: bool) -> dict:
+    """Sync memory system."""
+    import os
+    import sys
+    import warnings
+    import logging
+    from up.memory import MemoryManager, _check_chromadb
+    
+    # Suppress noisy warnings and logs
+    os.environ["TOKENIZERS_PARALLELISM"] = "false"
+    warnings.filterwarnings("ignore")
+    logging.getLogger("chromadb").setLevel(logging.ERROR)
+    
+    # Redirect stderr temporarily to suppress ChromaDB noise
+    old_stderr = sys.stderr
+    sys.stderr = open(os.devnull, 'w')
+    
+    try:
+        # Use JSON backend if ChromaDB not available (faster, no model download)
+        use_vectors = _check_chromadb()
+        
+        manager = MemoryManager(workspace, use_vectors=use_vectors)
+        results = manager.sync()
+        
+        return {
+            "commits": results.get("commits_indexed", 0),
+            "files": results.get("files_indexed", 0),
+            "backend": manager._backend,
+        }
+    finally:
+        sys.stderr.close()
+        sys.stderr = old_stderr
+
+
+def _sync_docs(workspace: Path, full: bool) -> dict:
+    """Sync documentation."""
+    from datetime import date
+    import re
+    
+    updated = 0
+    
+    # Update CONTEXT.md date
+    context_file = workspace / "docs" / "CONTEXT.md"
+    if context_file.exists():
+        content = context_file.read_text()
+        today = date.today().isoformat()
+        new_content = re.sub(
+            r'\*\*Updated\*\*:\s*[\d-]+',
+            f'**Updated**: {today}',
+            content
+        )
+        if new_content != content:
+            context_file.write_text(new_content)
+            updated += 1
+    
+    # Check INDEX.md exists
+    index_file = workspace / "docs" / "INDEX.md"
+    if index_file.exists():
+        updated += 0  # Just checking
+    
+    return {"updated": updated}
+
+
+def _sync_files(workspace: Path, full: bool) -> dict:
+    """Sync file index."""
+    from up.memory import MemoryManager
+    
+    manager = MemoryManager(workspace)
+    indexed = manager.index_file_changes()
+    
+    return {"indexed": indexed}
+
+
+def check_hooks_installed(workspace: Path) -> dict:
+    """Check if up-cli hooks are installed.
+    
+    Returns dict with status of each hook.
+    """
+    git_dir = workspace / ".git"
+    if not git_dir.exists():
+        return {"git": False, "post_commit": False, "post_checkout": False}
+    
+    hooks_dir = git_dir / "hooks"
+    
+    result = {"git": True, "post_commit": False, "post_checkout": False}
+    
+    post_commit = hooks_dir / "post-commit"
+    if post_commit.exists() and "up-cli" in post_commit.read_text():
+        result["post_commit"] = True
+    
+    post_checkout = hooks_dir / "post-checkout"
+    if post_checkout.exists() and "up-cli" in post_checkout.read_text():
+        result["post_checkout"] = True
+    
+    return result
+
+
+@click.command()
+@click.option("--uninstall", is_flag=True, help="Remove hooks instead of installing")
+@click.option("--check", is_flag=True, help="Check if hooks are installed")
+def hooks_cmd(uninstall: bool, check: bool):
+    """Install or uninstall git hooks for automatic syncing.
+    
+    Installs:
+    - post-commit: Auto-index commits to memory
+    - post-checkout: Update context on branch switch
+    
+    \b
+    Examples:
+      up hooks              # Install hooks
+      up hooks --check      # Check status
+      up hooks --uninstall  # Remove hooks
+    """
+    cwd = Path.cwd()
+    git_dir = cwd / ".git"
+    
+    if not git_dir.exists():
+        console.print("[red]Error:[/] Not a git repository")
+        raise SystemExit(1)
+    
+    hooks_dir = git_dir / "hooks"
+    hooks_dir.mkdir(exist_ok=True)
+    
+    if check:
+        status = check_hooks_installed(cwd)
+        console.print("\n[bold]Git Hooks Status:[/]")
+        console.print(f"  post-commit:   {'[green]✓ installed[/]' if status['post_commit'] else '[yellow]✗ not installed[/]'}")
+        console.print(f"  post-checkout: {'[green]✓ installed[/]' if status['post_checkout'] else '[yellow]✗ not installed[/]'}")
+        
+        if not status['post_commit'] or not status['post_checkout']:
+            console.print("\n[dim]Run 'up hooks' to install missing hooks[/]")
+        return
+    
+    if uninstall:
+        _uninstall_hooks(hooks_dir)
+    else:
+        _install_hooks(hooks_dir)
+
+
+def _install_hooks(hooks_dir: Path):
+    """Install git hooks."""
+    
+    # Post-commit hook
+    post_commit = hooks_dir / "post-commit"
+    post_commit_content = '''#!/bin/bash
+# up-cli auto-sync hook
+# Indexes commits to memory automatically
+
+# Run in background to not slow down commits
+(
+    # Wait a moment for git to finish
+    sleep 1
+    
+    # Sync memory with latest commit
+    if command -v up &> /dev/null; then
+        up memory sync 2>/dev/null
+    elif command -v python3 &> /dev/null; then
+        python3 -m up.memory sync 2>/dev/null
+    fi
+) &
+
+exit 0
+'''
+    
+    _write_hook(post_commit, post_commit_content)
+    console.print("[green]✓[/] Installed post-commit hook")
+    
+    # Post-checkout hook (for branch switches)
+    post_checkout = hooks_dir / "post-checkout"
+    post_checkout_content = '''#!/bin/bash
+# up-cli context update hook
+# Updates context when switching branches
+
+PREV_HEAD=$1
+NEW_HEAD=$2
+BRANCH_CHECKOUT=$3
+
+# Only run on branch checkout, not file checkout
+if [ "$BRANCH_CHECKOUT" = "1" ]; then
+    (
+        sleep 1
+        if command -v up &> /dev/null; then
+            up sync --no-memory 2>/dev/null
+        fi
+    ) &
+fi
+
+exit 0
+'''
+    
+    _write_hook(post_checkout, post_checkout_content)
+    console.print("[green]✓[/] Installed post-checkout hook")
+    
+    console.print("\n[bold]Hooks installed![/]")
+    console.print("Memory will auto-sync on commits.")
+
+
+def _write_hook(path: Path, content: str):
+    """Write hook file with executable permissions."""
+    import stat
+    
+    # Check for existing hook
+    if path.exists():
+        existing = path.read_text()
+        if "up-cli" in existing:
+            # Already our hook, overwrite
+            pass
+        else:
+            # User has custom hook, append
+            content = existing + "\n\n" + content
+    
+    path.write_text(content)
+    path.chmod(path.stat().st_mode | stat.S_IEXEC)
+
+
+def _uninstall_hooks(hooks_dir: Path):
+    """Remove up-cli hooks."""
+    
+    for hook_name in ["post-commit", "post-checkout"]:
+        hook_path = hooks_dir / hook_name
+        if hook_path.exists():
+            content = hook_path.read_text()
+            if "up-cli" in content:
+                # Remove our section
+                lines = content.split("\n")
+                new_lines = []
+                skip = False
+                for line in lines:
+                    if "up-cli" in line:
+                        skip = True
+                    elif skip and line.startswith("exit"):
+                        skip = False
+                        continue
+                    elif not skip:
+                        new_lines.append(line)
+                
+                new_content = "\n".join(new_lines).strip()
+                if new_content:
+                    hook_path.write_text(new_content)
+                else:
+                    hook_path.unlink()
+                
+                console.print(f"[green]✓[/] Removed {hook_name} hook")
+    
+    console.print("\n[bold]Hooks uninstalled![/]")
+
+
+if __name__ == "__main__":
+    sync_cmd()

up/commands/vibe.py

@@ -0,0 +1,304 @@
+"""up save/reset/diff - Vibe coding safety commands.
+
+These commands provide the core safety rails for AI-assisted development:
+- up save: Create checkpoint before AI work
+- up reset: Restore to checkpoint when AI fails
+- up diff: Review AI changes before accepting
+"""
+
+import subprocess
+import sys
+from pathlib import Path
+
+import click
+from rich.console import Console
+from rich.panel import Panel
+from rich.syntax import Syntax
+from rich.table import Table
+
+from up.core.checkpoint import (
+    get_checkpoint_manager,
+    CheckpointManager,
+    CheckpointNotFoundError,
+    NotAGitRepoError,
+)
+from up.core.state import get_state_manager
+
+console = Console()
+
+
+# =============================================================================
+# up save - Create checkpoint
+# =============================================================================
+
+@click.command("save")
+@click.argument("message", required=False)
+@click.option("--task", "-t", help="Associated task ID")
+@click.option("--quiet", "-q", is_flag=True, help="Minimal output")
+def save_cmd(message: str, task: str, quiet: bool):
+    """Create a checkpoint before AI work.
+    
+    Automatically commits any dirty files and creates a tag for easy recovery.
+    Use 'up reset' to restore if AI generation goes wrong.
+    
+    \b
+    Examples:
+      up save                      # Auto-named checkpoint
+      up save "before auth"        # Named checkpoint
+      up save -t US-004            # Link to task
+    """
+    cwd = Path.cwd()
+    
+    try:
+        manager = get_checkpoint_manager(cwd)
+        
+        # Check for changes
+        has_changes = manager._has_changes()
+        
+        if not has_changes and not quiet:
+            console.print("[dim]No changes to checkpoint (working tree clean)[/]")
+        
+        # Create checkpoint
+        metadata = manager.save(
+            message=message,
+            task_id=task,
+            auto_commit=True
+        )
+        
+        if quiet:
+            console.print(metadata.id)
+        else:
+            console.print(f"[green]✓[/] Checkpoint created: [cyan]{metadata.id}[/]")
+            if metadata.files_changed > 0:
+                console.print(f"  Committed {metadata.files_changed} file(s)")
+            console.print(f"  Commit: {metadata.commit_sha[:8]}")
+            console.print(f"  Tag: {metadata.tag_name}")
+            console.print(f"\nTo restore: [cyan]up reset {metadata.id}[/]")
+        
+    except NotAGitRepoError:
+        console.print("[red]Error:[/] Not a git repository")
+        console.print("Initialize with: [cyan]git init[/]")
+        sys.exit(1)
+    except Exception as e:
+        console.print(f"[red]Error:[/] {e}")
+        sys.exit(1)
+
+
+# =============================================================================
+# up reset - Restore checkpoint
+# =============================================================================
+
+@click.command("reset")
+@click.argument("checkpoint_id", required=False)
+@click.option("--hard", is_flag=True, default=True, help="Hard reset (discard changes)")
+@click.option("--soft", is_flag=True, help="Soft reset (keep changes staged)")
+@click.option("--list", "list_checkpoints", is_flag=True, help="List available checkpoints")
+@click.option("--yes", "-y", is_flag=True, help="Skip confirmation")
+def reset_cmd(checkpoint_id: str, hard: bool, soft: bool, list_checkpoints: bool, yes: bool):
+    """Reset to a checkpoint.
+    
+    Instantly restores your code to a previous checkpoint state.
+    Use when AI generation produces bad results.
+    
+    \b
+    Examples:
+      up reset                     # Reset to last checkpoint
+      up reset cp-20260204-1234    # Reset to specific checkpoint
+      up reset --list              # Show available checkpoints
+    """
+    cwd = Path.cwd()
+    
+    try:
+        manager = get_checkpoint_manager(cwd)
+        
+        # List mode
+        if list_checkpoints:
+            checkpoints = manager.list_checkpoints(limit=20)
+            
+            if not checkpoints:
+                console.print("[dim]No checkpoints available[/]")
+                return
+            
+            table = Table(title="Available Checkpoints")
+            table.add_column("ID", style="cyan")
+            table.add_column("Message")
+            table.add_column("Branch")
+            table.add_column("Time")
+            
+            for cp in checkpoints:
+                table.add_row(
+                    cp.id,
+                    cp.message[:40] + "..." if len(cp.message) > 40 else cp.message,
+                    cp.branch,
+                    cp.created_at[:19]
+                )
+            
+            console.print(table)
+            return
+        
+        # Get checkpoint info for confirmation
+        if checkpoint_id:
+            target = checkpoint_id
+        else:
+            last = manager.get_last_checkpoint()
+            if not last:
+                console.print("[yellow]No checkpoints available[/]")
+                console.print("Create one with: [cyan]up save[/]")
+                return
+            target = last.id
+        
+        # Show what will be reset
+        stats = manager.diff_stats(target)
+        
+        console.print(f"[bold]Reset to checkpoint:[/] {target}")
+        if stats["files"] > 0:
+            console.print(f"  Changes to discard: {stats['files']} files, "
+                         f"+{stats['insertions']} -{stats['deletions']}")
+        
+        # Confirm
+        if not yes:
+            if not click.confirm("Proceed with reset?"):
+                console.print("[dim]Cancelled[/]")
+                return
+        
+        # Perform reset
+        use_hard = not soft
+        metadata = manager.restore(checkpoint_id=target, hard=use_hard)
+        
+        console.print(f"\n[green]✓[/] Reset to [cyan]{metadata.id}[/]")
+        console.print(f"  Commit: {metadata.commit_sha[:8]}")
+        
+        # Update state
+        state_manager = get_state_manager(cwd)
+        state_manager.state.loop.consecutive_failures = 0  # Reset doom loop counter
+        state_manager.save()
+        
+    except NotAGitRepoError:
+        console.print("[red]Error:[/] Not a git repository")
+        sys.exit(1)
+    except CheckpointNotFoundError as e:
+        console.print(f"[red]Error:[/] {e}")
+        console.print("List available: [cyan]up reset --list[/]")
+        sys.exit(1)
+    except Exception as e:
+        console.print(f"[red]Error:[/] {e}")
+        sys.exit(1)
+
+
+# =============================================================================
+# up diff - Review AI changes
+# =============================================================================
+
+@click.command("diff")
+@click.argument("checkpoint_id", required=False)
+@click.option("--stat", is_flag=True, help="Show only stats")
+@click.option("--accept", "-a", is_flag=True, help="Accept changes and commit")
+@click.option("--reject", "-r", is_flag=True, help="Reject changes and reset")
+@click.option("--message", "-m", help="Commit message (with --accept)")
+def diff_cmd(checkpoint_id: str, stat: bool, accept: bool, reject: bool, message: str):
+    """Review AI changes before accepting.
+    
+    Shows a syntax-highlighted diff of changes since the last checkpoint.
+    Can accept (commit) or reject (reset) the changes.
+    
+    \b
+    Examples:
+      up diff                      # Show diff from last checkpoint
+      up diff --stat               # Show stats only
+      up diff --accept             # Accept and commit changes
+      up diff --reject             # Reject and reset to checkpoint
+    """
+    cwd = Path.cwd()
+    
+    try:
+        manager = get_checkpoint_manager(cwd)
+        
+        # Get diff
+        diff_output = manager.diff_from_checkpoint(checkpoint_id)
+        stats = manager.diff_stats(checkpoint_id)
+        
+        if not diff_output and stats["files"] == 0:
+            console.print("[dim]No changes since checkpoint[/]")
+            return
+        
+        # Stats mode
+        if stat:
+            console.print(Panel.fit(
+                f"[bold]Changes since checkpoint[/]\n\n"
+                f"Files: {stats['files']}\n"
+                f"Insertions: [green]+{stats['insertions']}[/]\n"
+                f"Deletions: [red]-{stats['deletions']}[/]",
+                border_style="blue"
+            ))
+            return
+        
+        # Reject mode
+        if reject:
+            console.print("[yellow]Rejecting changes...[/]")
+            manager.restore(checkpoint_id=checkpoint_id)
+            console.print("[green]✓[/] Changes rejected, reset to checkpoint")
+            return
+        
+        # Accept mode
+        if accept:
+            # Commit the changes
+            result = subprocess.run(
+                ["git", "add", "-A"],
+                cwd=cwd,
+                capture_output=True
+            )
+            
+            commit_msg = message or "Accept AI changes"
+            result = subprocess.run(
+                ["git", "commit", "-m", commit_msg],
+                cwd=cwd,
+                capture_output=True,
+                text=True
+            )
+            
+            if result.returncode == 0:
+                console.print(f"[green]✓[/] Changes accepted and committed")
+                console.print(f"  Message: {commit_msg}")
+            else:
+                console.print("[yellow]No changes to commit[/]")
+            return
+        
+        # Show diff
+        console.print(Panel.fit(
+            f"[bold]Changes since checkpoint[/] "
+            f"({stats['files']} files, +{stats['insertions']} -{stats['deletions']})",
+            border_style="blue"
+        ))
+        console.print()
+        
+        # Syntax-highlighted diff
+        syntax = Syntax(diff_output, "diff", theme="monokai", line_numbers=False)
+        console.print(syntax)
+        
+        # Interactive prompt
+        console.print()
+        console.print("[bold]Actions:[/]")
+        console.print("  [cyan]up diff --accept[/]  Accept changes")
+        console.print("  [cyan]up diff --reject[/]  Reject and reset")
+        
+    except NotAGitRepoError:
+        console.print("[red]Error:[/] Not a git repository")
+        sys.exit(1)
+    except Exception as e:
+        console.print(f"[red]Error:[/] {e}")
+        sys.exit(1)
+
+
+# =============================================================================
+# Command Group (for future expansion)
+# =============================================================================
+
+@click.group()
+def vibe():
+    """Vibe coding safety commands."""
+    pass
+
+
+vibe.add_command(save_cmd, name="save")
+vibe.add_command(reset_cmd, name="reset")
+vibe.add_command(diff_cmd, name="diff")