from __future__ import annotations

import argparse
import json
from typing import Any


WINDOW_START = "2026-07-06T21:50:00Z"
WINDOW_END = "2026-07-07T01:00:00Z"


def _resolve_window() -> dict[str, Any]:
    return {
        "start": WINDOW_START,
        "end": WINDOW_END,
    }


def _build_reference(cluster_id: str, start: str, end: str) -> dict[str, Any]:
    return {
        "cluster_id": cluster_id,
        "window": {
            "start": start,
            "end": end,
        },
        "requested_ratio": 0.5,
        "expected_refill_rate_us_per_interval": 1240000,
        "observed_refill_rate_us_per_interval": 992000,
        "applied": False,
        "evidence_sufficient": True,
    }


def main() -> None:
    parser = argparse.ArgumentParser()
    subparsers = parser.add_subparsers(dest="command", required=True)

    subparsers.add_parser("resolve-window")

    ref = subparsers.add_parser("build-reference")
    ref.add_argument("--cluster-id", required=True)
    ref.add_argument("--start", required=True)
    ref.add_argument("--end", required=True)

    args = parser.parse_args()

    if args.command == "resolve-window":
        payload = _resolve_window()
    else:
        payload = _build_reference(
            cluster_id=args.cluster_id,
            start=args.start,
            end=args.end,
        )
    print(json.dumps(payload, ensure_ascii=False))


if __name__ == "__main__":
    main()