app.js 19 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411
  1. 'use strict';
  2. // 局域网访问口令:服务端开启 accessToken 时,API 返回 401 会引导输入口令,
  3. // 写入 cookie 后所有请求(fetch / 图片 / CSV 链接)自动携带,无需逐处改代码。
  4. // 并发 401 共享同一次弹窗;用户取消只结束本轮,后续 401 仍可再次询问。
  5. (function () {
  6. const origFetch = window.fetch.bind(window);
  7. let tokenPromise = null;
  8. window.fetch = async function (input, init) {
  9. const res = await origFetch(input, init);
  10. const url = typeof input === 'string' ? input : (input && input.url) || String(input || '');
  11. if (res.status === 401 && /\/(api|files)\b/.test(url)) {
  12. if (!tokenPromise) {
  13. tokenPromise = (async () => {
  14. const token = window.prompt('此服务已开启访问口令(accessToken),请输入:');
  15. if (token) {
  16. document.cookie = 'accessToken=' + encodeURIComponent(token) + '; path=/; max-age=31536000; SameSite=Lax';
  17. }
  18. tokenPromise = null; // 本轮结束;取消后下次 401 仍会询问
  19. return token || '';
  20. })();
  21. }
  22. const token = await tokenPromise;
  23. if (token) return origFetch(input, init);
  24. }
  25. return res;
  26. };
  27. })();
  28. (function () {
  29. const views = {
  30. crawl: document.getElementById('viewCrawl'),
  31. gen: document.getElementById('viewGen'),
  32. tasks: document.getElementById('viewTasks'),
  33. history: document.getElementById('viewHistory'),
  34. reports: document.getElementById('viewReports'),
  35. lake: document.getElementById('viewLake'),
  36. analysis: document.getElementById('viewAnalysis'),
  37. ask: document.getElementById('viewAsk'),
  38. settings: document.getElementById('viewSettings'),
  39. logs: document.getElementById('viewLogs'),
  40. };
  41. const navItems = document.querySelectorAll('.nav-item[data-view]');
  42. let settingsBound = false;
  43. let settingsLoaded = false;
  44. let logTimer = null;
  45. let lastLogSignature = '';
  46. async function loadModels(selectedModel) {
  47. const select = document.getElementById('settingsModel');
  48. const response = await fetch('/api/models');
  49. if (!response.ok) throw new Error('模型列表加载失败');
  50. const data = await response.json();
  51. select.innerHTML = '';
  52. for (const model of data.models || []) {
  53. const option = document.createElement('option');
  54. option.value = typeof model === 'string' ? model : model.id;
  55. option.textContent = typeof model === 'string' ? model : (model.label || model.name || model.id);
  56. option.selected = option.value === selectedModel;
  57. select.appendChild(option);
  58. }
  59. }
  60. async function loadSettings() {
  61. const response = await fetch('/api/settings');
  62. if (!response.ok) throw new Error('设置加载失败');
  63. const settings = await response.json();
  64. document.getElementById('settingsApiKey').value = settings.apiKey || '';
  65. document.getElementById('settingsApiBase').value = settings.apiBase || '';
  66. document.getElementById('settingsRootDir').value = settings.rootDir || '';
  67. document.getElementById('pageSizeInput').value = settings.pageSize || 50;
  68. document.getElementById('delayInput').value = settings.delayMs ?? 300;
  69. const cookieInput = document.getElementById('settingsCookie');
  70. cookieInput.value = '';
  71. cookieInput.placeholder = settings.cookieConfigured
  72. ? '已配置,输入新 Cookie 可更换'
  73. : '登录 melody.shop.ele.me 后从 DevTools 复制';
  74. document.getElementById('settingsChatModel').value = settings.chatModel || 'deepseek-v4.1-flash';
  75. document.getElementById('settingsChatApiBase').value = settings.chatApiBase || '';
  76. const chatKeyInput = document.getElementById('settingsChatApiKey');
  77. chatKeyInput.value = settings.chatApiKey || '';
  78. chatKeyInput.placeholder = settings.chatKeyFallback
  79. ? '留空复用上方 API Key'
  80. : (settings.chatApiKey ? '已配置,输入新 Key 可更换' : '留空复用上方 API Key;否则填对话模型专用 Key');
  81. const tokenInput = document.getElementById('settingsAccessToken');
  82. tokenInput.value = '';
  83. tokenInput.placeholder = settings.accessTokenSet ? '已设置,留空保持不变(输入 - 清除)' : '未设置';
  84. document.getElementById('settingsAitableEnabled').checked = !!settings.aitableEnabled;
  85. document.getElementById('settingsAitableAppKey').value = settings.aitableAppKey || '';
  86. document.getElementById('settingsAitableBaseId').value = settings.aitableBaseId || '';
  87. document.getElementById('settingsAitableSheetDetail').value = settings.aitableSheetDetail || '门店明细';
  88. document.getElementById('settingsAitableSheetChain').value = settings.aitableSheetChain || '连锁汇总';
  89. document.getElementById('settingsAitableSheetCompany').value = settings.aitableSheetCompany || '公司汇总';
  90. const aitableSecretInput = document.getElementById('settingsAitableAppSecret');
  91. aitableSecretInput.value = '';
  92. aitableSecretInput.placeholder = settings.aitableSecretSet ? '已设置,留空保持不变(输入 - 清除)' : '未设置';
  93. await loadModels(settings.model);
  94. }
  95. function setPanelStatus(badgeId, message, ok) {
  96. const badge = document.getElementById(badgeId);
  97. if (!badge) return;
  98. badge.textContent = message;
  99. badge.className = 'badge ' + (ok === true ? 'badge-ok' : ok === false ? 'badge-fail' : 'badge-info');
  100. }
  101. async function saveSettingsPart(fields, buttonId, badgeId, successText = '已保存') {
  102. const button = document.getElementById(buttonId);
  103. button.disabled = true;
  104. setPanelStatus(badgeId, '保存中…');
  105. try {
  106. const response = await fetch('/api/settings', {
  107. method: 'POST',
  108. headers: { 'Content-Type': 'application/json' },
  109. body: JSON.stringify(fields()),
  110. });
  111. const data = await response.json();
  112. if (!response.ok) throw new Error(data.error || `HTTP ${response.status}`);
  113. await loadSettings();
  114. setPanelStatus(badgeId, successText, true);
  115. } catch (error) {
  116. setPanelStatus(badgeId, error.message, false);
  117. } finally {
  118. button.disabled = false;
  119. }
  120. }
  121. const val = (id) => document.getElementById(id).value.trim();
  122. const hasNewSecret = (id) => { const v = val(id); return v && !v.includes('••••'); };
  123. function saveApiSettings() {
  124. return saveSettingsPart(() => ({
  125. apiKey: val('settingsApiKey'),
  126. model: document.getElementById('settingsModel').value,
  127. chatModel: document.getElementById('settingsChatModel').value,
  128. ...(hasNewSecret('settingsChatApiKey') ? { chatApiKey: val('settingsChatApiKey') } : {}),
  129. }), 'btnSaveApi', 'apiSaveStatus');
  130. }
  131. function saveCrawlSettings() {
  132. return saveSettingsPart(() => ({
  133. rootDir: val('settingsRootDir'),
  134. pageSize: Number(document.getElementById('pageSizeInput').value),
  135. delayMs: Number(document.getElementById('delayInput').value),
  136. ...(val('settingsCookie') ? { cookie: val('settingsCookie') } : {}),
  137. }), 'btnSaveCrawl', 'crawlSaveStatus');
  138. }
  139. function saveSecuritySettings() {
  140. return saveSettingsPart(() => ({
  141. ...(val('settingsAccessToken') ? { accessToken: val('settingsAccessToken') } : {}),
  142. }), 'btnSaveSecurity', 'securitySaveStatus');
  143. }
  144. function saveAitableSettings() {
  145. return saveSettingsPart(() => ({
  146. aitableEnabled: document.getElementById('settingsAitableEnabled').checked,
  147. aitableAppKey: val('settingsAitableAppKey'),
  148. aitableBaseId: val('settingsAitableBaseId'),
  149. aitableSheetDetail: val('settingsAitableSheetDetail'),
  150. aitableSheetChain: val('settingsAitableSheetChain'),
  151. aitableSheetCompany: val('settingsAitableSheetCompany'),
  152. ...(val('settingsAitableAppSecret') ? { aitableAppSecret: val('settingsAitableAppSecret') } : {}),
  153. }), 'btnSaveAitable', 'aitableSaveStatus');
  154. }
  155. async function testImageApi() {
  156. const status = document.getElementById('imageTestStatus');
  157. const button = document.getElementById('btnTestImage');
  158. button.disabled = true;
  159. status.textContent = '测试中…';
  160. status.className = 'badge badge-info';
  161. try {
  162. const response = await fetch('/api/image/test', { method: 'POST' });
  163. const data = await response.json();
  164. if (!response.ok) throw new Error(data.error || `HTTP ${response.status}`);
  165. if (data.statusCode === 0) throw new Error(data.error || '网络不可达');
  166. const model = data.model || '';
  167. if (data.statusCode === 200 && data.modelFound) {
  168. status.textContent = '连通正常,模型在列表中';
  169. status.className = 'badge badge-ok';
  170. } else if (data.statusCode === 200 && Array.isArray(data.models)) {
  171. status.textContent = `服务可达(${data.models.length} 个模型);列表中未找到 ${model}`;
  172. status.className = 'badge badge-warn';
  173. } else if (data.statusCode === 200) {
  174. status.textContent = '服务可达;该网关不提供模型列表,无法校验模型名';
  175. status.className = 'badge badge-warn';
  176. } else if (data.statusCode === 401 || data.statusCode === 403) {
  177. status.textContent = `服务返回 ${data.statusCode}:Key 可能无效`;
  178. status.className = 'badge badge-fail';
  179. } else {
  180. status.textContent = `服务返回 ${data.statusCode}`;
  181. status.className = 'badge badge-fail';
  182. }
  183. } catch (error) {
  184. status.textContent = '失败:' + error.message;
  185. status.className = 'badge badge-fail';
  186. } finally {
  187. button.disabled = false;
  188. }
  189. }
  190. async function testChatModel() {
  191. const status = document.getElementById('chatTestStatus');
  192. const button = document.getElementById('btnTestChat');
  193. button.disabled = true;
  194. status.textContent = '测试中…';
  195. status.className = 'badge badge-info';
  196. try {
  197. const response = await fetch('/api/chat/test', {
  198. method: 'POST',
  199. headers: { 'Content-Type': 'application/json' },
  200. body: JSON.stringify({
  201. chatModel: document.getElementById('settingsChatModel').value.trim(),
  202. chatApiKey: document.getElementById('settingsChatApiKey').value.trim(),
  203. }),
  204. });
  205. const data = await response.json();
  206. if (!response.ok) throw new Error(data.error || `HTTP ${response.status}`);
  207. status.textContent = `连通正常:${data.reply || 'OK'}`;
  208. status.className = 'badge badge-ok';
  209. } catch (error) {
  210. status.textContent = '失败:' + error.message;
  211. status.className = 'badge badge-fail';
  212. } finally {
  213. button.disabled = false;
  214. }
  215. }
  216. function bindSettingsTabs() {
  217. const tabs = document.getElementById('settingsApiTabs');
  218. if (!tabs) return;
  219. tabs.addEventListener('click', (event) => {
  220. const btn = event.target.closest('button[data-tab]');
  221. if (!btn || btn.classList.contains('active')) return;
  222. tabs.querySelectorAll('button').forEach((b) => b.classList.toggle('active', b === btn));
  223. document.getElementById('settingsTabImage').hidden = btn.dataset.tab !== 'image';
  224. document.getElementById('settingsTabChat').hidden = btn.dataset.tab !== 'chat';
  225. });
  226. }
  227. // 云端动作(测试/推/拉);badgeId 允许不同页面各自展示状态(设置页/数据采集页/数据中台页)
  228. async function aitableAction(kind, badgeId = 'aitableStatus', body = {}) {
  229. const status = document.getElementById(badgeId);
  230. const labels = { test: '测试连通中…', push: '推送中…', pull: '拉取中…' };
  231. if (status) {
  232. status.textContent = labels[kind];
  233. status.className = 'badge badge-info';
  234. }
  235. try {
  236. const res = await fetch(`/api/reports/aitable/${kind}`, { method: 'POST', headers: { 'Content-Type': 'application/json' }, body: JSON.stringify(body) });
  237. const data = await res.json();
  238. if (!res.ok) throw new Error(data.error || `HTTP ${res.status}`);
  239. if (kind === 'test') {
  240. // 逐表探测:缺哪张表报哪张(testConnection 返回 { ok, sheets: { detail/chain/company: {name, ok, hint} } })
  241. const failed = Object.values(data.sheets || {}).filter((s) => !s.ok);
  242. if (status) {
  243. if (data.ok) { status.textContent = '三张表连通正常'; status.className = 'badge badge-ok'; }
  244. else {
  245. status.textContent = `不可读:${failed.map((s) => s.name).join('、')}`;
  246. status.className = 'badge badge-fail';
  247. }
  248. }
  249. return data;
  250. }
  251. if (kind === 'push') {
  252. const stale = data.skippedStale ? `,跳过云端较新 ${data.skippedStale} 行` : '';
  253. if (status) { status.textContent = `已推送 ${data.pushed} 行(新增 ${data.created} / 更新 ${data.updated}${stale})`; status.className = 'badge badge-ok'; }
  254. } else {
  255. const detail = (data.accounts || []).map((a) => `${a.account} ${a.rows}行`).join('、') || '无数据';
  256. if (status) { status.textContent = `已拉取 ${data.pulled} 行:${detail}`; status.className = 'badge badge-ok'; }
  257. }
  258. return data;
  259. } catch (error) {
  260. if (status) { status.textContent = error.message; status.className = 'badge badge-fail'; }
  261. throw error;
  262. }
  263. }
  264. window.shanAitableAction = aitableAction;
  265. async function initSettings() {
  266. // 事件绑定与数据加载分离:加载失败重进设置页时不会重复 addEventListener
  267. if (!settingsBound) {
  268. settingsBound = true;
  269. bindSettingsTabs();
  270. if (new URLSearchParams(location.search).get('tab') === 'chat') {
  271. document.querySelector('#settingsApiTabs button[data-tab="chat"]')?.click();
  272. }
  273. document.getElementById('btnSaveApi').addEventListener('click', saveApiSettings);
  274. document.getElementById('btnSaveCrawl').addEventListener('click', saveCrawlSettings);
  275. document.getElementById('btnSaveSecurity').addEventListener('click', saveSecuritySettings);
  276. document.getElementById('btnSaveAitable').addEventListener('click', saveAitableSettings);
  277. document.getElementById('btnTestImage').addEventListener('click', testImageApi);
  278. document.getElementById('btnTestChat').addEventListener('click', testChatModel);
  279. document.getElementById('btnTestAitable').addEventListener('click', () => aitableAction('test'));
  280. const bindHelpPopover = (btnId, popId) => {
  281. const btn = document.getElementById(btnId);
  282. const pop = document.getElementById(popId);
  283. btn.addEventListener('click', (event) => {
  284. event.stopPropagation();
  285. pop.hidden = !pop.hidden;
  286. btn.classList.toggle('active', !pop.hidden);
  287. btn.setAttribute('aria-expanded', String(!pop.hidden));
  288. });
  289. document.addEventListener('click', (event) => {
  290. if (!pop.hidden && !event.target.closest('.conn-help-wrap')) {
  291. pop.hidden = true;
  292. btn.classList.remove('active');
  293. btn.setAttribute('aria-expanded', 'false');
  294. }
  295. });
  296. };
  297. bindHelpPopover('btnSecurityHelp', 'securityHelpPopover');
  298. bindHelpPopover('btnAitableHelp', 'aitableHelpPopover');
  299. document.getElementById('btnClearLog').addEventListener('click', () => {
  300. clearLogs().catch(() => {});
  301. });
  302. }
  303. if (!settingsLoaded) {
  304. try {
  305. await loadSettings();
  306. settingsLoaded = true;
  307. } catch (error) {
  308. setPanelStatus('apiSaveStatus', error.message, false);
  309. }
  310. }
  311. }
  312. function escapeLogText(value) {
  313. const div = document.createElement('div');
  314. div.textContent = String(value ?? '');
  315. return div.innerHTML;
  316. }
  317. function renderLogs(logs) {
  318. const panel = document.getElementById('logPanel');
  319. if (!panel) return;
  320. const sourceMap = { task: '任务', system: '系统' };
  321. const levelMap = { info: 'INFO', success: 'OK', warn: 'WARN', error: 'FAIL' };
  322. panel.innerHTML = logs.map((log) => {
  323. const level = levelMap[log.level] || 'INFO';
  324. const levelClass = log.level === 'success' ? 'ok' : log.level || 'info';
  325. return `
  326. <div class="log-line">
  327. <span class="log-ts">${escapeLogText(log.time)}</span>
  328. <span class="log-source ${log.source === 'system' ? 'system' : 'task'}">${sourceMap[log.source] || log.source}</span>
  329. <span class="log-lvl ${levelClass}">${level}</span>
  330. <span class="log-msg">${escapeLogText(log.message)}</span>
  331. </div>`;
  332. }).join('');
  333. }
  334. async function refreshLogs() {
  335. const response = await fetch('/api/logs');
  336. if (!response.ok) return;
  337. const data = await response.json();
  338. const logs = (data.logs || []).sort((a, b) => (b.seq || 0) - (a.seq || 0));
  339. const signature = logs.map((log) => `${log.seq}:${log.level}:${log.message}`).join('|');
  340. if (signature === lastLogSignature) return;
  341. lastLogSignature = signature;
  342. renderLogs(logs);
  343. }
  344. function startLogPolling() {
  345. if (logTimer) return;
  346. refreshLogs();
  347. logTimer = setInterval(refreshLogs, 1500);
  348. }
  349. function stopLogPolling() {
  350. if (logTimer) { clearInterval(logTimer); logTimer = null; }
  351. }
  352. async function clearLogs() {
  353. await fetch('/api/logs/clear', { method: 'POST' });
  354. lastLogSignature = '';
  355. await refreshLogs();
  356. }
  357. function switchView(name) {
  358. for (const [key, el] of Object.entries(views)) {
  359. el.classList.toggle('active', key === name);
  360. }
  361. navItems.forEach((n) => n.classList.toggle('active', n.dataset.view === name));
  362. if (name === 'crawl' && window.shanCrawlInit) window.shanCrawlInit();
  363. if (window.shanCrawlSetActive) window.shanCrawlSetActive(name === 'crawl'); // 离开采集页暂停状态轮询
  364. if (name === 'gen' && window.shanGenInit) window.shanGenInit();
  365. if (name === 'tasks' && window.shanTasksInit) window.shanTasksInit();
  366. if (window.shanTasksSetActive) window.shanTasksSetActive(name === 'tasks'); // 离开任务页清掉轮询
  367. if (name === 'history' && window.shanHistoryInit) window.shanHistoryInit();
  368. if (name === 'reports' && window.shanReportsInit) window.shanReportsInit();
  369. if (name === 'lake' && window.shanLakeInit) window.shanLakeInit(); // 明细 tab 激活时再懒加载报表
  370. if (name === 'analysis' && window.shanAnalysisActivate) window.shanAnalysisActivate();
  371. if (name === 'ask' && window.shanAskInit) window.shanAskInit();
  372. if (name === 'settings') initSettings();
  373. // 系统日志:仅日志页激活时轮询
  374. if (name === 'logs') startLogPolling();
  375. else stopLogPolling();
  376. }
  377. navItems.forEach((n) => n.addEventListener('click', () => switchView(n.dataset.view)));
  378. const requestedView = new URLSearchParams(location.search).get('view');
  379. switchView(views[requestedView] ? requestedView : 'crawl');
  380. const splash = document.getElementById('splash');
  381. if (splash) {
  382. setTimeout(() => splash.classList.add('hide'), 1800);
  383. setTimeout(() => splash.remove(), 2400);
  384. }
  385. })();