dsh-memory-engine 0.6.1 → 0.7.0-beta.2b

package/lib/serve.js

@@ -1,150 +0,0 @@
-import { createServer } from 'node:http';
-import fs from 'node:fs';
-import path from 'node:path';
-import { generateMap } from './map.js';
-import { readConfig } from './config.js';
-
-export const DEFAULT_UI_PORT = 18886;
-const pending = '<!doctype html><html lang="zh-CN"><meta charset="utf-8"><meta http-equiv="refresh" content="1"><title>记忆图谱生成中</title><body style="background:#03060c;color:#d8e9f8;font:16px system-ui;padding:40px">正在生成记忆图谱，请稍候……</body></html>';
-const escapeBridge = '<script>window.addEventListener("keydown",function(e){if(e.key==="Escape")parent.postMessage({type:"dsh-memory-engine-close"},"*")});</script>';
-
-// 空库时的上手引导页。
-// 为什么要有这一页：新装用户点开图谱看到的是一片黑，只会以为插件坏了。
-// 这个页面在他最想用的那一刻告诉他「为什么是空的、两件事怎么做」。
-const ONBOARD_PROMPT = [
-  '聊天时遇到这些，用 memory_save 记下来：',
-  '  · 对方的称呼、偏好、忌讳、习惯（kind="印象"，who 填对方名字）',
-  '  · 答应过的事、约定、承诺（kind="对话"）',
-  '  · 情绪特别重的瞬间（kind="事件"，heat=1）',
-  '  · 新学到的知识、名词、梗的解释（kind="知识"）',
-  '  · 每天结束时，用一两句写当天小结（kind="日记"）',
-  '',
-  '不用每条都记。一天三五条就够，宁可少存也别灌水。',
-  '存之前先 memory_recall 查一下，避免重复。',
-].join('\n');
-
-const onboard = `<!doctype html><html lang="zh-CN"><head><meta charset="utf-8">
-<title>还没有记忆</title>
-<style>
-  :root{color-scheme:dark}
-  *{box-sizing:border-box}
-  body{margin:0;min-height:100vh;background:radial-gradient(ellipse at 50% 0%,#0b1a2e 0%,#03060c 60%);
-    color:#dbeaf8;font:15px/1.75 "Microsoft YaHei",system-ui,sans-serif;display:flex;align-items:center;justify-content:center;padding:32px}
-  .card{width:min(760px,100%)}
-  h1{font-size:26px;margin:0 0 6px;font-weight:600;letter-spacing:.5px}
-  .sub{color:#7f9dbb;margin:0 0 26px;font-size:14px}
-  .step{border-left:3px solid #1f4d73;padding:2px 0 2px 16px;margin:0 0 24px}
-  .step h2{font-size:16px;margin:0 0 8px;color:#9fd0f5;font-weight:600}
-  .step p{margin:0 0 10px;color:#b6cde2;font-size:14px}
-  pre{background:#07121f;border:1px solid #17324b;border-radius:8px;padding:14px;overflow:auto;
-    font:13px/1.7 ui-monospace,Consolas,monospace;color:#b9d8f2;margin:0 0 10px;white-space:pre-wrap}
-  button{background:#123a5c;color:#cfe9ff;border:1px solid #2a6698;border-radius:7px;
-    padding:8px 16px;font:14px "Microsoft YaHei",system-ui;cursor:pointer}
-  button:hover{background:#17496f}
-  button:active{transform:translateY(1px)}
-  code{background:#0d1e30;border-radius:4px;padding:1px 5px;color:#8fd0ff;font-size:13px}
-  .foot{color:#5f7a93;font-size:13px;margin-top:8px}
-</style></head><body><div class="card">
-  <h1>记忆库还是空的</h1>
-  <p class="sub">插件不会自己读你的聊天记录。记忆要靠下面两件事之一进来。</p>
-
-  <div class="step">
-    <h2>① 让它边聊边记（推荐）</h2>
-    <p>把这段加进 agent 的提示词或角色卡里：</p>
-    <pre id="prompt">${ONBOARD_PROMPT.replace(/&/g, '&amp;').replace(/</g, '&lt;').replace(/>/g, '&gt;')}</pre>
-    <button id="copy">复制这段提示词</button>
-  </div>
-
-  <div class="step">
-    <h2>② 或者把旧记忆导进来</h2>
-    <p>让 agent 调用 <code>memory_import</code>，支持三种格式：</p>
-    <pre>memory_import(source="json", path="D:\\\\bot\\\\memories.json")
-memory_import(source="markdown", path="D:\\\\bot\\\\notes")
-memory_import(source="text", path="D:\\\\bot\\\\memories.txt")</pre>
-    <p class="foot">会先预览「能新增多少条、多少条重复被跳过」，确认后再写。</p>
-  </div>
-
-  <div class="step">
-    <h2>③ 然后就有图了</h2>
-    <p>有了几条记忆之后，回到这个按钮，就能看到神经网络。</p>
-    <p class="foot">记忆只存在本机 <code>~/.dsh/memory-engine/</code>，不上传、不联网。</p>
-  </div>
-</div>
-<script>
-  var btn = document.getElementById('copy');
-  btn.addEventListener('click', function () {
-    var text = document.getElementById('prompt').textContent;
-    var done = function () { btn.textContent = '已复制 ✓'; setTimeout(function () { btn.textContent = '复制这段提示词'; }, 1800); };
-    if (navigator.clipboard && navigator.clipboard.writeText) {
-      navigator.clipboard.writeText(text).then(done, fallback);
-    } else { fallback(); }
-    function fallback() {
-      var area = document.createElement('textarea');
-      area.value = text; document.body.appendChild(area); area.select();
-      try { document.execCommand('copy'); done(); } catch (e) { btn.textContent = '请手动选中复制'; }
-      document.body.removeChild(area);
-    }
-  });
-</script>
-</body></html>`;
-
-export async function startMapServer({ store, port, warn = () => {} }) {
-  const directory = path.dirname(store.file);
-  const config = readConfig(path.dirname(directory));
-  port ??= config.uiPort ?? DEFAULT_UI_PORT;
-  if (!Number.isInteger(port) || port < 0 || port > 65535) throw new Error('uiPort 必须是有效端口号');
-  let state;
-  const server = createServer((request, response) => {
-    const reply = (status, body = '') => {
-      response.writeHead(status, { 'Content-Type': 'text/html; charset=utf-8', 'Cache-Control': 'no-store',
-        'X-Content-Type-Options': 'nosniff', 'Referrer-Policy': 'no-referrer' });
-      response.end(request.method === 'HEAD' ? '' : body);
-    };
-    try {
-      if (request.headers.host !== `127.0.0.1:${server.address()?.port}`) return reply(403);
-      if (!['GET', 'HEAD'].includes(request.method)) return reply(405);
-      const url = new URL(request.url, 'http://127.0.0.1');
-      const rootPage = ['/', '/index.html', '/starmap.html'].includes(url.pathname);
-      if (!rootPage && !/^\/pages\/[1-9]\d*\.html$/u.test(url.pathname)) return reply(404);
-      if (rootPage) {
-        // 空库直接返回上手引导页：不要让人对着黑屏猜「是不是坏了」
-        let rows = [];
-        try { rows = store.read(); } catch (error) { warn(`记忆库读取失败：${error.message}`); }
-        if (!Array.isArray(rows) || rows.length === 0) return reply(200, onboard + escapeBridge);
-        let signature = 'empty';
-        try { const info = fs.statSync(store.file); signature = `${info.mtimeMs}:${info.size}`; }
-        catch (error) { if (error.code !== 'ENOENT') throw error; }
-        // A new opening gets a new v token. Polling its loading page reuses
-        // that generation instead of continually starting another render.
-        const key = `${url.searchParams.get('v') ?? ''}:${signature}`;
-        if (!state || state.key !== key) {
-          const generation = { key, status: 'pending' };
-          state = generation;
-          setImmediate(() => {
-            if (!server.listening) return;
-            try { generateMap(store); generation.status = 'ready'; }
-            catch (error) { generation.status = 'error'; warn(`图谱生成失败：${error.message}`); }
-          });
-        }
-        if (state.status === 'pending') return reply(200, pending + escapeBridge);
-        if (state.status === 'error') return reply(500, '<meta charset="utf-8">图谱生成失败，请检查本地插件日志。' + escapeBridge);
-      }
-      const file = rootPage ? path.join(directory, 'starmap.html') : path.join(directory, url.pathname.slice(1));
-      let html;
-      try { html = fs.readFileSync(file, 'utf8'); }
-      catch (error) { if (error.code === 'ENOENT') return reply(404); throw error; }
-      // This transport-only key bridge leaves the copied renderer untouched.
-      reply(200, html.replace('</body>', escapeBridge + '</body>'));
-    } catch (error) {
-      warn(`图谱页面读取失败：${error.message}`);
-      reply(500, '<meta charset="utf-8">图谱页面暂不可用，请检查本地插件日志。');
-    }
-  });
-  await new Promise((resolve, reject) => {
-    server.once('error', reject);
-    server.listen(port, '127.0.0.1', () => { server.removeListener('error', reject); resolve(); });
-  });
-  server.on('error', error => warn(`图谱服务异常：${error.message}`));
-  server.unref();
-  return server;
-}