@aws-amplify/api-graphql 4.0.1-console-preview.b5a42af.0 → 4.0.1-console-preview.9607e4a.0

package/lib-esm/internals/InternalGraphQLAPI.js

@@ -5,7 +5,8 @@ import { print, parse, GraphQLError, } from 'graphql';
 import { Amplify, Cache, fetchAuthSession } from '@aws-amplify/core';
 import { ConsoleLogger as Logger, getAmplifyUserAgent, } from '@aws-amplify/core/internals/utils';
 import { GraphQLAuthError, } from '../types';
-import { post } from '@aws-amplify/api-rest/internals';
+import { isCancelError as isCancelErrorREST } from '@aws-amplify/api-rest';
+import { post, cancel as cancelREST, updateRequestToBeCancellable, } from '@aws-amplify/api-rest/internals';
 import { AWSAppSyncRealTimeProvider } from '../Providers/AWSAppSyncRealTimeProvider';
 var USER_AGENT_HEADER = 'x-amz-user-agent';
 var logger = new Logger('GraphQLAPI');
@@ -27,26 +28,27 @@ var InternalGraphQLAPIClass = /** @class */ (function () {
      */
     function InternalGraphQLAPIClass(options) {
         this.Cache = Cache;
-        this._api = { post: post };
+        this._api = { post: post, updateRequestToBeCancellable: updateRequestToBeCancellable };
         this._options = options;
         logger.debug('API Options', this._options);
     }
     InternalGraphQLAPIClass.prototype.getModuleName = function () {
         return 'InternalGraphQLAPI';
     };
-    InternalGraphQLAPIClass.prototype._headerBasedAuth = function (defaultAuthenticationType, additionalHeaders, customUserAgentDetails) {
+    InternalGraphQLAPIClass.prototype._headerBasedAuth = function (authMode, additionalHeaders) {
         if (additionalHeaders === void 0) { additionalHeaders = {}; }
-        var _a;
+        var _a, _b;
         return __awaiter(this, void 0, void 0, function () {
-            var config, _b, region, appSyncGraphqlEndpoint, apiKey, defaultAuthMode, headers, _c, session, token, e_1;
-            return __generator(this, function (_d) {
-                switch (_d.label) {
+            var config, _c, region, appSyncGraphqlEndpoint, apiKey, defaultAuthMode, authenticationType, headers, _d, session, token, e_1;
+            return __generator(this, function (_e) {
+                switch (_e.label) {
                     case 0:
                         config = Amplify.getConfig();
-                        _b = config.API.GraphQL, region = _b.region, appSyncGraphqlEndpoint = _b.endpoint, apiKey = _b.apiKey, defaultAuthMode = _b.defaultAuthMode;
+                        _c = ((_a = config.API) === null || _a === void 0 ? void 0 : _a.GraphQL) || {}, region = _c.region, appSyncGraphqlEndpoint = _c.endpoint, apiKey = _c.apiKey, defaultAuthMode = _c.defaultAuthMode;
+                        authenticationType = authMode || defaultAuthMode || 'iam';
                         headers = {};
-                        _c = defaultAuthMode;
-                        switch (_c) {
+                        _d = authenticationType;
+                        switch (_d) {
                             case 'apiKey': return [3 /*break*/, 1];
                             case 'iam': return [3 /*break*/, 2];
                             case 'oidc': return [3 /*break*/, 4];
@@ -65,17 +67,17 @@ var InternalGraphQLAPIClass = /** @class */ (function () {
                         return [3 /*break*/, 11];
                     case 2: return [4 /*yield*/, fetchAuthSession()];
                     case 3:
-                        session = _d.sent();
+                        session = _e.sent();
                         if (session.credentials === undefined) {
                             throw new Error(GraphQLAuthError.NO_CREDENTIALS);
                         }
                         return [3 /*break*/, 11];
                     case 4:
-                        _d.trys.push([4, 6, , 7]);
+                        _e.trys.push([4, 6, , 7]);
                         token = void 0;
                         return [4 /*yield*/, fetchAuthSession()];
                     case 5:
-                        token = (_a = (_d.sent()).tokens) === null || _a === void 0 ? void 0 : _a.accessToken.toString();
+                        token = (_b = (_e.sent()).tokens) === null || _b === void 0 ? void 0 : _b.accessToken.toString();
                         if (!token) {
                             throw new Error(GraphQLAuthError.NO_FEDERATED_JWT);
                         }
@@ -84,7 +86,7 @@ var InternalGraphQLAPIClass = /** @class */ (function () {
                         };
                         return [3 /*break*/, 7];
                     case 6:
-                        e_1 = _d.sent();
+                        e_1 = _e.sent();
                         throw new Error(GraphQLAuthError.NO_CURRENT_USER);
                     case 7: return [3 /*break*/, 11];
                     case 8:
@@ -138,7 +140,9 @@ var InternalGraphQLAPIClass = /** @class */ (function () {
         switch (operationType) {
             case 'query':
             case 'mutation':
-                var responsePromise = this._graphql({ query: query, variables: variables, authMode: authMode }, headers, customUserAgentDetails);
+                var abortController = new AbortController();
+                var responsePromise = this._graphql({ query: query, variables: variables, authMode: authMode }, headers, abortController, customUserAgentDetails);
+                this._api.updateRequestToBeCancellable(responsePromise, abortController);
                 return responsePromise;
             case 'subscription':
                 return this._graphqlSubscribe({ query: query, variables: variables, authMode: authMode }, headers, customUserAgentDetails);
@@ -146,48 +150,50 @@ var InternalGraphQLAPIClass = /** @class */ (function () {
                 throw new Error("invalid operation type: " + operationType);
         }
     };
-    InternalGraphQLAPIClass.prototype._graphql = function (_a, additionalHeaders, customUserAgentDetails) {
+    InternalGraphQLAPIClass.prototype._graphql = function (_a, additionalHeaders, abortController, customUserAgentDetails) {
         var query = _a.query, variables = _a.variables, authMode = _a.authMode;
         if (additionalHeaders === void 0) { additionalHeaders = {}; }
+        var _b;
         return __awaiter(this, void 0, void 0, function () {
-            var config, _b, region, appSyncGraphqlEndpoint, customGraphqlEndpoint, customEndpointRegion, headers, _c, _d, _e, _f, _g, body, endpoint, error, response, responsePayload, err_1, errors;
-            var _h;
-            return __generator(this, function (_j) {
-                switch (_j.label) {
+            var config, _c, region, appSyncGraphqlEndpoint, customGraphqlEndpoint, customEndpointRegion, headers, _d, _e, _f, _g, _h, body, endpoint, error, response, responseBody, result, err_1, errors;
+            var _j;
+            return __generator(this, function (_k) {
+                switch (_k.label) {
                     case 0:
                         config = Amplify.getConfig();
-                        _b = config.API.GraphQL, region = _b.region, appSyncGraphqlEndpoint = _b.endpoint;
+                        _c = ((_b = config.API) === null || _b === void 0 ? void 0 : _b.GraphQL) || {}, region = _c.region, appSyncGraphqlEndpoint = _c.endpoint;
                         customGraphqlEndpoint = null;
                         customEndpointRegion = null;
-                        _c = [{}];
-                        _d = !customGraphqlEndpoint;
-                        if (!_d) return [3 /*break*/, 2];
-                        return [4 /*yield*/, this._headerBasedAuth(authMode, additionalHeaders, customUserAgentDetails)];
+                        _d = [{}];
+                        _e = !customGraphqlEndpoint;
+                        if (!_e) return [3 /*break*/, 2];
+                        return [4 /*yield*/, this._headerBasedAuth(authMode, additionalHeaders)];
                     case 1:
-                        _d = (_j.sent());
-                        _j.label = 2;
+                        _e = (_k.sent());
+                        _k.label = 2;
                     case 2:
-                        _e = [__assign.apply(void 0, _c.concat([(_d)]))];
-                        _f = customGraphqlEndpoint;
-                        if (!_f) return [3 /*break*/, 6];
+                        _f = [__assign.apply(void 0, _d.concat([(_e)]))];
+                        _g = customGraphqlEndpoint;
+                        if (!_g) return [3 /*break*/, 6];
                         if (!customEndpointRegion) return [3 /*break*/, 4];
-                        return [4 /*yield*/, this._headerBasedAuth(authMode, additionalHeaders, customUserAgentDetails)];
+                        return [4 /*yield*/, this._headerBasedAuth(authMode, additionalHeaders)];
                     case 3:
-                        _g = _j.sent();
+                        _h = _k.sent();
                         return [3 /*break*/, 5];
                     case 4:
-                        _g = { Authorization: null };
-                        _j.label = 5;
+                        _h = { Authorization: null };
+                        _k.label = 5;
                     case 5:
-                        _f = (_g);
-                        _j.label = 6;
+                        _g = (_h);
+                        _k.label = 6;
                     case 6:
-                        headers = __assign.apply(void 0, [__assign.apply(void 0, [__assign.apply(void 0, _e.concat([(_f)])), additionalHeaders]), (!customGraphqlEndpoint && (_h = {},
-                                _h[USER_AGENT_HEADER] = getAmplifyUserAgent(customUserAgentDetails),
-                                _h))]);
+                        headers = __assign.apply(void 0, [__assign.apply(void 0, [__assign.apply(void 0, _f.concat([((_g) ||
+                                        {})])), additionalHeaders]), (!customGraphqlEndpoint && (_j = {},
+                                _j[USER_AGENT_HEADER] = getAmplifyUserAgent(customUserAgentDetails),
+                                _j))]);
                         body = {
                             query: print(query),
-                            variables: variables,
+                            variables: variables || null,
                         };
                         endpoint = customGraphqlEndpoint || appSyncGraphqlEndpoint;
                         if (!endpoint) {
@@ -197,9 +203,9 @@ var InternalGraphQLAPIClass = /** @class */ (function () {
                                 errors: [error],
                             };
                         }
-                        _j.label = 7;
+                        _k.label = 7;
                     case 7:
-                        _j.trys.push([7, 10, , 11]);
+                        _k.trys.push([7, 10, , 11]);
                         return [4 /*yield*/, this._api.post({
                                 url: new URL(endpoint),
                                 options: {
@@ -210,15 +216,23 @@ var InternalGraphQLAPIClass = /** @class */ (function () {
                                         region: region,
                                     },
                                 },
+                                abortController: abortController,
                             })];
                     case 8:
-                        responsePayload = (_j.sent()).body;
-                        return [4 /*yield*/, responsePayload.json()];
+                        responseBody = (_k.sent()).body;
+                        return [4 /*yield*/, responseBody.json()];
                     case 9:
-                        response = _j.sent();
+                        result = _k.sent();
+                        response = result;
                         return [3 /*break*/, 11];
                     case 10:
-                        err_1 = _j.sent();
+                        err_1 = _k.sent();
+                        // If the exception is because user intentionally
+                        // cancelled the request, do not modify the exception
+                        // so that clients can identify the exception correctly.
+                        if (isCancelErrorREST(err_1)) {
+                            throw err_1;
+                        }
                         response = {
                             data: {},
                             errors: [new GraphQLError(err_1.message, null, null, null, null, err_1)],
@@ -234,8 +248,24 @@ var InternalGraphQLAPIClass = /** @class */ (function () {
             });
         });
     };
+    /**
+     * Checks to see if an error thrown is from an api request cancellation
+     * @param {any} error - Any error
+     * @return {boolean} - A boolean indicating if the error was from an api request cancellation
+     */
+    InternalGraphQLAPIClass.prototype.isCancelError = function (error) {
+        return isCancelErrorREST(error);
+    };
+    /**
+     * Cancels an inflight request. Only applicable for graphql queries and mutations
+     * @param {any} request - request to cancel
+     * @returns - A boolean indicating if the request was cancelled
+     */
+    InternalGraphQLAPIClass.prototype.cancel = function (request, message) {
+        return cancelREST(request, message);
+    };
     InternalGraphQLAPIClass.prototype._graphqlSubscribe = function (_a, additionalHeaders, customUserAgentDetails) {
-        var query = _a.query, variables = _a.variables, defaultAuthenticationType = _a.authMode, authToken = _a.authToken;
+        var query = _a.query, variables = _a.variables, authMode = _a.authMode;
         if (additionalHeaders === void 0) { additionalHeaders = {}; }
         var _b;
         var GraphQL = ((_b = Amplify.getConfig().API) !== null && _b !== void 0 ? _b : {}).GraphQL;
@@ -247,9 +277,10 @@ var InternalGraphQLAPIClass = /** @class */ (function () {
             variables: variables,
             appSyncGraphqlEndpoint: GraphQL === null || GraphQL === void 0 ? void 0 : GraphQL.endpoint,
             region: GraphQL === null || GraphQL === void 0 ? void 0 : GraphQL.region,
-            authenticationType: GraphQL === null || GraphQL === void 0 ? void 0 : GraphQL.defaultAuthMode,
+            authenticationType: authMode !== null && authMode !== void 0 ? authMode : GraphQL === null || GraphQL === void 0 ? void 0 : GraphQL.defaultAuthMode,
             apiKey: GraphQL === null || GraphQL === void 0 ? void 0 : GraphQL.apiKey,
-        });
+            additionalHeaders: additionalHeaders,
+        }, customUserAgentDetails);
     };
     return InternalGraphQLAPIClass;
 }());

package/lib-esm/internals/index.d.ts

@@ -1,2 +1,2 @@
 export { InternalGraphQLAPI, InternalGraphQLAPIClass, } from './InternalGraphQLAPI';
-export { graphql } from './v6';
+export { graphql, cancel, isCancelError } from './v6';

package/lib-esm/internals/index.js

@@ -1,4 +1,4 @@
 // Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
 // SPDX-License-Identifier: Apache-2.0
 export { InternalGraphQLAPI, InternalGraphQLAPIClass, } from './InternalGraphQLAPI';
-export { graphql } from './v6';
+export { graphql, cancel, isCancelError } from './v6';

package/lib-esm/internals/v6.d.ts

@@ -86,4 +86,16 @@ import { GraphQLOptionsV6, GraphQLResponseV6 } from '../types';
 export declare function graphql<FALLBACK_TYPES = unknown, TYPED_GQL_STRING extends string = string>(options: GraphQLOptionsV6<FALLBACK_TYPES, TYPED_GQL_STRING>, additionalHeaders?: {
     [key: string]: string;
 }): GraphQLResponseV6<FALLBACK_TYPES, TYPED_GQL_STRING>;
+/**
+ * Cancels an inflight request. Only applicable for graphql queries and mutations
+ * @param {any} request - request to cancel
+ * @returns - A boolean indicating if the request was cancelled
+ */
+export declare function cancel(promise: Promise<any>, message?: string): boolean;
+/**
+ * Checks to see if an error thrown is from an api request cancellation
+ * @param {any} error - Any error
+ * @returns - A boolean indicating if the error was from an api request cancellation
+ */
+export declare function isCancelError(error: any): boolean;
 export { GraphQLOptionsV6, GraphQLResponseV6 };

package/lib-esm/internals/v6.js

@@ -94,3 +94,19 @@ export function graphql(options, additionalHeaders) {
     var result = GraphQLAPI.graphql(options, additionalHeaders);
     return result;
 }
+/**
+ * Cancels an inflight request. Only applicable for graphql queries and mutations
+ * @param {any} request - request to cancel
+ * @returns - A boolean indicating if the request was cancelled
+ */
+export function cancel(promise, message) {
+    return GraphQLAPI.cancel(promise, message);
+}
+/**
+ * Checks to see if an error thrown is from an api request cancellation
+ * @param {any} error - Any error
+ * @returns - A boolean indicating if the error was from an api request cancellation
+ */
+export function isCancelError(error) {
+    return GraphQLAPI.isCancelError(error);
+}