@armory-sh/middleware-elysia 0.3.27 → 0.3.28

package/dist/index.d.ts

@@ -16,6 +16,7 @@ interface PaymentConfig {
     facilitatorUrl?: string;
     facilitatorUrlByChain?: Record<string, string>;
     facilitatorUrlByToken?: Record<string, Record<string, string>>;
+    extensions?: Record<string, unknown>;
 }
 interface ResolvedRequirementsConfig {
     requirements: PaymentRequirementsV2[];

package/dist/index.js

@@ -1,7 +1,9 @@
 export { routeAwarePaymentMiddleware } from './chunk-L45RQOFN.js';
-import { resolveNetwork, isValidationError, resolveToken, createPaymentRequirements as createPaymentRequirements$1, PAYMENT_SIGNATURE_HEADER, createPaymentRequiredHeaders, decodePayloadHeader, findRequirementByAccepted, verifyPayment, settlePayment, createSettlementHeaders, TOKENS, registerToken } from '@armory-sh/base';
+import { resolveNetwork, isValidationError, resolveToken, createPaymentRequirements as createPaymentRequirements$1, PAYMENT_SIGNATURE_HEADER, decodePayloadHeader, findRequirementByAccepted, verifyPayment, settlePayment, createSettlementHeaders, createPaymentRequiredHeaders, TOKENS, registerToken, getSupported } from '@armory-sh/base';
 import { Elysia } from 'elysia';
 
+var extensionCapabilityCache = /* @__PURE__ */ new Map();
+var EXTENSION_CAPABILITY_TTL_MS = 5 * 60 * 1e3;
 function ensureTokensRegistered() {
   for (const token of Object.values(TOKENS)) {
     try {
@@ -18,17 +20,13 @@ function resolveFacilitatorUrlFromRequirement(config, requirement) {
       config.facilitatorUrlByToken
     )) {
       const resolvedChain = resolveNetwork(chainKey);
-      if (!isValidationError(resolvedChain) && resolvedChain.config.chainId === chainId) {
-        for (const [, url] of Object.entries(tokenMap)) {
-          const network = resolveNetwork(chainKey);
-          if (!isValidationError(network)) {
-            for (const tokenKey of Object.keys(tokenMap)) {
-              const resolvedToken = resolveToken(tokenKey, network);
-              if (!isValidationError(resolvedToken) && resolvedToken.config.contractAddress.toLowerCase() === assetAddress) {
-                return url;
-              }
-            }
-          }
+      if (isValidationError(resolvedChain) || resolvedChain.config.chainId !== chainId) {
+        continue;
+      }
+      for (const [tokenKey, url] of Object.entries(tokenMap)) {
+        const resolvedToken = resolveToken(tokenKey, resolvedChain);
+        if (!isValidationError(resolvedToken) && resolvedToken.config.contractAddress.toLowerCase() === assetAddress) {
+          return url;
         }
       }
     }
@@ -63,12 +61,60 @@ function createPaymentRequirements(config) {
   ensureTokensRegistered();
   return createPaymentRequirements$1(config);
 }
+async function resolvePaymentRequiredExtensions(config, requirements) {
+  if (!config.extensions) {
+    return {};
+  }
+  let filtered = { ...config.extensions };
+  for (const requirement of requirements) {
+    const facilitatorUrl = resolveFacilitatorUrlFromRequirement(config, requirement);
+    if (!facilitatorUrl) {
+      continue;
+    }
+    const cacheKey = `${facilitatorUrl}|${requirement.network.toLowerCase()}`;
+    const now = Date.now();
+    let keys = extensionCapabilityCache.get(cacheKey);
+    if (!keys || keys.expiresAt <= now) {
+      try {
+        const supported = await getSupported({ url: facilitatorUrl });
+        const nextKeys = /* @__PURE__ */ new Set();
+        for (const kind of supported.kinds) {
+          if (kind.network.toLowerCase() !== requirement.network.toLowerCase()) {
+            continue;
+          }
+          if (kind.extra && typeof kind.extra === "object") {
+            for (const key of Object.keys(kind.extra)) {
+              nextKeys.add(key);
+            }
+          }
+        }
+        keys = { expiresAt: now + EXTENSION_CAPABILITY_TTL_MS, keys: nextKeys };
+      } catch {
+        keys = { expiresAt: now + EXTENSION_CAPABILITY_TTL_MS, keys: /* @__PURE__ */ new Set() };
+      }
+      extensionCapabilityCache.set(cacheKey, keys);
+    }
+    filtered = Object.fromEntries(
+      Object.entries(filtered).filter(([key]) => keys.keys.has(key))
+    );
+    if (Object.keys(filtered).length === 0) {
+      return {};
+    }
+  }
+  return filtered;
+}
 var errorResponse = (error, status, headers) => new Response(JSON.stringify(error), {
   status,
   headers: { "Content-Type": "application/json", ...headers }
 });
 var paymentMiddleware = (config) => {
   const { requirements, error } = createPaymentRequirements(config);
+  const resolvePaymentRequiredHeaders = async () => createPaymentRequiredHeaders(requirements, {
+    extensions: await resolvePaymentRequiredExtensions(
+      config,
+      requirements
+    )
+  });
   return new Elysia({ name: "armory-payment" }).derive(() => ({
     payment: void 0
   })).onBeforeHandle(async (context) => {
@@ -95,10 +141,11 @@ var paymentMiddleware = (config) => {
       PAYMENT_SIGNATURE_HEADER
     );
     if (!paymentHeader) {
+      const requiredHeaders = await resolvePaymentRequiredHeaders();
       return errorResponse(
         { error: "Payment required", accepts: requirements },
         402,
-        createPaymentRequiredHeaders(requirements)
+        requiredHeaders
       );
     }
     let paymentPayload;
@@ -147,13 +194,14 @@ var paymentMiddleware = (config) => {
       { url: facilitatorUrl }
     );
     if (!verifyResult.isValid) {
+      const requiredHeaders = await resolvePaymentRequiredHeaders();
       return errorResponse(
         {
           error: "Payment verification failed",
           message: verifyResult.invalidReason
         },
         402,
-        createPaymentRequiredHeaders(requirements)
+        requiredHeaders
       );
     }
     const payerAddress = verifyResult.payer ?? paymentPayload.payload.authorization.from;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@armory-sh/middleware-elysia",
-  "version": "0.3.27",
+  "version": "0.3.28",
   "license": "MIT",
   "author": "Sawyer Cutler <sawyer@dirtroad.dev>",
   "keywords": [
@@ -47,7 +47,7 @@
     "elysia": "^1"
   },
   "dependencies": {
-    "@armory-sh/base": "0.2.29"
+    "@armory-sh/base": "0.2.30"
   },
   "devDependencies": {
     "bun-types": "latest",