challenger.py

import asyncio

from api import API
from botli_dataclasses import ApiChallengeResponse, ChallengeRequest, ChallengeResponse


class Challenger:
    def __init__(self, api: API) -> None:
        self.api = api
        self.tasks: set[asyncio.Task[None]] = set()

    async def create(self, challenge_request: ChallengeRequest) -> ChallengeResponse:
        challenge_id = None

        challenge_queue: asyncio.Queue[ApiChallengeResponse] = asyncio.Queue()
        task = asyncio.create_task(self.api.create_challenge(challenge_request, challenge_queue))
        self.tasks.add(task)
        task.add_done_callback(self.tasks.discard)

        while response := await challenge_queue.get():
            if response.challenge_id:
                challenge_id = response.challenge_id

            if response.was_accepted:
                return ChallengeResponse(challenge_id=challenge_id, success=True)

            if response.was_declined:
                return ChallengeResponse()

            if response.has_reached_rate_limit:
                print(f"Challenge against {challenge_request.opponent_username} failed due to Lichess rate limit.")
                return ChallengeResponse(has_reached_rate_limit=True, wait_seconds=response.wait_seconds)

            if response.invalid_initial:
                print("Challenge failed due to invalid initial time.")
                return ChallengeResponse(is_misconfigured=True)

            if response.invalid_increment:
                print("Challenge failed due to invalid increment time.")
                return ChallengeResponse(is_misconfigured=True)

            if response.has_timed_out:
                print(f"Challenge against {challenge_request.opponent_username} has timed out.")
                if challenge_id is not None:
                    await self.api.cancel_challenge(challenge_id)
                return ChallengeResponse()

            if response.error:
                print(response.error)
                return ChallengeResponse(wait_seconds=response.wait_seconds)

        return ChallengeResponse()