@slates/provider-handler 1.0.0-rc.42 → 1.0.0-rc.44

package/dist/index.cjs

@@ -77,7 +77,8 @@ var validate = (schema, data, entity, message) => {
   }
   return result.data;
 };
-var toJsonSchema = (schema) => schema.toJSONSchema({
+var toJsonSchema = (schema, io = "output") => schema.toJSONSchema({
+  io,
   unrepresentable: "any",
   override: (ctx) => {
     let def = ctx.zodSchema._zod.def;
@@ -240,15 +241,30 @@ var getWebhookAutoRegistration = (group) => {
   return group.webhook.autoRegistration;
 };
 var getWebhookManualRegistration = (group) => {
-  if (group.source !== "webhook" || !group.webhook?.manualRegistration) {
+  let registration = group.source === "webhook" ? group.webhook?.manualRegistration ?? group.webhook?.deliveryConfig : void 0;
+  if (!registration) {
     throw new import_error2.ServiceError(
       (0, import_error2.badRequestError)({
         message: `Trigger group does not support manual webhook registration: ${group.key}`
       })
     );
   }
-  return group.webhook.manualRegistration;
+  return registration;
 };
+var getWebhookDeliveryConfig = (group) => {
+  if (group.source !== "webhook" || !group.webhook?.deliveryConfig) {
+    throw new import_error2.ServiceError(
+      (0, import_error2.badRequestError)({
+        message: `Trigger group does not support webhook delivery authentication: ${group.key}`
+      })
+    );
+  }
+  return group.webhook.deliveryConfig;
+};
+var mapWebhookConfigSchemas = (config) => ({
+  userConfigSchema: toJsonSchema(config.userConfigSchema),
+  fullConfigSchema: toJsonSchema(config.fullConfigSchema)
+});
 var mapToolCategory = (category) => ({
   id: category.key,
   name: category.name,
@@ -268,20 +284,24 @@ var mapTriggerGroup = (group) => ({
   name: group.name,
   description: group.description,
   metadata: group.metadata,
+  ...group.selectionSchema ? { selectionSchema: toJsonSchema(group.selectionSchema, "input") } : {},
   invocation: group.source === "polling" ? {
     type: "polling",
     intervalSeconds: group.polling?.intervalSeconds ?? import_provider.SlateDefaultPollingIntervalSeconds
   } : group.source === "gateway" ? { type: "gateway" } : {
     type: "webhook",
+    ...group.webhook?.expandWebhook ? { supportsExpansion: true } : {},
+    ...group.webhook?.autoRegistration?.webhookRenew ? { supportsRenewal: true } : {},
+    ...group.webhook?.reconcileIntervalSeconds !== void 0 ? { reconcileIntervalSeconds: group.webhook.reconcileIntervalSeconds } : {},
     registration: group.webhook?.manualRegistration ? {
       mode: "manual",
-      userConfigSchema: toJsonSchema(
-        group.webhook.manualRegistration.userConfigSchema
-      ),
-      fullConfigSchema: toJsonSchema(
-        group.webhook.manualRegistration.fullConfigSchema
-      )
-    } : { mode: "auto" }
+      ...mapWebhookConfigSchemas(group.webhook.manualRegistration)
+    } : {
+      mode: "auto",
+      ...group.webhook?.deliveryConfig ? {
+        deliveryConfig: mapWebhookConfigSchemas(group.webhook.deliveryConfig)
+      } : {}
+    }
   }
 });
 
@@ -358,6 +378,23 @@ var serializeWebhookHttpResponse = async (response) => {
 
 // src/index.ts
 var DEFAULT_MAX_ATTACHMENT_SIZE_BYTES = 1e8;
+var validateTriggerGroupSelection = (group, selection) => group.selectionSchema ? validate(
+  group.selectionSchema,
+  selection === void 0 ? {} : selection,
+  "trigger_group.selection",
+  `Invalid selection for trigger group "${group.key}"`
+) : selection ?? {};
+var validateWebhookRegistration = (result) => validate(
+  import_proto.slatesTriggerGroupWebhookRegistrationResult,
+  result,
+  "trigger_group.webhook.registration",
+  "Webhook registration returned an invalid result"
+);
+var toWebhookRequest = (params) => new Request(params.url, {
+  method: params.method,
+  headers: params.headers,
+  body: params.body ? Uint8Array.from(atob(params.body.content), (c) => c.charCodeAt(0)) : null
+});
 var routeAttachmentsThroughDirectUpload = async (attachments, live) => {
   if (!attachments || attachments.length === 0 || !live) return attachments;
   let contentEntries = attachments.map((attachment, index) => ({ attachment, index })).filter((entry) => entry.attachment.content.type === "content");
@@ -454,6 +491,7 @@ var createProviderHandler = (slate, listeners) => (0, import_proto.createSlatesP
   let liveInvocation = new State(
     null
   );
+  let runtimeInfo = new State(null);
   let getLiveInvocation = () => {
     let directUpload = hubCapabilities.get()?.attachments?.directUpload;
     let live = liveInvocation.get();
@@ -543,8 +581,17 @@ var createProviderHandler = (slate, listeners) => (0, import_proto.createSlatesP
     };
   };
   let getAuthConfig = () => config.get()?.value ?? {};
-  let getEmptyContext = () => new import_provider2.SlateContext({}, {}, {}, slate.spec, logger);
-  let getAuthContext = () => new import_provider2.SlateContext(getAuthConfig(), {}, {}, slate.spec, logger);
+  let createContext = (config2, input, auth2, liveInvocationInfo = null) => new import_provider2.SlateContext(
+    config2,
+    input,
+    auth2,
+    slate.spec,
+    logger,
+    liveInvocationInfo,
+    runtimeInfo.get()
+  );
+  let getEmptyContext = () => createContext({}, {}, {});
+  let getAuthContext = () => createContext(getAuthConfig(), {}, {});
   let withRequestTraces = (context, result) => {
     let requestTraces = context.getHttpTraces();
     return requestTraces.length > 0 ? { ...result, requestTraces } : result;
@@ -651,11 +698,15 @@ var createProviderHandler = (slate, listeners) => (0, import_proto.createSlatesP
       baseUrl: params.baseUrl
     });
   });
+  manager.onNotification("slates/hub.runtime_info.set", async ({ params }) => {
+    runtimeInfo.set(params.runtimeInfo);
+  });
   manager.onRequest("slates/provider.capabilities.get", async () => ({
     capabilities: {
       hub: {
         capabilitiesNotification: true,
-        liveInvocation: true
+        liveInvocation: true,
+        runtimeInfo: true
       },
       provider: {
         triggerGroups: (slate.triggerGroups?.length ?? 0) > 0 && (slate.actions ?? []).every(isMappableTrigger),
@@ -1214,19 +1265,18 @@ var createProviderHandler = (slate, listeners) => (0, import_proto.createSlatesP
     };
     if (action.isPublic) {
       getContextBasic();
-      return invoke(new import_provider2.SlatePublicContext(input, slate.spec, logger, getLiveInvocation()));
+      return invoke(
+        new import_provider2.SlatePublicContext(
+          input,
+          slate.spec,
+          logger,
+          getLiveInvocation(),
+          runtimeInfo.get()
+        )
+      );
     }
     let ctx = getContextFull();
-    return invoke(
-      new import_provider2.SlateContext(
-        ctx.config,
-        input,
-        ctx.auth?.output,
-        slate.spec,
-        logger,
-        getLiveInvocation()
-      )
-    );
+    return invoke(createContext(ctx.config, input, ctx.auth?.output, getLiveInvocation()));
   });
   manager.onRequest("slates/action.trigger.map_event", async ({ params }) => {
     let ctx = getContextFull();
@@ -1237,7 +1287,7 @@ var createProviderHandler = (slate, listeners) => (0, import_proto.createSlatesP
       "input",
       `Invalid event for trigger ID: ${params.actionId}`
     );
-    let context = new import_provider2.SlateContext(ctx.config, input, ctx.auth?.output, slate.spec, logger);
+    let context = createContext(ctx.config, input, ctx.auth?.output);
     let res = await traceProviderCall(
       {
         component: "action",
@@ -1309,12 +1359,13 @@ var createProviderHandler = (slate, listeners) => (0, import_proto.createSlatesP
     let ctx = getContextFull();
     let group = getTriggerGroup(slate, params.triggerGroupId);
     let autoRegistration = getWebhookAutoRegistration(group);
-    let context = new import_provider2.SlateContext(
+    let context = createContext(
       ctx.config,
-      { pageToken: params.pageToken ?? null },
-      ctx.auth?.output,
-      slate.spec,
-      logger
+      {
+        pageToken: params.pageToken ?? null,
+        selection: validateTriggerGroupSelection(group, params.selection)
+      },
+      ctx.auth?.output
     );
     let res = await traceProviderCall(
       {
@@ -1344,16 +1395,17 @@ var createProviderHandler = (slate, listeners) => (0, import_proto.createSlatesP
     let ctx = getContextFull();
     let group = getTriggerGroup(slate, params.triggerGroupId);
     let autoRegistration = getWebhookAutoRegistration(group);
-    let context = new import_provider2.SlateContext(
+    let context = createContext(
       ctx.config,
       {
         webhookTargetIdentifier: params.webhookTargetIdentifier,
         webhookTargetPayload: params.webhookTargetPayload,
-        webhookUrl: params.webhookUrl
+        webhookUrl: params.webhookUrl,
+        operationId: params.operationId,
+        selection: validateTriggerGroupSelection(group, params.selection),
+        deliveryConfig: params.deliveryConfig
       },
-      ctx.auth?.output,
-      slate.spec,
-      logger
+      ctx.auth?.output
     );
     let res = await traceProviderCall(
       {
@@ -1368,26 +1420,66 @@ var createProviderHandler = (slate, listeners) => (0, import_proto.createSlatesP
           webhookTargetIdentifier: params.webhookTargetIdentifier
         }
       },
-      () => (0, import_provider2.runWithContext)(context, () => autoRegistration.webhookRegister(context))
+      () => (0, import_provider2.runWithContext)(
+        context,
+        async () => validateWebhookRegistration(await autoRegistration.webhookRegister(context))
+      )
     );
-    return withRequestTraces(context, {
-      webhookRegistrationIdentifier: res.webhookRegistrationIdentifier,
-      webhookRegistrationPayload: res.webhookRegistrationPayload
-    });
+    return withRequestTraces(context, res);
+  });
+  manager.onRequest("slates/trigger_group.webhook.renew", async ({ params }) => {
+    let ctx = getContextFull();
+    let group = getTriggerGroup(slate, params.triggerGroupId);
+    let autoRegistration = getWebhookAutoRegistration(group);
+    let renew = autoRegistration.webhookRenew;
+    if (!renew) {
+      throw new import_error4.ServiceError(
+        (0, import_error4.badRequestError)({
+          message: `Trigger group does not support webhook renewal: ${params.triggerGroupId}`
+        })
+      );
+    }
+    let context = createContext(
+      ctx.config,
+      {
+        webhookRegistrationIdentifier: params.webhookRegistrationIdentifier,
+        webhookRegistrationPayload: params.webhookRegistrationPayload,
+        webhookUrl: params.webhookUrl,
+        webhookTargetIdentifier: params.webhookTargetIdentifier,
+        webhookTargetPayload: params.webhookTargetPayload,
+        operationId: params.operationId,
+        selection: validateTriggerGroupSelection(group, params.selection),
+        deliveryConfig: params.deliveryConfig
+      },
+      ctx.auth?.output
+    );
+    let res = await traceProviderCall(
+      {
+        component: "action",
+        functionName: "webhookRenew",
+        message: `Renewing webhook for trigger group ${formatEntityLabel(group.name, group.key)}`,
+        successMessage: `Renewed webhook for trigger group ${formatEntityLabel(group.name, group.key)}`,
+        errorMessage: `Trigger group ${formatEntityLabel(group.name, group.key)} failed while renewing a webhook`,
+        metadata: { triggerGroupId: group.key, triggerGroupName: group.name }
+      },
+      () => (0, import_provider2.runWithContext)(
+        context,
+        async () => validateWebhookRegistration(await renew(context))
+      )
+    );
+    return withRequestTraces(context, res);
   });
   manager.onRequest("slates/trigger_group.webhook.unregister", async ({ params }) => {
     let ctx = getContextFull();
     let group = getTriggerGroup(slate, params.triggerGroupId);
     let autoRegistration = getWebhookAutoRegistration(group);
-    let context = new import_provider2.SlateContext(
+    let context = createContext(
       ctx.config,
       {
         webhookRegistrationIdentifier: params.webhookRegistrationIdentifier,
         webhookRegistrationPayload: params.webhookRegistrationPayload
       },
-      ctx.auth?.output,
-      slate.spec,
-      logger
+      ctx.auth?.output
     );
     await traceProviderCall(
       {
@@ -1410,13 +1502,7 @@ var createProviderHandler = (slate, listeners) => (0, import_proto.createSlatesP
     getContextBasic();
     let group = getTriggerGroup(slate, params.triggerGroupId);
     let manualRegistration = getWebhookManualRegistration(group);
-    let context = new import_provider2.SlateContext(
-      {},
-      { webhookUrl: params.webhookUrl },
-      {},
-      slate.spec,
-      logger
-    );
+    let context = createContext({}, { webhookUrl: params.webhookUrl }, {});
     let res = await traceProviderCall(
       {
         component: "action",
@@ -1447,16 +1533,14 @@ var createProviderHandler = (slate, listeners) => (0, import_proto.createSlatesP
         })
       );
     }
-    let context = new import_provider2.SlateContext(
+    let context = createContext(
       {},
       {
         webhookUrl: params.webhookUrl,
         partialWebhookRegistrationPayload: params.partialWebhookRegistrationPayload,
         userWebhookRegistrationPayload: params.userWebhookRegistrationPayload
       },
-      {},
-      slate.spec,
-      logger
+      {}
     );
     let res = await traceProviderCall(
       {
@@ -1485,20 +1569,13 @@ var createProviderHandler = (slate, listeners) => (0, import_proto.createSlatesP
         })
       );
     }
-    let req = new Request(params.url, {
-      method: params.method,
-      headers: params.headers,
-      body: params.body ? Uint8Array.from(atob(params.body.content), (c) => c.charCodeAt(0)) : null
-    });
-    let context = new import_provider2.SlateContext(
+    let context = createContext(
       {},
       {
-        request: req,
+        request: toWebhookRequest(params),
         webhookRegistrationPayload: params.webhookRegistrationPayload
       },
-      {},
-      slate.spec,
-      logger
+      {}
     );
     let res = await traceProviderCall(
       {
@@ -1536,16 +1613,110 @@ var createProviderHandler = (slate, listeners) => (0, import_proto.createSlatesP
         matchers: event.matchers,
         payload: event.payload,
         idempotencyKey: event.idempotencyKey,
-        triggerIds: evaluateTriggerMatches(slate, group.key, event.payload)
+        triggerIds: group.webhook?.expandWebhook ? [] : evaluateTriggerMatches(slate, group.key, event.payload)
       })),
+      ...group.webhook?.expandWebhook ? { requiresExpansion: true } : {},
       response,
       skipped: res.skipped ?? null
     });
   });
+  manager.onRequest(
+    "slates/trigger_group.webhook.delivery_authenticate",
+    async ({ params }) => {
+      let group = getTriggerGroup(slate, params.triggerGroupId);
+      let deliveryConfig = getWebhookDeliveryConfig(group);
+      let context = createContext(
+        {},
+        {
+          request: toWebhookRequest(params),
+          deliveryConfig: params.deliveryConfig
+        },
+        {}
+      );
+      let res = await traceProviderCall(
+        {
+          component: "action",
+          functionName: "webhookDeliveryAuthenticate",
+          message: `Authenticating a delivery for trigger group ${formatEntityLabel(group.name, group.key)}`,
+          successMessage: (result) => result.rejected ? `Rejected a delivery with status ${result.rejected.status} for trigger group ${formatEntityLabel(group.name, group.key)}` : `Authenticated a delivery for ${result.routingKeys.length} routing key(s) for trigger group ${formatEntityLabel(group.name, group.key)}`,
+          errorMessage: `Trigger group ${formatEntityLabel(group.name, group.key)} failed while authenticating a delivery`,
+          metadata: {
+            triggerGroupId: group.key,
+            triggerGroupName: group.name,
+            requestMethod: params.method,
+            hasRequestBody: !!params.body
+          },
+          onSuccess: (result) => ({
+            routingKeyCount: result.routingKeys.length,
+            rejectedStatus: result.rejected?.status ?? null
+          })
+        },
+        () => (0, import_provider2.runWithContext)(context, async () => {
+          let authentication = await deliveryConfig.authenticate(context);
+          return validate(
+            import_proto.slatesTriggerGroupWebhookDeliveryAuthentication,
+            authentication.rejected ? { routingKeys: [], rejected: authentication.rejected } : { routingKeys: authentication.routingKeys, rejected: null },
+            "trigger_group.webhook.delivery_authentication",
+            "Webhook delivery authentication returned an invalid result"
+          );
+        })
+      );
+      return withRequestTraces(context, res);
+    }
+  );
+  manager.onRequest("slates/trigger_group.webhook.expand", async ({ params }) => {
+    let ctx = getContextFull();
+    let group = getTriggerGroup(slate, params.triggerGroupId);
+    let expand = group.source === "webhook" ? group.webhook?.expandWebhook : void 0;
+    if (!expand) {
+      throw new import_error4.ServiceError(
+        (0, import_error4.badRequestError)({
+          message: `Trigger group does not support webhook expansion: ${params.triggerGroupId}`
+        })
+      );
+    }
+    let context = createContext(
+      ctx.config,
+      {
+        payload: params.payload,
+        state: params.state ?? null,
+        selection: validateTriggerGroupSelection(group, params.selection),
+        deliveryId: params.deliveryId,
+        webhookRegistrationPayload: params.webhookRegistrationPayload
+      },
+      ctx.auth?.output
+    );
+    let res = await traceProviderCall(
+      {
+        component: "action",
+        functionName: "expandWebhook",
+        message: `Expanding webhook delivery for trigger group ${formatEntityLabel(group.name, group.key)}`,
+        successMessage: `Expanded webhook delivery for trigger group ${formatEntityLabel(group.name, group.key)}`,
+        errorMessage: `Trigger group ${formatEntityLabel(group.name, group.key)} failed while expanding a webhook delivery`,
+        metadata: { triggerGroupId: group.key, triggerGroupName: group.name }
+      },
+      () => (0, import_provider2.runWithContext)(
+        context,
+        async () => validate(
+          import_proto.slatesTriggerGroupWebhookExpansionResult,
+          await expand(context),
+          "trigger_group.webhook.expansion",
+          "Webhook expansion returned an invalid result"
+        )
+      )
+    );
+    return withRequestTraces(context, {
+      ...res.updatedState !== void 0 ? { updatedState: res.updatedState } : {},
+      events: res.events.map((event) => ({
+        ...event,
+        triggerIds: evaluateTriggerMatches(slate, group.key, event.payload)
+      }))
+    });
+  });
   manager.onRequest("slates/trigger_group.routing_matchers.get", async ({ params }) => {
     let ctx = getContextFull();
     let group = getTriggerGroup(slate, params.triggerGroupId);
-    let context = new import_provider2.SlateContext(ctx.config, {}, ctx.auth?.output, slate.spec, logger);
+    let context = createContext(ctx.config, {}, ctx.auth?.output);
     let matchers = await traceProviderCall(
       {
         component: "action",
@@ -1575,12 +1746,13 @@ var createProviderHandler = (slate, listeners) => (0, import_proto.createSlatesP
         })
       );
     }
-    let context = new import_provider2.SlateContext(
+    let context = createContext(
       ctx.config,
-      { state: params.state },
-      ctx.auth?.output,
-      slate.spec,
-      logger
+      {
+        state: params.state,
+        selection: validateTriggerGroupSelection(group, params.selection)
+      },
+      ctx.auth?.output
     );
     let res = await traceProviderCall(
       {
@@ -1624,13 +1796,7 @@ var createProviderHandler = (slate, listeners) => (0, import_proto.createSlatesP
   manager.onRequest("slates/trigger_group.gateway.connect", async ({ params }) => {
     let ctx = getContextFull();
     let { group, gateway } = getGatewayTriggerGroup(params.triggerGroupId);
-    let context = new import_provider2.SlateContext(
-      ctx.config,
-      { state: params.state },
-      ctx.auth?.output,
-      slate.spec,
-      logger
-    );
+    let context = createContext(ctx.config, { state: params.state }, ctx.auth?.output);
     let res = await traceProviderCall(
       {
         component: "action",
@@ -1651,12 +1817,10 @@ var createProviderHandler = (slate, listeners) => (0, import_proto.createSlatesP
   manager.onRequest("slates/trigger_group.gateway.receive", async ({ params }) => {
     let ctx = getContextFull();
     let { group, gateway } = getGatewayTriggerGroup(params.triggerGroupId);
-    let context = new import_provider2.SlateContext(
+    let context = createContext(
       ctx.config,
       { state: params.state, frames: params.frames, closed: params.closed ?? null },
-      ctx.auth?.output,
-      slate.spec,
-      logger
+      ctx.auth?.output
     );
     let res = await traceProviderCall(
       {