#!/usr/bin/env python3
from __future__ import annotations

import argparse
import sys

from eth_abi import encode
from web3 import Web3


def main() -> int:
    parser = argparse.ArgumentParser(
        description="Encode PolyWeatherCheckout constructor args for PolygonScan verification.",
    )
    parser.add_argument(
        "--token",
        help="Initial allowed token address (e.g. USDC.e or Native USDC)",
    )
    parser.add_argument("--usdc", help="Backward-compatible alias of --token")
    parser.add_argument("--treasury", required=True, help="Treasury address")
    args = parser.parse_args()

    token = args.token or args.usdc
    if not token:
        print("missing --token (or --usdc)", file=sys.stderr)
        return 1
    if not Web3.is_address(token):
        print("invalid --token address", file=sys.stderr)
        return 1
    if not Web3.is_address(args.treasury):
        print("invalid --treasury address", file=sys.stderr)
        return 1

    encoded = encode(
        ["address", "address"],
        [Web3.to_checksum_address(token), Web3.to_checksum_address(args.treasury)],
    ).hex()
    print(encoded)
    return 0


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