@solvapay/server 1.0.2 → 1.0.3

package/dist/edge.js

@@ -162,6 +162,21 @@ function createSolvaPayClient(opts) {
       }
       return await res.json();
     },
+    // PUT: /v1/sdk/products/{productRef}/mcp/plans
+    async configureMcpPlans(productRef, params) {
+      const url = `${base}/v1/sdk/products/${productRef}/mcp/plans`;
+      const res = await fetch(url, {
+        method: "PUT",
+        headers,
+        body: JSON.stringify(params)
+      });
+      if (!res.ok) {
+        const error = await res.text();
+        log(`\u274C API Error: ${res.status} - ${error}`);
+        throw new SolvaPayError(`Configure MCP plans failed (${res.status}): ${error}`);
+      }
+      return await res.json();
+    },
     // DELETE: /v1/sdk/products/{productRef}
     async deleteProduct(productRef) {
       const url = `${base}/v1/sdk/products/${productRef}`;
@@ -919,10 +934,10 @@ async function createAdapterHandler(adapter, paywall, metadata, businessLogic) {
   const backendRefCache = /* @__PURE__ */ new Map();
   const getCustomerRef = (args) => args.auth?.customer_ref || "anonymous";
   const protectedHandler = await paywall.protect(businessLogic, metadata, getCustomerRef);
-  return async (context) => {
+  return async (context, extra) => {
     try {
       const args = await adapter.extractArgs(context);
-      const customerRef = await adapter.getCustomerRef(context);
+      const customerRef = await adapter.getCustomerRef(context, extra);
       let backendRef = backendRefCache.get(customerRef);
       if (!backendRef) {
         backendRef = await paywall.ensureCustomer(customerRef, customerRef);
@@ -1113,18 +1128,18 @@ var McpAdapter = class {
   extractArgs(args) {
     return args;
   }
-  async getCustomerRef(args) {
+  async getCustomerRef(args, extra) {
     if (this.options.getCustomerRef) {
-      const ref = await this.options.getCustomerRef(args);
+      const ref = await this.options.getCustomerRef(args, extra);
       return AdapterUtils.ensureCustomerRef(ref);
     }
-    const auth = args?.auth;
-    const customerRef = auth?.customer_ref || "anonymous";
+    const customerRefFromExtra = extra?.authInfo?.extra?.customer_ref;
+    const customerRef = typeof customerRefFromExtra === "string" && customerRefFromExtra.trim() ? customerRefFromExtra.trim() : "anonymous";
     return AdapterUtils.ensureCustomerRef(customerRef);
   }
   formatResponse(result, _context) {
     const transformed = this.options.transformResponse ? this.options.transformResponse(result) : result;
-    return {
+    const response = {
       content: [
         {
           type: "text",
@@ -1132,6 +1147,10 @@ var McpAdapter = class {
         }
       ]
     };
+    if (transformed && typeof transformed === "object" && !Array.isArray(transformed)) {
+      response.structuredContent = transformed;
+    }
+    return response;
   }
   formatError(error, _context) {
     if (error instanceof PaywallError) {
@@ -1218,8 +1237,8 @@ function mcpErrorResult(message) {
   return { content: [{ type: "text", text: JSON.stringify({ error: message }) }], isError: true };
 }
 function createGetUserInfoHandler(apiClient, productRef, getCustomerRef) {
-  return async (args) => {
-    const customerRef = getCustomerRef(args);
+  return async (args, extra) => {
+    const customerRef = getCustomerRef(args, extra);
     try {
       if (!apiClient.getUserInfo) {
         return mcpErrorResult("getUserInfo is not available on this API client");
@@ -1234,12 +1253,12 @@ function createGetUserInfoHandler(apiClient, productRef, getCustomerRef) {
   };
 }
 function createUpgradeHandler(apiClient, productRef, getCustomerRef) {
-  return async (args) => {
-    const customerRef = getCustomerRef(args);
+  return async (args, extra) => {
+    const customerRef = getCustomerRef(args, extra);
     const planRef = typeof args.planRef === "string" ? args.planRef : void 0;
     try {
       const result = await apiClient.createCheckoutSession({
-        customerReference: customerRef,
+        customerRef,
         productRef,
         ...planRef && { planRef }
       });
@@ -1266,8 +1285,8 @@ You'll be able to compare options and select the one that's right for you.`;
   };
 }
 function createManageAccountHandler(apiClient, productRef, getCustomerRef) {
-  return async (args) => {
-    const customerRef = getCustomerRef(args);
+  return async (args, extra) => {
+    const customerRef = getCustomerRef(args, extra);
     try {
       const session = await apiClient.createCustomerSession({ customerRef, productRef });
       const portalUrl = session.customerUrl;
@@ -1310,6 +1329,98 @@ function createVirtualTools(apiClient, options) {
   return allTools.filter((t) => !excludeSet.has(t.name));
 }
 
+// src/register-virtual-tools-mcp.ts
+import { createRequire } from "module";
+function getNodeRequire() {
+  try {
+    return (0, eval)("require");
+  } catch {
+    return createRequire(`${process.cwd()}/package.json`);
+  }
+}
+var nodeRequire = getNodeRequire();
+function getZod() {
+  try {
+    const zodModule = nodeRequire("zod");
+    return zodModule.z ?? zodModule;
+  } catch {
+    throw new Error(
+      "zod is required to use registerVirtualToolsMcp(). Install it as a dependency in your MCP server project."
+    );
+  }
+}
+function isJsonSchemaObject(value) {
+  return Boolean(value && typeof value === "object");
+}
+function toZodSchema(property) {
+  const z = getZod();
+  if (Array.isArray(property.enum) && property.enum.length > 0) {
+    if (property.enum.every((value) => typeof value === "string")) {
+      const [first, ...rest] = property.enum;
+      return z.enum([first, ...rest]);
+    }
+    if (property.enum.length === 1) {
+      return z.literal(property.enum[0]);
+    }
+    return z.union(property.enum.map((value) => z.literal(value)));
+  }
+  switch (property.type) {
+    case "string":
+      return z.string();
+    case "number":
+      return z.number();
+    case "integer":
+      return z.number().int();
+    case "boolean":
+      return z.boolean();
+    case "array":
+      return z.array(
+        isJsonSchemaObject(property.items) ? toZodSchema(property.items) : z.any()
+      );
+    case "object":
+      return z.object(jsonSchemaToZodRawShape(property.properties || {}));
+    default:
+      return z.any();
+  }
+}
+function jsonSchemaToZodRawShape(properties, required = []) {
+  const requiredSet = new Set(required);
+  return Object.fromEntries(
+    Object.entries(properties).map(([name, property]) => {
+      const schema = toZodSchema(property);
+      const finalSchema = requiredSet.has(name) ? schema : schema.optional();
+      return [name, finalSchema];
+    })
+  );
+}
+function defaultGetCustomerRef(_args, extra) {
+  const fromExtra = extra?.authInfo?.extra?.customer_ref;
+  return typeof fromExtra === "string" && fromExtra.trim() ? fromExtra.trim() : "anonymous";
+}
+function registerVirtualToolsMcpImpl(server, apiClient, options) {
+  const { filter, mapDefinition, wrapHandler, getCustomerRef, ...virtualToolsOptions } = options;
+  const virtualTools = createVirtualTools(apiClient, {
+    ...virtualToolsOptions,
+    getCustomerRef: getCustomerRef || defaultGetCustomerRef
+  });
+  for (const toolDefinition of virtualTools) {
+    if (filter && !filter(toolDefinition)) continue;
+    const mappedDefinition = mapDefinition ? mapDefinition(toolDefinition) : toolDefinition;
+    const wrappedHandler = wrapHandler ? wrapHandler(mappedDefinition.handler, mappedDefinition) : mappedDefinition.handler;
+    server.registerTool(
+      mappedDefinition.name,
+      {
+        description: mappedDefinition.description,
+        inputSchema: jsonSchemaToZodRawShape(
+          mappedDefinition.inputSchema.properties,
+          mappedDefinition.inputSchema.required || []
+        )
+      },
+      wrappedHandler
+    );
+  }
+}
+
 // src/factory.ts
 function createSolvaPay(config) {
   let resolvedConfig;
@@ -1366,7 +1477,7 @@ function createSolvaPay(config) {
     },
     createCheckoutSession(params) {
       return apiClient.createCheckoutSession({
-        customerReference: params.customerRef,
+        customerRef: params.customerRef,
         productRef: params.productRef,
         planRef: params.planRef
       });
@@ -1380,9 +1491,18 @@ function createSolvaPay(config) {
       }
       return apiClient.bootstrapMcpProduct(params);
     },
+    configureMcpPlans(productRef, params) {
+      if (!apiClient.configureMcpPlans) {
+        throw new SolvaPayError2("configureMcpPlans is not available on this API client");
+      }
+      return apiClient.configureMcpPlans(productRef, params);
+    },
     getVirtualTools(options) {
       return createVirtualTools(apiClient, options);
     },
+    async registerVirtualToolsMcp(server, options) {
+      await registerVirtualToolsMcpImpl(server, apiClient, options);
+    },
     // Payable API for framework-specific handlers
     payable(options = {}) {
       const product = options.productRef || options.product || process.env.SOLVAPAY_PRODUCT || "default-product";
@@ -1421,9 +1541,9 @@ function createSolvaPay(config) {
             getCustomerRef: adapterOptions?.getCustomerRef || options.getCustomerRef
           });
           const handlerPromise = createAdapterHandler(adapter, paywall, metadata, businessLogic);
-          return async (args) => {
+          return async (args, extra) => {
             const handler = await handlerPromise;
-            return handler(args);
+            return handler(args, extra);
           };
         },
         // eslint-disable-next-line @typescript-eslint/no-explicit-any