"""Hello Xbox - presses A on the Xbox controller via Titan Two.

Requires the bridge.gpc script to be loaded and running on the Titan Two

(load once via Gtuner IV, then it runs standalone). Defaults to pressing A

for 150 ms.

"""

from __future__ import annotations

import argparse

import sys

import time

from pathlib import Path

from rich.console import Console

sys.path.insert(0, str(Path(__file__).resolve().parents[1]))

from io_titan.bridge import TitanBridge, tap

console = Console()

def main() -> int:

    ap = argparse.ArgumentParser()

    ap.add_argument("--button", default="A", help="Button to tap: A,B,X,Y,LB,RB,LS,RS,UP,DOWN,LEFT,RIGHT,BACK,START")

    ap.add_argument("--hold-ms", type=int, default=150)

    ap.add_argument("--count", type=int, default=1)

    ap.add_argument("--sweep", action="store_true", help="Instead of tapping, sweep the left stick in a circle.")

    args = ap.parse_args()

    console.print("[cyan]Opening Titan Two HID...[/cyan]")

    with TitanBridge() as t:

        console.print("[green]Opened.[/green] Sending to Xbox.")

        if args.sweep:

            console.print("Sweeping left stick (3 seconds). Watch any stick-driven UI on screen.")

            t0 = time.perf_counter()

            import math

            while time.perf_counter() - t0 < 3.0:

                theta = (time.perf_counter() - t0) * 2 * math.pi

                t.send(lx=80 * math.cos(theta), ly=80 * math.sin(theta))

                time.sleep(0.016)

            t.send()

            console.print("[green]Sweep complete.[/green]")

        else:

            for i in range(args.count):

                console.print(f"  tap {args.button} ({i+1}/{args.count})")

                tap(t, args.button, args.hold_ms / 1000.0)

                time.sleep(0.25)

        console.print("[green]Done. Disarming and releasing device.[/green]")

    return 0

if __name__ == "__main__":

    sys.exit(main())