#!/usr/bin/env python3
"""Compute Deployment + Reporting DGA candidate domains from seeds (offline)."""

from __future__ import annotations

import argparse
import json
import sys
from pathlib import Path

from _common import (
    MODULE_HUNT,
    ORIGINAL_DEPLOYMENT_SEED,
    ORIGINAL_REPORTING_SEED,
    validate_seed_arg,
)

sys.path.insert(0, str(MODULE_HUNT))
from reproduce_coruna_dga import generate_domains  # noqa: E402


def main() -> int:
    parser = argparse.ArgumentParser(
        description="Print Coruna PLServerPool DGA candidates for Deployment + Reporting seeds"
    )
    parser.add_argument(
        "--deployment-seed",
        default=ORIGINAL_DEPLOYMENT_SEED,
        help=f"default: original campaign seed ({ORIGINAL_DEPLOYMENT_SEED})",
    )
    parser.add_argument(
        "--reporting-seed",
        default=ORIGINAL_REPORTING_SEED,
        help=f"default: original campaign seed ({ORIGINAL_REPORTING_SEED})",
    )
    parser.add_argument(
        "-n",
        "--count",
        type=int,
        default=5,
        help="candidates per pool (operational pool uses 5; max 512)",
    )
    parser.add_argument("--json", action="store_true", help="emit JSON")
    args = parser.parse_args()
    dep = validate_seed_arg("--deployment-seed", args.deployment_seed)
    rep = validate_seed_arg("--reporting-seed", args.reporting_seed)

    payload = {
        "deployment": {
            "seed": dep,
            "domains": generate_domains(dep, args.count),
        },
        "reporting": {
            "seed": rep,
            "domains": generate_domains(rep, args.count),
        },
        "note": (
            "Native helper generates 512 strings; PLServerPool keeps the first 5 "
            "as the live candidate pool."
        ),
    }

    if args.json:
        print(json.dumps(payload, indent=2))
        return 0

    print(f"deployment seed={dep}")
    for i, domain in enumerate(payload["deployment"]["domains"], 1):
        print(f"  {i:03d} {domain}")
    print(f"reporting seed={rep}")
    for i, domain in enumerate(payload["reporting"]["domains"], 1):
        print(f"  {i:03d} {domain}")
    return 0


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