feat: first pass slash commands

Author: damassi

src/agent_chat_cli/components/header.py

@@ -33,7 +33,7 @@ def compose(self) -> ComposeResult:
         yield Spacer()
 
         yield Label(
-            "[dim]Type your message and press Enter. Type 'exit' to quit.[/dim]",
+            "[dim]Type your message and press Enter. Press / for commands.[/dim]",
             id="header-instructions",
             classes="header-instructions",
         )

src/agent_chat_cli/components/slash_command_menu.py

@@ -0,0 +1,45 @@
+from textual.widget import Widget
+from textual.app import ComposeResult
+from textual.widgets import OptionList
+from textual.widgets.option_list import Option
+
+from agent_chat_cli.core.actions import Actions
+
+
+class SlashCommandMenu(Widget):
+    def __init__(self, actions: Actions) -> None:
+        super().__init__()
+        self.actions = actions
+
+    def compose(self) -> ComposeResult:
+        yield OptionList(
+            Option("/new   - Start new conversation", id="new"),
+            Option("/clear - Clear chat history", id="clear"),
+            Option("/exit  - Exit", id="exit"),
+        )
+
+    def show(self) -> None:
+        self.add_class("visible")
+        option_list = self.query_one(OptionList)
+        option_list.highlighted = 0
+        option_list.focus()
+
+    def hide(self) -> None:
+        self.remove_class("visible")
+
+    @property
+    def is_visible(self) -> bool:
+        return self.has_class("visible")
+
+    async def on_option_list_option_selected(
+        self, event: OptionList.OptionSelected
+    ) -> None:
+        self.hide()
+
+        match event.option_id:
+            case "exit":
+                self.actions.quit()
+            case "clear":
+                await self.actions.clear()
+            case "new":
+                await self.actions.new()

src/agent_chat_cli/components/tool_permission_prompt.py

@@ -10,6 +10,7 @@
 from agent_chat_cli.components.flex import Flex
 from agent_chat_cli.components.spacer import Spacer
 from agent_chat_cli.utils import get_tool_info
+from agent_chat_cli.utils.enums import Key
 from agent_chat_cli.utils.logger import log_json
 
 if TYPE_CHECKING:
@@ -22,7 +23,7 @@ class ToolPermissionPrompt(Widget):
     tool_input: dict[str, Any] = reactive({}, init=False)  # type: ignore[assignment]
 
     BINDINGS = [
-        Binding("enter", "submit", "Submit", priority=True),
+        Binding(Key.ENTER.value, "submit", "Submit", priority=True),
     ]
 
     def __init__(self, actions: "Actions") -> None:
@@ -92,7 +93,7 @@ def on_descendant_blur(self) -> None:
             input_widget.focus()
 
     async def on_key(self, event) -> None:
-        if event.key == "escape":
+        if event.key == Key.ESCAPE.value:
             log_json({"event": "permission_escape_pressed"})
 
             event.stop()

src/agent_chat_cli/components/user_input.py

@@ -1,18 +1,20 @@
 from textual.widget import Widget
 from textual.app import ComposeResult
-from textual.widgets import TextArea
+from textual.widgets import TextArea, OptionList
 from textual.binding import Binding
 from textual.events import DescendantBlur
 
 from agent_chat_cli.components.caret import Caret
 from agent_chat_cli.components.flex import Flex
+from agent_chat_cli.components.slash_command_menu import SlashCommandMenu
 from agent_chat_cli.core.actions import Actions
-from agent_chat_cli.utils.enums import ControlCommand
+from agent_chat_cli.utils.enums import Key
 
 
 class UserInput(Widget):
     BINDINGS = [
-        Binding("enter", "submit", "Submit", priority=True),
+        Binding(Key.ENTER.value, "submit", "Submit", priority=True),
+        Binding(Key.ESCAPE.value, "hide_menu", "Hide Menu", priority=True),
     ]
 
     def __init__(self, actions: Actions) -> None:
