@aws-amplify/api-graphql 4.0.1-api-v6-models.3eb7748.0 → 4.0.1-api-v6-models.01b2bb8.0

package/lib/internals/APIClient.d.ts

@@ -42,6 +42,10 @@ export declare const graphQLOperationsInfo: {
         operationPrefix: "onDelete";
         usePlural: boolean;
     };
+    OBSERVE_QUERY: {
+        operationPrefix: "observeQuery";
+        usePlural: boolean;
+    };
 };
 export type ModelOperation = keyof typeof graphQLOperationsInfo;
 /**

package/lib/internals/APIClient.js

@@ -145,9 +145,10 @@ exports.graphQLOperationsInfo = {
     ONCREATE: { operationPrefix: 'onCreate', usePlural: false },
     ONUPDATE: { operationPrefix: 'onUpdate', usePlural: false },
     ONDELETE: { operationPrefix: 'onDelete', usePlural: false },
+    OBSERVE_QUERY: { operationPrefix: 'observeQuery', usePlural: false },
 };
 var graphQLDocumentsCache = new Map();
-var SELECTION_SET_ALL_NESTED = '*';
+var SELECTION_SET_WILDCARD = '*';
 function defaultSelectionSetForModel(modelDefinition) {
     // fields that are explicitly part of the graphql schema; not
     // inferred from owner auth rules.
@@ -184,71 +185,54 @@ var FIELD_IR = '';
  * ```
  */
 function customSelectionSetToIR(modelIntrospection, modelName, selectionSet) {
-    var e_2, _a;
-    var _b, _c, _d, _e;
     var modelDefinition = modelIntrospection[modelName];
-    var fields = modelDefinition.fields;
-    var intermediateSelectionSet = {};
-    try {
-        for (var selectionSet_1 = tslib_1.__values(selectionSet), selectionSet_1_1 = selectionSet_1.next(); !selectionSet_1_1.done; selectionSet_1_1 = selectionSet_1.next()) {
-            var f = selectionSet_1_1.value;
-            var nested = f.includes('.');
-            if (nested) {
-                var _f = tslib_1.__read(f.split('.'), 2), modelFieldName = _f[0], selectedField = _f[1];
-                var relatedModel = (_c = (_b = fields[modelFieldName]) === null || _b === void 0 ? void 0 : _b.type) === null || _c === void 0 ? void 0 : _c.model;
-                if (!relatedModel) {
-                    // TODO: may need to change this to support custom types
-                    throw Error("".concat(modelFieldName, " is not a model field"));
-                }
-                if (selectedField === SELECTION_SET_ALL_NESTED) {
-                    var relatedModelDefinition = modelIntrospection[relatedModel];
-                    var defaultSelectionSet = defaultSelectionSetForModel(relatedModelDefinition);
-                    var reduced = defaultSelectionSet.reduce(function (acc, curVal) {
-                        acc[curVal] = FIELD_IR;
-                        return acc;
-                    }, {});
-                    if ((_d = fields[modelFieldName]) === null || _d === void 0 ? void 0 : _d.isArray) {
-                        intermediateSelectionSet[modelFieldName] = {
-                            items: reduced,
-                        };
-                    }
-                    else {
-                        intermediateSelectionSet[modelFieldName] = reduced;
-                    }
-                }
-                else {
-                    var getNestedSelSet = customSelectionSetToIR(modelIntrospection, relatedModel, [selectedField]);
-                    if ((_e = fields[modelFieldName]) === null || _e === void 0 ? void 0 : _e.isArray) {
-                        var existing = intermediateSelectionSet[modelFieldName] || { items: {} };
-                        var merged = tslib_1.__assign(tslib_1.__assign({}, existing.items), getNestedSelSet);
-                        intermediateSelectionSet[modelFieldName] = { items: merged };
-                    }
-                    else {
-                        var existingItems = intermediateSelectionSet[modelFieldName] || {};
-                        var merged = tslib_1.__assign(tslib_1.__assign({}, existingItems), getNestedSelSet);
-                        intermediateSelectionSet[modelFieldName] = merged;
-                    }
-                }
+    var modelFields = modelDefinition.fields;
+    return selectionSet.reduce(function (resultObj, path) {
+        var _a, _b, _c;
+        var _d = tslib_1.__read(path.split('.')), fieldName = _d[0], nested = _d[1], rest = _d.slice(2);
+        if (nested) {
+            var relatedModel = (_b = (_a = modelFields[fieldName]) === null || _a === void 0 ? void 0 : _a.type) === null || _b === void 0 ? void 0 : _b.model;
+            if (!relatedModel) {
+                // TODO: may need to change this to support custom types
+                throw Error("".concat(fieldName, " is not a model field"));
             }
-            else {
-                var exists = Boolean(fields[f]);
-                if (!exists) {
-                    throw Error("".concat(f, " is not a field of model ").concat(modelName));
-                }
-                intermediateSelectionSet[f] = FIELD_IR;
+            var relatedModelDefinition = modelIntrospection[relatedModel];
+            var selectionSet_1 = nested === SELECTION_SET_WILDCARD
+                ? defaultSelectionSetIR(relatedModelDefinition)
+                : // if we have a path like 'field.anotherField' recursively build up selection set IR
+                    customSelectionSetToIR(modelIntrospection, relatedModel, [
+                        tslib_1.__spreadArray([nested], tslib_1.__read(rest), false).join('.'),
+                    ]);
+            if ((_c = modelFields[fieldName]) === null || _c === void 0 ? void 0 : _c.isArray) {
+                var existing = resultObj[fieldName] || {
+                    items: {},
+                };
+                var merged_1 = tslib_1.__assign(tslib_1.__assign({}, existing.items), selectionSet_1);
+                resultObj[fieldName] = { items: merged_1 };
+                return resultObj;
             }
+            var existingItems = resultObj[fieldName] || {};
+            var merged = tslib_1.__assign(tslib_1.__assign({}, existingItems), selectionSet_1);
+            resultObj[fieldName] = merged;
+            return resultObj;
         }
-    }
-    catch (e_2_1) { e_2 = { error: e_2_1 }; }
-    finally {
-        try {
-            if (selectionSet_1_1 && !selectionSet_1_1.done && (_a = selectionSet_1.return)) _a.call(selectionSet_1);
+        var exists = Boolean(modelFields[fieldName]);
+        if (!exists) {
+            throw Error("".concat(fieldName, " is not a field of model ").concat(modelName));
         }
-        finally { if (e_2) throw e_2.error; }
-    }
-    return intermediateSelectionSet;
+        resultObj[fieldName] = FIELD_IR;
+        return resultObj;
+    }, {});
 }
 exports.customSelectionSetToIR = customSelectionSetToIR;
+var defaultSelectionSetIR = function (relatedModelDefinition) {
+    var defaultSelectionSet = defaultSelectionSetForModel(relatedModelDefinition);
+    var reduced = defaultSelectionSet.reduce(function (acc, curVal) {
+        acc[curVal] = FIELD_IR;
+        return acc;
+    }, {});
+    return reduced;
+};
 /**
  * Stringifies selection set IR
  * * @example
@@ -334,9 +318,11 @@ function generateGraphQLDocument(modelIntrospection, modelName, modelOperation,
         case 'LIST':
             graphQLArguments !== null && graphQLArguments !== void 0 ? graphQLArguments : (graphQLArguments = {
                 filter: "Model".concat(name, "FilterInput"),
+                limit: 'Int',
+                nextToken: 'String',
             });
             graphQLOperationType !== null && graphQLOperationType !== void 0 ? graphQLOperationType : (graphQLOperationType = 'query');
-            graphQLSelectionSet !== null && graphQLSelectionSet !== void 0 ? graphQLSelectionSet : (graphQLSelectionSet = "items { ".concat(selectionSetFields, " }"));
+            graphQLSelectionSet !== null && graphQLSelectionSet !== void 0 ? graphQLSelectionSet : (graphQLSelectionSet = "items { ".concat(selectionSetFields, " } nextToken __typename"));
         case 'ONCREATE':
         case 'ONUPDATE':
         case 'ONDELETE':
@@ -345,6 +331,10 @@ function generateGraphQLDocument(modelIntrospection, modelName, modelOperation,
             });
             graphQLOperationType !== null && graphQLOperationType !== void 0 ? graphQLOperationType : (graphQLOperationType = 'subscription');
             graphQLSelectionSet !== null && graphQLSelectionSet !== void 0 ? graphQLSelectionSet : (graphQLSelectionSet = selectionSetFields);
+            break;
+        case 'OBSERVE_QUERY':
+        default:
+            throw new Error('Internal error: Attempted to generate graphql document for observeQuery. Please report this error.');
     }
     var graphQLDocument = "".concat(graphQLOperationType).concat(graphQLArguments
         ? "(".concat(Object.entries(graphQLArguments).map(function (_a) {
@@ -393,6 +383,13 @@ function buildGraphQLVariables(modelDefinition, operation, arg, modelIntrospecti
             }
             break;
         case 'LIST':
+            if (arg === null || arg === void 0 ? void 0 : arg.filter) {
+                variables.filter = arg.filter;
+            }
+            if (arg === null || arg === void 0 ? void 0 : arg.nextToken) {
+                variables.nextToken = arg.nextToken;
+            }
+            break;
         case 'ONCREATE':
         case 'ONUPDATE':
         case 'ONDELETE':
@@ -400,6 +397,9 @@ function buildGraphQLVariables(modelDefinition, operation, arg, modelIntrospecti
                 variables = { filter: arg.filter };
             }
             break;
+        case 'OBSERVE_QUERY':
+            throw new Error('Internal error: Attempted to build variables for observeQuery. Please report this error.');
+            break;
         default:
             var exhaustiveCheck = operation;
             throw new Error("Unhandled operation case: ".concat(exhaustiveCheck));

package/lib/internals/generateModelsProperty.js

@@ -3,6 +3,7 @@ Object.defineProperty(exports, "__esModule", { value: true });
 exports.generateModelsProperty = void 0;
 var tslib_1 = require("tslib");
 var APIClient_1 = require("./APIClient");
+var utils_1 = require("../utils");
 var rxjs_1 = require("rxjs");
 function generateModelsProperty(client, params) {
     var e_1, _a;
@@ -22,7 +23,7 @@ function generateModelsProperty(client, params) {
             var operation = key;
             if (operation === 'LIST') {
                 models[name_1][operationPrefix] = function (args) { return tslib_1.__awaiter(_this, void 0, void 0, function () {
-                    var query, variables, _a, data, nextToken, extensions, _b, key_1, flattenedResult, initialized, error_1;
+                    var query, variables, _a, data, extensions, _b, key_1, flattenedResult, initialized, error_1;
                     return tslib_1.__generator(this, function (_c) {
                         switch (_c.label) {
                             case 0:
@@ -36,7 +37,7 @@ function generateModelsProperty(client, params) {
                                         variables: variables,
                                     })];
                             case 2:
-                                _a = (_c.sent()), data = _a.data, nextToken = _a.nextToken, extensions = _a.extensions;
+                                _a = (_c.sent()), data = _a.data, extensions = _a.extensions;
                                 // flatten response
                                 if (data !== undefined) {
                                     _b = tslib_1.__read(Object.keys(data), 1), key_1 = _b[0];
@@ -46,7 +47,7 @@ function generateModelsProperty(client, params) {
                                         if (args === null || args === void 0 ? void 0 : args.selectionSet) {
                                             return [2 /*return*/, {
                                                     data: flattenedResult,
-                                                    nextToken: nextToken,
+                                                    nextToken: data[key_1].nextToken,
                                                     extensions: extensions,
                                                 }];
                                         }
@@ -54,14 +55,14 @@ function generateModelsProperty(client, params) {
                                             initialized = (0, APIClient_1.initializeModel)(client, name_1, flattenedResult, modelIntrospection);
                                             return [2 /*return*/, {
                                                     data: initialized,
-                                                    nextToken: nextToken,
+                                                    nextToken: data[key_1].nextToken,
                                                     extensions: extensions,
                                                 }];
                                         }
                                     }
                                     return [2 /*return*/, {
                                             data: data[key_1],
-                                            nextToken: nextToken,
+                                            nextToken: data[key_1].nextToken,
                                             extensions: extensions,
                                         }];
                                 }
