promptgraph-mcp 2.8.8 → 2.9.0

package/bundle-counts.js

@@ -75,6 +75,15 @@ async function countSubdirMdFiles(ownerRepo) {
   }).length;
 }
 
+// Write a known real count (e.g. after install) — skips API, sets TTL to 24h
+export function setCachedCount(repoUrl, count) {
+  try {
+    const cache = loadCache();
+    cache[repoUrl] = { count, ts: Date.now() };
+    saveCache(cache);
+  } catch {}
+}
+
 // Read cached count for a bundle (returns null if stale or missing)
 export function getCachedCount(repoUrl) {
   const cache = loadCache();

package/commands/marketplace.js

@@ -39,9 +39,22 @@ export default async function handler(args, bin) {
 
   if (skills?.error) { error(skills.error); process.exit(1); }
 
-  const { getCachedCount, refreshCountsInBackground } = await import('../bundle-counts.js');
+  const { getCachedCount, setCachedCount, refreshCountsInBackground } = await import('../bundle-counts.js');
+  const { SKILLS_STORE_DIR } = await import('../config.js');
+  const { globSync } = await import('glob');
+  const githubDir = path.join(SKILLS_STORE_DIR, 'github');
+
+  // For each bundle: if installed on disk — use real file count; otherwise use cache
   const bundlesWithCounts = (Array.isArray(bundles) ? bundles : []).map(b => {
     if (!b.repo_url) return b;
+    const owner = b.repo_url.split('/')[0];
+    const repo  = b.repo_url.split('/')[1];
+    const clonedDir = path.join(githubDir, `${owner}-${repo}`);
+    if (fs.existsSync(clonedDir) && fs.readdirSync(clonedDir).length > 0) {
+      const realCount = globSync(`${clonedDir}/**/*.md`).length;
+      setCachedCount(b.repo_url, realCount);
+      return { ...b, skillCount: realCount };
+    }
     const cached = getCachedCount(b.repo_url);
     return cached !== null ? { ...b, skillCount: cached } : b;
   });
@@ -51,18 +64,13 @@ export default async function handler(args, bin) {
   try {
     const cfg = _lcMkt();
     const db = _getDbMkt();
-    const { SKILLS_STORE_DIR } = await import('../config.js');
-    const githubDir = path.join(SKILLS_STORE_DIR, 'github');
 
     for (const b of (Array.isArray(bundles) ? bundles : [])) {
       if (b.repo_url) {
         const owner = b.repo_url.split('/')[0];
         const repo  = b.repo_url.split('/')[1];
-        const clonedName = `${owner}-${repo}`;
-        const clonedDir  = path.join(githubDir, clonedName);
-        const dirExists  = fs.existsSync(clonedDir) &&
-          fs.readdirSync(clonedDir).length > 0;
-        if (dirExists) installedSet.add(b.id);
+        const clonedDir = path.join(githubDir, `${owner}-${repo}`);
+        if (fs.existsSync(clonedDir) && fs.readdirSync(clonedDir).length > 0) installedSet.add(b.id);
       } else if (Array.isArray(b.skills)) {
         const allOnDisk = b.skills.every(sid => {
           const row = db.prepare('SELECT path FROM skills WHERE id = ?').get(sid);

package/github-import.js

@@ -735,20 +735,20 @@ export async function importFromGitHubLight(repoUrl) {
   process.stderr.write('\n');
 
   const realCount = globSync(`${destBase}/**/*.md`).length;
-  const cacheKey = url.replace(/\.git$/, '');
-  const cachePath = path.join(PROMPTGRAPH_DIR, 'skill-counts.json');
-  try {
-    fs.mkdirSync(path.dirname(cachePath), { recursive: true });
-    const cache = JSON.parse(fs.readFileSync(cachePath, 'utf8') || '{}');
-    cache[cacheKey] = { count: realCount, ts: Date.now() };
-    fs.writeFileSync(cachePath, JSON.stringify(cache, null, 2));
-  } catch {}
 
   if (realCount < 1) {
     fs.rmSync(destBase, { recursive: true, force: true });
     throw new Error('No valid skills in repo — all files were filtered out');
   }
 
+  // Update both caches with the real post-filter count
+  try {
+    const { setCachedCount } = await import('./bundle-counts.js');
+    const cacheKey = url.replace(/\.git$/, '').replace(/^https?:\/\/github\.com\//, '');
+    setCachedCount(cacheKey, realCount);
+    setCachedCount(url.replace(/\.git$/, ''), realCount);
+  } catch {}
+
   const { dir: localDir, label: localLabel } = detectSkillsDirLocal(destBase);
   const skillsDir = subdir ? path.join(destBase, subdir) : localDir;
   const label = subdir || localLabel;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "promptgraph-mcp",
-  "version": "2.8.8",
+  "version": "2.9.0",
   "files": [
     "*.js",
     "commands/",