2
0

index.html 49 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281
  1. <!DOCTYPE html>
  2. <html lang="zh-CN">
  3. <head>
  4. <meta charset="UTF-8">
  5. <meta name="viewport" content="width=device-width, initial-scale=1.0">
  6. <title>MCP 助手</title>
  7. <script src="https://cdn.tailwindcss.com"></script>
  8. <style>
  9. .chat-container {
  10. height: calc(100vh - 200px);
  11. }
  12. /* 移动端优化 */
  13. @media (max-width: 768px) {
  14. .chat-container {
  15. height: calc(100vh - 180px);
  16. }
  17. /* 头部固定 */
  18. header {
  19. position: sticky;
  20. top: 0;
  21. z-index: 50;
  22. }
  23. /* 消息气泡宽度 */
  24. .message-bubble {
  25. max-width: 85vw !important;
  26. }
  27. /* 输入区域固定 */
  28. .input-area {
  29. position: sticky;
  30. bottom: 0;
  31. z-index: 50;
  32. background: white;
  33. border-top-left-radius: 0;
  34. border-top-right-radius: 0;
  35. }
  36. }
  37. /* 参数/结果可折叠区域 */
  38. .collapsible-section {
  39. margin-top: 6px;
  40. }
  41. .collapsible-header {
  42. display: flex;
  43. align-items: center;
  44. gap: 4px;
  45. padding: 4px 8px;
  46. background: rgba(255, 255, 255, 0.05);
  47. border-radius: 4px;
  48. cursor: pointer;
  49. font-size: 11px;
  50. color: #9ca3af;
  51. user-select: none;
  52. }
  53. .collapsible-header:hover {
  54. background: rgba(255, 255, 255, 0.08);
  55. }
  56. .collapsible-content {
  57. max-height: 0;
  58. overflow: hidden;
  59. transition: max-height 0.3s ease-out;
  60. }
  61. .collapsible-content.expanded {
  62. max-height: 300px;
  63. overflow-y: auto;
  64. }
  65. .collapsible-icon {
  66. transition: transform 0.3s;
  67. font-size: 10px;
  68. }
  69. .collapsible-icon.expanded {
  70. transform: rotate(90deg);
  71. }
  72. /* MCP 服务器横向滚动 */
  73. .mcp-servers-scroll {
  74. display: flex;
  75. gap: 8px;
  76. overflow-x: auto;
  77. padding: 4px 0;
  78. scrollbar-width: thin;
  79. -webkit-overflow-scrolling: touch;
  80. }
  81. .mcp-servers-scroll::-webkit-scrollbar {
  82. height: 4px;
  83. }
  84. .mcp-servers-scroll::-webkit-scrollbar-thumb {
  85. background: rgba(0, 0, 0, 0.2);
  86. border-radius: 2px;
  87. }
  88. .mcp-server-tag {
  89. flex-shrink: 0;
  90. display: flex;
  91. align-items: center;
  92. gap: 6px;
  93. padding: 8px 12px;
  94. border-radius: 8px;
  95. font-size: 12px;
  96. white-space: nowrap;
  97. }
  98. .mcp-server-tag.enabled {
  99. background: #d1fae5;
  100. color: #065f46;
  101. }
  102. .mcp-server-tag.disabled {
  103. background: #f3f4f6;
  104. color: #6b7280;
  105. }
  106. .mcp-status-dot {
  107. width: 8px;
  108. height: 8px;
  109. border-radius: 50%;
  110. }
  111. .mcp-status-dot.online {
  112. background: #10b981;
  113. }
  114. .mcp-status-dot.offline {
  115. background: #9ca3af;
  116. }
  117. .message-user {
  118. background: linear-gradient(135deg, #667eea 0%, #764ba2 100%);
  119. }
  120. .message-assistant {
  121. background: linear-gradient(135deg, #f5f7fa 0%, #c3cfe2 100%);
  122. }
  123. .typing-indicator span {
  124. animation: bounce 1.4s infinite ease-in-out both;
  125. }
  126. .typing-indicator span:nth-child(1) { animation-delay: -0.32s; }
  127. .typing-indicator span:nth-child(2) { animation-delay: -0.16s; }
  128. @keyframes bounce {
  129. 0%, 80%, 100% { transform: scale(0); }
  130. 40% { transform: scale(1); }
  131. }
  132. /* 工具调用面板样式 */
  133. .tool-call-panel {
  134. background: linear-gradient(135deg, #1a1a2e 0%, #16213e 100%);
  135. border-radius: 8px;
  136. margin: 12px 0;
  137. overflow: hidden;
  138. border: 1px solid #3a3a5a;
  139. }
  140. .tool-panel-header {
  141. background: rgba(255, 255, 255, 0.05);
  142. padding: 10px 14px;
  143. cursor: pointer;
  144. display: flex;
  145. align-items: center;
  146. justify-content: space-between;
  147. transition: background 0.2s;
  148. }
  149. .tool-panel-header:hover {
  150. background: rgba(255, 255, 255, 0.08);
  151. }
  152. .tool-panel-title {
  153. display: flex;
  154. align-items: center;
  155. gap: 8px;
  156. color: #e0e0e0;
  157. font-size: 13px;
  158. font-weight: 500;
  159. }
  160. .tool-panel-content {
  161. padding: 0;
  162. max-height: 0;
  163. overflow: hidden;
  164. transition: max-height 0.3s ease-out, padding 0.3s ease-out;
  165. }
  166. .tool-panel-content.collapsed {
  167. max-height: 0;
  168. }
  169. .tool-panel-content.expanded {
  170. max-height: 800px;
  171. padding: 12px;
  172. overflow-y: auto;
  173. }
  174. .tool-panel-content::-webkit-scrollbar {
  175. width: 6px;
  176. }
  177. .tool-panel-content::-webkit-scrollbar-track {
  178. background: rgba(255, 255, 255, 0.05);
  179. }
  180. .tool-panel-content::-webkit-scrollbar-thumb {
  181. background: rgba(255, 255, 255, 0.2);
  182. border-radius: 3px;
  183. }
  184. /* 工具调用项样式 */
  185. .tool-call-item {
  186. background: rgba(255, 255, 255, 0.03);
  187. border-radius: 6px;
  188. padding: 10px 12px;
  189. margin-bottom: 8px;
  190. border-left: 3px solid #555;
  191. transition: all 0.2s;
  192. }
  193. .tool-call-item:last-child {
  194. margin-bottom: 0;
  195. }
  196. .tool-call-item.pending {
  197. border-left-color: #f59e0b;
  198. animation: pulse-border 1.5s infinite;
  199. }
  200. .tool-call-item.success {
  201. border-left-color: #10b981;
  202. }
  203. .tool-call-item.error {
  204. border-left-color: #ef4444;
  205. }
  206. @keyframes pulse-border {
  207. 0%, 100% { border-left-color: #f59e0b; }
  208. 50% { border-left-color: #fbbf24; }
  209. }
  210. .tool-call-header {
  211. display: flex;
  212. align-items: center;
  213. justify-content: space-between;
  214. margin-bottom: 6px;
  215. }
  216. .tool-name {
  217. color: #e0e0e0;
  218. font-size: 13px;
  219. font-weight: 500;
  220. display: flex;
  221. align-items: center;
  222. gap: 6px;
  223. }
  224. .tool-status {
  225. font-size: 11px;
  226. padding: 2px 8px;
  227. border-radius: 10px;
  228. font-weight: 500;
  229. }
  230. .tool-status.pending {
  231. background: rgba(245, 158, 11, 0.2);
  232. color: #fbbf24;
  233. }
  234. .tool-status.success {
  235. background: rgba(16, 185, 129, 0.2);
  236. color: #34d399;
  237. }
  238. .tool-status.error {
  239. background: rgba(239, 68, 68, 0.2);
  240. color: #f87171;
  241. }
  242. .tool-params {
  243. background: rgba(0, 0, 0, 0.3);
  244. border-radius: 4px;
  245. padding: 8px;
  246. margin-top: 6px;
  247. font-size: 11px;
  248. color: #9ca3af;
  249. overflow-x: auto;
  250. }
  251. .tool-params::-webkit-scrollbar {
  252. height: 4px;
  253. }
  254. .tool-params::-webkit-scrollbar-thumb {
  255. background: rgba(255, 255, 255, 0.2);
  256. border-radius: 2px;
  257. }
  258. .tool-result {
  259. background: rgba(16, 185, 129, 0.1);
  260. border-radius: 4px;
  261. padding: 8px;
  262. margin-top: 6px;
  263. font-size: 11px;
  264. color: #6ee7b7;
  265. }
  266. .tool-result.error {
  267. background: rgba(239, 68, 68, 0.1);
  268. color: #fca5a5;
  269. }
  270. .tool-result-content {
  271. max-height: 100px;
  272. overflow-y: auto;
  273. word-break: break-all;
  274. }
  275. .tool-result-content::-webkit-scrollbar {
  276. width: 4px;
  277. }
  278. .tool-result-content::-webkit-scrollbar-thumb {
  279. background: rgba(255, 255, 255, 0.2);
  280. border-radius: 2px;
  281. }
  282. .expand-icon {
  283. transition: transform 0.3s;
  284. }
  285. .expand-icon.expanded {
  286. transform: rotate(180deg);
  287. }
  288. /* 加载动画 */
  289. .spinner {
  290. width: 14px;
  291. height: 14px;
  292. border: 2px solid rgba(251, 191, 36, 0.3);
  293. border-top-color: #fbbf24;
  294. border-radius: 50%;
  295. animation: spin 0.8s linear infinite;
  296. }
  297. @keyframes spin {
  298. to { transform: rotate(360deg); }
  299. }
  300. /* 状态指示器 */
  301. .process-indicator {
  302. display: flex;
  303. align-items: center;
  304. gap: 6px;
  305. padding: 6px 10px;
  306. background: rgba(245, 158, 11, 0.1);
  307. border-radius: 6px;
  308. font-size: 12px;
  309. color: #fbbf24;
  310. }
  311. .process-indicator.success {
  312. background: rgba(16, 185, 129, 0.1);
  313. color: #34d399;
  314. }
  315. .process-indicator .spinner {
  316. width: 12px;
  317. height: 12px;
  318. }
  319. pre {
  320. background: #1e1e1e;
  321. color: #d4d4d4;
  322. padding: 1rem;
  323. border-radius: 0.5rem;
  324. overflow-x: auto;
  325. }
  326. code {
  327. font-family: 'Courier New', monospace;
  328. }
  329. </style>
  330. </head>
  331. <body class="bg-gray-100 min-h-screen">
  332. <div class="container mx-auto px-4 py-6 max-w-4xl">
  333. <!-- Header -->
  334. <header class="bg-white rounded-lg shadow-md p-4 mb-4">
  335. <div class="flex items-center justify-between">
  336. <div class="flex items-center space-x-3">
  337. <div class="w-10 h-10 bg-gradient-to-br from-purple-500 to-indigo-600 rounded-lg flex items-center justify-center">
  338. <span class="text-white font-bold">AI</span>
  339. </div>
  340. <div>
  341. <h1 class="text-xl font-bold text-gray-800">AI MCP Web UI</h1>
  342. <p class="text-sm text-gray-500">通用 MCP 服务器 Web 界面</p>
  343. </div>
  344. </div>
  345. <div id="status-indicator" class="flex items-center space-x-2">
  346. <span class="w-2 h-2 bg-green-500 rounded-full animate-pulse"></span>
  347. <span class="text-sm text-gray-600">已连接</span>
  348. </div>
  349. </div>
  350. </header>
  351. <!-- Chat Container -->
  352. <div class="bg-white rounded-lg shadow-md">
  353. <!-- Messages -->
  354. <div id="chat-messages" class="chat-container overflow-y-auto p-4 space-y-4">
  355. <!-- Welcome Message -->
  356. <div class="flex justify-start">
  357. <div class="message-assistant message-bubble rounded-lg p-4 max-w-[80%] shadow-sm">
  358. <p class="text-gray-800">你好!我是 AI MCP Web UI 助手。我可以帮助你:</p>
  359. <ul class="mt-2 text-gray-700 list-disc list-inside">
  360. <li>与 Claude AI 进行对话</li>
  361. <li>调用 MCP 服务器工具</li>
  362. <li>自动化各种任务</li>
  363. </ul>
  364. <p class="mt-2 text-gray-600 text-sm">请问有什么可以帮助你的?</p>
  365. </div>
  366. </div>
  367. </div>
  368. <!-- Input Area -->
  369. <div class="border-t border-gray-200 p-4 input-area">
  370. <form id="chat-form" class="flex space-x-3">
  371. <input
  372. type="text"
  373. id="user-input"
  374. placeholder="输入你的消息..."
  375. class="flex-1 px-4 py-3 border border-gray-300 rounded-lg focus:outline-none focus:ring-2 focus:ring-purple-500 focus:border-transparent"
  376. style="font-size: 16px;"
  377. autocomplete="off"
  378. >
  379. <button
  380. type="submit"
  381. id="send-button"
  382. class="px-6 py-3 bg-gradient-to-r from-purple-500 to-indigo-600 text-white rounded-lg hover:from-purple-600 hover:to-indigo-700 focus:outline-none focus:ring-2 focus:ring-purple-500 focus:ring-offset-2 transition-all disabled:opacity-50 disabled:cursor-not-allowed"
  383. style="font-size: 16px;"
  384. >
  385. 发送
  386. </button>
  387. </form>
  388. </div>
  389. </div>
  390. <!-- MCP Servers Status -->
  391. <div class="mt-4 bg-white rounded-lg shadow-md p-4">
  392. <h2 class="text-sm font-semibold text-gray-800 mb-2">MCP 服务器</h2>
  393. <div id="mcp-servers" class="mcp-servers-scroll">
  394. <!-- MCP server tags will be inserted here -->
  395. </div>
  396. </div>
  397. </div>
  398. <!-- 登录对话框 -->
  399. <div id="login-modal" class="hidden fixed inset-0 bg-black bg-opacity-50 flex items-center justify-center z-50">
  400. <div class="bg-white rounded-lg p-6 max-w-md w-full mx-4 shadow-2xl">
  401. <div class="flex items-center justify-between mb-4">
  402. <h3 class="text-xl font-bold text-gray-800" id="login-title">登录 Novel Platform</h3>
  403. <button id="login-close-x" class="text-gray-500 hover:text-gray-700 text-2xl">&times;</button>
  404. </div>
  405. <p id="login-description" class="text-sm text-gray-600 mb-4">请输入您的账号密码以访问需要认证的 MCP 工具</p>
  406. <form id="login-form">
  407. <div class="mb-3">
  408. <label for="login-username" class="block text-sm font-medium text-gray-700 mb-1">用户名/邮箱</label>
  409. <input type="text" id="login-username" placeholder="请输入用户名或邮箱" class="w-full px-3 py-2 border border-gray-300 rounded-lg focus:outline-none focus:ring-2 focus:ring-purple-500 focus:border-transparent">
  410. </div>
  411. <div class="mb-3">
  412. <label for="login-password" class="block text-sm font-medium text-gray-700 mb-1">密码</label>
  413. <input type="password" id="login-password" placeholder="请输入密码" class="w-full px-3 py-2 border border-gray-300 rounded-lg focus:outline-none focus:ring-2 focus:ring-purple-500 focus:border-transparent">
  414. </div>
  415. <p id="login-error" class="text-red-500 text-sm mb-3 hidden"></p>
  416. <div class="flex justify-end gap-2">
  417. <button type="button" id="login-cancel" class="px-4 py-2 border border-gray-300 rounded-lg hover:bg-gray-50 transition-colors">取消</button>
  418. <button type="submit" id="login-submit" class="px-4 py-2 bg-purple-600 text-white rounded-lg hover:bg-purple-700 transition-colors disabled:opacity-50 disabled:cursor-not-allowed">登录</button>
  419. </div>
  420. </form>
  421. </div>
  422. </div>
  423. <script>
  424. // State
  425. let conversationHistory = [];
  426. let isTyping = false;
  427. // DOM Elements
  428. const chatMessages = document.getElementById('chat-messages');
  429. const chatForm = document.getElementById('chat-form');
  430. const userInput = document.getElementById('user-input');
  431. const sendButton = document.getElementById('send-button');
  432. const mcpServers = document.getElementById('mcp-servers');
  433. const statusIndicator = document.getElementById('status-indicator');
  434. // API Base URL
  435. const API_BASE = window.location.origin;
  436. // ========== MCP 服务器配置 ==========
  437. const MCP_SERVERS = {
  438. "novel-translator": {
  439. "name": "Novel Translator MCP",
  440. "auth_type": "none"
  441. },
  442. "novel-platform-user": {
  443. "name": "Novel Platform User MCP",
  444. "url": "https://d8d-ai-vscode-8080-223-238-template-6-group.dev.d8d.fun/mcp/",
  445. "auth_type": "jwt",
  446. "login_api": "/api/v1/auth/login",
  447. "base_url": "https://d8d-ai-vscode-8080-223-238-template-6-group.dev.d8d.fun"
  448. },
  449. "novel-platform-admin": {
  450. "name": "Novel Platform Admin MCP",
  451. "url": "https://d8d-ai-vscode-8080-223-238-template-6-group.dev.d8d.fun/mcp/",
  452. "auth_type": "jwt",
  453. "login_api": "/api/v1/auth/admin-login",
  454. "base_url": "https://d8d-ai-vscode-8080-223-238-template-6-group.dev.d8d.fun"
  455. }
  456. };
  457. // ========== Token 管理器 ==========
  458. const TokenManager = {
  459. // 保存 token
  460. saveToken(mcpType, token, username = null) {
  461. localStorage.setItem(`mcp_token_${mcpType}`, token);
  462. localStorage.setItem(`mcp_token_${mcpType}_time`, Date.now());
  463. if (username) {
  464. localStorage.setItem(`mcp_username_${mcpType}`, username);
  465. }
  466. },
  467. // 获取 token
  468. getToken(mcpType) {
  469. return localStorage.getItem(`mcp_token_${mcpType}`);
  470. },
  471. // 获取用户名
  472. getUsername(mcpType) {
  473. return localStorage.getItem(`mcp_username_${mcpType}`);
  474. },
  475. // 清除 token
  476. clearToken(mcpType) {
  477. localStorage.removeItem(`mcp_token_${mcpType}`);
  478. localStorage.removeItem(`mcp_token_${mcpType}_time`);
  479. localStorage.removeItem(`mcp_username_${mcpType}`);
  480. },
  481. // 检查是否已登录
  482. isLoggedIn(mcpType) {
  483. return !!this.getToken(mcpType);
  484. },
  485. // 获取 token 存储时长(毫秒)
  486. getTokenAge(mcpType) {
  487. const time = localStorage.getItem(`mcp_token_${mcpType}_time`);
  488. return time ? (Date.now() - parseInt(time)) : null;
  489. }
  490. };
  491. // ========== 登录相关 ==========
  492. let currentLoginMcpType = null;
  493. const loginModal = document.getElementById('login-modal');
  494. const loginForm = document.getElementById('login-form');
  495. const loginUsername = document.getElementById('login-username');
  496. const loginPassword = document.getElementById('login-password');
  497. const loginError = document.getElementById('login-error');
  498. const loginTitle = document.getElementById('login-title');
  499. const loginSubmit = document.getElementById('login-submit');
  500. // 显示登录对话框
  501. function showLoginModal(mcpType) {
  502. currentLoginMcpType = mcpType;
  503. const config = MCP_SERVERS[mcpType];
  504. if (!config) return;
  505. loginTitle.textContent = `登录 ${config.name}`;
  506. loginUsername.value = '';
  507. loginPassword.value = '';
  508. loginError.classList.add('hidden');
  509. loginSubmit.disabled = false;
  510. loginModal.classList.remove('hidden');
  511. loginUsername.focus();
  512. }
  513. // 隐藏登录对话框
  514. function hideLoginModal() {
  515. loginModal.classList.add('hidden');
  516. currentLoginMcpType = null;
  517. }
  518. // 执行登录 - 使用本地后端代理避免 CORS 问题
  519. async function performLogin(mcpType, username, password) {
  520. const config = MCP_SERVERS[mcpType];
  521. if (!config || config.auth_type !== 'jwt') {
  522. return { success: false, error: '该 MCP 不需要登录' };
  523. }
  524. // 使用本地后端代理端点,避免直接调用外部 API 导致 CORS 错误
  525. const is_admin = mcpType === 'novel-platform-admin';
  526. const proxyUrl = is_admin ? '/api/auth/admin-login' : '/api/auth/login';
  527. try {
  528. const response = await fetch(proxyUrl, {
  529. method: 'POST',
  530. headers: { 'Content-Type': 'application/json' },
  531. body: JSON.stringify({
  532. email: username,
  533. password: password
  534. })
  535. });
  536. const data = await response.json();
  537. if (response.ok && data.success) {
  538. const token = data.token;
  539. TokenManager.saveToken(mcpType, token, data.username || username);
  540. return { success: true, token, username: data.username || username };
  541. } else {
  542. return { success: false, error: data.detail || data.error || '登录失败' };
  543. }
  544. } catch (e) {
  545. return { success: false, error: e.message };
  546. }
  547. }
  548. // 登录表单提交处理
  549. loginForm.addEventListener('submit', async (e) => {
  550. e.preventDefault();
  551. if (!currentLoginMcpType) return;
  552. const username = loginUsername.value.trim();
  553. const password = loginPassword.value;
  554. if (!username || !password) {
  555. loginError.textContent = '请输入用户名和密码';
  556. loginError.classList.remove('hidden');
  557. return;
  558. }
  559. loginSubmit.disabled = true;
  560. loginError.classList.add('hidden');
  561. const result = await performLogin(currentLoginMcpType, username, password);
  562. if (result.success) {
  563. hideLoginModal();
  564. loadMCPServers(); // 刷新 MCP 服务器列表
  565. } else {
  566. loginError.textContent = result.error;
  567. loginError.classList.remove('hidden');
  568. loginSubmit.disabled = false;
  569. }
  570. });
  571. // 登录对话框事件绑定
  572. document.getElementById('login-cancel').addEventListener('click', hideLoginModal);
  573. document.getElementById('login-close-x').addEventListener('click', hideLoginModal);
  574. // 点击背景关闭
  575. loginModal.addEventListener('click', (e) => {
  576. if (e.target === loginModal) {
  577. hideLoginModal();
  578. }
  579. });
  580. // ========== Token 传递到后端 ==========
  581. // 为每个请求添加 Authorization header
  582. async function fetchWithAuth(url, options = {}) {
  583. // 检查 URL 路径,确定是哪个 MCP
  584. let mcpType = null;
  585. if (url.includes('/chat') || url.includes('/mcp')) {
  586. // 对于需要认证的 MCP,检查本地存储的 token
  587. for (const [type, config] of Object.entries(MCP_SERVERS)) {
  588. if (config.auth_type === 'jwt' && TokenManager.isLoggedIn(type)) {
  589. mcpType = type;
  590. break;
  591. }
  592. }
  593. }
  594. // 添加 token 到请求头
  595. if (mcpType) {
  596. const token = TokenManager.getToken(mcpType);
  597. if (token) {
  598. options.headers = options.headers || {};
  599. options.headers['X-MCP-Token'] = token;
  600. options.headers['X-MCP-Type'] = mcpType;
  601. }
  602. }
  603. return fetch(url, options);
  604. }
  605. // ========== Format message content (handle code blocks, etc.) ==========
  606. function formatMessage(content) {
  607. // Simple markdown-like formatting
  608. let formatted = content
  609. .replace(/```(\w+)?\n([\s\S]*?)```/g, '<pre><code>$2</code></pre>')
  610. .replace(/`([^`]+)`/g, '<code class="bg-gray-200 px-1 rounded">$1</code>')
  611. .replace(/\n/g, '<br>');
  612. return formatted;
  613. }
  614. // Add message to chat
  615. function addMessage(role, content) {
  616. const messageDiv = document.createElement('div');
  617. messageDiv.className = `flex ${role === 'user' ? 'justify-end' : 'justify-start'}`;
  618. const bubbleClass = role === 'user' ? 'message-user text-white' : 'message-assistant text-gray-800';
  619. messageDiv.innerHTML = `
  620. <div class="${bubbleClass} message-bubble rounded-lg p-4 max-w-[80%] shadow-sm">
  621. <div>${formatMessage(content)}</div>
  622. </div>
  623. `;
  624. chatMessages.appendChild(messageDiv);
  625. chatMessages.scrollTop = chatMessages.scrollHeight;
  626. }
  627. // Add typing indicator
  628. function addTypingIndicator() {
  629. const typingDiv = document.createElement('div');
  630. typingDiv.id = 'typing-indicator';
  631. typingDiv.className = 'flex justify-start';
  632. typingDiv.innerHTML = `
  633. <div class="message-assistant rounded-lg p-4 shadow-sm">
  634. <div class="typing-indicator flex space-x-1">
  635. <span class="w-2 h-2 bg-gray-500 rounded-full"></span>
  636. <span class="w-2 h-2 bg-gray-500 rounded-full"></span>
  637. <span class="w-2 h-2 bg-gray-500 rounded-full"></span>
  638. </div>
  639. </div>
  640. `;
  641. chatMessages.appendChild(typingDiv);
  642. chatMessages.scrollTop = chatMessages.scrollHeight;
  643. }
  644. // Remove typing indicator
  645. function removeTypingIndicator() {
  646. const indicator = document.getElementById('typing-indicator');
  647. if (indicator) indicator.remove();
  648. }
  649. // 创建工具调用面板
  650. function createToolCallPanel() {
  651. const panel = document.createElement('div');
  652. panel.className = 'tool-call-panel';
  653. panel.innerHTML = `
  654. <div class="tool-panel-header" onclick="toggleToolPanel(this)">
  655. <div class="tool-panel-title">
  656. <span class="expand-icon">▶</span>
  657. <span>🔧 工具调用</span>
  658. <span class="tool-count">(0)</span>
  659. </div>
  660. <div class="process-indicator">
  661. <div class="spinner"></div>
  662. <span>处理中...</span>
  663. </div>
  664. </div>
  665. <div class="tool-panel-content">
  666. <div class="tool-calls-container"></div>
  667. </div>
  668. `;
  669. return panel;
  670. }
  671. // 切换工具面板展开/收起
  672. function toggleToolPanel(header) {
  673. const content = header.nextElementSibling;
  674. const icon = header.querySelector('.expand-icon');
  675. content.classList.toggle('expanded');
  676. icon.classList.toggle('expanded');
  677. }
  678. // 切换参数/结果折叠区域
  679. function toggleCollapsible(header) {
  680. const content = header.nextElementSibling;
  681. const icon = header.querySelector('.collapsible-icon');
  682. content.classList.toggle('expanded');
  683. icon.classList.toggle('expanded');
  684. }
  685. // 添加工具调用项
  686. function addToolCallItem(container, toolName, params, toolUseId = null) {
  687. const item = document.createElement('div');
  688. item.className = 'tool-call-item pending';
  689. const uniqueId = `tool-${Date.now()}-${Math.random().toString(36).substr(2, 9)}`;
  690. item.id = uniqueId;
  691. // 存储 tool_use_id 和 tool_name 以便后续匹配
  692. if (toolUseId) {
  693. item.dataset.toolUseId = toolUseId;
  694. }
  695. item.dataset.toolName = toolName;
  696. console.log('[addToolCallItem] Created item:', { id: uniqueId, toolName, toolUseId });
  697. // 简化工具名称显示 - 去掉 mcp__ 前缀和下划线
  698. let displayName = toolName;
  699. if (toolName.startsWith('mcp__')) {
  700. displayName = toolName.replace('mcp__', '').replace(/_/g, ' ');
  701. }
  702. // 首字母大写
  703. displayName = displayName.charAt(0).toUpperCase() + displayName.slice(1);
  704. const paramsStr = params ? JSON.stringify(params, null, 2) : '{}';
  705. const hasParams = params && Object.keys(params).length > 0;
  706. item.innerHTML = `
  707. <div class="tool-call-header">
  708. <div class="tool-name">
  709. <span>🔧</span>
  710. <span>${escapeHtml(displayName)}</span>
  711. </div>
  712. <div class="tool-status pending">
  713. <div class="spinner"></div>
  714. <span>执行中</span>
  715. </div>
  716. </div>
  717. ${hasParams ? `
  718. <div class="collapsible-section">
  719. <div class="collapsible-header" onclick="toggleCollapsible(this)">
  720. <span class="collapsible-icon">▶</span>
  721. <span>查看参数</span>
  722. </div>
  723. <div class="collapsible-content">
  724. <div class="tool-params"><pre style="margin:0;padding:0;background:transparent;font-size:10px;">${escapeHtml(paramsStr)}</pre></div>
  725. </div>
  726. </div>
  727. ` : ''}
  728. `;
  729. container.appendChild(item);
  730. return { id: uniqueId, toolUseId: toolUseId, toolName: toolName };
  731. }
  732. // 更新工具调用状态
  733. function updateToolCallStatus(itemId, status, result = null) {
  734. const item = document.getElementById(itemId);
  735. if (!item) {
  736. console.warn('[updateToolCallStatus] Item not found:', itemId);
  737. return;
  738. }
  739. const statusEl = item.querySelector('.tool-status');
  740. item.classList.remove('pending', 'success', 'error');
  741. if (status === 'success') {
  742. item.classList.add('success');
  743. statusEl.className = 'tool-status success';
  744. statusEl.innerHTML = '✅ 完成';
  745. // 总是显示结果区域(即使结果为空)
  746. const resultText = result ? (typeof result === 'string' ? result : JSON.stringify(result, null, 2)) : '';
  747. const displayResult = resultText || '(无返回结果)';
  748. const truncatedResult = displayResult.substring(0, 1000) + (displayResult.length > 1000 ? '...' : '');
  749. const resultDiv = document.createElement('div');
  750. resultDiv.className = 'collapsible-section';
  751. resultDiv.innerHTML = `
  752. <div class="collapsible-header" onclick="toggleCollapsible(this)">
  753. <span class="collapsible-icon">▶</span>
  754. <span>查看结果</span>
  755. </div>
  756. <div class="collapsible-content">
  757. <div class="tool-result"><div class="tool-result-content">${escapeHtml(truncatedResult)}</div></div>
  758. </div>
  759. `;
  760. item.appendChild(resultDiv);
  761. } else if (status === 'error') {
  762. item.classList.add('error');
  763. statusEl.className = 'tool-status error';
  764. statusEl.innerHTML = '❌ 失败';
  765. const errorDiv = document.createElement('div');
  766. errorDiv.className = 'collapsible-section';
  767. errorDiv.innerHTML = `
  768. <div class="collapsible-header" onclick="toggleCollapsible(this)">
  769. <span class="collapsible-icon">▶</span>
  770. <span>查看错误</span>
  771. </div>
  772. <div class="collapsible-content expanded">
  773. <div class="tool-result error">${escapeHtml(result || '未知错误')}</div>
  774. </div>
  775. `;
  776. item.appendChild(errorDiv);
  777. }
  778. }
  779. // 更新工具面板状态
  780. function updateToolPanelStatus(panel, status, totalCalls = 0) {
  781. const indicator = panel.querySelector('.process-indicator');
  782. const countEl = panel.querySelector('.tool-count');
  783. const content = panel.querySelector('.tool-panel-content');
  784. const icon = panel.querySelector('.expand-icon');
  785. countEl.textContent = `(${totalCalls})`;
  786. if (status === 'running') {
  787. // 保持处理中状态,不改变
  788. indicator.className = 'process-indicator';
  789. indicator.innerHTML = `<div class="spinner"></div><span>处理中...</span>`;
  790. } else if (status === 'complete') {
  791. indicator.className = 'process-indicator success';
  792. indicator.innerHTML = `<span>✅ 完成 - ${totalCalls} 个工具调用</span>`;
  793. // 自动展开面板以便查看结果
  794. if (content && !content.classList.contains('expanded')) {
  795. content.classList.add('expanded');
  796. if (icon) icon.classList.add('expanded');
  797. }
  798. } else if (status === 'error') {
  799. indicator.className = 'process-indicator';
  800. indicator.style.background = 'rgba(239, 68, 68, 0.1)';
  801. indicator.style.color = '#f87171';
  802. indicator.innerHTML = `<span>❌ 部分失败</span>`;
  803. }
  804. }
  805. // HTML 转义
  806. function escapeHtml(text) {
  807. const div = document.createElement('div');
  808. div.textContent = text;
  809. return div.innerHTML;
  810. }
  811. function scrollToBottom() {
  812. chatMessages.scrollTop = chatMessages.scrollHeight;
  813. }
  814. // ========== 流式聊天功能 ==========
  815. async function chatStream(message, onEvent) {
  816. const events = [];
  817. try {
  818. // 收集所有已登录 MCP 的 tokens
  819. const mcpTokens = {};
  820. for (const [type, config] of Object.entries(MCP_SERVERS)) {
  821. if (config.auth_type === 'jwt' && TokenManager.isLoggedIn(type)) {
  822. mcpTokens[type] = TokenManager.getToken(type);
  823. }
  824. }
  825. const headers = { 'Content-Type': 'application/json' };
  826. // 将 tokens 作为 JSON 字符串传递
  827. if (Object.keys(mcpTokens).length > 0) {
  828. headers['X-MCP-Tokens'] = JSON.stringify(mcpTokens);
  829. }
  830. const response = await fetch('/api/chat/stream', {
  831. method: 'POST',
  832. headers: headers,
  833. body: JSON.stringify({ message, history: conversationHistory })
  834. });
  835. if (!response.ok) {
  836. throw new Error(`HTTP error! status: ${response.status}`);
  837. }
  838. const reader = response.body.getReader();
  839. const decoder = new TextDecoder();
  840. let buffer = '';
  841. while (true) {
  842. const { done, value } = await reader.read();
  843. if (done) break;
  844. buffer += decoder.decode(value, { stream: true });
  845. // 处理 SSE 格式: "event: xxx\ndata: {...}\n\n"
  846. const lines = buffer.split('\n');
  847. buffer = lines.pop() || ''; // 保留未完成的行
  848. for (let i = 0; i < lines.length; i++) {
  849. const line = lines[i].trim();
  850. if (!line) continue;
  851. if (line.startsWith('event:')) {
  852. const eventType = line.substring(6).trim();
  853. events.push({ type: eventType, data: null });
  854. } else if (line.startsWith('data:')) {
  855. const data = line.substring(5).trim();
  856. if (events.length > 0) {
  857. events[events.length - 1].data = data;
  858. }
  859. }
  860. // 处理完整的事件
  861. while (events.length > 0 && events[0].data) {
  862. const event = events.shift();
  863. try {
  864. const data = JSON.parse(event.data);
  865. onEvent(event.type, data);
  866. } catch (e) {
  867. console.error('Parse error:', e, event.data);
  868. }
  869. }
  870. }
  871. }
  872. } catch (error) {
  873. onEvent('error', { error: error.message });
  874. }
  875. }
  876. async function sendMessageStream(message) {
  877. if (isTyping) return;
  878. isTyping = true;
  879. sendButton.disabled = true;
  880. // 添加用户消息
  881. addMessage('user', message);
  882. conversationHistory.push({ role: 'user', content: message });
  883. // 创建助手消息容器
  884. const assistantMsgDiv = document.createElement('div');
  885. assistantMsgDiv.className = 'flex justify-start';
  886. assistantMsgDiv.innerHTML = `
  887. <div class="message-assistant message-bubble rounded-lg p-4 shadow-sm">
  888. <div class="response-content">
  889. <div class="text" id="currentResponse"></div>
  890. </div>
  891. <div id="toolCallContainer"></div>
  892. </div>
  893. `;
  894. chatMessages.appendChild(assistantMsgDiv);
  895. const responseDiv = assistantMsgDiv.querySelector('#currentResponse');
  896. const toolContainer = assistantMsgDiv.querySelector('#toolCallContainer');
  897. let currentText = '';
  898. let toolCallsCount = 0;
  899. let toolCallItems = []; // 改为数组存储,按索引匹配
  900. let finalResponse = '';
  901. let toolCallPanel = null;
  902. let toolCallsContainer = null;
  903. await chatStream(message, (eventType, data) => {
  904. switch (eventType) {
  905. case 'start':
  906. responseDiv.innerHTML = '<span style="color: #888; font-style: italic;">🤔 AI 正在思考...</span>';
  907. break;
  908. case 'token':
  909. // 移除思考提示
  910. if (currentText === '') {
  911. responseDiv.innerHTML = '';
  912. }
  913. currentText += data.text || '';
  914. responseDiv.innerHTML = formatMessage(currentText);
  915. scrollToBottom();
  916. break;
  917. case 'tool_call':
  918. // 首次工具调用时创建面板
  919. if (!toolCallPanel) {
  920. toolCallPanel = createToolCallPanel();
  921. toolContainer.appendChild(toolCallPanel);
  922. toolCallsContainer = toolCallPanel.querySelector('.tool-calls-container');
  923. // 清空思考提示
  924. if (currentText === '' || currentText.includes('正在思考')) {
  925. responseDiv.innerHTML = '<span style="color: #888; font-style: italic;">🔧 正在调用工具...</span>';
  926. }
  927. }
  928. // 添加工具调用项
  929. const toolName = data.tool || 'unknown';
  930. const params = data.args || data.params || null;
  931. const toolUseId = data.tool_id || null;
  932. console.log('[tool_call] Adding tool:', { toolName, toolUseId, params });
  933. const itemInfo = addToolCallItem(toolCallsContainer, toolName, params, toolUseId);
  934. toolCallItems.push(itemInfo);
  935. toolCallsCount++;
  936. console.log('[tool_call] Current toolCallItems:', toolCallItems);
  937. updateToolPanelStatus(toolCallPanel, 'running', toolCallsCount);
  938. scrollToBottom();
  939. break;
  940. case 'tools_start':
  941. // 批量工具开始
  942. if (!toolCallPanel) {
  943. toolCallPanel = createToolCallPanel();
  944. toolContainer.appendChild(toolCallPanel);
  945. toolCallsContainer = toolCallPanel.querySelector('.tool-calls-container');
  946. }
  947. if (currentText === '' || currentText.includes('正在思考')) {
  948. responseDiv.innerHTML = '<span style="color: #888; font-style: italic;">🔧 正在调用工具...</span>';
  949. }
  950. scrollToBottom();
  951. break;
  952. case 'tool_done':
  953. // 工具执行完成 - 按 tool_use_id 或 tool_name 匹配
  954. const doneToolUseId = data.tool_id;
  955. const doneToolName = data.tool;
  956. const doneResult = data.result;
  957. console.log('[tool_done] Looking for tool:', { tool_id: doneToolUseId, tool: doneToolName });
  958. console.log('[tool_done] Current toolCallItems:', toolCallItems);
  959. // 查找匹配的工具调用项
  960. const doneItem = toolCallItems.find(item =>
  961. item.toolUseId === doneToolUseId || item.toolName === doneToolName
  962. );
  963. if (doneItem) {
  964. console.log('[tool_done] Found item, updating status:', doneItem.id);
  965. updateToolCallStatus(doneItem.id, 'success', doneResult);
  966. } else {
  967. console.warn('[tool_done] Tool item not found for:', { tool_id: doneToolUseId, tool: doneToolName });
  968. // 如果找不到匹配项,尝试通过 DOM 查找
  969. const fallbackItem = Array.from(document.querySelectorAll('.tool-call-item')).find(el => {
  970. return el.dataset.toolUseId === doneToolUseId || el.dataset.toolName === doneToolName;
  971. });
  972. if (fallbackItem) {
  973. console.log('[tool_done] Found via DOM fallback:', fallbackItem.id);
  974. updateToolCallStatus(fallbackItem.id, 'success', doneResult);
  975. }
  976. }
  977. scrollToBottom();
  978. break;
  979. case 'tool_error':
  980. // 工具执行错误 - 按 tool_use_id 或 tool_name 匹配
  981. const errorToolUseId = data.tool_id;
  982. const errorToolName = data.tool;
  983. const errorItem = toolCallItems.find(item =>
  984. item.toolUseId === errorToolUseId || item.toolName === errorToolName
  985. );
  986. if (errorItem) {
  987. updateToolCallStatus(errorItem.id, 'error', data.error);
  988. }
  989. if (toolCallPanel) {
  990. updateToolPanelStatus(toolCallPanel, 'error', toolCallsCount);
  991. }
  992. scrollToBottom();
  993. break;
  994. case 'complete':
  995. // 完成
  996. console.log('[complete] Tool calls finished, updating panel status');
  997. if (toolCallPanel) {
  998. updateToolPanelStatus(toolCallPanel, 'complete', toolCallsCount);
  999. }
  1000. // 清空临时提示并显示最终响应
  1001. if (data.response || currentText) {
  1002. finalResponse = data.response || currentText;
  1003. responseDiv.innerHTML = formatMessage(finalResponse);
  1004. }
  1005. conversationHistory.push({ role: 'assistant', content: finalResponse });
  1006. scrollToBottom();
  1007. break;
  1008. case 'error':
  1009. responseDiv.innerHTML = `<span style="color: #ef4444;">❌ 错误: ${escapeHtml(data.error || '未知错误')}</span>`;
  1010. if (toolCallPanel) {
  1011. updateToolPanelStatus(toolCallPanel, 'error', toolCallsCount);
  1012. }
  1013. scrollToBottom();
  1014. break;
  1015. }
  1016. });
  1017. // 重置状态
  1018. isTyping = false;
  1019. sendButton.disabled = false;
  1020. userInput.focus();
  1021. }
  1022. // Load MCP servers
  1023. async function loadMCPServers() {
  1024. try {
  1025. const response = await fetch(`${API_BASE}/api/mcp/servers`);
  1026. const data = await response.json();
  1027. mcpServers.innerHTML = data.servers.map(server => {
  1028. const mcpType = server.id;
  1029. const isLoggedIn = TokenManager.isLoggedIn(mcpType);
  1030. const username = TokenManager.getUsername(mcpType);
  1031. let authStatus = '';
  1032. let clickHandler = '';
  1033. if (server.auth_type === 'jwt') {
  1034. if (isLoggedIn) {
  1035. authStatus = `<span class="text-green-600 font-medium" title="已登录为 ${username}">✅ ${username || '已登录'}</span>`;
  1036. clickHandler = `onclick="handleMcpClick('${mcpType}', '${server.auth_type}')"`;
  1037. } else {
  1038. authStatus = '<span class="text-amber-600" title="需要登录">🔒 需要登录</span>';
  1039. clickHandler = `onclick="handleMcpClick('${mcpType}', '${server.auth_type}')"`;
  1040. }
  1041. } else {
  1042. authStatus = '<span class="text-gray-500">🔓 公开</span>';
  1043. }
  1044. return `
  1045. <div class="mcp-server-tag ${server.enabled ? 'enabled' : 'disabled'}" ${clickHandler} style="cursor: ${server.auth_type === 'jwt' ? 'pointer' : 'default'}">
  1046. <span class="mcp-status-dot ${server.enabled ? 'online' : 'offline'}"></span>
  1047. <span class="font-medium">${server.name}</span>
  1048. ${authStatus}
  1049. </div>
  1050. `;
  1051. }).join('');
  1052. } catch (error) {
  1053. console.error('Failed to load MCP servers:', error);
  1054. }
  1055. }
  1056. // 处理 MCP 服务器点击
  1057. function handleMcpClick(mcpType, authType) {
  1058. if (authType !== 'jwt') return;
  1059. if (TokenManager.isLoggedIn(mcpType)) {
  1060. // 已登录,询问是否登出
  1061. const username = TokenManager.getUsername(mcpType);
  1062. if (confirm(`当前已登录为 ${username}\n\n是否要登出?`)) {
  1063. TokenManager.clearToken(mcpType);
  1064. loadMCPServers();
  1065. }
  1066. } else {
  1067. // 未登录,显示登录对话框
  1068. showLoginModal(mcpType);
  1069. }
  1070. }
  1071. // 将 handleMcpClick 暴露到全局作用域
  1072. window.handleMcpClick = handleMcpClick;
  1073. // Check health
  1074. async function checkHealth() {
  1075. try {
  1076. const response = await fetch(`${API_BASE}/api/health`);
  1077. if (response.ok) {
  1078. statusIndicator.innerHTML = `
  1079. <span class="w-2 h-2 bg-green-500 rounded-full animate-pulse"></span>
  1080. <span class="text-sm text-gray-600">已连接</span>
  1081. `;
  1082. } else {
  1083. throw new Error('Health check failed');
  1084. }
  1085. } catch (error) {
  1086. statusIndicator.innerHTML = `
  1087. <span class="w-2 h-2 bg-red-500 rounded-full"></span>
  1088. <span class="text-sm text-gray-600">连接失败</span>
  1089. `;
  1090. }
  1091. }
  1092. // Event listeners
  1093. chatForm.addEventListener('submit', (e) => {
  1094. e.preventDefault();
  1095. const message = userInput.value.trim();
  1096. if (message) {
  1097. sendMessageStream(message);
  1098. userInput.value = '';
  1099. }
  1100. });
  1101. // Initialize
  1102. loadMCPServers();
  1103. checkHealth();
  1104. userInput.focus();
  1105. // Make toggle functions global
  1106. window.toggleToolPanel = toggleToolPanel;
  1107. window.toggleCollapsible = toggleCollapsible;
  1108. </script>
  1109. </body>
  1110. </html>