CHI-3660: Migrate adjustChatCapacity and transferStart

lambdas/account-scoped/src/conversation/adjustChatCapacity.ts

@@ -0,0 +1,143 @@
+/**
+ * Copyright (C) 2021-2023 Technology Matters
+ * This program is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU Affero General Public License as published
+ * by the Free Software Foundation, either version 3 of the License, or
+ * (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU Affero General Public License for more details.
+ *
+ * You should have received a copy of the GNU Affero General Public License
+ * along with this program.  If not, see https://www.gnu.org/licenses/.
+ */
+
+import '@twilio-labs/serverless-runtime-types';
+import { WorkerChannelInstance } from 'twilio/lib/rest/taskrouter/v1/workspace/worker/workerChannel';
+import { AccountScopedHandler, HttpRequest } from '../httpTypes';
+import { newMissingParameterResult } from '../httpErrors';
+import type { AccountSID, WorkerSID } from '@tech-matters/twilio-types';
+import { getTwilioClient, getWorkspaceSid } from '@tech-matters/twilio-configuration';
+import { newOk } from '../Result';
+
+export type AdjustChatCapacityParameters = {
+  workerSid: WorkerSID;
+  adjustment: 'increaseUntilCapacityAvailable' | 'setTo1';
+};
+
+const increaseChatCapacity = async (
+  channel: WorkerChannelInstance,
+  maxMessageCapacity: number,
+) => {
+  if (channel.availableCapacityPercentage > 0) {
+    return {
+      result: {
+        status: 412,
+        message: 'Still have available capacity, no need to increase.',
+      },
+      updatedChannel: channel,
+    };
+  }
+
+  if (!(channel.configuredCapacity < maxMessageCapacity)) {
+    return {
+      result: { status: 412, message: 'Reached the max capacity.' },
+      updatedChannel: channel,
+    };
+  }
+
+  const updatedChannel = await channel.update({
+    capacity: channel.configuredCapacity + 1,
+  });
+  return {
+    result: { status: 200, message: 'Successfully increased channel capacity' },
+    updatedChannel,
+  };
+};
+
+export const adjustChatCapacity = async (
+  accountSid: AccountSID,
+  body: AdjustChatCapacityParameters,
+): Promise<{ status: number; message: string }> => {
+  const client = await getTwilioClient(accountSid);
+
+  const worker = await client.taskrouter.v1.workspaces
+    .get(await getWorkspaceSid(accountSid))
+    .workers(body.workerSid)
+    .fetch();
+
+  if (!worker) return { status: 404, message: 'Could not find worker.' };
+
+  const attributes = JSON.parse(worker.attributes);
+  const maxMessageCapacity = parseInt(attributes.maxMessageCapacity, 10);
+
+  if (!maxMessageCapacity) {
+    return {
+      status: 409,
+      message: `Worker ${body.workerSid} does not have a "maxMessageCapacity" attribute, can't adjust capacity.`,
+    };
+  }
+
+  const channels = await worker.workerChannels().list();
+  let channel = channels.find(c => c.taskChannelUniqueName === 'chat');
+
+  if (!channel) return { status: 404, message: 'Could not find chat channel.' };
+
+  if (body.adjustment === 'increaseUntilCapacityAvailable') {
+    let result: Awaited<ReturnType<typeof increaseChatCapacity>>['result'] = {
+      status: 200,
+      message: '',
+    };
+    while (result.status === 200) {
+      // eslint-disable-next-line no-await-in-loop
+      ({ result, updatedChannel: channel } = await increaseChatCapacity(
+        channel,
+        maxMessageCapacity,
+      ));
+    }
+    if (
+      channel.configuredCapacity === maxMessageCapacity &&
+      channel.availableCapacityPercentage === 0
+    ) {
+      return {
+        status: 412,
+        message: 'Reached the max capacity with no available capacity.',
+      };
+    }
+    return {
+      status: 200,
+      message: 'Adjusted chat capacity until there is capacity available',
+    };
+  }
+
+  if (body.adjustment === 'setTo1') {
+    if (channel.configuredCapacity !== 1) {
+      await channel.update({ capacity: 1 });
+      // If configuredCapacity is already 1, send status 200 to avoid error on client side
+      return { status: 200, message: 'Successfully reset channel capacity to 1' };
+    }
+
+    // If configuredCapacity is already 1, send status 200 to avoid error on client side
+    return { status: 200, message: 'Channel capacity already 1, no adjustment made.' };
+  }
+
+  return { status: 400, message: 'Invalid adjustment argument' };
+};
+
+export const adjustChatCapacityHandler: AccountScopedHandler = async (
+  { body: event }: HttpRequest,
+  accountSid: AccountSID,
+) => {
+  const { workerSid, adjustment } = event;
+
+  if (workerSid === undefined) return newMissingParameterResult('workerSid');
+  if (adjustment === undefined) return newMissingParameterResult('adjustment');
+
+  const validBody = { workerSid, adjustment };
+
+  const { status, message } = await adjustChatCapacity(accountSid, validBody);
+
+  return newOk({ message, status });
+};

lambdas/account-scoped/src/router.ts

@@ -50,6 +50,8 @@ import { initWebchatHandler } from './webchatAuthentication/initWebchat';
 import { refreshTokenHandler } from './webchatAuthentication/refreshToken';
 import { getAccountSid } from '@tech-matters/twilio-configuration';
 import { validateRequestWithTwilioJwtToken } from './validation/twilioJwt';
+import { transferStartHandler } from './transfer/transferStart';
+import { adjustChatCapacityHandler } from './conversation/adjustChatCapacity';
 
 /**
  * Super simple router sufficient for directly ported Twilio Serverless functions
@@ -115,6 +117,10 @@ const ACCOUNTSID_ROUTES: Record<
     requestPipeline: [validateWebhookRequest],
     handler: handleConversationEvent,
   },
+  'conversation/adjustChatCapacity': {
+    requestPipeline: [validateFlexTokenRequest({ tokenMode: 'worker' })],
+    handler: adjustChatCapacityHandler,
+  },
   'customChannels/instagram/instagramToFlex': {
     requestPipeline: [],
     handler: instagramToFlexHandler,
@@ -155,6 +161,10 @@ const ACCOUNTSID_ROUTES: Record<
     requestPipeline: [validateFlexTokenRequest({ tokenMode: 'worker' })],
     handler: getTaskAndReservationsHandler,
   },
+  'transfer/transferStart': {
+    requestPipeline: [validateFlexTokenRequest({ tokenMode: 'worker' })],
+    handler: transferStartHandler,
+  },
   updateWorkersSkills: {
     requestPipeline: [validateFlexTokenRequest({ tokenMode: 'supervisor' })],
     handler: handleUpdateWorkersSkills,