#!/usr/bin/env python3

"""Plan, execute and resume bounded local coding-agent campaigns."""

from __future__ import annotations

import argparse

import json

from pathlib import Path

import sys

from oakbench.agent_campaign import (CampaignError, create_plan, load_plan, load_events,

                                    campaign_status, run_campaign)

def main(argv=None) -> int:

    parser = argparse.ArgumentParser(description=__doc__)

    sub = parser.add_subparsers(dest="command", required=True)

    plan = sub.add_parser("plan", help="freeze a manifest, schedule and provenance; invokes no agent task")

    plan.add_argument("--manifest", type=Path, required=True)

    plan.add_argument("--campaign-dir", type=Path, required=True)

    for name in ("run", "resume", "status", "report"):

        child = sub.add_parser(name)

        child.add_argument("--campaign-dir", type=Path, required=True)

        if name in ("run", "resume"):

            child.add_argument("--enable-real", action="store_true")

            child.add_argument("--limit", type=int, help="maximum new trials this invocation")

        if name == "resume":

            child.add_argument("--acknowledge-torn-journal", action="store_true", help="archive torn journal, reserve uncertain trial, and resume without retrying it")

        if name == "report":

            child.add_argument("--format", choices=("json", "markdown"), default="json")

    args = parser.parse_args(argv)

    try:

        if args.command == "plan":

            frozen = create_plan(args.manifest, args.campaign_dir)

            print(json.dumps({"campaign_id": frozen["campaign_id"], "plan_sha256": frozen["plan_sha256"],

                              "scheduled": len(frozen["schedule"]), "plan": str(args.campaign_dir / "plan.json")}, indent=2))

            return 0

        if args.command in ("run", "resume"):

            state = run_campaign(args.campaign_dir, resume=args.command == "resume", enable_real=args.enable_real, limit=args.limit, acknowledge_torn=getattr(args, "acknowledge_torn_journal", False))

        else:

            frozen = load_plan(args.campaign_dir)

            events = load_events(args.campaign_dir, frozen, tolerate_torn=True)

            if args.command == "report":

                from oakbench.campaign_report import report_campaign, render_markdown

                report = report_campaign(frozen, events)

                print(json.dumps(report, indent=2) if args.format == "json" else render_markdown(report))

                return 0

            state = campaign_status(frozen, events)

        print(json.dumps(state, indent=2))

        if args.command in ("run", "resume"):

            if state["pending_trial_ids"] or state["interrupted_trial_ids"]:

                return 3

            events = load_events(args.campaign_dir, load_plan(args.campaign_dir))

            return 1 if any(event.get("outcome") in {"error", "fail", "timeout"} for event in events) else 0

        return 0

    except (ValueError, OSError) as exc:

        print("campaign error: " + str(exc), file=sys.stderr)

        return 2

if __name__ == "__main__":

    raise SystemExit(main())