Add extra protection to source

matthewturner · Dec 13, 2019 · b1c5dbf · b1c5dbf
1 parent 8805a35
commit b1c5dbf
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/aws/handlers/Handler.js b/aws/handlers/Handler.js
@@ -22,7 +22,7 @@ class Handler {
         const userId = profile.user_id;
         const shortUserId = helpers.truncateUserId(userId);
         this._logger.prefix = shortUserId;
-        const source = event.directive.payload.source || 'user';
+        const source = this.retrieveSource(event);
         const context = {
             userId: userId,
             shortUserId: shortUserId,
@@ -59,6 +59,13 @@ class Handler {
         return new MockProfileGateway();
     }
 
+    retrieveSource(event) {
+        if (event.directive.payload && event.directive.payload.source) {
+            return event.directive.payload.source;
+        }
+        return 'user';
+    }
+
     async retrieveProfile(event) {
         if (event.directive.endpoint && event.directive.endpoint.userId) {
             this._logger.debug('UserId found; skipping profile data...');