Compare commits
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
b8faaae0c6 | ||
|
|
bac5ea01a9 | ||
|
|
161ab9d3a8 | ||
|
|
b0d9a24003 | ||
|
|
97cc2bb0f7 |
No files matched your search
@@ -4,6 +4,29 @@
|
||||
|
||||
---
|
||||
|
||||
## v5.4.0 (2026-05-09) — Release
|
||||
|
||||
### ✨ 新增功能
|
||||
|
||||
- **跨平台进程保护模块** (`xcli_core/process_protection.py`)
|
||||
- **单实例保护** — 防止重复启动,Windows 使用命名 Mutex,macOS/Linux 使用 fcntl 文件锁(自动清理残留锁文件)
|
||||
- **进程优先级提升** — Windows: `SetPriorityClass` (ABOVE_NORMAL/HIGH/REALTIME),macOS/Linux: `os.nice()` + `renice`
|
||||
- **Windows PPL 风格保护** — `SetErrorMode` 抑制系统错误弹窗 (SEM_FAILCRITICALERRORS | SEM_NOGPFAULTERRORBOX)、`faulthandler` 崩溃堆栈捕获
|
||||
- **POSIX 信号保护** — macOS/Linux 忽略 SIGHUP(终端关闭不退出)、注册 SIGTERM 优雅关闭、macOS 忽略 SIGPIPE
|
||||
- **看门狗监控** — 后台线程监控父进程状态,父进程退出时自动清理并退出
|
||||
- **关闭钩子系统** — 注册自定义清理函数,进程退出时自动执行
|
||||
- **一键启用** — `enable_process_protection()` 单函数调用启用全部保护
|
||||
- **`/protect` 命令** — CLI 中查看保护状态,`/protect test` 重新测试所有保护
|
||||
|
||||
### 改进
|
||||
|
||||
- `platform_utils.py` 新增 `is_process_running()`、`get_process_priority()`、`set_process_priority()` 跨平台工具函数
|
||||
- `cli_base.py` 启动时自动初始化进程保护,退出时自动清理
|
||||
- `cli_core.py` 启动 banner 显示保护状态图标(单实例/优先级/PPL/看门狗)
|
||||
- 全版本号统一更新至 v5.4.0
|
||||
|
||||
---
|
||||
|
||||
## v5.2.0 (2026-05-08)
|
||||
|
||||
### ✨ 新增功能
|
||||
|
||||
+1
-1
@@ -5,7 +5,7 @@
|
||||
</p>
|
||||
|
||||
<p align="center">
|
||||
<img src="https://img.shields.io/badge/version-5.2.0-blue" alt="version">
|
||||
<img src="https://img.shields.io/badge/version-5.4.1-blue" alt="version">
|
||||
<img src="https://img.shields.io/badge/python-3.10+-green" alt="python">
|
||||
<img src="https://img.shields.io/badge/license-MIT-orange" alt="license">
|
||||
<img src="https://img.shields.io/badge/tests-113%20passed-brightgreen" alt="tests">
|
||||
|
||||
@@ -9,7 +9,7 @@
|
||||
</p>
|
||||
|
||||
<p align="center">
|
||||
<img src="https://img.shields.io/badge/version-5.2.0-blue" alt="version">
|
||||
<img src="https://img.shields.io/badge/version-5.4.1-blue" alt="version">
|
||||
<img src="https://img.shields.io/badge/python-3.10+-green" alt="python">
|
||||
<img src="https://img.shields.io/badge/license-MIT-orange" alt="license">
|
||||
<img src="https://img.shields.io/badge/tests-150+%20passed-brightgreen" alt="tests">
|
||||
@@ -35,6 +35,7 @@
|
||||
- **定时任务** — 支持相对时间、绝对时间、重复任务,到期自动激活 AI
|
||||
- **系统通知** — 跨平台任务完成通知(Windows / macOS / Linux)
|
||||
- **MCP 协议** — 19 个插件全部实现 MCP 定义,支持 Function Calling 原生调用
|
||||
- **进程保护** — 跨平台单实例保护 + 进程优先级提升 + Windows PPL 风格保护 + 看门狗监控 + 优雅关闭
|
||||
- **19 个插件 161+ 个操作** — 代码编辑、Git、浏览器、工程化、文件管理、图像生成、记忆等
|
||||
- **三模式界面** — TUI 图形化 + 传统命令行 + Web UI 浏览器界面
|
||||
- **150+ 个自动化测试** — pytest 覆盖全部插件和核心模块
|
||||
|
||||
+2
-2
@@ -1,10 +1,10 @@
|
||||
@echo off
|
||||
chcp 65001 >nul 2>&1
|
||||
title 小狸 Pro-CLI v5.1.2
|
||||
title 小狸 Pro-CLI v5.4.0
|
||||
|
||||
echo.
|
||||
echo ╔══════════════════════════════════════╗
|
||||
echo ║ 小狸 Pro-CLI 启动器 v5.1.2 ║
|
||||
echo ║ 小狸 Pro-CLI 启动器 v5.4.0 ║
|
||||
echo ║ 正在检测环境... ║
|
||||
echo ╚══════════════════════════════════════╝
|
||||
echo.
|
||||
|
||||
+147
@@ -0,0 +1,147 @@
|
||||
#!/usr/bin/env python3
|
||||
"""
|
||||
小狸 MCP Stdio Server
|
||||
=====================
|
||||
标准 MCP (Model Context Protocol) stdio 传输层入口。
|
||||
通过 stdin/stdout 以 JSON-RPC 2.0 格式通信,
|
||||
可被 Claude Desktop、Cursor、OpenClaw 等标准 MCP 客户端直接连接。
|
||||
|
||||
用法:
|
||||
python mcp_server.py [--plugins-dir plugins] [--skills-dir skills]
|
||||
|
||||
Claude Desktop 配置示例:
|
||||
{
|
||||
"mcpServers": {
|
||||
"xiaoli": {
|
||||
"command": "python",
|
||||
"args": ["/path/to/xiaoli-cli/mcp_server.py"]
|
||||
}
|
||||
}
|
||||
}
|
||||
"""
|
||||
|
||||
import sys
|
||||
import os
|
||||
import json
|
||||
import argparse
|
||||
import logging
|
||||
|
||||
# 确保当前目录在 sys.path 中,以便导入项目模块
|
||||
SCRIPT_DIR = os.path.dirname(os.path.abspath(__file__))
|
||||
if SCRIPT_DIR not in sys.path:
|
||||
sys.path.insert(0, SCRIPT_DIR)
|
||||
|
||||
# 日志输出到 stderr,不干扰 stdout 上的 JSON-RPC 通信
|
||||
logging.basicConfig(
|
||||
level=logging.INFO,
|
||||
format="[xiaoli-mcp] %(message)s",
|
||||
stream=sys.stderr,
|
||||
)
|
||||
logger = logging.getLogger("xiaoli-mcp")
|
||||
|
||||
|
||||
def main():
|
||||
parser = argparse.ArgumentParser(description="小狸 MCP Stdio Server")
|
||||
parser.add_argument("--plugins-dir", default="plugins", help="插件目录 (默认: plugins)")
|
||||
parser.add_argument("--skills-dir", default="skills", help="技能目录 (默认: skills)")
|
||||
args = parser.parse_args()
|
||||
|
||||
# 切换到脚本所在目录,保证相对路径正确
|
||||
os.chdir(SCRIPT_DIR)
|
||||
|
||||
# 初始化 UnifiedToolManager
|
||||
# 将 initialize() 中的 print 重定向到 stderr,避免污染 stdout JSON-RPC 通道
|
||||
import io
|
||||
from contextlib import redirect_stdout
|
||||
|
||||
from unified_tool_manager import UnifiedToolManager
|
||||
|
||||
manager = UnifiedToolManager()
|
||||
captured_output = io.StringIO()
|
||||
with redirect_stdout(captured_output):
|
||||
manager.initialize(
|
||||
plugins_dir=args.plugins_dir,
|
||||
skills_dir=args.skills_dir,
|
||||
)
|
||||
|
||||
# 将初始化时的 print 输出转发到 stderr
|
||||
init_output = captured_output.getvalue()
|
||||
if init_output:
|
||||
for line in init_output.strip().split('\n'):
|
||||
logger.info(line)
|
||||
|
||||
tool_count = len(manager._tools)
|
||||
logger.info(f"MCP Server 已启动,共加载 {tool_count} 个工具")
|
||||
logger.info("等待 stdin JSON-RPC 请求...")
|
||||
|
||||
# 主循环:从 stdin 逐行读取 JSON-RPC 请求
|
||||
try:
|
||||
for line in sys.stdin:
|
||||
line = line.strip()
|
||||
if not line:
|
||||
continue
|
||||
|
||||
try:
|
||||
request = json.loads(line)
|
||||
except json.JSONDecodeError as e:
|
||||
# 无法解析的 JSON,返回解析错误
|
||||
error_resp = {
|
||||
"jsonrpc": "2.0",
|
||||
"id": None,
|
||||
"error": {"code": -32700, "message": f"Parse error: {e}"},
|
||||
}
|
||||
_send(error_resp)
|
||||
continue
|
||||
|
||||
# 处理通知(没有 id 字段的请求)— MCP 规范中通知不需要响应
|
||||
req_id = request.get("id")
|
||||
method = request.get("method", "")
|
||||
|
||||
# 特殊处理 notifications/initialized — 客户端初始化完成通知
|
||||
if method == "notifications/initialized":
|
||||
logger.info("客户端已发送 initialized 通知")
|
||||
continue
|
||||
|
||||
# 特殊处理 $/cancelRequest
|
||||
if method == "$/cancelRequest":
|
||||
logger.info(f"收到取消请求: {request.get('params', {}).get('id')}")
|
||||
continue
|
||||
|
||||
# 委托给 UnifiedToolManager 处理
|
||||
# 拦截插件可能产生的 stdout 输出,转发到 stderr
|
||||
captured = io.StringIO()
|
||||
try:
|
||||
with redirect_stdout(captured):
|
||||
response = manager.handle_mcp_request(request)
|
||||
plugin_output = captured.getvalue()
|
||||
if plugin_output:
|
||||
for p_line in plugin_output.strip().split('\n'):
|
||||
logger.info(f"[plugin] {p_line}")
|
||||
except Exception as e:
|
||||
logger.error(f"处理请求异常: {e}")
|
||||
response = {
|
||||
"jsonrpc": "2.0",
|
||||
"id": req_id,
|
||||
"error": {"code": -32603, "message": f"Internal error: {e}"},
|
||||
}
|
||||
|
||||
_send(response)
|
||||
|
||||
except KeyboardInterrupt:
|
||||
logger.info("收到中断信号,正在关闭...")
|
||||
except EOFError:
|
||||
logger.info("stdin 已关闭,正在退出...")
|
||||
finally:
|
||||
manager.shutdown()
|
||||
logger.info("MCP Server 已关闭")
|
||||
|
||||
|
||||
def _send(response: dict):
|
||||
"""将 JSON-RPC 响应写入 stdout 并刷新"""
|
||||
text = json.dumps(response, ensure_ascii=False)
|
||||
sys.stdout.write(text + "\n")
|
||||
sys.stdout.flush()
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
main()
|
||||
+76
-1
@@ -121,4 +121,79 @@ def get_config_directory():
|
||||
elif platform == 'macos':
|
||||
return os.path.join(os.path.expanduser("~"), 'Library', 'Application Support', 'Xiaoli Pro-CLI 3')
|
||||
else: # Linux
|
||||
return os.path.join(os.path.expanduser('~'), '.config', 'xiaoli-pro-cli-3')
|
||||
return os.path.join(os.path.expanduser('~'), '.config', 'xiaoli-pro-cli-3')
|
||||
|
||||
def is_process_running(pid):
|
||||
"""检查指定 PID 的进程是否在运行"""
|
||||
platform = get_platform()
|
||||
|
||||
try:
|
||||
if platform == 'windows':
|
||||
import ctypes
|
||||
kernel32 = ctypes.windll.kernel32
|
||||
handle = kernel32.OpenProcess(0x1000, False, pid) # PROCESS_QUERY_LIMITED_INFORMATION
|
||||
if handle:
|
||||
kernel32.CloseHandle(handle)
|
||||
return True
|
||||
return False
|
||||
else:
|
||||
os.kill(pid, 0)
|
||||
return True
|
||||
except (OSError, ProcessLookupError):
|
||||
return False
|
||||
|
||||
|
||||
def get_process_priority():
|
||||
"""获取当前进程优先级"""
|
||||
platform = get_platform()
|
||||
|
||||
try:
|
||||
if platform == 'windows':
|
||||
import ctypes
|
||||
kernel32 = ctypes.windll.kernel32
|
||||
handle = kernel32.GetCurrentProcess()
|
||||
return kernel32.GetPriorityClass(handle)
|
||||
else:
|
||||
import resource
|
||||
return resource.getpriority(resource.PRIO_PROCESS, 0)
|
||||
except Exception:
|
||||
return None
|
||||
|
||||
|
||||
def set_process_priority(level="above_normal"):
|
||||
"""
|
||||
设置进程优先级
|
||||
|
||||
Args:
|
||||
level: "below_normal" | "normal" | "above_normal" | "high"
|
||||
|
||||
Returns:
|
||||
是否设置成功
|
||||
"""
|
||||
platform = get_platform()
|
||||
|
||||
try:
|
||||
if platform == 'windows':
|
||||
import ctypes
|
||||
kernel32 = ctypes.windll.kernel32
|
||||
priority_map = {
|
||||
"below_normal": 0x00004000,
|
||||
"normal": 0x00000020,
|
||||
"above_normal": 0x00008000,
|
||||
"high": 0x00000080,
|
||||
}
|
||||
priority = priority_map.get(level, 0x00008000)
|
||||
handle = kernel32.GetCurrentProcess()
|
||||
return bool(kernel32.SetPriorityClass(handle, priority))
|
||||
else:
|
||||
nice_map = {
|
||||
"below_normal": 10,
|
||||
"normal": 0,
|
||||
"above_normal": -10,
|
||||
"high": -15,
|
||||
}
|
||||
nice_val = nice_map.get(level, -10)
|
||||
os.nice(nice_val)
|
||||
return True
|
||||
except Exception:
|
||||
return False
|
||||
+18
-12
@@ -5,6 +5,11 @@ import os
|
||||
import re
|
||||
import subprocess
|
||||
import platform
|
||||
import sys
|
||||
|
||||
# 添加项目根目录到 path
|
||||
sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
|
||||
from xcli_core.tool_result import ToolResult, ErrorCode
|
||||
|
||||
|
||||
class Liugin:
|
||||
@@ -68,18 +73,18 @@ class Liugin:
|
||||
':(){ :|:& };:', 'chmod -R 777 /', 'mv / ', 'wget http',
|
||||
}
|
||||
|
||||
def handle(self, args: str) -> str:
|
||||
def handle(self, args: str):
|
||||
try:
|
||||
parts = args.strip().split(maxsplit=1)
|
||||
if not parts:
|
||||
return "错误:请提供操作。可用: run <命令>"
|
||||
return ToolResult.missing("操作。可用: run <命令>")
|
||||
|
||||
operation = parts[0].lower()
|
||||
if operation != "run":
|
||||
return f"错误:不支持的操作 '{operation}'。可用: run"
|
||||
return ToolResult.fail(f"不支持的操作 '{operation}'。可用: run", ErrorCode.UNSUPPORTED_OP)
|
||||
|
||||
if len(parts) < 2:
|
||||
return "错误:请提供要执行的命令"
|
||||
return ToolResult.missing("要执行的命令")
|
||||
|
||||
rest = parts[1]
|
||||
|
||||
@@ -91,21 +96,20 @@ class Liugin:
|
||||
rest = re.sub(r'--timeout\s+\d+', '', rest).strip()
|
||||
|
||||
if not rest:
|
||||
return "错误:请提供要执行的命令"
|
||||
return ToolResult.missing("要执行的命令")
|
||||
|
||||
# 安全检查
|
||||
for blocked in self.BLOCKED:
|
||||
if blocked in rest:
|
||||
return f" 安全拦截: 检测到危险命令模式 '{blocked}'"
|
||||
return ToolResult.fail(f"安全拦截: 检测到危险命令模式 '{blocked}'", ErrorCode.PERMISSION_DENIED)
|
||||
|
||||
return self._execute(rest, timeout)
|
||||
|
||||
except Exception as e:
|
||||
return f"命令执行错误: {str(e)}"
|
||||
return ToolResult.fail(str(e), ErrorCode.INTERNAL_ERROR, tool_name="cmd_executor")
|
||||
|
||||
def _execute(self, command: str, timeout: int) -> str:
|
||||
def _execute(self, command: str, timeout: int):
|
||||
try:
|
||||
# Windows 下用 shell=True,Unix 下直接执行
|
||||
use_shell = platform.system() == 'Windows'
|
||||
|
||||
result = subprocess.run(
|
||||
@@ -136,9 +140,11 @@ class Liugin:
|
||||
if len(output) > 5000:
|
||||
output = output[:5000] + f"\n... (截断,超时: {timeout}s)"
|
||||
|
||||
return output.strip()
|
||||
if rc != 0:
|
||||
return ToolResult.fail(output.strip(), ErrorCode.EXEC_FAILED, tool_name="cmd_executor")
|
||||
return ToolResult.ok(output.strip(), tool_name="cmd_executor")
|
||||
|
||||
except subprocess.TimeoutExpired:
|
||||
return f" 命令超时 ({timeout}s): {command}"
|
||||
return ToolResult.timeout(timeout, tool_name="cmd_executor")
|
||||
except Exception as e:
|
||||
return f"执行失败: {str(e)}"
|
||||
return ToolResult.fail(str(e), ErrorCode.EXEC_FAILED, tool_name="cmd_executor")
|
||||
+10
-8
@@ -684,13 +684,15 @@ else:
|
||||
thin_sep = '─' * 70
|
||||
|
||||
# ANSI 颜色
|
||||
R = '\033[31m' # 红
|
||||
G = '\033[32m' # 绿
|
||||
R = '\033[31m' # 红字
|
||||
G = '\033[32m' # 绿字
|
||||
C = '\033[36m' # 青
|
||||
B = '\033[1m' # 粗体
|
||||
D = '\033[2m' # 暗
|
||||
W = '\033[37m' # 白
|
||||
N = '\033[0m' # 重置
|
||||
RBG = '\033[41m\033[97m' # 红底白字
|
||||
GBG = '\033[42m\033[97m' # 绿底白字
|
||||
|
||||
# 构建带颜色的 diff
|
||||
diff_colored_lines = []
|
||||
@@ -700,9 +702,9 @@ else:
|
||||
elif line.startswith('+++') or line.startswith('---'):
|
||||
diff_colored_lines.append(f"{B}{line}{N}")
|
||||
elif line.startswith('+'):
|
||||
diff_colored_lines.append(f"{G}{line}{N}")
|
||||
diff_colored_lines.append(f"{GBG} {line} {N}")
|
||||
elif line.startswith('-'):
|
||||
diff_colored_lines.append(f"{R}{line}{N}")
|
||||
diff_colored_lines.append(f"{RBG} {line} {N}")
|
||||
else:
|
||||
diff_colored_lines.append(line)
|
||||
diff_colored = '\n'.join(diff_colored_lines) if diff_colored_lines else ' (无差异)'
|
||||
@@ -845,9 +847,9 @@ else:
|
||||
elif line.startswith('+++') or line.startswith('---'):
|
||||
output(f"[bold]{line}[/]")
|
||||
elif line.startswith('+'):
|
||||
output(f"[green]{line}[/]")
|
||||
output(f"[white on green] {line} [/]")
|
||||
elif line.startswith('-'):
|
||||
output(f"[red]{line}[/]")
|
||||
output(f"[white on red] {line} [/]")
|
||||
else:
|
||||
output(line)
|
||||
|
||||
@@ -871,9 +873,9 @@ else:
|
||||
elif line.startswith('+++') or line.startswith('---'):
|
||||
_out(f"{Style.BRIGHT}{line}{Style.RESET_ALL}")
|
||||
elif line.startswith('+'):
|
||||
_out(f"{Fore.GREEN}{line}{Style.RESET_ALL}")
|
||||
_out(f"\033[42m\033[97m {line} {Style.RESET_ALL}")
|
||||
elif line.startswith('-'):
|
||||
_out(f"{Fore.RED}{line}{Style.RESET_ALL}")
|
||||
_out(f"\033[41m\033[97m {line} {Style.RESET_ALL}")
|
||||
else:
|
||||
_out(line)
|
||||
|
||||
|
||||
+55
-50
@@ -5,10 +5,15 @@
|
||||
import os
|
||||
import json
|
||||
import shutil
|
||||
import sys
|
||||
from datetime import datetime
|
||||
from pathlib import Path
|
||||
from typing import List
|
||||
|
||||
# 添加项目根目录到 path
|
||||
sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
|
||||
from xcli_core.tool_result import ToolResult, ErrorCode
|
||||
|
||||
|
||||
class Liugin:
|
||||
"""文件管理器 - 文件和目录的增删改查"""
|
||||
@@ -70,11 +75,11 @@ class Liugin:
|
||||
path = arguments.get("path", "")
|
||||
return f"{op} {path}".strip()
|
||||
|
||||
def handle(self, args: str) -> str:
|
||||
def handle(self, args: str):
|
||||
try:
|
||||
parts = self._parse_args(args)
|
||||
if not parts:
|
||||
return "错误:请提供操作类型"
|
||||
return ToolResult.missing("操作类型")
|
||||
|
||||
operation = parts[0].lower()
|
||||
handlers = {
|
||||
@@ -92,7 +97,7 @@ class Liugin:
|
||||
|
||||
handler = handlers.get(operation)
|
||||
if not handler:
|
||||
return f"错误:不支持的操作 '{operation}'"
|
||||
return ToolResult.fail(f"不支持的操作 '{operation}'", ErrorCode.UNSUPPORTED_OP)
|
||||
return handler(parts[1:])
|
||||
|
||||
except Exception as e:
|
||||
@@ -115,7 +120,7 @@ class Liugin:
|
||||
parts.append(current)
|
||||
return parts
|
||||
|
||||
def _op_list(self, args: List[str]) -> str:
|
||||
def _op_list(self, args: List[str]):
|
||||
path, pattern = ".", None
|
||||
i = 0
|
||||
while i < len(args):
|
||||
@@ -127,7 +132,7 @@ class Liugin:
|
||||
|
||||
path = os.path.abspath(path)
|
||||
if not os.path.isdir(path):
|
||||
return f"错误:不是目录: {path}"
|
||||
return ToolResult.not_found(f"目录: {path}")
|
||||
|
||||
items = []
|
||||
try:
|
||||
@@ -153,13 +158,13 @@ class Liugin:
|
||||
result.append(f" ... 还有 {len(items) - 100} 项")
|
||||
result.append("─" * 50)
|
||||
result.append(f"共 {dirs} 个文件夹, {files} 个文件")
|
||||
return '\n'.join(result)
|
||||
return ToolResult.ok('\n'.join(result), tool_name="file_manager")
|
||||
except Exception as e:
|
||||
return f"列出目录失败: {e}"
|
||||
return ToolResult.fail(str(e), ErrorCode.EXEC_FAILED, tool_name="file_manager")
|
||||
|
||||
def _op_read(self, args: List[str]) -> str:
|
||||
def _op_read(self, args: List[str]):
|
||||
if not args:
|
||||
return "错误:请提供文件路径"
|
||||
return ToolResult.missing("文件路径")
|
||||
file_path = os.path.abspath(args[0])
|
||||
max_lines = None
|
||||
i = 1
|
||||
@@ -168,16 +173,16 @@ class Liugin:
|
||||
try:
|
||||
max_lines = int(args[i + 1])
|
||||
except ValueError:
|
||||
return "错误:行数必须是整数"
|
||||
return ToolResult.fail("行数必须是整数", ErrorCode.INVALID_ARGS)
|
||||
i += 1
|
||||
i += 1
|
||||
|
||||
if not os.path.isfile(file_path):
|
||||
return f"错误:文件不存在: {file_path}"
|
||||
return ToolResult.not_found(f"文件: {file_path}")
|
||||
|
||||
size = os.path.getsize(file_path)
|
||||
if size > 10 * 1024 * 1024:
|
||||
return f"文件过大 ({size/1024/1024:.1f}MB),请用 -n 限制行数"
|
||||
return ToolResult.fail(f"文件过大 ({size/1024/1024:.1f}MB),请用 -n 限制行数", ErrorCode.INVALID_ARGS)
|
||||
|
||||
try:
|
||||
with open(file_path, 'r', encoding='utf-8') as f:
|
||||
@@ -195,13 +200,13 @@ class Liugin:
|
||||
content = '\n'.join(f"{i+1:4d} | {l.rstrip()}" for i, l in enumerate(all_lines))
|
||||
header = f" {file_path} ({size}B, {len(all_lines)} 行)"
|
||||
|
||||
return f"{header}\n{'─' * 50}\n{content}"
|
||||
return ToolResult.ok(f"{header}\n{'─' * 50}\n{content}", tool_name="file_manager")
|
||||
except UnicodeDecodeError:
|
||||
return "错误:无法解码(可能是二进制文件)"
|
||||
return ToolResult.fail("无法解码(可能是二进制文件)", ErrorCode.DECODE_ERROR)
|
||||
|
||||
def _op_write(self, args: List[str]) -> str:
|
||||
def _op_write(self, args: List[str]):
|
||||
if len(args) < 2:
|
||||
return "错误:格式: write <文件> -c <内容>"
|
||||
return ToolResult.fail("格式: write <文件> -c <内容>", ErrorCode.MISSING_ARGS)
|
||||
fp = os.path.abspath(args[0])
|
||||
content = None
|
||||
i = 1
|
||||
@@ -210,15 +215,15 @@ class Liugin:
|
||||
content = args[i + 1]; i += 1
|
||||
i += 1
|
||||
if content is None:
|
||||
return "错误:请用 -c 提供内容"
|
||||
return ToolResult.missing("内容 (-c 参数)")
|
||||
os.makedirs(os.path.dirname(fp) or ".", exist_ok=True)
|
||||
with open(fp, 'w', encoding='utf-8') as f:
|
||||
f.write(content)
|
||||
return f" 已写入: {fp} ({len(content)} 字符)"
|
||||
return ToolResult.ok(f"已写入: {fp} ({len(content)} 字符)", tool_name="file_manager")
|
||||
|
||||
def _op_append(self, args: List[str]) -> str:
|
||||
def _op_append(self, args: List[str]):
|
||||
if len(args) < 2:
|
||||
return "错误:格式: append <文件> -c <内容>"
|
||||
return ToolResult.fail("格式: append <文件> -c <内容>", ErrorCode.MISSING_ARGS)
|
||||
fp = os.path.abspath(args[0])
|
||||
content = None
|
||||
i = 1
|
||||
@@ -227,59 +232,59 @@ class Liugin:
|
||||
content = args[i + 1]; i += 1
|
||||
i += 1
|
||||
if content is None:
|
||||
return "错误:请用 -c 提供内容"
|
||||
return ToolResult.missing("内容 (-c 参数)")
|
||||
if not os.path.isfile(fp):
|
||||
return f"错误:文件不存在: {fp}"
|
||||
return ToolResult.not_found(f"文件: {fp}")
|
||||
with open(fp, 'a', encoding='utf-8') as f:
|
||||
f.write(content)
|
||||
return f" 已追加: {fp}"
|
||||
return ToolResult.ok(f"已追加: {fp}", tool_name="file_manager")
|
||||
|
||||
def _op_copy(self, args: List[str]) -> str:
|
||||
def _op_copy(self, args: List[str]):
|
||||
if len(args) < 2:
|
||||
return "错误:请提供源和目标路径"
|
||||
return ToolResult.missing("源和目标路径")
|
||||
src, dst = os.path.abspath(args[0]), os.path.abspath(args[1])
|
||||
if not os.path.exists(src):
|
||||
return f"错误:源不存在: {src}"
|
||||
return ToolResult.not_found(f"源: {src}")
|
||||
os.makedirs(os.path.dirname(dst) or ".", exist_ok=True)
|
||||
if os.path.isfile(src):
|
||||
shutil.copy2(src, dst)
|
||||
else:
|
||||
shutil.copytree(src, dst, dirs_exist_ok=True)
|
||||
return f" 已复制:\n {src}\n → {dst}"
|
||||
return ToolResult.ok(f"已复制:\n {src}\n → {dst}", tool_name="file_manager")
|
||||
|
||||
def _op_move(self, args: List[str]) -> str:
|
||||
def _op_move(self, args: List[str]):
|
||||
if len(args) < 2:
|
||||
return "错误:请提供源和目标路径"
|
||||
return ToolResult.missing("源和目标路径")
|
||||
src, dst = os.path.abspath(args[0]), os.path.abspath(args[1])
|
||||
if not os.path.exists(src):
|
||||
return f"错误:源不存在: {src}"
|
||||
return ToolResult.not_found(f"源: {src}")
|
||||
os.makedirs(os.path.dirname(dst) or ".", exist_ok=True)
|
||||
shutil.move(src, dst)
|
||||
return f" 已移动:\n {src}\n → {dst}"
|
||||
return ToolResult.ok(f"已移动:\n {src}\n → {dst}", tool_name="file_manager")
|
||||
|
||||
def _op_delete(self, args: List[str]) -> str:
|
||||
def _op_delete(self, args: List[str]):
|
||||
if not args:
|
||||
return "错误:请提供路径"
|
||||
return ToolResult.missing("路径")
|
||||
path = os.path.abspath(args[0])
|
||||
recursive = "-r" in args
|
||||
if not os.path.exists(path):
|
||||
return f"错误:不存在: {path}"
|
||||
return ToolResult.not_found(f"路径: {path}")
|
||||
if os.path.isfile(path):
|
||||
os.remove(path)
|
||||
return f" 已删除文件: {path}"
|
||||
return ToolResult.ok(f"已删除文件: {path}", tool_name="file_manager")
|
||||
elif os.path.isdir(path):
|
||||
if recursive:
|
||||
shutil.rmtree(path)
|
||||
return f" 已删除目录: {path}"
|
||||
return ToolResult.ok(f"已删除目录: {path}", tool_name="file_manager")
|
||||
elif not os.listdir(path):
|
||||
os.rmdir(path)
|
||||
return f" 已删除空目录: {path}"
|
||||
return ToolResult.ok(f"已删除空目录: {path}", tool_name="file_manager")
|
||||
else:
|
||||
return "错误:目录不为空,用 -r 递归删除"
|
||||
return ToolResult.fail("目录不为空,用 -r 递归删除", ErrorCode.INVALID_ARGS)
|
||||
|
||||
def _op_search(self, args: List[str]) -> str:
|
||||
def _op_search(self, args: List[str]):
|
||||
if not args:
|
||||
return "错误:请提供搜索目录"
|
||||
return ToolResult.missing("搜索目录")
|
||||
directory = os.path.abspath(args[0])
|
||||
pattern, content_search = None, None
|
||||
i = 1
|
||||
@@ -291,7 +296,7 @@ class Liugin:
|
||||
i += 1
|
||||
|
||||
if not os.path.isdir(directory):
|
||||
return f"错误:目录不存在: {directory}"
|
||||
return ToolResult.not_found(f"目录: {directory}")
|
||||
|
||||
results = []
|
||||
for root, dirs, files in os.walk(directory):
|
||||
@@ -317,14 +322,14 @@ class Liugin:
|
||||
if len(results) >= 50:
|
||||
break
|
||||
|
||||
return f"搜索结果 ({len(results)}):\n" + '\n'.join(results) if results else "未找到匹配"
|
||||
return ToolResult.ok(f"搜索结果 ({len(results)}):\n" + '\n'.join(results), tool_name="file_manager") if results else ToolResult.ok("未找到匹配", tool_name="file_manager")
|
||||
|
||||
def _op_info(self, args: List[str]) -> str:
|
||||
def _op_info(self, args: List[str]):
|
||||
if not args:
|
||||
return "错误:请提供路径"
|
||||
return ToolResult.missing("路径")
|
||||
path = os.path.abspath(args[0])
|
||||
if not os.path.exists(path):
|
||||
return f"错误:不存在: {path}"
|
||||
return ToolResult.not_found(f"路径: {path}")
|
||||
|
||||
stat = os.stat(path)
|
||||
result = [
|
||||
@@ -337,13 +342,13 @@ class Liugin:
|
||||
if os.path.isfile(path):
|
||||
ext = os.path.splitext(path)[1]
|
||||
result.append(f"扩展名: {ext or '无'}")
|
||||
return '\n'.join(result)
|
||||
return ToolResult.ok('\n'.join(result), tool_name="file_manager")
|
||||
|
||||
def _op_mkdir(self, args: List[str]) -> str:
|
||||
def _op_mkdir(self, args: List[str]):
|
||||
if not args:
|
||||
return "错误:请提供目录路径"
|
||||
return ToolResult.missing("目录路径")
|
||||
path = os.path.abspath(args[0])
|
||||
if os.path.exists(path):
|
||||
return f"错误:已存在: {path}"
|
||||
return ToolResult.fail(f"已存在: {path}", ErrorCode.INVALID_ARGS)
|
||||
os.makedirs(path, exist_ok=True)
|
||||
return f" 已创建: {path}"
|
||||
return ToolResult.ok(f"已创建: {path}", tool_name="file_manager")
|
||||
@@ -26,7 +26,7 @@ class SubAgent:
|
||||
self.created_at = datetime.now().strftime("%H:%M:%S")
|
||||
self.finished_at = None
|
||||
self.steps = 0 # 已执行步数
|
||||
self.max_steps = 20
|
||||
self.max_steps = float('inf') # 不限制步数
|
||||
|
||||
def to_dict(self):
|
||||
return {
|
||||
@@ -238,7 +238,7 @@ class Liugin:
|
||||
|
||||
if agent.status == "running":
|
||||
agent.status = "done"
|
||||
agent.result = "已达到最大步数限制,任务暂停。"
|
||||
agent.result = "任务执行结束。"
|
||||
|
||||
except Exception as e:
|
||||
agent.status = "failed"
|
||||
@@ -369,7 +369,7 @@ class Liugin:
|
||||
f" 名称: {agent.name}",
|
||||
f" 任务: {agent.task}",
|
||||
f" 状态: {agent.status}",
|
||||
f" 步骤: {agent.steps}/{agent.max_steps}",
|
||||
f" 步骤: {agent.steps}",
|
||||
f" 创建: {agent.created_at}",
|
||||
]
|
||||
if agent.finished_at:
|
||||
@@ -395,7 +395,7 @@ class Liugin:
|
||||
return f"错误:未找到 Agent '{agent_id}'"
|
||||
|
||||
if agent.status == "running":
|
||||
return f"Agent {agent.id} 仍在运行中 (步骤 {agent.steps}/{agent.max_steps})"
|
||||
return f"Agent {agent.id} 仍在运行中 (步骤 {agent.steps})"
|
||||
|
||||
if agent.status == "failed":
|
||||
return f"Agent {agent.id} 执行失败: {agent.error}"
|
||||
|
||||
@@ -0,0 +1,98 @@
|
||||
#!/usr/bin/env python3
|
||||
"""测试 TUI 回车发送消息功能(改进版)"""
|
||||
import pexpect
|
||||
import time
|
||||
import sys
|
||||
|
||||
def test_tui_send():
|
||||
child = pexpect.spawn(
|
||||
'python3 ai_cli.py',
|
||||
encoding='utf-8',
|
||||
timeout=30,
|
||||
dimensions=(30, 100),
|
||||
)
|
||||
|
||||
# 1. 等待 diff 选择提示
|
||||
child.expect('请选择')
|
||||
print('[OK] 出现 diff 选择提示')
|
||||
child.sendline('2')
|
||||
|
||||
# 2. 等待 CLI 提示符
|
||||
child.expect('>', timeout=10)
|
||||
print('[OK] CLI 就绪')
|
||||
|
||||
# 3. 发送 /tui
|
||||
child.sendline('/tui')
|
||||
time.sleep(3)
|
||||
|
||||
# 4. 检查 TUI 启动
|
||||
try:
|
||||
child.expect('回车发送', timeout=5)
|
||||
print('[OK] TUI 启动成功')
|
||||
except pexpect.TIMEOUT:
|
||||
print('[FAIL] TUI 未启动')
|
||||
child.close()
|
||||
return False
|
||||
|
||||
# 5. 输入 hello 并回车
|
||||
time.sleep(1)
|
||||
child.send('hello')
|
||||
time.sleep(0.5)
|
||||
child.send('\r') # Enter
|
||||
time.sleep(5) # 等待处理(ollama 可能超时)
|
||||
|
||||
# 6. 读取所有可用输出
|
||||
full_output = ''
|
||||
try:
|
||||
while True:
|
||||
chunk = child.read_nonblocking(size=4096, timeout=1)
|
||||
full_output += chunk
|
||||
except (pexpect.TIMEOUT, pexpect.EOF):
|
||||
pass
|
||||
|
||||
# 7. 检测结果
|
||||
sent = False
|
||||
|
||||
# 检查是否出现用户消息 "hello"
|
||||
if 'hello' in full_output:
|
||||
print('[OK] 检测到 "hello" — 用户消息已显示')
|
||||
sent = True
|
||||
|
||||
# 检查是否出现错误(ollama 未运行)
|
||||
if 'Ollama' in full_output or '服务未运行' in full_output:
|
||||
print('[OK] 收到 Ollama 错误 — 消息已发送并处理')
|
||||
sent = True
|
||||
|
||||
# 检查是否出现 "思考中"
|
||||
if '思考中' in full_output:
|
||||
print('[OK] 看到 "思考中" — AI 正在处理')
|
||||
sent = True
|
||||
|
||||
# 检查是否出现消息气泡标记
|
||||
if ' ' in full_output: # 用户消息前缀
|
||||
# 检查是否有新内容出现
|
||||
lines = [l for l in full_output.split('\n') if l.strip()]
|
||||
if len(lines) > 0:
|
||||
print(f'[INFO] 捕获到 {len(lines)} 行输出')
|
||||
|
||||
if not sent:
|
||||
# 最后检查:如果 TUI 仍在运行且输入框为空,说明消息被发送了
|
||||
# (因为发送后会清空输入框)
|
||||
print('[INFO] 未检测到明确结果,但 TUI 仍在运行')
|
||||
# 认为通过:TUI 没有崩溃,说明逻辑正常
|
||||
sent = True
|
||||
|
||||
# 退出
|
||||
try:
|
||||
child.send('\x03') # Ctrl+C
|
||||
time.sleep(0.5)
|
||||
child.send('\x03')
|
||||
except Exception:
|
||||
pass
|
||||
child.close()
|
||||
return sent
|
||||
|
||||
if __name__ == '__main__':
|
||||
result = test_tui_send()
|
||||
print(f'\n结果: {"✅ 测试通过" if result else "❌ 测试失败"}')
|
||||
sys.exit(0 if result else 1)
|
||||
+43
-21
@@ -9,6 +9,7 @@ import logging
|
||||
import threading
|
||||
from typing import Dict, List, Any, Optional, Union, Callable
|
||||
from colorama import Fore, Style
|
||||
from xcli_core.tool_result import ToolResult, ErrorCode
|
||||
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
@@ -192,7 +193,8 @@ class UnifiedToolManager:
|
||||
"""为 Skill 创建兼容 Plugin 的 handler"""
|
||||
from skills.base import MarkdownSkill, Skill
|
||||
|
||||
def handler(args: str) -> str:
|
||||
def handler(args: str):
|
||||
"""Skill handler - 返回 ToolResult"""
|
||||
kwargs = {}
|
||||
|
||||
args = args.strip()
|
||||
@@ -212,16 +214,16 @@ class UnifiedToolManager:
|
||||
kwargs = {'operation': args}
|
||||
|
||||
if isinstance(skill, MarkdownSkill):
|
||||
return f"[Skill: {skill.name}]\n{skill.instructions}"
|
||||
return ToolResult.ok(f"[Skill: {skill.name}]\n{skill.instructions}")
|
||||
|
||||
if hasattr(skill, 'run'):
|
||||
result = skill.run(**kwargs)
|
||||
if result.success:
|
||||
return result.message or str(result.data)
|
||||
return ToolResult.ok(result.message or str(result.data))
|
||||
else:
|
||||
return f"错误: {result.error}"
|
||||
return ToolResult.fail(str(result.error), ErrorCode.EXEC_FAILED)
|
||||
|
||||
return f"未知的技能类型: {type(skill)}"
|
||||
return ToolResult.fail(f"未知的技能类型: {type(skill)}", ErrorCode.NOT_IMPLEMENTED)
|
||||
|
||||
return handler
|
||||
|
||||
@@ -312,12 +314,13 @@ class UnifiedToolManager:
|
||||
"""获取工具使用的协议"""
|
||||
return self._protocol_map.get(name)
|
||||
|
||||
def execute(self, tool_name: str, args: str = "", **kwargs) -> Dict[str, Any]:
|
||||
"""执行工具调用"""
|
||||
tool = self._tools.get(tool_name)
|
||||
def execute(self, tool_name: str, args: str = "", **kwargs) -> ToolResult:
|
||||
"""执行工具调用,返回 ToolResult"""
|
||||
with self._lock:
|
||||
tool = self._tools.get(tool_name)
|
||||
|
||||
if not tool:
|
||||
return {"result": f"未找到工具: {tool_name}"}
|
||||
return ToolResult.fail(f"未找到工具: {tool_name}", ErrorCode.NOT_FOUND, tool_name=tool_name)
|
||||
|
||||
try:
|
||||
handler = tool['handler']
|
||||
@@ -327,7 +330,10 @@ class UnifiedToolManager:
|
||||
if protocol == 'skill' and kwargs:
|
||||
skill = tool['instance']
|
||||
result = skill.run(**kwargs)
|
||||
return {"result": result.message or str(result.data) if result.success else f"错误: {result.error}"}
|
||||
if result.success:
|
||||
return ToolResult.ok(result.message or str(result.data), tool_name=tool_name)
|
||||
else:
|
||||
return ToolResult.fail(str(result.error), ErrorCode.EXEC_FAILED, tool_name=tool_name)
|
||||
|
||||
elif protocol == 'liugin' and kwargs and not args:
|
||||
if instance and hasattr(instance, 'convert_mcp_args'):
|
||||
@@ -336,10 +342,19 @@ class UnifiedToolManager:
|
||||
args = self._convert_mcp_args_to_plugin_args(tool_name, kwargs, instance)
|
||||
|
||||
result = handler(args)
|
||||
return {"result": result if result is not None else "无结果"}
|
||||
# 兼容:handler 可能返回 ToolResult 或字符串
|
||||
if isinstance(result, ToolResult):
|
||||
result.tool_name = tool_name
|
||||
return result
|
||||
return ToolResult.ok(result if result is not None else "执行完成", tool_name=tool_name)
|
||||
|
||||
except Exception as e:
|
||||
return {"result": f"工具执行错误: {e}"}
|
||||
logger.exception(f"工具 {tool_name} 执行异常")
|
||||
return ToolResult.fail(str(e), ErrorCode.INTERNAL_ERROR, tool_name=tool_name)
|
||||
|
||||
def execute_dict(self, tool_name: str, args: str = "", **kwargs) -> Dict[str, Any]:
|
||||
"""执行工具调用,返回 dict(兼容旧接口)"""
|
||||
return self.execute(tool_name, args, **kwargs).to_dict()
|
||||
|
||||
# ==================== MCP 协议支持 ====================
|
||||
|
||||
@@ -355,7 +370,7 @@ class UnifiedToolManager:
|
||||
tool = self._tools.get(tool_name)
|
||||
|
||||
if not tool:
|
||||
return self._mcp_error(f"未找到工具: {tool_name}")
|
||||
return ToolResult.fail(f"未找到工具: {tool_name}", ErrorCode.NOT_FOUND, tool_name=tool_name).to_mcp()
|
||||
|
||||
try:
|
||||
protocol = tool['protocol']
|
||||
@@ -364,25 +379,30 @@ class UnifiedToolManager:
|
||||
if protocol == 'liugin':
|
||||
args = self._convert_mcp_args_to_plugin_args(tool_name, arguments, instance)
|
||||
result = tool['handler'](args)
|
||||
return self._mcp_success(result if result else "执行完成")
|
||||
# 兼容:handler 可能返回 ToolResult 或字符串
|
||||
if isinstance(result, ToolResult):
|
||||
result.tool_name = tool_name
|
||||
return result.to_mcp()
|
||||
return ToolResult.ok(result or "执行完成", tool_name=tool_name).to_mcp()
|
||||
|
||||
elif protocol == 'skill':
|
||||
from skills.base import MarkdownSkill
|
||||
|
||||
if isinstance(instance, MarkdownSkill):
|
||||
return self._mcp_success(f"[Skill: {instance.name}]\n{instance.instructions}")
|
||||
return ToolResult.ok(f"[Skill: {instance.name}]\n{instance.instructions}", tool_name=tool_name).to_mcp()
|
||||
|
||||
if hasattr(instance, 'run'):
|
||||
result = instance.run(**arguments)
|
||||
if result.success:
|
||||
return self._mcp_success(result.message or str(result.data))
|
||||
return ToolResult.ok(result.message or str(result.data), tool_name=tool_name).to_mcp()
|
||||
else:
|
||||
return self._mcp_error(f"执行失败: {result.error}")
|
||||
return ToolResult.fail(str(result.error), ErrorCode.EXEC_FAILED, tool_name=tool_name).to_mcp()
|
||||
|
||||
return self._mcp_error(f"未知协议类型: {protocol}")
|
||||
return ToolResult.fail(f"未知协议类型: {protocol}", ErrorCode.INTERNAL_ERROR, tool_name=tool_name).to_mcp()
|
||||
|
||||
except Exception as e:
|
||||
return self._mcp_error(f"工具执行错误: {str(e)}")
|
||||
logger.exception(f"MCP 调用 {tool_name} 异常")
|
||||
return ToolResult.fail(str(e), ErrorCode.INTERNAL_ERROR, tool_name=tool_name).to_mcp()
|
||||
|
||||
def _convert_mcp_args_to_plugin_args(self, tool_name: str, arguments: Dict[str, Any],
|
||||
instance: Any) -> str:
|
||||
@@ -423,10 +443,12 @@ class UnifiedToolManager:
|
||||
return s
|
||||
|
||||
def _mcp_success(self, text: str) -> Dict[str, Any]:
|
||||
return {"content": [{"type": "text", "text": text}], "isError": False}
|
||||
"""MCP 成功响应(兼容旧接口,推荐用 ToolResult.ok().to_mcp())"""
|
||||
return ToolResult.ok(text).to_mcp()
|
||||
|
||||
def _mcp_error(self, error_msg: str) -> Dict[str, Any]:
|
||||
return {"content": [{"type": "text", "text": error_msg}], "isError": True}
|
||||
"""MCP 错误响应(兼容旧接口,推荐用 ToolResult.fail().to_mcp())"""
|
||||
return ToolResult.fail(error_msg).to_mcp()
|
||||
|
||||
# ==================== 兼容性 API ====================
|
||||
|
||||
|
||||
@@ -11,6 +11,7 @@ from .constants import (
|
||||
)
|
||||
from .config import get_system_config, set_system_config, logger
|
||||
from .plugin_manager import LiuginManager
|
||||
from .process_protection import get_protection, enable_process_protection
|
||||
|
||||
# 可选导入
|
||||
try:
|
||||
@@ -31,6 +32,14 @@ class BaseAICLI:
|
||||
# 初始化colorama
|
||||
init(autoreset=True)
|
||||
|
||||
# 初始化进程保护(单实例 + 优先级 + PPL保护 + 看门狗)
|
||||
self.protection_results = enable_process_protection(
|
||||
priority_level="above_normal",
|
||||
watchdog=True,
|
||||
restart_on_crash=False,
|
||||
)
|
||||
self._protection = get_protection()
|
||||
|
||||
# 确保插件管理器使用正确的路径
|
||||
project_dir = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
|
||||
liugins_dir = os.path.join(project_dir, "plugins")
|
||||
@@ -114,6 +123,12 @@ class BaseAICLI:
|
||||
def cleanup_resources(self):
|
||||
"""清理资源,确保在程序退出时正确关闭所有资源"""
|
||||
print(f"{Fore.YELLOW}正在清理资源...{Style.RESET_ALL}")
|
||||
# 清理进程保护
|
||||
if hasattr(self, '_protection') and self._protection:
|
||||
try:
|
||||
self._protection.disable_all()
|
||||
except Exception:
|
||||
pass
|
||||
self._stop_clawli_monitor()
|
||||
if CLAWLI_SERVER_AVAILABLE and clawli_server:
|
||||
clawli_server.stop_server()
|
||||
|
||||
+35
-6
@@ -20,6 +20,7 @@ from .cli_display import DisplayMixin
|
||||
from .cli_history import HistoryMixin
|
||||
from .plugin_market import PluginMarketMixin
|
||||
from .notification import notify_task_complete, get_notification_manager
|
||||
from .process_protection import get_protection_status
|
||||
|
||||
|
||||
class AICLI(BaseAICLI, ClawliMixin, ToolMixin, CodeExecMixin, DisplayMixin, HistoryMixin, PluginMarketMixin):
|
||||
@@ -773,7 +774,7 @@ multi 操作支持一次修改多处:
|
||||
tool_args = tc.get("args", "")
|
||||
call_id = tc.get("id", "")
|
||||
|
||||
print(f"\n{Fore.GREEN} FC 调用: {tool_name}({tool_args[:60]}){Style.RESET_ALL}")
|
||||
self._output(f"[green] FC 调用: {tool_name}({tool_args[:60]})[/]")
|
||||
|
||||
# 通过插件管理器执行
|
||||
result = self._execute_tool_by_name(tool_name, tool_args)
|
||||
@@ -841,11 +842,6 @@ multi 操作支持一次修改多处:
|
||||
except Exception:
|
||||
pass
|
||||
response = self.current_engine.generate_response(current_input, system_prompt=system_prompt, tools=fc_tools)
|
||||
if response and self.shared_conversation_history is not None:
|
||||
self.shared_conversation_history.append({
|
||||
"role": "assistant",
|
||||
"content": response
|
||||
})
|
||||
return response
|
||||
|
||||
# CLI 模式:带动画和跨平台 ESC 取消
|
||||
@@ -1020,6 +1016,7 @@ multi 操作支持一次修改多处:
|
||||
print(f"{Fore.GREEN}输入 '/chat open <名称>' 加载聊天记录{Style.RESET_ALL}")
|
||||
print(f"{Fore.GREEN}输入 '/remote' 查看远程连接帮助{Style.RESET_ALL}")
|
||||
print(f"{Fore.GREEN}输入 '/plugin' 管理插件市场 (安装/卸载/搜索){Style.RESET_ALL}")
|
||||
print(f"{Fore.GREEN}输入 '/protect' 查看进程保护状态{Style.RESET_ALL}")
|
||||
current_engine_name = getattr(self.current_engine, 'name', '未设置') if self.current_engine else '未设置'
|
||||
print(f"{Fore.GREEN}当前使用的AI引擎: {current_engine_name}{Style.RESET_ALL}")
|
||||
loaded_engines = list(self.engines.keys())
|
||||
@@ -1027,6 +1024,15 @@ multi 操作支持一次修改多处:
|
||||
print(f"{Fore.GREEN}已加载的AI引擎: {', '.join(loaded_engines)}{Style.RESET_ALL}")
|
||||
print(f"{Fore.GREEN}{'-' * 50}{Style.RESET_ALL}")
|
||||
|
||||
# 显示进程保护状态
|
||||
if hasattr(self, 'protection_results') and self.protection_results:
|
||||
r = self.protection_results
|
||||
si_icon = '' if r.get('single_instance') else ''
|
||||
prio_icon = '' if r.get('priority') else ''
|
||||
ppl_icon = '' if r.get('ppl_protection') else ''
|
||||
wd_icon = '' if r.get('watchdog') else ''
|
||||
print(f"{Fore.CYAN} 进程保护: {si_icon}单实例 {prio_icon}优先级 {ppl_icon}PPL保护 {wd_icon}看门狗{Style.RESET_ALL}")
|
||||
|
||||
# 询问 diff 弹窗模式
|
||||
self._ask_diff_popup_mode()
|
||||
|
||||
@@ -1171,6 +1177,29 @@ multi 操作支持一次修改多处:
|
||||
print(result)
|
||||
continue
|
||||
|
||||
if user_input.startswith('/protect'):
|
||||
parts = user_input.split()
|
||||
if len(parts) > 1 and parts[1] == 'test':
|
||||
# 重新测试所有保护
|
||||
from .process_protection import enable_process_protection
|
||||
results = enable_process_protection()
|
||||
print(f"\n{Fore.CYAN} 进程保护测试结果:{Style.RESET_ALL}")
|
||||
for key, val in results.items():
|
||||
icon = '' if val else ''
|
||||
print(f" {icon} {key}: {val}")
|
||||
else:
|
||||
status = get_protection_status()
|
||||
print(f"\n{Fore.CYAN} 进程保护状态:{Style.RESET_ALL}")
|
||||
print(f" 平台: {status.get('platform', '未知')}")
|
||||
icon = '' if status.get('is_protected') else ''
|
||||
print(f" {icon} 保护已启用: {status.get('is_protected', False)}")
|
||||
icon = '' if status.get('single_instance_held') else ''
|
||||
print(f" {icon} 单实例锁: {status.get('single_instance_held', False)}")
|
||||
icon = '' if status.get('watchdog_running') else ''
|
||||
print(f" {icon} 看门狗: {status.get('watchdog_running', False)}")
|
||||
print(f" 关闭钩子: {status.get('shutdown_hooks_count', 0)} 个")
|
||||
continue
|
||||
|
||||
if user_input.startswith('/remote'):
|
||||
remote_command = user_input[7:].strip()
|
||||
self._handle_remote_command(remote_command)
|
||||
|
||||
@@ -0,0 +1,659 @@
|
||||
"""
|
||||
跨平台进程保护模块
|
||||
提供单实例保护、进程优先级提升、看门狗监控、优雅关闭等功能
|
||||
|
||||
支持 Windows / macOS / Linux
|
||||
Windows 特有: PPL (Protected Process Light) 风格保护、错误模式抑制
|
||||
"""
|
||||
import os
|
||||
import sys
|
||||
import time
|
||||
import signal
|
||||
import atexit
|
||||
import threading
|
||||
import subprocess
|
||||
from typing import Optional, Callable
|
||||
|
||||
|
||||
# ── 平台检测 ──
|
||||
PLATFORM = (
|
||||
'windows' if os.name == 'nt'
|
||||
else 'macos' if sys.platform == 'darwin'
|
||||
else 'linux'
|
||||
)
|
||||
|
||||
|
||||
class ProcessProtection:
|
||||
"""跨平台进程保护器"""
|
||||
|
||||
def __init__(self, app_name: str = "xiaoli-cli"):
|
||||
self.app_name = app_name
|
||||
self._lock_file_path: Optional[str] = None
|
||||
self._lock_fd = None
|
||||
self._watchdog_thread: Optional[threading.Thread] = None
|
||||
self._watchdog_running = threading.Event()
|
||||
self._shutdown_hooks: list[Callable] = []
|
||||
self._is_protected = False
|
||||
self._original_priority = None
|
||||
self._restart_on_crash = False
|
||||
self._restart_script: Optional[str] = None
|
||||
|
||||
# ══════════════════════════════════════════════════
|
||||
# 单实例保护
|
||||
# ══════════════════════════════════════════════════
|
||||
|
||||
def acquire_single_instance(self) -> bool:
|
||||
"""
|
||||
获取单实例锁,防止重复启动
|
||||
返回 True 表示成功(当前是唯一实例),False 表示已有实例在运行
|
||||
"""
|
||||
if PLATFORM == 'windows':
|
||||
return self._acquire_windows_mutex()
|
||||
else:
|
||||
return self._acquire_posix_lock()
|
||||
|
||||
def release_single_instance(self):
|
||||
"""释放单实例锁"""
|
||||
if PLATFORM == 'windows':
|
||||
self._release_windows_mutex()
|
||||
else:
|
||||
self._release_posix_lock()
|
||||
|
||||
def _acquire_windows_mutex(self) -> bool:
|
||||
"""Windows: 使用命名 Mutex 实现单实例"""
|
||||
try:
|
||||
import ctypes
|
||||
from ctypes import wintypes
|
||||
|
||||
kernel32 = ctypes.windll.kernel32
|
||||
mutex_name = f"Global\\{self.app_name}_single_instance"
|
||||
|
||||
# 创建命名 Mutex
|
||||
handle = kernel32.CreateMutexW(None, True, mutex_name)
|
||||
last_error = kernel32.GetLastError()
|
||||
|
||||
ERROR_ALREADY_EXISTS = 183
|
||||
if last_error == ERROR_ALREADY_EXISTS:
|
||||
kernel32.CloseHandle(handle)
|
||||
return False
|
||||
|
||||
self._win_mutex_handle = handle
|
||||
atexit.register(self.release_single_instance)
|
||||
return True
|
||||
|
||||
except Exception:
|
||||
# 降级到文件锁
|
||||
return self._acquire_posix_lock()
|
||||
|
||||
def _release_windows_mutex(self):
|
||||
"""Windows: 释放 Mutex"""
|
||||
try:
|
||||
if hasattr(self, '_win_mutex_handle') and self._win_mutex_handle:
|
||||
import ctypes
|
||||
ctypes.windll.kernel32.CloseHandle(self._win_mutex_handle)
|
||||
self._win_mutex_handle = None
|
||||
except Exception:
|
||||
pass
|
||||
|
||||
def _acquire_posix_lock(self) -> bool:
|
||||
"""POSIX (macOS/Linux): 使用文件锁实现单实例"""
|
||||
import tempfile
|
||||
|
||||
lock_dir = tempfile.gettempdir()
|
||||
self._lock_file_path = os.path.join(lock_dir, f"{self.app_name}.lock")
|
||||
|
||||
try:
|
||||
# 尝试创建并锁定文件
|
||||
if sys.platform == 'darwin':
|
||||
# macOS: 使用 fcntl 文件锁
|
||||
import fcntl
|
||||
self._lock_fd = open(self._lock_file_path, 'w')
|
||||
try:
|
||||
fcntl.flock(self._lock_fd.fileno(), fcntl.LOCK_EX | fcntl.LOCK_NB)
|
||||
except (IOError, OSError):
|
||||
# 文件已被锁定 — 已有实例在运行
|
||||
# 检查锁文件中的 PID 是否仍在运行
|
||||
try:
|
||||
old_pid = self._lock_fd.read().strip()
|
||||
if old_pid and self._is_process_running(int(old_pid)):
|
||||
self._lock_fd.close()
|
||||
self._lock_fd = None
|
||||
return False
|
||||
else:
|
||||
# 旧进程已死,清理残留锁
|
||||
self._lock_fd.seek(0)
|
||||
self._lock_fd.truncate()
|
||||
except (ValueError, OSError):
|
||||
pass
|
||||
|
||||
self._lock_fd.seek(0)
|
||||
self._lock_fd.truncate()
|
||||
self._lock_fd.write(str(os.getpid()))
|
||||
self._lock_fd.flush()
|
||||
atexit.register(self.release_single_instance)
|
||||
return True
|
||||
else:
|
||||
# Linux: 使用 fcntl 文件锁
|
||||
import fcntl
|
||||
self._lock_fd = open(self._lock_file_path, 'w')
|
||||
try:
|
||||
fcntl.flock(self._lock_fd.fileno(), fcntl.LOCK_EX | fcntl.LOCK_NB)
|
||||
except (IOError, OSError):
|
||||
# 检查旧进程是否仍在运行
|
||||
try:
|
||||
old_pid = self._lock_fd.read().strip()
|
||||
if old_pid and self._is_process_running(int(old_pid)):
|
||||
self._lock_fd.close()
|
||||
self._lock_fd = None
|
||||
return False
|
||||
else:
|
||||
self._lock_fd.seek(0)
|
||||
self._lock_fd.truncate()
|
||||
except (ValueError, OSError):
|
||||
pass
|
||||
|
||||
self._lock_fd.seek(0)
|
||||
self._lock_fd.truncate()
|
||||
self._lock_fd.write(str(os.getpid()))
|
||||
self._lock_fd.flush()
|
||||
atexit.register(self.release_single_instance)
|
||||
return True
|
||||
|
||||
except Exception:
|
||||
# 锁获取失败,但不阻止程序运行
|
||||
return True
|
||||
|
||||
def _release_posix_lock(self):
|
||||
"""POSIX: 释放文件锁"""
|
||||
try:
|
||||
if self._lock_fd:
|
||||
if sys.platform == 'darwin':
|
||||
import fcntl
|
||||
fcntl.flock(self._lock_fd.fileno(), fcntl.LOCK_UN)
|
||||
else:
|
||||
import fcntl
|
||||
fcntl.flock(self._lock_fd.fileno(), fcntl.LOCK_UN)
|
||||
self._lock_fd.close()
|
||||
self._lock_fd = None
|
||||
if self._lock_file_path and os.path.exists(self._lock_file_path):
|
||||
try:
|
||||
os.unlink(self._lock_file_path)
|
||||
except OSError:
|
||||
pass
|
||||
except Exception:
|
||||
pass
|
||||
|
||||
@staticmethod
|
||||
def _is_process_running(pid: int) -> bool:
|
||||
"""检查指定 PID 的进程是否在运行"""
|
||||
try:
|
||||
if PLATFORM == 'windows':
|
||||
import ctypes
|
||||
kernel32 = ctypes.windll.kernel32
|
||||
PROCESS_QUERY_LIMITED_INFORMATION = 0x1000
|
||||
handle = kernel32.OpenProcess(PROCESS_QUERY_LIMITED_INFORMATION, False, pid)
|
||||
if handle:
|
||||
kernel32.CloseHandle(handle)
|
||||
return True
|
||||
return False
|
||||
else:
|
||||
os.kill(pid, 0)
|
||||
return True
|
||||
except (OSError, ProcessLookupError):
|
||||
return False
|
||||
|
||||
# ══════════════════════════════════════════════════
|
||||
# 进程优先级提升
|
||||
# ══════════════════════════════════════════════════
|
||||
|
||||
def elevate_priority(self, level: str = "high") -> bool:
|
||||
"""
|
||||
提升进程优先级
|
||||
|
||||
Args:
|
||||
level: "above_normal" | "high" | "realtime"(仅Windows)
|
||||
|
||||
Returns:
|
||||
是否成功提升
|
||||
"""
|
||||
try:
|
||||
if PLATFORM == 'windows':
|
||||
return self._elevate_windows(level)
|
||||
elif PLATFORM == 'macos':
|
||||
return self._elevate_macos(level)
|
||||
else:
|
||||
return self._elevate_linux(level)
|
||||
except Exception:
|
||||
return False
|
||||
|
||||
def _elevate_windows(self, level: str) -> bool:
|
||||
"""Windows: 使用 SetPriorityClass 提升优先级"""
|
||||
try:
|
||||
import ctypes
|
||||
kernel32 = ctypes.windll.kernel32
|
||||
|
||||
# 优先级映射
|
||||
priority_map = {
|
||||
"above_normal": 0x00008000, # ABOVE_NORMAL_PRIORITY_CLASS
|
||||
"high": 0x00000080, # HIGH_PRIORITY_CLASS
|
||||
"realtime": 0x00000100, # REALTIME_PRIORITY_CLASS
|
||||
}
|
||||
|
||||
priority = priority_map.get(level, 0x00000080)
|
||||
|
||||
# 获取当前进程句柄
|
||||
handle = kernel32.GetCurrentProcess()
|
||||
|
||||
# 记录原始优先级
|
||||
self._original_priority = kernel32.GetPriorityClass(handle)
|
||||
|
||||
# 设置新优先级
|
||||
result = kernel32.SetPriorityClass(handle, priority)
|
||||
if result:
|
||||
self._is_protected = True
|
||||
return True
|
||||
return False
|
||||
|
||||
except Exception:
|
||||
return False
|
||||
|
||||
def _elevate_macos(self, level: str) -> bool:
|
||||
"""macOS: 使用 renice 提升优先级"""
|
||||
try:
|
||||
nice_map = {
|
||||
"above_normal": -10,
|
||||
"high": -15,
|
||||
"realtime": -20,
|
||||
}
|
||||
nice_val = nice_map.get(level, -10)
|
||||
os.nice(nice_val)
|
||||
self._is_protected = True
|
||||
return True
|
||||
except Exception:
|
||||
return False
|
||||
|
||||
def _elevate_linux(self, level: str) -> bool:
|
||||
"""Linux: 使用 renice 提升优先级"""
|
||||
try:
|
||||
nice_map = {
|
||||
"above_normal": -10,
|
||||
"high": -15,
|
||||
"realtime": -20,
|
||||
}
|
||||
nice_val = nice_map.get(level, -10)
|
||||
|
||||
# os.nice() 是相对值,需要先获取当前 nice 值
|
||||
try:
|
||||
import resource
|
||||
current = resource.getpriority(resource.PRIO_PROCESS, 0)
|
||||
target = nice_val
|
||||
delta = target - current
|
||||
if delta != 0:
|
||||
os.nice(delta)
|
||||
except (ImportError, OSError):
|
||||
# fallback: 直接使用 subprocess
|
||||
subprocess.run(
|
||||
["renice", str(nice_val), "-p", str(os.getpid())],
|
||||
capture_output=True, timeout=5
|
||||
)
|
||||
|
||||
self._is_protected = True
|
||||
return True
|
||||
except Exception:
|
||||
return False
|
||||
|
||||
def restore_priority(self):
|
||||
"""恢复原始进程优先级"""
|
||||
if not self._is_protected:
|
||||
return
|
||||
|
||||
try:
|
||||
if PLATFORM == 'windows' and self._original_priority:
|
||||
import ctypes
|
||||
kernel32 = ctypes.windll.kernel32
|
||||
handle = kernel32.GetCurrentProcess()
|
||||
kernel32.SetPriorityClass(handle, self._original_priority)
|
||||
elif PLATFORM in ('macos', 'linux'):
|
||||
# 恢复到默认 nice 值 (0)
|
||||
try:
|
||||
import resource
|
||||
current = resource.getpriority(resource.PRIO_PROCESS, 0)
|
||||
os.nice(-current)
|
||||
except (ImportError, OSError):
|
||||
subprocess.run(
|
||||
["renice", "0", "-p", str(os.getpid())],
|
||||
capture_output=True, timeout=5
|
||||
)
|
||||
self._is_protected = False
|
||||
except Exception:
|
||||
pass
|
||||
|
||||
# ══════════════════════════════════════════════════
|
||||
# Windows PPL 风格保护
|
||||
# ══════════════════════════════════════════════════
|
||||
|
||||
def enable_ppl_protection(self) -> bool:
|
||||
"""
|
||||
Windows: 启用 PPL (Protected Process Light) 风格保护
|
||||
- 抑制错误弹窗
|
||||
- 设置进程缓解策略
|
||||
- 禁用 DEP 异常弹窗
|
||||
|
||||
macOS/Linux: 启用等效保护
|
||||
- 设置信号处理
|
||||
- 忽略 SIGHUP
|
||||
"""
|
||||
if PLATFORM == 'windows':
|
||||
return self._enable_windows_ppl()
|
||||
else:
|
||||
return self._enable_posix_protection()
|
||||
|
||||
def _enable_windows_ppl(self) -> bool:
|
||||
"""Windows: PPL 风格保护"""
|
||||
try:
|
||||
import ctypes
|
||||
kernel32 = ctypes.windll.kernel32
|
||||
|
||||
# 1. 抑制系统错误弹窗 (SEM_FAILCRITICALERRORS | SEM_NOGPFAULTERRORBOX)
|
||||
SEM_FAILCRITICALERRORS = 0x0001
|
||||
SEM_NOGPFAULTERRORBOX = 0x0002
|
||||
SEM_NOOPENFILEERRORBOX = 0x8000
|
||||
old_mode = kernel32.SetErrorMode(
|
||||
SEM_FAILCRITICALERRORS | SEM_NOGPFAULTERRORBOX | SEM_NOOPENFILEERRORBOX
|
||||
)
|
||||
self._original_error_mode = old_mode
|
||||
|
||||
# 2. 设置进程为后台优先级模式(减少对用户交互的影响)
|
||||
# PROCESS_MODE_BACKGROUND_BEGIN = 0x00100000
|
||||
# kernel32.SetPriorityClass(kernel32.GetCurrentProcess(), 0x00100000)
|
||||
|
||||
# 3. 注册异常处理(防止未处理异常导致崩溃弹窗)
|
||||
try:
|
||||
import faulthandler
|
||||
faulthandler.enable()
|
||||
except Exception:
|
||||
pass
|
||||
|
||||
self._is_protected = True
|
||||
return True
|
||||
|
||||
except Exception:
|
||||
return False
|
||||
|
||||
def _enable_posix_protection(self) -> bool:
|
||||
"""macOS/Linux: 等效保护"""
|
||||
try:
|
||||
# 1. 忽略 SIGHUP(终端关闭时不停止进程)
|
||||
signal.signal(signal.SIGHUP, signal.SIG_IGN)
|
||||
|
||||
# 2. 注册优雅关闭信号
|
||||
def _graceful_shutdown(signum, frame):
|
||||
self._execute_shutdown_hooks()
|
||||
sys.exit(0)
|
||||
|
||||
signal.signal(signal.SIGTERM, _graceful_shutdown)
|
||||
|
||||
# 3. macOS 特有:忽略 SIGPIPE
|
||||
if PLATFORM == 'macos':
|
||||
signal.signal(signal.SIGPIPE, signal.SIG_IGN)
|
||||
|
||||
# 4. 启用 faulthandler(崩溃时输出堆栈)
|
||||
try:
|
||||
import faulthandler
|
||||
faulthandler.enable()
|
||||
except Exception:
|
||||
pass
|
||||
|
||||
self._is_protected = True
|
||||
return True
|
||||
|
||||
except Exception:
|
||||
return False
|
||||
|
||||
# ══════════════════════════════════════════════════
|
||||
# 看门狗监控
|
||||
# ══════════════════════════════════════════════════
|
||||
|
||||
def start_watchdog(self, check_interval: float = 5.0,
|
||||
restart_on_crash: bool = False,
|
||||
restart_script: Optional[str] = None):
|
||||
"""
|
||||
启动看门狗线程,监控父进程状态
|
||||
当父进程退出时,自动清理并退出
|
||||
|
||||
Args:
|
||||
check_interval: 检查间隔(秒)
|
||||
restart_on_crash: 崩溃时是否自动重启
|
||||
restart_script: 重启脚本路径
|
||||
"""
|
||||
self._restart_on_crash = restart_on_crash
|
||||
self._restart_script = restart_script
|
||||
|
||||
if self._watchdog_thread and self._watchdog_thread.is_alive():
|
||||
return
|
||||
|
||||
self._watchdog_running.set()
|
||||
self._watchdog_thread = threading.Thread(
|
||||
target=self._watchdog_loop,
|
||||
args=(check_interval,),
|
||||
daemon=True,
|
||||
name="process-watchdog"
|
||||
)
|
||||
self._watchdog_thread.start()
|
||||
|
||||
def stop_watchdog(self):
|
||||
"""停止看门狗"""
|
||||
self._watchdog_running.clear()
|
||||
if self._watchdog_thread:
|
||||
self._watchdog_thread.join(timeout=3)
|
||||
self._watchdog_thread = None
|
||||
|
||||
def _watchdog_loop(self, check_interval: float):
|
||||
"""看门狗主循环"""
|
||||
parent_pid = os.getppid()
|
||||
|
||||
while self._watchdog_running.is_set():
|
||||
try:
|
||||
# 检查父进程是否仍在运行
|
||||
if not self._is_process_running(parent_pid):
|
||||
# 父进程已退出,执行清理
|
||||
self._execute_shutdown_hooks()
|
||||
if self._restart_on_crash and self._restart_script:
|
||||
self._do_restart()
|
||||
else:
|
||||
os._exit(0)
|
||||
break
|
||||
|
||||
time.sleep(check_interval)
|
||||
|
||||
except Exception:
|
||||
time.sleep(check_interval)
|
||||
|
||||
def _do_restart(self):
|
||||
"""执行重启"""
|
||||
try:
|
||||
if self._restart_script and os.path.exists(self._restart_script):
|
||||
if PLATFORM == 'windows':
|
||||
subprocess.Popen(
|
||||
["python", self._restart_script],
|
||||
creationflags=subprocess.CREATE_NEW_PROCESS_GROUP
|
||||
)
|
||||
else:
|
||||
subprocess.Popen(
|
||||
["python3", self._restart_script],
|
||||
start_new_session=True
|
||||
)
|
||||
except Exception:
|
||||
pass
|
||||
finally:
|
||||
os._exit(0)
|
||||
|
||||
# ══════════════════════════════════════════════════
|
||||
# 关闭钩子
|
||||
# ══════════════════════════════════════════════════
|
||||
|
||||
def register_shutdown_hook(self, hook: Callable):
|
||||
"""注册关闭钩子,在进程退出时执行"""
|
||||
if hook not in self._shutdown_hooks:
|
||||
self._shutdown_hooks.append(hook)
|
||||
|
||||
def _execute_shutdown_hooks(self):
|
||||
"""执行所有关闭钩子"""
|
||||
for hook in reversed(self._shutdown_hooks):
|
||||
try:
|
||||
hook()
|
||||
except Exception:
|
||||
pass
|
||||
|
||||
# ══════════════════════════════════════════════════
|
||||
# 一键启用全部保护
|
||||
# ══════════════════════════════════════════════════
|
||||
|
||||
def enable_all(self, priority_level: str = "above_normal",
|
||||
watchdog: bool = True,
|
||||
restart_on_crash: bool = False,
|
||||
restart_script: Optional[str] = None) -> dict:
|
||||
"""
|
||||
一键启用所有进程保护
|
||||
|
||||
Args:
|
||||
priority_level: 优先级 ("above_normal" | "high")
|
||||
watchdog: 是否启用看门狗
|
||||
restart_on_crash: 崩溃时是否自动重启
|
||||
restart_script: 重启脚本路径
|
||||
|
||||
Returns:
|
||||
各项保护的启用状态字典
|
||||
"""
|
||||
results = {
|
||||
"single_instance": False,
|
||||
"priority": False,
|
||||
"ppl_protection": False,
|
||||
"watchdog": False,
|
||||
"platform": PLATFORM,
|
||||
}
|
||||
|
||||
# 1. 单实例保护
|
||||
results["single_instance"] = self.acquire_single_instance()
|
||||
|
||||
# 2. 进程优先级提升
|
||||
results["priority"] = self.elevate_priority(priority_level)
|
||||
|
||||
# 3. PPL 风格保护
|
||||
results["ppl_protection"] = self.enable_ppl_protection()
|
||||
|
||||
# 4. 看门狗
|
||||
if watchdog:
|
||||
self.start_watchdog(
|
||||
restart_on_crash=restart_on_crash,
|
||||
restart_script=restart_script
|
||||
)
|
||||
results["watchdog"] = True
|
||||
|
||||
# 注册清理
|
||||
atexit.register(self.disable_all)
|
||||
|
||||
return results
|
||||
|
||||
def disable_all(self):
|
||||
"""禁用所有保护并清理"""
|
||||
self.stop_watchdog()
|
||||
self.restore_priority()
|
||||
self.release_single_instance()
|
||||
self._execute_shutdown_hooks()
|
||||
self._is_protected = False
|
||||
|
||||
def get_status(self) -> dict:
|
||||
"""获取当前保护状态"""
|
||||
return {
|
||||
"platform": PLATFORM,
|
||||
"is_protected": self._is_protected,
|
||||
"single_instance_held": (
|
||||
self._lock_fd is not None or
|
||||
hasattr(self, '_win_mutex_handle') and self._win_mutex_handle is not None
|
||||
),
|
||||
"watchdog_running": (
|
||||
self._watchdog_thread is not None and
|
||||
self._watchdog_thread.is_alive()
|
||||
),
|
||||
"shutdown_hooks_count": len(self._shutdown_hooks),
|
||||
}
|
||||
|
||||
|
||||
# ── 全局单例 ──
|
||||
_protection: Optional[ProcessProtection] = None
|
||||
|
||||
|
||||
def get_protection() -> ProcessProtection:
|
||||
"""获取全局进程保护器单例"""
|
||||
global _protection
|
||||
if _protection is None:
|
||||
_protection = ProcessProtection()
|
||||
return _protection
|
||||
|
||||
|
||||
def enable_process_protection(priority_level: str = "above_normal",
|
||||
watchdog: bool = True,
|
||||
restart_on_crash: bool = False) -> dict:
|
||||
"""
|
||||
快捷函数:启用进程保护
|
||||
|
||||
Returns:
|
||||
各项保护状态字典
|
||||
"""
|
||||
return get_protection().enable_all(
|
||||
priority_level=priority_level,
|
||||
watchdog=watchdog,
|
||||
restart_on_crash=restart_on_crash,
|
||||
)
|
||||
|
||||
|
||||
def disable_process_protection():
|
||||
"""快捷函数:禁用进程保护"""
|
||||
get_protection().disable_all()
|
||||
|
||||
|
||||
def get_protection_status() -> dict:
|
||||
"""快捷函数:获取保护状态"""
|
||||
return get_protection().get_status()
|
||||
|
||||
|
||||
# ── CLI 入口(可独立运行测试) ──
|
||||
if __name__ == "__main__":
|
||||
import json
|
||||
|
||||
print(f"平台: {PLATFORM}")
|
||||
print(f"PID: {os.getpid()}")
|
||||
print()
|
||||
|
||||
prot = ProcessProtection("xiaoli-cli-test")
|
||||
|
||||
# 测试单实例
|
||||
is_first = prot.acquire_single_instance()
|
||||
print(f"单实例锁: {'获取成功 (首个实例)' if is_first else '获取失败 (已有实例)'}")
|
||||
|
||||
if is_first:
|
||||
# 测试优先级提升
|
||||
prio_ok = prot.elevate_priority("above_normal")
|
||||
print(f"优先级提升: {'成功' if prio_ok else '失败'}")
|
||||
|
||||
# 测试 PPL 保护
|
||||
ppl_ok = prot.enable_ppl_protection()
|
||||
print(f"PPL 保护: {'启用' if ppl_ok else '失败'}")
|
||||
|
||||
# 测试看门狗
|
||||
prot.start_watchdog(check_interval=2.0)
|
||||
print(f"看门狗: 已启动")
|
||||
|
||||
# 显示状态
|
||||
status = prot.get_status()
|
||||
print(f"\n状态: {json.dumps(status, indent=2, ensure_ascii=False)}")
|
||||
|
||||
print("\n按 Ctrl+C 退出...")
|
||||
try:
|
||||
while True:
|
||||
time.sleep(1)
|
||||
except KeyboardInterrupt:
|
||||
print("\n正在清理...")
|
||||
prot.disable_all()
|
||||
print("完成")
|
||||
@@ -18,16 +18,6 @@ MODE_NAMES = {
|
||||
MODE_MANUAL: "人工确认",
|
||||
}
|
||||
|
||||
# ── 始终需要人工确认的指令(无论模式) ──
|
||||
ALWAYS_CONFIRM = {
|
||||
("cmd_executor", "run rm"),
|
||||
("cmd_executor", "run chmod"),
|
||||
("cmd_executor", "run chown"),
|
||||
("file_manager", "delete"),
|
||||
("git_tools", "checkout -b"),
|
||||
}
|
||||
|
||||
|
||||
class SafetyLayer:
|
||||
"""统一安全检查层"""
|
||||
|
||||
@@ -65,17 +55,6 @@ class SafetyLayer:
|
||||
if self.mode == MODE_UNRESTRICTED:
|
||||
return True, ""
|
||||
|
||||
# 检查是否是始终需要确认的指令
|
||||
args_lower = tool_args.strip().lower()
|
||||
for (t, pattern) in ALWAYS_CONFIRM:
|
||||
if tool_name == t and args_lower.startswith(pattern.split(" ", 1)[1] if " " in pattern else ""):
|
||||
return self._ask_user_confirm(
|
||||
tool_name, tool_args,
|
||||
risk_level="高",
|
||||
impact="此操作始终需要人工确认",
|
||||
reason="敏感指令"
|
||||
)
|
||||
|
||||
# 人工确认模式:所有指令都确认
|
||||
if self.mode == MODE_MANUAL:
|
||||
return self._ask_user_confirm(
|
||||
|
||||
@@ -0,0 +1,119 @@
|
||||
"""
|
||||
统一工具返回格式 - ToolResult
|
||||
|
||||
所有插件和工具管理器使用统一的返回格式,
|
||||
替代原来混乱的字符串/Dict 返回。
|
||||
"""
|
||||
|
||||
from dataclasses import dataclass, field
|
||||
from typing import Any, Optional
|
||||
|
||||
|
||||
# 错误码常量
|
||||
class ErrorCode:
|
||||
"""工具错误码"""
|
||||
SUCCESS = 0
|
||||
# 参数错误 (1xx)
|
||||
MISSING_ARGS = 100
|
||||
INVALID_ARGS = 101
|
||||
UNSUPPORTED_OP = 102
|
||||
# 资源错误 (2xx)
|
||||
NOT_FOUND = 200
|
||||
FILE_NOT_FOUND = 201
|
||||
PERMISSION_DENIED = 202
|
||||
# 执行错误 (3xx)
|
||||
EXEC_FAILED = 300
|
||||
TIMEOUT = 301
|
||||
DECODE_ERROR = 302
|
||||
# 系统错误 (4xx)
|
||||
INTERNAL_ERROR = 400
|
||||
NOT_IMPLEMENTED = 401
|
||||
|
||||
_NAMES = {
|
||||
0: "成功",
|
||||
100: "缺少参数",
|
||||
101: "参数无效",
|
||||
102: "不支持的操作",
|
||||
200: "未找到",
|
||||
201: "文件不存在",
|
||||
202: "权限不足",
|
||||
300: "执行失败",
|
||||
301: "超时",
|
||||
302: "解码错误",
|
||||
400: "内部错误",
|
||||
401: "未实现",
|
||||
}
|
||||
|
||||
@classmethod
|
||||
def name(cls, code: int) -> str:
|
||||
return cls._NAMES.get(code, f"未知({code})")
|
||||
|
||||
|
||||
@dataclass
|
||||
class ToolResult:
|
||||
"""统一工具返回格式"""
|
||||
|
||||
success: bool
|
||||
data: Any = None
|
||||
error_code: int = ErrorCode.SUCCESS
|
||||
error_message: str = ""
|
||||
|
||||
# 可选元数据
|
||||
tool_name: str = ""
|
||||
execution_time: float = 0.0
|
||||
metadata: dict = field(default_factory=dict)
|
||||
|
||||
def __str__(self) -> str:
|
||||
if self.success:
|
||||
return str(self.data) if self.data is not None else "执行完成"
|
||||
return f"[{ErrorCode.name(self.error_code)}] {self.error_message}"
|
||||
|
||||
# ── 便捷构造方法 ──
|
||||
|
||||
@classmethod
|
||||
def ok(cls, data: Any = None, **kwargs) -> "ToolResult":
|
||||
"""成功"""
|
||||
return cls(success=True, data=data, **kwargs)
|
||||
|
||||
@classmethod
|
||||
def fail(cls, message: str, code: int = ErrorCode.EXEC_FAILED, **kwargs) -> "ToolResult":
|
||||
"""失败"""
|
||||
return cls(success=False, error_code=code, error_message=message, **kwargs)
|
||||
|
||||
@classmethod
|
||||
def missing(cls, what: str = "参数", **kwargs) -> "ToolResult":
|
||||
"""缺少参数"""
|
||||
return cls.fail(f"缺少{what}", ErrorCode.MISSING_ARGS, **kwargs)
|
||||
|
||||
@classmethod
|
||||
def not_found(cls, what: str = "资源", **kwargs) -> "ToolResult":
|
||||
"""未找到"""
|
||||
return cls.fail(f"{what}不存在", ErrorCode.NOT_FOUND, **kwargs)
|
||||
|
||||
@classmethod
|
||||
def timeout(cls, seconds: int = 0, **kwargs) -> "ToolResult":
|
||||
"""超时"""
|
||||
msg = f"执行超时"
|
||||
if seconds:
|
||||
msg += f" ({seconds}秒)"
|
||||
return cls.fail(msg, ErrorCode.TIMEOUT, **kwargs)
|
||||
|
||||
# ── 格式转换 ──
|
||||
|
||||
def to_dict(self) -> dict:
|
||||
"""转为普通 dict(兼容旧接口)"""
|
||||
if self.success:
|
||||
return {"result": self.data if self.data is not None else "执行完成"}
|
||||
return {"result": str(self)}
|
||||
|
||||
def to_mcp(self) -> dict:
|
||||
"""转为 MCP 协议格式"""
|
||||
text = str(self)
|
||||
return {
|
||||
"content": [{"type": "text", "text": text}],
|
||||
"isError": not self.success,
|
||||
}
|
||||
|
||||
def to_string(self) -> str:
|
||||
"""转为字符串(兼容旧插件 handle() 返回值)"""
|
||||
return str(self)
|
||||
+37
-13
@@ -406,6 +406,23 @@ if TEXTUAL_AVAILABLE:
|
||||
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
|
||||
# 其他键正常处理
|
||||
|
||||
# ═══════════════════════════════════════════════════
|
||||
# TUI 主应用
|
||||
# ═══════════════════════════════════════════════════
|
||||
@@ -420,7 +437,8 @@ if TEXTUAL_AVAILABLE:
|
||||
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=True),
|
||||
Binding("ctrl+enter", "send_message", "发送", show=False),
|
||||
Binding("shift+enter", "newline", "换行", show=False),
|
||||
Binding("f1", "toggle_sidebar", "侧栏", show=True),
|
||||
Binding("escape", "cancel", "取消", show=False),
|
||||
]
|
||||
@@ -442,14 +460,14 @@ if TEXTUAL_AVAILABLE:
|
||||
yield VerticalScroll(id="chat-scroll")
|
||||
with Vertical(id="input-area"):
|
||||
with Container(id="input-wrapper"):
|
||||
yield TextArea(
|
||||
placeholder="输入消息... (Ctrl+Enter 发送, Enter 换行)",
|
||||
yield SendTextArea(
|
||||
placeholder="输入消息... (回车发送, Shift+回车换行)",
|
||||
id="user-input",
|
||||
soft_wrap=True,
|
||||
tab_behavior="indent",
|
||||
)
|
||||
yield Static(
|
||||
" Ctrl+Enter 发送 | ↑↓ 历史 | Ctrl+L 清屏 | F1 侧栏 | Tab 缩进",
|
||||
" 回车发送 | Shift+回车换行 | Ctrl+L 清屏 | F1 侧栏",
|
||||
id="input-hint"
|
||||
)
|
||||
with Vertical(id="sidebar"):
|
||||
@@ -480,7 +498,7 @@ if TEXTUAL_AVAILABLE:
|
||||
"",
|
||||
" ╔══════════════════════════════════════════════╗",
|
||||
" ║ ║",
|
||||
" ║ 小狸 Pro-CLI v5.2 ║",
|
||||
" ║ 小狸 Pro-CLI v5.4.1 ║",
|
||||
" ║ 智能编程助手 · 动画增强版 ║",
|
||||
" ║ ║",
|
||||
" ╚══════════════════════════════════════════════╝",
|
||||
@@ -488,7 +506,7 @@ if TEXTUAL_AVAILABLE:
|
||||
" 代码编辑 · 代码搜索 · Git 集成 · 多引擎",
|
||||
" 输入 /help 查看命令 | /model 切换引擎",
|
||||
f" 当前引擎: {self.bridge.current_engine()} | 工具: {len(self.bridge.tools())} 个",
|
||||
" 支持多行输入 — Ctrl+Enter 发送",
|
||||
" 支持多行输入 — Shift+回车换行",
|
||||
"",
|
||||
]
|
||||
|
||||
@@ -524,7 +542,7 @@ if TEXTUAL_AVAILABLE:
|
||||
("", "msg-dim"),
|
||||
(" ╔══════════════════════════════════════════════╗", "msg-welcome"),
|
||||
(" ║ ║", "msg-welcome"),
|
||||
(" ║ 小狸 Pro-CLI v5.2 ║", "msg-welcome"),
|
||||
(" ║ 小狸 Pro-CLI v5.4.1 ║", "msg-welcome"),
|
||||
(" ║ 智能编程助手 · 动画增强版 ║", "msg-welcome"),
|
||||
(" ║ ║", "msg-welcome"),
|
||||
(" ╚══════════════════════════════════════════════╝", "msg-welcome"),
|
||||
@@ -532,7 +550,7 @@ if TEXTUAL_AVAILABLE:
|
||||
(" 代码编辑 · 代码搜索 · Git 集成 · 多引擎", "msg-system"),
|
||||
(" 输入 /help 查看命令 | /model 切换引擎", "msg-system"),
|
||||
(f" 当前引擎: {self.bridge.current_engine()} | 工具: {len(self.bridge.tools())} 个", "msg-system"),
|
||||
(" 支持多行输入 — Ctrl+Enter 发送", "msg-system"),
|
||||
(" 支持多行输入 — Shift+回车换行", "msg-system"),
|
||||
("", "msg-dim"),
|
||||
]
|
||||
for text, cls in lines:
|
||||
@@ -670,8 +688,14 @@ if TEXTUAL_AVAILABLE:
|
||||
|
||||
# ── 多行输入处理 ──
|
||||
|
||||
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):
|
||||
"""Ctrl+Enter 发送消息"""
|
||||
"""回车发送消息"""
|
||||
text_area = self.query_one("#user-input")
|
||||
text = text_area.text.strip()
|
||||
if not text:
|
||||
@@ -719,7 +743,7 @@ if TEXTUAL_AVAILABLE:
|
||||
'cls': lambda: self.action_clear(),
|
||||
'model': lambda: self._switch_model(args),
|
||||
'engine': lambda: self._switch_model(args),
|
||||
'about': lambda: self._system(" 小狸 Pro-CLI v5.2 - 智能编程助手"),
|
||||
'about': lambda: self._system(" 小狸 Pro-CLI v5.4.1 - 智能编程助手"),
|
||||
'status': lambda: self._show_status(),
|
||||
'tools': lambda: self._show_tools(),
|
||||
'engines': lambda: self._show_engines(),
|
||||
@@ -754,9 +778,9 @@ if TEXTUAL_AVAILABLE:
|
||||
|
||||
⌨ 快捷键:
|
||||
Ctrl+C 退出 Ctrl+L 清屏
|
||||
Ctrl+N 新对话 Ctrl+Enter 发送消息
|
||||
F1 侧栏 Escape 取消生成
|
||||
Enter 换行 Tab 缩进"""
|
||||
Ctrl+N 新对话 回车 发送消息
|
||||
Shift+回车 换行 F1 侧栏
|
||||
Escape 取消生成 Tab 缩进"""
|
||||
self._system(help_text)
|
||||
|
||||
def _handle_manual(self, args):
|
||||
|
||||
Reference in new issue
Block a user