#!/usr/bin/env python3
"""Assist filling the first external oracle packet without inventing oracle values."""

from __future__ import annotations

import argparse
import json
import subprocess
import sys
from pathlib import Path
from typing import Any


ROOT = Path(__file__).resolve().parents[1]
PYTHON = sys.executable

FRONTS = {
    "dasha": {
        "status_command": [
            PYTHON,
            "scripts/dasha_oracle_closure_status.py",
            "--oracle-file",
            "references/oracle/dasha_shadbala_oracle_cases.json",
            "--format",
            "json",
        ],
        "operator_card": "docs/benchmark/dasha_steve_jobs_first_packet_operator_card.md",
        "packet_template": "references/oracle/evidence_packet_templates/dasha_steve_jobs_lahiri_first_packet_only.json",
        "oracle_file": "references/oracle/dasha_shadbala_oracle_cases.json",
        "apply_script": "scripts/oracle_collection_queue.py",
        "external_sources": [
            "JHora Vimshottari Dasha screen",
            "PyJHora black-box dasha output",
            "documented printed/software example",
        ],
    },
    "tajika_sahams": {
        "status_command": [
            PYTHON,
            "scripts/tajika_annual_closure_status.py",
            "--oracle-file",
            "references/oracle/tajika_annual_oracle_cases.json",
            "--format",
            "json",
        ],
        "operator_card": "docs/benchmark/tajika_einstein_1905_first_packet_operator_card.md",
        "packet_template": "references/oracle/artifacts/pending_packets/external_template_einstein_varshaphala_1905_lahiri.json",
        "oracle_file": "references/oracle/tajika_annual_oracle_cases.json",
        "apply_script": "scripts/tajika_annual_oracle_queue.py",
        "external_sources": [
            "JHora Varshaphala/Tajika annual chart screen",
            "PyJHora black-box annual output",
            "printed Tajika/Varshaphala example",
        ],
    },
    "shadbala": {
        "status_command": [
            PYTHON,
            "scripts/shadbala_oracle_closure_status.py",
            "--oracle-file",
            "references/oracle/dasha_shadbala_oracle_cases.json",
            "--format",
            "json",
        ],
        "operator_card": "docs/benchmark/shadbala_synthetic_north_china_raman_first_packet_operator_card.md",
        "packet_template": "references/oracle/evidence_packet_templates/shadbala_synthetic_north_china_raman_first_packet.json",
        "oracle_file": "references/oracle/dasha_shadbala_oracle_cases.json",
        "apply_script": "scripts/oracle_collection_queue.py",
        "external_sources": [
            "JHora Shadbala component table screenshot",
            "PyJHora black-box shadbala output",
            "documented printed/software Shadbala example",
        ],
    }
}


def _fallback_apply_command(config: dict[str, Any]) -> str:
    return (
        f"python3 {config['apply_script']} "
        f"--oracle-file {config['oracle_file']} "
        f"--apply-packet {config['packet_template']} "
        "--format json"
    )


def _fallback_validate_command(config: dict[str, Any]) -> str:
    queue_path = "/tmp/jyotish_oracle_queue_filled.json"
    return (
        f"python3 scripts/oracle_collection_queue.py --oracle-file {config['oracle_file']} "
        f"--format json > {queue_path} && "
        f"python3 scripts/oracle_evidence_validator.py --queue-file {queue_path}"
    )


def _group_missing_fields(missing_fields: list[str]) -> dict[str, Any]:
    metadata_fields = [field for field in missing_fields if field.startswith("metadata.")]
    target_fields = [field for field in missing_fields if field.startswith("target.")]
    body_groups: dict[str, list[str]] = {}
    prefix = "target.shadbala_components."
    for field in target_fields:
        if not field.startswith(prefix):
            continue
        remainder = field[len(prefix):]
        body, _, component = remainder.partition(".")
        if not body or not component:
            continue
        body_groups.setdefault(body, []).append(field)
    grouped_bodies = {
        body: {
            "count": len(fields),
            "fields": fields,
        }
        for body, fields in sorted(body_groups.items())
    }
    return {
        "metadata": {
            "count": len(metadata_fields),
            "fields": metadata_fields,
        },
        "target": {
            "count": len(target_fields),
            "fields": target_fields,
        },
        "bodies": grouped_bodies,
    }


