mirror of
https://github.com/Minidoracat/mcp-feedback-enhanced.git
synced 2025-07-27 10:42:25 +08:00
442 lines
14 KiB
Python
442 lines
14 KiB
Python
#!/usr/bin/env python3
|
|
"""
|
|
MCP Interactive Feedback Enhanced - 主程式入口
|
|
==============================================
|
|
|
|
此檔案允許套件透過 `python -m mcp_feedback_enhanced` 執行。
|
|
|
|
使用方法:
|
|
python -m mcp_feedback_enhanced # 啟動 MCP 伺服器
|
|
python -m mcp_feedback_enhanced test # 執行測試
|
|
"""
|
|
|
|
import argparse
|
|
import asyncio
|
|
import os
|
|
import sys
|
|
import warnings
|
|
|
|
|
|
# 抑制 Windows 上的 asyncio ResourceWarning
|
|
if sys.platform == "win32":
|
|
warnings.filterwarnings(
|
|
"ignore", category=ResourceWarning, message=".*unclosed transport.*"
|
|
)
|
|
warnings.filterwarnings("ignore", category=ResourceWarning, message=".*unclosed.*")
|
|
|
|
# 設置 asyncio 事件循環策略以減少警告
|
|
try:
|
|
asyncio.set_event_loop_policy(asyncio.WindowsProactorEventLoopPolicy())
|
|
except AttributeError:
|
|
pass
|
|
|
|
|
|
def main():
|
|
"""主程式入口點"""
|
|
parser = argparse.ArgumentParser(
|
|
description="MCP Feedback Enhanced Enhanced - 互動式回饋收集 MCP 伺服器"
|
|
)
|
|
|
|
subparsers = parser.add_subparsers(dest="command", help="可用命令")
|
|
|
|
# 伺服器命令（預設）
|
|
server_parser = subparsers.add_parser("server", help="啟動 MCP 伺服器（預設）")
|
|
|
|
# 測試命令
|
|
test_parser = subparsers.add_parser("test", help="執行測試")
|
|
test_parser.add_argument(
|
|
"--web", action="store_true", help="測試 Web UI (自動持續運行)"
|
|
)
|
|
test_parser.add_argument(
|
|
"--desktop", action="store_true", help="測試桌面應用 (啟動 Electron 應用)"
|
|
)
|
|
test_parser.add_argument(
|
|
"--full", action="store_true", help="完整整合測試 (Web + 桌面)"
|
|
)
|
|
test_parser.add_argument(
|
|
"--electron-only", action="store_true", help="僅測試 Electron 環境"
|
|
)
|
|
test_parser.add_argument(
|
|
"--timeout", type=int, default=60, help="測試超時時間 (秒)"
|
|
)
|
|
|
|
# 版本命令
|
|
version_parser = subparsers.add_parser("version", help="顯示版本資訊")
|
|
|
|
args = parser.parse_args()
|
|
|
|
if args.command == "test":
|
|
run_tests(args)
|
|
elif args.command == "version":
|
|
show_version()
|
|
elif args.command == "server" or args.command is None:
|
|
run_server()
|
|
else:
|
|
# 不應該到達這裡
|
|
parser.print_help()
|
|
sys.exit(1)
|
|
|
|
|
|
def run_server():
|
|
"""啟動 MCP 伺服器"""
|
|
from .server import main as server_main
|
|
|
|
return server_main()
|
|
|
|
|
|
def run_tests(args):
|
|
"""執行測試"""
|
|
# 啟用調試模式以顯示測試過程
|
|
os.environ["MCP_DEBUG"] = "true"
|
|
|
|
# 在 Windows 上抑制 asyncio 警告
|
|
if sys.platform == "win32":
|
|
os.environ["PYTHONWARNINGS"] = "ignore::ResourceWarning"
|
|
|
|
if args.web:
|
|
print("🧪 執行 Web UI 測試...")
|
|
success = test_web_ui_simple()
|
|
if not success:
|
|
sys.exit(1)
|
|
elif args.desktop:
|
|
print("🧪 執行桌面應用測試...")
|
|
success = test_desktop_app()
|
|
if not success:
|
|
sys.exit(1)
|
|
elif args.full:
|
|
print("🧪 執行完整整合測試...")
|
|
success = test_full_integration()
|
|
if not success:
|
|
sys.exit(1)
|
|
elif args.electron_only:
|
|
print("🧪 執行 Electron 環境測試...")
|
|
success = test_electron_environment()
|
|
if not success:
|
|
sys.exit(1)
|
|
else:
|
|
print("❌ 測試功能已簡化")
|
|
print("💡 可用的測試選項：")
|
|
print(" --web 測試 Web UI")
|
|
print(" --desktop 測試桌面應用")
|
|
print(" --full 完整整合測試")
|
|
print(" --electron-only 僅測試 Electron 環境")
|
|
print("💡 對於開發者：使用 'uv run pytest' 執行完整測試")
|
|
sys.exit(1)
|
|
|
|
|
|
def test_web_ui_simple():
|
|
"""簡單的 Web UI 測試"""
|
|
try:
|
|
import tempfile
|
|
import time
|
|
import webbrowser
|
|
|
|
from .web.main import WebUIManager
|
|
|
|
# 設置測試模式，禁用自動清理避免權限問題
|
|
os.environ["MCP_TEST_MODE"] = "true"
|
|
# 設置更高的端口範圍避免系統保留端口
|
|
os.environ["MCP_WEB_PORT"] = "9765"
|
|
|
|
print("🔧 創建 Web UI 管理器...")
|
|
manager = WebUIManager(host="127.0.0.1") # 使用動態端口分配
|
|
|
|
print("🔧 創建測試會話...")
|
|
with tempfile.TemporaryDirectory() as temp_dir:
|
|
session_id = manager.create_session(temp_dir, "Web UI 測試 - 驗證基本功能")
|
|
|
|
if session_id:
|
|
print("✅ 會話創建成功")
|
|
|
|
print("🚀 啟動 Web 服務器...")
|
|
manager.start_server()
|
|
time.sleep(5) # 等待服務器完全啟動
|
|
|
|
if manager.server_thread and manager.server_thread.is_alive():
|
|
print("✅ Web 服務器啟動成功")
|
|
url = f"http://{manager.host}:{manager.port}"
|
|
print(f"🌐 服務器運行在: {url}")
|
|
|
|
# 嘗試開啟瀏覽器
|
|
print("🌐 正在開啟瀏覽器...")
|
|
try:
|
|
webbrowser.open(url)
|
|
print("✅ 瀏覽器已開啟")
|
|
except Exception as e:
|
|
print(f"⚠️ 無法自動開啟瀏覽器: {e}")
|
|
print(f"💡 請手動開啟瀏覽器並訪問: {url}")
|
|
|
|
print("📝 Web UI 測試完成，進入持續模式...")
|
|
print("💡 提示：服務器將持續運行，可在瀏覽器中測試互動功能")
|
|
print("💡 按 Ctrl+C 停止服務器")
|
|
|
|
try:
|
|
# 保持服務器運行
|
|
while True:
|
|
time.sleep(1)
|
|
except KeyboardInterrupt:
|
|
print("\n🛑 停止服務器...")
|
|
return True
|
|
else:
|
|
print("❌ Web 服務器啟動失敗")
|
|
return False
|
|
else:
|
|
print("❌ 會話創建失敗")
|
|
return False
|
|
|
|
except Exception as e:
|
|
print(f"❌ Web UI 測試失敗: {e}")
|
|
import traceback
|
|
|
|
traceback.print_exc()
|
|
return False
|
|
finally:
|
|
# 清理測試環境變數
|
|
os.environ.pop("MCP_TEST_MODE", None)
|
|
os.environ.pop("MCP_WEB_PORT", None)
|
|
|
|
|
|
def test_desktop_app():
|
|
"""測試桌面應用"""
|
|
try:
|
|
print("🔧 檢查桌面環境...")
|
|
|
|
# 檢查桌面環境可用性
|
|
from .desktop import is_desktop_available
|
|
|
|
if not is_desktop_available():
|
|
print("❌ 桌面環境不可用")
|
|
print("💡 請確保 Node.js 已安裝且不在遠程環境中")
|
|
return False
|
|
|
|
print("✅ 桌面環境檢查通過")
|
|
|
|
# 設置桌面模式
|
|
os.environ["MCP_FEEDBACK_MODE"] = "desktop"
|
|
|
|
print("🔧 創建 Electron 管理器...")
|
|
import asyncio
|
|
|
|
async def run_desktop_test():
|
|
print("🚀 啟動完整桌面應用測試...")
|
|
print("💡 這將啟動 Web 服務器和 Electron 應用視窗")
|
|
print("💡 請在應用中測試基本功能，然後關閉視窗")
|
|
|
|
# 使用完整的桌面應用啟動函數
|
|
from .desktop import launch_desktop_app
|
|
|
|
try:
|
|
# 這會啟動 Web 服務器和 Electron 應用
|
|
result = await launch_desktop_app(
|
|
os.getcwd(),
|
|
"桌面應用測試 - 驗證 Electron 整合功能",
|
|
300, # 5分鐘超時
|
|
)
|
|
|
|
print("✅ 桌面應用測試完成")
|
|
print(f"收到回饋: {result.get('interactive_feedback', '無回饋')}")
|
|
return True
|
|
|
|
except Exception as e:
|
|
print(f"❌ 桌面應用測試失敗: {e}")
|
|
return False
|
|
|
|
return asyncio.run(run_desktop_test())
|
|
|
|
except Exception as e:
|
|
print(f"❌ 桌面應用測試失敗: {e}")
|
|
import traceback
|
|
|
|
traceback.print_exc()
|
|
return False
|
|
|
|
|
|
async def wait_for_process(process):
|
|
"""等待進程結束"""
|
|
try:
|
|
# 等待進程自然結束
|
|
await process.wait()
|
|
|
|
# 確保管道正確關閉
|
|
try:
|
|
if hasattr(process, "stdout") and process.stdout:
|
|
process.stdout.close()
|
|
if hasattr(process, "stderr") and process.stderr:
|
|
process.stderr.close()
|
|
if hasattr(process, "stdin") and process.stdin:
|
|
process.stdin.close()
|
|
except Exception as close_error:
|
|
print(f"關閉進程管道時出錯: {close_error}")
|
|
|
|
except Exception as e:
|
|
print(f"等待進程時出錯: {e}")
|
|
|
|
|
|
def test_electron_environment():
|
|
"""測試 Electron 環境"""
|
|
try:
|
|
print("🔧 檢查 Electron 環境...")
|
|
|
|
# 檢查 Node.js
|
|
import subprocess
|
|
|
|
try:
|
|
result = subprocess.run(
|
|
["node", "--version"],
|
|
capture_output=True,
|
|
text=True,
|
|
timeout=10,
|
|
check=False,
|
|
)
|
|
if result.returncode == 0:
|
|
print(f"✅ Node.js 版本: {result.stdout.strip()}")
|
|
else:
|
|
print("❌ Node.js 不可用")
|
|
return False
|
|
except (subprocess.TimeoutExpired, FileNotFoundError):
|
|
print("❌ Node.js 不可用")
|
|
return False
|
|
|
|
# 檢查桌面模組
|
|
from .desktop import is_desktop_available
|
|
|
|
if is_desktop_available():
|
|
print("✅ 桌面環境可用")
|
|
else:
|
|
print("❌ 桌面環境不可用")
|
|
return False
|
|
|
|
# 檢查 Electron 管理器
|
|
from .desktop.electron_manager import ElectronManager
|
|
|
|
manager = ElectronManager()
|
|
|
|
if manager.is_electron_available():
|
|
print("✅ Electron 環境可用")
|
|
else:
|
|
print("❌ Electron 環境不可用")
|
|
return False
|
|
|
|
# 檢查文件結構
|
|
desktop_dir = manager.desktop_dir
|
|
required_files = ["main.js", "preload.js", "package.json"]
|
|
|
|
for file_name in required_files:
|
|
file_path = desktop_dir / file_name
|
|
if file_path.exists():
|
|
print(f"✅ {file_name} 存在")
|
|
else:
|
|
print(f"❌ {file_name} 不存在")
|
|
return False
|
|
|
|
# 檢查 node_modules
|
|
node_modules = desktop_dir / "node_modules"
|
|
if node_modules.exists():
|
|
print("✅ node_modules 存在")
|
|
else:
|
|
print("❌ node_modules 不存在")
|
|
return False
|
|
|
|
print("🎉 Electron 環境測試完成，所有檢查通過")
|
|
return True
|
|
|
|
except Exception as e:
|
|
print(f"❌ Electron 環境測試失敗: {e}")
|
|
return False
|
|
|
|
|
|
def test_full_integration():
|
|
"""完整整合測試"""
|
|
try:
|
|
print("🧪 執行完整整合測試...")
|
|
|
|
# 1. 環境變數測試
|
|
print("\n📋 1. 測試環境變數控制...")
|
|
test_cases = [("auto", "auto"), ("web", "web"), ("desktop", "desktop")]
|
|
|
|
for env_value, expected in test_cases:
|
|
os.environ["MCP_FEEDBACK_MODE"] = env_value
|
|
|
|
# 重新導入以獲取新的環境變數值
|
|
import sys
|
|
|
|
if "mcp_feedback_enhanced.server" in sys.modules:
|
|
del sys.modules["mcp_feedback_enhanced.server"]
|
|
|
|
from .server import get_feedback_mode
|
|
|
|
actual = get_feedback_mode().value
|
|
|
|
if actual == expected:
|
|
print(f" ✅ MCP_FEEDBACK_MODE='{env_value}' → {actual}")
|
|
else:
|
|
print(
|
|
f" ❌ MCP_FEEDBACK_MODE='{env_value}' → {actual} (期望: {expected})"
|
|
)
|
|
return False
|
|
|
|
# 2. Electron 環境測試
|
|
print("\n📋 2. 測試 Electron 環境...")
|
|
if not test_electron_environment():
|
|
print("❌ Electron 環境測試失敗")
|
|
return False
|
|
|
|
# 3. Web UI 基本功能測試
|
|
print("\n📋 3. 測試 Web UI 基本功能...")
|
|
import tempfile
|
|
|
|
from .web.main import WebUIManager
|
|
|
|
# 設置測試模式
|
|
os.environ["MCP_TEST_MODE"] = "true"
|
|
os.environ["MCP_WEB_PORT"] = "9766"
|
|
|
|
with tempfile.TemporaryDirectory() as temp_dir:
|
|
manager = WebUIManager(host="127.0.0.1") # 使用動態端口分配避免衝突
|
|
session_id = manager.create_session(temp_dir, "整合測試會話")
|
|
|
|
if session_id:
|
|
print(" ✅ Web UI 會話創建成功")
|
|
else:
|
|
print(" ❌ Web UI 會話創建失敗")
|
|
return False
|
|
|
|
# 4. 桌面模式檢測測試
|
|
print("\n📋 4. 測試桌面模式檢測...")
|
|
os.environ["MCP_FEEDBACK_MODE"] = "desktop"
|
|
|
|
manager = WebUIManager()
|
|
if manager.should_use_desktop_mode():
|
|
print(" ✅ 桌面模式檢測正常")
|
|
else:
|
|
print(" ❌ 桌面模式檢測失敗")
|
|
return False
|
|
|
|
print("\n🎉 完整整合測試通過！")
|
|
print("💡 所有核心功能正常運作")
|
|
return True
|
|
|
|
except Exception as e:
|
|
print(f"❌ 完整整合測試失敗: {e}")
|
|
import traceback
|
|
|
|
traceback.print_exc()
|
|
return False
|
|
finally:
|
|
# 清理測試環境變數
|
|
os.environ.pop("MCP_TEST_MODE", None)
|
|
os.environ.pop("MCP_WEB_PORT", None)
|
|
|
|
|
|
def show_version():
|
|
"""顯示版本資訊"""
|
|
from . import __author__, __version__
|
|
|
|
print(f"MCP Feedback Enhanced Enhanced v{__version__}")
|
|
print(f"作者: {__author__}")
|
|
print("GitHub: https://github.com/Minidoracat/mcp-feedback-enhanced")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|