"""Handle text input and output to and from bidi agent."""

import logging
from typing import Any

from prompt_toolkit import PromptSession

from ....types.content import TextBlock
from ..types.events import (
    BidiBargeInEvent,
    BidiConnectionStopEvent,
    BidiOutputEvent,
    BidiTranscriptDeltaEvent,
)
from ..types.io import InputStream, OutputStream

logger = logging.getLogger(__name__)


class _ConsoleInputStream(InputStream):
    """Handle text input from user."""

    def __init__(self, config: dict[str, Any]) -> None:
        """Extract configs and setup prompt session."""
        prompt = config.get("input_prompt", "")
        self._session: PromptSession = PromptSession(prompt)

    async def __call__(self) -> TextBlock:
        """Read user input from stdin."""
        text = await self._session.prompt_async()
        return TextBlock(text.strip())


class _ConsoleOutputStream(OutputStream):
    """Handle text output from bidi agent."""

    async def __call__(self, event: BidiOutputEvent) -> None:
        """Print text events to stdout."""
        if isinstance(event, BidiBargeInEvent):
            logger.debug("reason=<%s> | barge-in detected", event["reason"])
            print("barge-in")

        elif isinstance(event, BidiConnectionStopEvent):
            if event.reason == "user_request":
                print("user requested connection close using the stop tool.")
                logger.debug("connection_id=<%s> | user requested connection close", event.connection_id)
        elif isinstance(event, BidiTranscriptDeltaEvent):
            logger.debug(
                "role=<%s>, text_length=<%d> | text transcript received",
                event.role,
                len(event.delta),
            )
            print(event.delta)


class ConsoleIO:
    """Handle text input and output to and from bidi agent.

    Accepts input from stdin and outputs to stdout.
    """

    def __init__(self, **config: Any) -> None:
        """Initialize I/O.

        Args:
            **config: Optional I/O configurations.

                - input_prompt (str): Input prompt to display on screen (default: blank)
        """
        self._config = config

    def input(self) -> _ConsoleInputStream:
        """Return the standard-input stream."""
        return _ConsoleInputStream(self._config)

    def output(self) -> _ConsoleOutputStream:
        """Return the standard-output stream."""
        return _ConsoleOutputStream()