@aws-amplify/api-graphql 4.0.1-console-preview.be08038.0 → 4.0.1-server-generate-client.1e317a1.0

package/lib-esm/internals/generateModelsProperty.js

@@ -0,0 +1,165 @@
+import { __awaiter, __generator, __read, __values } from "tslib";
+import { initializeModel, generateGraphQLDocument, buildGraphQLVariables, graphQLOperationsInfo, flattenItems, } from './APIClient';
+import { map } from 'rxjs';
+import { Amplify } from '@aws-amplify/core';
+export function generateModelsProperty(client, params) {
+    var e_1, _a;
+    var _this = this;
+    var _b, _c, _d;
+    var models = {};
+    // params.config gets populated in SSR only; otherwise we get it from the singleton
+    var config = (_b = params.config) !== null && _b !== void 0 ? _b : Amplify.getConfig();
+    var modelIntrospection = (_d = (_c = config.API) === null || _c === void 0 ? void 0 : _c.GraphQL) === null || _d === void 0 ? void 0 : _d.modelIntrospection;
+    if (!modelIntrospection) {
+        return {};
+    }
+    var _loop_1 = function (model) {
+        var name_1 = model.name;
+        models[name_1] = {};
+        Object.entries(graphQLOperationsInfo).forEach(function (_a) {
+            var _b = __read(_a, 2), key = _b[0], operationPrefix = _b[1].operationPrefix;
+            var operation = key;
+            if (operation === 'LIST') {
+                models[name_1][operationPrefix] = function (args) { return __awaiter(_this, void 0, void 0, function () {
+                    var query, variables, _a, data, nextToken, extensions, _b, key_1, flattenedResult, initialized, error_1;
+                    return __generator(this, function (_c) {
+                        switch (_c.label) {
+                            case 0:
+                                query = generateGraphQLDocument(modelIntrospection.models, name_1, 'LIST', args);
+                                variables = buildGraphQLVariables(model, 'LIST', args, modelIntrospection);
+                                _c.label = 1;
+                            case 1:
+                                _c.trys.push([1, 3, , 4]);
+                                return [4 /*yield*/, client.graphql({
+                                        query: query,
+                                        variables: variables,
+                                    })];
+                            case 2:
+                                _a = (_c.sent()), data = _a.data, nextToken = _a.nextToken, extensions = _a.extensions;
+                                // flatten response
+                                if (data !== undefined) {
+                                    _b = __read(Object.keys(data), 1), key_1 = _b[0];
+                                    if (data[key_1].items) {
+                                        flattenedResult = flattenItems(data)[key_1];
+                                        // don't init if custom selection set
+                                        if (args === null || args === void 0 ? void 0 : args.selectionSet) {
+                                            return [2 /*return*/, {
+                                                    data: flattenedResult,
+                                                    nextToken: nextToken,
+                                                    extensions: extensions,
+                                                }];
+                                        }
+                                        else {
+                                            initialized = initializeModel(client, name_1, flattenedResult, modelIntrospection);
+                                            return [2 /*return*/, {
+                                                    data: initialized,
+                                                    nextToken: nextToken,
+                                                    extensions: extensions,
+                                                }];
+                                        }
+                                    }
+                                    return [2 /*return*/, {
+                                            data: data[key_1],
+                                            nextToken: nextToken,
+                                            extensions: extensions,
+                                        }];
+                                }
+                                return [3 /*break*/, 4];
+                            case 3:
+                                error_1 = _c.sent();
+                                if (error_1.errors) {
+                                    // graphql errors pass through
+                                    return [2 /*return*/, error_1];
+                                }
+                                else {
+                                    // non-graphql errors re re-thrown
+                                    throw error_1;
+                                }
+                                return [3 /*break*/, 4];
+                            case 4: return [2 /*return*/];
+                        }
+                    });
+                }); };
+            }
+            else if (['ONCREATE', 'ONUPDATE', 'ONDELETE'].includes(operation)) {
+                models[name_1][operationPrefix] = function (arg, options) {
+                    var query = generateGraphQLDocument(modelIntrospection.models, name_1, operation);
+                    var variables = buildGraphQLVariables(model, operation, arg, modelIntrospection);
+                    var observable = client.graphql({
+                        query: query,
+                        variables: variables,
+                    });
+                    return observable.pipe(map(function (value) {
+                        var _a = __read(Object.keys(value.data), 1), key = _a[0];
+                        return value.data[key];
+                    }));
+                };
+            }
+            else {
+                models[name_1][operationPrefix] = function (arg, options) { return __awaiter(_this, void 0, void 0, function () {
+                    var query, variables, _a, data, extensions, _b, key_2, flattenedResult, _c, initialized, error_2;
+                    return __generator(this, function (_d) {
+                        switch (_d.label) {
+                            case 0:
+                                query = generateGraphQLDocument(modelIntrospection.models, name_1, operation, options);
+                                variables = buildGraphQLVariables(model, operation, arg, modelIntrospection);
+                                _d.label = 1;
+                            case 1:
+                                _d.trys.push([1, 3, , 4]);
+                                return [4 /*yield*/, client.graphql({
+                                        query: query,
+                                        variables: variables,
+                                    })];
+                            case 2:
+                                _a = (_d.sent()), data = _a.data, extensions = _a.extensions;
+                                // flatten response
+                                if (data !== undefined) {
+                                    _b = __read(Object.keys(data), 1), key_2 = _b[0];
+                                    flattenedResult = flattenItems(data)[key_2];
+                                    if (options === null || options === void 0 ? void 0 : options.selectionSet) {
+                                        return [2 /*return*/, { data: flattenedResult, extensions: extensions }];
+                                    }
+                                    else {
+                                        _c = __read(initializeModel(client, name_1, [flattenedResult], modelIntrospection), 1), initialized = _c[0];
+                                        return [2 /*return*/, { data: initialized, extensions: extensions }];
+                                    }
+                                }
+                                else {
+                                    return [2 /*return*/, { data: null, extensions: extensions }];
+                                }
+                                return [3 /*break*/, 4];
+                            case 3:
+                                error_2 = _d.sent();
+                                if (error_2.errors) {
+                                    // graphql errors pass through
+                                    return [2 /*return*/, error_2];
+                                }
+                                else {
+                                    // non-graphql errors re re-thrown
+                                    throw error_2;
+                                }
+                                return [3 /*break*/, 4];
+                            case 4: return [2 /*return*/];
+                        }
+                    });
+                }); };
+            }
+        });
+    };
+    try {
+        // TODO: refactor this to use separate methods for each CRUDL.
+        // Doesn't make sense to gen the methods dynamically given the different args and return values
+        for (var _e = __values(Object.values(modelIntrospection.models)), _f = _e.next(); !_f.done; _f = _e.next()) {
+            var model = _f.value;
+            _loop_1(model);
+        }
+    }
+    catch (e_1_1) { e_1 = { error: e_1_1 }; }
+    finally {
+        try {
+            if (_f && !_f.done && (_a = _e.return)) _a.call(_e);
+        }
+        finally { if (e_1) throw e_1.error; }
+    }
+    return models;
+}

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

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

package/lib-esm/internals/index.js

@@ -2,3 +2,4 @@
 // SPDX-License-Identifier: Apache-2.0
 export { InternalGraphQLAPI, InternalGraphQLAPIClass, } from './InternalGraphQLAPI';
 export { graphql, cancel, isCancelError } from './v6';
+export { generateClient } from './generateClient';

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

@@ -0,0 +1,10 @@
+import { AmplifyClassV6 } from '@aws-amplify/core';
+/**
+ * @private
+ *
+ * The knobs available for configuring `generateClient` internally.
+ */
+export type ClientGenerationParams = {
+    amplify: AmplifyClassV6 | ((fn: (amplify: any) => Promise<any>) => Promise<AmplifyClassV6>);
+    config?: object;
+};

package/lib-esm/internals/types.js

@@ -0,0 +1 @@
+export {};