@aws-amplify/api-graphql 4.0.1-api-v6-models.9351bcf.0 → 4.0.1-api-v6-models.891fe0d.0

package/lib/types/index.d.ts

@@ -1,9 +1,10 @@
-import { AmplifyClassV6 } from '@aws-amplify/core';
+import { AmplifyClassV6, ResourcesConfig } from '@aws-amplify/core';
 import { ModelTypes } from '@aws-amplify/amplify-api-next-types-alpha';
 import { Source, DocumentNode, GraphQLError } from 'graphql';
 export { OperationTypeNode } from 'graphql';
 import { Observable } from 'rxjs';
 import { GraphQLAuthMode, DocumentType } from '@aws-amplify/core/internals/utils';
+import { AmplifyServer } from '@aws-amplify/core/internals/adapter-core';
 export { CONTROL_MSG, ConnectionState } from './PubSub';
 export { SelectionSet } from '@aws-amplify/amplify-api-next-types-alpha';
 export { CommonPublicClientOptions } from '../internals/types';
@@ -288,10 +289,41 @@ export type V6Client<T extends Record<any, any> = never> = ExcludeNeverFields<{
     [__amplify]: AmplifyClassV6;
     [__authMode]?: GraphQLAuthMode;
     [__authToken]?: string;
-    graphql: <FALLBACK_TYPES = unknown, TYPED_GQL_STRING extends string = string>(options: GraphQLOptionsV6<FALLBACK_TYPES, TYPED_GQL_STRING>, additionalHeaders?: {
-        [key: string]: string;
-    } | undefined) => GraphQLResponseV6<FALLBACK_TYPES, TYPED_GQL_STRING>;
+    graphql: GraphQLMethod;
     cancel: (promise: Promise<any>, message?: string) => boolean;
     isCancelError: (error: any) => boolean;
     models: ModelTypes<T>;
 }>;
+export type V6ClientSSRRequest<T extends Record<any, any> = never> = ExcludeNeverFields<{
+    [__amplify]: AmplifyClassV6;
+    [__authMode]?: GraphQLAuthMode;
+    [__authToken]?: string;
+    graphql: GraphQLMethodSSR;
+    cancel: (promise: Promise<any>, message?: string) => boolean;
+    isCancelError: (error: any) => boolean;
+    models: ModelTypes<T, 'REQUEST'>;
+}>;
+export type V6ClientSSRCookies<T extends Record<any, any> = never> = ExcludeNeverFields<{
+    [__amplify]: AmplifyClassV6;
+    [__authMode]?: GraphQLAuthMode;
+    [__authToken]?: string;
+    graphql: GraphQLMethod;
+    cancel: (promise: Promise<any>, message?: string) => boolean;
+    isCancelError: (error: any) => boolean;
+    models: ModelTypes<T, 'COOKIES'>;
+}>;
+export type GraphQLMethod = <FALLBACK_TYPES = unknown, TYPED_GQL_STRING extends string = string>(options: GraphQLOptionsV6<FALLBACK_TYPES, TYPED_GQL_STRING>, additionalHeaders?: {
+    [key: string]: string;
+} | undefined) => GraphQLResponseV6<FALLBACK_TYPES, TYPED_GQL_STRING>;
+export type GraphQLMethodSSR = <FALLBACK_TYPES = unknown, TYPED_GQL_STRING extends string = string>(contextSpec: AmplifyServer.ContextSpec, options: GraphQLOptionsV6<FALLBACK_TYPES, TYPED_GQL_STRING>, additionalHeaders?: {
+    [key: string]: string;
+} | undefined) => GraphQLResponseV6<FALLBACK_TYPES, TYPED_GQL_STRING>;
+/**
+ * @private
+ *
+ * The knobs available for configuring `server/generateClient` internally.
+ */
+export type ServerClientGenerationParams = {
+    amplify: null | ((fn: (amplify: AmplifyClassV6) => Promise<any>) => Promise<any>);
+    config: ResourcesConfig;
+};

package/lib/utils/ConnectionStateMonitor.js

