@aws-amplify/api-graphql 2.2.13-unstable.4 → 2.2.13-unstable.9

package/dist/aws-amplify-api-graphql.js

@@ -11346,7 +11346,7 @@ var __read = undefined && undefined.__read || function (o, n) {
   return ar;
 };
 /*
- * Copyright 2017-2019 Amazon.com, Inc. or its affiliates. All Rights Reserved.
+ * Copyright 2017-2021 Amazon.com, Inc. or its affiliates. All Rights Reserved.
  *
  * Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance with
  * the License. A copy of the License is located at
@@ -11482,6 +11482,8 @@ var START_ACK_TIMEOUT = 15000;
  */
 
 var DEFAULT_KEEP_ALIVE_TIMEOUT = 5 * 60 * 1000;
+var standardDomainPattern = /^https:\/\/\w{26}\.appsync\-api\.\w{2}(?:(?:\-\w{2,})+)\-\d\.amazonaws.com\/graphql$/i;
+var customDomainPath = '/realtime';
 
 var AWSAppSyncRealTimeProvider =
 /** @class */
@@ -11512,6 +11514,11 @@ function (_super) {
         throw new Error('Operation not supported');
       });
     });
+  }; // Check if url matches standard domain pattern
+
+
+  AWSAppSyncRealTimeProvider.prototype.isCustomDomain = function (url) {
+    return url.match(standardDomainPattern) === null;
   };
 
   AWSAppSyncRealTimeProvider.prototype.subscribe = function (_topics, options) {
@@ -11994,7 +12001,7 @@ function (_super) {
 
     return new Promise(function (res, rej) {
       return __awaiter(_this, void 0, void 0, function () {
-        var protocol, discoverableEndpoint, payloadString, headerString, _a, _b, headerQs, payloadQs, awsRealTimeUrl, err_3;
+        var payloadString, headerString, _a, _b, headerQs, payloadQs, discoverableEndpoint, protocol, awsRealTimeUrl, err_3;
 
         return __generator(this, function (_c) {
           switch (_c.label) {
@@ -12012,8 +12019,6 @@ function (_super) {
               _c.trys.push([1, 4,, 5]);
 
               this.socketStatus = SOCKET_STATUS.CONNECTING;
-              protocol = this.isSSLEnabled ? 'wss://' : 'ws://';
-              discoverableEndpoint = appSyncGraphqlEndpoint.replace('https://', protocol).replace('http://', protocol).replace('appsync-api', 'appsync-realtime-api').replace('gogi-beta', 'grt-beta');
               payloadString = '{}';
               _b = (_a = JSON).stringify;
               return [4
@@ -12032,6 +12037,16 @@ function (_super) {
               headerString = _b.apply(_a, [_c.sent()]);
               headerQs = buffer__WEBPACK_IMPORTED_MODULE_4__["Buffer"].from(headerString).toString('base64');
               payloadQs = buffer__WEBPACK_IMPORTED_MODULE_4__["Buffer"].from(payloadString).toString('base64');
+              discoverableEndpoint = appSyncGraphqlEndpoint;
+
+              if (this.isCustomDomain(discoverableEndpoint)) {
+                discoverableEndpoint = discoverableEndpoint.concat(customDomainPath);
+              } else {
+                discoverableEndpoint = discoverableEndpoint.replace('appsync-api', 'appsync-realtime-api').replace('gogi-beta', 'grt-beta');
+              }
+
+              protocol = this.isSSLEnabled ? 'wss://' : 'ws://';
+              discoverableEndpoint = discoverableEndpoint.replace('https://', protocol).replace('http://', protocol);
               awsRealTimeUrl = discoverableEndpoint + "?header=" + headerQs + "&payload=" + payloadQs;
               return [4
               /*yield*/