@@ -96,6 +97,163 @@ function generateModelsProperty(client, params) {
                     }));
                 };
             }
+            else if (operation === 'OBSERVE_QUERY') {
+                models[name_1][operationPrefix] = function (arg, options) {
+                    return new rxjs_1.Observable(function (subscriber) {
+                        // what we'll be sending to our subscribers
+                        var items = [];
+                        // To enqueue subscription messages while we collect our initial
+                        // result set.
+                        var messageQueue = [];
+                        // operation to take when message(s) arrive.
+                        // this operation will be swapped out once initial "sync" is complete
+                        // to immediately ingest messsages.
+                        var receiveMessages = function () {
+                            var messages = [];
+                            for (var _i = 0; _i < arguments.length; _i++) {
+                                messages[_i] = arguments[_i];
+                            }
+                            return messageQueue.push.apply(messageQueue, tslib_1.__spreadArray([], tslib_1.__read(messages), false));
+                        };
+                        // start subscriptions
+                        var onCreateSub = models[name_1]
+                            .onCreate(arg, options)
+                            .subscribe({
+                            next: function (item) {
+                                receiveMessages({ item: item, type: 'create' });
+                            },
+                            error: function (error) {
+                                subscriber.error({ type: 'onCreate', error: error });
+                            },
+                        });
+                        var onUpdateSub = models[name_1]
+                            .onUpdate(arg, options)
+                            .subscribe({
+                            next: function (item) {
+                                receiveMessages({ item: item, type: 'update' });
+                            },
+                            error: function (error) {
+                                subscriber.error({ type: 'onUpdate', error: error });
+                            },
+                        });
+                        var onDeleteSub = models[name_1]
+                            .onDelete(arg, options)
+                            .subscribe({
+                            next: function (item) {
+                                receiveMessages({ item: item, type: 'delete' });
+                            },
+                            error: function (error) {
+                                subscriber.error({ type: 'onDelete', error: error });
+                            },
+                        });
+                        // consumes a list of messages and sends a snapshot
+                        function ingestMessages(messages) {
+                            var e_2, _a;
+                            try {
+                                for (var messages_1 = tslib_1.__values(messages), messages_1_1 = messages_1.next(); !messages_1_1.done; messages_1_1 = messages_1.next()) {
+                                    var message = messages_1_1.value;
+                                    var idx = (0, utils_1.findIndexByFields)(message.item, items, pkFields);
+                                    switch (message.type) {
+                                        case 'create':
+                                            if (idx < 0)
+                                                items.push(message.item);
+                                            break;
+                                        case 'update':
+                                            if (idx >= 0)
+                                                items[idx] = message.item;
+                                            break;
+                                        case 'delete':
+                                            if (idx >= 0)
+                                                items.splice(idx, 1);
+                                            break;
+                                        default:
+                                            console.error('Unrecognized message in observeQuery.', message);
+                                    }
+                                }
+                            }
+                            catch (e_2_1) { e_2 = { error: e_2_1 }; }
+                            finally {
+                                try {
+                                    if (messages_1_1 && !messages_1_1.done && (_a = messages_1.return)) _a.call(messages_1);
+                                }
+                                finally { if (e_2) throw e_2.error; }
+                            }
+                            subscriber.next({
+                                items: items,
+                                isSynced: true,
+                            });
+                        }
+                        var pkFields = (0, utils_1.resolvePKFields)(model);
+                        // initial results
+                        (function () { return tslib_1.__awaiter(_this, void 0, void 0, function () {
+                            var firstPage, nextToken, _a, page, errors, _nextToken, isSynced, errors_1, errors_1_1, error;
+                            var e_3, _b;
+                            return tslib_1.__generator(this, function (_c) {
+                                switch (_c.label) {
+                                    case 0:
+                                        firstPage = true;
+                                        nextToken = null;
+                                        _c.label = 1;
+                                    case 1:
+                                        if (!(!subscriber.closed && (firstPage || nextToken))) return [3 /*break*/, 3];
+                                        firstPage = false;
+                                        return [4 /*yield*/, models[name_1].list(tslib_1.__assign(tslib_1.__assign({}, arg), { nextToken: nextToken }), options)];
+                                    case 2:
+                                        _a = _c.sent(), page = _a.data, errors = _a.errors, _nextToken = _a.nextToken;
+                                        nextToken = _nextToken;
+                                        items.push.apply(items, tslib_1.__spreadArray([], tslib_1.__read(page), false));
+                                        isSynced = messageQueue.length === 0 &&
+                                            (nextToken === null || nextToken === undefined);
+                                        subscriber.next({
+                                            items: items,
+                                            isSynced: isSynced,
+                                        });
+                                        if (Array.isArray(errors)) {
+                                            try {
+                                                for (errors_1 = (e_3 = void 0, tslib_1.__values(errors)), errors_1_1 = errors_1.next(); !errors_1_1.done; errors_1_1 = errors_1.next()) {
+                                                    error = errors_1_1.value;
+                                                    subscriber.error(error);
+                                                }
+                                            }
+                                            catch (e_3_1) { e_3 = { error: e_3_1 }; }
+                                            finally {
+                                                try {
+                                                    if (errors_1_1 && !errors_1_1.done && (_b = errors_1.return)) _b.call(errors_1);
+                                                }
+                                                finally { if (e_3) throw e_3.error; }
+                                            }
+                                        }
+                                        return [3 /*break*/, 1];
+                                    case 3:
+                                        // play through the queue
+                                        if (messageQueue.length > 0) {
+                                            ingestMessages(messageQueue);
+                                        }
+                                        // switch the queue to write directly to the items collection
+                                        receiveMessages = function () {
+                                            var messages = [];
+                                            for (var _i = 0; _i < arguments.length; _i++) {
+                                                messages[_i] = arguments[_i];
+                                            }
+                                            ingestMessages(messages);
+                                            return items.length;
+                                        };
+                                        return [2 /*return*/];
+                                }
+                            });
+                        }); })();
+                        // when subscriber unsubscribes, tear down internal subs
+                        return function () {
+                            // 1. tear down internal subs
+                            onCreateSub.unsubscribe();
+                            onUpdateSub.unsubscribe();
+                            onDeleteSub.unsubscribe();
+                            // 2. there is no need to explicitly stop paging. instead, we
+                            // just check `subscriber.closed` above before fetching each page.
+                        };
+                    });
+                };
+            }
             else {
                 models[name_1][operationPrefix] = function (arg, options) { return tslib_1.__awaiter(_this, void 0, void 0, function () {
                     var query, variables, _a, data, extensions, _b, key_2, flattenedResult, _c, initialized, error_2;