init

sendgrid · Nov 17, 2023 · 91b4cb6 · 91b4cb6
1 parent b8125d8
commit 91b4cb6
Show file tree

Hide file tree

Showing 2 changed files with 23 additions and 0 deletions.
diff --git a/packages/client/src/classes/client.js b/packages/client/src/classes/client.js
@@ -15,6 +15,11 @@ const API_KEY_PREFIX = 'SG.';
 const SENDGRID_BASE_URL = 'https://api.sendgrid.com/';
 const TWILIO_BASE_URL = 'https://email.twilio.com/';
 
+// Initialize the allowed regions and their corresponding hosts
+const REGION_HOST_MAP = {
+  eu: 'api.eu.sendgrid.com',
+  global: 'api.sendgrid.com',
+};
 class Client {
   constructor() {
     this.auth = '';
@@ -94,6 +99,14 @@ class Client {
     return this;
   }
 
+  setDataResidency(region) {
+    if (!REGION_HOST_MAP.hasOwnProperty(region)) {
+      console.warn('Region can only be "global" or "eu".');
+    } else {
+      this.setDefaultRequest('baseUrl', REGION_HOST_MAP[region]);
+    }
+  }
+
   createHeaders(data) {
     // Merge data with default headers.
     const headers = mergeData(this.defaultHeaders, data);

diff --git a/packages/client/src/client.spec.js b/packages/client/src/client.spec.js
@@ -1,5 +1,6 @@
 'use strict';
 const nock = require('nock');
+const sgClient = require('./client');
 
 const testRequest = (request, statusCode) => {
   const sgClient = require('./client');
@@ -3091,3 +3092,12 @@ describe('test_whitelabel_links__link_id__subuser_post', () => {
     return testRequest(request, 200);
   });
 });
+
+describe('setDataResidency', () => {
+  const sgClient = require('./client');
+  sgClient.setDataResidency('eu');
+
+  it('should have host as eu', () => {
+    expect(sgClient.baseUrl).to.equal('api.eu.sendgrid.com');
+  });
+});