feat: add ip pool override

apps/worker/src/app/workflow/usecases/send-message/send-message-email.usecase.ts

@@ -244,7 +244,7 @@ export class SendMessageEmail extends SendMessageBase {
         id: message._id,
         replyTo: replyToAddress,
       },
-      command.overrides?.email || {}
+      overrides || {}
     );
 
     if (command.overrides?.email?.replyTo) {
@@ -468,6 +468,7 @@ export const createMailData = (options: IEmailOptions, overrides: Record<string,
     text: overrides?.text,
     cc: overrides?.cc || [],
     bcc: overrides?.bcc || [],
+    ipPoolName: overrides?.ipPoolName,
   };
 };
 

libs/shared/src/types/events/index.ts

@@ -21,6 +21,7 @@ export interface IEmailOptions {
   bcc?: string[];
   payloadDetails?: any;
   notificationDetails?: any;
+  ipPoolName?: string;
 }
 
 export interface ITriggerPayload {

packages/stateless/src/lib/provider/provider.interface.ts

@@ -22,6 +22,7 @@ export interface IEmailOptions {
   bcc?: string[];
   payloadDetails?: any;
   notificationDetails?: any;
+  ipPoolName?: string;
 }
 
 export interface ISmsOptions {

providers/sendgrid/src/lib/sendgrid.provider.spec.ts

@@ -15,6 +15,7 @@ const mockNovuMessage = {
   attachments: [
     { mime: 'text/plain', file: Buffer.from('dGVzdA=='), name: 'test.txt' },
   ],
+  id: 'message_id',
 };
 
 test('should trigger sendgrid correctly', async () => {
@@ -47,7 +48,7 @@ test('should trigger sendgrid correctly', async () => {
       },
     ],
     customArgs: {
-      id: undefined,
+      id: 'message_id',
     },
   });
 });
@@ -65,3 +66,36 @@ test('should check provider integration correctly', async () => {
   expect(spy).toHaveBeenCalled();
   expect(response.success).toBe(true);
 });
+
+test('should get ip pool name from credentials', async () => {
+  const provider = new SendgridEmailProvider({
+    ...mockConfig,
+    ...{ ipPoolName: 'config_ip' },
+  });
+  const sendMock = jest.fn().mockResolvedValue([{ statusCode: 202 }]);
+  jest.spyOn(MailService.prototype, 'send').mockImplementation(sendMock);
+
+  await provider.sendMessage({
+    ...mockNovuMessage,
+  });
+  expect(sendMock).toHaveBeenCalledWith(
+    expect.objectContaining({ ipPoolName: 'config_ip' })
+  );
+});
+
+test('should override credentials with mail data', async () => {
+  const provider = new SendgridEmailProvider({
+    ...mockConfig,
+    ...{ ipPoolName: 'config_ip' },
+  });
+  const sendMock = jest.fn().mockResolvedValue([{ statusCode: 202 }]);
+  jest.spyOn(MailService.prototype, 'send').mockImplementation(sendMock);
+
+  await provider.sendMessage({
+    ...mockNovuMessage,
+    ...{ ipPoolName: 'ip_from_mail_data' },
+  });
+  expect(sendMock).toHaveBeenCalledWith(
+    expect.objectContaining({ ipPoolName: 'ip_from_mail_data' })
+  );
+});

providers/sendgrid/src/lib/sendgrid.provider.ts

@@ -70,7 +70,7 @@ export class SendgridEmailProvider implements IEmailProvider {
         email: options.from || this.config.from,
         name: this.config.senderName,
       },
-      ipPoolName: this.config.ipPoolName,
+      ipPoolName: options.ipPoolName || this.config.ipPoolName,
       to: options.to.map((email) => ({ email })),
       cc: options.cc?.map((ccItem) => ({ email: ccItem })),
       bcc: options.bcc?.map((ccItem) => ({ email: ccItem })),