emdash-cli 0.1.46__py3-none-any.whl → 0.1.70__py3-none-any.whl

emdash_cli/commands/agent/session_restore.py

@@ -0,0 +1,210 @@
+"""Session restore prompt for emdash CLI.
+
+Detects recent sessions and offers to restore them with zen styling.
+"""
+
+from datetime import datetime, timedelta
+from pathlib import Path
+from typing import Optional
+
+from rich.console import Console
+from prompt_toolkit import Application
+from prompt_toolkit.key_binding import KeyBindings
+from prompt_toolkit.layout import Layout, HSplit, Window, FormattedTextControl
+from prompt_toolkit.styles import Style
+
+from ...design import (
+    Colors,
+    STATUS_ACTIVE,
+    STATUS_INACTIVE,
+    DOT_BULLET,
+    ARROW_PROMPT,
+    header,
+    footer,
+)
+
+console = Console()
+
+
+def get_recent_session(client, max_age_hours: int = 24) -> Optional[dict]:
+    """Get the most recent session if within max_age.
+
+    Args:
+        client: Emdash client instance
+        max_age_hours: Maximum age in hours for session to be considered recent
+
+    Returns:
+        Session info dict or None
+    """
+    try:
+        sessions = client.list_sessions()
+        if not sessions:
+            return None
+
+        # Sort by updated_at (most recent first)
+        sessions = sorted(sessions, key=lambda s: s.updated_at or "", reverse=True)
+
+        if not sessions:
+            return None
+
+        recent = sessions[0]
+
+        # Check if session is recent enough
+        if recent.updated_at:
+            try:
+                updated = datetime.fromisoformat(recent.updated_at.replace("Z", "+00:00"))
+                cutoff = datetime.now(updated.tzinfo) - timedelta(hours=max_age_hours)
+                if updated < cutoff:
+                    return None
+            except (ValueError, TypeError):
+                pass
+
+        # Only offer to restore if it has messages
+        if recent.message_count and recent.message_count > 0:
+            return {
+                "name": recent.name,
+                "summary": recent.summary,
+                "mode": recent.mode,
+                "message_count": recent.message_count,
+                "updated_at": recent.updated_at,
+            }
+
+        return None
+    except Exception:
+        return None
+
+
+def format_relative_time(iso_time: str) -> str:
+    """Format ISO time as relative time (e.g., '2 hours ago')."""
+    try:
+        dt = datetime.fromisoformat(iso_time.replace("Z", "+00:00"))
+        now = datetime.now(dt.tzinfo)
+        delta = now - dt
+
+        if delta.days > 0:
+            return f"{delta.days} day{'s' if delta.days > 1 else ''} ago"
+        elif delta.seconds >= 3600:
+            hours = delta.seconds // 3600
+            return f"{hours} hour{'s' if hours > 1 else ''} ago"
+        elif delta.seconds >= 60:
+            minutes = delta.seconds // 60
+            return f"{minutes} minute{'s' if minutes > 1 else ''} ago"
+        else:
+            return "just now"
+    except (ValueError, TypeError):
+        return ""
+
+
+def show_session_restore_prompt(session_info: dict) -> tuple[str, Optional[dict]]:
+    """Show prompt to restore a recent session.
+
+    Args:
+        session_info: Dict with session details
+
+    Returns:
+        Tuple of (choice, session_data) where choice is:
+        - 'restore': Restore the session
+        - 'new': Start new session
+        - 'view': View session details first
+    """
+    name = session_info.get("name", "unnamed")
+    summary = session_info.get("summary", "")
+    mode = session_info.get("mode", "code")
+    msg_count = session_info.get("message_count", 0)
+    updated = session_info.get("updated_at", "")
+
+    relative_time = format_relative_time(updated) if updated else ""
+
+    console.print()
+    console.print(f"[{Colors.MUTED}]{header('Session Found', 40)}[/{Colors.MUTED}]")
+    console.print()
+    console.print(f"  [{Colors.DIM}]Previous session from {relative_time}:[/{Colors.DIM}]")
+    console.print()
+    console.print(f"    {DOT_BULLET} [{Colors.MUTED}]{msg_count} messages[/{Colors.MUTED}]")
+    if summary:
+        truncated = summary[:60] + "..." if len(summary) > 60 else summary
+        console.print(f"    {DOT_BULLET} [{Colors.MUTED}]{truncated}[/{Colors.MUTED}]")
+    console.print(f"    {DOT_BULLET} [{Colors.MUTED}]Mode: {mode}[/{Colors.MUTED}]")
+    console.print()
+    console.print(f"[{Colors.MUTED}]{footer(40)}[/{Colors.MUTED}]")
+
+    selected_index = [0]
+    result = [("new", None)]
+
+    options = [
+        ("restore", "Restore this session"),
+        ("new", "Start new session"),
+    ]
+
+    kb = KeyBindings()
+
+    @kb.add("up")
+    @kb.add("k")
+    def move_up(event):
+        selected_index[0] = (selected_index[0] - 1) % len(options)
+
+    @kb.add("down")
+    @kb.add("j")
+    def move_down(event):
+        selected_index[0] = (selected_index[0] + 1) % len(options)
+
+    @kb.add("enter")
+    def select(event):
+        result[0] = (options[selected_index[0]][0], session_info if options[selected_index[0]][0] == "restore" else None)
+        event.app.exit()
+
+    @kb.add("r")
+    def restore(event):
+        result[0] = ("restore", session_info)
+        event.app.exit()
+
+    @kb.add("n")
+    def new(event):
+        result[0] = ("new", None)
+        event.app.exit()
+
+    @kb.add("c-c")
+    @kb.add("escape")
+    def cancel(event):
+        result[0] = ("new", None)
+        event.app.exit()
+
+    def get_formatted_options():
+        lines = []
+        for i, (key, desc) in enumerate(options):
+            indicator = STATUS_ACTIVE if i == selected_index[0] else STATUS_INACTIVE
+            style_class = "selected" if i == selected_index[0] else "option"
+            lines.append((f"class:{style_class}", f"  {indicator} {desc}\n"))
+        lines.append(("class:hint", f"\n{ARROW_PROMPT} r restore  n new  Esc skip"))
+        return lines
+
+    style = Style.from_dict({
+        "selected": f"{Colors.SUCCESS} bold",
+        "option": Colors.MUTED,
+        "hint": f"{Colors.DIM} italic",
+    })
+
+    layout = Layout(
+        HSplit([
+            Window(
+                FormattedTextControl(get_formatted_options),
+                height=5,
+            ),
+        ])
+    )
+
+    app = Application(
+        layout=layout,
+        key_bindings=kb,
+        style=style,
+        full_screen=False,
+    )
+
+    console.print()
+
+    try:
+        app.run()
+    except (KeyboardInterrupt, EOFError):
+        return ("new", None)
+
+    return result[0]

