@aistoragedepot/mcp 0.1.0 → 0.2.0

package/README.md

@@ -16,6 +16,9 @@ Your stored content shows up where you actually work:
 
 ## Setup
 
+> **👉 Easiest path:** step-by-step instructions for each app (config-file locations and all)
+> are at **<https://www.aistoragedepot.com/connect>**. The manual version:
+
 1. In AIStorageDepot (a **PLUS** plan is required for API tokens), go to **Settings → API
    tokens** and create a token. Copy it.
 2. Add the server to your MCP client's config. Example (Claude Desktop /
@@ -44,6 +47,7 @@ Your stored content shows up where you actually work:
 | --- | --- | --- | --- |
 | `AISD_TOKEN` | yes | — | Token from Settings → API tokens. Has full access to your account — keep it secret. |
 | `AISD_BASE_URL` | no | `https://www.aistoragedepot.com` | Point at a self-hosted / dev instance if needed. |
+| `AISD_WORKSPACES` | no | your personal library | Which workspaces expose **prompts** (slash-commands). Comma-separated workspace names (e.g. `My Library,Engineering`), or `all`. Keeps the slash menu short; **search and resources still cover every workspace** regardless. |
 
 ## Notes
 

package/index.mjs

@@ -43,15 +43,13 @@ let cache = { at: 0, workspaces: [], items: [] };
 async function snapshot() {
   if (cache.items.length && Date.now() - cache.at < 30_000) return cache;
   const workspaces = await api("/api/workspaces");
-  const items = [];
-  for (const ws of workspaces) {
-    try {
-      const lib = await api(`/api/library?workspace=${encodeURIComponent(ws.id)}`);
-      for (const it of lib.items || []) items.push(it);
-    } catch {
-      // a workspace we can't read (or that errored) — skip it, keep the rest.
-    }
-  }
+  // Fetch every workspace's library in parallel — the sequential loop was the slow part.
+  const libs = await Promise.all(
+    workspaces.map((ws) =>
+      api(`/api/library?workspace=${encodeURIComponent(ws.id)}`).catch(() => ({ items: [] })),
+    ),
+  );
+  const items = libs.flatMap((lib) => lib.items || []);
   cache = { at: Date.now(), workspaces, items };
   return cache;
 }
@@ -76,6 +74,31 @@ function promptIndex(items) {
   return byName;
 }
 
+// Which workspaces' prompts appear as slash-commands. Default: your personal library only —
+// keeps the slash menu short + fast (clients bury/limit long prompt lists). Override with
+// AISD_WORKSPACES="Name1,Name2" (comma-separated workspace names) or AISD_WORKSPACES="all".
+const PROMPT_WS = (process.env.AISD_WORKSPACES || "")
+  .split(",")
+  .map((s) => s.trim().toLowerCase())
+  .filter(Boolean);
+
+function promptScopeIds(workspaces) {
+  const all = () => new Set(workspaces.map((w) => w.id));
+  if (PROMPT_WS.includes("all") || PROMPT_WS.includes("*")) return all();
+  if (PROMPT_WS.length) {
+    const named = workspaces.filter((w) => PROMPT_WS.includes((w.name || "").toLowerCase()));
+    if (named.length) return new Set(named.map((w) => w.id));
+  }
+  const personal = workspaces.filter((w) => w.type === "PERSONAL");
+  return personal.length ? new Set(personal.map((w) => w.id)) : all();
+}
+
+// promptIndex, but only over items in the in-scope workspaces.
+function scopedPromptIndex(cache) {
+  const ids = promptScopeIds(cache.workspaces);
+  return promptIndex(cache.items.filter((it) => ids.has(it.workspaceId)));
+}
+
 // MCP argument names should be simple identifiers; map them back to the original
 // "[placeholder]" text so we can substitute into the body.
 function argMap(placeholders) {
@@ -100,7 +123,7 @@ function fillBody(body, placeholders, args) {
 }
 
 const server = new Server(
-  { name: "aistoragedepot", version: "0.1.0" },
+  { name: "aistoragedepot", version: "0.2.0" },
   { capabilities: { resources: {}, prompts: {}, tools: {} } },
 );
 
@@ -128,9 +151,8 @@ server.setRequestHandler(ReadResourceRequestSchema, async (req) => {
 
 // ---- Prompts: prompt-format items, [fields] → arguments ----
 server.setRequestHandler(ListPromptsRequestSchema, async () => {
-  const { items } = await snapshot();
   const prompts = [];
-  for (const [name, it] of promptIndex(items)) {
+  for (const [name, it] of scopedPromptIndex(await snapshot())) {
     const am = argMap(it.placeholders);
     prompts.push({
       name,
@@ -146,8 +168,7 @@ server.setRequestHandler(ListPromptsRequestSchema, async () => {
 });
 
 server.setRequestHandler(GetPromptRequestSchema, async (req) => {
-  const { items } = await snapshot();
-  const it = promptIndex(items).get(req.params.name);
+  const it = scopedPromptIndex(await snapshot()).get(req.params.name);
   if (!it) throw new Error(`Unknown prompt: ${req.params.name}`);
   const text = fillBody(it.body, it.placeholders, req.params.arguments || {});
   return {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@aistoragedepot/mcp",
-  "version": "0.1.0",
+  "version": "0.2.0",
   "description": "Model Context Protocol server for your AIStorageDepot library — exposes your prompts, rules, docs, and skills to MCP-aware AI clients (Claude, Cursor, Cline, …).",
   "type": "module",
   "bin": {