def _prefilled_fields(packet: dict[str, Any], missing_fields: list[str]) -> dict[str, Any]:
    missing = set(missing_fields)
    metadata = {
        key: value
        for key, value in packet.get("metadata", {}).items()
        if f"metadata.{key}" not in missing and value not in ("", None, [], {})
    }
    settings = {
        key: value
        for key, value in packet.get("settings", {}).items()
        if value not in ("", None, [], {})
    }
    return {
        "status": packet.get("status"),
        "promotion_status_after_fill": packet.get("promotion_status_after_fill"),
        "metadata": metadata,
        "settings": settings,
    }


def _manual_fill_plan(packet: dict[str, Any], missing_fields: list[str]) -> dict[str, Any]:
    return {
        "status_value": packet.get("promotion_status_after_fill", "external_verified"),
        "manual_entry_count": len(missing_fields),
        "remaining_manual_fields": missing_fields,
    }


def _run_json(command: list[str]) -> dict[str, Any]:
    completed = subprocess.run(
        command,
        cwd=ROOT,
        text=True,
        capture_output=True,
        timeout=60,
        check=False,
    )
    if completed.returncode != 0:
        raise RuntimeError(completed.stderr.strip() or completed.stdout.strip())
    return json.loads(completed.stdout)


def _packet_missing(packet_path: str) -> list[str]:
    packet = json.loads((ROOT / packet_path).read_text(encoding="utf-8"))
    missing: list[str] = []
    metadata = packet.get("metadata", {})
    for field in packet.get("required_metadata_fields", []):
        value = metadata.get(field)
        if value is None or value == "" or value == [] or value == {}:
            missing.append(f"metadata.{field}")
    if metadata.get("source_artifact") in {"references/oracle/artifacts/", "references/oracle/artifacts", "", None}:
        if "metadata.source_artifact" not in missing:
            missing.append("metadata.source_artifact")
    def walk(prefix: str, value: Any) -> None:
        if prefix == "target.tajika_yogas":
            if value is None or value == "" or value == {}:
                missing.append(prefix)
            return
        if isinstance(value, dict):
            for key, child in value.items():
                walk(f"{prefix}.{key}" if prefix else key, child)
            return
        if value is None or value == "" or value == [] or value == {}:
            missing.append(prefix)

    for field, value in packet.get("target_placeholders", {}).items():
        walk(field, value)
    return missing


def build_report(front: str) -> dict[str, Any]:
    if front not in FRONTS:
        raise RuntimeError(f"Unsupported front: {front}")
    config = FRONTS[front]
    status = _run_json(config["status_command"])
    packet = json.loads((ROOT / config["packet_template"]).read_text(encoding="utf-8"))
    status_first = status.get("first_priority")
    if status_first is None:
        return {
            "scope": "first_external_oracle_packet_assistant",
            "schema_version": 1,
            "front": front,
            "case_id": packet.get("case_id"),
            "capture_id": packet.get("capture_id"),
            "operator_card": config["operator_card"],
            "packet_template": config["packet_template"],
            "missing_fields": [],
            "missing_groups": _group_missing_fields([]),
            "prefilled_fields": _prefilled_fields(packet, []),
            "manual_fill_plan": _manual_fill_plan(packet, []),
            "ready_to_apply": True,
            "external_sources": config["external_sources"],
            "apply_command": "",
            "validate_command": "",
            "boundary": (
                "This assistant only reports what to fill. The current target set for this front is already "
                "closed, so this entry now acts as a historical operator reference rather than an active fill task."
            ),
        }
    first = status.get("first_priority") or {
        "case_id": packet.get("case_id"),
        "capture_id": packet.get("capture_id"),
        "apply_command": _fallback_apply_command(config),
        "validate_command": _fallback_validate_command(config),
        "packet_path": config["packet_template"],
    }
    packet_template = first.get("packet_path") or config["packet_template"]
    if not Path(packet_template).is_absolute():
        packet_path = str(ROOT / packet_template)
    else:
        packet_path = packet_template
    packet_missing = _packet_missing(str(Path(packet_path).relative_to(ROOT)))
    missing_groups = _group_missing_fields(packet_missing)
    apply_command = first.get("apply_command") or _fallback_apply_command(config)
    return {
        "scope": "first_external_oracle_packet_assistant",
        "schema_version": 1,
        "front": front,
        "case_id": first["case_id"],
        "capture_id": first["capture_id"],
        "operator_card": config["operator_card"],
        "packet_template": str(Path(packet_path).relative_to(ROOT)),
        "missing_fields": packet_missing,
        "missing_groups": missing_groups,
        "prefilled_fields": _prefilled_fields(json.loads(Path(packet_path).read_text(encoding="utf-8")), packet_missing),
        "manual_fill_plan": _manual_fill_plan(json.loads(Path(packet_path).read_text(encoding="utf-8")), packet_missing),
        "ready_to_apply": not packet_missing,
        "external_sources": config["external_sources"],
        "apply_command": apply_command,
        "validate_command": first.get("validate_command", ""),
        "boundary": (
            "This assistant only reports what to fill. It must not invent external oracle values, "
            "must not use local engine output as evidence, and must not copy incompatible external code."
        ),
    }