emdash_cli/commands/index.py

@@ -23,16 +23,18 @@ def index():
 
 @index.command("start")
 @click.argument("repo_path", required=False)
-@click.option("--changed-only", is_flag=True, help="Only index changed files")
-@click.option("--skip-git", is_flag=True, help="Skip git history indexing")
-@click.option("--github-prs", default=0, help="Number of GitHub PRs to index")
+@click.option("--full", is_flag=True, help="Force full reindex (default: incremental)")
+@click.option("--with-git", is_flag=True, help="Include git history (Layer B)")
+@click.option("--with-github", is_flag=True, help="Include GitHub PRs (Layer C)")
+@click.option("--github-prs", default=50, help="Number of GitHub PRs to index (when --with-github)")
 @click.option("--detect-communities", is_flag=True, default=True, help="Run community detection")
 @click.option("--describe-communities", is_flag=True, help="Use LLM to describe communities")
 @click.option("--model", "-m", default=None, help="Model for community descriptions")
 def index_start(
     repo_path: str | None,
-    changed_only: bool,
-    skip_git: bool,
+    full: bool,
+    with_git: bool,
+    with_github: bool,
     github_prs: int,
     detect_communities: bool,
     describe_communities: bool,
@@ -42,11 +44,18 @@ def index_start(
 
     If REPO_PATH is not provided, indexes the current directory.
 
+    By default, indexes only code structure (Layer A - AST parsing).
+    Git history (Layer B) and GitHub PRs (Layer C) are skipped unless enabled.
+
+    Environment variables:
+        EMDASH_INDEX_GIT=true      Enable git history indexing
+        EMDASH_INDEX_GITHUB=true   Enable GitHub PR indexing
+
     Examples:
-        emdash index start                    # Index current directory
-        emdash index start /path/to/repo      # Index specific repo
-        emdash index start --changed-only     # Only index changed files
-        emdash index start --github-prs 50    # Also index 50 PRs
+        emdash index start                    # Code only (Layer A)
+        emdash index start --with-git         # Include git history
+        emdash index start --with-github      # Include GitHub PRs
+        emdash index start --full             # Force full reindex
     """
     # Default to current directory
     if not repo_path:
@@ -58,11 +67,20 @@ def index_start(
 
     console.print(f"\n[bold cyan]Indexing[/bold cyan] {repo_path}\n")
 
+    # Check environment variables for Layer B and C
+    env_index_git = os.environ.get("EMDASH_INDEX_GIT", "").lower() == "true"
+    env_index_github = os.environ.get("EMDASH_INDEX_GITHUB", "").lower() == "true"
+
+    # Enable git/github if flag passed OR env var set
+    index_git = with_git or env_index_git
+    index_github = with_github or env_index_github
+
     # Build options
+    # Incremental mode is default (changed_only=True), unless --full is passed
     options = {
-        "changed_only": changed_only,
-        "index_git": not skip_git,
-        "index_github": github_prs,
+        "changed_only": not full,
+        "index_git": index_git,
+        "index_github": github_prs if index_github else 0,
         "detect_communities": detect_communities,
         "describe_communities": describe_communities,
     }
@@ -82,7 +100,7 @@ def index_start(
         ) as progress:
             task = progress.add_task("Starting...", total=100)
 
-            for line in client.index_start_stream(repo_path, changed_only):
+            for line in client.index_start_stream(repo_path, not full):
                 line = line.strip()
                 if line.startswith("event: "):
                     continue
@@ -160,6 +178,86 @@ def _show_completion(repo_path: str, stats: dict, client: EmdashClient) -> None:
     console.print()
 
 
+@index.command("hook")
+@click.argument("action", type=click.Choice(["install", "uninstall"]))
+@click.argument("repo_path", required=False)
+def index_hook(action: str, repo_path: str | None):
+    """Install or uninstall the post-commit hook for automatic indexing.
+
+    The hook runs 'emdash index start' after each commit to keep the index updated.
+
+    Examples:
+        emdash index hook install             # Install in current repo
+        emdash index hook install /path/repo  # Install in specific repo
+        emdash index hook uninstall           # Remove the hook
+    """
+    from pathlib import Path
+
+    # Default to current directory
+    if not repo_path:
+        repo_path = os.getcwd()
+
+    hooks_dir = Path(repo_path) / ".git" / "hooks"
+    hook_path = hooks_dir / "post-commit"
+
+    if not hooks_dir.exists():
+        console.print(f"[red]Error:[/red] Not a git repository: {repo_path}")
+        raise click.Abort()
+
+    hook_content = """#!/bin/sh
+# emdash post-commit hook - auto-reindex on commit
+# Installed by: emdash index hook install
+
+# Run indexing in background to not block the commit
+emdash index start > /dev/null 2>&1 &
+"""
+
+    if action == "install":
+        # Check if hook already exists
+        if hook_path.exists():
+            existing = hook_path.read_text()
+            if "emdash" in existing:
+                console.print("[yellow]Hook already installed[/yellow]")
+                return
+            else:
+                # Append to existing hook
+                console.print("[yellow]Appending to existing post-commit hook[/yellow]")
+                with open(hook_path, "a") as f:
+                    f.write("\n# emdash auto-index\nemdash index start > /dev/null 2>&1 &\n")
+        else:
+            # Create new hook
+            hook_path.write_text(hook_content)
+
+        # Make executable
+        hook_path.chmod(0o755)
+        console.print(f"[green]Post-commit hook installed:[/green] {hook_path}")
+
+    elif action == "uninstall":
+        if not hook_path.exists():
+            console.print("[yellow]No post-commit hook found[/yellow]")
+            return
+
+        existing = hook_path.read_text()
+        if "emdash" not in existing:
+            console.print("[yellow]No emdash hook found in post-commit[/yellow]")
+            return
+
+        # Check if it's our hook entirely or just contains our line
+        if existing.strip() == hook_content.strip():
+            # It's only our hook, remove the file
+            hook_path.unlink()
+            console.print("[green]Post-commit hook removed[/green]")
+        else:
+            # Remove just our lines
+            lines = existing.split("\n")
+            new_lines = [
+                line for line in lines
+                if "emdash" not in line and "auto-reindex" not in line
+            ]
+            hook_path.write_text("\n".join(new_lines))
+            console.print("[green]Emdash hook lines removed from post-commit[/green]")
+
+
 @index.command("status")
 @click.argument("repo_path", required=False)
 def index_status(repo_path: str | None):