feat: add custom variables

chatwoot · Dec 7, 2023 · 9034696 · 9034696
1 parent c413ba9
commit 9034696
Show file tree

Hide file tree

Showing 3 changed files with 44 additions and 2 deletions.
diff --git a/src/canned.ts b/src/canned.ts
@@ -1,4 +1,9 @@
-import { Conversation, Sender, Variables } from './types/conversation';
+import {
+  Conversation,
+  Sender,
+  Variables,
+  CustomAttributes,
+} from './types/conversation';
 const MESSAGE_VARIABLES_REGEX = /{{(.*?)}}/g;
 
 const skipCodeBlocks = (str: string) => str.replace(/```(?:.|\n)+?```/g, '');
@@ -29,9 +34,11 @@ export const getMessageVariables = ({
   const {
     meta: { assignee, sender },
     id,
+    custom_attributes: conversationCustomAttributes = {},
   } = conversation;
+  const { custom_attributes: contactCustomAttributes } = sender || {};
 
-  return {
+  const standardVariables = {
     'contact.name': capitalizeName(sender?.name || ''),
     'contact.first_name': getFirstName({ user: sender }),
     'contact.last_name': getLastName({ user: sender }),
@@ -44,6 +51,27 @@ export const getMessageVariables = ({
     'agent.last_name': getLastName({ user: assignee }),
     'agent.email': assignee?.email ?? '',
   };
+  const conversationCustomAttributeVariables = Object.entries(
+    conversationCustomAttributes as CustomAttributes
+  ).reduce((acc: CustomAttributes, [key, value]) => {
+    acc[`conversation.custom_attribute.${key}`] = value;
+    return acc;
+  }, {});
+
+  const contactCustomAttributeVariables = Object.entries(
+    contactCustomAttributes as CustomAttributes
+  ).reduce((acc: CustomAttributes, [key, value]) => {
+    acc[`contact.custom_attribute.${key}`] = value;
+    return acc;
+  }, {});
+
+  const variables = {
+    ...standardVariables,
+    ...conversationCustomAttributeVariables,
+    ...contactCustomAttributeVariables,
+  };
+
+  return variables;
 };
 
 export const replaceVariablesInMessage = ({

diff --git a/src/types/conversation.ts b/src/types/conversation.ts
@@ -1,6 +1,7 @@
 export interface Conversation {
   meta: Meta;
   id: number;
+  custom_attributes: CustomAttributes;
 }
 
 export interface Meta {
@@ -13,6 +14,7 @@ export interface Sender {
   email?: string;
   name?: string;
   phone_number?: string;
+  custom_attributes?: CustomAttributes;
 }
 
 export interface Assignee {
@@ -25,3 +27,7 @@ export interface Assignee {
 export interface Variables {
   [key: string]: string | number;
 }
+
+export interface CustomAttributes {
+  [key: string]: any;
+}
diff --git a/test/canned.test.ts b/test/canned.test.ts
@@ -98,9 +98,14 @@ describe('#getMessageVariables', () => {
           name: 'john Doe',
           email: '[email protected]',
           phone_number: '1234567890',
+          custom_attributes: { priority: 'high' },
         },
       },
       id: 1,
+      custom_attributes: {
+        car_model: 'Tesla Model S',
+        car_year: '2022',
+      },
     };
     expect(getMessageVariables({ conversation })).toEqual({
       'contact.name': 'John Doe',
@@ -114,6 +119,9 @@ describe('#getMessageVariables', () => {
       'agent.first_name': 'Samuel',
       'agent.last_name': 'Smith',
       'agent.email': '[email protected]',
+      'contact.custom_attribute.priority': 'high',
+      'conversation.custom_attribute.car_model': 'Tesla Model S',
+      'conversation.custom_attribute.car_year': '2022',
     });
   });
 });