htmlgraph 0.26.5__py3-none-any.whl → 0.26.6__py3-none-any.whl

htmlgraph/cli/main.py

@@ -0,0 +1,143 @@
+"""HtmlGraph CLI - Main entry point.
+
+Entry point with argument parsing and command routing.
+Keeps main() thin by delegating to command modules.
+"""
+
+from __future__ import annotations
+
+import argparse
+import sys
+
+from rich.console import Console
+
+from htmlgraph.cli.constants import (
+    DEFAULT_GRAPH_DIR,
+    DEFAULT_OUTPUT_FORMAT,
+    OUTPUT_FORMATS,
+)
+
+
+def create_parser() -> argparse.ArgumentParser:
+    """Create and configure the argument parser.
+
+    Returns:
+        Configured ArgumentParser with all subcommands
+    """
+    parser = argparse.ArgumentParser(
+        description="HtmlGraph - HTML is All You Need",
+        formatter_class=argparse.RawDescriptionHelpFormatter,
+        epilog="""
+Examples:
+  htmlgraph init                    # Initialize .htmlgraph in current dir
+  htmlgraph serve                   # Start server on port 8080
+  htmlgraph status                  # Show graph status
+  htmlgraph query "[data-status='todo']"  # Query nodes
+
+Session Management:
+  htmlgraph session start           # Start a new session (auto-ID)
+  htmlgraph session end my-session  # End a session
+  htmlgraph session list            # List all sessions
+
+Feature Management:
+  htmlgraph feature list            # List all features
+  htmlgraph feature start feat-001  # Start working on a feature
+  htmlgraph feature complete feat-001  # Mark feature as done
+
+Track Management:
+  htmlgraph track new "User Auth"  # Create a new track
+  htmlgraph track list              # List all tracks
+
+Analytics:
+  htmlgraph analytics               # Project-wide analytics
+  htmlgraph analytics --recent 10   # Analyze last 10 sessions
+
+For more help: https://github.com/Shakes-tzd/htmlgraph
+""",
+    )
+
+    # Global output control flags
+    parser.add_argument(
+        "--format",
+        choices=OUTPUT_FORMATS,
+        default=DEFAULT_OUTPUT_FORMAT,
+        help="Output format: text (default), json, or plain",
+    )
+    parser.add_argument(
+        "--quiet",
+        "-q",
+        action="store_true",
+        help="Suppress progress messages and non-essential output",
+    )
+    parser.add_argument(
+        "--verbose",
+        "-v",
+        action="count",
+        default=0,
+        help="Increase verbosity (can be used multiple times: -v, -vv, -vvv)",
+    )
+
+    subparsers = parser.add_subparsers(dest="command", help="Command to run")
+
+    # Import command registration functions
+    from htmlgraph.cli import analytics, core, work
+
+    # Register commands from each module
+    core.register_commands(subparsers)
+    work.register_commands(subparsers)
+    analytics.register_commands(subparsers)
+
+    return parser
+
+
+def main() -> None:
+    """Main entry point for the CLI."""
+    parser = create_parser()
+    args = parser.parse_args()
+
+    # If no command specified, show help
+    if not args.command:
+        parser.print_help()
+        sys.exit(0)
+
+    # Get the command handler (set by register_commands via set_defaults)
+    if not hasattr(args, "func"):
+        parser.print_help()
+        sys.exit(1)
+
+    # Determine graph directory and agent
+    graph_dir = getattr(args, "graph_dir", DEFAULT_GRAPH_DIR)
+    agent = getattr(args, "agent", None)
+
+    # Get output format from args
+    output_format = args.format
+
+    # Execute the command
+    try:
+        # Create command instance from args
+        command = args.func(args)
+
+        # Run command with context
+        command.run(
+            graph_dir=graph_dir,
+            agent=agent,
+            output_format=output_format,
+        )
+    except KeyboardInterrupt:
+        err_console = Console(stderr=True)
+        err_console.print("\n\n[yellow]Interrupted by user[/yellow]")
+        sys.exit(130)
+    except Exception as e:
+        from htmlgraph.cli.base import save_traceback
+
+        err_console = Console(stderr=True)
+        log_file = save_traceback(
+            e, context={"command": args.command, "cwd": graph_dir}
+        )
+        err_console.print(f"[red]Error:[/red] {e}")
+        err_console.print(f"[dim]Full traceback saved to:[/dim] {log_file}")
+        sys.exit(1)
+
+
+if __name__ == "__main__":
+    main()

