design-learn-server 0.1.2 → 0.1.4

package/README.md

@@ -2,10 +2,6 @@
 
 A minimal single-process server entry that registers REST, WebSocket, and MCP (SSE) routes.
 
-## Node version
-
-Requires Node 18 or 20 (LTS). Node 22+ is not supported because `better-sqlite3` uses native bindings.
-
 ## Start
 
 ```bash

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "design-learn-server",
-  "version": "0.1.2",
+  "version": "0.1.4",
   "private": false,
   "description": "Design-Learn server skeleton for REST/WS/MCP routes",
   "main": "src/server.js",
@@ -9,9 +9,6 @@
     "design-learn-mcp": "src/stdio.js",
     "design-learn-stdio": "src/stdio.js"
   },
-  "engines": {
-    "node": ">=18 <23"
-  },
   "files": [
     "src",
     "README.md"
@@ -21,13 +18,12 @@
   },
   "scripts": {
     "start": "node src/server.js",
-    "dev": "node src/server.js",
-    "preinstall": "node src/check-node.js"
+    "dev": "node src/server.js"
   },
   "dependencies": {
     "@modelcontextprotocol/sdk": "^1.0.0",
-    "better-sqlite3": "^11.10.0",
     "playwright": "^1.57.0",
+    "sql.js": "^1.13.0",
     "zod": "^3.24.1"
   }
 }

package/src/mcp/index.js

@@ -445,8 +445,8 @@ function createMcpServer({ name, version, storage, uipro }) {
   return server;
 }
 
-function createMcpHandler(options = {}) {
-  const storage = options.storage || createStorage({ dataDir: options.dataDir });
+async function createMcpHandler(options = {}) {
+  const storage = options.storage || (await createStorage({ dataDir: options.dataDir }));
   const ownsStorage = !options.storage;
   const serverName = options.serverName || 'design-learn';
   const serverVersion = options.serverVersion || '0.1.0';

package/src/server.js

@@ -22,69 +22,64 @@ const { readJson, writeJson } = require('./storage/fileStore');
 const defaultDataDir = path.join(os.homedir(), '.design-learn', 'data');
 const dataDir = process.env.DESIGN_LEARN_DATA_DIR || process.env.DATA_DIR || defaultDataDir;
 
-const storage = createStorage({ dataDir });
-const uipro = createUipro({ dataDir });
-const extractionPipeline = createExtractionPipeline({ storage });
-const previewPipeline = createPreviewPipeline({ storage });
+let storage;
+let uipro;
+let extractionPipeline;
+let previewPipeline;
+let mcpHandler;
 
 // import job -> designId，用于将 pipeline 状态回写到 design metadata（便于 UI 与 MCP 立刻可见）
 const importJobDesignMap = new Map();
 
-extractionPipeline.onProgress((event) => {
-  const designId = importJobDesignMap.get(event.job.id);
-  if (!designId) {
-    return;
-  }
-
-  const now = new Date().toISOString();
-  void (async () => {
-    const design = await storage.getDesign(designId);
-    const meta = design?.metadata || {};
-    const aiRequested = !!meta.aiRequested;
-    const aiCompleted = meta.processingMessage === 'ai_completed' || meta.aiCompleted;
-
-    const metaPatch = {
-      processingStatus: event.event === 'failed' ? 'failed' : event.event === 'completed' ? 'completed' : 'processing',
-      processingJobId: event.job.id,
-      processingUpdatedAt: now,
-    };
+function registerPipelineHandlers() {
+  extractionPipeline.onProgress((event) => {
+    const designId = importJobDesignMap.get(event.job.id);
+    if (!designId) {
+      return;
+    }
 
-    if (event.event === 'failed') {
-      metaPatch.processingError = event.job.error?.message || 'unknown_error';
-      metaPatch.processingMessage = 'failed';
-      metaPatch.processingProgress = 100;
-    } else if (event.event === 'completed') {
-      metaPatch.processingError = null;
-      metaPatch.lastImportAt = now;
-      metaPatch.lastImportVersionId = event.job.result?.versionId || null;
-      if (aiRequested && !aiCompleted) {
-        metaPatch.processingStatus = 'analyzing';
-        metaPatch.processingMessage = 'ai_pending';
-        metaPatch.processingProgress = 80;
-      } else {
-        metaPatch.processingMessage = 'completed';
+    const now = new Date().toISOString();
+    void (async () => {
+      const design = await storage.getDesign(designId);
+      const meta = design?.metadata || {};
+      const aiRequested = !!meta.aiRequested;
+      const aiCompleted = meta.processingMessage === 'ai_completed' || meta.aiCompleted;
+
+      const metaPatch = {
+        processingStatus: event.event === 'failed' ? 'failed' : event.event === 'completed' ? 'completed' : 'processing',
+        processingJobId: event.job.id,
+        processingUpdatedAt: now,
+      };
+
+      if (event.event === 'failed') {
+        metaPatch.processingError = event.job.error?.message || 'unknown_error';
+        metaPatch.processingMessage = 'failed';
         metaPatch.processingProgress = 100;
+      } else if (event.event === 'completed') {
+        metaPatch.processingError = null;
+        metaPatch.lastImportAt = now;
+        metaPatch.lastImportVersionId = event.job.result?.versionId || null;
+        if (aiRequested && !aiCompleted) {
+          metaPatch.processingStatus = 'analyzing';
+          metaPatch.processingMessage = 'ai_pending';
+          metaPatch.processingProgress = 80;
+        } else {
+          metaPatch.processingMessage = 'completed';
+          metaPatch.processingProgress = 100;
+        }
+      } else {
+        metaPatch.processingMessage = event.job.message || 'processing';
+        metaPatch.processingProgress = event.job.progress || 0;
       }
-    } else {
-      metaPatch.processingMessage = event.job.message || 'processing';
-      metaPatch.processingProgress = event.job.progress || 0;
-    }
 
-    await storage.updateDesign(designId, { metadata: metaPatch });
-  })().catch(() => undefined);
+      await storage.updateDesign(designId, { metadata: metaPatch });
+    })().catch(() => undefined);
 
-  if (event.event === 'failed' || event.event === 'completed') {
-    importJobDesignMap.delete(event.job.id);
-  }
-});
-const mcpHandler = createMcpHandler({
-  storage,
-  dataDir: process.env.DESIGN_LEARN_DATA_DIR,
-  serverName: process.env.MCP_SERVER_NAME,
-  serverVersion: process.env.MCP_SERVER_VERSION,
-  authToken: process.env.MCP_AUTH_TOKEN,
-  uipro,
-});
+    if (event.event === 'failed' || event.event === 'completed') {
+      importJobDesignMap.delete(event.job.id);
+    }
+  });
+}
 
 const routes = [
   {
@@ -1367,17 +1362,40 @@ server.on('clientError', (err, socket) => {
   socket.end('HTTP/1.1 400 Bad Request\r\n\r\n');
 });
 
-server.listen(DEFAULT_PORT, () => {
-  console.log(`[design-learn-server] listening on http://localhost:${DEFAULT_PORT}`);
-  console.log(`[design-learn-server] data dir: ${storage.dataDir}`);
+async function start() {
+  storage = await createStorage({ dataDir });
+  uipro = createUipro({ dataDir });
+  extractionPipeline = createExtractionPipeline({ storage });
+  previewPipeline = createPreviewPipeline({ storage });
+  registerPipelineHandlers();
+  mcpHandler = await createMcpHandler({
+    storage,
+    dataDir: process.env.DESIGN_LEARN_DATA_DIR,
+    serverName: process.env.MCP_SERVER_NAME,
+    serverVersion: process.env.MCP_SERVER_VERSION,
+    authToken: process.env.MCP_AUTH_TOKEN,
+    uipro,
+  });
+
+  server.listen(DEFAULT_PORT, () => {
+    console.log(`[design-learn-server] listening on http://localhost:${DEFAULT_PORT}`);
+    console.log(`[design-learn-server] data dir: ${storage.dataDir}`);
+  });
+}
+
+start().catch((error) => {
+  console.error('[design-learn-server] startup failed', error);
+  process.exit(1);
 });
 
 function shutdown(signal) {
   console.log(`[design-learn-server] received ${signal}, shutting down`);
-  mcpHandler.close().catch((error) => console.error('[mcp] close error', error));
-  extractionPipeline.close();
-  previewPipeline.close();
-  storage.close();
+  if (mcpHandler) {
+    mcpHandler.close().catch((error) => console.error('[mcp] close error', error));
+  }
+  extractionPipeline?.close();
+  previewPipeline?.close();
+  storage?.close();
   server.close(() => process.exit(0));
 }
 

