htmlgraph 0.20.1__py3-none-any.whl → 0.27.5__py3-none-any.whl

htmlgraph/operations/analytics.py

@@ -0,0 +1,339 @@
+from __future__ import annotations
+
+"""Analytics operations for HtmlGraph."""
+
+
+from dataclasses import dataclass
+from pathlib import Path
+from typing import Any
+
+from htmlgraph import SDK
+from htmlgraph.converter import html_to_session
+
+
+@dataclass(frozen=True)
+class AnalyticsSessionResult:
+    """Result of analyzing a single session."""
+
+    session_id: str
+    metrics: dict[str, Any]
+    warnings: list[str]
+
+
+@dataclass(frozen=True)
+class AnalyticsProjectResult:
+    """Result of analyzing project-wide analytics."""
+
+    metrics: dict[str, Any]
+    warnings: list[str]
+
+
+@dataclass(frozen=True)
+class RecommendationsResult:
+    """Result of getting work recommendations."""
+
+    recommendations: list[dict[str, Any]]
+    reasoning: dict[str, Any]
+    warnings: list[str]
+
+
+class AnalyticsOperationError(RuntimeError):
+    """Base error for analytics operations."""
+
+
+def analyze_session(*, graph_dir: Path, session_id: str) -> AnalyticsSessionResult:
+    """
+    Compute analytics for a single session.
+
+    Args:
+        graph_dir: Path to .htmlgraph directory
+        session_id: ID of the session to analyze
+
+    Returns:
+        AnalyticsSessionResult with session metrics and warnings
+
+    Raises:
+        AnalyticsOperationError: If session cannot be analyzed
+    """
+    warnings: list[str] = []
+
+    # Validate inputs
+    if not graph_dir.exists():
+        raise AnalyticsOperationError(f"Graph directory does not exist: {graph_dir}")
+
+    session_path = graph_dir / "sessions" / f"{session_id}.html"
+    if not session_path.exists():
+        raise AnalyticsOperationError(f"Session not found: {session_id}")
+
+    try:
+        # Load session
+        session = html_to_session(session_path)
+    except Exception as e:
+        raise AnalyticsOperationError(f"Failed to load session {session_id}: {e}")
+
+    try:
+        # Initialize SDK with minimal agent
+        sdk = SDK(directory=str(graph_dir), agent="analytics-ops")
+
+        # Compute metrics
+        metrics: dict[str, Any] = {}
+
+        # Work distribution
+        try:
+            work_dist = sdk.analytics.work_type_distribution(session_id=session_id)
+            metrics["work_distribution"] = work_dist
+        except Exception as e:
+            warnings.append(f"Failed to compute work distribution: {e}")
+            metrics["work_distribution"] = {}
+
+        # Spike-to-feature ratio
+        try:
+            spike_ratio = sdk.analytics.spike_to_feature_ratio(session_id=session_id)
+            metrics["spike_to_feature_ratio"] = spike_ratio
+        except Exception as e:
+            warnings.append(f"Failed to compute spike ratio: {e}")
+            metrics["spike_to_feature_ratio"] = 0.0
+
+        # Maintenance burden
+        try:
+            maintenance = sdk.analytics.maintenance_burden(session_id=session_id)
+            metrics["maintenance_burden"] = maintenance
+        except Exception as e:
+            warnings.append(f"Failed to compute maintenance burden: {e}")
+            metrics["maintenance_burden"] = 0.0
+
+        # Primary work type
+        try:
+            primary = sdk.analytics.calculate_session_primary_work_type(session_id)
+            metrics["primary_work_type"] = primary
+        except Exception as e:
+            warnings.append(f"Failed to compute primary work type: {e}")
+            metrics["primary_work_type"] = None
+
+        # Work breakdown (event counts)
+        try:
+            breakdown = sdk.analytics.calculate_session_work_breakdown(session_id)
+            metrics["work_breakdown"] = breakdown
+            metrics["total_events"] = sum(breakdown.values()) if breakdown else 0
+        except Exception as e:
+            warnings.append(f"Failed to compute work breakdown: {e}")
+            metrics["work_breakdown"] = {}
+            metrics["total_events"] = session.event_count
+
+        # Transition time metrics
+        try:
+            transition = sdk.analytics.transition_time_metrics(session_id=session_id)
+            metrics["transition_metrics"] = transition
+        except Exception as e:
+            warnings.append(f"Failed to compute transition metrics: {e}")
+            metrics["transition_metrics"] = {}
+
+        # Session metadata
+        metrics["session_id"] = session.id
+        metrics["agent"] = session.agent
+        metrics["status"] = session.status
+        metrics["started_at"] = session.started_at.isoformat()
+        if session.ended_at:
+            metrics["ended_at"] = session.ended_at.isoformat()
+
+        return AnalyticsSessionResult(
+            session_id=session_id, metrics=metrics, warnings=warnings
+        )
+
+    except AnalyticsOperationError:
+        raise
+    except Exception as e:
+        raise AnalyticsOperationError(f"Failed to analyze session {session_id}: {e}")
+
+
+def analyze_project(*, graph_dir: Path) -> AnalyticsProjectResult:
+    """
+    Compute analytics for the project.
+
+    Args:
+        graph_dir: Path to .htmlgraph directory
+
+    Returns:
+        AnalyticsProjectResult with project metrics and warnings
+
+    Raises:
+        AnalyticsOperationError: If project cannot be analyzed
+    """
+    warnings: list[str] = []
+
+    # Validate inputs
+    if not graph_dir.exists():
+        raise AnalyticsOperationError(f"Graph directory does not exist: {graph_dir}")
+
+    sessions_dir = graph_dir / "sessions"
+    if not sessions_dir.exists():
+        warnings.append("No sessions directory found")
+        return AnalyticsProjectResult(metrics={"total_sessions": 0}, warnings=warnings)
+
+    try:
+        # Initialize SDK
+        sdk = SDK(directory=str(graph_dir), agent="analytics-ops")
+
+        # Get session count
+        session_files = sorted(
+            sessions_dir.glob("*.html"), key=lambda p: p.stat().st_mtime, reverse=True
+        )
+        total_sessions = len(session_files)
+
+        # Compute metrics
+        metrics: dict[str, Any] = {
+            "total_sessions": total_sessions,
+        }
+
+        if total_sessions == 0:
+            warnings.append("No sessions found in project")
+            return AnalyticsProjectResult(metrics=metrics, warnings=warnings)
+
+        # Project-wide work distribution
+        try:
+            work_dist = sdk.analytics.work_type_distribution()
+            metrics["work_distribution"] = work_dist
+        except Exception as e:
+            warnings.append(f"Failed to compute work distribution: {e}")
+            metrics["work_distribution"] = {}
+
+        # Project-wide spike-to-feature ratio
+        try:
+            spike_ratio = sdk.analytics.spike_to_feature_ratio()
+            metrics["spike_to_feature_ratio"] = spike_ratio
+        except Exception as e:
+            warnings.append(f"Failed to compute spike ratio: {e}")
+            metrics["spike_to_feature_ratio"] = 0.0
+
+        # Project-wide maintenance burden
+        try:
+            maintenance = sdk.analytics.maintenance_burden()
+            metrics["maintenance_burden"] = maintenance
+        except Exception as e:
+            warnings.append(f"Failed to compute maintenance burden: {e}")
+            metrics["maintenance_burden"] = 0.0
+
+        # Project-wide transition metrics
+        try:
+            transition = sdk.analytics.transition_time_metrics()
+            metrics["transition_metrics"] = transition
+        except Exception as e:
+            warnings.append(f"Failed to compute transition metrics: {e}")
+            metrics["transition_metrics"] = {}
+
+        # Session type breakdown
+        try:
+            from htmlgraph import WorkType
+
+            spike_sessions = sdk.analytics.get_sessions_by_work_type(
+                WorkType.SPIKE.value
+            )
+            feature_sessions = sdk.analytics.get_sessions_by_work_type(
+                WorkType.FEATURE.value
+            )
+            maintenance_sessions = sdk.analytics.get_sessions_by_work_type(
+                WorkType.MAINTENANCE.value
+            )
+
+            metrics["session_types"] = {
+                "spike": len(spike_sessions),
+                "feature": len(feature_sessions),
+                "maintenance": len(maintenance_sessions),
+            }
+        except Exception as e:
+            warnings.append(f"Failed to compute session types: {e}")
+            metrics["session_types"] = {}
+
+        # Recent sessions (metadata only)
+        try:
+            recent_sessions = []
+            for session_path in session_files[:5]:  # Top 5 most recent
+                try:
+                    session = html_to_session(session_path)
+                    primary = (
+                        sdk.analytics.calculate_session_primary_work_type(session.id)
+                        or "unknown"
+                    )
+                    recent_sessions.append(
+                        {
+                            "session_id": session.id,
+                            "agent": session.agent,
+                            "started_at": session.started_at.isoformat(),
+                            "status": session.status,
+                            "primary_work_type": primary,
+                        }
+                    )
+                except Exception as e:
+                    warnings.append(f"Failed to load session {session_path.name}: {e}")
+                    continue
+
+            metrics["recent_sessions"] = recent_sessions
+        except Exception as e:
+            warnings.append(f"Failed to load recent sessions: {e}")
+            metrics["recent_sessions"] = []
+
+        return AnalyticsProjectResult(metrics=metrics, warnings=warnings)
+
+    except AnalyticsOperationError:
+        raise
+    except Exception as e:
+        raise AnalyticsOperationError(f"Failed to analyze project: {e}")
+
+
+def get_recommendations(*, graph_dir: Path) -> RecommendationsResult:
+    """
+    Get work recommendations based on project state.
+
+    Args:
+        graph_dir: Path to .htmlgraph directory
+
+    Returns:
+        RecommendationsResult with recommendations, reasoning, and warnings
+
+    Raises:
+        AnalyticsOperationError: If recommendations cannot be generated
+    """
+    warnings: list[str] = []
+
+    # Validate inputs
+    if not graph_dir.exists():
+        raise AnalyticsOperationError(f"Graph directory does not exist: {graph_dir}")
+
+    try:
+        # Initialize SDK
+        sdk = SDK(directory=str(graph_dir), agent="analytics-ops")
+
+        # Get recommendations
+        try:
+            task_recs = sdk.dep_analytics.recommend_next_tasks(agent_count=5)
+            recommendations = [
+                {
+                    "id": rec.id,
+                    "title": rec.title,
+                    "priority": rec.priority,
+                    "score": rec.score,
+                    "reasons": rec.reasons,
+                    "unlocks": rec.unlocks,
+                    "estimated_effort": rec.estimated_effort,
+                }
+                for rec in task_recs.recommendations
+            ]
+            reasoning = {
+                "recommendation_count": len(task_recs.recommendations),
+                "parallel_suggestions": task_recs.parallel_suggestions,
+            }
+        except Exception as e:
+            raise AnalyticsOperationError(f"Failed to generate recommendations: {e}")
+
+        # Add contextual warnings based on recommendations
+        if not recommendations:
+            warnings.append("No recommendations available - project may be empty")
+
+        return RecommendationsResult(
+            recommendations=recommendations, reasoning=reasoning, warnings=warnings
+        )
+
+    except AnalyticsOperationError:
+        raise
+    except Exception as e:
+        raise AnalyticsOperationError(f"Failed to get recommendations: {e}")

