Add muting support for alerts

x-pack/legacy/plugins/alerting/mappings.json

@@ -46,6 +46,12 @@
       },
       "throttle": {
         "type": "keyword"
+      },
+      "muteAll": {
+        "type": "boolean"
+      },
+      "mutedInstanceIds": {
+        "type": "keyword"
       }
     }
   }

x-pack/legacy/plugins/alerting/server/alerts_client.mock.ts

@@ -18,6 +18,10 @@ const createAlertsClientMock = () => {
     enable: jest.fn(),
     disable: jest.fn(),
     updateApiKey: jest.fn(),
+    muteAll: jest.fn(),
+    unmuteAll: jest.fn(),
+    muteInstance: jest.fn(),
+    unmuteInstance: jest.fn(),
   };
   return mocked;
 };

x-pack/legacy/plugins/alerting/server/alerts_client.ts

@@ -56,7 +56,13 @@ interface FindResult {
 }
 
 interface CreateOptions {
-  data: Pick<Alert, Exclude<keyof Alert, 'createdBy' | 'updatedBy' | 'apiKey' | 'apiKeyOwner'>>;
+  data: Pick<
+    Alert,
+    Exclude<
+      keyof Alert,
+      'createdBy' | 'updatedBy' | 'apiKey' | 'apiKeyOwner' | 'muteAll' | 'mutedInstanceIds'
+    >
+  >;
   options?: {
     migrationVersion?: Record<string, string>;
   };
@@ -69,7 +75,6 @@ interface UpdateOptions {
     actions: AlertAction[];
     alertTypeParams: Record<string, any>;
   };
-  options?: { version?: string };
 }
 
 export class AlertsClient {
@@ -117,6 +122,8 @@ export class AlertsClient {
         ? Buffer.from(`${apiKey.result.id}:${apiKey.result.api_key}`).toString('base64')
         : undefined,
       alertTypeParams: validatedAlertTypeParams,
+      muteAll: false,
+      mutedInstanceIds: [],
     });
     const createdAlert = await this.savedObjectsClient.create('alert', rawAlert, {
       ...options,
@@ -188,7 +195,7 @@ export class AlertsClient {
     return removeResult;
   }
 
-  public async update({ id, data, options = {} }: UpdateOptions) {
+  public async update({ id, data }: UpdateOptions) {
     const existingObject = await this.savedObjectsClient.get('alert', id);
     const { alertTypeId } = existingObject.attributes;
     const alertType = this.alertTypeRegistry.get(alertTypeId);
@@ -204,6 +211,7 @@ export class AlertsClient {
       'alert',
       id,
       {
+        ...existingObject.attributes,
         ...data,
         alertTypeParams: validatedAlertTypeParams,
         actions,
@@ -214,30 +222,32 @@ export class AlertsClient {
           : null,
       },
       {
-        ...options,
         references,
+        version: existingObject.version,
       }
     );
     return this.getAlertFromRaw(id, updatedObject.attributes, updatedObject.references);
   }
 
   public async updateApiKey({ id }: { id: string }) {
-    const { references } = await this.savedObjectsClient.get('alert', id);
+    const existingObject = await this.savedObjectsClient.get('alert', id);
 
     const apiKey = await this.createAPIKey();
     const username = await this.getUserName();
     await this.savedObjectsClient.update(
       'alert',
       id,
       {
+        ...existingObject.attributes,
         updatedBy: username,
         apiKeyOwner: apiKey.created ? username : null,
         apiKey: apiKey.created
           ? Buffer.from(`${apiKey.result.id}:${apiKey.result.api_key}`).toString('base64')
           : null,
       },
       {
-        references,
+        version: existingObject.version,
+        references: existingObject.references,
       }
     );
   }
@@ -256,6 +266,7 @@ export class AlertsClient {
         'alert',
         id,
         {
+          ...existingObject.attributes,
           enabled: true,
           updatedBy: username,
           apiKeyOwner: apiKey.created ? username : null,
@@ -264,7 +275,10 @@ export class AlertsClient {
             ? Buffer.from(`${apiKey.result.id}:${apiKey.result.api_key}`).toString('base64')
             : null,
         },
-        { references: existingObject.references }
+        {
+          version: existingObject.version,
+          references: existingObject.references,
+        }
       );
     }
   }
@@ -276,18 +290,111 @@ export class AlertsClient {
         'alert',
         id,
         {
+          ...existingObject.attributes,
           enabled: false,
           scheduledTaskId: null,
           apiKey: null,
           apiKeyOwner: null,
           updatedBy: await this.getUserName(),
         },
-        { references: existingObject.references }
+        {
+          version: existingObject.version,
+          references: existingObject.references,
+        }
       );
       await this.taskManager.remove(existingObject.attributes.scheduledTaskId);
     }
   }
 
+  public async muteAll({ id }: { id: string }) {
+    const {
+      references,
+      attributes: { muteAll },
+    } = await this.savedObjectsClient.get('alert', id);
+    if (!muteAll) {
+      await this.savedObjectsClient.update(
+        'alert',
+        id,
+        {
+          muteAll: true,
+          mutedInstanceIds: [],
+          updatedBy: await this.getUserName(),
+        },
+        { references }
+      );
+    }
+  }
+
+  public async unmuteAll({ id }: { id: string }) {
+    const {
+      references,
+      attributes: { muteAll },
+    } = await this.savedObjectsClient.get('alert', id);
+    if (muteAll) {
+      await this.savedObjectsClient.update(
+        'alert',
+        id,
+        {
+          muteAll: false,
+          mutedInstanceIds: [],
+          updatedBy: await this.getUserName(),
+        },
+        { references }
+      );
+    }
+  }
+
+  public async muteInstance({
+    alertId,
+    alertInstanceId,
+  }: {
+    alertId: string;
+    alertInstanceId: string;
+  }) {
+    const existingObject = await this.savedObjectsClient.get('alert', alertId);
+    const mutedInstanceIds = existingObject.attributes.mutedInstanceIds || [];
+    if (!existingObject.attributes.muteAll && !mutedInstanceIds.includes(alertInstanceId)) {
+      mutedInstanceIds.push(alertInstanceId);
+      await this.savedObjectsClient.update(
+        'alert',
+        alertId,
+        {
+          mutedInstanceIds,
+          updatedBy: await this.getUserName(),
+        },
+        {
+          version: existingObject.version,
+          references: existingObject.references,
+        }
+      );
+    }
+  }
+
+  public async unmuteInstance({
+    alertId,
+    alertInstanceId,
+  }: {
+    alertId: string;
+    alertInstanceId: string;
+  }) {
+    const existingObject = await this.savedObjectsClient.get('alert', alertId);
+    const mutedInstanceIds = existingObject.attributes.mutedInstanceIds || [];
+    if (!existingObject.attributes.muteAll && mutedInstanceIds.includes(alertInstanceId)) {
+      await this.savedObjectsClient.update(
+        'alert',
+        alertId,
+        {
+          updatedBy: await this.getUserName(),
+          mutedInstanceIds: mutedInstanceIds.filter((id: string) => id !== alertInstanceId),
+        },
+        {
+          version: existingObject.version,
+          references: existingObject.references,
+        }
+      );
+    }
+  }
+
   private async scheduleAlert(id: string, alertTypeId: string, interval: string) {
     return await this.taskManager.schedule({
       taskType: `alerting:${alertTypeId}`,

x-pack/legacy/plugins/alerting/server/init.ts

@@ -28,6 +28,10 @@ import {
   enableAlertRoute,
   disableAlertRoute,
   updateApiKeyRoute,
+  muteAllAlertRoute,
+  unmuteAllAlertRoute,
+  muteAlertInstanceRoute,
+  unmuteAlertInstanceRoute,
 } from './routes';
 
 // Extend PluginProperties to indicate which plugins are guaranteed to exist
@@ -88,7 +92,12 @@ export function init(server: Server) {
   server.plugins.encrypted_saved_objects.registerType({
     type: 'alert',
     attributesToEncrypt: new Set(['apiKey']),
-    attributesToExcludeFromAAD: new Set(['scheduledTaskId']),
+    attributesToExcludeFromAAD: new Set([
+      'scheduledTaskId',
+      'muted',
+      'mutedInstanceIds',
+      'updatedBy',
+    ]),
   });
 
   function getServices(request: any): Services {
@@ -127,6 +136,10 @@ export function init(server: Server) {
   enableAlertRoute(server);
   disableAlertRoute(server);
   updateApiKeyRoute(server);
+  muteAllAlertRoute(server);
+  unmuteAllAlertRoute(server);
+  muteAlertInstanceRoute(server);
+  unmuteAlertInstanceRoute(server);
 
   // Expose functions
   server.decorate('request', 'getAlertsClient', function() {

x-pack/legacy/plugins/alerting/server/lib/get_create_task_runner_function.test.ts

@@ -68,6 +68,7 @@ const mockedAlertTypeSavedObject = {
     enabled: true,
     alertTypeId: '123',
     interval: '10s',
+    mutedInstanceIds: [],
     alertTypeParams: {
       bar: true,
     },

x-pack/legacy/plugins/alerting/server/lib/get_create_task_runner_function.ts

@@ -74,7 +74,7 @@ export function getCreateTaskRunnerFunction({
         const services = getServices(fakeRequest);
         // Ensure API key is still valid and user has access
         const {
-          attributes: { alertTypeParams, actions, interval, throttle },
+          attributes: { alertTypeParams, actions, interval, throttle, muteAll, mutedInstanceIds },
           references,
         } = await services.savedObjectsClient.get<RawAlert>('alert', alertId);
 
@@ -128,6 +128,9 @@ export function getCreateTaskRunnerFunction({
           Object.keys(alertInstances).map(alertInstanceId => {
             const alertInstance = alertInstances[alertInstanceId];
             if (alertInstance.hasScheduledActions(throttle)) {
+              if (muteAll || mutedInstanceIds.includes(alertInstanceId)) {
+                return;
+              }
               const { actionGroup, context, state } = alertInstance.getSechduledActionOptions()!;
               alertInstance.updateLastScheduledActions(actionGroup);
               alertInstance.unscheduleActions();

x-pack/legacy/plugins/alerting/server/routes/index.ts

@@ -13,3 +13,7 @@ export { updateAlertRoute } from './update';
 export { enableAlertRoute } from './enable';
 export { disableAlertRoute } from './disable';
 export { updateApiKeyRoute } from './update_api_key';
+export { muteAlertInstanceRoute } from './mute_instance';
+export { unmuteAlertInstanceRoute } from './unmute_instance';
+export { muteAllAlertRoute } from './mute_all';
+export { unmuteAllAlertRoute } from './unmute_all';

x-pack/legacy/plugins/alerting/server/routes/mute_all.test.ts

@@ -0,0 +1,22 @@
+/*
+ * Copyright Elasticsearch B.V. and/or licensed to Elasticsearch B.V. under one
+ * or more contributor license agreements. Licensed under the Elastic License;
+ * you may not use this file except in compliance with the Elastic License.
+ */
+
+import { createMockServer } from './_mock_server';
+import { muteAllAlertRoute } from './mute_all';
+
+const { server, alertsClient } = createMockServer();
+muteAllAlertRoute(server);
+
+test('mutes an alert', async () => {
+  const request = {
+    method: 'POST',
+    url: '/api/alert/1/_mute_all',
+  };
+
+  const { statusCode } = await server.inject(request);
+  expect(statusCode).toBe(204);
+  expect(alertsClient.muteAll).toHaveBeenCalledWith({ id: '1' });
+});

x-pack/legacy/plugins/alerting/server/routes/mute_all.ts

@@ -0,0 +1,31 @@
+/*
+ * Copyright Elasticsearch B.V. and/or licensed to Elasticsearch B.V. under one
+ * or more contributor license agreements. Licensed under the Elastic License;
+ * you may not use this file except in compliance with the Elastic License.
+ */
+
+import Hapi from 'hapi';
+
+interface MuteAllRequest extends Hapi.Request {
+  params: {
+    id: string;
+  };
+}
+
+export function muteAllAlertRoute(server: Hapi.Server) {
+  server.route({
+    method: 'POST',
+    path: '/api/alert/{id}/_mute_all',
+    options: {
+      tags: ['access:alerting-all'],
+      response: {
+        emptyStatusCode: 204,
+      },
+    },
+    async handler(request: MuteAllRequest, h: Hapi.ResponseToolkit) {
+      const alertsClient = request.getAlertsClient!();
+      await alertsClient.muteAll(request.params);
+      return h.response();
+    },
+  });
+}

x-pack/legacy/plugins/alerting/server/routes/mute_instance.test.ts

@@ -0,0 +1,22 @@
+/*
+ * Copyright Elasticsearch B.V. and/or licensed to Elasticsearch B.V. under one
+ * or more contributor license agreements. Licensed under the Elastic License;
+ * you may not use this file except in compliance with the Elastic License.
+ */
+
+import { createMockServer } from './_mock_server';
+import { muteAlertInstanceRoute } from './mute_instance';
+
+const { server, alertsClient } = createMockServer();
+muteAlertInstanceRoute(server);
+
+test('mutes an alert instance', async () => {
+  const request = {
+    method: 'POST',
+    url: '/api/alert/1/alert_instance/2/_mute',
+  };
+
+  const { statusCode } = await server.inject(request);
+  expect(statusCode).toBe(204);
+  expect(alertsClient.muteInstance).toHaveBeenCalledWith({ alertId: '1', alertInstanceId: '2' });
+});