qa-memorize-mcp 0.1.0

package/bin/run.js

@@ -0,0 +1,53 @@
+#!/usr/bin/env node
+'use strict';
+
+const path = require('path');
+const fs = require('fs');
+const { spawnSync } = require('child_process');
+const zlib = require('zlib');
+
+const PLATFORMS = {
+  'win32-x64':    { pkg: 'qa-memorize-mcp-win-x64',    bin: 'memorize_mcp.exe' },
+  'linux-x64':    { pkg: 'qa-memorize-mcp-linux-x64',   bin: 'memorize_mcp' },
+  'darwin-x64':   { pkg: 'qa-memorize-mcp-darwin-x64',  bin: 'memorize_mcp' },
+  'darwin-arm64': { pkg: 'qa-memorize-mcp-darwin-arm64', bin: 'memorize_mcp' },
+};
+
+const key = `${process.platform}-${process.arch}`;
+const entry = PLATFORMS[key];
+if (!entry) {
+  console.error('Unsupported platform: ' + key);
+  process.exit(1);
+}
+
+let pkgDir;
+try {
+  pkgDir = path.dirname(require.resolve(entry.pkg + '/package.json'));
+} catch (e) {
+  console.error(`Platform package ${entry.pkg} not found: ${e.message}`);
+  process.exit(1);
+}
+
+const binDir = path.join(pkgDir, 'bin');
+const binPath = path.join(binDir, entry.bin);
+const modelDir = path.join(binDir, 'embedding_model');
+
+for (const name of ['model_ort.onnx', 'tokenizer.json']) {
+  const dest = path.join(modelDir, name);
+  const src = dest + '.gz';
+  if (!fs.existsSync(dest) && fs.existsSync(src)) {
+    fs.writeFileSync(dest, zlib.gunzipSync(fs.readFileSync(src)));
+  }
+}
+
+const args = process.argv.slice(2);
+const finalArgs = args.includes('--model-dir')
+  ? args
+  : ['--model-dir', modelDir, ...args];
+
+const result = spawnSync(binPath, finalArgs, {
+  stdio: 'inherit',
+  env: { ...process.env, ORT_DYLIB_PATH: binDir },
+});
+
+process.exit(result.status ?? 1);

package/package.json

@@ -0,0 +1,23 @@
+{
+  "name": "qa-memorize-mcp",
+  "version": "0.1.0",
+  "description": "RAG-based memory MCP server with local ONNX embeddings",
+  "bin": {
+    "qa-memorize-mcp": "bin/run.js"
+  },
+  "scripts": {
+    "postinstall": "node scripts/postinstall.mjs"
+  },
+  "optionalDependencies": {
+    "qa-memorize-mcp-win-x64": "0.1.0",
+    "qa-memorize-mcp-linux-x64": "0.1.0",
+    "qa-memorize-mcp-darwin-x64": "0.1.0",
+    "qa-memorize-mcp-darwin-arm64": "0.1.0"
+  },
+  "engines": {
+    "node": ">=18"
+  },
+  "keywords": ["mcp", "memorize", "rag", "embedding", "onnx"],
+  "license": "MIT",
+  "author": "whereslow"
+}

package/scripts/postinstall.mjs

@@ -0,0 +1,36 @@
+import { createRequire } from 'module';
+import { readFileSync, writeFileSync, unlinkSync, existsSync } from 'fs';
+import { gunzipSync } from 'zlib';
+import { join, dirname } from 'path';
+
+const require = createRequire(import.meta.url);
+
+const platforms = [
+  'qa-memorize-mcp-win-x64',
+  'qa-memorize-mcp-linux-x64',
+  'qa-memorize-mcp-darwin-x64',
+  'qa-memorize-mcp-darwin-arm64',
+];
+
+let pkgDir = null;
+for (const pkg of platforms) {
+  try {
+    const pkgJson = require.resolve(`${pkg}/package.json`);
+    pkgDir = dirname(pkgJson);
+    break;
+  } catch {}
+}
+
+if (!pkgDir) process.exit(0);
+
+const modelDir = join(pkgDir, 'bin', 'embedding_model');
+
+for (const file of ['model_ort.onnx.gz', 'tokenizer.json.gz']) {
+  const gz = join(modelDir, file);
+  const out = gz.slice(0, -3);
+  if (!existsSync(gz)) continue;
+  if (existsSync(out)) { unlinkSync(gz); continue; }
+  process.stderr.write(`postinstall: decompressing ${file}\n`);
+  writeFileSync(out, gunzipSync(readFileSync(gz)));
+  unlinkSync(gz);
+}