htmlgraph/operations/bootstrap.py

@@ -0,0 +1,289 @@
+from __future__ import annotations
+
+"""HtmlGraph bootstrap operations.
+
+One-command setup to go from installation to first value in under 60 seconds.
+This module provides functions for bootstrapping a project with HtmlGraph.
+
+The bootstrap process includes:
+1. Auto-detecting project type (Python, Node, etc.)
+2. Creating .htmlgraph directory structure
+3. Initializing database with schema
+4. Installing Claude Code plugin hooks automatically
+5. Printing next steps for the user
+
+This is designed for simplicity and speed - the minimal viable setup.
+"""
+
+
+import json
+import subprocess
+from pathlib import Path
+from typing import TYPE_CHECKING, Any
+
+if TYPE_CHECKING:
+    from htmlgraph.cli.models import BootstrapConfig
+
+
+def detect_project_type(project_dir: Path) -> str:
+    """
+    Auto-detect project type from files in directory.
+
+    Args:
+        project_dir: Project directory to inspect
+
+    Returns:
+        Detected project type: "python", "node", "multi", or "unknown"
+    """
+    # Check for Python project markers
+    has_python = any(
+        [
+            (project_dir / "pyproject.toml").exists(),
+            (project_dir / "setup.py").exists(),
+            (project_dir / "requirements.txt").exists(),
+            (project_dir / "Pipfile").exists(),
+        ]
+    )
+
+    # Check for Node project markers
+    has_node = (project_dir / "package.json").exists()
+
+    # Determine project type
+    if has_python and has_node:
+        return "multi"
+    elif has_python:
+        return "python"
+    elif has_node:
+        return "node"
+    else:
+        return "unknown"
+
+
+def create_gitignore_template() -> str:
+    """
+    Create .gitignore template content for .htmlgraph directory.
+
+    Returns:
+        Gitignore template content
+    """
+    return """# HtmlGraph cache and regenerable files
+.htmlgraph/htmlgraph.db
+.htmlgraph/sessions/*.jsonl
+.htmlgraph/events/*.jsonl
+.htmlgraph/parent-activity.json
+.htmlgraph/logs/
+"""
+
+
+def check_already_initialized(project_dir: Path) -> bool:
+    """
+    Check if project is already initialized with HtmlGraph.
+
+    Args:
+        project_dir: Project directory to check
+
+    Returns:
+        True if already initialized, False otherwise
+    """
+    graph_dir = project_dir / ".htmlgraph"
+    return graph_dir.exists()
+
+
+def create_bootstrap_structure(project_dir: Path) -> dict[str, list[str]]:
+    """
+    Create minimal .htmlgraph directory structure for bootstrap.
+
+    Args:
+        project_dir: Project directory
+
+    Returns:
+        Dictionary with lists of created directories and files
+    """
+    graph_dir = project_dir / ".htmlgraph"
+    created_dirs: list[str] = []
+    created_files: list[str] = []
+
+    # Create main .htmlgraph directory
+    if not graph_dir.exists():
+        graph_dir.mkdir(parents=True)
+        created_dirs.append(str(graph_dir))
+
+    # Create subdirectories
+    subdirs = [
+        "sessions",
+        "features",
+        "spikes",
+        "tracks",
+        "events",
+        "logs",
+        "logs/errors",
+    ]
+
+    for subdir in subdirs:
+        subdir_path = graph_dir / subdir
+        if not subdir_path.exists():
+            subdir_path.mkdir(parents=True)
+            created_dirs.append(str(subdir_path))
+
+    # Create .gitignore in .htmlgraph
+    gitignore = graph_dir / ".gitignore"
+    if not gitignore.exists():
+        gitignore.write_text(create_gitignore_template())
+        created_files.append(str(gitignore))
+
+    # Create config.json
+    config_file = graph_dir / "config.json"
+    if not config_file.exists():
+        config_data = {
+            "bootstrapped": True,
+            "version": "1.0",
+        }
+        config_file.write_text(json.dumps(config_data, indent=2) + "\n")
+        created_files.append(str(config_file))
+
+    return {"directories": created_dirs, "files": created_files}
+
+
+def initialize_database(graph_dir: Path) -> str:
+    """
+    Initialize HtmlGraph database with schema.
+
+    Args:
+        graph_dir: Path to .htmlgraph directory
+
+    Returns:
+        Path to created database file
+    """
+    from htmlgraph.db.schema import HtmlGraphDB
+
+    db_path = graph_dir / "htmlgraph.db"
+
+    # Create database using HtmlGraphDB (auto-creates tables)
+    db = HtmlGraphDB(db_path=str(db_path))
+    db.disconnect()
+
+    return str(db_path)
+
+
+def check_claude_code_available() -> bool:
+    """
+    Check if Claude Code CLI is available.
+
+    Returns:
+        True if claude command is available, False otherwise
+    """
+    try:
+        result = subprocess.run(
+            ["claude", "--version"],
+            capture_output=True,
+            check=False,
+            timeout=5,
+        )
+        return result.returncode == 0
+    except (subprocess.TimeoutExpired, FileNotFoundError):
+        return False
+
+
+def get_next_steps(
+    project_type: str, has_claude: bool, plugin_installed: bool
+) -> list[str]:
+    """
+    Generate next steps message based on project state.
+
+    Args:
+        project_type: Detected project type
+        has_claude: Whether Claude Code CLI is available
+        plugin_installed: Whether plugin hooks were installed
+
+    Returns:
+        List of next step messages
+    """
+    steps = []
+
+    if has_claude:
+        if plugin_installed:
+            steps.append("1. Use Claude Code: Run 'claude --dev' in this project")
+        else:
+            steps.append(
+                "1. Install HtmlGraph plugin: Run 'claude plugin install htmlgraph'"
+            )
+            steps.append("2. Use Claude Code: Run 'claude --dev' in this project")
+    else:
+        steps.append(
+            "1. Install Claude Code CLI: Visit https://code.claude.com/docs/installation"
+        )
+        steps.append(
+            "2. Install HtmlGraph plugin: Run 'claude plugin install htmlgraph'"
+        )
+        steps.append("3. Use Claude Code: Run 'claude --dev' in this project")
+
+    steps.append(
+        f"{len(steps) + 1}. Track work: Create features with 'htmlgraph feature create \"Title\"'"
+    )
+    steps.append(f"{len(steps) + 1}. View progress: Run 'htmlgraph status'")
+    steps.append(
+        f"{len(steps) + 1}. See what Claude did: Run 'htmlgraph serve' and open http://localhost:8080"
+    )
+
+    return steps
+
+
+def bootstrap_htmlgraph(config: BootstrapConfig) -> dict[str, Any]:
+    """
+    Bootstrap HtmlGraph in a project directory.
+
+    This is the main entry point for the bootstrap command.
+
+    Args:
+        config: BootstrapConfig with bootstrap settings
+
+    Returns:
+        Dictionary with bootstrap results
+    """
+    project_dir = Path(config.project_path).resolve()
+
+    # Check if already initialized
+    if check_already_initialized(project_dir):
+        # Ask user if they want to overwrite
+        print(f"\n⚠️  HtmlGraph already initialized in {project_dir}")
+        response = input("Do you want to reinitialize? (y/N): ").strip().lower()
+        if response not in ["y", "yes"]:
+            return {
+                "success": False,
+                "message": "Bootstrap cancelled - already initialized",
+            }
+
+    # Detect project type
+    project_type = detect_project_type(project_dir)
+
+    # Create directory structure
+    created = create_bootstrap_structure(project_dir)
+    graph_dir = project_dir / ".htmlgraph"
+
+    # Initialize database
+    db_path = initialize_database(graph_dir)
+    created["files"].append(db_path)
+
+    # Check for Claude Code
+    has_claude = check_claude_code_available()
+
+    # Check if plugin is already available (skip installation check for now)
+    plugin_installed = False
+    if not config.no_plugins and has_claude:
+        # We'll consider it "installed" if hooks can be configured
+        # The actual plugin installation happens via marketplace
+        plugin_installed = True
+
+    # Generate next steps
+    next_steps = get_next_steps(project_type, has_claude, plugin_installed)
+
+    return {
+        "success": True,
+        "project_type": project_type,
+        "graph_dir": str(graph_dir),
+        "directories_created": created["directories"],
+        "files_created": created["files"],
+        "has_claude": has_claude,
+        "plugin_installed": plugin_installed,
+        "next_steps": next_steps,
+    }