@@ -27,38 +29,68 @@ def compose(self) -> ComposeResult:
                 show_line_numbers=False,
                 soft_wrap=True,
             )
+        yield SlashCommandMenu(actions=self.actions)
 
     def on_mount(self) -> None:
         input_widget = self.query_one(TextArea)
         input_widget.focus()
 
     def on_descendant_blur(self, event: DescendantBlur) -> None:
-        if isinstance(event.widget, TextArea):
+        menu = self.query_one(SlashCommandMenu)
+
+        if isinstance(event.widget, TextArea) and not menu.is_visible:
             event.widget.focus(scroll_visible=False)
+        elif isinstance(event.widget, OptionList) and menu.is_visible:
+            menu.hide()
+            self.query_one(TextArea).focus(scroll_visible=False)
+
+    def on_text_area_changed(self, event: TextArea.Changed) -> None:
+        menu = self.query_one(SlashCommandMenu)
+        text = event.text_area.text
+
+        if text == Key.SLASH.value:
+            event.text_area.clear()
+            menu.show()
 
     async def on_key(self, event) -> None:
-        if event.key == "ctrl+j":
+        if event.key == Key.CTRL_J.value:
             event.stop()
             event.prevent_default()
             input_widget = self.query_one(TextArea)
             input_widget.insert("\n")
+            return
+
+        menu = self.query_one(SlashCommandMenu)
+
+        if menu.is_visible and event.key in (Key.BACKSPACE.value, Key.DELETE.value):
+            event.stop()
+            event.prevent_default()
+            menu.hide()
+            self.query_one(TextArea).focus()
+
+    def action_hide_menu(self) -> None:
+        menu = self.query_one(SlashCommandMenu)
+
+        if menu.is_visible:
+            menu.hide()
+            input_widget = self.query_one(TextArea)
+            input_widget.clear()
+            input_widget.focus()
 
     async def action_submit(self) -> None:
+        menu = self.query_one(SlashCommandMenu)
+
+        if menu.is_visible:
+            option_list = menu.query_one(OptionList)
+            option_list.action_select()
+            self.query_one(TextArea).focus()
+            return
+
         input_widget = self.query_one(TextArea)
         user_message = input_widget.text.strip()
 
         if not user_message:
             return
 
-        if user_message.lower() == ControlCommand.EXIT.value:
-            self.actions.quit()
-            return
-
         input_widget.clear()
-
-        if user_message.lower() == ControlCommand.CLEAR.value:
-            await self.actions.interrupt()
-            await self.actions.new()
-            return
-
         await self.actions.submit_user_message(user_message)

src/agent_chat_cli/core/actions.py

@@ -37,14 +37,16 @@ async def interrupt(self) -> None:
         await self.app.agent_loop.client.interrupt()
         self.app.ui_state.stop_thinking()
 
-    async def new(self) -> None:
-        await self.app.agent_loop.query_queue.put(ControlCommand.NEW_CONVERSATION)
-
+    async def clear(self) -> None:
         chat_history = self.app.query_one(ChatHistory)
         await chat_history.remove_children()
 
         self.app.ui_state.stop_thinking()
 