htmlgraph/cli/models.py

@@ -0,0 +1,462 @@
+"""HtmlGraph CLI - Pydantic models for command filters and configuration.
+
+This module provides type-safe models for validating command inputs:
+- Filter models for list commands (features, sessions, tracks)
+- Configuration models for infrastructure commands (init, serve)
+"""
+
+from __future__ import annotations
+
+from datetime import datetime
+from typing import Literal
+
+from pydantic import BaseModel, Field, field_validator
+
+# ============================================================================
+# Filter Models
+# ============================================================================
+
+
+class FeatureFilter(BaseModel):
+    """Filter options for feature listing.
+
+    Attributes:
+        status: Filter by status (todo, in_progress, completed, blocked, all)
+        priority: Filter by priority (high, medium, low, critical, all)
+        agent: Filter by agent name
+        collection: Collection name to query (default: features)
+    """
+
+    status: Literal["todo", "in_progress", "completed", "blocked", "all"] | None = None
+    priority: Literal["high", "medium", "low", "critical", "all"] | None = None
+    agent: str | None = None
+    collection: str = Field(default="features")
+
+    @field_validator("status")
+    @classmethod
+    def validate_status(cls, v: str | None) -> str | None:
+        """Validate status value."""
+        if v and v not in ["todo", "in_progress", "completed", "blocked", "all"]:
+            raise ValueError(
+                f"Invalid status: {v}. "
+                f"Valid values: todo, in_progress, completed, blocked, all"
+            )
+        return v
+
+    @field_validator("priority")
+    @classmethod
+    def validate_priority(cls, v: str | None) -> str | None:
+        """Validate priority value."""
+        if v and v not in ["high", "medium", "low", "critical", "all"]:
+            raise ValueError(
+                f"Invalid priority: {v}. Valid values: high, medium, low, critical, all"
+            )
+        return v
+
+
+class SessionFilter(BaseModel):
+    """Filter options for session listing.
+
+    Attributes:
+        status: Filter by status (active, ended, all)
+        agent: Filter by agent name
+        since: Only show sessions since this date
+    """
+
+    status: Literal["active", "ended", "all"] | None = None
+    agent: str | None = None
+    since: datetime | None = None
+
+    @field_validator("status")
+    @classmethod
+    def validate_status(cls, v: str | None) -> str | None:
+        """Validate status value."""
+        if v and v not in ["active", "ended", "all"]:
+            raise ValueError(f"Invalid status: {v}. Valid values: active, ended, all")
+        return v
+
+
+class TrackFilter(BaseModel):
+    """Filter options for track listing.
+
+    Attributes:
+        status: Filter by status (todo, in_progress, completed, all)
+        priority: Filter by priority (high, medium, low, all)
+        has_spec: Filter for tracks with specs
+        has_plan: Filter for tracks with plans
+    """
+
+    status: Literal["todo", "in_progress", "completed", "all"] | None = None
+    priority: Literal["high", "medium", "low", "all"] | None = None
+    has_spec: bool | None = None
+    has_plan: bool | None = None
+
+    @field_validator("status")
+    @classmethod
+    def validate_status(cls, v: str | None) -> str | None:
+        """Validate status value."""
+        if v and v not in ["todo", "in_progress", "completed", "all"]:
+            raise ValueError(
+                f"Invalid status: {v}. Valid values: todo, in_progress, completed, all"
+            )
+        return v
+
+    @field_validator("priority")
+    @classmethod
+    def validate_priority(cls, v: str | None) -> str | None:
+        """Validate priority value."""
+        if v and v not in ["high", "medium", "low", "all"]:
+            raise ValueError(
+                f"Invalid priority: {v}. Valid values: high, medium, low, all"
+            )
+        return v
+
+
+# ============================================================================
+# Configuration Models
+# ============================================================================
+
+
+class InitConfig(BaseModel):
+    """Configuration for htmlgraph init command.
+
+    Attributes:
+        dir: Directory to initialize (default: .)
+        install_hooks: Install Git hooks for event logging
+        interactive: Interactive setup wizard
+        no_index: Do not create the analytics cache (index.sqlite)
+        no_update_gitignore: Do not update/create .gitignore for cache files
+        no_events_keep: Do not create .htmlgraph/events/.gitkeep
+    """
+
+    dir: str = Field(default=".")
+    install_hooks: bool = Field(default=False)
+    interactive: bool = Field(default=False)
+    no_index: bool = Field(default=False)
+    no_update_gitignore: bool = Field(default=False)
+    no_events_keep: bool = Field(default=False)
+
+
+class ServeConfig(BaseModel):
+    """Configuration for htmlgraph serve command.
+
+    Attributes:
+        port: Port to bind to (must be between 1024-65535)
+        host: Host to bind to (default: 0.0.0.0)
+        graph_dir: Graph directory path
+        static_dir: Static files directory
+        no_watch: Disable file watching (auto-reload disabled)
+        auto_port: Automatically find available port if default is occupied
+    """
+
+    port: int = Field(default=8080, ge=1024, le=65535)
+    host: str = Field(default="0.0.0.0")
+    graph_dir: str = Field(default=".htmlgraph")
+    static_dir: str = Field(default=".")
+    no_watch: bool = Field(default=False)
+    auto_port: bool = Field(default=False)
+
+    @field_validator("port")
+    @classmethod
+    def validate_port(cls, v: int) -> int:
+        """Validate port is in valid range."""
+        if not 1024 <= v <= 65535:
+            raise ValueError(f"Port must be between 1024 and 65535, got {v}")
+        return v
+
+    @field_validator("host")
+    @classmethod
+    def validate_host(cls, v: str) -> str:
+        """Validate host is not empty."""
+        if not v or not v.strip():
+            raise ValueError("Host cannot be empty")
+        return v.strip()
+
+
+class ServeApiConfig(BaseModel):
+    """Configuration for htmlgraph serve-api command.
+
+    Attributes:
+        port: Port to bind to (must be between 1024-65535)
+        host: Host to bind to (default: 127.0.0.1)
+        db: Path to SQLite database file
+        auto_port: Automatically find available port if default is occupied
+        reload: Enable auto-reload on file changes (development mode)
+    """
+
+    port: int = Field(default=8000, ge=1024, le=65535)
+    host: str = Field(default="127.0.0.1")
+    db: str | None = None
+    auto_port: bool = Field(default=False)
+    reload: bool = Field(default=False)
+
+    @field_validator("port")
+    @classmethod
+    def validate_port(cls, v: int) -> int:
+        """Validate port is in valid range."""
+        if not 1024 <= v <= 65535:
+            raise ValueError(f"Port must be between 1024 and 65535, got {v}")
+        return v
+
+    @field_validator("host")
+    @classmethod
+    def validate_host(cls, v: str) -> str:
+        """Validate host is not empty."""
+        if not v or not v.strip():
+            raise ValueError("Host cannot be empty")
+        return v.strip()
+
+
+# ============================================================================
+# Result Models
+# ============================================================================
+
+
+class InitResult(BaseModel):
+    """Result from htmlgraph init command.
+
+    Attributes:
+        success: Whether initialization succeeded
+        graph_dir: Path to initialized .htmlgraph directory
+        directories_created: List of directories created
+        files_created: List of files created
+        hooks_installed: Whether Git hooks were installed
+        warnings: List of warning messages
+        errors: List of error messages
+    """
+
+    success: bool = Field(default=True)
+    graph_dir: str = Field(...)
+    directories_created: list[str] = Field(default_factory=list)
+    files_created: list[str] = Field(default_factory=list)
+    hooks_installed: bool = Field(default=False)
+    warnings: list[str] = Field(default_factory=list)
+    errors: list[str] = Field(default_factory=list)
+
+    @property
+    def summary(self) -> str:
+        """Human-readable summary of initialization."""
+        lines = [f"Initialized {self.graph_dir}"]
+        if self.directories_created:
+            lines.append(f"  • Created {len(self.directories_created)} directories")
+        if self.files_created:
+            lines.append(f"  • Created {len(self.files_created)} files")
+        if self.hooks_installed:
+            lines.append("  • Installed Git hooks")
+        if self.warnings:
+            lines.append(f"  • {len(self.warnings)} warnings")
+        if self.errors:
+            lines.append(f"  • {len(self.errors)} errors")
+        return "\n".join(lines)
+
+
+class ValidationResult(BaseModel):
+    """Result from directory validation.
+
+    Attributes:
+        valid: Whether directory is valid for initialization
+        exists: Whether directory already exists
+        is_initialized: Whether directory is already initialized
+        has_git: Whether directory is in a Git repository
+        errors: List of validation errors
+    """
+
+    valid: bool = Field(default=True)
+    exists: bool = Field(default=False)
+    is_initialized: bool = Field(default=False)
+    has_git: bool = Field(default=False)
+    errors: list[str] = Field(default_factory=list)
+
+
+# ============================================================================
+# Display Models
+# ============================================================================
+
+
+class SessionDisplay(BaseModel):
+    """Validated session data for CLI display."""
+
+    id: str = Field(..., description="Session identifier")
+    status: str = Field(default="active", description="Session status")
+    agent: str = Field(default="unknown", description="Agent name")
+    event_count: int = Field(default=0, ge=0, description="Number of events")
+    started_at: datetime = Field(..., description="Session start time")
+    ended_at: datetime | None = Field(None, description="Session end time")
+    title: str | None = Field(None, description="Session title")
+
+    @classmethod
+    def from_node(cls, node: object) -> SessionDisplay:
+        """
+        Create SessionDisplay from graph node.
+
+        Args:
+            node: Session node from graph
+
+        Returns:
+            SessionDisplay instance
+        """
+        return cls(
+            id=getattr(node, "id"),
+            status=getattr(node, "status", "active"),
+            agent=getattr(node, "agent", "unknown"),
+            event_count=getattr(node, "event_count", 0),
+            started_at=getattr(node, "started_at"),
+            ended_at=getattr(node, "ended_at", None),
+            title=getattr(node, "title", None),
+        )
+
+    @property
+    def started_str(self) -> str:
+        """Formatted start time for display."""
+        return self.started_at.strftime("%Y-%m-%d %H:%M")
+
+    @property
+    def ended_str(self) -> str | None:
+        """Formatted end time for display."""
+        if self.ended_at:
+            return self.ended_at.strftime("%Y-%m-%d %H:%M")
+        return None
+
+    def sort_key(self) -> datetime:
+        """
+        Return sort key for session ordering.
+
+        Returns timezone-naive datetime for consistent sorting.
+        """
+        ts = self.started_at
+        if ts.tzinfo is None:
+            return ts
+        return ts.replace(tzinfo=None)
+
+
+class FeatureDisplay(BaseModel):
+    """Validated feature data for CLI display."""
+
+    id: str = Field(..., description="Feature identifier")
+    title: str = Field(default="Untitled", description="Feature title")
+    status: str = Field(default="unknown", description="Feature status")
+    priority: Literal["low", "medium", "high", "critical"] = Field(
+        default="medium", description="Feature priority"
+    )
+    updated: datetime = Field(..., description="Last update time")
+    created: datetime | None = Field(None, description="Creation time")
+    track_id: str | None = Field(None, description="Associated track ID")
+    agent_assigned: str | None = Field(None, description="Assigned agent")
+
+    @classmethod
+    def from_node(cls, node: object) -> FeatureDisplay:
+        """
+        Create FeatureDisplay from graph node.
+
+        Args:
+            node: Feature node from graph
+
+        Returns:
+            FeatureDisplay instance
+        """
+        return cls(
+            id=getattr(node, "id"),
+            title=getattr(node, "title", "Untitled"),
+            status=getattr(node, "status", "unknown"),
+            priority=getattr(node, "priority", "medium"),
+            updated=getattr(node, "updated"),
+            created=getattr(node, "created", None),
+            track_id=getattr(node, "track_id", None),
+            agent_assigned=getattr(node, "agent_assigned", None),
+        )
+
+    @property
+    def updated_str(self) -> str:
+        """Formatted update time for display."""
+        return self.updated.strftime("%Y-%m-%d %H:%M")
+
+    @property
+    def created_str(self) -> str | None:
+        """Formatted creation time for display."""
+        if self.created:
+            return self.created.strftime("%Y-%m-%d %H:%M")
+        return None
+
+    def sort_key(self) -> tuple[int, datetime]:
+        """
+        Return sort key for feature ordering (priority, then updated time).
+
+        Returns:
+            Tuple of (priority_rank, timezone_naive_updated_time)
+        """
+        priority_order = {"critical": 0, "high": 1, "medium": 2, "low": 3}
+        priority_rank = priority_order.get(self.priority, 99)
+
+        updated_ts = self.updated
+        if updated_ts.tzinfo is None:
+            updated_naive = updated_ts
+        else:
+            updated_naive = updated_ts.replace(tzinfo=None)
+
+        return (priority_rank, updated_naive)
+
+
+class TrackDisplay(BaseModel):
+    """Validated track data for CLI display."""
+
+    id: str = Field(..., description="Track identifier")
+    title: str = Field(default="Untitled", description="Track title")
+    status: str = Field(default="planning", description="Track status")
+    priority: Literal["low", "medium", "high"] = Field(
+        default="medium", description="Track priority"
+    )
+    has_spec: bool = Field(default=False, description="Has specification")
+    has_plan: bool = Field(default=False, description="Has plan")
+    format_type: str = Field(
+        default="consolidated", description="File format (consolidated or directory)"
+    )
+    feature_count: int = Field(default=0, ge=0, description="Number of features")
+
+    @classmethod
+    def from_track_id(
+        cls,
+        track_id: str,
+        has_spec: bool = False,
+        has_plan: bool = False,
+        format_type: str = "consolidated",
+    ) -> TrackDisplay:
+        """
+        Create TrackDisplay from track ID and metadata.
+
+        Args:
+            track_id: Track identifier
+            has_spec: Whether track has specification
+            has_plan: Whether track has plan
+            format_type: File format type
+
+        Returns:
+            TrackDisplay instance
+        """
+        return cls(
+            id=track_id,
+            title=track_id,  # Default to ID if title not available
+            has_spec=has_spec,
+            has_plan=has_plan,
+            format_type=format_type,
+        )
+
+    @property
+    def components_str(self) -> str:
+        """Formatted components list for display."""
+        components = []
+        if self.has_spec:
+            components.append("spec")
+        if self.has_plan:
+            components.append("plan")
+        return ", ".join(components) if components else "empty"
+
+    def sort_key(self) -> tuple[int, str]:
+        """
+        Return sort key for track ordering (priority, then ID).
+
+        Returns:
+            Tuple of (priority_rank, track_id)
+        """
+        priority_order = {"high": 0, "medium": 1, "low": 2}
+        priority_rank = priority_order.get(self.priority, 99)
+        return (priority_rank, self.id)

htmlgraph/cli/templates/__init__.py

@@ -0,0 +1 @@
+"""HTML templates for CLI output."""