"""Submit a MARKET order through the API Gateway and print the result.

Usage::

    python3 submit_market_order.py --side BUY  --symbol AAPL --qty 100
    python3 submit_market_order.py --side SELL --symbol MSFT --qty 50 --wait-ack
    python3 submit_market_order.py --side BUY  --symbol AAPL --qty 100 --client-tag demo-1

Environment variables:

    EDUMATCHER_API_URL  Base URL of the API gateway (default: http://127.0.0.1:8080)
    EDUMATCHER_API_KEY  Bearer token              (default: key-trader-demo)
"""

from __future__ import annotations

import argparse
import json
import os
import sys
import uuid
from typing import Any

from api_gateway_client import ApiGatewayClient, ApiUnreachable


def _print_result(result: dict[str, Any]) -> None:
    print(f"order_id  : {result.get('order_id', '-')}")
    print(f"status    : {result.get('status', '-')}")
    if result.get("client_tag"):
        print(f"client_tag: {result['client_tag']}")
    if result.get("accepted") is not None:
        print(f"accepted  : {result['accepted']}")
    if result.get("ack_timeout"):
        # Sent, but the ack was late: follow it, do not submit it again.
        print("ack       : not received in time; the order was sent")
    if result.get("event"):
        print("engine ack:")
        print(json.dumps(result["event"], indent=2))


def main() -> None:
    parser = argparse.ArgumentParser(
        description="Submit a MARKET order via the API Gateway"
    )
    parser.add_argument(
        "--side", required=True, choices=["BUY", "SELL"], help="Order side"
    )
    parser.add_argument("--symbol", required=True, help="Instrument symbol, e.g. AAPL")
    parser.add_argument(
        "--qty", required=True, type=int, metavar="N", help="Order quantity"
    )
    parser.add_argument(
        "--wait-ack",
        action="store_true",
        help="Block until the matching engine ACKs the order",
    )
    parser.add_argument(
        "--client-tag",
        default=None,
        help="Optional opaque client correlation tag",
    )
    parser.add_argument(
        "--url",
        default=os.environ.get("EDUMATCHER_API_URL", "http://127.0.0.1:8080"),
        help="API gateway base URL",
    )
    parser.add_argument(
        "--key",
        default=os.environ.get("EDUMATCHER_API_KEY", "key-trader-demo"),
        help="Bearer API key",
    )
    args = parser.parse_args()

    client = ApiGatewayClient(args.url, args.key)
    path = "/api/v1/orders" + ("?wait=ack" if args.wait_ack else "")
    payload = {
        "symbol": args.symbol.upper(),
        "side": args.side,
        "order_type": "MARKET",
        "quantity": args.qty,
    }
    if args.client_tag:
        payload["client_tag"] = args.client_tag

    # One key per logical order, reused on a retry, so a retry after a
    # dropped connection can never submit the order twice.
    key = str(uuid.uuid4())
    try:
        try:
            result = client.post_json(path, payload, idempotency_key=key)
        except ApiUnreachable:
            # No answer: the order may or may not have gone out. The same key
            # makes the retry safe -- the gateway replays the first reply.
            result = client.post_json(path, payload, idempotency_key=key)
    except RuntimeError as exc:
        print(f"error: {exc}", file=sys.stderr)
        sys.exit(1)

    _print_result(result)


if __name__ == "__main__":
    main()