@scemoon/cdh 1.0.0

package/tui/screens/main.py

@@ -0,0 +1,411 @@
+from functools import partial
+from pathlib import Path
+import random
+
+from textual import on
+from textual.app import ComposeResult
+from textual import getters
+from textual.binding import Binding
+from textual.command import Hit, Hits, Provider, DiscoveryHit
+from textual.content import Content
+from textual.events import ScreenResume, Click
+from textual.screen import Screen
+from textual.reactive import var, reactive
+from textual.widgets import Footer, OptionList, DirectoryTree, Tree, Static
+from textual import containers
+from textual.widget import Widget
+
+
+from tui.app import A2TUIApp
+from tui import messages
+from tui.agent_schema import Agent
+from tui.acp import messages as acp_messages
+
+from tui.widgets.context_stats import ContextStats
+from tui.widgets.modified_files import ModifiedFiles
+from tui.widgets.plan import Plan
+from tui.widgets.throbber import Throbber
+from tui.widgets.conversation import Conversation
+from tui.widgets.project_directory_tree import ProjectDirectoryTree
+from tui.widgets.side_bar import SideBar
+from tui.widgets.session_tabs import SessionsTabs, SessionLabel
+from onecode.agent.cdh_loader import CdhProjectLoader
+
+
+class ModeProvider(Provider):
+    async def search(self, query: str) -> Hits:
+        """Search for Python files."""
+        matcher = self.matcher(query)
+
+        screen = self.screen
+        assert isinstance(screen, MainScreen)
+
+        for mode in sorted(
+            screen.conversation.modes.values(), key=lambda mode: mode.name
+        ):
+            command = mode.name
+            score = matcher.match(command)
+            if score > 0:
+                yield Hit(
+                    score,
+                    matcher.highlight(command),
+                    partial(screen.conversation.set_mode, mode.id),
+                    help=mode.description,
+                )
+
+    async def discover(self) -> Hits:
+        screen = self.screen
+        assert isinstance(screen, MainScreen)
+
+        for mode in sorted(
+            screen.conversation.modes.values(), key=lambda mode: mode.name
+        ):
+            yield DiscoveryHit(
+                mode.name,
+                partial(screen.conversation.set_mode, mode.id),
+                help=mode.description,
+            )
+
+
+class MainScreen(Screen, can_focus=False):
+    AUTO_FOCUS = "Conversation Prompt TextArea"
+
+    CSS_PATH = "main.tcss"
+
+    COMMANDS = {ModeProvider}
+
+    SESSION_NAVIGATION_GROUP = Binding.Group(description="Sessions")
+    BINDINGS = [
+        Binding("ctrl+b,f20", "show_sidebar", "Sidebar", priority=True),
+        Binding("ctrl+g", "go_home", "Home", priority=True),
+        Binding(
+            "ctrl+left_square_bracket",
+            "session_previous",
+            "Previous session",
+            group=SESSION_NAVIGATION_GROUP,
+        ),
+        Binding(
+            "ctrl+right_square_bracket",
+            "session_next",
+            "Next session",
+            group=SESSION_NAVIGATION_GROUP,
+        ),
+    ]
+
+    BINDING_GROUP_TITLE = "Screen"
+    busy_count = var(0)
+    throbber: getters.query_one[Throbber] = getters.query_one("#throbber")
+    conversation = getters.query_one(Conversation)
+    side_bar = getters.query_one(SideBar)
+    project_directory_tree = getters.query_one("#project_directory_tree")
+
+    column = reactive(False)
+    column_width = reactive(80)
+    scrollbar = reactive("")
+    project_path: var[Path] = var(Path("./").expanduser().absolute())
+
+    app = getters.app(A2TUIApp)
+
+    def __init__(
+        self,
+        project_path: Path,
+        agent: Agent | None = None,
+        agent_session_id: str | None = None,
+        agent_session_title: str | None = None,
+        session_pk: int | None = None,
+        initial_prompt: str | None = None,
+    ) -> None:
+        super().__init__()
+        self.set_reactive(MainScreen.project_path, project_path)
+        self._agent = agent
+        self._agent_session_id = agent_session_id
+        self._agent_session_title = agent_session_title
+        self._session_pk = session_pk
+        self._initial_prompt = initial_prompt
+
+    def watch_title(self, title: str) -> None:
+        self.app.update_terminal_title()
+
+    def get_loading_widget(self) -> Widget:
+        throbber = self.app.settings.get("ui.throbber", str)
+        if throbber == "quotes":
+            from tui.app import QUOTES
+            from tui.widgets.future_text import FutureText
+
+            quotes = QUOTES.copy()
+            random.shuffle(quotes)
+            return FutureText([Content(quote) for quote in quotes])
+        return super().get_loading_widget()
+
+    @on(ScreenResume)
+    def on_screen_resume(self, event: ScreenResume) -> None:
+        self.conversation.update_title()
+
+    def compose(self) -> ComposeResult:
+        panels: list[SideBar.Panel] = [
+            SideBar.Panel(
+                "Plan",
+                Plan([], placeholder="no plan yet"),
+            ),
+            SideBar.Panel(
+                "Context",
+                ContextStats(id="context-stats"),
+                collapsed=False,
+                id="context-panel",
+            ),
+        ]
+        cdh_dir = CdhProjectLoader.find_cdh_dir(self.project_path)
+        if cdh_dir is not None:
+            panels.append(
+                SideBar.Panel(
+                    "Project",
+                    ProjectDirectoryTree(
+                        self.project_path,
+                        id="project_directory_tree",
+                    ),
+                    flex=True,
+                    id="project-panel",
+                ),
+            )
+        else:
+            panels.append(
+                SideBar.Panel(
+                    "No Project",
+                    Static(
+                        "[bold]No .cdh project[/]\n\nClick to initialize .cdh in this directory",
+                        id="init-cdh-hint",
+                    ),
+                    id="no-project-panel",
+                ),
+            )
+        panels.append(
+            SideBar.Panel(
+                "Modified Files",
+                ModifiedFiles(self.project_path, id="modified_files"),
+                collapsed=False,
+                id="modified-files-panel",
+            ),
+        )
+        with containers.Center():
+            yield SideBar(*panels)
+            yield Conversation(
+                self.project_path,
+                self._agent,
+                self._agent_session_id,
+                self._session_pk,
+                initial_prompt=self._initial_prompt,
+            ).data_bind(
+                project_path=MainScreen.project_path,
+                column=MainScreen.column,
+            )
+        yield Footer()
+
+    def run_prompt(self, prompt: str) -> None:
+        self.conversation
+
+    def update_node_styles(self, animate: bool = True) -> None:
+        self.conversation.update_node_styles(animate=animate)
+        self.query_one(Footer).update_node_styles(animate=animate)
+        self.query_one(SideBar).update_node_styles(animate=animate)
+
+    def action_session_previous(self) -> None:
+        if self.screen.id is not None:
+            self.post_message(messages.SessionNavigate(self.screen.id, -1))
+
+    def action_session_next(self) -> None:
+        if self.screen.id is not None:
+            self.post_message(messages.SessionNavigate(self.screen.id, +1))
+
+    @on(messages.ProjectDirectoryUpdated)
+    async def on_project_directory_update(self) -> None:
+        self.project_path = self.app.project_dir
+        sidebar = self.query_one(SideBar)
+        cdh_dir = CdhProjectLoader.find_cdh_dir(self.project_path)
+        if cdh_dir is not None:
+            sidebar.remove_panel("no-project-panel")
+            if not sidebar.has_panel("project-panel"):
+                collapsible = await sidebar.add_panel(
+                    SideBar.Panel(
+                        "Project",
+                        ProjectDirectoryTree(
+                            self.project_path,
+                            id="project_directory_tree",
+                        ),
+                        flex=True,
+                        id="project-panel",
+                    ),
+                )
+                tree = collapsible.query_one(ProjectDirectoryTree)
+                tree.data_bind(path=MainScreen.project_path)
+                tree.guide_depth = 3
+            else:
+                tree = sidebar.query_one(ProjectDirectoryTree)
+                tree.path = self.project_path
+                await tree.reload()
+        else:
+            sidebar.remove_panel("project-panel")
+            if not sidebar.has_panel("no-project-panel"):
+                await sidebar.add_panel(
+                    SideBar.Panel(
+                        "No Project",
+                        Static(
+                            "[bold]No .cdh project[/]\n\nClick to initialize .cdh in this directory",
+                            id="init-cdh-hint",
+                        ),
+                        id="no-project-panel",
+                    ),
+                )
+        if mf := sidebar.query_one_optional("#modified_files", ModifiedFiles):
+            mf.refresh_files()
+
+    @on(Click, "#init-cdh-hint")
+    def on_init_cdh_click(self) -> None:
+        from pathlib import Path
+        from onecode.config_screen import EditFieldScreen
+        from onecode.agent.cdh_loader import CdhProjectLoader
+
+        default_path = str(self.project_path.resolve())
+        self.app.push_screen(
+            EditFieldScreen("Directory to initialize .cdh", default_path),
+            self._on_init_cdh_path,
+        )
+
+    def _on_init_cdh_path(self, path_str: str | None) -> None:
+        from pathlib import Path
+        from onecode.agent.cdh_loader import CdhProjectLoader
+
+        if not path_str:
+            return
+        try:
+            target = Path(path_str).expanduser().resolve()
+            if not target.is_dir():
+                self.notify(f"Not a directory: {target}", severity="error")
+                return
+        except Exception:
+            self.notify("Invalid path", severity="error")
+            return
+        existing = CdhProjectLoader.find_cdh_dir(target)
+        if existing is not None:
+            self.notify(f".cdh already exists at {existing}", severity="warning")
+            return
+        name = target.name
+        from cdh.scaffold import scaffold_dlc_project
+        scaffold_dlc_project(target, name)
+        CdhProjectLoader.init_project(target, name)
+        self.notify(f"Initialized .cdh in {target}")
+        self.post_message(messages.ProjectDirectoryUpdated())
+
+    @on(DirectoryTree.FileSelected, "ProjectDirectoryTree")
+    def on_project_directory_tree_selected(self, event: Tree.NodeSelected):
+        if (data := event.node.data) is not None:
+            self.conversation.insert_path_into_prompt(data.path)
+
+    @on(acp_messages.Plan)
+    async def on_acp_plan(self, message: acp_messages.Plan):
+        message.stop()
+        entries = [
+            Plan.Entry(
+                Content(entry["content"]),
+                entry.get("priority", "medium"),
+                entry.get("status", "pending"),
+            )
+            for entry in message.entries
+        ]
+        self.query_one("SideBar Plan", Plan).entries = entries
+
+    @on(messages.SessionUpdate)
+    async def on_session_update(self, event: messages.SessionUpdate) -> None:
+        if event.name is not None:
+            self._agent_session_title = event.name
+            self.conversation._agent_session_title = event.name
+        if self.id is not None:
+            self.app.session_tracker.update_session(
+                self.id,
+                title=event.name,
+                subtitle=event.subtitle,
+                path=event.path,
+                state=event.state,
+            )
+        if event.name is not None:
+            self.conversation.update_title()
+
+    @on(messages.SessionClose)
+    async def on_session_close(self, event: messages.SessionClose) -> None:
+
+        if self.id is None:
+            return
+        current_mode = self.id
+        session_tracker = self.app.session_tracker
+
+        if session_tracker.session_count <= 1:
+            session_tracker.close_session(current_mode)
+            self.app.exit()
+            return
+
+        next_mode = session_tracker.session_cursor_move(current_mode, +1)
+        session_tracker.close_session(current_mode)
+        if next_mode:
+            self.app.switch_mode(next_mode)
+            # Directly remove the closed tab from the visible screen's SessionsTabs
+            if sessions_tabs := self.app.screen.query_one_optional(
+                SessionsTabs
+            ):
+                if current_tab := sessions_tabs.query_one_optional(
+                    f"#{current_mode}", SessionLabel
+                ):
+                    await current_tab.remove()
+        self.app.call_later(self.app.remove_mode, current_mode)
+
+    def on_mount(self) -> None:
+        import gc
+
+        gc.freeze()
+        if self.query_one(SideBar).has_panel("project-panel"):
+            for tree in self.query("#project_directory_tree").results(DirectoryTree):
+                tree.data_bind(path=MainScreen.project_path)
+            for tree in self.query(DirectoryTree):
+                tree.guide_depth = 3
+        mf = self.query_one("#modified_files", ModifiedFiles)
+        mf.data_bind(path=MainScreen.project_path)
+
+    @on(OptionList.OptionHighlighted)
+    def on_option_list_option_highlighted(
+        self, event: OptionList.OptionHighlighted
+    ) -> None:
+        if event.option.id is not None:
+            self.conversation.prompt.suggest(event.option.id)
+
+    def check_action(self, action: str, parameters: tuple[object, ...]) -> bool | None:
+        if action == "show_sidebar" and self.side_bar.has_focus_within:
+            return False
+        return True
+
+    def action_show_sidebar(self) -> None:
+        self.side_bar.query_one("Collapsible CollapsibleTitle").focus()
+
+    def action_focus_prompt(self) -> None:
+        self.conversation.focus_prompt()
+
+    async def action_go_home(self) -> None:
+        self.app.push_screen("store")
+
+    @on(SideBar.Dismiss)
+    def on_side_bar_dismiss(self, message: SideBar.Dismiss):
+        message.stop()
+        self.conversation.focus_prompt()
+
+    def watch_column(self, column: bool) -> None:
+        self.conversation.styles.max_width = (
+            f"{max(40, self.column_width)}%" if column else None
+        )
+
+    def watch_column_width(self, column_width: int) -> None:
+        self.conversation.styles.max_width = (
+            f"{max(40, column_width)}%" if self.column else None
+        )
+
+    def watch_scrollbar(self, old_scrollbar: str, scrollbar: str) -> None:
+        if old_scrollbar:
+            self.conversation.remove_class(f"-scrollbar-{old_scrollbar}")
+        if scrollbar:
+            self.conversation.add_class(f"-scrollbar-{scrollbar}")