repr-cli 0.2.8__py3-none-any.whl → 0.2.11__py3-none-any.whl

repr/cli.py

@@ -21,7 +21,8 @@ import os
 import sys
 from datetime import datetime, timedelta
 from pathlib import Path
-from typing import Optional, List
+from typing import Optional, List, Dict
+from collections import defaultdict
 
 import typer
 from rich.prompt import Confirm, Prompt
@@ -247,6 +248,81 @@ def init(
 # GENERATE
 # =============================================================================
 
+def _parse_date_reference(date_str: str) -> str | None:
+    """
+    Parse a date reference string into an ISO date string.
+    
+    Supports:
+    - ISO dates: "2024-01-01"
+    - Day names: "monday", "tuesday", etc.
+    - Relative: "3 days ago", "2 weeks ago", "1 month ago"
+    
+    Returns ISO date string or None if parsing fails.
+    """
+    import re
+    from datetime import datetime, timedelta
+    
+    date_str = date_str.lower().strip()
+    
+    # Try ISO format first
+    try:
+        parsed = datetime.fromisoformat(date_str)
+        return parsed.isoformat()
+    except ValueError:
+        pass
+    
+    # Day names (find previous occurrence)
+    day_names = ["monday", "tuesday", "wednesday", "thursday", "friday", "saturday", "sunday"]
+    if date_str in day_names:
+        today = datetime.now()
+        target_day = day_names.index(date_str)
+        current_day = today.weekday()
+        
+        # Calculate days back to that day
+        days_back = (current_day - target_day) % 7
+        if days_back == 0:
+            days_back = 7  # Go to last week's occurrence
+        
+        target_date = today - timedelta(days=days_back)
+        return target_date.replace(hour=0, minute=0, second=0, microsecond=0).isoformat()
+    
+    # Relative time: "N days/weeks/months ago"
+    match = re.match(r"(\d+)\s+(day|days|week|weeks|month|months)\s+ago", date_str)
+    if match:
+        amount = int(match.group(1))
+        unit = match.group(2).rstrip("s")  # Normalize to singular
+        
+        if unit == "day":
+            delta = timedelta(days=amount)
+        elif unit == "week":
+            delta = timedelta(weeks=amount)
+        elif unit == "month":
+            delta = timedelta(days=amount * 30)  # Approximate
+        else:
+            return None
+        
+        target_date = datetime.now() - delta
+        return target_date.replace(hour=0, minute=0, second=0, microsecond=0).isoformat()
+    
+    # "yesterday" / "today"
+    if date_str == "yesterday":
+        target_date = datetime.now() - timedelta(days=1)
+        return target_date.replace(hour=0, minute=0, second=0, microsecond=0).isoformat()
+    if date_str == "today":
+        target_date = datetime.now()
+        return target_date.replace(hour=0, minute=0, second=0, microsecond=0).isoformat()
+    
+    # "last week" / "last month"
+    if date_str == "last week":
+        target_date = datetime.now() - timedelta(weeks=1)
+        return target_date.replace(hour=0, minute=0, second=0, microsecond=0).isoformat()
+    if date_str == "last month":
+        target_date = datetime.now() - timedelta(days=30)
+        return target_date.replace(hour=0, minute=0, second=0, microsecond=0).isoformat()
+    
+    return None
+
+
 @app.command()
 def generate(
     local: bool = typer.Option(
@@ -267,6 +343,14 @@ def generate(
         None, "--commits",
         help="Generate from specific commits (comma-separated)",
     ),
+    since_date: Optional[str] = typer.Option(
+        None, "--since",
+        help="Generate from commits since date (e.g., '2024-01-01', 'monday', '2 weeks ago')",
+    ),
+    days: Optional[int] = typer.Option(
+        None, "--days",
+        help="Generate from commits in the last N days",
+    ),
     batch_size: int = typer.Option(
         5, "--batch-size",
         help="Commits per story",
@@ -283,6 +367,10 @@ def generate(
         None, "--prompt", "-p",
         help="Custom prompt to append",
     ),
+    json_output: bool = typer.Option(
+        False, "--json",
+        help="Output as JSON",
+    ),
 ):
     """
     Generate stories from commits.
@@ -290,6 +378,8 @@ def generate(
     Examples:
         repr generate --local
         repr generate --cloud
+        repr generate --since "2 weeks ago"
+        repr generate --days 30
         repr generate --template changelog
         repr generate --commits abc123,def456
     """
@@ -299,6 +389,9 @@ def generate(
     if cloud:
         allowed, reason = check_cloud_permission("cloud_generation")
         if not allowed:
+            if json_output:
+                print(json.dumps({"generated": 0, "stories": [], "error": f"Cloud generation blocked: {reason}"}, indent=2))
+                raise typer.Exit(1)
             print_error("Cloud generation blocked")
             print_info(reason)
             console.print()
@@ -313,11 +406,24 @@ def generate(
         else:
             local = True
     
-    print_header()
+    if not json_output:
+        print_header()
     
     mode_str = "local LLM" if local else "cloud LLM"
-    console.print(f"Generating stories ({mode_str})...")
-    console.print()
+    
+    # Build timeframe string for display
+    if commits:
+        timeframe_str = f"specific commits"
+    elif since_date:
+        timeframe_str = f"since {since_date}"
+    elif days:
+        timeframe_str = f"last {days} days"
+    else:
+        timeframe_str = "last 90 days"
+    
+    if not json_output:
+        console.print(f"Generating stories ({mode_str}, {timeframe_str})...")
+        console.print()
     
     # Get repos to analyze
     if repo:
@@ -325,12 +431,18 @@ def generate(
     else:
         tracked = get_tracked_repos()
         if not tracked:
+            if json_output:
+                print(json.dumps({"generated": 0, "stories": [], "error": "No repositories tracked"}, indent=2))
+                raise typer.Exit(1)
             print_warning("No repositories tracked.")
             print_info("Run `repr init` or `repr repos add <path>` first.")
             raise typer.Exit(1)
         repo_paths = [Path(r["path"]) for r in tracked if Path(r["path"]).exists()]
     
     if not repo_paths:
+        if json_output:
+            print(json.dumps({"generated": 0, "stories": [], "error": "No valid repositories found"}, indent=2))
+            raise typer.Exit(1)
         print_error("No valid repositories found")
         raise typer.Exit(1)
     
@@ -339,21 +451,37 @@ def generate(
     from .discovery import analyze_repo
     
     total_stories = 0
+    all_stories = []  # Collect all generated stories for JSON output
     
     for repo_path in repo_paths:
         repo_info = analyze_repo(repo_path)
-        console.print(f"[bold]{repo_info.name}[/]")
+        if not json_output:
+            console.print(f"[bold]{repo_info.name}[/]")
         
         if commits:
             # Specific commits
             commit_shas = [s.strip() for s in commits.split(",")]
             commit_list = get_commits_by_shas(repo_path, commit_shas)
         else:
-            # Recent commits
-            commit_list = get_commits_with_diffs(repo_path, count=50, days=90)
+            # Determine timeframe
+            timeframe_days = days if days is not None else 90  # Default 90 days
+            since_str = None
+            
+            # Parse natural language date if provided
+            if since_date:
+                since_str = _parse_date_reference(since_date)
+            
+            # Recent commits within timeframe
+            commit_list = get_commits_with_diffs(
+                repo_path, 
+                count=500,  # Higher limit when filtering by time
+                days=timeframe_days,
+                since=since_str,
+            )
         
         if not commit_list:
-            console.print(f"  [{BRAND_MUTED}]No commits found[/]")
+            if not json_output:
+                console.print(f"  [{BRAND_MUTED}]No commits found[/]")
             continue
         
         # Dry run: show what would be sent
@@ -434,8 +562,10 @@ def generate(
         )
         
         for story in stories:
-            console.print(f"  • {story['summary']}")
+            if not json_output:
+                console.print(f"  • {story['summary']}")
             total_stories += 1
+            all_stories.append(story)
         
         # Log cloud operation if using cloud
         if cloud and stories:
@@ -450,7 +580,12 @@ def generate(
                 bytes_sent=len(str(commit_list)) // 2,  # Rough estimate
             )
         
-        console.print()
+        if not json_output:
+            console.print()
+    
+    if json_output:
+        print(json.dumps({"generated": total_stories, "stories": all_stories}, indent=2, default=str))
+        return
     
     if dry_run:
         console.print()
@@ -465,7 +600,7 @@ def generate(
         ])
 
 
-def _generate_stories(
+async def _generate_stories_async(
     commits: list[dict],
     repo_info,
     batch_size: int,
@@ -473,9 +608,9 @@ def _generate_stories(
     template: str = "resume",
     custom_prompt: str | None = None,
 ) -> list[dict]:
-    """Generate stories from commits using LLM."""
+    """Generate stories from commits using LLM (async implementation)."""
     from .openai_analysis import get_openai_client, extract_commit_batch
-    from .templates import build_generation_prompt
+    from .templates import build_generation_prompt, StoryOutput
     
     stories = []
     
@@ -485,186 +620,119 @@ def _generate_stories(
         for i in range(0, len(commits), batch_size)
     ]
     
-    for i, batch in enumerate(batches):
-        try:
-            # Build prompt with template
-            system_prompt, user_prompt = build_generation_prompt(
-                template_name=template,
-                repo_name=repo_info.name,
-                commits=batch,
-                custom_prompt=custom_prompt,
-            )
-            
-            # Get LLM client
-            if local:
-                llm_config = get_llm_config()
-                client = get_openai_client(
-                    api_key=llm_config.get("local_api_key") or "ollama",
-                    base_url=llm_config.get("local_api_url") or "http://localhost:11434/v1",
+    # Create client once, reuse for all batches
+    if local:
+        llm_config = get_llm_config()
+        client = get_openai_client(
+            api_key=llm_config.get("local_api_key") or "ollama",
+            base_url=llm_config.get("local_api_url") or "http://localhost:11434/v1",
+        )
+        model = llm_config.get("local_model") or llm_config.get("extraction_model") or "llama3.2"
+    else:
+        client = get_openai_client()
+        model = None  # Use default
+    
+    try:
+        for i, batch in enumerate(batches):
+            try:
+                # Build prompt with template
+                system_prompt, user_prompt = build_generation_prompt(
+                    template_name=template,
+                    repo_name=repo_info.name,
+                    commits=batch,
+                    custom_prompt=custom_prompt,
                 )
-                model = llm_config.get("local_model") or llm_config.get("extraction_model") or "llama3.2"
-            else:
-                client = get_openai_client()
-                model = None  # Use default
-            
-            # Extract story from batch
-            content = asyncio.run(extract_commit_batch(
-                client=client,
-                commits=batch,
-                batch_num=i + 1,
-                total_batches=len(batches),
-                model=model,
-                system_prompt=system_prompt,
-                user_prompt=user_prompt,
-            ))
-            
-            if not content or content.startswith("[Batch"):
-                continue
-            
-            # Extract summary (first non-empty line)
-            lines = [l.strip() for l in content.split("\n") if l.strip()]
-            summary = lines[0][:100] if lines else "Story"
-            # Clean up summary
-            summary = summary.lstrip("#-•* ").strip()
-            
-            # Build metadata
-            commit_shas = [c["full_sha"] for c in batch]
-            first_date = min(c["date"] for c in batch)
-            last_date = max(c["date"] for c in batch)
-            total_files = sum(len(c.get("files", [])) for c in batch)
-            total_adds = sum(c.get("insertions", 0) for c in batch)
-            total_dels = sum(c.get("deletions", 0) for c in batch)
-            
-            metadata = {
-                "summary": summary,
-                "repo_name": repo_info.name,
-                "repo_path": str(repo_info.path),
-                "commit_shas": commit_shas,
-                "first_commit_at": first_date,
-                "last_commit_at": last_date,
-                "files_changed": total_files,
-                "lines_added": total_adds,
-                "lines_removed": total_dels,
-                "generated_locally": local,
-                "template": template,
-                "needs_review": False,
-            }
-            
-            # Save story
-            story_id = save_story(content, metadata)
-            metadata["id"] = story_id
-            stories.append(metadata)
-            
-        except Exception as e:
-            console.print(f"  [{BRAND_MUTED}]Batch {i+1} failed: {e}[/]")
+                
+                # Extract story from batch using structured output
+                result = await extract_commit_batch(
+                    client=client,
+                    commits=batch,
+                    batch_num=i + 1,
+                    total_batches=len(batches),
+                    model=model,
+                    system_prompt=system_prompt,
+                    user_prompt=user_prompt,
+                    structured=True,
+                )
+                
+                # Handle structured output - now returns list[StoryOutput]
+                story_outputs: list[StoryOutput] = []
+                if isinstance(result, list):
+                    story_outputs = result
+                elif isinstance(result, StoryOutput):
+                    story_outputs = [result]
+                else:
+                    # Fallback for string response
+                    content = result
+                    if not content or content.startswith("[Batch"):
+                        continue
+                    lines = [l.strip() for l in content.split("\n") if l.strip()]
+                    summary = lines[0] if lines else "Story"
+                    summary = summary.lstrip("#-•* ").strip()
+                    story_outputs = [StoryOutput(summary=summary, content=content)]
+
+                # Build shared metadata for all stories from this batch
+                commit_shas = [c["full_sha"] for c in batch]
+                first_date = min(c["date"] for c in batch)
+                last_date = max(c["date"] for c in batch)
+                total_files = sum(len(c.get("files", [])) for c in batch)
+                total_adds = sum(c.get("insertions", 0) for c in batch)
+                total_dels = sum(c.get("deletions", 0) for c in batch)
+
+                # Save each story from this batch
+                for story_output in story_outputs:
+                    content = story_output.content
+                    summary = story_output.summary
+
+                    if not content or content.startswith("[Batch"):
+                        continue
+
+                    metadata = {
+                        "summary": summary,
+                        "repo_name": repo_info.name,
+                        "repo_path": str(repo_info.path),
+                        "commit_shas": commit_shas,
+                        "first_commit_at": first_date,
+                        "last_commit_at": last_date,
+                        "files_changed": total_files,
+                        "lines_added": total_adds,
+                        "lines_removed": total_dels,
+                        "generated_locally": local,
+                        "template": template,
+                        "needs_review": False,
+                    }
+
+                    # Save story
+                    story_id = save_story(content, metadata)
+                    metadata["id"] = story_id
+                    stories.append(metadata)
+                
+            except Exception as e:
+                console.print(f"  [{BRAND_MUTED}]Batch {i+1} failed: {e}[/]")
+    finally:
+        # Properly close the async client
+        await client.close()
     
     return stories
 
 
-# =============================================================================
-# QUICK REFLECTION COMMANDS
-# =============================================================================
-
-@app.command()
-def week(
-    save: bool = typer.Option(
-        False, "--save",
-        help="Save as a permanent story",
-    ),
-):
-    """
-    Show what you worked on this week.
-    
-    Example:
-        repr week
-        repr week --save
-    """
-    _show_summary_since("this week", days=7, save=save)
-
-
-@app.command()
-def since(
-    date: str = typer.Argument(..., help="Date or time reference (e.g., 'monday', '2026-01-01', '3 days ago')"),
-    save: bool = typer.Option(
-        False, "--save",
-        help="Save as a permanent story",
-    ),
-):
-    """
-    Show work since a specific date or time.
-    
-    Examples:
-        repr since monday
-        repr since "3 days ago"
-        repr since 2026-01-01
-    """
-    _show_summary_since(date, save=save)
-
-
-@app.command()
-def standup(
-    days: int = typer.Option(
-        3, "--days",
-        help="Number of days to look back",
-    ),
-):
-    """
-    Quick summary for daily standup (last 3 days).
-    
-    Example:
-        repr standup
-        repr standup --days 5
-    """
-    _show_summary_since(f"last {days} days", days=days, save=False)
-
-
-def _show_summary_since(label: str, days: int = None, save: bool = False):
-    """Show summary of work since a date."""
-    from .tools import get_commits_with_diffs
-    from .discovery import analyze_repo
-    
-    print_header()
-    
-    tracked = get_tracked_repos()
-    if not tracked:
-        print_warning("No repositories tracked.")
-        print_info("Run `repr init` first.")
-        raise typer.Exit(1)
-    
-    console.print(f"📊 Work since {label}")
-    console.print()
-    
-    total_commits = 0
-    all_summaries = []
-    
-    for repo_info in tracked:
-        repo_path = Path(repo_info["path"])
-        if not repo_path.exists():
-            continue
-        
-        commits = get_commits_with_diffs(repo_path, count=100, days=days or 30)
-        if not commits:
-            continue
-        
-        repo_name = repo_path.name
-        console.print(f"[bold]{repo_name}[/] ({len(commits)} commits):")
-        
-        # Group commits by rough topic (simple heuristic)
-        for c in commits[:5]:
-            msg = c["message"].split("\n")[0][:60]
-            console.print(f"  • {msg}")
-        
-        if len(commits) > 5:
-            console.print(f"  [{BRAND_MUTED}]... and {len(commits) - 5} more[/]")
-        
-        total_commits += len(commits)
-        console.print()
-    
-    console.print(f"Total: {total_commits} commits")
-    
-    if not save:
-        console.print()
-        console.print(f"[{BRAND_MUTED}]This summary wasn't saved. Run with --save to create a story.[/]")
+def _generate_stories(
+    commits: list[dict],
+    repo_info,
+    batch_size: int,
+    local: bool,
+    template: str = "resume",
+    custom_prompt: str | None = None,
+) -> list[dict]:
+    """Generate stories from commits using LLM."""
+    return asyncio.run(_generate_stories_async(
+        commits=commits,
+        repo_info=repo_info,
+        batch_size=batch_size,
+        local=local,
+        template=template,
+        custom_prompt=custom_prompt,
+    ))
 
 
 # =============================================================================
@@ -698,24 +766,37 @@ def stories(
     
     console.print(f"[bold]Stories[/] ({len(story_list)} total)")
     console.print()
-    
+
+    # Group stories by repository
+    by_repo = defaultdict(list)
     for story in story_list[:20]:
-        # Status indicator
-        if story.get("needs_review"):
-            status = f"[{BRAND_WARNING}]⚠[/]"
-        elif story.get("pushed_at"):
-            status = f"[{BRAND_SUCCESS}]✓[/]"
-        else:
-            status = f"[{BRAND_MUTED}]○[/]"
-        
-        summary = story.get("summary", "Untitled")[:60]
-        repo_name = story.get("repo_name", "unknown")
-        created = format_relative_time(story.get("created_at", ""))
-        
-        console.print(f"{status} {summary}")
-        console.print(f"  [{BRAND_MUTED}]{repo_name} • {created}[/]")
+        r_name = story.get("repo_name", "unknown")
+        by_repo[r_name].append(story)
+    
+    # Sort repositories by their most recent story
+    sorted_repos = sorted(
+        by_repo.keys(),
+        key=lambda r: max(s.get("created_at", "") for s in by_repo[r]),
+        reverse=True
+    )
+
+    for repo_name in sorted_repos:
+        console.print(f"[bold]{repo_name}[/]")
+        for story in by_repo[repo_name]:
+            # Status indicator
+            if story.get("needs_review"):
+                status = f"[{BRAND_WARNING}]⚠[/]"
+            elif story.get("pushed_at"):
+                status = f"[{BRAND_SUCCESS}]✓[/]"
+            else:
+                status = f"[{BRAND_MUTED}]○[/]"
+            
+            summary = story.get("summary", "Untitled")
+            created = format_relative_time(story.get("created_at", ""))
+            
+            console.print(f"  {status} {summary} [{BRAND_MUTED}]• {created}[/]")
         console.print()
-    
+
     if len(story_list) > 20:
         console.print(f"[{BRAND_MUTED}]... and {len(story_list) - 20} more[/]")
 
@@ -723,7 +804,8 @@ def stories(
 @app.command()
 def story(
     action: str = typer.Argument(..., help="Action: view, edit, delete, hide, feature, regenerate"),
-    story_id: str = typer.Argument(..., help="Story ID (ULID)"),
+    story_id: Optional[str] = typer.Argument(None, help="Story ID (ULID)"),
+    all_stories: bool = typer.Option(False, "--all", help="Apply to all stories (for delete)"),
 ):
     """
     Manage a single story.
@@ -731,7 +813,38 @@ def story(
     Examples:
         repr story view 01ARYZ6S41TSV4RRFFQ69G5FAV
         repr story delete 01ARYZ6S41TSV4RRFFQ69G5FAV
+        repr story delete --all
     """
+    # Handle --all flag for delete
+    if all_stories:
+        if action != "delete":
+            print_error("--all flag only works with 'delete' action")
+            raise typer.Exit(1)
+        
+        story_list = list_stories()
+        if not story_list:
+            print_info("No stories to delete")
+            raise typer.Exit()
+        
+        console.print(f"This will delete [bold]{len(story_list)}[/] stories.")
+        if confirm("Delete all stories?"):
+            deleted = 0
+            for s in story_list:
+                try:
+                    delete_story(s["id"])
+                    deleted += 1
+                except Exception:
+                    pass
+            print_success(f"Deleted {deleted} stories")
+        else:
+            print_info("Cancelled")
+        raise typer.Exit()
+    
+    # Require story_id for single-story operations
+    if not story_id:
+        print_error("Story ID required (or use --all for delete)")
+        raise typer.Exit(1)
+    
     result = load_story(story_id)
     
     if not result:
@@ -935,7 +1048,9 @@ def push(
     for s in to_push:
         try:
             content, meta = load_story(s["id"])
-            asyncio.run(api_push_story({**meta, "content": content}))
+            # Use local story ID as client_id for sync
+            payload = {**meta, "content": content, "client_id": s["id"]}
+            asyncio.run(api_push_story(payload))
             mark_story_pushed(s["id"])
             console.print(f"  [{BRAND_SUCCESS}]✓[/] {s.get('summary', s.get('id'))[:50]}")
             pushed += 1
@@ -984,7 +1099,8 @@ def sync():
             for s in unpushed:
                 try:
                     content, meta = load_story(s["id"])
-                    asyncio.run(api_push_story({**meta, "content": content}))
+                    payload = {**meta, "content": content, "client_id": s["id"]}
+                    asyncio.run(api_push_story(payload))
                     mark_story_pushed(s["id"])
                 except Exception:
                     pass
@@ -1025,15 +1141,20 @@ def pull():
 def list_commits(
     repo: Optional[str] = typer.Option(None, "--repo", help="Filter by repo name"),
     limit: int = typer.Option(50, "--limit", "-n", help="Number of commits"),
-    since: Optional[str] = typer.Option(None, "--since", help="Since date"),
+    days: Optional[int] = typer.Option(None, "--days", "-d", help="Show commits from last N days (e.g., 7 for week, 3 for standup)"),
+    since: Optional[str] = typer.Option(None, "--since", help="Since date (e.g., '2024-01-01', 'monday', '2 weeks ago')"),
     json_output: bool = typer.Option(False, "--json", help="Output as JSON"),
 ):
     """
     List recent commits across all tracked repos.
     
-    Example:
+    Examples:
         repr commits --limit 10
         repr commits --repo myproject
+        repr commits --days 7           # Last week
+        repr commits --days 3           # For standup
+        repr commits --since monday
+        repr commits --since "2 weeks ago"
     """
     from .tools import get_commits_with_diffs
     
@@ -1042,6 +1163,18 @@ def list_commits(
         print_warning("No repositories tracked.")
         raise typer.Exit(1)
     
+    # Determine days filter
+    filter_days = days if days is not None else 90
+    
+    # Parse natural language date if --since provided
+    since_str = None
+    if since:
+        since_str = _parse_date_reference(since)
+        if not since_str:
+            print_error(f"Could not parse date: {since}")
+            print_info("Try: '2024-01-01', 'monday', '3 days ago', 'last week'")
+            raise typer.Exit(1)
+    
     all_commits = []
     
     for repo_info in tracked:
@@ -1051,7 +1184,7 @@ def list_commits(
         if not repo_path.exists():
             continue
         
-        commits = get_commits_with_diffs(repo_path, count=limit, days=90)
+        commits = get_commits_with_diffs(repo_path, count=limit, days=filter_days, since=since_str)
         for c in commits:
             c["repo_name"] = repo_path.name
         all_commits.extend(commits)
@@ -1068,7 +1201,15 @@ def list_commits(
         print_info("No commits found")
         raise typer.Exit()
     
-    console.print(f"[bold]Recent Commits[/] ({len(all_commits)})")
+    # Build label for display
+    if days:
+        label = f"last {days} days"
+    elif since:
+        label = f"since {since}"
+    else:
+        label = "recent"
+    
+    console.print(f"[bold]Commits ({label})[/] — {len(all_commits)} total")
     console.print()
     
     current_repo = None
@@ -2331,22 +2472,64 @@ def status(
 
 
 @app.command()
-def mode():
+def mode(
+    json_output: bool = typer.Option(False, "--json", help="Output as JSON"),
+):
     """
     Show current execution mode and settings.
-    
+
     Example:
         repr mode
+        repr mode --json
     """
     from .llm import get_llm_status
-    
+
     authenticated = is_authenticated()
     privacy = get_privacy_settings()
     llm_status = get_llm_status()
-    
+
+    # Determine mode
+    if privacy.get("lock_local_only"):
+        if privacy.get("lock_permanent"):
+            mode = "LOCAL-ONLY"
+            locked = True
+        else:
+            mode = "LOCAL-ONLY"
+            locked = True
+    elif authenticated:
+        mode = "CLOUD"
+        locked = False
+    else:
+        mode = "LOCAL-ONLY"
+        locked = False
+
+    # Build LLM provider string
+    if llm_status["local"]["available"]:
+        llm_provider = f"{llm_status['local']['name']} ({llm_status['local']['model']})"
+    else:
+        llm_provider = "none"
+
+    # Check cloud allowed
+    cloud_allowed = authenticated and is_cloud_allowed()
+
+    if json_output:
+        import json
+        output = {
+            "mode": mode,
+            "locked": locked,
+            "llm_provider": llm_provider,
+            "network_policy": "cloud-enabled" if authenticated else "local-only",
+            "authenticated": authenticated,
+            "cloud_generation_available": cloud_allowed,
+            "sync_available": cloud_allowed,
+            "publishing_available": cloud_allowed,
+        }
+        print(json.dumps(output, indent=2))
+        return
+
     console.print("[bold]Mode[/]")
     console.print()
-    
+
     if privacy.get("lock_local_only"):
         if privacy.get("lock_permanent"):
             console.print("Data boundary: local only (permanently locked)")
@@ -2356,22 +2539,22 @@ def mode():
         console.print("Data boundary: cloud-enabled")
     else:
         console.print("Data boundary: local only")
-    
+
     console.print(f"Default inference: {llm_status['default_mode']}")
-    
+
     console.print()
-    
+
     # LLM info
     if llm_status["local"]["available"]:
         console.print(f"Local LLM: {llm_status['local']['name']} ({llm_status['local']['model']})")
     else:
         console.print(f"[{BRAND_MUTED}]Local LLM: not detected[/]")
-    
+
     console.print()
-    
+
     console.print("Available:")
     console.print(f"  [{BRAND_SUCCESS}]✓[/] Local generation")
-    if authenticated and is_cloud_allowed():
+    if cloud_allowed:
         console.print(f"  [{BRAND_SUCCESS}]✓[/] Cloud generation")
         console.print(f"  [{BRAND_SUCCESS}]✓[/] Sync")
         console.print(f"  [{BRAND_SUCCESS}]✓[/] Publishing")