promptgraph-mcp 1.5.19 → 1.5.21

package/index.js

@@ -1,15 +1,7 @@
 #!/usr/bin/env node
-import { Server } from '@modelcontextprotocol/sdk/server/index.js';
-import { StdioServerTransport } from '@modelcontextprotocol/sdk/server/stdio.js';
-import { CallToolRequestSchema, ListToolsRequestSchema } from '@modelcontextprotocol/sdk/types.js';
-import { search, getContext, getCallers, getCallees, getImpact, listAll } from './search.js';
-import { indexAll } from './indexer.js';
-import { startWatcher } from './watcher.js';
-import { promptConfig } from './config.js';
-import { importFromGitHub } from './github-import.js';
-import { detectPlatforms, PLATFORMS } from './platform.js';
-import { browseMarketplace, installSkill, publishSkill, getTopRated, recordUse, recordSuccess, recordFail, browseBundles, installBundle } from './marketplace.js';
-
+// Only lightweight imports at top. Heavy modules (fastembed/ONNX, vectra,
+// better-sqlite3) are dynamically imported inside the command that needs them,
+// so fast CLI commands (help, marketplace) start instantly.
 import { colors, banner, success, error, info, section, table } from './cli.js';
 import boxen from 'boxen';
 import chalk from 'chalk';
