"""One demo agent. It only knows MSGBOARD_URL, MSGBOARD_TOKEN and MSGBOARD_CHANNEL from its environment."""

from __future__ import annotations

import argparse
import os
import sys

from .client import BoardError, Client


def main() -> int:
    parser = argparse.ArgumentParser()
    parser.add_argument("--mode", choices=["ping", "pong"], required=True)
    parser.add_argument("--rounds", type=int, default=3)
    args = parser.parse_args()

    with Client.from_env(timeout=10) as client:
        me = client.me()
        name = me["external_id"]
        channel = client.channel_id(os.environ["MSGBOARD_CHANNEL"])

        # Workers cannot create channels or mint tokens; show that the board refuses.
        if args.mode == "ping":
            try:
                client.create_channel("not-allowed")
            except BoardError as exc:
                print(f"[{name}] create_channel refused as expected: {exc.status} {exc.code}", flush=True)
            try:
                client.rotate_token(me["id"])
            except BoardError as exc:
                print(f"[{name}] making itself a new token refused as expected: {exc.status} {exc.code}", flush=True)

        if args.mode == "ping":
            for i in range(1, args.rounds + 1):
                # A request ID derived from the agent's own state survives restarts of this process.
                sent = client.post(channel, f"ping {i}", request_id=f"{name}-ping-{i}")
                print(f"[{name}] posted seq={sent['seq']}: {sent['body']}", flush=True)
                # Replies always come after the ping, so start looking right after it.
                for msg in client.follow(channel, after=sent["seq"], wait=10):
                    if msg["reply_to"] == sent["id"]:
                        print(f"[{name}] got reply seq={msg['seq']} from {msg['sender']['external_id']}: "
                              f"{msg['body']}", flush=True)
                        break
        else:
            answered = 0
            for msg in client.follow(channel, after=0, wait=10):
                if msg["sender"]["id"] != me["id"] and msg["body"].startswith("ping"):
                    reply = client.post(channel, msg["body"].replace("ping", "pong"), reply_to=msg["id"],
                                        request_id=f"{name}-reply-{msg['id']}")
                    print(f"[{name}] replied seq={reply['seq']}: {reply['body']}", flush=True)
                    answered += 1
                    if answered == args.rounds:
                        break
    return 0


if __name__ == "__main__":
    sys.exit(main())