package/src/stdio.js

@@ -55,11 +55,12 @@ async function ensurePlaywright() {
 const os = require('os');
 const defaultDataDir = path.join(os.homedir(), '.design-learn', 'data');
 const dataDir = process.env.DESIGN_LEARN_DATA_DIR || process.env.DATA_DIR || defaultDataDir;
-const storage = createStorage({ dataDir });
+const storagePromise = createStorage({ dataDir });
 const uipro = createUipro({ dataDir });
 
-function matchDesigns(query) {
+async function matchDesigns(query) {
   const needle = query.toLowerCase();
+  const storage = await storagePromise;
   const designs = storage.listDesigns();
   return designs.filter((design) => {
     const tags = Array.isArray(design.metadata?.tags) ? design.metadata.tags.join(' ') : '';
@@ -91,6 +92,7 @@ server.tool(
   'List stored design resources',
   { limit: z.number().min(1).max(100).optional() },
   async ({ limit }) => {
+    const storage = await storagePromise;
     const designs = storage.listDesigns();
     const data = typeof limit === 'number' ? designs.slice(0, limit) : designs;
     return {
@@ -108,7 +110,7 @@ server.tool(
     limit: z.number().min(1).max(100).optional(),
   },
   async ({ query, limit }) => {
-    const matches = matchDesigns(query);
+    const matches = await matchDesigns(query);
     const data = typeof limit === 'number' ? matches.slice(0, limit) : matches;
     return {
       content: [{ type: 'text', text: JSON.stringify(data, null, 2) }],
@@ -135,14 +137,12 @@ server.tool(
   },
   async ({ query, sources, domain, stack, limit, designLimit }) => {
     const effectiveSources = Array.isArray(sources) && sources.length > 0 ? sources : ['designs', 'uipro'];
-    const designs =
-      effectiveSources.includes('designs')
-        ? (() => {
-            const matches = matchDesigns(query);
-            const max = typeof designLimit === 'number' ? designLimit : typeof limit === 'number' ? limit : undefined;
-            return typeof max === 'number' ? matches.slice(0, max) : matches;
-          })()
-        : [];
+    let designs = [];
+    if (effectiveSources.includes('designs')) {
+      const matches = await matchDesigns(query);
+      const max = typeof designLimit === 'number' ? designLimit : typeof limit === 'number' ? limit : undefined;
+      designs = typeof max === 'number' ? matches.slice(0, max) : matches;
+    }
 
     const resolvedDomain = domain === 'auto' ? undefined : domain;
     const maxUipro = typeof limit === 'number' ? limit : 10;
@@ -201,6 +201,7 @@ server.tool(
   'Get styleguide markdown by design ID (latest version).',
   { designId: z.string() },
   async ({ designId }) => {
+    const storage = await storagePromise;
     const versions = storage.listVersions(designId);
     if (!versions || versions.length === 0) {
       return {

package/src/storage/index.js

@@ -19,9 +19,9 @@ const {
 const { ensureDir, writeJson, readJson, writeText, readText, removePath } = require('./fileStore');
 const { openDatabase } = require('./sqliteStore');
 
-function createStorage(options = {}) {
+async function createStorage(options = {}) {
   const dataDir = resolveDataDir(options.dataDir);
-  const db = openDatabase(getDatabasePath(dataDir));
+  const db = await openDatabase(getDatabasePath(dataDir));
 
   return {
     dataDir,

package/src/storage/sqliteStore.js

@@ -1,6 +1,6 @@
 const fs = require('fs');
 const path = require('path');
-const Database = require('better-sqlite3');
+const initSqlJs = require('sql.js');
 
 const SCHEMA_VERSION = 2;
 
@@ -8,9 +8,118 @@ function ensureDir(dirPath) {
   fs.mkdirSync(dirPath, { recursive: true });
 }
 
-function openDatabase(dbPath) {
+function readDatabaseFile(dbPath) {
+  try {
+    return fs.readFileSync(dbPath);
+  } catch (error) {
+    if (error.code === 'ENOENT') {
+      return null;
+    }
+    throw error;
+  }
+}
+
+function wrapDatabase(rawDb, dbPath) {
+  let closed = false;
+
+  function persist() {
+    if (closed) {
+      return;
+    }
+    const data = rawDb.export();
+    const buffer = Buffer.from(data);
+    fs.writeFileSync(dbPath, buffer);
+  }
+
+  function prepare(sql) {
+    const stmt = rawDb.prepare(sql);
+    const normalizeParams = (params) => (params.length === 1 && Array.isArray(params[0]) ? params[0] : params);
+    const finalize = () => {
+      try {
+        stmt.free();
+      } catch {
+        return;
+      }
+    };
+
+    return {
+      run: (...params) => {
+        const bound = normalizeParams(params);
+        stmt.run(bound);
+        const changes = typeof rawDb.getRowsModified === 'function' ? rawDb.getRowsModified() : 0;
+        finalize();
+        persist();
+        return { changes };
+      },
+      get: (...params) => {
+        const bound = normalizeParams(params);
+        stmt.bind(bound);
+        if (!stmt.step()) {
+          finalize();
+          return undefined;
+        }
+        const row = stmt.getAsObject();
+        finalize();
+        return row;
+      },
+      all: (...params) => {
+        const bound = normalizeParams(params);
+        const rows = [];
+        stmt.bind(bound);
+        while (stmt.step()) {
+          rows.push(stmt.getAsObject());
+        }
+        finalize();
+        return rows;
+      },
+    };
+  }
+
+  function exec(sql) {
+    rawDb.exec(sql);
+    persist();
+  }
+
+  function pragma(statement, options = {}) {
+    const trimmed = statement.trim();
+    if (trimmed.includes('=')) {
+      rawDb.exec(`PRAGMA ${statement}`);
+      persist();
+      return undefined;
+    }
+    const result = rawDb.exec(`PRAGMA ${statement}`);
+    if (options.simple) {
+      return result?.[0]?.values?.[0]?.[0] ?? 0;
+    }
+    return result;
+  }
+
+  function close() {
+    if (closed) {
+      return;
+    }
+    persist();
+    rawDb.close();
+    closed = true;
+  }
+
+  return {
+    prepare,
+    exec,
+    pragma,
+    close,
+  };
+}
+
+async function openDatabase(dbPath) {
   ensureDir(path.dirname(dbPath));
-  const db = new Database(dbPath);
+  const wasmPath = require.resolve('sql.js/dist/sql-wasm.wasm');
+  const SQL = await initSqlJs({
+    locateFile: () => wasmPath,
+  });
+  const fileBuffer = readDatabaseFile(dbPath);
+  const rawDb = fileBuffer ? new SQL.Database(fileBuffer) : new SQL.Database();
+  const db = wrapDatabase(rawDb, dbPath);
   db.pragma('journal_mode = WAL');
   migrate(db);
   return db;

package/src/check-node.js

@@ -1,14 +0,0 @@
-"use strict";
-
-const major = Number(process.versions.node.split(".")[0]);
-const isSupported = major >= 18 && major < 23;
-
-if (!isSupported) {
-  console.error("=== Node 版本不支持 ===");
-  console.error(
-    `[design-learn] 仅支持 Node 18/20，当前版本 ${process.versions.node}。`
-  );
-  console.error("[design-learn] 请切换到 Node 18/20 后重试。");
-  process.exit(1);
-}
-