"""TUI 界面模块 - Textual TUI 实现""" from .constants import TEXTUAL_AVAILABLE if TEXTUAL_AVAILABLE: import re import asyncio from textual.app import App, ComposeResult from textual.containers import Horizontal, Vertical, VerticalScroll, Container from textual.widgets import ( Header, Footer, Input, Static, Rule, Label, DataTable, ProgressBar, TabbedContent, TabPane ) from textual.reactive import reactive, var from textual.message import Message from textual.binding import Binding from textual import work, on from rich.text import Text from rich.markdown import Markdown as RichMarkdown from rich.syntax import Syntax from rich.panel import Panel from rich.table import Table from rich.tree import Tree as RichTree from rich.box import ROUNDED, HEAVY, DOUBLE # ── TUI 主题色 ── class _Theme: BG = "#0d1117" BG_LIGHT = "#161b22" BG_INPUT = "#0d1117" BORDER = "#30363d" BORDER_FOCUS = "#58a6ff" TEXT = "#c9d1d9" TEXT_DIM = "#484f58" TEXT_MUTED = "#8b949e" ACCENT = "#58a6ff" SUCCESS = "#3fb950" WARNING = "#d29922" ERROR = "#f85149" USER = "#79c0ff" AI = "#c9d1d9" TOOL = "#56d364" TOOL_ERR = "#f85149" CODE_BG = "#161b22" _TUI_CSS = f""" Screen {{ background: {_Theme.BG}; }} #app-container {{ height: 100%; width: 100%; }} #main {{ width: 1fr; height: 1fr; }} #chat-scroll {{ height: 1fr; background: {_Theme.BG}; scrollbar-color: {_Theme.BORDER}; scrollbar-color-hover: {_Theme.TEXT_MUTED}; padding: 0 1; }} #input-area {{ height: auto; min-height: 4; max-height: 10; padding: 1 1 0 1; }} #user-input {{ height: auto; min-height: 3; background: {_Theme.BG_INPUT}; border: tall {_Theme.BORDER}; color: {_Theme.TEXT}; padding: 0 1; }} #user-input:focus {{ border: tall {_Theme.BORDER_FOCUS}; }} #input-hint {{ height: 1; color: {_Theme.TEXT_DIM}; padding: 0 1; text-size: 80%; }} #sidebar {{ width: 32; min-width: 32; height: 1fr; background: {_Theme.BG_LIGHT}; border-left: wide {_Theme.BORDER}; display: block; }} #sidebar.hidden {{ display: none; width: 0; min-width: 0; }} .sidebar-header {{ width: 100%; text-align: center; color: {_Theme.ACCENT}; text-style: bold; padding: 1 0 0 0; text-size: 90%; }} .sidebar-section {{ height: auto; padding: 0 1; margin: 0 0 1 0; }} .engine-item {{ padding: 0 1; color: {_Theme.TEXT_MUTED}; text-size: 85%; }} .engine-item-active {{ padding: 0 1; color: {_Theme.SUCCESS}; text-style: bold; text-size: 85%; }} .tool-item {{ padding: 0 0 0 1; color: {_Theme.TEXT_MUTED}; text-size: 80%; }} .tool-item-name {{ color: {_Theme.TEXT}; text-style: bold; }} #status-bar {{ height: 1; width: 100%; dock: bottom; background: {_Theme.BG_LIGHT}; color: {_Theme.TEXT_MUTED}; padding: 0 1; text-size: 80%; }} .msg-user {{ color: {_Theme.USER}; padding: 1 0 0 1; text-style: bold; }} .msg-ai {{ color: {_Theme.AI}; padding: 0 1; }} .msg-system {{ color: {_Theme.ACCENT}; padding: 0 1; text-size: 85%; }} .msg-tool-ok {{ color: {_Theme.TOOL}; padding: 0 1; text-size: 85%; }} .msg-tool-err {{ color: {_Theme.TOOL_ERR}; padding: 0 1; text-size: 85%; }} .msg-thinking {{ color: {_Theme.TEXT_DIM}; text-style: italic; padding: 0 1; }} .msg-error {{ color: {_Theme.ERROR}; padding: 0 1; }} .msg-dim {{ color: {_Theme.TEXT_DIM}; padding: 0 1; text-size: 85%; }} .msg-welcome {{ color: {_Theme.ACCENT}; padding: 0 1; text-style: bold; }} .code-block {{ background: {_Theme.CODE_BG}; border: wide {_Theme.BORDER}; padding: 0 1; margin: 0 2 0 2; text-size: 85%; }} Tab {{ background: {_Theme.BG}; }} Tab.-active {{ background: {_Theme.BG_LIGHT}; }} TabbedContent > Tabs {{ background: {_Theme.BG}; }} """ class _TUIBridge: """桥接 TUI ↔ AICLI""" def __init__(self, cli): self.cli = cli def engines(self) -> list: return list(self.cli.engines.keys()) def current_engine(self) -> str: return self.cli.get_current_engine_name() def tools(self) -> list: return self.cli.liugin_manager.tools def history(self) -> list: return self.cli.shared_conversation_history def switch_engine(self, name: str) -> bool: if name in self.cli.engines: self.cli.switch_engine(name) return True return False class XiaoliTUI(App): """小狸 TUI v2 - GitHub 暗色主题""" CSS = _TUI_CSS TITLE = "🐱 小狸 Pro-CLI" SUB_TITLE = "智能编程助手" BINDINGS = [ Binding("ctrl+c", "quit", "退出", show=True), Binding("ctrl+l", "clear", "清屏", show=True), Binding("ctrl+n", "new_chat", "新对话", show=True), Binding("f1", "toggle_sidebar", "侧栏", show=True), Binding("escape", "cancel", "取消", show=False), ] sidebar_visible = var(True) is_generating = var(False) def __init__(self, cli): super().__init__() self.cli = cli self.bridge = _TUIBridge(cli) def compose(self) -> ComposeResult: yield Header(show_clock=True) with Horizontal(id="app-container"): with Vertical(id="main"): yield VerticalScroll(id="chat-scroll") with Vertical(id="input-area"): yield Input( placeholder=" 输入消息... (Enter 发送, /help 帮助)", id="user-input" ) yield Static( " Tab 补全 | ↑↓ 历史 | Ctrl+L 清屏 | F1 侧栏", id="input-hint" ) with Vertical(id="sidebar"): yield Static("⚙️ 引擎", classes="sidebar-header") yield Vertical(id="engine-list", classes="sidebar-section") yield Rule(line_style="heavy") yield Static("🔧 工具", classes="sidebar-header") yield Vertical(id="tool-list", classes="sidebar-section") yield Rule(line_style="heavy") yield Static("📊 状态", classes="sidebar-header") yield Vertical(id="status-info", classes="sidebar-section") yield Static(" 就绪 | Ctrl+C 退出", id="status-bar") def on_mount(self): self._render_welcome() self._update_sidebar() self.query_one("#user-input").focus() def _render_welcome(self): scroll = self.query_one("#chat-scroll") lines = [ ("", "msg-dim"), (" ╔══════════════════════════════════════════════╗", "msg-welcome"), (" ║ ║", "msg-welcome"), (" ║ 🐱 小狸 Pro-CLI v5.1 ║", "msg-welcome"), (" ║ 智能编程助手 · 对标 Claude Code ║", "msg-welcome"), (" ║ ║", "msg-welcome"), (" ╚══════════════════════════════════════════════╝", "msg-welcome"), ("", "msg-dim"), (" 💡 代码编辑 · 代码搜索 · Git 集成 · 多引擎", "msg-system"), (" 📝 输入 /help 查看命令 | /model 切换引擎", "msg-system"), (f" 🔧 当前引擎: {self.bridge.current_engine()} | 工具: {len(self.bridge.tools())} 个", "msg-system"), ("", "msg-dim"), ] for text, cls in lines: scroll.mount(Static(text, classes=cls)) def _update_sidebar(self): engine_container = self.query_one("#engine-list") engine_container.remove_children() current = self.bridge.current_engine() for name in self.bridge.engines(): if name == current: engine_container.mount(Static(f" ▸ {name}", classes="engine-item-active")) else: engine_container.mount(Static(f" {name}", classes="engine-item")) tool_container = self.query_one("#tool-list") tool_container.remove_children() for tool in self.bridge.tools(): name = tool.get('name', '?') tool_container.mount(Static(f" • {name}", classes="tool-item")) status_container = self.query_one("#status-info") status_container.remove_children() status_container.mount(Static(f" 对话: {len(self.bridge.history())} 条", classes="tool-item")) status_container.mount(Static(f" 工具: {len(self.bridge.tools())} 个", classes="tool-item")) status_container.mount(Static(f" 引擎: {len(self.bridge.engines())} 个", classes="tool-item")) def _update_status(self, text: str): bar = self.query_one("#status-bar") engine = self.bridge.current_engine() bar.update(f" {text} | {engine} | {len(self.bridge.history())} 条对话") def _append(self, widget): scroll = self.query_one("#chat-scroll") scroll.mount(widget) scroll.scroll_end(animate=False) def _add(self, text, cls="msg-dim"): self._append(Static(text, classes=cls)) def _user_msg(self, text): self._add(f" 👤 {text}", "msg-user") def _ai_msg(self, text): if "```" in text: self._render_with_code(text) else: self._add(f" ✦ {text}", "msg-ai") def _render_with_code(self, text): parts = re.split(r'```(\w*)\n(.*?)```', text, flags=re.DOTALL) i = 0 while i < len(parts): if i % 3 == 0: if parts[i].strip(): for line in parts[i].strip().split('\n'): self._add(f" ✦ {line}", "msg-ai") elif i % 3 == 2: code = parts[i] lang = parts[i-1] if i > 1 else "" try: syntax = Syntax(code, lang or "python", theme="monokai", line_numbers=True, word_wrap=True) self._append(Panel(syntax, border_style=f"dim {_Theme.BORDER}", box=ROUNDED, padding=(0, 1))) except Exception: for line in code.split('\n'): self._add(f" {line}", "msg-dim") i += 1 def _tool_ok(self, name, args): self._add(f" ✅ {name}: {args[:60]}", "msg-tool-ok") def _tool_err(self, name, args): self._add(f" ❌ {name}: {args[:60]}", "msg-tool-err") def _thinking(self): self._add(" 💭 思考中...", "msg-thinking") def _system(self, text): self._add(f" ℹ️ {text}", "msg-system") def _error(self, text): self._add(f" ⚠️ {text}", "msg-error") @on(Input.Submitted, "#user-input") def on_input(self, event): text = event.value.strip() if not text: return event.input.value = "" if text.startswith('/'): self._handle_command(text) return self._user_msg(text) self._thinking() self.is_generating = True self._update_status("🔄 思考中...") self.run_worker(self._generate(text), exclusive=True) def _handle_command(self, cmd): parts = cmd[1:].split(maxsplit=1) name = parts[0].lower() args = parts[1] if len(parts) > 1 else "" cmds = { 'help': lambda: self._show_help(), 'quit': lambda: self.exit(), 'q': lambda: self.exit(), 'exit': lambda: self.exit(), 'cli': lambda: self._switch_cli(), 'clear': lambda: self.action_clear(), 'cls': lambda: self.action_clear(), 'model': lambda: self._switch_model(args), 'engine': lambda: self._switch_model(args), 'about': lambda: self._system("🐱 小狸 Pro-CLI v5.1 - 智能编程助手"), 'status': lambda: self._show_status(), 'tools': lambda: self._show_tools(), 'engines': lambda: self._show_engines(), 'tui': lambda: self._system("已在 TUI 模式中"), } handler = cmds.get(name) if handler: handler() else: if name in self.cli.liugin_commands: try: result = self.cli.liugin_commands[name](args) if result: self._system(result[:500]) except Exception as e: self._error(f"插件命令失败: {e}") else: self._error(f"未知命令: /{name}，输入 /help 查看帮助") def _show_help(self): help_text = """ 📖 命令: /help 帮助信息 /quit 退出 /cli 切换命令行模式 /model <引擎> 切换 AI 引擎 /engines 列出引擎 /tools 列出工具 /status 系统状态 /clear 清屏 ⌨️ 快捷键: Ctrl+C 退出 Ctrl+L 清屏 Ctrl+N 新对话 F1 侧栏 Escape 取消生成""" self._system(help_text) def _switch_cli(self): self._system("切换到命令行模式...") self.cli.tui_output_callback = None self.exit() def _switch_model(self, name): name = name.strip() if not name: self._show_engines() return if self.bridge.switch_engine(name): self._system(f"已切换到: {name}") self._update_sidebar() else: self._error(f"未找到引擎: {name}") def _show_engines(self): current = self.bridge.current_engine() lines = [" 可用引擎:"] for name in self.bridge.engines(): marker = "▸" if name == current else " " lines.append(f" {marker} {name}") self._system('\n'.join(lines)) def _show_tools(self): tools = self.bridge.tools() lines = [f" 可用工具 ({len(tools)} 个):"] for t in tools: name = t.get('name', '?') desc = t.get('description', '')[:40] lines.append(f" • {name}: {desc}") self._system('\n'.join(lines)) def _show_status(self): self._system(f""" 系统状态: 引擎: {self.bridge.current_engine()} 引擎数: {len(self.bridge.engines())} 工具数: {len(self.bridge.tools())} 对话数: {len(self.bridge.history())}""") async def _generate(self, user_input): try: self.call_after_refresh(self._remove_thinking) def tui_output(msg): self.call_after_refresh(self._write_raw, msg) original = self.cli.tui_output_callback self.cli.tui_output_callback = tui_output try: loop = asyncio.get_running_loop() await loop.run_in_executor( None, self.cli.process_conversation, user_input ) finally: self.cli.tui_output_callback = original except Exception as e: self.call_after_refresh(self._error, str(e)) finally: self.is_generating = False self.call_after_refresh(lambda: self._update_status("就绪")) self.call_after_refresh(self._update_sidebar) def _remove_thinking(self): try: scroll = self.query_one("#chat-scroll") for child in reversed(list(scroll.children)): renderable = getattr(child, 'renderable', None) text = str(renderable) if renderable is not None else "" if '思考中' in text: child.remove() break except Exception: pass def _write_raw(self, msg): if '✅' in msg or 'OK 工具' in msg: self._add(f" {msg}", "msg-tool-ok") elif '❌' in msg or 'X 工具' in msg or '错误' in msg: self._add(f" {msg}", "msg-tool-err") elif '✦' in msg: self._add(f" {msg}", "msg-ai") elif '工具' in msg and ('调用' in msg or '执行' in msg): self._add(f" {msg}", "msg-tool-ok") else: self._add(f" {msg}", "msg-dim") def action_clear(self): scroll = self.query_one("#chat-scroll") scroll.remove_children() self._render_welcome() def action_new_chat(self): self.cli.shared_conversation_history.clear() self.action_clear() self._system("已开始新对话") def action_toggle_sidebar(self): sidebar = self.query_one("#sidebar") sidebar.visible = not sidebar.visible def action_cancel(self): if self.is_generating: self._system("已取消") self.is_generating = False else: # Textual 不可用时的占位类 class XiaoliTUI: """Textual 未安装时的占位类""" def __init__(self, *args, **kwargs): pass def run(self, *args, **kwargs): print("TUI 模式不可用：Textual 库未安装") print("请运行: pip install textual rich")