@amplis/mcp-server 0.3.2 → 0.4.0

package/dist/index.js

@@ -431,22 +431,22 @@ async function handleMemorySearch(query) {
   return result;
 }
 async function handleRead5(uri) {
-  const projectContextMatch = uri.match(/^context:\/\/project\/(.+)$/);
+  const projectContextMatch = uri.match(/^context:\/\/project\/([^/]+)$/);
   if (projectContextMatch) {
     return resourceContent(uri, await handleProjectContext(projectContextMatch[1]));
   }
-  const clientContextMatch = uri.match(/^context:\/\/client\/(.+)$/);
+  const clientContextMatch = uri.match(/^context:\/\/client\/([^/]+)$/);
   if (clientContextMatch) {
     return resourceContent(uri, await handleClientContext(clientContextMatch[1]));
   }
-  const projectDecisionsMatch = uri.match(/^decisions:\/\/project\/(.+)$/);
+  const projectDecisionsMatch = uri.match(/^decisions:\/\/project\/([^/]+)$/);
   if (projectDecisionsMatch) {
     return resourceContent(uri, await handleProjectDecisions(projectDecisionsMatch[1]));
   }
   if (uri === "decisions://org") {
     return resourceContent(uri, await handleOrgDecisions());
   }
-  const projectMeetingsMatch = uri.match(/^meetings:\/\/project\/(.+)$/);
+  const projectMeetingsMatch = uri.match(/^meetings:\/\/project\/([^/]+)$/);
   if (projectMeetingsMatch) {
     return resourceContent(uri, await handleProjectMeetings(projectMeetingsMatch[1]));
   }
@@ -1244,8 +1244,8 @@ var tools4 = [
   { name: "pipeline_move", description: "Move an opportunity to a different pipeline stage", inputSchema: { type: "object", properties: { opportunityId: { type: "string" }, stage: { type: "string", enum: ["lead", "discovery", "qualified", "proposal_draft", "proposal_submitted", "negotiation", "won", "lost"] }, notes: { type: "string" } }, required: ["opportunityId", "stage"] } }
 ];
 async function handlePipelineMove(args) {
-  const input = PipelineMoveSchema.parse(args);
-  return await apiPatch(`/pipeline/${input.opportunityId}/stage`, input);
+  const { opportunityId, ...data } = PipelineMoveSchema.parse(args);
+  return await apiPatch(`/pipeline/${encodeURIComponent(opportunityId)}/stage`, data);
 }
 async function handleToolCall4(name, args) {
   try {
@@ -1335,8 +1335,8 @@ async function handleSiteVisitLog(args) {
   return await apiPost("/compliance/site-visits", input);
 }
 async function handleComplianceUpdate(args) {
-  const input = ComplianceUpdateSchema.parse(args);
-  const updated = await apiPatch(`/compliance/${input.complianceId}`, input);
+  const { complianceId, ...data } = ComplianceUpdateSchema.parse(args);
+  const updated = await apiPatch(`/compliance/${encodeURIComponent(complianceId)}`, data);
   return { id: updated.id, name: updated.name, status: updated.status };
 }
 async function handleToolCall6(name, args) {
@@ -1698,8 +1698,8 @@ async function handleCredentialCreate(args) {
   return { id: cred.id, name: cred.name, type: cred.type, status: cred.status };
 }
 async function handleCredentialUpdate(args) {
-  const input = CredentialUpdateSchema.parse(args);
-  const updated = await apiPatch(`/credentials/${input.credentialId}`, input);
+  const { credentialId, ...data } = CredentialUpdateSchema.parse(args);
+  const updated = await apiPatch(`/credentials/${encodeURIComponent(credentialId)}`, data);
   return { id: updated.id, name: updated.name, status: updated.status, expiresAt: updated.expiresAt };
 }
 async function handleCredentialSearch(args) {
@@ -1763,13 +1763,14 @@ async function handleCapabilityAdd(args) {
   return { id: cap.id, name: cap.name, category: cap.category, proficiency: cap.proficiency };
 }
 async function handleCapabilityUpdate(args) {
-  const input = CapabilityUpdateSchema.parse(args);
-  const updated = await apiPatch(`/capabilities/${input.capabilityId}`, input);
+  const { capabilityId, ...data } = CapabilityUpdateSchema.parse(args);
+  const updated = await apiPatch(`/capabilities/${encodeURIComponent(capabilityId)}`, data);
   return { id: updated.id, name: updated.name, proficiency: updated.proficiency };
 }
 async function handleCapabilityEvidenceAdd(args) {
   const input = CapabilityEvidenceAddSchema.parse(args);
-  const evidence = await apiPost(`/capabilities/${input.capabilityId}/evidence`, input);
+  const { capabilityId, ...data } = input;
+  const evidence = await apiPost(`/capabilities/${encodeURIComponent(capabilityId)}/evidence`, data);
   return evidence;
 }
 async function handleToolCall13(name, args) {
@@ -1894,7 +1895,7 @@ var tools15 = [
         notes: { type: "string", description: "Notes" },
         linkedInUrl: { type: "string", description: "LinkedIn URL" }
       },
-      required: ["firstName"]
+      required: ["firstName", "lastName"]
     }
   }
 ];
@@ -1950,8 +1951,8 @@ async function handleDeliverableCreate(args) {
   return { id: d.id, name: d.name, type: d.type, status: d.status };
 }
 async function handleDeliverableUpdate(args) {
-  const input = DeliverableUpdateSchema.parse(args);
-  const updated = await apiPatch(`/deliverables/${input.deliverableId}`, input);
+  const { deliverableId, ...data } = DeliverableUpdateSchema.parse(args);
+  const updated = await apiPatch(`/deliverables/${encodeURIComponent(deliverableId)}`, data);
   return { id: updated.id, name: updated.name, status: updated.status, percentComplete: updated.percentComplete };
 }
 async function handleToolCall16(name, args) {
@@ -2033,6 +2034,182 @@ async function handleToolCall17(name, args) {
   }
 }
 
+// src/tools/asset-tools.ts
+var asset_tools_exports = {};
+__export(asset_tools_exports, {
+  handleToolCall: () => handleToolCall18,
+  tools: () => tools18
+});
+import { z as z19 } from "zod";
+var SiteListSchema = z19.object({
+  query: z19.string().optional(),
+  limit: z19.number().optional()
+});
+var SiteGetSchema = z19.object({
+  siteId: z19.string()
+});
+var AssetSearchSchema = z19.object({
+  query: z19.string().optional(),
+  siteId: z19.string().optional(),
+  assetClass: z19.string().optional(),
+  assetType: z19.string().optional(),
+  limit: z19.number().optional()
+});
+var AssetGetSchema = z19.object({
+  assetId: z19.string()
+});
+var AssetDocSearchSchema = z19.object({
+  query: z19.string(),
+  assetId: z19.string().optional(),
+  siteId: z19.string().optional(),
+  limit: z19.number().optional()
+});
+var AssetDocListSchema = z19.object({
+  assetId: z19.string().optional(),
+  siteId: z19.string().optional(),
+  limit: z19.number().optional()
+});
+var tools18 = [
+  {
+    name: "site_list",
+    description: "List or search sites by name",
+    inputSchema: {
+      type: "object",
+      properties: {
+        query: { type: "string", description: "Search site names" },
+        limit: { type: "number", description: "Max results (default 50)" }
+      }
+    }
+  },
+  {
+    name: "site_get",
+    description: "Get site details including asset/area counts",
+    inputSchema: {
+      type: "object",
+      properties: {
+        siteId: { type: "string", description: "Site ID" }
+      },
+      required: ["siteId"]
+    }
+  },
+  {
+    name: "asset_search",
+    description: "Search physical assets by name, tag, class, or type. Optionally scope to a site.",
+    inputSchema: {
+      type: "object",
+      properties: {
+        query: { type: "string", description: "Search asset names, tags, descriptions" },
+        siteId: { type: "string", description: "Filter to assets at this site" },
+        assetClass: { type: "string", description: "Filter by asset class (e.g., Rotating Equipment)" },
+        assetType: { type: "string", description: "Filter by asset type (e.g., Pump, Motor)" },
+        limit: { type: "number", description: "Max results (default 50)" }
+      }
+    }
+  },
+  {
+    name: "asset_get",
+    description: "Get full asset details with hierarchy, failure modes, and reliability metrics",
+    inputSchema: {
+      type: "object",
+      properties: {
+        assetId: { type: "string", description: "Asset ID" }
+      },
+      required: ["assetId"]
+    }
+  },
+  {
+    name: "asset_documentation_search",
+    description: "Semantic search across documents indexed for a specific asset or site",
+    inputSchema: {
+      type: "object",
+      properties: {
+        query: { type: "string", description: "Search query" },
+        assetId: { type: "string", description: "Scope to this asset" },
+        siteId: { type: "string", description: "Scope to this site" },
+        limit: { type: "number", description: "Max results (default 10)" }
+      },
+      required: ["query"]
+    }
+  },
+  {
+    name: "asset_documentation_list",
+    description: "List all indexed documents for an asset or site",
+    inputSchema: {
+      type: "object",
+      properties: {
+        assetId: { type: "string", description: "List docs for this asset" },
+        siteId: { type: "string", description: "List docs for this site" },
+        limit: { type: "number", description: "Max results (default 50)" }
+      }
+    }
+  }
+];
+async function handleSiteList(args) {
+  const input = SiteListSchema.parse(args);
+  return apiGet("/sites", {
+    query: input.query,
+    limit: input.limit ?? 50
+  });
+}
+async function handleSiteGet(args) {
+  const input = SiteGetSchema.parse(args);
+  return apiGet(`/sites/${encodeURIComponent(input.siteId)}`);
+}
+async function handleAssetSearch(args) {
+  const input = AssetSearchSchema.parse(args);
+  return apiGet("/physical-assets", {
+    query: input.query,
+    siteId: input.siteId,
+    assetClass: input.assetClass,
+    assetType: input.assetType,
+    limit: input.limit ?? 50
+  });
+}
+async function handleAssetGet(args) {
+  const input = AssetGetSchema.parse(args);
+  return apiGet(`/physical-assets/${encodeURIComponent(input.assetId)}`);
+}
+async function handleAssetDocSearch(args) {
+  const input = AssetDocSearchSchema.parse(args);
+  return apiGet("/physical-assets/documentation", {
+    query: input.query,
+    assetId: input.assetId,
+    siteId: input.siteId,
+    limit: input.limit ?? 10
+  });
+}
+async function handleAssetDocList(args) {
+  const input = AssetDocListSchema.parse(args);
+  return apiGet("/physical-assets/documentation", {
+    assetId: input.assetId,
+    siteId: input.siteId,
+    limit: input.limit ?? 50
+  });
+}
+async function handleToolCall18(name, args) {
+  try {
+    switch (name) {
+      case "site_list":
+        return success(await handleSiteList(args));
+      case "site_get":
+        return success(await handleSiteGet(args));
+      case "asset_search":
+        return success(await handleAssetSearch(args));
+      case "asset_get":
+        return success(await handleAssetGet(args));
+      case "asset_documentation_search":
+        return success(await handleAssetDocSearch(args));
+      case "asset_documentation_list":
+        return success(await handleAssetDocList(args));
+      default:
+        return null;
+    }
+  } catch (err) {
+    console.error(`Tool ${name} failed:`, err instanceof Error ? err.message : err);
+    return safeError(err);
+  }
+}
+
 // src/tools/index.ts
 var modules2 = [
   task_tools_exports,
@@ -2051,10 +2228,11 @@ var modules2 = [
   availability_tools_exports,
   contact_tools_exports,
   deliverable_tools_exports,
-  memory_tools_exports
+  memory_tools_exports,
+  asset_tools_exports
 ];
-var tools18 = modules2.flatMap((m) => [...m.tools]);
-async function handleToolCall18(name, args) {
+var tools19 = modules2.flatMap((m) => [...m.tools]);
+async function handleToolCall19(name, args) {
   try {
     checkRateLimit(name);
   } catch (err) {
@@ -3267,11 +3445,11 @@ async function generatePromptContent(name, args) {
 }
 
 // src/auth/api-key.ts
-import { z as z19 } from "zod";
-var apiKeyPayloadSchema = z19.object({
-  orgId: z19.string(),
-  permissions: z19.array(z19.enum(["read", "write", "admin"])),
-  expiresAt: z19.date().optional()
+import { z as z20 } from "zod";
+var apiKeyPayloadSchema = z20.object({
+  orgId: z20.string(),
+  permissions: z20.array(z20.enum(["read", "write", "admin"])),
+  expiresAt: z20.date().optional()
 });
 function isValidApiKeyFormat(key) {
   return /^amplis_[a-f0-9]{64}$/.test(key);
@@ -3446,8 +3624,48 @@ import open from "open";
 var DEFAULT_API_URL2 = "https://app.amplis.com.au";
 var CALLBACK_PORT = 19284;
 var TIMEOUT_MS = 12e4;
+function escapeHtml(s) {
+  return s.replace(/&/g, "&amp;").replace(/</g, "&lt;").replace(/>/g, "&gt;").replace(/"/g, "&quot;");
+}
+function renderPage(opts) {
+  const iconSvg = opts.success ? '<svg width="48" height="48" viewBox="0 0 24 24" fill="none" stroke="#22c55e" stroke-width="2" stroke-linecap="round" stroke-linejoin="round"><path d="M22 11.08V12a10 10 0 1 1-5.93-9.14"/><polyline points="22 4 12 14.01 9 11.01"/></svg>' : '<svg width="48" height="48" viewBox="0 0 24 24" fill="none" stroke="#ef4444" stroke-width="2" stroke-linecap="round" stroke-linejoin="round"><circle cx="12" cy="12" r="10"/><line x1="15" y1="9" x2="9" y2="15"/><line x1="9" y1="9" x2="15" y2="15"/></svg>';
+  const permissionBadges = opts.permissions ? `<div style="display:flex;gap:8px;justify-content:center;margin-top:16px">${opts.permissions.map((p) => {
+    const colors = p === "admin" ? "background:rgba(251,191,36,0.15);color:#f59e0b" : p === "write" ? "background:rgba(74,141,255,0.15);color:#4A8DFF" : "background:rgba(52,211,153,0.15);color:#34d399";
+    return `<span style="padding:4px 12px;border-radius:9999px;font-size:12px;font-weight:500;${colors}">${escapeHtml(p)}</span>`;
+  }).join("")}</div>` : "";
+  const orgLine = opts.orgName ? `<p style="font-size:15px;color:#cbd5e1;line-height:1.5">Logged in to <strong style="color:#f1f5f9">${escapeHtml(opts.orgName)}</strong></p>` : `<p style="font-size:15px;color:#cbd5e1;line-height:1.5">${escapeHtml(opts.message)}</p>`;
+  const autoCloseScript = opts.autoClose ? `<p id="countdown" style="font-size:13px;color:#64748b;margin-top:20px">This window will close in <span id="secs">5</span>s</p>
+       <script>let s=5;const el=document.getElementById('secs');setInterval(()=>{s--;if(el)el.textContent=String(s);if(s<=0)window.close();},1000);</script>` : "";
+  return `<!DOCTYPE html>
+<html lang="en">
+<head>
+  <meta charset="utf-8">
+  <meta name="viewport" content="width=device-width, initial-scale=1">
+  <title>AMPLIS - ${escapeHtml(opts.title)}</title>
+  <style>
+    *{margin:0;padding:0;box-sizing:border-box}
+    body{font-family:-apple-system,BlinkMacSystemFont,'Segoe UI',Roboto,sans-serif;min-height:100vh;display:flex;align-items:center;justify-content:center;background:#0a0e1a}
+    .card{background:#111827;border:1px solid rgba(255,255,255,0.08);border-radius:16px;padding:48px 40px;max-width:420px;width:100%;text-align:center;box-shadow:0 25px 50px rgba(0,0,0,0.4)}
+    .icon{margin:0 auto 24px}
+    h1{font-size:22px;font-weight:700;color:${opts.success ? "#22c55e" : "#ef4444"};margin-bottom:12px}
+    .hint{font-size:13px;color:#64748b;margin-top:20px}
+    .brand{margin-top:32px;padding-top:20px;border-top:1px solid rgba(255,255,255,0.06);font-size:12px;color:#475569;letter-spacing:0.5px}
+  </style>
+</head>
+<body>
+  <div class="card">
+    <div class="icon">${iconSvg}</div>
+    <h1>${escapeHtml(opts.title)}</h1>
+    ${orgLine}
+    ${permissionBadges}
+    ${opts.success ? `<p class="hint">You can close this window and return to your terminal.</p>${autoCloseScript}` : '<p class="hint">Please close this window and try again.</p>'}
+    <div class="brand">AMPLIS</div>
+  </div>
+</body>
+</html>`;
+}
 async function browserLoginCommand() {
-  console.log("\n  \u{1F510} AMPLIS MCP Server \u2014 Browser Authentication\n");
+  console.log("\n  AMPLIS MCP Server - Browser Authentication\n");
   const apiUrl = process.env.AMPLIS_API_URL || DEFAULT_API_URL2;
   const state = crypto.randomBytes(32).toString("hex");
   const callbackUrl = `http://localhost:${CALLBACK_PORT}/callback`;
@@ -3459,40 +3677,22 @@ async function browserLoginCommand() {
         const returnedState = url.searchParams.get("state");
         const error2 = url.searchParams.get("error");
         if (error2) {
-          res.writeHead(400, { "Content-Type": "text/html" });
-          res.end(`
-            <html><body style="font-family: system-ui; padding: 40px; text-align: center;">
-              <h1 style="color: #dc2626;">\u274C Authentication Failed</h1>
-              <p>${error2}</p>
-              <p>You can close this window.</p>
-            </body></html>
-          `);
+          res.writeHead(400, { "Content-Type": "text/html; charset=utf-8" });
+          res.end(renderPage({ success: false, title: "Authentication Failed", message: error2 }));
           server.close();
           reject(new Error(error2));
           return;
         }
         if (returnedState !== state) {
-          res.writeHead(400, { "Content-Type": "text/html" });
-          res.end(`
-            <html><body style="font-family: system-ui; padding: 40px; text-align: center;">
-              <h1 style="color: #dc2626;">\u274C Invalid State</h1>
-              <p>Security validation failed. Please try again.</p>
-              <p>You can close this window.</p>
-            </body></html>
-          `);
+          res.writeHead(400, { "Content-Type": "text/html; charset=utf-8" });
+          res.end(renderPage({ success: false, title: "Invalid State", message: "Security validation failed. Please try again." }));
           server.close();
           reject(new Error("Invalid state parameter"));
           return;
         }
         if (!code) {
-          res.writeHead(400, { "Content-Type": "text/html" });
-          res.end(`
-            <html><body style="font-family: system-ui; padding: 40px; text-align: center;">
-              <h1 style="color: #dc2626;">\u274C Missing Code</h1>
-              <p>No authorization code received.</p>
-              <p>You can close this window.</p>
-            </body></html>
-          `);
+          res.writeHead(400, { "Content-Type": "text/html; charset=utf-8" });
+          res.end(renderPage({ success: false, title: "Missing Code", message: "No authorization code received." }));
           server.close();
           reject(new Error("Missing authorization code"));
           return;
@@ -3508,26 +3708,20 @@ async function browserLoginCommand() {
             throw new Error(errorBody.error || `HTTP ${exchangeResponse.status}`);
           }
           const result = await exchangeResponse.json();
-          res.writeHead(200, { "Content-Type": "text/html" });
-          res.end(`
-            <html><body style="font-family: system-ui; padding: 40px; text-align: center;">
-              <h1 style="color: #16a34a;">\u2705 Authentication Successful!</h1>
-              <p>Logged in to <strong>${result.orgName}</strong></p>
-              <p>You can close this window and return to your terminal.</p>
-              <script>setTimeout(() => window.close(), 3000);</script>
-            </body></html>
-          `);
+          res.writeHead(200, { "Content-Type": "text/html; charset=utf-8" });
+          res.end(renderPage({
+            success: true,
+            title: "Authentication Successful",
+            orgName: result.orgName,
+            message: "",
+            permissions: result.permissions,
+            autoClose: true
+          }));
           server.close();
           resolve(result);
         } catch (err) {
-          res.writeHead(500, { "Content-Type": "text/html" });
-          res.end(`
-            <html><body style="font-family: system-ui; padding: 40px; text-align: center;">
-              <h1 style="color: #dc2626;">\u274C Exchange Failed</h1>
-              <p>${err instanceof Error ? err.message : "Unknown error"}</p>
-              <p>You can close this window.</p>
-            </body></html>
-          `);
+          res.writeHead(500, { "Content-Type": "text/html; charset=utf-8" });
+          res.end(renderPage({ success: false, title: "Exchange Failed", message: err instanceof Error ? err.message : "Unknown error" }));
           server.close();
           reject(err);
         }
@@ -3537,7 +3731,7 @@ async function browserLoginCommand() {
       }
     });
     server.listen(CALLBACK_PORT, "127.0.0.1", () => {
-      console.log(`  \u{1F4E1} Callback server listening on port ${CALLBACK_PORT}`);
+      console.log(`  Callback server listening on port ${CALLBACK_PORT}`);
     });
     server.on("error", (err) => {
       reject(new Error(`Failed to start callback server: ${err.message}`));
@@ -3550,7 +3744,7 @@ async function browserLoginCommand() {
   const authUrl = new URL(`${apiUrl}/auth/mcp`);
   authUrl.searchParams.set("callback", callbackUrl);
   authUrl.searchParams.set("state", state);
-  console.log(`  \u{1F310} Opening browser to authenticate...
+  console.log(`  Opening browser to authenticate...
 `);
   console.log(`  If the browser doesn't open, visit:`);
   console.log(`  ${authUrl.toString()}
@@ -3559,7 +3753,7 @@ async function browserLoginCommand() {
     await open(authUrl.toString());
   } catch {
   }
-  console.log("  \u23F3 Waiting for authentication...\n");
+  console.log("  Waiting for authentication...\n");
   try {
     const result = await authPromise;
     saveCredentials({
@@ -3569,14 +3763,14 @@ async function browserLoginCommand() {
       userId: result.userId,
       permissions: result.permissions
     });
-    console.log("  \u2705 Authentication successful!\n");
+    console.log("  Authentication successful!\n");
     console.log(`  Organization : ${result.orgName}`);
     console.log(`  Permissions  : ${result.permissions.join(", ")}`);
     console.log(`  Key prefix   : ${result.apiKey.substring(0, 16)}...`);
     console.log("\n  Credentials saved to ~/.amplis/credentials.json\n");
   } catch (err) {
     console.error(`
-  \u274C Authentication failed: ${err instanceof Error ? err.message : "Unknown error"}
+  Authentication failed: ${err instanceof Error ? err.message : "Unknown error"}
 `);
     process.exit(1);
   }
@@ -3694,7 +3888,7 @@ if (command !== "serve") {
   const server = new Server(
     {
       name: "amplis-mcp-server",
-      version: "0.3.0"
+      version: "0.4.0"
     },
     {
       capabilities: {
@@ -3714,10 +3908,10 @@ if (command !== "serve") {
     return handleReadResource(request.params.uri);
   });
   server.setRequestHandler(ListToolsRequestSchema, async () => {
-    return { tools: tools18 };
+    return { tools: tools19 };
   });
   server.setRequestHandler(CallToolRequestSchema, async (request) => {
-    return handleToolCall18(request.params.name, request.params.arguments);
+    return handleToolCall19(request.params.name, request.params.arguments);
   });
   server.setRequestHandler(ListPromptsRequestSchema, async () => {
     return { prompts };

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@amplis/mcp-server",
-  "version": "0.3.2",
+  "version": "0.4.0",
   "description": "MCP server for AMPLIS - connect Claude, Cursor, and other AI tools to your live consulting data",
   "author": "AMPLIS (https://amplis.com.au)",
   "license": "MIT",