@@ -222,11 +214,13 @@ if (args[0] === 'validate') {
 }
 
 if (args[0] === 'import') {
+  const { importFromGitHub } = await import('./github-import.js');
   await importFromGitHub(args[1]);
   process.exit(0);
 }
 
 if (args[0] === 'setup') {
+  const { detectPlatforms, PLATFORMS } = await import('./platform.js');
   const platformId = args[1];
   if (!platformId) {
     section('Detected platforms');
@@ -243,6 +237,8 @@ if (args[0] === 'setup') {
 }
 
 if (args[0] === 'init') {
+  const { promptConfig } = await import('./config.js');
+  const { indexAll } = await import('./indexer.js');
   const config = await promptConfig();
   await indexAll();
   console.log();
@@ -257,10 +253,19 @@ if (args[0] === 'init') {
 }
 
 if (args[0] === 'reindex') {
+  const { indexAll } = await import('./indexer.js');
   await indexAll();
   process.exit(0);
 }
 
+// ── MCP server mode (no CLI command) ──
+const { Server } = await import('@modelcontextprotocol/sdk/server/index.js');
+const { StdioServerTransport } = await import('@modelcontextprotocol/sdk/server/stdio.js');
+const { CallToolRequestSchema, ListToolsRequestSchema } = await import('@modelcontextprotocol/sdk/types.js');
+const { search, getContext, getCallers, getCallees, getImpact, listAll } = await import('./search.js');
+const { startWatcher } = await import('./watcher.js');
+const { browseMarketplace, installSkill, publishSkill, getTopRated, recordUse, recordSuccess, recordFail, browseBundles, installBundle } = await import('./marketplace.js');
+
 const server = new Server(
   { name: 'promptgraph', version: '1.0.0' },
   { capabilities: { tools: {} } }

package/marketplace.js

@@ -2,6 +2,7 @@ import fs from 'fs';
 import path from 'path';
 import os from 'os';
 import https from 'https';
+import { createHash } from 'crypto';
 import { spawnSync } from 'child_process';
 import { getDb } from './db.js';
 import { validateSkill } from './validator.js';
@@ -9,10 +10,16 @@ import { validateSkill } from './validator.js';
 const REGISTRY_URL = 'https://raw.githubusercontent.com/NeiP4n/promptgraph-registry/main/registry.json';
 const SKILLS_DIR = path.join(os.homedir(), '.claude', 'skills-store', 'marketplace');
 
-// Robust fetch: try undici fetch, fall back to node:https (works where undici fails on Windows)
+// Deterministic short code from an id. Same id always yields the same code,
+// so codes auto-generate — no need to assign them by hand.
+export function codeFor(id) {
+  return 'pg-' + createHash('md5').update(String(id)).digest('hex').slice(0, 6);
+}
+
+// node:https GET — reliable and fast on Windows (undici fetch can hang ~10s there).
 function httpGet(url) {
   return new Promise((resolve, reject) => {
-    https.get(url, { headers: { 'User-Agent': 'promptgraph-mcp' } }, (res) => {
+    const req = https.get(url, { headers: { 'User-Agent': 'promptgraph-mcp' }, timeout: 8000, family: 4 }, (res) => {
       if (res.statusCode >= 300 && res.statusCode < 400 && res.headers.location) {
         return httpGet(res.headers.location).then(resolve, reject);
       }
@@ -24,26 +31,58 @@ function httpGet(url) {
       res.setEncoding('utf8');
       res.on('data', c => data += c);
       res.on('end', () => resolve(data));
-    }).on('error', reject);
+    });
+    req.on('timeout', () => { req.destroy(new Error('request timed out')); });
+    req.on('error', reject);
   });
 }
 
-async function fetchText(url) {
+// Primary path is httpGet (fast/reliable on Windows); undici fetch only as fallback.
+async function rawFetch(url) {
   try {
+    return await httpGet(url);
+  } catch {
     const res = await fetch(url, { headers: { 'User-Agent': 'promptgraph-mcp' } });
     if (!res.ok) throw new Error(`HTTP ${res.status}`);
     return await res.text();
-  } catch {
-    return await httpGet(url);
   }
 }
 
+// Disk cache for the registry (network to GitHub raw can be slow on some networks).
+const CACHE_DIR = path.join(os.homedir(), '.claude', '.promptgraph');
+const CACHE_TTL = 5 * 60 * 1000; // 5 minutes
+
+async function fetchText(url) {
+  const cacheFile = path.join(CACHE_DIR, 'registry-cache.json');
+  const isRegistry = url === REGISTRY_URL;
+
+  if (isRegistry && fs.existsSync(cacheFile)) {
+    try {
+      const stat = fs.statSync(cacheFile);
+      if (Date.now() - stat.mtimeMs < CACHE_TTL) {
+        return fs.readFileSync(cacheFile, 'utf8');
+      }
+    } catch {}
+  }
+
+  const text = await rawFetch(url);
+
+  if (isRegistry) {
+    try {
+      fs.mkdirSync(CACHE_DIR, { recursive: true });
+      fs.writeFileSync(cacheFile, text);
+    } catch {}
+  }
+  return text;
+}
+
 export async function browseMarketplace(topK = 20) {
   try {
     const text = await fetchText(REGISTRY_URL);
     const registry = JSON.parse(text);
     if (!Array.isArray(registry.skills)) return { error: 'Invalid registry format' };
     return registry.skills
+      .map(s => ({ ...s, code: s.code || codeFor(s.id) })) // auto-fill code
       .sort((a, b) => (b.stars || 0) - (a.stars || 0))
       .slice(0, topK);
   } catch (e) {
@@ -56,15 +95,16 @@ export async function installSkill(query) {
     const text = await fetchText(REGISTRY_URL);
     const registry = JSON.parse(text);
     const q = String(query).trim().toLowerCase();
-    // match by code, id, or name (case-insensitive)
+    // match by code (stored OR auto-generated), id, or name
     const skill = registry.skills?.find(s =>
-      s.code?.toLowerCase() === q ||
+      (s.code || codeFor(s.id)).toLowerCase() === q ||
       s.id?.toLowerCase() === q ||
       s.name?.toLowerCase() === q
     );
     if (!skill) {
-      // maybe it's a bundle code/id — hint the user
-      const bundle = (registry.bundles || []).find(b => b.code?.toLowerCase() === q || b.id?.toLowerCase() === q);
+      const bundle = (registry.bundles || []).find(b =>
+        (b.code || codeFor(b.id)).toLowerCase() === q || b.id?.toLowerCase() === q
+      );
       if (bundle) return { error: `"${query}" is a bundle. Use pg_bundle_install("${bundle.id}") instead.` };
       return { error: `No skill matching "${query}" (try a code like pg-xxxxxx, an id, or a name)` };
     }
@@ -89,6 +129,7 @@ export async function browseBundles(topK = 20) {
     const registry = JSON.parse(text);
     const bundles = registry.bundles || [];
     return bundles
+      .map(b => ({ ...b, code: b.code || codeFor(b.id) }))
       .sort((a, b) => (b.stars || 0) - (a.stars || 0))
       .slice(0, topK);
   } catch (e) {
@@ -102,7 +143,7 @@ export async function installBundle(bundleId) {
     const registry = JSON.parse(text);
     const q = String(bundleId).trim().toLowerCase();
     const bundle = (registry.bundles || []).find(b =>
-      b.code?.toLowerCase() === q || b.id?.toLowerCase() === q || b.name?.toLowerCase() === q
+      (b.code || codeFor(b.id)).toLowerCase() === q || b.id?.toLowerCase() === q || b.name?.toLowerCase() === q
     );
     if (!bundle) return { error: `No bundle matching "${bundleId}"` };
 

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "promptgraph-mcp",
-  "version": "1.5.19",
+  "version": "1.5.21",
   "main": "index.js",
   "type": "module",
   "bin": {