erk 0.4.5__py3-none-any.whl

erk/cli/commands/plan/extraction/create_raw_cmd.py

@@ -0,0 +1,63 @@
+"""Command to create a raw extraction plan from session data."""
+
+import json
+from dataclasses import asdict
+
+import click
+
+from erk.cli.core import discover_repo_context
+from erk.core.context import ErkContext
+from erk_shared.extraction.raw_extraction import create_raw_extraction_plan
+from erk_shared.scratch.markers import PENDING_EXTRACTION_MARKER, delete_marker
+
+
+@click.command("raw")
+@click.option(
+    "--min-size",
+    default=1024,
+    type=int,
+    help="Minimum session size in bytes for selection (default: 1024)",
+)
+@click.option(
+    "--session-id",
+    default=None,
+    type=str,
+    help="Current session ID (for extraction)",
+)
+@click.pass_obj
+def create_raw(ctx: ErkContext, min_size: int, session_id: str | None) -> None:
+    """Create extraction plan with raw session context.
+
+    This command:
+    1. Discovers Claude Code sessions in the project directory
+    2. Auto-selects sessions based on branch context and size
+    3. Preprocesses sessions to compressed XML format
+    4. Creates a GitHub issue with erk-plan and erk-extraction labels
+    5. Posts session content as chunked comments
+    6. Deletes the pending-extraction marker if successful
+
+    Output is JSON with success status, issue URL, and chunk count.
+    """
+    repo = discover_repo_context(ctx, ctx.cwd)
+    repo_root = repo.root
+
+    # Call the orchestrator
+    result = create_raw_extraction_plan(
+        github_issues=ctx.issues,
+        git=ctx.git,
+        claude_installation=ctx.claude_installation,
+        repo_root=repo_root,
+        cwd=ctx.cwd,
+        current_session_id=session_id,
+        min_size=min_size,
+    )
+
+    # Delete pending extraction marker if successful
+    if result.success:
+        delete_marker(repo_root, PENDING_EXTRACTION_MARKER)
+
+    # Output JSON result
+    click.echo(json.dumps(asdict(result)))
+
+    if not result.success:
+        raise SystemExit(1)

erk/cli/commands/plan/get.py

@@ -0,0 +1,71 @@
+"""Command to fetch and display a single plan."""
+
+import click
+
+from erk.cli.core import discover_repo_context
+from erk.core.context import ErkContext
+from erk.core.repo_discovery import ensure_erk_metadata_dir
+from erk_shared.output.output import user_output
+
+
+@click.command("get")
+@click.argument("identifier", type=str)
+@click.pass_obj
+def get_plan(ctx: ErkContext, identifier: str) -> None:
+    """Fetch and display a plan by identifier.
+
+    Args:
+        identifier: Plan identifier (e.g., "42" for GitHub)
+    """
+    repo = discover_repo_context(ctx, ctx.cwd)
+    ensure_erk_metadata_dir(repo)  # Ensure erk metadata directories exist
+    repo_root = repo.root  # Use git repository root for GitHub operations
+
+    try:
+        plan = ctx.plan_store.get_plan(repo_root, identifier)
+    except RuntimeError as e:
+        user_output(click.style("Error: ", fg="red") + str(e))
+        raise SystemExit(1) from e
+
+    # Display plan details
+    user_output("")
+    user_output(click.style(plan.title, bold=True))
+    user_output("")
+
+    # Display metadata with clickable ID
+    state_color = "green" if plan.state.value == "OPEN" else "red"
+
+    # Make ID clickable using OSC 8 if URL is available
+    id_text = f"#{identifier}"
+    if plan.url:
+        colored_id = click.style(id_text, fg="cyan")
+        clickable_id = f"\033]8;;{plan.url}\033\\{colored_id}\033]8;;\033\\"
+    else:
+        clickable_id = click.style(id_text, fg="cyan")
+
+    user_output(f"State: {click.style(plan.state.value, fg=state_color)} | ID: {clickable_id}")
+    user_output(f"URL: {plan.url}")
+
+    # Display labels
+    if plan.labels:
+        labels_str = ", ".join(
+            click.style(f"[{label}]", fg="bright_magenta") for label in plan.labels
+        )
+        user_output(f"Labels: {labels_str}")
+
+    # Display assignees
+    if plan.assignees:
+        assignees_str = ", ".join(plan.assignees)
+        user_output(f"Assignees: {assignees_str}")
+
+    # Display timestamps
+    created = plan.created_at.strftime("%Y-%m-%d %H:%M:%S UTC")
+    updated = plan.updated_at.strftime("%Y-%m-%d %H:%M:%S UTC")
+    user_output(f"Created: {created}")
+    user_output(f"Updated: {updated}")
+
+    # Display body if present
+    if plan.body:
+        user_output("")
+        user_output(click.style("Description:", bold=True))
+        user_output(plan.body)