cowork-flow 0.0.1

package/template/.cowork-flow/scripts/common/git_context.py

@@ -0,0 +1,628 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+"""
+Git and Session Context utilities.
+
+Provides:
+    output_json - Output context in JSON format
+    output_text - Output context in text format
+"""
+
+from __future__ import annotations
+
+import json
+import subprocess
+from dataclasses import dataclass
+from pathlib import Path
+
+from .files import read_json_file as _read_json_file
+from .paths import (
+    DIR_SCRIPTS,
+    DIR_SPEC,
+    DIR_TASKS,
+    DIR_WORKFLOW,
+    DIR_WORKSPACE,
+    FILE_TASK_JSON,
+    count_lines,
+    get_active_journal_file,
+    get_current_task,
+    get_developer,
+    get_repo_root,
+    get_tasks_dir,
+)
+
+# =============================================================================
+# Helper Functions
+# =============================================================================
+
+
+@dataclass(frozen=True)
+class GitSnapshot:
+    branch: str
+    status_count: int
+    log_out: str
+    recent_commits: list[dict]
+
+
+@dataclass(frozen=True)
+class CurrentTaskSnapshot:
+    path: str | None
+    data: dict | None
+    has_prd: bool
+
+
+def _run_git_command(args: list[str], cwd: Path | None = None) -> tuple[int, str, str]:
+    """Run a git command and return (returncode, stdout, stderr).
+
+    Uses UTF-8 encoding with -c i18n.logOutputEncoding=UTF-8 to ensure
+    consistent output across all platforms (Windows, macOS, Linux).
+    """
+    try:
+        # Force git to output UTF-8 for consistent cross-platform behavior
+        git_args = ["git", "-c", "i18n.logOutputEncoding=UTF-8"] + args
+        result = subprocess.run(
+            git_args,
+            cwd=cwd,
+            capture_output=True,
+            text=True,
+            encoding="utf-8",
+            errors="replace",
+        )
+        return result.returncode, result.stdout, result.stderr
+    except Exception as e:
+        return 1, "", str(e)
+
+
+def _iter_task_dirs(tasks_dir: Path, sort: bool = True):
+    """Yield active task directories."""
+    if not tasks_dir.is_dir():
+        return
+
+    dirs = sorted(tasks_dir.iterdir()) if sort else tasks_dir.iterdir()
+    for d in dirs:
+        if d.is_dir() and d.name != "archive":
+            yield d
+
+
+def _load_task_json_by_dir(tasks_dir: Path, sort: bool = True) -> dict[str, dict]:
+    """Load task.json data keyed by task directory name."""
+    tasks: dict[str, dict] = {}
+    for d in _iter_task_dirs(tasks_dir, sort=sort):
+        task_json = d / FILE_TASK_JSON
+        if task_json.is_file():
+            data = _read_json_file(task_json)
+            if data:
+                tasks[d.name] = data
+    return tasks
+
+
+def _parse_recent_commits(log_out: str, include_empty_message: bool = True) -> list[dict]:
+    """Parse git log --oneline output into JSON-ready commit summaries."""
+    commits = []
+    for line in log_out.splitlines():
+        if not line.strip():
+            continue
+
+        parts = line.split(" ", 1)
+        if len(parts) >= 2:
+            commits.append({"hash": parts[0], "message": parts[1]})
+        elif include_empty_message:
+            commits.append({"hash": parts[0], "message": ""})
+    return commits
+
+
+def _get_git_snapshot(
+    repo_root: Path,
+    include_empty_commit_message: bool = True,
+) -> GitSnapshot:
+    """Collect git data shared by JSON and text renderers."""
+    _, branch_out, _ = _run_git_command(["branch", "--show-current"], cwd=repo_root)
+    _, status_out, _ = _run_git_command(["status", "--porcelain"], cwd=repo_root)
+    _, log_out, _ = _run_git_command(["log", "--oneline", "-5"], cwd=repo_root)
+
+    status_lines = [line for line in status_out.splitlines() if line.strip()]
+    return GitSnapshot(
+        branch=branch_out.strip() or "unknown",
+        status_count=len(status_lines),
+        log_out=log_out,
+        recent_commits=_parse_recent_commits(
+            log_out,
+            include_empty_message=include_empty_commit_message,
+        ),
+    )
+
+
+def _git_json(git_snapshot: GitSnapshot) -> dict:
+    """Build the git JSON object used by context output."""
+    status_count = git_snapshot.status_count
+    return {
+        "branch": git_snapshot.branch,
+        "isClean": status_count == 0,
+        "uncommittedChanges": status_count,
+        "recentCommits": git_snapshot.recent_commits,
+    }
+
+
+def _append_git_status(lines: list[str], git_snapshot: GitSnapshot, repo_root: Path) -> None:
+    """Append the GIT STATUS section to text output."""
+    lines.append("## GIT STATUS")
+    lines.append(f"Branch: {git_snapshot.branch}")
+
+    status_count = git_snapshot.status_count
+    if status_count == 0:
+        lines.append("Working directory: Clean")
+    else:
+        lines.append(f"Working directory: {status_count} uncommitted change(s)")
+        lines.append("")
+        lines.append("Changes:")
+        _, short_out, _ = _run_git_command(["status", "--short"], cwd=repo_root)
+        for line in short_out.splitlines()[:10]:
+            lines.append(line)
+    lines.append("")
+
+
+def _append_recent_commits(lines: list[str], git_snapshot: GitSnapshot) -> None:
+    """Append the RECENT COMMITS section to text output."""
+    lines.append("## RECENT COMMITS")
+    log_out = git_snapshot.log_out
+    if log_out.strip():
+        for line in log_out.splitlines():
+            lines.append(line)
+    else:
+        lines.append("(no commits)")
+    lines.append("")
+
+
+def _get_current_task_snapshot(repo_root: Path) -> CurrentTaskSnapshot:
+    """Collect current-task metadata once for text and JSON renderers."""
+    current_task = get_current_task(repo_root)
+    if not current_task:
+        return CurrentTaskSnapshot(path=None, data=None, has_prd=False)
+
+    current_task_dir = repo_root / current_task
+    task_json_path = current_task_dir / FILE_TASK_JSON
+    data = None
+    if task_json_path.is_file():
+        data = _read_json_file(task_json_path)
+    return CurrentTaskSnapshot(
+        path=current_task,
+        data=data,
+        has_prd=(current_task_dir / "prd.md").is_file(),
+    )
+
+
+def _current_task_json(snapshot: CurrentTaskSnapshot) -> dict | None:
+    """Build the record-mode currentTask JSON object."""
+    current_task = snapshot.path
+    data = snapshot.data
+    if not current_task or not data:
+        return None
+
+    return {
+        "path": current_task,
+        "name": data.get("name") or data.get("id") or "unknown",
+        "status": data.get("status", "unknown"),
+    }
+
+
+def _append_current_task(
+    lines: list[str],
+    snapshot: CurrentTaskSnapshot,
+    include_created: bool = False,
+    include_description: bool = False,
+    include_prd_hint: bool = False,
+) -> None:
+    """Append the CURRENT TASK section to text output."""
+    lines.append("## CURRENT TASK")
+    current_task = snapshot.path
+    if not current_task:
+        lines.append("(none)")
+        lines.append("")
+        return
+
+    lines.append(f"Path: {current_task}")
+    data = snapshot.data
+    if data:
+        lines.append(f"Name: {data.get('name') or data.get('id') or 'unknown'}")
+        lines.append(f"Status: {data.get('status', 'unknown')}")
+        if include_created:
+            lines.append(f"Created: {data.get('createdAt', 'unknown')}")
+        if include_description:
+            description = data.get("description", "")
+            if description:
+                lines.append(f"Description: {description}")
+
+    if include_prd_hint and snapshot.has_prd:
+        lines.append("")
+        lines.append("[!] This task has prd.md - read it for task details")
+    lines.append("")
+
+
+def _load_task_context_by_dir(tasks_dir: Path) -> dict[str, dict]:
+    """Load task display data, preserving dirs without task.json as unknown."""
+    all_task_data: dict[str, dict] = {}
+    for d in _iter_task_dirs(tasks_dir):
+        data = _read_json_file(d / FILE_TASK_JSON) or {}
+        all_task_data[d.name] = {
+            "status": data.get("status", "unknown"),
+            "my_status": data.get("status", "planning"),
+            "assignee": data.get("assignee", "-"),
+            "my_assignee": data.get("assignee", ""),
+            "title": data.get("title") or data.get("name") or "unknown",
+            "priority": data.get("priority", "P2"),
+            "children": data.get("children", []),
+            "parent": data.get("parent"),
+        }
+    return all_task_data
+
+
+def _task_statuses(task_data_by_dir: dict[str, dict]) -> dict[str, str]:
+    """Return task status lookup by directory name."""
+    return {
+        dir_name: data.get("status", "unknown")
+        for dir_name, data in task_data_by_dir.items()
+    }
+
+
+def _children_done_count(children: list[str], statuses: dict[str, str]) -> int:
+    """Count completed child tasks."""
+    return sum(1 for child in children if statuses.get(child) in ("completed", "done"))
+
+
+def _children_progress(children: list[str], statuses: dict[str, str]) -> str:
+    """Render children progress like '[2/3 done]'."""
+    if not children:
+        return ""
+    return f" [{_children_done_count(children, statuses)}/{len(children)} done]"
+
+
+# =============================================================================
+# JSON Output
+# =============================================================================
+
+
+def get_context_json(repo_root: Path | None = None) -> dict:
+    """Get context as a dictionary.
+
+    Args:
+        repo_root: Repository root path. Defaults to auto-detected.
+
+    Returns:
+        Context dictionary.
+    """
+    if repo_root is None:
+        repo_root = get_repo_root()
+
+    developer = get_developer(repo_root)
+    tasks_dir = get_tasks_dir(repo_root)
+    journal_file = get_active_journal_file(repo_root)
+    git_snapshot = _get_git_snapshot(repo_root)
+
+    journal_lines = 0
+    journal_relative = ""
+    if journal_file and developer:
+        journal_lines = count_lines(journal_file)
+        journal_relative = (
+            f"{DIR_WORKFLOW}/{DIR_WORKSPACE}/{developer}/{journal_file.name}"
+        )
+
+    # Tasks
+    tasks = []
+    for dir_name, data in _load_task_json_by_dir(tasks_dir, sort=False).items():
+        tasks.append(
+            {
+                "dir": dir_name,
+                "name": data.get("name") or data.get("id") or "unknown",
+                "status": data.get("status", "unknown"),
+                "children": data.get("children", []),
+                "parent": data.get("parent"),
+            }
+        )
+
+    return {
+        "developer": developer or "",
+        "git": _git_json(git_snapshot),
+        "tasks": {
+            "active": tasks,
+            "directory": f"{DIR_WORKFLOW}/{DIR_TASKS}",
+        },
+        "journal": {
+            "file": journal_relative,
+            "lines": journal_lines,
+            "nearLimit": journal_lines > 1800,
+        },
+    }
+
+
+def output_json(repo_root: Path | None = None) -> None:
+    """Output context in JSON format.
+
+    Args:
+        repo_root: Repository root path. Defaults to auto-detected.
+    """
+    context = get_context_json(repo_root)
+    print(json.dumps(context, indent=2, ensure_ascii=False))
+
+
+# =============================================================================
+# Text Output
+# =============================================================================
+
+
+def get_context_text(repo_root: Path | None = None) -> str:
+    """Get context as formatted text.
+
+    Args:
+        repo_root: Repository root path. Defaults to auto-detected.
+
+    Returns:
+        Formatted text output.
+    """
+    if repo_root is None:
+        repo_root = get_repo_root()
+
+    lines = []
+    lines.append("========================================")
+    lines.append("SESSION CONTEXT")
+    lines.append("========================================")
+    lines.append("")
+
+    developer = get_developer(repo_root)
+
+    # Developer section
+    lines.append("## DEVELOPER")
+    if not developer:
+        lines.append(
+            f"ERROR: Not initialized. Run: python3 ./{DIR_WORKFLOW}/{DIR_SCRIPTS}/init_developer.py <name>"
+        )
+        return "\n".join(lines)
+
+    lines.append(f"Name: {developer}")
+    lines.append("")
+
+    git_snapshot = _get_git_snapshot(repo_root)
+    _append_git_status(lines, git_snapshot, repo_root)
+    _append_recent_commits(lines, git_snapshot)
+    _append_current_task(
+        lines,
+        _get_current_task_snapshot(repo_root),
+        include_created=True,
+        include_description=True,
+        include_prd_hint=True,
+    )
+
+    # Active tasks
+    lines.append("## ACTIVE TASKS")
+    tasks_dir = get_tasks_dir(repo_root)
+    task_count = 0
+
+    all_task_data = _load_task_context_by_dir(tasks_dir)
+    all_task_statuses = {
+        dir_name: data["status"]
+        for dir_name, data in all_task_data.items()
+    }
+
+    def _print_task_tree(name: str, indent: int = 0) -> None:
+        nonlocal task_count
+        info = all_task_data[name]
+        progress = _children_progress(info["children"], all_task_statuses)
+        prefix = "  " * indent
+        lines.append(f"{prefix}- {name}/ ({info['status']}){progress} @{info['assignee']}")
+        task_count += 1
+        for child in info["children"]:
+            if child in all_task_data:
+                _print_task_tree(child, indent + 1)
+
+    for dir_name in sorted(all_task_data.keys()):
+        if not all_task_data[dir_name]["parent"]:
+            _print_task_tree(dir_name)
+
+    if task_count == 0:
+        lines.append("(no active tasks)")
+    lines.append(f"Total: {task_count} active task(s)")
+    lines.append("")
+
+    # My tasks
+    lines.append("## MY TASKS (Assigned to me)")
+    my_task_count = 0
+
+    for dir_name in sorted(all_task_data.keys()):
+        info = all_task_data[dir_name]
+        assignee = info["my_assignee"]
+        status = info["my_status"]
+
+        if assignee == developer and status != "done":
+            children_list = info["children"]
+            progress = _children_progress(children_list, all_task_statuses)
+            lines.append(f"- [{info['priority']}] {info['title']} ({status}){progress}")
+            my_task_count += 1
+
+    if my_task_count == 0:
+        lines.append("(no tasks assigned to you)")
+    lines.append("")
+
+    # Journal file
+    lines.append("## JOURNAL FILE")
+    journal_file = get_active_journal_file(repo_root)
+    if journal_file:
+        journal_lines = count_lines(journal_file)
+        relative = f"{DIR_WORKFLOW}/{DIR_WORKSPACE}/{developer}/{journal_file.name}"
+        lines.append(f"Active file: {relative}")
+        lines.append(f"Line count: {journal_lines} / 2000")
+        if journal_lines > 1800:
+            lines.append("[!] WARNING: Approaching 2000 line limit!")
+    else:
+        lines.append("No journal file found")
+    lines.append("")
+
+    # Paths
+    lines.append("## PATHS")
+    lines.append(f"Workspace: {DIR_WORKFLOW}/{DIR_WORKSPACE}/{developer}/")
+    lines.append(f"Tasks: {DIR_WORKFLOW}/{DIR_TASKS}/")
+    lines.append(f"Spec: {DIR_WORKFLOW}/{DIR_SPEC}/")
+    lines.append("")
+
+    lines.append("========================================")
+
+    return "\n".join(lines)
+
+
+def get_context_record_json(repo_root: Path | None = None) -> dict:
+    """Get record-mode context as a dictionary.
+
+    Focused on: my active tasks, git status, current task.
+    """
+    if repo_root is None:
+        repo_root = get_repo_root()
+
+    developer = get_developer(repo_root)
+    tasks_dir = get_tasks_dir(repo_root)
+
+    git_snapshot = _get_git_snapshot(
+        repo_root,
+        include_empty_commit_message=False,
+    )
+
+    # My tasks
+    my_tasks = []
+    task_data_by_dir = _load_task_json_by_dir(tasks_dir)
+    all_task_statuses = _task_statuses(task_data_by_dir)
+
+    for dir_name, data in task_data_by_dir.items():
+        if data.get("assignee") == developer:
+            children_list = data.get("children", [])
+            done = _children_done_count(children_list, all_task_statuses)
+            my_tasks.append({
+                "dir": dir_name,
+                "title": data.get("title") or data.get("name") or "unknown",
+                "status": data.get("status", "unknown"),
+                "priority": data.get("priority", "P2"),
+                "children": children_list,
+                "childrenDone": done,
+                "parent": data.get("parent"),
+                "meta": data.get("meta", {}),
+            })
+
+    return {
+        "developer": developer or "",
+        "git": _git_json(git_snapshot),
+        "myTasks": my_tasks,
+        "currentTask": _current_task_json(_get_current_task_snapshot(repo_root)),
+    }
+
+
+def get_context_text_record(repo_root: Path | None = None) -> str:
+    """Get context as formatted text for record-session mode.
+
+    Focused output: MY ACTIVE TASKS first (with [!!!] emphasis),
+    then GIT STATUS, RECENT COMMITS, CURRENT TASK.
+
+    Args:
+        repo_root: Repository root path. Defaults to auto-detected.
+
+    Returns:
+        Formatted text output for record-session.
+    """
+    if repo_root is None:
+        repo_root = get_repo_root()
+
+    lines: list[str] = []
+    lines.append("========================================")
+    lines.append("SESSION CONTEXT (RECORD MODE)")
+    lines.append("========================================")
+    lines.append("")
+
+    developer = get_developer(repo_root)
+    if not developer:
+        lines.append(
+            f"ERROR: Not initialized. Run: python3 ./{DIR_WORKFLOW}/{DIR_SCRIPTS}/init_developer.py <name>"
+        )
+        return "\n".join(lines)
+
+    # MY ACTIVE TASKS - first and prominent
+    lines.append(f"## [!!!] MY ACTIVE TASKS (Assigned to {developer})")
+    lines.append("[!] Review whether any should be archived before recording this session.")
+    lines.append("")
+
+    tasks_dir = get_tasks_dir(repo_root)
+    my_task_count = 0
+
+    # Collect task data for children progress
+    task_data_by_dir = _load_task_json_by_dir(tasks_dir)
+    all_task_statuses = _task_statuses(task_data_by_dir)
+
+    for dir_name, data in task_data_by_dir.items():
+        assignee = data.get("assignee", "")
+        status = data.get("status", "planning")
+
+        if assignee == developer:
+            title = data.get("title") or data.get("name") or "unknown"
+            priority = data.get("priority", "P2")
+            children_list = data.get("children", [])
+            progress = _children_progress(children_list, all_task_statuses)
+            lines.append(f"- [{priority}] {title} ({status}){progress} - {dir_name}")
+            my_task_count += 1
+
+    if my_task_count == 0:
+        lines.append("(no active tasks assigned to you)")
+    lines.append("")
+
+    git_snapshot = _get_git_snapshot(repo_root)
+    _append_git_status(lines, git_snapshot, repo_root)
+    _append_recent_commits(lines, git_snapshot)
+    _append_current_task(lines, _get_current_task_snapshot(repo_root))
+
+    lines.append("========================================")
+
+    return "\n".join(lines)
+
+
+def output_text(repo_root: Path | None = None) -> None:
+    """Output context in text format.
+
+    Args:
+        repo_root: Repository root path. Defaults to auto-detected.
+    """
+    print(get_context_text(repo_root))
+
+
+# =============================================================================
+# Main Entry
+# =============================================================================
+
+
+def main() -> None:
+    """CLI entry point."""
+    import argparse
+
+    parser = argparse.ArgumentParser(description="Get Session Context for AI Agent")
+    parser.add_argument(
+        "--json",
+        "-j",
+        action="store_true",
+        help="Output in JSON format (works with any --mode)",
+    )
+    parser.add_argument(
+        "--mode",
+        "-m",
+        choices=["default", "record"],
+        default="default",
+        help="Output mode: default (full context) or record (for record-session)",
+    )
+
+    args = parser.parse_args()
+
+    if args.mode == "record":
+        if args.json:
+            print(json.dumps(get_context_record_json(), indent=2, ensure_ascii=False))
+        else:
+            print(get_context_text_record())
+    else:
+        if args.json:
+            output_json()
+        else:
+            output_text()
+
+
+if __name__ == "__main__":
+    main()