@@ -3,10 +3,9 @@
 // SPDX-License-Identifier: Apache-2.0
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.ConnectionStateMonitor = exports.CONNECTION_CHANGE = void 0;
-var tslib_1 = require("tslib");
-var rxjs_1 = require("rxjs");
-var PubSub_1 = require("../types/PubSub");
-var ReachabilityMonitor_1 = require("./ReachabilityMonitor");
+const rxjs_1 = require("rxjs");
+const PubSub_1 = require("../types/PubSub");
+const ReachabilityMonitor_1 = require("./ReachabilityMonitor");
 exports.CONNECTION_CHANGE = {
     KEEP_ALIVE_MISSED: { keepAliveState: 'unhealthy' },
     KEEP_ALIVE: { keepAliveState: 'healthy' },
@@ -24,9 +23,8 @@ exports.CONNECTION_CHANGE = {
     ONLINE: { networkState: 'connected' },
     OFFLINE: { networkState: 'disconnected' },
 };
-var ConnectionStateMonitor = /** @class */ (function () {
-    function ConnectionStateMonitor() {
-        var _this = this;
+class ConnectionStateMonitor {
+    constructor() {
         this._networkMonitoringSubscription = undefined;
         this._linkedConnectionState = {
             networkState: 'connected',
@@ -35,73 +33,61 @@ var ConnectionStateMonitor = /** @class */ (function () {
             keepAliveState: 'healthy',
         };
         // Attempt to update the state with the current actual network state
-        this._initialNetworkStateSubscription = (0, ReachabilityMonitor_1.ReachabilityMonitor)().subscribe(function (_a) {
-            var _b;
-            var online = _a.online;
-            _this.record(online ? exports.CONNECTION_CHANGE.ONLINE : exports.CONNECTION_CHANGE.OFFLINE);
-            (_b = _this._initialNetworkStateSubscription) === null || _b === void 0 ? void 0 : _b.unsubscribe();
+        this._initialNetworkStateSubscription = (0, ReachabilityMonitor_1.ReachabilityMonitor)().subscribe(({ online }) => {
+            this.record(online ? exports.CONNECTION_CHANGE.ONLINE : exports.CONNECTION_CHANGE.OFFLINE);
+            this._initialNetworkStateSubscription?.unsubscribe();
         });
         this._linkedConnectionStateObservable =
-            new rxjs_1.Observable(function (connectionStateObserver) {
-                connectionStateObserver.next(_this._linkedConnectionState);
-                _this._linkedConnectionStateObserver = connectionStateObserver;
+            new rxjs_1.Observable(connectionStateObserver => {
+                connectionStateObserver.next(this._linkedConnectionState);
+                this._linkedConnectionStateObserver = connectionStateObserver;
             });
     }
     /**
      * Turn network state monitoring on if it isn't on already
      */
-    ConnectionStateMonitor.prototype.enableNetworkMonitoring = function () {
-        var _this = this;
-        var _a;
+    enableNetworkMonitoring() {
         // If no initial network state was discovered, stop trying
-        (_a = this._initialNetworkStateSubscription) === null || _a === void 0 ? void 0 : _a.unsubscribe();
+        this._initialNetworkStateSubscription?.unsubscribe();
         // Maintain the network state based on the reachability monitor
         if (this._networkMonitoringSubscription === undefined) {
-            this._networkMonitoringSubscription = (0, ReachabilityMonitor_1.ReachabilityMonitor)().subscribe(function (_a) {
-                var online = _a.online;
-                _this.record(online ? exports.CONNECTION_CHANGE.ONLINE : exports.CONNECTION_CHANGE.OFFLINE);
+            this._networkMonitoringSubscription = (0, ReachabilityMonitor_1.ReachabilityMonitor)().subscribe(({ online }) => {
+                this.record(online ? exports.CONNECTION_CHANGE.ONLINE : exports.CONNECTION_CHANGE.OFFLINE);
             });
         }
-    };
+    }
     /**
      * Turn network state monitoring off if it isn't off already
      */
-    ConnectionStateMonitor.prototype.disableNetworkMonitoring = function () {
-        var _a;
-        (_a = this._networkMonitoringSubscription) === null || _a === void 0 ? void 0 : _a.unsubscribe();
+    disableNetworkMonitoring() {
+        this._networkMonitoringSubscription?.unsubscribe();
         this._networkMonitoringSubscription = undefined;
-    };
-    Object.defineProperty(ConnectionStateMonitor.prototype, "connectionStateObservable", {
-        /**
-         * Get the observable that allows us to monitor the connection state
-         *
-         * @returns {Observable<ConnectionState>} - The observable that emits ConnectionState updates
-         */
-        get: function () {
-            var _this = this;
-            var previous;
-            // The linked state aggregates state changes to any of the network, connection,
-            // intendedConnection and keepAliveHealth. Some states will change these independent
-            // states without changing the overall connection state.
-            // After translating from linked states to ConnectionState, then remove any duplicates
-            return this._linkedConnectionStateObservable
-                .pipe((0, rxjs_1.map)(function (value) {
-                return _this.connectionStatesTranslator(value);
-            }))
-                .pipe((0, rxjs_1.filter)(function (current) {
-                var toInclude = current !== previous;
-                previous = current;
-                return toInclude;
-            }));
-        },
-        enumerable: false,
-        configurable: true
-    });
+    }
+    /**
+     * Get the observable that allows us to monitor the connection state
+     *
+     * @returns {Observable<ConnectionState>} - The observable that emits ConnectionState updates
+     */
+    get connectionStateObservable() {
+        let previous;
+        // The linked state aggregates state changes to any of the network, connection,
+        // intendedConnection and keepAliveHealth. Some states will change these independent
+        // states without changing the overall connection state.
+        // After translating from linked states to ConnectionState, then remove any duplicates
+        return this._linkedConnectionStateObservable
+            .pipe((0, rxjs_1.map)(value => {
+            return this.connectionStatesTranslator(value);
+        }))
+            .pipe((0, rxjs_1.filter)(current => {
+            const toInclude = current !== previous;
+            previous = current;
+            return toInclude;
+        }));
+    }
     /*
      * Updates local connection state and emits the full state to the observer.
      */
-    ConnectionStateMonitor.prototype.record = function (statusUpdates) {
-        var _a;
+    record(statusUpdates) {
         // Maintain the network monitor
         if (statusUpdates.intendedConnectionState === 'connected') {
             this.enableNetworkMonitoring();
@@ -110,15 +96,17 @@ var ConnectionStateMonitor = /** @class */ (function () {
             this.disableNetworkMonitoring();
         }
         // Maintain the socket state
-        var newSocketStatus = tslib_1.__assign(tslib_1.__assign({}, this._linkedConnectionState), statusUpdates);
-        this._linkedConnectionState = tslib_1.__assign({}, newSocketStatus);
-        (_a = this._linkedConnectionStateObserver) === null || _a === void 0 ? void 0 : _a.next(this._linkedConnectionState);
-    };
+        const newSocketStatus = {
+            ...this._linkedConnectionState,
+            ...statusUpdates,
+        };
+        this._linkedConnectionState = { ...newSocketStatus };
+        this._linkedConnectionStateObserver?.next(this._linkedConnectionState);
+    }
     /*
      * Translate the ConnectionState structure into a specific ConnectionState string literal union
      */
-    ConnectionStateMonitor.prototype.connectionStatesTranslator = function (_a) {
-        var connectionState = _a.connectionState, networkState = _a.networkState, intendedConnectionState = _a.intendedConnectionState, keepAliveState = _a.keepAliveState;
+    connectionStatesTranslator({ connectionState, networkState, intendedConnectionState, keepAliveState, }) {
         if (connectionState === 'connected' && networkState === 'disconnected')
             return PubSub_1.ConnectionState.ConnectedPendingNetwork;
         if (connectionState === 'connected' &&
@@ -139,7 +127,6 @@ var ConnectionStateMonitor = /** @class */ (function () {
         if (connectionState === 'disconnected')
             return PubSub_1.ConnectionState.Disconnected;
         return PubSub_1.ConnectionState.Connected;
-    };
-    return ConnectionStateMonitor;
-}());
+    }
+}
 exports.ConnectionStateMonitor = ConnectionStateMonitor;

package/lib/utils/ReachabilityMonitor/index.js

@@ -3,6 +3,6 @@ Object.defineProperty(exports, "__esModule", { value: true });
 exports.ReachabilityMonitor = void 0;
 // Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
 // SPDX-License-Identifier: Apache-2.0
-var utils_1 = require("@aws-amplify/core/internals/utils");
-var ReachabilityMonitor = function () { return new utils_1.Reachability().networkMonitor(); };
+const utils_1 = require("@aws-amplify/core/internals/utils");
+const ReachabilityMonitor = () => new utils_1.Reachability().networkMonitor();
 exports.ReachabilityMonitor = ReachabilityMonitor;

package/lib/utils/ReachabilityMonitor/index.native.js

@@ -1,12 +1,10 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.ReachabilityMonitor = void 0;
-var tslib_1 = require("tslib");
+const tslib_1 = require("tslib");
 // Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
 // SPDX-License-Identifier: Apache-2.0
-var utils_1 = require("@aws-amplify/core/internals/utils");
-var netinfo_1 = tslib_1.__importDefault(require("@react-native-community/netinfo"));
-var ReachabilityMonitor = function () {
-    return new utils_1.Reachability().networkMonitor(netinfo_1.default);
-};
+const utils_1 = require("@aws-amplify/core/internals/utils");
+const netinfo_1 = tslib_1.__importDefault(require("@react-native-community/netinfo"));
+const ReachabilityMonitor = () => new utils_1.Reachability().networkMonitor(netinfo_1.default);
 exports.ReachabilityMonitor = ReachabilityMonitor;

package/lib/utils/ReconnectionMonitor.js

@@ -1,7 +1,7 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.ReconnectionMonitor = exports.ReconnectEvent = void 0;
-var constants_1 = require("../Providers/constants");
+const constants_1 = require("../Providers/constants");
 var ReconnectEvent;
 (function (ReconnectEvent) {
     ReconnectEvent["START_RECONNECT"] = "START_RECONNECT";
@@ -13,31 +13,30 @@ var ReconnectEvent;
  *   Attempting to reconnect only once creates unrecoverable states when the network state isn't
  *   supported by the browser, so this keeps retrying every minute until halted.
  */
-var ReconnectionMonitor = /** @class */ (function () {
-    function ReconnectionMonitor() {
+class ReconnectionMonitor {
+    constructor() {
         this.reconnectObservers = [];
     }
     /**
      * Add reconnect observer to the list of observers to alert on reconnect
      */
-    ReconnectionMonitor.prototype.addObserver = function (reconnectObserver) {
+    addObserver(reconnectObserver) {
         this.reconnectObservers.push(reconnectObserver);
-    };
+    }
     /**
      * Given a reconnect event, start the appropriate behavior
      */
-    ReconnectionMonitor.prototype.record = function (event) {
-        var _this = this;
+    record(event) {
         if (event === ReconnectEvent.START_RECONNECT) {
             // If the reconnection hasn't been started
             if (this.reconnectSetTimeoutId === undefined &&
                 this.reconnectIntervalId === undefined) {
-                this.reconnectSetTimeoutId = setTimeout(function () {
+                this.reconnectSetTimeoutId = setTimeout(() => {
                     // Reconnect now
-                    _this._triggerReconnect();
+                    this._triggerReconnect();
                     // Retry reconnect every periodically until it works
-                    _this.reconnectIntervalId = setInterval(function () {
-                        _this._triggerReconnect();
+                    this.reconnectIntervalId = setInterval(() => {
+                        this._triggerReconnect();
                     }, constants_1.RECONNECT_INTERVAL);
                 }, constants_1.RECONNECT_DELAY);
             }
@@ -52,22 +51,19 @@ var ReconnectionMonitor = /** @class */ (function () {
                 this.reconnectSetTimeoutId = undefined;
             }
         }
-    };
+    }
     /**
      * Complete all reconnect observers
      */
-    ReconnectionMonitor.prototype.close = function () {
-        this.reconnectObservers.forEach(function (reconnectObserver) {
-            var _a;
-            (_a = reconnectObserver.complete) === null || _a === void 0 ? void 0 : _a.call(reconnectObserver);
+    close() {
+        this.reconnectObservers.forEach(reconnectObserver => {
+            reconnectObserver.complete?.();
         });
-    };
-    ReconnectionMonitor.prototype._triggerReconnect = function () {
-        this.reconnectObservers.forEach(function (reconnectObserver) {
-            var _a;
-            (_a = reconnectObserver.next) === null || _a === void 0 ? void 0 : _a.call(reconnectObserver);
+    }
+    _triggerReconnect() {
+        this.reconnectObservers.forEach(reconnectObserver => {
+            reconnectObserver.next?.();
         });
-    };
-    return ReconnectionMonitor;
-}());
+    }
+}
 exports.ReconnectionMonitor = ReconnectionMonitor;

package/lib/utils/errors/APIError.js

@@ -3,21 +3,17 @@
 // SPDX-License-Identifier: Apache-2.0
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.APIError = void 0;
-var tslib_1 = require("tslib");
-var utils_1 = require("@aws-amplify/core/internals/utils");
+const utils_1 = require("@aws-amplify/core/internals/utils");
 /**
  * @internal
  */
-var APIError = /** @class */ (function (_super) {
-    tslib_1.__extends(APIError, _super);
-    function APIError(params) {
-        var _this = _super.call(this, params) || this;
+class APIError extends utils_1.AmplifyError {
+    constructor(params) {
+        super(params);
         // Hack for making the custom error class work when transpiled to es5
         // TODO: Delete the following 2 lines after we change the build target to >= es2015
-        _this.constructor = APIError;
-        Object.setPrototypeOf(_this, APIError.prototype);
-        return _this;
+        this.constructor = APIError;
+        Object.setPrototypeOf(this, APIError.prototype);
     }
-    return APIError;
-}(utils_1.AmplifyError));
+}
 exports.APIError = APIError;

package/lib/utils/errors/assertValidationError.js

@@ -3,15 +3,15 @@
 // SPDX-License-Identifier: Apache-2.0
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.assertValidationError = void 0;
-var APIError_1 = require("./APIError");
-var validation_1 = require("./validation");
+const APIError_1 = require("./APIError");
+const validation_1 = require("./validation");
 /**
  * @internal
  */
 function assertValidationError(assertion, name) {
-    var _a = validation_1.validationErrorMap[name], message = _a.message, recoverySuggestion = _a.recoverySuggestion;
+    const { message, recoverySuggestion } = validation_1.validationErrorMap[name];
     if (!assertion) {
-        throw new APIError_1.APIError({ name: name, message: message, recoverySuggestion: recoverySuggestion });
+        throw new APIError_1.APIError({ name, message, recoverySuggestion });
     }
 }
 exports.assertValidationError = assertValidationError;

package/lib/utils/errors/validation.js

@@ -1,7 +1,6 @@
 "use strict";
 // Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
 // SPDX-License-Identifier: Apache-2.0
-var _a;
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.validationErrorMap = exports.APIValidationErrorCode = void 0;
 var APIValidationErrorCode;
@@ -10,18 +9,18 @@ var APIValidationErrorCode;
     APIValidationErrorCode["NoRegion"] = "NoRegion";
     APIValidationErrorCode["NoCustomEndpoint"] = "NoCustomEndpoint";
 })(APIValidationErrorCode || (exports.APIValidationErrorCode = APIValidationErrorCode = {}));
-exports.validationErrorMap = (_a = {},
-    _a[APIValidationErrorCode.NoAuthSession] = {
+exports.validationErrorMap = {
+    [APIValidationErrorCode.NoAuthSession]: {
         message: 'Auth session should not be empty.',
     },
     // TODO: re-enable when working in all test environments:
     // [APIValidationErrorCode.NoEndpoint]: {
     // 	message: 'Missing endpoint',
     // },
-    _a[APIValidationErrorCode.NoRegion] = {
+    [APIValidationErrorCode.NoRegion]: {
         message: 'Missing region.',
     },
-    _a[APIValidationErrorCode.NoCustomEndpoint] = {
+    [APIValidationErrorCode.NoCustomEndpoint]: {
         message: 'Custom endpoint region is present without custom endpoint.',
     },
-    _a);
+};

package/lib/utils/findIndexByFields.js

@@ -12,16 +12,11 @@ exports.findIndexByFields = void 0;
  * @returns Index of `needle` in `haystack`, otherwise -1 if not found.
  */
 function findIndexByFields(needle, haystack, keyFields) {
-    var searchObject = Object.fromEntries(keyFields.map(function (fieldName) { return [fieldName, needle[fieldName]]; }));
-    var _loop_1 = function (i) {
-        if (Object.keys(searchObject).every(function (k) { return searchObject[k] === haystack[i][k]; })) {
-            return { value: i };
+    const searchObject = Object.fromEntries(keyFields.map(fieldName => [fieldName, needle[fieldName]]));
+    for (let i = 0; i < haystack.length; i++) {
+        if (Object.keys(searchObject).every(k => searchObject[k] === haystack[i][k])) {
+            return i;
         }
-    };
-    for (var i = 0; i < haystack.length; i++) {
-        var state_1 = _loop_1(i);
-        if (typeof state_1 === "object")
-            return state_1.value;
     }
     return -1;
 }

package/lib/utils/resolveConfig.js

@@ -3,23 +3,22 @@
 // SPDX-License-Identifier: Apache-2.0
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.resolveConfig = void 0;
-var errors_1 = require("./errors");
+const errors_1 = require("./errors");
 /**
  * @internal
  */
-var resolveConfig = function (amplify) {
-    var _a, _b;
-    var _c = (_b = (_a = amplify.getConfig().API) === null || _a === void 0 ? void 0 : _a.GraphQL) !== null && _b !== void 0 ? _b : {}, apiKey = _c.apiKey, customEndpoint = _c.customEndpoint, customEndpointRegion = _c.customEndpointRegion, defaultAuthMode = _c.defaultAuthMode, endpoint = _c.endpoint, region = _c.region;
+const resolveConfig = (amplify) => {
+    const { apiKey, customEndpoint, customEndpointRegion, defaultAuthMode, endpoint, region, } = amplify.getConfig().API?.GraphQL ?? {};
     // TODO: re-enable when working in all test environments:
     // assertValidationError(!!endpoint, APIValidationErrorCode.NoEndpoint);
     (0, errors_1.assertValidationError)(!(!customEndpoint && customEndpointRegion), errors_1.APIValidationErrorCode.NoCustomEndpoint);
     return {
-        apiKey: apiKey,
-        customEndpoint: customEndpoint,
-        customEndpointRegion: customEndpointRegion,
-        defaultAuthMode: defaultAuthMode,
-        endpoint: endpoint,
-        region: region,
+        apiKey,
+        customEndpoint,
+        customEndpointRegion,
+        defaultAuthMode,
+        endpoint,
+        region,
     };
 };
 exports.resolveConfig = resolveConfig;

package/lib/utils/resolveLibraryOptions.js

@@ -6,10 +6,9 @@ exports.resolveLibraryOptions = void 0;
 /**
  * @internal
  */
-var resolveLibraryOptions = function (amplify) {
-    var _a, _b, _c, _d, _e, _f;
-    var headers = (_c = (_b = (_a = amplify.libraryOptions) === null || _a === void 0 ? void 0 : _a.API) === null || _b === void 0 ? void 0 : _b.GraphQL) === null || _c === void 0 ? void 0 : _c.headers;
-    var withCredentials = (_f = (_e = (_d = amplify.libraryOptions) === null || _d === void 0 ? void 0 : _d.API) === null || _e === void 0 ? void 0 : _e.GraphQL) === null || _f === void 0 ? void 0 : _f.withCredentials;
-    return { headers: headers, withCredentials: withCredentials };
+const resolveLibraryOptions = (amplify) => {
+    const headers = amplify.libraryOptions?.API?.GraphQL?.headers;
+    const withCredentials = amplify.libraryOptions?.API?.GraphQL?.withCredentials;
+    return { headers, withCredentials };
 };
 exports.resolveLibraryOptions = resolveLibraryOptions;

package/lib/utils/resolveOwnerFields.js

@@ -1,7 +1,6 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.resolveOwnerFields = void 0;
-var tslib_1 = require("tslib");
 /**
  * Given an introspection schema model, returns all owner fields.
  *
@@ -9,37 +8,16 @@ var tslib_1 = require("tslib");
  * @returns List of owner field names
  */
 function resolveOwnerFields(model) {
-    var e_1, _a, e_2, _b;
-    var ownerFields = new Set();
-    try {
-        for (var _c = tslib_1.__values(model.attributes || []), _d = _c.next(); !_d.done; _d = _c.next()) {
-            var attr = _d.value;
-            if (isAuthAttribute(attr)) {
-                try {
-                    for (var _e = (e_2 = void 0, tslib_1.__values(attr.properties.rules)), _f = _e.next(); !_f.done; _f = _e.next()) {
-                        var rule = _f.value;
-                        if (rule.allow === 'owner') {
-                            ownerFields.add(rule.ownerField || 'owner');
-                        }
-                    }
-                }
-                catch (e_2_1) { e_2 = { error: e_2_1 }; }
-                finally {
-                    try {
-                        if (_f && !_f.done && (_b = _e.return)) _b.call(_e);
-                    }
-                    finally { if (e_2) throw e_2.error; }
+    const ownerFields = new Set();
+    for (const attr of model.attributes || []) {
+        if (isAuthAttribute(attr)) {
+            for (const rule of attr.properties.rules) {
+                if (rule.allow === 'owner') {
+                    ownerFields.add(rule.ownerField || 'owner');
                 }
             }
         }
     }
-    catch (e_1_1) { e_1 = { error: e_1_1 }; }
-    finally {
-        try {
-            if (_d && !_d.done && (_a = _c.return)) _a.call(_c);
-        }
-        finally { if (e_1) throw e_1.error; }
-    }
     return Array.from(ownerFields);
 }
 exports.resolveOwnerFields = resolveOwnerFields;
@@ -51,11 +29,10 @@ exports.resolveOwnerFields = resolveOwnerFields;
  * @returns True if given object is an auth attribute
  */
 function isAuthAttribute(attribute) {
-    var _a, _b;
-    if ((attribute === null || attribute === void 0 ? void 0 : attribute.type) === 'auth') {
-        if (typeof (attribute === null || attribute === void 0 ? void 0 : attribute.properties) === 'object') {
-            if (Array.isArray((_a = attribute === null || attribute === void 0 ? void 0 : attribute.properties) === null || _a === void 0 ? void 0 : _a.rules)) {
-                return ((_b = attribute === null || attribute === void 0 ? void 0 : attribute.properties) === null || _b === void 0 ? void 0 : _b.rules).every(function (rule) { return !!rule.allow; });
+    if (attribute?.type === 'auth') {
+        if (typeof attribute?.properties === 'object') {
+            if (Array.isArray(attribute?.properties?.rules)) {
+                return (attribute?.properties?.rules).every(rule => !!rule.allow);
             }
         }
     }

package/lib/utils/resolvePKFields.js

@@ -1,7 +1,6 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.resolvePKFields = void 0;
-var tslib_1 = require("tslib");
 /**
  * Given a SchemaModel from a ModelIntrospectionSchema, returns the primary key
  * as an array of field names.
@@ -10,7 +9,7 @@ var tslib_1 = require("tslib");
  * @returns Array of field names
  */
 function resolvePKFields(model) {
-    var _a = model.primaryKeyInfo, primaryKeyFieldName = _a.primaryKeyFieldName, sortKeyFieldNames = _a.sortKeyFieldNames;
-    return tslib_1.__spreadArray([primaryKeyFieldName], tslib_1.__read(sortKeyFieldNames), false);
+    const { primaryKeyFieldName, sortKeyFieldNames } = model.primaryKeyInfo;
+    return [primaryKeyFieldName, ...sortKeyFieldNames];
 }
 exports.resolvePKFields = resolvePKFields;

package/lib-esm/GraphQLAPI.d.ts

@@ -19,7 +19,7 @@ export declare class GraphQLAPIClass extends InternalGraphQLAPIClass {
      * @param [additionalHeaders] - headers to merge in after any `graphql_headers` set in the config
      * @returns An Observable if the query is a subscription query, else a promise of the graphql result.
      */
-    graphql<T = any>(amplify: AmplifyClassV6, options: GraphQLOptions, additionalHeaders?: {
+    graphql<T = any>(amplify: AmplifyClassV6 | (() => Promise<AmplifyClassV6>), options: GraphQLOptions, additionalHeaders?: {
         [key: string]: string;
     }): Observable<GraphQLResult<T>> | Promise<GraphQLResult<T>>;
     /**