#!/usr/bin/env python3

"""Approve a pending-ack task so the watcher promotes it to processing/.

Usage:

  dispatch-ack <task-id-or-prefix> [--side <node-id>]

"""

from __future__ import annotations

import argparse

import os

import pathlib

import sys

sys.path.insert(0, str(pathlib.Path(__file__).resolve().parent))

import dispatch_lib as d

def main() -> int:

    ap = argparse.ArgumentParser()

    ap.add_argument("task_id", help="task id or unique prefix")

    ap.add_argument("--side", default=os.environ.get("DISPATCH_SIDE"),

                    help="this node's id (or set DISPATCH_SIDE)")

    args = ap.parse_args()

    if not args.side:

        print("ERR: --side required (or set DISPATCH_SIDE)", file=sys.stderr)

        return 2

    other = d.NODE_B if args.side == d.NODE_A else d.NODE_A

    pending = d.ROOT / f"{other}-to-{args.side}" / "pending-ack"

    pending.mkdir(parents=True, exist_ok=True)

    matches = list(pending.glob(f"{args.task_id}*.json"))

    if not matches:

        print(f"ERR: no pending task matching '{args.task_id}'", file=sys.stderr)

        return 1

    if len(matches) > 1:

        print(f"ERR: prefix '{args.task_id}' matches {len(matches)} tasks; be more specific",

              file=sys.stderr)

        return 1

    task_path = matches[0]

    ack_path = task_path.with_suffix(".ack")

    ack_path.write_text("ack\n")

    print(f"acked: {task_path.name}")

    return 0

if __name__ == "__main__":

    raise SystemExit(main())