Merge pull request #279 from pantheon-systems/ag/pcc-1210/support-for…

…-filtering-sent-webhooks-by-event [PCC-1210] Add support for setting preferred webhook events
pantheon-systems · Jun 24, 2024 · ef28b57 · ef28b57
2 parents 2b3f273 + 9d1248e
commit ef28b57
Show file tree

Hide file tree

Showing 5 changed files with 80 additions and 5 deletions.
diff --git a/.changeset/tame-mails-own.md b/.changeset/tame-mails-own.md
@@ -0,0 +1,5 @@
+---
+"@pantheon-systems/pcc-cli": patch
+---
+
+Added support for setting preferred webhook events. Webhook notifications will only be sent on events matching preferred events
diff --git a/packages/cli/src/cli/commands/sites/webhooks.ts b/packages/cli/src/cli/commands/sites/webhooks.ts
@@ -0,0 +1,38 @@
+import inquirer from "inquirer";
+import ora from "ora";
+import AddOnApiHelper from "../../../lib/addonApiHelper";
+import { errorHandler } from "../../exceptions";
+
+async function configurePreferredWebhookEvents(siteId: string) {
+  // Fetch available events
+  const availableEventsSpinner = ora("Fetching available events...").start();
+  const availableEvents =
+    await AddOnApiHelper.fetchAvailableWebhookEvents(siteId);
+  availableEventsSpinner.succeed("Fetched available events");
+
+  // Prompt user to select events
+  const { selectedEvents } = await inquirer.prompt([
+    {
+      type: "checkbox",
+      name: "selectedEvents",
+      message:
+        "Select events to receive notifications for. Select none to receive notifications for all events.",
+      choices: availableEvents,
+    },
+  ]);
+
+  if (selectedEvents.length === 0) {
+    console.info(
+      "No events selected. Your webhook will receive notifications for all events.",
+    );
+  }
+
+  // Update events for the site
+  const updateEventsSpinner = ora("Updating preferred events...").start();
+  await AddOnApiHelper.updateSiteConfig(siteId, {
+    preferredEvents: selectedEvents,
+  });
+  updateEventsSpinner.succeed("Updated preferred events");
+}
+
+export default errorHandler(configurePreferredWebhookEvents);
diff --git a/packages/cli/src/cli/exceptions.ts b/packages/cli/src/cli/exceptions.ts
@@ -32,24 +32,24 @@ export function errorHandler<T>(
       if (cleanup) cleanup(arg);
 
       if (e instanceof UserNotLoggedIn) {
-        console.log(chalk.red("Error: User is not logged in."));
-        console.log(chalk.yellow('Please run "pcc login" to login.'));
+        console.log(chalk.red("\nError: User is not logged in."));
+        console.log(chalk.yellow('\nPlease run "pcc login" to login.'));
       } else {
         if (
           axios.isAxiosError(e) &&
           (e.response?.status ?? 500) < 500 && // Treat internal server errors as unhandled errors
           e.response?.data?.message
         ) {
           // Operational error
-          console.log(chalk.red(`Error: ${e.response.data.message}`));
+          console.log(chalk.red(`\nError: ${e.response.data.message}`));
         } else {
           // Unhandled error
           console.log(
             chalk.yellow("\nStack trace:", (e as { stack: string }).stack),
           );
           console.log(
             chalk.red(
-              "Error: Something went wrong. Please contact Pantheon support team.",
+              "\nError: Something went wrong. Please contact Pantheon support team.",
             ),
           );
         }

diff --git a/packages/cli/src/cli/index.ts b/packages/cli/src/cli/index.ts
@@ -34,6 +34,7 @@ import {
   SITE_EXAMPLES,
   updateSiteConfig,
 } from "./commands/sites/site";
+import configurePreferredWebhookEvents from "./commands/sites/webhooks";
 import {
   createToken,
   listTokens,
@@ -552,6 +553,19 @@ yargs(hideBin(process.argv))
                     id: args.id as string,
                     limit: args.limit as number,
                   }),
+              )
+              .command(
+                "preferred-events <id>",
+                "Set preferred webhook events for a given site. Your webhook will only receive notifications for events that you specify.",
+                (yargs) => {
+                  yargs.strictCommands().positional("<id>", {
+                    describe: "ID of the site for which you want to configure.",
+                    demandOption: true,
+                    type: "string",
+                  });
+                },
+                async (args) =>
+                  configurePreferredWebhookEvents(String(args.id)),
               );
           },
         )

diff --git a/packages/cli/src/lib/addonApiHelper.ts b/packages/cli/src/lib/addonApiHelper.ts
@@ -430,15 +430,17 @@ class AddOnApiHelper {
       url,
       webhookUrl,
       webhookSecret,
+      preferredEvents,
     }: {
       url?: string;
       webhookUrl?: string;
       webhookSecret?: string;
+      preferredEvents?: string[];
     },
   ): Promise<void> {
     const idToken = await this.getIdToken();
 
-    const configuredWebhook = webhookUrl || webhookSecret;
+    const configuredWebhook = webhookUrl || webhookSecret || preferredEvents;
 
     await axios.patch(
       `${(await getApiConfig()).SITE_ENDPOINT}/${id}`,
@@ -448,6 +450,7 @@ class AddOnApiHelper {
           webhookConfig: {
             ...(webhookUrl && { webhookUrl: webhookUrl }),
             ...(webhookSecret && { webhookSecret: webhookSecret }),
+            ...(preferredEvents && { preferredEvents }),
           },
         }),
       },
@@ -486,6 +489,21 @@ class AddOnApiHelper {
 
     return resp.data as WebhookDeliveryLog[];
   }
+
+  static async fetchAvailableWebhookEvents(siteId: string) {
+    const idToken = await this.getIdToken();
+
+    const resp = await axios.get(
+      `${(await getApiConfig()).SITE_ENDPOINT}/${siteId}/availableWebhookEvents`,
+      {
+        headers: {
+          Authorization: `Bearer ${idToken}`,
+        },
+      },
+    );
+
+    return resp.data as string[];
+  }
 }
 
 export default AddOnApiHelper;