shuiyu67andWorkBuddy
749fb3f91c
- 会话数据改用 SQLite（chat_history/conversations.db，WAL）：sessions / snapshots 两表， 首次启动自动迁移旧 JSON，/chat 与 /resume 改走 SQLite，memory 搜索兼容数据库 - 去掉流式显示（_typeprint 改整段打印）与常驻加载动画（移除 show_animation 线程及逐帧输出） - 展示给用户的消息过长自动截断（仅展示层，底层 messages 完整保存） - 工具调用解析加固：_loads_json 四级兜底修复；新增 _extract_last_action_json 成对错括号扫描； 续轮也下发 Function Calling 工具，减少长文本格式错误浪费 token - 版本号统一更新至 8.0.4（README/about.txt/constants/TUI banner/LSP·MCP clientInfo） - 12 项 session 单测 + 8 项 JSON 提取边界用例全部通过 Co-Authored-By: WorkBuddy <workbuddy@tencent.com>
1607 lines
58 KiB
Python
1607 lines
58 KiB
Python
"""TUI 界面模块 - Textual TUI 实现 (动画增强版 + 多行输入)"""
|
||
|
||
from .constants import TEXTUAL_AVAILABLE, VERSION
|
||
|
||
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, Static, Rule, TextArea
|
||
from textual.reactive import var
|
||
from textual.binding import Binding
|
||
from textual import on
|
||
from .vim_keys import VimInputState
|
||
import os
|
||
from rich.syntax import Syntax
|
||
from rich.panel import Panel
|
||
from rich.box import ROUNDED
|
||
|
||
# ── 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"
|
||
# 动画额外色
|
||
GLOW = "#1f6feb"
|
||
THINKING = "#8b949e"
|
||
WELCOME_ACCENT = "#58a6ff"
|
||
|
||
_TUI_CSS = f"""
|
||
Screen {{
|
||
background: {_Theme.BG};
|
||
/* 全局过渡 */
|
||
transition: background 300ms in_out_cubic;
|
||
}}
|
||
|
||
#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: 5;
|
||
max-height: 14;
|
||
padding: 0 1 1 1;
|
||
}}
|
||
|
||
#input-wrapper {{
|
||
height: auto;
|
||
min-height: 4;
|
||
max-height: 12;
|
||
background: {_Theme.BG_INPUT};
|
||
border: tall {_Theme.BORDER};
|
||
padding: 0;
|
||
/* 聚焦时边框颜色过渡 */
|
||
transition: border-color 300ms in_out_cubic;
|
||
}}
|
||
|
||
#input-wrapper:focus-within {{
|
||
border: tall {_Theme.BORDER_FOCUS};
|
||
background: {_Theme.BG};
|
||
}}
|
||
|
||
#user-input {{
|
||
height: auto;
|
||
min-height: 3;
|
||
max-height: 10;
|
||
background: transparent;
|
||
color: {_Theme.TEXT};
|
||
padding: 0 1;
|
||
border: none;
|
||
}}
|
||
|
||
#user-input:focus {{
|
||
background: transparent;
|
||
}}
|
||
|
||
/* TextArea 内部样式覆盖 */
|
||
#user-input .text-area--cursor {{
|
||
color: {_Theme.ACCENT};
|
||
}}
|
||
|
||
#user-input .text-area--cursor-line {{
|
||
background: {_Theme.BG_LIGHT} 50%;
|
||
}}
|
||
|
||
#user-input .text-area--selection {{
|
||
background: {_Theme.GLOW} 40%;
|
||
}}
|
||
|
||
#input-hint {{
|
||
height: 1;
|
||
color: {_Theme.TEXT_DIM};
|
||
padding: 0 1;
|
||
/* 渐入效果 */
|
||
opacity: 0;
|
||
transition: opacity 500ms in_out_cubic;
|
||
}}
|
||
|
||
#input-hint.visible {{
|
||
opacity: 1;
|
||
}}
|
||
|
||
/* ── 侧栏 ── */
|
||
#sidebar {{
|
||
width: 32;
|
||
min-width: 32;
|
||
height: 1fr;
|
||
background: {_Theme.BG_LIGHT};
|
||
border-left: wide {_Theme.BORDER};
|
||
display: block;
|
||
/* 侧栏滑入/滑出动画 */
|
||
transition: width 300ms in_out_cubic, opacity 300ms in_out_cubic;
|
||
overflow: hidden;
|
||
}}
|
||
|
||
#sidebar.hidden {{
|
||
width: 0;
|
||
min-width: 0;
|
||
opacity: 0;
|
||
border: none;
|
||
overflow: hidden;
|
||
}}
|
||
|
||
.sidebar-header {{
|
||
width: 100%;
|
||
text-align: center;
|
||
color: {_Theme.ACCENT};
|
||
text-style: bold;
|
||
padding: 1 0 0 0;
|
||
}}
|
||
|
||
.sidebar-section {{
|
||
height: auto;
|
||
padding: 0 1;
|
||
margin: 0 0 1 0;
|
||
}}
|
||
|
||
.engine-item {{
|
||
padding: 0 1;
|
||
color: {_Theme.TEXT_MUTED};
|
||
}}
|
||
|
||
.engine-item-active {{
|
||
padding: 0 1;
|
||
color: {_Theme.SUCCESS};
|
||
text-style: bold;
|
||
}}
|
||
|
||
.tool-item {{
|
||
padding: 0 0 0 1;
|
||
color: {_Theme.TEXT_MUTED};
|
||
}}
|
||
|
||
.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;
|
||
/* 状态变化过渡 */
|
||
transition: color 300ms in_out_cubic;
|
||
}}
|
||
|
||
/* ── 文件树 / 会话列表 ── */
|
||
.file-tree-item {{
|
||
padding: 0 0 0 1;
|
||
color: {_Theme.TEXT_MUTED};
|
||
}}
|
||
.file-tree-dir {{
|
||
color: {_Theme.ACCENT};
|
||
text-style: bold;
|
||
}}
|
||
.session-item {{
|
||
padding: 0 0 0 1;
|
||
color: {_Theme.TEXT_MUTED};
|
||
}}
|
||
.session-item-active {{
|
||
color: {_Theme.SUCCESS};
|
||
text-style: bold;
|
||
}}
|
||
.plan-badge {{
|
||
color: {_Theme.WARNING};
|
||
text-style: bold;
|
||
}}
|
||
|
||
/* ── 消息样式 ── */
|
||
.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;
|
||
}}
|
||
|
||
.msg-tool-ok {{
|
||
color: {_Theme.TOOL};
|
||
padding: 0 1;
|
||
}}
|
||
|
||
.msg-tool-err {{
|
||
color: {_Theme.TOOL_ERR};
|
||
padding: 0 1;
|
||
}}
|
||
|
||
.msg-thinking {{
|
||
color: {_Theme.THINKING};
|
||
text-style: italic;
|
||
padding: 0 1;
|
||
}}
|
||
|
||
.msg-error {{
|
||
color: {_Theme.ERROR};
|
||
padding: 0 1;
|
||
}}
|
||
|
||
.msg-dim {{
|
||
color: {_Theme.TEXT_DIM};
|
||
padding: 0 1;
|
||
}}
|
||
|
||
.msg-welcome {{
|
||
color: {_Theme.WELCOME_ACCENT};
|
||
padding: 0 1;
|
||
text-style: bold;
|
||
}}
|
||
|
||
.msg-welcome-line {{
|
||
color: {_Theme.WELCOME_ACCENT};
|
||
padding: 0 1;
|
||
text-style: bold;
|
||
/* 每行依次淡入 */
|
||
opacity: 0;
|
||
transition: opacity 400ms in_out_cubic;
|
||
}}
|
||
|
||
.msg-welcome-line.visible {{
|
||
opacity: 1;
|
||
}}
|
||
|
||
.msg-system-line {{
|
||
color: {_Theme.ACCENT};
|
||
padding: 0 1;
|
||
opacity: 0;
|
||
transition: opacity 400ms in_out_cubic;
|
||
}}
|
||
|
||
.msg-system-line.visible {{
|
||
opacity: 1;
|
||
}}
|
||
|
||
/* 动画思考指示器 */
|
||
.thinking-indicator {{
|
||
color: {_Theme.THINKING};
|
||
text-style: italic;
|
||
padding: 0 1;
|
||
}}
|
||
|
||
.code-block {{
|
||
background: {_Theme.CODE_BG};
|
||
border: wide {_Theme.BORDER};
|
||
padding: 0 1;
|
||
margin: 0 2 0 2;
|
||
}}
|
||
|
||
/* ── Tab 样式 ── */
|
||
Tab {{
|
||
background: {_Theme.BG};
|
||
}}
|
||
|
||
Tab.-active {{
|
||
background: {_Theme.BG_LIGHT};
|
||
}}
|
||
|
||
TabbedContent > Tabs {{
|
||
background: {_Theme.BG};
|
||
}}
|
||
|
||
/* ── 自定义顶栏 (opencode 风格) ── */
|
||
#tui-header {{
|
||
height: 1;
|
||
width: 100%;
|
||
dock: top;
|
||
background: {_Theme.BG_LIGHT};
|
||
color: {_Theme.TEXT_MUTED};
|
||
padding: 0 1;
|
||
border-bottom: solid {_Theme.BORDER};
|
||
}}
|
||
#tui-header .hdr-left {{ color: {_Theme.ACCENT}; text-style: bold; }}
|
||
#tui-header .hdr-mid {{ color: {_Theme.TEXT_MUTED}; }}
|
||
#tui-header .hdr-right {{ color: {_Theme.SUCCESS}; }}
|
||
|
||
/* ── 侧栏标签页 ── */
|
||
#sidebar-tabs {{
|
||
height: 1;
|
||
width: 100%;
|
||
dock: top;
|
||
background: {_Theme.BG};
|
||
padding: 0;
|
||
}}
|
||
.sidebar-tab {{
|
||
display: block;
|
||
height: 1;
|
||
padding: 0 1;
|
||
color: {_Theme.TEXT_DIM};
|
||
text-style: bold;
|
||
}}
|
||
.sidebar-tab:hover {{
|
||
color: {_Theme.TEXT_MUTED};
|
||
}}
|
||
.sidebar-tab-active {{
|
||
color: {_Theme.ACCENT};
|
||
border-bottom: solid {_Theme.ACCENT};
|
||
}}
|
||
|
||
/* ── 侧栏内容面板 ── */
|
||
#sidebar-content {{
|
||
height: 1fr;
|
||
padding: 0 1;
|
||
}}
|
||
.panel-section {{
|
||
padding: 1 0 0 0;
|
||
border-bottom: solid {_Theme.BORDER};
|
||
}}
|
||
.panel-title {{
|
||
color: {_Theme.TEXT_DIM};
|
||
padding: 0 0 1 0;
|
||
}}
|
||
|
||
/* ── 消息角色色条 (opencode 风格) ── */
|
||
.msg-user {{
|
||
color: {_Theme.USER};
|
||
padding: 0 0 0 2;
|
||
border-left: solid {_Theme.USER};
|
||
}}
|
||
.msg-ai {{
|
||
color: {_Theme.AI};
|
||
padding: 0 0 0 2;
|
||
border-left: solid {_Theme.GLOW};
|
||
}}
|
||
.msg-tool-ok {{
|
||
color: {_Theme.TOOL};
|
||
padding: 0 0 0 2;
|
||
border-left: solid {_Theme.TOOL};
|
||
}}
|
||
.msg-tool-err {{
|
||
color: {_Theme.TOOL_ERR};
|
||
padding: 0 0 0 2;
|
||
border-left: solid {_Theme.TOOL_ERR};
|
||
}}
|
||
.msg-system {{
|
||
color: {_Theme.ACCENT};
|
||
padding: 0 0 0 2;
|
||
border-left: solid {_Theme.ACCENT};
|
||
}}
|
||
.msg-error {{
|
||
color: {_Theme.ERROR};
|
||
padding: 0 0 0 2;
|
||
border-left: solid {_Theme.ERROR};
|
||
}}
|
||
|
||
/* ── 精简欢迎框 ── */
|
||
.welcome-compact {{
|
||
color: {_Theme.WELCOME_ACCENT};
|
||
padding: 1 2;
|
||
border-bottom: solid {_Theme.BORDER};
|
||
text-style: bold;
|
||
}}
|
||
|
||
"""
|
||
|
||
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
|
||
|
||
def sessions(self) -> list:
|
||
mgr = getattr(self.cli, 'session_manager', None)
|
||
if mgr and hasattr(mgr, 'list_sessions'):
|
||
try:
|
||
return mgr.list_sessions()
|
||
except Exception:
|
||
return []
|
||
return []
|
||
|
||
def plan_mode(self) -> bool:
|
||
return bool(getattr(self.cli, 'plan_mode', False))
|
||
|
||
def current_plan(self) -> str:
|
||
return getattr(self.cli, 'current_plan', '') or ""
|
||
|
||
def cwd(self) -> str:
|
||
return os.getcwd()
|
||
|
||
|
||
# ═══════════════════════════════════════════════════
|
||
# 动画辅助组件
|
||
# ═══════════════════════════════════════════════════
|
||
|
||
class ThinkingSpinner(Static):
|
||
"""动态思考指示器 - 旋转 + 脉冲"""
|
||
FRAMES = ["⠋", "⠙", "⠹", "⠸", "⠼", "⠴", "⠦", "⠧", "⠇", "⠏"]
|
||
DOTS_FRAMES = [" ", ". ", ".. ", "..."]
|
||
|
||
def __init__(self, **kwargs):
|
||
super().__init__("", **kwargs)
|
||
self._frame = 0
|
||
self._dot_frame = 0
|
||
self._running = False
|
||
|
||
def on_mount(self):
|
||
self._running = True
|
||
self.set_interval(0.08, self._animate)
|
||
|
||
def _animate(self):
|
||
if not self._running:
|
||
return
|
||
spinner = self.FRAMES[self._frame % len(self.FRAMES)]
|
||
dots = self.DOTS_FRAMES[self._dot_frame % len(self.DOTS_FRAMES)]
|
||
self.update(f" {spinner} 思考中{dots}")
|
||
self._frame += 1
|
||
if self._frame % 3 == 0:
|
||
self._dot_frame += 1
|
||
|
||
def stop(self):
|
||
self._running = False
|
||
|
||
|
||
class AnimatedWelcome(Static):
|
||
"""逐行动画显示的欢迎组件"""
|
||
def __init__(self, lines, **kwargs):
|
||
super().__init__("", **kwargs)
|
||
self._lines = lines
|
||
self._current = 0
|
||
|
||
def on_mount(self):
|
||
self.set_interval(0.12, self._show_next_line)
|
||
|
||
def _show_next_line(self):
|
||
if self._current < len(self._lines):
|
||
current_text = self.renderable
|
||
new_line = self._lines[self._current]
|
||
if self._current == 0:
|
||
self.update(new_line)
|
||
else:
|
||
self.update(str(current_text) + "\n" + new_line)
|
||
self._current += 1
|
||
|
||
|
||
class MessageBubble(Static):
|
||
"""带入场动画的消息气泡"""
|
||
def __init__(self, text, **kwargs):
|
||
super().__init__(text, **kwargs)
|
||
self.styles.opacity = 0
|
||
|
||
def on_mount(self):
|
||
# textual >= 8: Widget.opacity 是只读属性，直接 animate 会抛
|
||
# "property 'opacity' has no setter"。改为在 styles 上动画，失败则直接到位。
|
||
try:
|
||
self.styles.animate("opacity", value=1.0, duration=0.3, easing="out_cubic")
|
||
except Exception:
|
||
self.styles.opacity = 1.0
|
||
|
||
|
||
class TypingIndicator(Static):
|
||
"""打字指示器 - 模拟 AI 正在打字"""
|
||
FRAMES = ["●○○", "○●○", "○○●", "○●○"]
|
||
|
||
def __init__(self, **kwargs):
|
||
super().__init__("", **kwargs)
|
||
self._frame = 0
|
||
self._running = False
|
||
|
||
def on_mount(self):
|
||
self._running = True
|
||
self.set_interval(0.3, self._tick)
|
||
|
||
def _tick(self):
|
||
if not self._running:
|
||
return
|
||
dots = self.FRAMES[self._frame % len(self.FRAMES)]
|
||
self.update(f" {dots} AI 正在输入")
|
||
self._frame += 1
|
||
|
||
def stop(self):
|
||
self._running = False
|
||
|
||
|
||
# ═══════════════════════════════════════════════════
|
||
# 自定义 TextArea：回车发送，Shift+回车换行
|
||
# ═══════════════════════════════════════════════════
|
||
|
||
class SendTextArea(TextArea):
|
||
"""回车发送消息，Shift+回车换行的输入框"""
|
||
async def _on_key(self, event):
|
||
if event.key in ("enter", "\r", "\n"):
|
||
# prevent_default 阻止父类 TextArea._on_key 执行
|
||
event.prevent_default()
|
||
try:
|
||
self.app.action_send_message()
|
||
except Exception:
|
||
pass
|
||
return
|
||
# 其他键正常处理
|
||
|
||
|
||
# ═══════════════════════════════════════════════════
|
||
# Vim 风格模态输入框（NORMAL / INSERT）
|
||
# ═══════════════════════════════════════════════════
|
||
|
||
class VimSendTextArea(SendTextArea):
|
||
"""
|
||
带 Vim 模态编辑的输入框。
|
||
|
||
模式：
|
||
INSERT —— 默认，正常打字；Esc 进入 NORMAL
|
||
NORMAL —— h/j/k/l 移动，i/a/o 进入插入，dd 删行，x 删字，
|
||
w/b 跳词，0/$ 行首行尾，gg/G 文首文尾，u 撤销 …
|
||
|
||
坐标：VimInputState 用字符偏移 offset，这里负责与 TextArea 的
|
||
(row, col) 光标互相转换。所有 vim 逻辑都在 vim_keys.py（纯函数，
|
||
可单测）；本类只做「按键映射 + 把结果写回 TextArea」。
|
||
"""
|
||
def __init__(self, *args, **kwargs):
|
||
super().__init__(*args, **kwargs)
|
||
self.vim_enabled = True
|
||
self.vim = VimInputState(mode="INSERT")
|
||
|
||
# ── offset <-> (row, col) ──
|
||
def _cur_off(self) -> int:
|
||
r, c = self.cursor_location
|
||
text = self.text
|
||
lines = text.split("\n")
|
||
return sum(len(l) + 1 for l in lines[:r]) + c
|
||
|
||
def _set_off(self, off: int):
|
||
text = self.text
|
||
off = max(0, min(off, len(text)))
|
||
r = text.count("\n", 0, off)
|
||
line_start = text.rfind("\n", 0, off)
|
||
c = off - (line_start + 1)
|
||
self.cursor_location = (r, c)
|
||
|
||
# ── key 名字 -> vim token ──
|
||
@staticmethod
|
||
def _vim_token(key: str):
|
||
m = {
|
||
"escape": "escape", "h": "h", "l": "l", "j": "j", "k": "k",
|
||
"w": "w", "b": "b", "e": "e", "0": "0", "$": "$", "^": "^",
|
||
"g": "g", "G": "G", "d": "d", "x": "x", "u": "u",
|
||
"i": "i", "a": "a", "o": "o", "O": "O", "I": "I", "A": "A",
|
||
"left": "left", "right": "right", "up": "up", "down": "down",
|
||
"shift+i": "I", "shift+a": "A", "shift+o": "O",
|
||
"shift+g": "G", "shift+6": "^", "shift+4": "$",
|
||
}
|
||
return m.get(key)
|
||
|
||
def _notify_mode(self):
|
||
app = self.app
|
||
if hasattr(app, "_update_vim_mode_display"):
|
||
try:
|
||
app._update_vim_mode_display()
|
||
except Exception:
|
||
pass
|
||
|
||
def _apply(self, res):
|
||
try:
|
||
if res.text != self.text:
|
||
self.text = res.text
|
||
self._set_off(res.offset)
|
||
self.vim.mode = res.mode
|
||
self.vim.pending = res.pending
|
||
if res.action == "undo":
|
||
try:
|
||
self.undo()
|
||
except Exception:
|
||
pass
|
||
self._notify_mode()
|
||
except Exception:
|
||
# 任何意外都不应让输入框崩；退回默认行为
|
||
pass
|
||
|
||
async def _on_key(self, event):
|
||
# 回车：INSERT（或 vim 关闭）发送；NORMAL 下移一行
|
||
if event.key in ("enter", "\r", "\n"):
|
||
if self.vim_enabled and self.vim.mode == "NORMAL":
|
||
event.prevent_default()
|
||
res = self.vim.feed(self.text, self._cur_off(), "j")
|
||
self._apply(res)
|
||
else:
|
||
event.prevent_default()
|
||
try:
|
||
self.app.action_send_message()
|
||
except Exception:
|
||
pass
|
||
return
|
||
|
||
if not self.vim_enabled:
|
||
return # 退化为普通 TextArea
|
||
|
||
if self.vim.mode == "INSERT":
|
||
if event.key == "escape":
|
||
event.prevent_default()
|
||
res = self.vim.feed(self.text, self._cur_off(), "escape")
|
||
self._apply(res)
|
||
# 其余键交给 TextArea 默认处理（打字）
|
||
return
|
||
|
||
# ── NORMAL 模式 ──
|
||
if event.key == "shift+enter":
|
||
token = "o"
|
||
else:
|
||
token = self._vim_token(event.key)
|
||
if token is None:
|
||
# ctrl+/shift+ 组合键放回去，让 App 级绑定处理（聚焦切换等）
|
||
if event.key.startswith("ctrl+") or event.key.startswith("shift+"):
|
||
return
|
||
# 普通单键未映射：吞掉，避免 TextArea 在 NORMAL 下误编辑
|
||
event.prevent_default()
|
||
return
|
||
event.prevent_default()
|
||
res = self.vim.feed(self.text, self._cur_off(), token)
|
||
self._apply(res)
|
||
|
||
# ═══════════════════════════════════════════════════
|
||
# TUI 主应用
|
||
# ═══════════════════════════════════════════════════
|
||
|
||
class XiaoliTUI(App):
|
||
"""小狸 TUI v3 - 动画增强版 + 多行输入"""
|
||
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("ctrl+enter", "send_message", "发送", show=False),
|
||
Binding("shift+enter", "newline", "换行", show=False),
|
||
Binding("f1", "toggle_sidebar", "侧栏", show=True),
|
||
Binding("ctrl+p", "toggle_plan", "PLAN", show=True),
|
||
Binding("1", "tab_engine", "引擎", show=False),
|
||
Binding("2", "tab_files", "文件", show=False),
|
||
Binding("3", "tab_sessions", "会话", show=False),
|
||
Binding("4", "tab_status", "状态", show=False),
|
||
Binding("escape", "cancel", "取消", show=False),
|
||
# ── Vim 风格导航（输入框未聚焦时生效）──
|
||
Binding("j", "chat_down", "↓聊天", show=False),
|
||
Binding("k", "chat_up", "↑聊天", show=False),
|
||
Binding("i", "focus_insert", "插入", show=False),
|
||
Binding("ctrl+f", "chat_page_down", "翻页↓", show=True),
|
||
Binding("ctrl+b", "chat_page_up", "翻页↑", show=True),
|
||
Binding("ctrl+d", "chat_half_down", "半页↓", show=True),
|
||
Binding("ctrl+u", "chat_half_up", "半页↑", show=True),
|
||
Binding("ctrl+space", "toggle_focus", "聚焦切换", show=True),
|
||
]
|
||
|
||
sidebar_visible = var(True)
|
||
is_generating = var(False)
|
||
plan_mode = var(False)
|
||
_sidebar_tab: str = "engine" # 当前侧栏标签: engine/files/sessions/status
|
||
|
||
def __init__(self, cli):
|
||
super().__init__()
|
||
self.cli = cli
|
||
self.bridge = _TUIBridge(cli)
|
||
self._thinking_widget = None
|
||
self._typing_widget = None
|
||
|
||
def compose(self) -> ComposeResult:
|
||
# ── 自定义顶栏 (opencode 风格: 左项目名 | 中模型 | 右状态) ──
|
||
yield Static("", id="tui-header")
|
||
with Horizontal(id="app-container"):
|
||
with Vertical(id="main"):
|
||
yield VerticalScroll(id="chat-scroll")
|
||
with Vertical(id="input-area"):
|
||
with Container(id="input-wrapper"):
|
||
yield VimSendTextArea(
|
||
placeholder="输入消息... (回车发送, Shift+回车换行, Esc 进 Vim 普通模式)",
|
||
id="user-input",
|
||
soft_wrap=True,
|
||
tab_behavior="indent",
|
||
)
|
||
yield Static(
|
||
" 回车发送 | Shift+回车换行 | Ctrl+L 清屏 | F1 侧栏 | Ctrl+P PLAN",
|
||
id="input-hint"
|
||
)
|
||
# ── 侧栏：标签页 + 内容区 ──
|
||
with Vertical(id="sidebar"):
|
||
# 标签行
|
||
with Horizontal(id="sidebar-tabs"):
|
||
yield Static(" 引擎 ", classes="sidebar-tab sidebar-tab-active", id="tab-engine")
|
||
yield Static(" 文件 ", classes="sidebar-tab", id="tab-files")
|
||
yield Static(" 会话 ", classes="sidebar-tab", id="tab-sessions")
|
||
yield Static(" 状态 ", classes="sidebar-tab", id="tab-status")
|
||
# 内容区（根据当前标签切换显示）
|
||
with VerticalScroll(id="sidebar-content"):
|
||
yield Vertical(id="panel-engine", classes="panel-section")
|
||
yield Vertical(id="panel-files", classes="panel-section")
|
||
yield Vertical(id="panel-sessions", classes="panel-section")
|
||
yield Vertical(id="panel-status", classes="panel-section")
|
||
yield Static(" 就绪 | Ctrl+C 退出", id="status-bar")
|
||
|
||
def on_mount(self):
|
||
# 初始化顶栏
|
||
self._update_header()
|
||
# 初始化侧栏: 默认显示引擎面板,隐藏其余
|
||
self._switch_sidebar_tab("engine")
|
||
# 精简欢迎(一行)
|
||
self._render_welcome_compact()
|
||
# 初始化 Vim 键位开关（默认开，可被 config.vim_mode 关闭）
|
||
self._init_vim_mode()
|
||
self.query_one("#user-input").focus()
|
||
def _show_hint():
|
||
try:
|
||
self.query_one("#input-hint").add_class("visible")
|
||
except Exception:
|
||
pass
|
||
self.set_timer(1.5, _show_hint)
|
||
|
||
def _init_vim_mode(self):
|
||
try:
|
||
ta = self.query_one("#user-input")
|
||
except Exception:
|
||
return
|
||
cfg = getattr(self.cli, "config", None)
|
||
enabled = True
|
||
if isinstance(cfg, dict) and "vim_mode" in cfg:
|
||
enabled = bool(cfg["vim_mode"])
|
||
ta.vim_enabled = enabled
|
||
self._update_vim_mode_display()
|
||
|
||
# ── 欢迎界面 ──
|
||
|
||
def _render_welcome_compact(self):
|
||
"""opencode 风格: 精简一行欢迎"""
|
||
scroll = self.query_one("#chat-scroll")
|
||
engine = self.bridge.current_engine()
|
||
welcome = Static(
|
||
f" ▸ 小狸 Pro-CLI v8.0 │ {engine} │ "
|
||
f"{len(self.bridge.tools())} 工具 │ /help 查看命令",
|
||
classes="welcome-compact"
|
||
)
|
||
scroll.mount(welcome)
|
||
|
||
def _render_welcome_animated(self):
|
||
"""逐行动画显示欢迎界面"""
|
||
scroll = self.query_one("#chat-scroll")
|
||
|
||
welcome_lines = [
|
||
"",
|
||
" ╔══════════════════════════════════════════════╗",
|
||
" ║ ║",
|
||
" ║ 小狸 Pro-CLI v8.0.4 ║",
|
||
" ║ 智能编程助手 · 动画增强版 ║",
|
||
" ║ ║",
|
||
" ╚══════════════════════════════════════════════╝",
|
||
"",
|
||
" 代码编辑 · 代码搜索 · Git 集成 · 多引擎",
|
||
" 输入 /help 查看命令 | /model 切换引擎",
|
||
f" 当前引擎: {self.bridge.current_engine()} | 工具: {len(self.bridge.tools())} 个",
|
||
" 支持多行输入 — Shift+回车换行",
|
||
"",
|
||
]
|
||
|
||
# 逐行渲染，带延迟
|
||
for i, line in enumerate(welcome_lines):
|
||
if "╔" in line or "║" in line or "╚" in line:
|
||
cls = "msg-welcome-line"
|
||
elif line.strip().startswith("小狸"):
|
||
cls = "msg-welcome-line"
|
||
elif line.strip().startswith("智能"):
|
||
cls = "msg-welcome-line"
|
||
elif "代码编辑" in line or "输入 /help" in line or "当前引擎" in line:
|
||
cls = "msg-system-line"
|
||
elif "多行输入" in line:
|
||
cls = "msg-system-line"
|
||
else:
|
||
cls = "msg-dim"
|
||
|
||
widget = Static(line, classes=cls)
|
||
scroll.mount(widget)
|
||
# 延迟显示每一行（避免 delay=0 导致 Textual 除零错误）
|
||
delay = max(0.01, i * 0.1)
|
||
self.set_timer(delay, lambda w=widget: w.add_class("visible"))
|
||
|
||
# 最后滚动到底部
|
||
self.set_timer(max(0.01, len(welcome_lines) * 0.1 + 0.1),
|
||
lambda: scroll.scroll_end(animate=True, duration=0.3))
|
||
|
||
def _render_welcome(self):
|
||
"""静态欢迎界面（用于清屏后快速恢复）"""
|
||
scroll = self.query_one("#chat-scroll")
|
||
lines = [
|
||
("", "msg-dim"),
|
||
(" ╔══════════════════════════════════════════════╗", "msg-welcome"),
|
||
(" ║ ║", "msg-welcome"),
|
||
(" ║ 小狸 Pro-CLI v8.0.4 ║", "msg-welcome"),
|
||
(" ║ 智能编程助手 · 动画增强版 ║", "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"),
|
||
(" 支持多行输入 — Shift+回车换行", "msg-system"),
|
||
("", "msg-dim"),
|
||
]
|
||
for text, cls in lines:
|
||
scroll.mount(Static(text, classes=cls))
|
||
|
||
# ── 侧栏更新 ──
|
||
|
||
def _update_sidebar(self):
|
||
"""按当前标签页更新侧栏内容"""
|
||
tab = self._sidebar_tab
|
||
|
||
# ── 引擎面板 ──
|
||
engine_panel = self.query_one("#panel-engine")
|
||
engine_panel.remove_children()
|
||
current = self.bridge.current_engine()
|
||
for name in self.bridge.engines():
|
||
if name == current:
|
||
engine_panel.mount(Static(f" ▸ {name}", classes="engine-item-active"))
|
||
else:
|
||
engine_panel.mount(Static(f" {name}", classes="engine-item"))
|
||
# 工具列表也放引擎面板下方
|
||
engine_panel.mount(Static(" 工具", classes="panel-title"))
|
||
for tool in self.bridge.tools()[:12]:
|
||
name = tool.get('name', '?')
|
||
engine_panel.mount(Static(f" • {name}", classes="tool-item"))
|
||
|
||
# ── 文件树面板 ──
|
||
self._update_file_tree()
|
||
|
||
# ── 会话面板 ──
|
||
self._update_session_list()
|
||
|
||
# ── 状态面板 ──
|
||
status_panel = self.query_one("#panel-status")
|
||
status_panel.remove_children()
|
||
status_panel.mount(Static(f" 对话: {len(self.bridge.history())} 条", classes="tool-item"))
|
||
status_panel.mount(Static(f" 工具: {len(self.bridge.tools())} 个", classes="tool-item"))
|
||
status_panel.mount(Static(f" 引擎: {len(self.bridge.engines())} 个", classes="tool-item"))
|
||
if self.bridge.plan_mode():
|
||
status_panel.mount(Static(" 模式: 📋 PLAN", classes="plan-badge"))
|
||
else:
|
||
status_panel.mount(Static(" 模式: 普通", classes="tool-item"))
|
||
|
||
# 更新标签高亮
|
||
self._sync_tab_highlight()
|
||
|
||
# ── 文件树 ──
|
||
|
||
@staticmethod
|
||
def _build_file_tree(root, max_depth=2, max_items=40):
|
||
"""生成 (name, is_dir) 列表，用于侧栏文件树展示。"""
|
||
items = []
|
||
try:
|
||
entries = sorted(os.listdir(root))
|
||
except OSError:
|
||
return items
|
||
dirs = [e for e in entries if os.path.isdir(os.path.join(root, e))]
|
||
files = [e for e in entries if os.path.isfile(os.path.join(root, e))]
|
||
skip = {'.git', '__pycache__', '.tui_snapshots', '.pytest_cache',
|
||
'node_modules', '.workbuddy', 'chat_history', 'memory',
|
||
'.idea', '.vscode', '.venv', 'venv'}
|
||
shown = 0
|
||
for name in dirs + files:
|
||
if name in skip or name.startswith('.'):
|
||
continue
|
||
if shown >= max_items:
|
||
items.append(("…", False))
|
||
break
|
||
is_dir = os.path.isdir(os.path.join(root, name))
|
||
items.append((name, is_dir))
|
||
shown += 1
|
||
return items
|
||
|
||
def _update_file_tree(self):
|
||
container = self.query_one("#panel-files")
|
||
container.remove_children()
|
||
root = self.bridge.cwd()
|
||
container.mount(Static(f" 📂 {os.path.basename(root)}", classes="file-tree-dir"))
|
||
for name, is_dir in self._build_file_tree(root):
|
||
icon = "📁" if is_dir else "📄"
|
||
container.mount(Static(f" {icon} {name}", classes="file-tree-item"))
|
||
|
||
# ── 会话列表 ──
|
||
|
||
@staticmethod
|
||
def _format_rel_time(ts: str) -> str:
|
||
if not ts:
|
||
return ""
|
||
try:
|
||
from datetime import datetime
|
||
dt = datetime.fromisoformat(ts.replace('Z', '+00:00'))
|
||
diff = (datetime.now() - dt).total_seconds()
|
||
if diff < 60:
|
||
return "刚刚"
|
||
if diff < 3600:
|
||
return f"{int(diff // 60)}分钟前"
|
||
if diff < 86400:
|
||
return f"{int(diff // 3600)}小时前"
|
||
return f"{int(diff // 86400)}天前"
|
||
except Exception:
|
||
return ""
|
||
|
||
def _update_session_list(self):
|
||
container = self.query_one("#panel-sessions")
|
||
container.remove_children()
|
||
sessions = self.bridge.sessions()[:8]
|
||
if not sessions:
|
||
container.mount(Static(" (无历史会话)", classes="session-item"))
|
||
return
|
||
for i, s in enumerate(sessions, 1):
|
||
title = (s.title or s.id)[:22]
|
||
t = self._format_rel_time(s.updated_at)
|
||
container.mount(Static(f" {i}. {title} {t}", classes="session-item"))
|
||
|
||
# ── 标签页切换 ──
|
||
|
||
def _sync_tab_highlight(self):
|
||
"""同步标签高亮状态"""
|
||
tab_map = {"engine": "tab-engine", "files": "tab-files",
|
||
"sessions": "tab-sessions", "status": "tab-status"}
|
||
for tid in tab_map.values():
|
||
try:
|
||
w = self.query_one(f"#{tid}")
|
||
if tid == tab_map.get(self._sidebar_tab, ""):
|
||
w.add_class("sidebar-tab-active")
|
||
else:
|
||
w.remove_class("sidebar-tab-active")
|
||
except Exception:
|
||
pass
|
||
|
||
def _switch_sidebar_tab(self, tab: str):
|
||
"""切换侧栏标签页"""
|
||
self._sidebar_tab = tab
|
||
# 显示/隐藏面板
|
||
panel_map = {
|
||
"engine": "panel-engine", "files": "panel-files",
|
||
"sessions": "panel-sessions", "status": "panel-status",
|
||
}
|
||
for pt, pid in panel_map.items():
|
||
try:
|
||
p = self.query_one(f"#{pid}")
|
||
if pt == tab:
|
||
p.remove_class("hidden")
|
||
else:
|
||
p.add_class("hidden")
|
||
except Exception:
|
||
pass
|
||
self._sync_tab_highlight()
|
||
self._update_sidebar()
|
||
|
||
def action_tab_engine(self):
|
||
self._switch_sidebar_tab("engine")
|
||
|
||
def action_tab_files(self):
|
||
self._switch_sidebar_tab("files")
|
||
|
||
def action_tab_sessions(self):
|
||
self._switch_sidebar_tab("sessions")
|
||
|
||
def action_tab_status(self):
|
||
self._switch_sidebar_tab("status")
|
||
|
||
def _update_header(self):
|
||
"""更新自定义顶栏内容"""
|
||
hdr = self.query_one("#tui-header", expect_type=Static)
|
||
engine = self.bridge.current_engine()
|
||
plan_tag = " [PLAN]" if self.bridge.plan_mode() else ""
|
||
hist_len = len(self.bridge.history())
|
||
tok = self._token_str().strip()
|
||
hdr.update(
|
||
f" 小狸 v8.0{plan_tag} │ {engine} │ {hist_len}条对话{tok}"
|
||
)
|
||
|
||
def _update_status(self, text: str):
|
||
bar = self.query_one("#status-bar")
|
||
engine = self.bridge.current_engine()
|
||
prefix = "📋 PLAN | " if self.plan_mode else ""
|
||
bar.update(f" {prefix}{text} | {engine} | {len(self.bridge.history())} 条对话{self._token_str()}")
|
||
|
||
def _token_str(self):
|
||
"""token 用量进度（v8.0 token 感知压缩联动，引擎支持时显示）"""
|
||
e = getattr(self.cli, 'current_engine', None)
|
||
if not e:
|
||
return ""
|
||
used = getattr(e, 'last_prompt_tokens', None)
|
||
win = None
|
||
if hasattr(e, 'context_window'):
|
||
try:
|
||
win = e.context_window()
|
||
except Exception:
|
||
win = None
|
||
if used and win:
|
||
ratio = min(1.0, used / win)
|
||
filled = int(ratio * 10)
|
||
bar_s = "▓" * filled + "░" * (10 - filled)
|
||
return f" | {used}/{win} {bar_s}"
|
||
if used:
|
||
return f" | {used} tok"
|
||
return ""
|
||
|
||
# ── 消息追加 ──
|
||
|
||
def _append(self, widget):
|
||
scroll = self.query_one("#chat-scroll")
|
||
scroll.mount(widget)
|
||
scroll.scroll_end(animate=True, duration=0.2)
|
||
|
||
def _add(self, text, cls="msg-dim"):
|
||
widget = MessageBubble(text, classes=cls)
|
||
self._append(widget)
|
||
|
||
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)
|
||
panel = Panel(syntax, border_style=f"dim {_Theme.BORDER}",
|
||
box=ROUNDED, padding=(0, 1))
|
||
widget = MessageBubble(str(panel), classes="code-block")
|
||
self._append(widget)
|
||
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 _show_thinking(self):
|
||
"""显示动画思考指示器"""
|
||
self._remove_thinking()
|
||
spinner = ThinkingSpinner(classes="thinking-indicator")
|
||
scroll = self.query_one("#chat-scroll")
|
||
scroll.mount(spinner)
|
||
scroll.scroll_end(animate=True, duration=0.2)
|
||
self._thinking_widget = spinner
|
||
|
||
def _remove_thinking(self):
|
||
"""移除思考指示器"""
|
||
if self._thinking_widget:
|
||
try:
|
||
self._thinking_widget.stop()
|
||
self._thinking_widget.remove()
|
||
except Exception:
|
||
pass
|
||
self._thinking_widget = None
|
||
|
||
def _show_typing_indicator(self):
|
||
"""显示打字指示器"""
|
||
self._remove_typing_indicator()
|
||
indicator = TypingIndicator(classes="thinking-indicator")
|
||
scroll = self.query_one("#chat-scroll")
|
||
scroll.mount(indicator)
|
||
scroll.scroll_end(animate=True, duration=0.2)
|
||
self._typing_widget = indicator
|
||
|
||
def _remove_typing_indicator(self):
|
||
"""移除打字指示器"""
|
||
if self._typing_widget:
|
||
try:
|
||
self._typing_widget.stop()
|
||
self._typing_widget.remove()
|
||
except Exception:
|
||
pass
|
||
self._typing_widget = None
|
||
|
||
def _thinking(self):
|
||
"""兼容旧接口"""
|
||
self._show_thinking()
|
||
|
||
def _system(self, text):
|
||
self._add(f" ℹ {text}", "msg-system")
|
||
|
||
def _error(self, text):
|
||
self._add(f" {text}", "msg-error")
|
||
|
||
# ── 多行输入处理 ──
|
||
|
||
def action_newline(self):
|
||
"""Shift+Enter 在输入框插入换行"""
|
||
text_area = self.query_one("#user-input")
|
||
if text_area.has_focus:
|
||
text_area.action_edit_insert_newline()
|
||
|
||
def action_send_message(self):
|
||
"""回车发送消息"""
|
||
text_area = self.query_one("#user-input")
|
||
text = text_area.text.strip()
|
||
if not text:
|
||
return
|
||
|
||
# 清空输入框
|
||
text_area.clear()
|
||
|
||
if text.startswith('/'):
|
||
self._handle_command(text)
|
||
return
|
||
|
||
self._user_msg(text)
|
||
self._show_thinking()
|
||
self.is_generating = True
|
||
self._update_status(" 思考中...")
|
||
self.run_worker(self._generate(text), exclusive=True)
|
||
|
||
# ── 兼容旧 Input 提交（保留以防万一） ──
|
||
|
||
@on(TextArea.Changed, "#user-input")
|
||
def on_textarea_changed(self, event):
|
||
"""输入框内容变化时调整高度"""
|
||
text_area = event.text_area
|
||
line_count = text_area.text.count('\n') + 1
|
||
# 动态调整输入区域高度
|
||
wrapper = self.query_one("#input-wrapper")
|
||
new_height = min(max(line_count + 1, 4), 12)
|
||
wrapper.styles.height = new_height
|
||
|
||
# ── 命令处理 ──
|
||
|
||
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(f" 小狸 Pro-CLI v{VERSION} - 智能编程助手"),
|
||
'status': lambda: self._show_status(),
|
||
'tools': lambda: self._show_tools(),
|
||
'engines': lambda: self._show_engines(),
|
||
'tui': lambda: self._system("已在 TUI 模式中"),
|
||
'manual': lambda: self._handle_manual(args),
|
||
'plan': lambda: self._tui_plan(args),
|
||
'build': lambda: self._tui_build(),
|
||
'sessions': lambda: self._tui_sessions(),
|
||
'session': lambda: self._tui_session(args),
|
||
'resume': lambda: self._tui_resume(args),
|
||
'snapshot': lambda: self._save_screenshot(),
|
||
'screenshot': lambda: self._save_screenshot(),
|
||
'vim': lambda: self._toggle_vim(),
|
||
}
|
||
|
||
handler = cmds.get(name)
|
||
if handler:
|
||
handler()
|
||
return
|
||
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}")
|
||
return
|
||
# 未知命令转发给 cli 完整路由（/agent /providers /chat 等）
|
||
self._forward_to_cli(cmd)
|
||
|
||
def _show_help(self):
|
||
help_text = """ 命令:
|
||
/help 帮助信息
|
||
/quit 退出
|
||
/cli 切换命令行模式
|
||
/model <引擎> 切换 AI 引擎
|
||
/engines 列出引擎
|
||
/tools 列出工具
|
||
/status 系统状态
|
||
/clear 清屏
|
||
/vim 开关 Vim 键位
|
||
|
||
⌨ 快捷键:
|
||
Ctrl+C 退出 Ctrl+L 清屏
|
||
Ctrl+N 新对话 回车 发送消息
|
||
Shift+回车 换行 F1 侧栏
|
||
Escape 取消生成 Tab 缩进
|
||
Ctrl+Space 焦点切换 i 聚焦并输入
|
||
Ctrl+F/B 翻页下/上 Ctrl+D/U 半页下/上
|
||
|
||
VIM 输入(默认开): 输入框内 Esc 进普通模式
|
||
h j k l 移动 w/b 跳词 0/$ 行首/行尾
|
||
i a o 插入/追加/换行 I A O 行首/行尾/上方
|
||
dd 删行 x 删字 u 撤销 gg/G 文首/文尾
|
||
普通模式下 回车 = 下移一行"""
|
||
self._system(help_text)
|
||
|
||
def _handle_manual(self, args):
|
||
"""处理 /manual 命令"""
|
||
if 'manual' not in self.cli.engines:
|
||
self._error("manual 引擎未加载")
|
||
return
|
||
if not args:
|
||
# 切换到 manual 引擎
|
||
if self.bridge.switch_engine('manual'):
|
||
self._system("已切换到 manual 引擎")
|
||
self._update_sidebar()
|
||
else:
|
||
self._error("切换失败")
|
||
else:
|
||
# 执行 manual 引擎子命令
|
||
result = self.cli.engines['manual'].handle_command(args)
|
||
if result:
|
||
self._system(result)
|
||
|
||
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):
|
||
status_lines = [
|
||
f" 系统状态:",
|
||
f" 引擎: {self.bridge.current_engine()}",
|
||
f" 引擎数: {len(self.bridge.engines())}",
|
||
f" 工具数: {len(self.bridge.tools())}",
|
||
f" 对话数: {len(self.bridge.history())}",
|
||
]
|
||
|
||
self._system('\n'.join(status_lines))
|
||
|
||
# ── Plan 模式 / 会话 委派 ──
|
||
|
||
def _tui_plan(self, args):
|
||
args = (args or "").strip()
|
||
if args in ('off', 'exit', '退出', 'cancel'):
|
||
self._set_plan_mode(False)
|
||
self._system("已退出 PLAN 模式（计划未执行）")
|
||
return
|
||
self._set_plan_mode(True)
|
||
if args:
|
||
self._system(f"📋 已进入 PLAN 模式，正在只读调研: {args}")
|
||
self._user_msg(f"/plan {args}")
|
||
self._show_thinking()
|
||
self.is_generating = True
|
||
self._update_status("思考中 (PLAN)...")
|
||
self.run_worker(self._generate(args), exclusive=True)
|
||
else:
|
||
self._system("📋 已进入 PLAN 模式。描述任务，AI 将只做只读调研并给出实施计划；/build 批准执行，/plan off 取消。")
|
||
|
||
def _tui_build(self):
|
||
if not self.bridge.plan_mode() and not self.bridge.current_plan().strip():
|
||
self._system("当前不在 PLAN 模式，且无已生成的计划")
|
||
return
|
||
self._set_plan_mode(False)
|
||
self._system("✅ 已批准 PLAN，开始执行。")
|
||
|
||
def _tui_sessions(self):
|
||
sessions = self.bridge.sessions()
|
||
if not sessions:
|
||
self._system(" 无历史会话")
|
||
return
|
||
lines = [" 历史会话:"]
|
||
for i, s in enumerate(sessions[:10], 1):
|
||
title = (s.title or s.id)[:30]
|
||
t = self._format_rel_time(s.updated_at)
|
||
lines.append(f" {i}. {title} ({t})")
|
||
self._system('\n'.join(lines))
|
||
|
||
def _tui_session(self, args):
|
||
args = (args or "").strip()
|
||
if not args:
|
||
self._tui_sessions()
|
||
return
|
||
if args == 'new':
|
||
self.action_new_chat()
|
||
return
|
||
self._forward_to_cli(f"/resume {args}")
|
||
|
||
def _tui_resume(self, args):
|
||
self._forward_to_cli(f"/resume {args}")
|
||
|
||
def _forward_to_cli(self, cmd: str):
|
||
"""把未知/委派命令交给 cli 完整路由处理。"""
|
||
try:
|
||
self.cli.process_conversation(cmd)
|
||
except Exception as e:
|
||
self._error(f"命令执行失败: {e}")
|
||
self.call_after_refresh(self._update_sidebar)
|
||
self.call_after_refresh(self._sync_plan)
|
||
|
||
def _set_plan_mode(self, on: bool):
|
||
self.plan_mode = on
|
||
try:
|
||
self.cli.plan_mode = on
|
||
if not on:
|
||
self.cli.current_plan = ""
|
||
except Exception:
|
||
pass
|
||
self._update_sidebar()
|
||
self._update_status("📋 PLAN" if on else "就绪")
|
||
|
||
def _sync_plan(self):
|
||
"""从 cli 同步 plan 状态到 TUI 反应变量。"""
|
||
remote = self.bridge.plan_mode()
|
||
if remote != self.plan_mode:
|
||
self.plan_mode = remote
|
||
self._update_sidebar()
|
||
|
||
# ── AI 生成 ──
|
||
|
||
async def _generate(self, user_input):
|
||
try:
|
||
self.call_after_refresh(self._remove_thinking)
|
||
self.call_after_refresh(self._show_typing_indicator)
|
||
|
||
def tui_output(msg):
|
||
self.call_after_refresh(self._remove_typing_indicator)
|
||
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._remove_typing_indicator)
|
||
self.call_after_refresh(self._error, str(e))
|
||
finally:
|
||
self.call_after_refresh(self._remove_thinking)
|
||
self.call_after_refresh(self._remove_typing_indicator)
|
||
self.is_generating = False
|
||
self.call_after_refresh(self._sync_plan)
|
||
self.call_after_refresh(lambda: self._update_status("就绪"))
|
||
self.call_after_refresh(self._update_sidebar)
|
||
self.call_after_refresh(self._update_header)
|
||
|
||
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")
|
||
# 通过 CSS 类控制侧栏显隐
|
||
if sidebar.has_class("hidden"):
|
||
sidebar.remove_class("hidden")
|
||
self.sidebar_visible = True
|
||
else:
|
||
sidebar.add_class("hidden")
|
||
self.sidebar_visible = False
|
||
|
||
def action_toggle_plan(self):
|
||
self._set_plan_mode(not self.plan_mode)
|
||
self._system("📋 PLAN 模式: " + ("开" if self.plan_mode else "关"))
|
||
|
||
def action_cancel(self):
|
||
if self.is_generating:
|
||
self._remove_thinking()
|
||
self._remove_typing_indicator()
|
||
self._system("已取消")
|
||
self.is_generating = False
|
||
|
||
# ── Vim 风格：聊天区滚动 / 聚焦切换 ──
|
||
def _user_input(self):
|
||
return self.query_one("#user-input")
|
||
|
||
def _chat_scroll(self):
|
||
return self.query_one("#chat-scroll")
|
||
|
||
def action_toggle_focus(self):
|
||
"""Ctrl+Space：在输入框与聊天区之间切换焦点。"""
|
||
ta = self._user_input()
|
||
if ta.has_focus:
|
||
self.screen.set_focus(None)
|
||
else:
|
||
ta.focus()
|
||
if getattr(ta, "vim_enabled", False):
|
||
ta.vim.mode = "INSERT"
|
||
self._update_vim_mode_display()
|
||
|
||
def action_focus_insert(self):
|
||
"""i：输入框未聚焦时聚焦并进入 INSERT（模仿 vim 按 i 开始输入）。"""
|
||
ta = self._user_input()
|
||
if ta.has_focus:
|
||
return
|
||
ta.focus()
|
||
if getattr(ta, "vim_enabled", False):
|
||
ta.vim.mode = "INSERT"
|
||
self._update_vim_mode_display()
|
||
|
||
def _guard_chat(self):
|
||
"""聊天导航仅在输入框未聚焦时生效，避免与输入冲突。"""
|
||
return self._user_input().has_focus
|
||
|
||
def action_chat_down(self):
|
||
if self._guard_chat():
|
||
return
|
||
self._chat_scroll().scroll_down(3)
|
||
|
||
def action_chat_up(self):
|
||
if self._guard_chat():
|
||
return
|
||
self._chat_scroll().scroll_up(3)
|
||
|
||
def action_chat_page_down(self):
|
||
if self._guard_chat():
|
||
return
|
||
self._chat_scroll().scroll_page_down()
|
||
|
||
def action_chat_page_up(self):
|
||
if self._guard_chat():
|
||
return
|
||
self._chat_scroll().scroll_page_up()
|
||
|
||
def action_chat_half_down(self):
|
||
if self._guard_chat():
|
||
return
|
||
self._chat_scroll().scroll_relative(y=20)
|
||
|
||
def action_chat_half_up(self):
|
||
if self._guard_chat():
|
||
return
|
||
self._chat_scroll().scroll_relative(y=-20)
|
||
|
||
def _update_vim_mode_display(self):
|
||
"""在输入框下方提示栏显示当前 Vim 模式。"""
|
||
try:
|
||
hint = self.query_one("#input-hint")
|
||
ta = self._user_input()
|
||
except Exception:
|
||
return
|
||
if not getattr(ta, "vim_enabled", False):
|
||
hint.update(" 回车发送 | Shift+回车换行 | Ctrl+L 清屏 | F1 侧栏 | Ctrl+P PLAN")
|
||
return
|
||
mode = getattr(getattr(ta, "vim", None), "mode", "INSERT")
|
||
tag = "-- INSERT --" if mode == "INSERT" else "-- NORMAL --"
|
||
hint.update(f" {tag} │ Esc 切模式 │ i插入 a追加 o换行 hjkl移动 dd删行 x删字 u撤销")
|
||
|
||
def _toggle_vim(self):
|
||
ta = self._user_input()
|
||
ta.vim_enabled = not getattr(ta, "vim_enabled", True)
|
||
if ta.vim_enabled:
|
||
self._system("Vim 键位: 开 (Esc 在 NORMAL/INSERT 间切换)")
|
||
else:
|
||
self._system("Vim 键位: 关 (普通输入)")
|
||
self._update_vim_mode_display()
|
||
|
||
def _save_screenshot(self):
|
||
"""导出当前 TUI 屏幕截图（SVG）——AI 可据此自验实时渲染（/snapshot）"""
|
||
import os
|
||
out_dir = os.path.join(
|
||
os.path.dirname(os.path.dirname(os.path.abspath(__file__))), ".tui_snapshots"
|
||
)
|
||
os.makedirs(out_dir, exist_ok=True)
|
||
path = os.path.join(out_dir, "live.svg")
|
||
try:
|
||
svg = self.export_screenshot()
|
||
with open(path, "w", encoding="utf-8") as f:
|
||
f.write(svg)
|
||
self._system(f"截图已保存: {path}")
|
||
except Exception as e:
|
||
self._error(f"截图失败: {e}")
|
||
|
||
else:
|
||
# Textual 不可用时的占位类
|
||
class XiaoliTUI:
|
||
"""Textual 未安装时的占位类"""
|
||
def __init__(self, *args, **kwargs):
|
||
pass
|
||
def run(self, *args, **kwargs):
|
||
print("TUI 模式不可用：Textual 库未安装")
|
||
print("请运行: pip install textual rich")
|