+    async def new(self) -> None:
+        await self.app.agent_loop.query_queue.put(ControlCommand.NEW_CONVERSATION)
+        await self.clear()
+
     async def respond_to_tool_permission(self, response: str) -> None:
         log_json(
             {

src/agent_chat_cli/core/agent_loop.py

@@ -72,6 +72,8 @@ async def start(self) -> None:
                 if user_input == ControlCommand.NEW_CONVERSATION:
                     await self.client.disconnect()
 
+                    self.session_id = None
+
                     mcp_servers = {
                         name: config.model_dump()
                         for name, config in self.available_servers.items()

src/agent_chat_cli/core/styles.tcss

@@ -94,3 +94,20 @@ TextArea {
 .tool-message {
     padding-left: 2;
 }
+
+SlashCommandMenu {
+    height: auto;
+    max-height: 10;
+    display: none;
+}
+
+SlashCommandMenu.visible {
+    display: block;
+}
+
+SlashCommandMenu OptionList {
+    height: auto;
+    max-height: 10;
+    border: solid $primary;
+    background: $surface;
+}

src/agent_chat_cli/utils/enums.py

@@ -22,3 +22,12 @@ class ControlCommand(Enum):
     NEW_CONVERSATION = "new_conversation"
     EXIT = "exit"
     CLEAR = "clear"
+
+
+class Key(Enum):
+    ENTER = "enter"
+    ESCAPE = "escape"
+    BACKSPACE = "backspace"
+    DELETE = "delete"
+    CTRL_J = "ctrl+j"
+    SLASH = "/"

tests/components/test_slash_command_menu.py

@@ -0,0 +1,99 @@
+import pytest
+from unittest.mock import AsyncMock, MagicMock
+
+from textual.app import App, ComposeResult
+from textual.widgets import OptionList
+
+from agent_chat_cli.components.slash_command_menu import SlashCommandMenu
+
+
+class SlashCommandMenuApp(App):
+    def __init__(self):
+        super().__init__()
+        self.mock_actions = MagicMock()
+        self.mock_actions.quit = MagicMock()
+        self.mock_actions.clear = AsyncMock()
+        self.mock_actions.new = AsyncMock()
+
+    def compose(self) -> ComposeResult:
+        yield SlashCommandMenu(actions=self.mock_actions)
+
+
+class TestSlashCommandMenuVisibility:
+    @pytest.fixture
+    def app(self):
+        return SlashCommandMenuApp()
+
+    async def test_hidden_by_default(self, app):
+        async with app.run_test():
+            menu = app.query_one(SlashCommandMenu)
+
+            assert menu.is_visible is False
+
+    async def test_show_makes_visible(self, app):
+        async with app.run_test():
+            menu = app.query_one(SlashCommandMenu)
+            menu.show()
+
+            assert menu.is_visible is True
+
+    async def test_hide_makes_invisible(self, app):
+        async with app.run_test():
+            menu = app.query_one(SlashCommandMenu)
+            menu.show()
+            menu.hide()
+
+            assert menu.is_visible is False
+
+    async def test_show_highlights_first_option(self, app):
+        async with app.run_test():
+            menu = app.query_one(SlashCommandMenu)
+            menu.show()
+
+            option_list = menu.query_one(OptionList)
+            assert option_list.highlighted == 0
+
+
+class TestSlashCommandMenuSelection:
+    @pytest.fixture
+    def app(self):
+        return SlashCommandMenuApp()
+
+    async def test_new_command_calls_new(self, app):
+        async with app.run_test() as pilot:
+            menu = app.query_one(SlashCommandMenu)
+            menu.show()
+
+            await pilot.press("enter")
+
+            app.mock_actions.new.assert_called_once()
+
+    async def test_clear_command_calls_clear(self, app):
+        async with app.run_test() as pilot:
+            menu = app.query_one(SlashCommandMenu)
+            menu.show()
+
+            await pilot.press("down")
+            await pilot.press("enter")
+
+            app.mock_actions.clear.assert_called_once()
+
+    async def test_exit_command_calls_quit(self, app):
+        async with app.run_test() as pilot:
+            menu = app.query_one(SlashCommandMenu)
+            menu.show()
+
+            await pilot.press("down")
+            await pilot.press("down")
+            await pilot.press("enter")
+
+            app.mock_actions.quit.assert_called_once()
+
+    async def test_selection_hides_menu(self, app):
+        async with app.run_test() as pilot:
+            menu = app.query_one(SlashCommandMenu)
+            menu.show()
+
+            await pilot.press("enter")
+
+            assert menu.is_visible is False