Crossing the Tool Planes: State Continuity Across Tools

# work/gh-600/scripts/inject_cross_surface_context.py
"""
Reads context from the current GitHub surface (issue, PR, or Actions run)
and generates a context injection prompt for the next agent session.
"""

import os
import json
import sys
import subprocess
from datetime import datetime, timezone


def get_github_context() -> dict:
    """Read current GitHub environment context."""
    return {
        "repository": os.environ.get("GITHUB_REPOSITORY", ""),
        "run_id": os.environ.get("GITHUB_RUN_ID", ""),
        "event_name": os.environ.get("GITHUB_EVENT_NAME", ""),
        "sha": os.environ.get("GITHUB_SHA", ""),
        "ref": os.environ.get("GITHUB_REF", ""),
        "actor": os.environ.get("GITHUB_ACTOR", ""),
    }


def load_handoff(path: str) -> dict:
    """Load existing handoff object or create a new one."""
    if os.path.exists(path):
        with open(path) as f:
            return json.load(f)
    return {
        "task_id": "unknown",
        "original_intent": "",
        "current_stage": "issue",
        "history": []
    }


def update_handoff(handoff: dict, stage: str, outcome: str) -> dict:
    """Add current surface visit to history."""
    ctx = get_github_context()
    handoff["current_stage"] = stage
    handoff["history"].append({
        "surface": stage,
        "timestamp": datetime.now(timezone.utc).isoformat(),
        "outcome": outcome,
        "run_id": ctx.get("run_id", ""),
        "sha": ctx.get("sha", "")
    })
    return handoff


def generate_injection_prompt(handoff: dict) -> str:
    """Generate a context injection prompt for the next agent session."""
    history_summary = "\n".join([
        f"  - {h['surface']} ({h['timestamp'][:10]}): {h['outcome']}"
        for h in handoff["history"]
    ])
    
    return f"""# Cross-Surface Context Injection

You are continuing a task that was started in a previous session/surface.

## Original Task
{handoff['original_intent']}

## Task ID
{handoff['task_id']}

## Current Stage
{handoff['current_stage']}

## History (what has been done)
{history_summary or '  (no prior history)'}

## Instructions
- Do NOT re-do any work listed in the history above
- Continue from where the previous session left off
- If unclear what was completed, read the linked artifacts/commits
- Maintain the original task scope — do not expand it
"""


if __name__ == "__main__":
    handoff_path = sys.argv[1] if len(sys.argv) > 1 else ".agent-memory/handoff.json"
    stage = sys.argv[2] if len(sys.argv) > 2 else "unknown"
    outcome = sys.argv[3] if len(sys.argv) > 3 else "in-progress"

    handoff = load_handoff(handoff_path)
    handoff = update_handoff(handoff, stage, outcome)
    
    with open(handoff_path, "w") as f:
        json.dump(handoff, f, indent=2)
    
    prompt = generate_injection_prompt(handoff)
    print(prompt)
    
    with open(".agent-memory/injection-prompt.md", "w") as f:
        f.write(prompt)
    
    print(f"\n✅ Context handoff updated and injection prompt written")