def render_markdown(report: dict[str, Any]) -> str:
    lines = [
        "# First External Oracle Packet Assistant",
        "",
        f"- front: `{report['front']}`",
        f"- case_id: `{report['case_id']}`",
        f"- capture_id: `{report['capture_id']}`",
        f"- ready_to_apply: `{str(report['ready_to_apply']).lower()}`",
        f"- operator_card: `{report['operator_card']}`",
        f"- packet_template: `{report['packet_template']}`",
        "",
        "## Missing Summary",
        "",
        f"- metadata: `{report['missing_groups']['metadata']['count']}`",
        f"- target: `{report['missing_groups']['target']['count']}`",
    ]
    if report["missing_groups"]["bodies"]:
        lines.extend(["- bodies:", ""])
        lines.extend(
            f"  - {body}: `{payload['count']}`"
            for body, payload in report["missing_groups"]["bodies"].items()
        )
    lines.extend(
        [
            "",
            "## Prefilled Fields",
            "",
            f"- status: `{report['prefilled_fields']['status']}`",
            f"- promotion_status_after_fill: `{report['prefilled_fields']['promotion_status_after_fill']}`",
            "",
        ]
    )
    if report["prefilled_fields"]["metadata"]:
        lines.append("- metadata:")
        lines.append("")
        lines.extend(
            f"  - {key}: `{value}`"
            for key, value in report["prefilled_fields"]["metadata"].items()
        )
        lines.append("")
    if report["prefilled_fields"]["settings"]:
        lines.append("- settings:")
        lines.append("")
        lines.extend(
            f"  - {key}: `{value}`"
            for key, value in report["prefilled_fields"]["settings"].items()
        )
        lines.append("")
    lines.extend(
        [
            "",
            "## Manual Fill Plan",
            "",
            f"- status_value: `{report['manual_fill_plan']['status_value']}`",
            f"- manual_entry_count: `{report['manual_fill_plan']['manual_entry_count']}`",
            "",
            "## Missing Fields",
            "",
        ]
    )
    lines.extend(f"- `{field}`" for field in report["missing_fields"])
    lines.extend(
        [
            "",
            "## External Sources",
            "",
        ]
    )
    lines.extend(f"- {source}" for source in report["external_sources"])
    lines.extend(
        [
            "",
            "## Apply",
            "",
            "```bash",
            report["apply_command"],
            "```",
            "",
            "## Validate",
            "",
            "```bash",
            report["validate_command"],
            "```",
            "",
            "## Boundary",
            "",
            report["boundary"],
            "",
        ]
    )
    return "\n".join(lines)


def parse_args(argv: list[str] | None = None) -> argparse.Namespace:
    parser = argparse.ArgumentParser(description="Assist the first external oracle packet")
    parser.add_argument("--front", choices=sorted(FRONTS), required=True)
    parser.add_argument("--format", choices=["json", "markdown"], default="json")
    parser.add_argument("--output", help="Optional output path")
    return parser.parse_args(argv)


def main(argv: list[str] | None = None) -> int:
    args = parse_args(argv)
    report = build_report(args.front)
    text = json.dumps(report, ensure_ascii=False, indent=2) if args.format == "json" else render_markdown(report)
    if args.output:
        output = Path(args.output)
        output.parent.mkdir(parents=True, exist_ok=True)
        output.write_text(text, encoding="utf-8")
    print(text)
    return 0


if __name__ == "__main__":
    raise SystemExit(main())