#!/usr/bin/env python3
"""prompt_envelope.py — wraps every dispatched job prompt so agents DO work.

Why: agents treated job DMs as informational and replied with prose; swarms and
timers never got created. The envelope puts an executable subagent-spawn tool
call at the TOP (before the task) and repeats it at the BOTTOM (after the task),
with a work-first mandate and explicit tool/curl surface in between.

Profiles tune the spawn recipe per job family; add new ones to PROFILES.
"""
import json
import re

BOX_EXEC = "https://box.muse-dev.online/exec"
UUID_RE = re.compile(r"[0-9a-f]{8}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{12}")

# profile -> (spawn count, followup minutes, spawn task hint)
PROFILES = {
    "pulse":  (2, 30, "execute your scope's standing work: set timers, verify them, report per-slot verdicts"),
    "health": (2, 60, "run the prescribed health gates read-only and report per-gate verdicts"),
    "swarm":  (2, 20, "sweep swarms and sidechats read-only and report stale or unarchived items"),
    "work":   (2, 15, "split the task above into two independent halves and complete each"),
}


def pick_profile(job_name):
    n = job_name or ""
    if n.startswith("autonomy-pulse"):
        return "pulse"
    if "health" in n:
        return "health"
    if "swarm" in n:
        return "swarm"
    return "work"


def _tool(op, args):
    # no ']' inside the JSON: the harvester's [TOOL ...] regex stops at the first one
    return "[TOOL %s %s]" % (op, json.dumps(args, separators=(", ", ": ")))


def spawn_call(job_id, job_name, profile):
    count, _, hint = PROFILES[profile]
    # no brackets in the task text: the harvester's [TOOL ...] regex stops at the first ']'
    task = "Subagent for job %s (%s): %s." % (job_id, job_name, hint)
    return _tool("swarm.spawn", {"count": count, "task": task[:900], "label": (job_name or "job")[:60]})


def native_followup_call(job_id, job_name, profile, agent):
    """Muse-native one-shot cron (cron.create runonce) - bridged to followup.create."""
    _, mins, _ = PROFILES[profile]
    return _tool("cron.create", {
        "kind": "runonce", "agent": agent, "in_m": mins,
        "prompt": "Continue job %s (%s): verify with cron.runs and swarm.list then act or close." % (job_id, job_name)})


def followup_call(job_id, job_name, profile, agent):
    _, mins, _ = PROFILES[profile]
    return _tool("followup.create", {
        "agent": agent, "in_m": mins,
        "prompt": "Continue job %s (%s): check swarm results with swarm.list then act or close." % (job_id, job_name)})


def thread_url(target):
    t = str(target)
    return "https://box.muse-dev.online/%s/%s" % ("thread" if UUID_RE.fullmatch(t.lower()) else "sidechat", t)


def wrap(job_name, job_id, agent, target, rendered):
    """Return the full DM body (without the [JOB id] prefix)."""
    profile = pick_profile(job_name)
    spawn = spawn_call(job_id, job_name, profile)
    follow = native_followup_call(job_id, job_name, profile, agent)
    top = (
        "WORK ORDER - ACTION REQUIRED, NOT INFORMATIONAL.\n"
        "Your FIRST output must be tool calls, not prose. Start now:\n"
        "  1) %s\n"
        "  2) %s\n"
        "Each tool call returns a result message; keep going until the task is done.\n"
        "---- TASK ----\n" % (spawn, follow)
    )
    has_result = "[RESULT" in rendered
    bottom = (
        "\n---- END TASK ----\n"
        "REMINDER (importance: critical): work is only real if a tool call ran. "
        "If you have not yet spawned subagents or set a follow-up timer, do it now:\n"
        "  %s\n"
        "  %s\n"
        "Native tools (preferred): cron.create (runonce/interval), cron.status, cron.view, cron.runs "
        "(truth about what ran), cron.run, subagent spawn; hooks.dry_run before hooks.enable. "
        "Native cron form: %s\n"
        "Box fallback tools: [TOOL health.check {}] [TOOL swarm.list {}] [TOOL swarm.status {\"swarm_id\": \"<id>\"}] "
        "[TOOL swarm.results {\"swarm_id\": \"<id>\"}] [TOOL quality.check {}]\n"
        "Direct: curl -sk -X POST %s -H \"Authorization: Bearer <token>\" "
        "-H \"Content-Type: application/json\" -d '{\"op\":\"<op>\",\"args\":{...}}'\n"
        "Thread: %s\n" % (spawn, follow, native_followup_call(job_id, job_name, profile, agent), BOX_EXEC, thread_url(target))
    )
    if not has_result:
        bottom += "Close with: [RESULT %s] <OK|FAILED>: <summary of tool outputs>\n" % job_id
    else:
        bottom += "Close using the [RESULT ...] line specified in the task, summarising tool outputs.\n"
    return top + rendered.strip() + "\n" + bottom