actions.py

from typing import TYPE_CHECKING

from agent_chat_cli.utils.enums import ControlCommand
from agent_chat_cli.components.chat_history import ChatHistory, MessagePosted
from agent_chat_cli.components.messages import Message
from agent_chat_cli.components.thinking_indicator import ThinkingIndicator
from agent_chat_cli.components.tool_permission_prompt import ToolPermissionPrompt
from agent_chat_cli.utils.logger import log_json

if TYPE_CHECKING:
    from agent_chat_cli.app import AgentChatCLIApp


class Actions:
    def __init__(self, app: "AgentChatCLIApp") -> None:
        self.app = app

    def quit(self) -> None:
        self.app.exit()

    async def query(self, user_input: str) -> None:
        await self.app.agent_loop.query_queue.put(user_input)

    async def submit_user_message(self, message: str) -> None:
        from textual.widgets import TextArea

        self.app.post_message(MessagePosted(Message.user(message)))

        thinking_indicator = self.app.query_one(ThinkingIndicator)
        thinking_indicator.is_thinking = True

        input_widget = self.app.query_one(TextArea)
        input_widget.cursor_blink = False

        await self.query(message)

    def post_system_message(self, message: str) -> None:
        self.app.post_message(MessagePosted(Message.system(message)))

    async def handle_agent_message(self, message) -> None:
        await self.app.message_bus.handle_agent_message(message)

    async def interrupt(self) -> None:
        permission_prompt = self.app.query_one(ToolPermissionPrompt)
        if permission_prompt.is_visible:
            return

        self.app.agent_loop.interrupting = True
        await self.app.agent_loop.client.interrupt()

        thinking_indicator = self.app.query_one(ThinkingIndicator)
        thinking_indicator.is_thinking = False

    async def new(self) -> None:
        await self.app.agent_loop.query_queue.put(ControlCommand.NEW_CONVERSATION)

        chat_history = self.app.query_one(ChatHistory)
        await chat_history.remove_children()

        thinking_indicator = self.app.query_one(ThinkingIndicator)
        thinking_indicator.is_thinking = False

    async def respond_to_tool_permission(self, response: str) -> None:
        from agent_chat_cli.components.user_input import UserInput
        from textual.widgets import TextArea

        log_json(
            {
                "event": "permission_response_action",
                "response": response,
            }
        )

        await self.app.agent_loop.permission_response_queue.put(response)

        permission_prompt = self.app.query_one(ToolPermissionPrompt)
        permission_prompt.is_visible = False

        user_input = self.app.query_one(UserInput)
        user_input.display = True

        input_widget = self.app.query_one(TextArea)
        input_widget.focus()

        thinking_indicator = self.app.query_one(ThinkingIndicator)
        thinking_indicator.is_thinking = True
        input_widget.cursor_blink = False

        # Check if it's a deny or custom response (anything except yes/allow)
        normalized = response.lower().strip()
        if normalized not in ["y", "yes", "allow", ""]:
            if normalized in ["n", "no", "deny"]:
                await self.query("The user has denied the tool")
            else:
                await self.submit_user_message(response)