@aws-amplify/graphql-api-construct 1.3.0 → 1.4.0-nov-14-cut.0

package/node_modules/@aws-amplify/graphql-index-transformer/lib/resolvers/resolvers.js

@@ -23,21 +23,26 @@ var __importStar = (this && this.__importStar) || function (mod) {
     return result;
 };
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.getVTLGenerator = exports.getDBType = exports.getDBInfo = exports.generateAuthExpressionForSandboxMode = exports.constructSyncVTL = exports.addIndexToResolverSlot = exports.mergeInputsAndDefaultsSnippet = exports.updateResolversForIndex = exports.appendSecondaryIndex = exports.validateSortDirectionInput = exports.setQuerySnippet = exports.ensureCompositeKeySnippet = exports.getResolverObject = exports.attributeTypeFromType = exports.updateResolvers = exports.replaceDdbPrimaryKey = void 0;
+exports.getVTLGenerator = exports.getDBType = exports.getDBInfo = exports.generateAuthExpressionForSandboxMode = exports.constructSyncVTL = exports.addIndexToResolverSlot = exports.mergeInputsAndDefaultsSnippet = exports.makeQueryResolver = exports.updateResolversForIndex = exports.overrideIndexAtCfnLevel = exports.appendSecondaryIndex = exports.validateSortDirectionInput = exports.setQuerySnippet = exports.ensureCompositeKeySnippet = exports.getResolverObject = exports.attributeTypeFromType = exports.updateResolvers = exports.replaceDdbPrimaryKey = void 0;
 const graphql_model_transformer_1 = require("@aws-amplify/graphql-model-transformer");
 const graphql_transformer_core_1 = require("@aws-amplify/graphql-transformer-core");
 const cdk = __importStar(require("aws-cdk-lib"));
 const graphql_1 = require("graphql");
 const graphql_mapping_template_1 = require("graphql-mapping-template");
 const graphql_transformer_common_1 = require("graphql-transformer-common");
+const graphql_transformer_core_2 = require("@aws-amplify/graphql-transformer-core");
 const utils_1 = require("../utils");
 const generators_1 = require("./generators");
 const API_KEY = 'API Key Authorization';
-function replaceDdbPrimaryKey(config, ctx) {
+const replaceDdbPrimaryKey = (config, ctx) => {
     var _a, _b, _c;
     const { field, object } = config;
+    const tableProvisionStrategy = (0, graphql_transformer_core_1.getDatasourceProvisionStrategy)(ctx, object.name.value);
+    const useAmplifyManagedTableResources = tableProvisionStrategy
+        ? tableProvisionStrategy === "AMPLIFY_TABLE"
+        : false;
     const table = getTable(ctx, object);
-    const cfnTable = table.table;
+    const cfnTable = useAmplifyManagedTableResources ? table.node.defaultChild.node.defaultChild : table.table;
     const tableAttrDefs = table.attributeDefinitions;
     const tableKeySchema = table.keySchema;
     const keySchema = getDdbKeySchema(config);
@@ -58,49 +63,55 @@ function replaceDdbPrimaryKey(config, ctx) {
             table.attributeDefinitions.push(attr);
         }
     }
-    cfnTable.keySchema = table.keySchema;
-    cfnTable.attributeDefinitions = table.attributeDefinitions;
-}
+    if (useAmplifyManagedTableResources) {
+        cfnTable.addPropertyOverride('keySchema', table.keySchema);
+        cfnTable.addPropertyOverride('attributeDefinitions', table.attributeDefinitions);
+    }
+    else {
+        cfnTable.keySchema = table.keySchema;
+        cfnTable.attributeDefinitions = table.attributeDefinitions;
+    }
+};
 exports.replaceDdbPrimaryKey = replaceDdbPrimaryKey;
-function updateResolvers(config, ctx, resolverMap) {
-    const getResolver = getResolverObject(config, ctx, 'get');
-    const listResolver = getResolverObject(config, ctx, 'list');
-    const createResolver = getResolverObject(config, ctx, 'create');
-    const updateResolver = getResolverObject(config, ctx, 'update');
-    const deleteResolver = getResolverObject(config, ctx, 'delete');
-    const syncResolver = getResolverObject(config, ctx, 'sync');
+const updateResolvers = (config, ctx, resolverMap) => {
+    const getResolver = (0, exports.getResolverObject)(config, ctx, 'get');
+    const listResolver = (0, exports.getResolverObject)(config, ctx, 'list');
+    const createResolver = (0, exports.getResolverObject)(config, ctx, 'create');
+    const updateResolver = (0, exports.getResolverObject)(config, ctx, 'update');
+    const deleteResolver = (0, exports.getResolverObject)(config, ctx, 'delete');
+    const syncResolver = (0, exports.getResolverObject)(config, ctx, 'sync');
     if (getResolver) {
-        addIndexToResolverSlot(getResolver, [setPrimaryKeySnippet(config, false)]);
+        (0, exports.addIndexToResolverSlot)(getResolver, [setPrimaryKeySnippet(config, false)]);
     }
     if (listResolver) {
-        addIndexToResolverSlot(listResolver, [
-            (0, graphql_mapping_template_1.print)(setQuerySnippet(config, ctx, true)),
+        (0, exports.addIndexToResolverSlot)(listResolver, [
+            (0, graphql_mapping_template_1.print)((0, exports.setQuerySnippet)(config, ctx, true)),
             `$util.qr($ctx.stash.put("${graphql_transformer_common_1.ResourceConstants.SNIPPETS.ModelQueryExpression}", $${graphql_transformer_common_1.ResourceConstants.SNIPPETS.ModelQueryExpression}))`,
         ]);
     }
     if (createResolver) {
-        addIndexToResolverSlot(createResolver, [
-            mergeInputsAndDefaultsSnippet(),
+        (0, exports.addIndexToResolverSlot)(createResolver, [
+            (0, exports.mergeInputsAndDefaultsSnippet)(),
             setPrimaryKeySnippet(config, true),
-            ensureCompositeKeySnippet(config, false),
+            (0, exports.ensureCompositeKeySnippet)(config, false),
         ]);
     }
     if (updateResolver) {
-        addIndexToResolverSlot(updateResolver, [
-            mergeInputsAndDefaultsSnippet(),
+        (0, exports.addIndexToResolverSlot)(updateResolver, [
+            (0, exports.mergeInputsAndDefaultsSnippet)(),
             setPrimaryKeySnippet(config, true),
-            ensureCompositeKeySnippet(config, false),
+            (0, exports.ensureCompositeKeySnippet)(config, false),
         ]);
     }
     if (deleteResolver) {
-        addIndexToResolverSlot(deleteResolver, [mergeInputsAndDefaultsSnippet(), setPrimaryKeySnippet(config, true)]);
+        (0, exports.addIndexToResolverSlot)(deleteResolver, [(0, exports.mergeInputsAndDefaultsSnippet)(), setPrimaryKeySnippet(config, true)]);
     }
     if (syncResolver) {
         makeSyncResolver('dbTable', config, ctx, syncResolver, resolverMap);
     }
-}
+};
 exports.updateResolvers = updateResolvers;
-function getTable(context, object) {
+const getTable = (context, object) => {
     const ddbDataSource = context.dataSources.get(object);
     const tableName = graphql_transformer_common_1.ModelResourceIDs.ModelTableResourceID(object.name.value);
     const table = ddbDataSource.ds.stack.node.findChild(tableName);
@@ -108,30 +119,30 @@ function getTable(context, object) {
         throw new Error(`Table not found in stack with table name ${tableName}`);
     }
     return table;
-}
-function getDdbKeySchema(config) {
+};
+const getDdbKeySchema = (config) => {
     const schema = [{ attributeName: config.field.name.value, keyType: 'HASH' }];
     if (config.sortKey.length > 0) {
         schema.push({ attributeName: getSortKeyName(config), keyType: 'RANGE' });
     }
     return schema;
-}
-function attributeTypeFromType(type, ctx) {
+};
+const attributeTypeFromType = (type, ctx) => {
     const baseTypeName = (0, graphql_transformer_common_1.getBaseType)(type);
     const ofType = ctx.output.getType(baseTypeName);
     if (ofType && ofType.kind === graphql_1.Kind.ENUM_TYPE_DEFINITION) {
         return 'S';
     }
     return (0, graphql_transformer_common_1.attributeTypeFromScalar)(type);
-}
+};
 exports.attributeTypeFromType = attributeTypeFromType;
-function attributeDefinitions(config, ctx) {
+const attributeDefinitions = (config, ctx) => {
     const { field, sortKey, sortKeyFields } = config;
-    const definitions = [{ attributeName: field.name.value, attributeType: attributeTypeFromType(field.type, ctx) }];
+    const definitions = [{ attributeName: field.name.value, attributeType: (0, exports.attributeTypeFromType)(field.type, ctx) }];
     if (sortKeyFields.length === 1) {
         definitions.push({
             attributeName: sortKeyFields[0],
-            attributeType: attributeTypeFromType(sortKey[0].type, ctx),
+            attributeType: (0, exports.attributeTypeFromType)(sortKey[0].type, ctx),
         });
     }
     else if (sortKeyFields.length > 1) {
@@ -141,11 +152,11 @@ function attributeDefinitions(config, ctx) {
         });
     }
     return definitions;
-}
-function getSortKeyName(config) {
+};
+const getSortKeyName = (config) => {
     return config.sortKeyFields.join(graphql_transformer_common_1.ModelResourceIDs.ModelCompositeKeySeparator());
-}
-function getResolverObject(config, ctx, op) {
+};
+const getResolverObject = (config, ctx, op) => {
     var _a;
     const resolverName = (0, utils_1.lookupResolverName)(config, ctx, op);
     if (!resolverName) {
@@ -156,15 +167,15 @@ function getResolverObject(config, ctx, op) {
         return null;
     }
     return (_a = ctx.resolvers.getResolver(objectName, resolverName)) !== null && _a !== void 0 ? _a : null;
-}
+};
 exports.getResolverObject = getResolverObject;
-function setPrimaryKeySnippet(config, isMutation) {
+const setPrimaryKeySnippet = (config, isMutation) => {
     const cmds = [
         (0, graphql_mapping_template_1.qref)((0, graphql_mapping_template_1.methodCall)((0, graphql_mapping_template_1.ref)('ctx.stash.metadata.put'), (0, graphql_mapping_template_1.str)(graphql_transformer_common_1.ResourceConstants.SNIPPETS.ModelObjectKey), modelObjectKeySnippet(config, isMutation))),
     ];
     return (0, graphql_mapping_template_1.printBlock)('Set the primary key')((0, graphql_mapping_template_1.compoundExpression)(cmds));
-}
-function modelObjectKeySnippet(config, isMutation) {
+};
+const modelObjectKeySnippet = (config, isMutation) => {
     const { field, sortKeyFields } = config;
     const argsPrefix = isMutation ? 'mergedValues' : 'ctx.args';
     const modelObject = {
@@ -181,8 +192,8 @@ function modelObjectKeySnippet(config, isMutation) {
         modelObject[sortKeyFields[0]] = (0, graphql_mapping_template_1.ref)(`util.dynamodb.toDynamoDB($${argsPrefix}.${sortKeyFields[0]})`);
     }
     return (0, graphql_mapping_template_1.obj)(modelObject);
-}
-function ensureCompositeKeySnippet(config, conditionallySetSortKey) {
+};
+const ensureCompositeKeySnippet = (config, conditionallySetSortKey) => {
     const { sortKeyFields } = config;
     if (sortKeyFields.length < 2) {
         return '';
@@ -199,19 +210,19 @@ function ensureCompositeKeySnippet(config, conditionallySetSortKey) {
             ? (0, graphql_mapping_template_1.iff)((0, graphql_mapping_template_1.ref)(graphql_transformer_common_1.ResourceConstants.SNIPPETS.HasSeenSomeKeyArg), (0, graphql_mapping_template_1.qref)(`$ctx.args.input.put('${condensedSortKey}',"${condensedSortKeyValue}")`))
             : (0, graphql_mapping_template_1.qref)(`$ctx.args.input.put('${condensedSortKey}',"${condensedSortKeyValue}")`),
     ]));
-}
+};
 exports.ensureCompositeKeySnippet = ensureCompositeKeySnippet;
-function setQuerySnippet(config, ctx, isListResolver) {
+const setQuerySnippet = (config, ctx, isListResolver) => {
     const { field, sortKey, sortKeyFields } = config;
     const keyFields = [field, ...sortKey];
     const keyNames = [field.name.value, ...sortKeyFields];
-    const keyTypes = keyFields.map((k) => attributeTypeFromType(k.type, ctx));
-    const expressions = validateSortDirectionInput(config, isListResolver);
+    const keyTypes = keyFields.map((k) => (0, exports.attributeTypeFromType)(k.type, ctx));
+    const expressions = (0, exports.validateSortDirectionInput)(config, isListResolver);
     expressions.push((0, graphql_mapping_template_1.set)((0, graphql_mapping_template_1.ref)(graphql_transformer_common_1.ResourceConstants.SNIPPETS.ModelQueryExpression), (0, graphql_mapping_template_1.obj)({})), (0, graphql_transformer_common_1.applyKeyExpressionForCompositeKey)(keyNames, keyTypes, graphql_transformer_common_1.ResourceConstants.SNIPPETS.ModelQueryExpression));
     return (0, graphql_mapping_template_1.block)('Set query expression for key', expressions);
-}
+};
 exports.setQuerySnippet = setQuerySnippet;
-function validateSortDirectionInput(config, isListResolver) {
+const validateSortDirectionInput = (config, isListResolver) => {
     const { field, sortKeyFields } = config;
     const keyNames = [field.name.value, ...sortKeyFields];
     const expressions = [];
@@ -224,13 +235,12 @@ function validateSortDirectionInput(config, isListResolver) {
         expressions.push(sortDirectionValidation);
     }
     return expressions;
-}
+};
 exports.validateSortDirectionInput = validateSortDirectionInput;
-function appendSecondaryIndex(config, ctx) {
+const appendSecondaryIndex = (config, ctx) => {
     var _a, _b, _c, _d, _e, _f, _g, _h;
     const { name, object, primaryKeyField } = config;
-    const dbType = getDBType(ctx, object.name.value);
-    if (dbType === 'MySQL') {
+    if ((0, graphql_transformer_core_2.isRDSModel)(ctx, object.name.value)) {
         return;
     }
     const table = getTable(ctx, object);
@@ -270,8 +280,7 @@ function appendSecondaryIndex(config, ctx) {
             readCapacity: cdk.Fn.ref(graphql_transformer_common_1.ResourceConstants.PARAMETERS.DynamoDBModelTableReadIOPS),
             writeCapacity: cdk.Fn.ref(graphql_transformer_common_1.ResourceConstants.PARAMETERS.DynamoDBModelTableWriteIOPS),
         });
-        const cfnTable = table.table;
-        cfnTable.globalSecondaryIndexes = appendIndex(cfnTable.globalSecondaryIndexes, {
+        const newIndex = {
             indexName: name,
             keySchema,
             projection: { projectionType: 'ALL' },
@@ -279,55 +288,73 @@ function appendSecondaryIndex(config, ctx) {
                 ReadCapacityUnits: cdk.Fn.ref(graphql_transformer_common_1.ResourceConstants.PARAMETERS.DynamoDBModelTableReadIOPS),
                 WriteCapacityUnits: cdk.Fn.ref(graphql_transformer_common_1.ResourceConstants.PARAMETERS.DynamoDBModelTableWriteIOPS),
             }),
-        });
+        };
+        (0, exports.overrideIndexAtCfnLevel)(ctx, object.name.value, table, newIndex);
     }
-}
+};
 exports.appendSecondaryIndex = appendSecondaryIndex;
-function appendIndex(list, newIndex) {
-    if (Array.isArray(list)) {
-        list.push(newIndex);
-        return list;
+const overrideIndexAtCfnLevel = (ctx, typeName, table, indexInfo) => {
+    const tableProvisionStrategy = (0, graphql_transformer_core_1.getDatasourceProvisionStrategy)(ctx, typeName);
+    const useAmplifyManagedTableResources = tableProvisionStrategy
+        ? tableProvisionStrategy === "AMPLIFY_TABLE"
+        : false;
+    if (!useAmplifyManagedTableResources) {
+        const cfnTable = table.table;
+        cfnTable.globalSecondaryIndexes = appendIndex(cfnTable.globalSecondaryIndexes, indexInfo);
+    }
+    else {
+        const cfnTable = table.table.node.defaultChild;
+        const idx = table.globalSecondaryIndexes.length - 1;
+        cfnTable.addOverride(`Properties.globalSecondaryIndexes.${idx}`, indexInfo);
+    }
+};
+exports.overrideIndexAtCfnLevel = overrideIndexAtCfnLevel;
+const appendIndex = (targetList, newIndex) => {
+    if (Array.isArray(targetList)) {
+        targetList.push(newIndex);
+        return targetList;
     }
     return [newIndex];
-}
-function updateResolversForIndex(config, ctx, resolverMap) {
+};
+const updateResolversForIndex = (config, ctx, resolverMap) => {
     const { name, queryField, object } = config;
     if (!name) {
         throw new Error('Expected name while updating index resolvers.');
     }
-    const createResolver = getResolverObject(config, ctx, 'create');
-    const updateResolver = getResolverObject(config, ctx, 'update');
-    const deleteResolver = getResolverObject(config, ctx, 'delete');
-    const syncResolver = getResolverObject(config, ctx, 'sync');
-    const dbType = getDBType(ctx, object.name.value);
+    const createResolver = (0, exports.getResolverObject)(config, ctx, 'create');
+    const updateResolver = (0, exports.getResolverObject)(config, ctx, 'update');
+    const deleteResolver = (0, exports.getResolverObject)(config, ctx, 'delete');
+    const syncResolver = (0, exports.getResolverObject)(config, ctx, 'sync');
+    const dbType = (0, exports.getDBType)(ctx, object.name.value);
     const isDynamoDB = dbType === 'DDB';
     if (isDynamoDB && createResolver) {
-        const checks = [validateIndexArgumentSnippet(config, 'create'), ensureCompositeKeySnippet(config, true)];
+        const checks = [validateIndexArgumentSnippet(config, 'create'), (0, exports.ensureCompositeKeySnippet)(config, true)];
         if (checks[0] || checks[1]) {
-            addIndexToResolverSlot(createResolver, [mergeInputsAndDefaultsSnippet(), ...checks]);
+            (0, exports.addIndexToResolverSlot)(createResolver, [(0, exports.mergeInputsAndDefaultsSnippet)(), ...checks]);
         }
     }
     if (isDynamoDB && updateResolver) {
-        const checks = [validateIndexArgumentSnippet(config, 'update'), ensureCompositeKeySnippet(config, true)];
+        const checks = [validateIndexArgumentSnippet(config, 'update'), (0, exports.ensureCompositeKeySnippet)(config, true)];
         if (checks[0] || checks[1]) {
-            addIndexToResolverSlot(updateResolver, [mergeInputsAndDefaultsSnippet(), ...checks]);
+            (0, exports.addIndexToResolverSlot)(updateResolver, [(0, exports.mergeInputsAndDefaultsSnippet)(), ...checks]);
         }
     }
     if (isDynamoDB && deleteResolver) {
-        const checks = [ensureCompositeKeySnippet(config, false)];
+        const checks = [(0, exports.ensureCompositeKeySnippet)(config, false)];
         if (checks[0]) {
-            addIndexToResolverSlot(deleteResolver, [mergeInputsAndDefaultsSnippet(), ...checks]);
+            (0, exports.addIndexToResolverSlot)(deleteResolver, [(0, exports.mergeInputsAndDefaultsSnippet)(), ...checks]);
         }
     }
     if (queryField) {
-        makeQueryResolver(config, ctx, dbType);
+        (0, exports.makeQueryResolver)(config, ctx, dbType);
     }
     if (isDynamoDB && syncResolver) {
         makeSyncResolver(name, config, ctx, syncResolver, resolverMap);
     }
-}
+};
 exports.updateResolversForIndex = updateResolversForIndex;
-function makeQueryResolver(config, ctx, dbType) {
+const makeQueryResolver = (config, ctx, dbType) => {
+    var _a;
     const { RDSLambdaDataSourceLogicalID } = graphql_transformer_common_1.ResourceConstants.RESOURCES;
     const isDynamoDB = dbType === graphql_transformer_core_1.DDB_DB_TYPE;
     const { name, object, queryField } = config;
@@ -335,9 +362,9 @@ function makeQueryResolver(config, ctx, dbType) {
         throw new Error('Expected name and queryField to be defined while generating resolver.');
     }
     const modelName = object.name.value;
-    const dbInfo = getDBInfo(ctx, modelName);
+    const dbInfo = (0, exports.getDBInfo)(ctx, modelName);
     let dataSourceName = `${object.name.value}Table`;
-    if (dbType === graphql_transformer_core_1.MYSQL_DB_TYPE) {
+    if (!isDynamoDB) {
         dataSourceName = RDSLambdaDataSourceLogicalID;
     }
     const dataSource = ctx.api.host.getDataSource(dataSourceName);
@@ -357,9 +384,17 @@ function makeQueryResolver(config, ctx, dbType) {
     ])), `${queryTypeName}.${queryField}.res.vtl`));
     resolver.addToSlot('postAuth', graphql_transformer_core_1.MappingTemplate.s3MappingTemplateFromString((0, exports.generateAuthExpressionForSandboxMode)(ctx.transformParameters.sandboxModeEnabled), `${queryTypeName}.${queryField}.{slotName}.{slotIndex}.res.vtl`));
     resolver.setScope(ctx.stackManager.getScopeFor(resolverResourceId, stackId));
-    ctx.resolvers.addResolver(object.name.value, queryField, resolver);
-}
-function validateIndexArgumentSnippet(config, keyOperation) {
+    ctx.resolvers.addResolver(queryTypeName, queryField, resolver);
+    if (!isDynamoDB) {
+        const modelFieldMap = ctx.resourceHelper.getModelFieldMap((_a = object === null || object === void 0 ? void 0 : object.name) === null || _a === void 0 ? void 0 : _a.value);
+        if (!modelFieldMap.getMappedFields().length) {
+            return;
+        }
+        modelFieldMap.addResolverReference({ typeName: queryTypeName, fieldName: queryField, isList: false });
+    }
+};
+exports.makeQueryResolver = makeQueryResolver;
+const validateIndexArgumentSnippet = (config, keyOperation) => {
     const { name, sortKeyFields } = config;
     if (sortKeyFields.length < 2) {
         return '';
@@ -375,17 +410,17 @@ function validateIndexArgumentSnippet(config, keyOperation) {
                 " you must provide all fields for the key. Missing key: '$keyFieldName'.\")")),
         ]),
     ]));
-}
-function mergeInputsAndDefaultsSnippet() {
+};
+const mergeInputsAndDefaultsSnippet = () => {
     return (0, graphql_mapping_template_1.printBlock)('Merge default values and inputs')((0, graphql_model_transformer_1.generateApplyDefaultsToInputTemplate)('mergedValues'));
-}
+};
 exports.mergeInputsAndDefaultsSnippet = mergeInputsAndDefaultsSnippet;
-function addIndexToResolverSlot(resolver, lines, isSync = false) {
+const addIndexToResolverSlot = (resolver, lines, isSync = false) => {
     const res = resolver;
     res.addToSlot('preAuth', graphql_transformer_core_1.MappingTemplate.s3MappingTemplateFromString(`${lines.join('\n')}\n${!isSync ? '{}' : ''}`, `${res.typeName}.${res.fieldName}.{slotName}.{slotIndex}.req.vtl`));
-}
+};
 exports.addIndexToResolverSlot = addIndexToResolverSlot;
-function makeSyncResolver(name, config, ctx, syncResolver, resolverMap) {
+const makeSyncResolver = (name, config, ctx, syncResolver, resolverMap) => {
     if (!ctx.isProjectUsingDataStore())
         return;
     if (resolverMap.has(syncResolver)) {
@@ -395,18 +430,18 @@ function makeSyncResolver(name, config, ctx, syncResolver, resolverMap) {
     else {
         resolverMap.set(syncResolver, (0, graphql_mapping_template_1.print)(setSyncQueryMapSnippet(name, config)));
     }
-}
-function joinSnippets(lines) {
+};
+const joinSnippets = (lines) => {
     return lines.join('\n');
-}
-function setSyncQueryMapSnippet(name, config) {
+};
+const setSyncQueryMapSnippet = (name, config) => {
     const { field, sortKeyFields } = config;
     const expressions = [];
     const keys = [field.name.value, ...(sortKeyFields !== null && sortKeyFields !== void 0 ? sortKeyFields : [])];
     expressions.push((0, graphql_mapping_template_1.raw)(`$util.qr($QueryMap.put('${keys.join('+')}' , '${name}'))`), (0, graphql_mapping_template_1.raw)(`$util.qr($PkMap.put('${field.name.value}' , '${name}'))`), (0, graphql_mapping_template_1.qref)((0, graphql_mapping_template_1.methodCall)((0, graphql_mapping_template_1.ref)('SkMap.put'), (0, graphql_mapping_template_1.str)(name), (0, graphql_mapping_template_1.list)(sortKeyFields.map(graphql_mapping_template_1.str)))));
     return (0, graphql_mapping_template_1.block)('Set query expression for @key', expressions);
-}
-function constructSyncVTL(syncVTLContent, resolver) {
+};
+const constructSyncVTL = (syncVTLContent, resolver) => {
     const checks = [
         (0, graphql_mapping_template_1.print)(generateSyncResolverInit()),
         syncVTLContent,
@@ -415,10 +450,10 @@ function constructSyncVTL(syncVTLContent, resolver) {
         (0, graphql_mapping_template_1.print)(setSyncKeyExpressionForRangeKey(graphql_transformer_common_1.ResourceConstants.SNIPPETS.ModelQueryExpression)),
         (0, graphql_mapping_template_1.print)(makeSyncQueryResolver()),
     ];
-    addIndexToResolverSlot(resolver, checks, true);
-}
+    (0, exports.addIndexToResolverSlot)(resolver, checks, true);
+};
 exports.constructSyncVTL = constructSyncVTL;
-function setSyncQueryFilterSnippet() {
+const setSyncQueryFilterSnippet = () => {
     const expressions = [];
     expressions.push((0, graphql_mapping_template_1.compoundExpression)([
         (0, graphql_mapping_template_1.set)((0, graphql_mapping_template_1.ref)('filterArgsMap'), (0, graphql_mapping_template_1.ref)('ctx.args.filter.get("and")')),
@@ -434,7 +469,7 @@ function setSyncQueryFilterSnippet() {
                         (0, graphql_mapping_template_1.set)((0, graphql_mapping_template_1.ref)('queryRequestVariables.partitionKey'), (0, graphql_mapping_template_1.ref)('pk')),
                         (0, graphql_mapping_template_1.set)((0, graphql_mapping_template_1.ref)('queryRequestVariables.sortKeys'), (0, graphql_mapping_template_1.ref)('SkMap.get($PkMap.get($pk))')),
                         (0, graphql_mapping_template_1.set)((0, graphql_mapping_template_1.ref)('queryRequestVariables.partitionKeyFilter'), (0, graphql_mapping_template_1.obj)({})),
-                        (0, graphql_mapping_template_1.raw)(`$util.qr($queryRequestVariables.partitionKeyFilter.put($pk, {'eq': $entry.value.eq}))`),
+                        (0, graphql_mapping_template_1.raw)("$util.qr($queryRequestVariables.partitionKeyFilter.put($pk, {'eq': $entry.value.eq}))"),
                         (0, graphql_mapping_template_1.raw)('$util.qr($ctx.args.put($pk,$entry.value.eq))'),
                         (0, graphql_mapping_template_1.set)((0, graphql_mapping_template_1.ref)('index'), (0, graphql_mapping_template_1.ref)('PkMap.get($pk)')),
                     ])),
@@ -448,11 +483,11 @@ function setSyncQueryFilterSnippet() {
         ]), (0, graphql_mapping_template_1.set)((0, graphql_mapping_template_1.ref)('filterMap'), (0, graphql_mapping_template_1.raw)('$ctx.args.filter'))),
     ]));
     return (0, graphql_mapping_template_1.block)('Set query expression for @key', expressions);
-}
+};
 const generateDeltaTableTTLCheck = (deltaTTLCheckRefName, lastSyncRefName) => {
     return (0, graphql_mapping_template_1.compoundExpression)([
         (0, graphql_mapping_template_1.set)((0, graphql_mapping_template_1.ref)(deltaTTLCheckRefName), (0, graphql_mapping_template_1.bool)(false)),
-        (0, graphql_mapping_template_1.set)((0, graphql_mapping_template_1.ref)('minLastSync'), (0, graphql_mapping_template_1.raw)(`$util.time.nowEpochMilliSeconds() - $ctx.stash.deltaSyncTableTtl * 60 * 1000`)),
+        (0, graphql_mapping_template_1.set)((0, graphql_mapping_template_1.ref)('minLastSync'), (0, graphql_mapping_template_1.raw)('$util.time.nowEpochMilliSeconds() - $ctx.stash.deltaSyncTableTtl * 60 * 1000')),
         (0, graphql_mapping_template_1.iff)((0, graphql_mapping_template_1.and)([
             (0, graphql_mapping_template_1.not)((0, graphql_mapping_template_1.methodCall)((0, graphql_mapping_template_1.ref)('util.isNull'), (0, graphql_mapping_template_1.ref)(lastSyncRefName))),
             (0, graphql_mapping_template_1.notEquals)((0, graphql_mapping_template_1.ref)(lastSyncRefName), (0, graphql_mapping_template_1.int)(0)),
@@ -460,7 +495,7 @@ const generateDeltaTableTTLCheck = (deltaTTLCheckRefName, lastSyncRefName) => {
         ]), (0, graphql_mapping_template_1.set)((0, graphql_mapping_template_1.ref)(deltaTTLCheckRefName), (0, graphql_mapping_template_1.bool)(true))),
     ]);
 };
-function setSyncKeyExpressionForHashKey(queryExprReference) {
+const setSyncKeyExpressionForHashKey = (queryExprReference) => {
     const expressions = [];
     expressions.push((0, graphql_mapping_template_1.set)((0, graphql_mapping_template_1.ref)(graphql_transformer_common_1.ResourceConstants.SNIPPETS.ModelQueryExpression), (0, graphql_mapping_template_1.obj)({})), (0, graphql_mapping_template_1.iff)((0, graphql_mapping_template_1.raw)('!$util.isNull($pk)'), (0, graphql_mapping_template_1.compoundExpression)([
         (0, graphql_mapping_template_1.set)((0, graphql_mapping_template_1.ref)(`${queryExprReference}.expression`), (0, graphql_mapping_template_1.str)('#pk = :pk')),
@@ -468,8 +503,8 @@ function setSyncKeyExpressionForHashKey(queryExprReference) {
         (0, graphql_mapping_template_1.set)((0, graphql_mapping_template_1.ref)(`${queryExprReference}.expressionValues`), (0, graphql_mapping_template_1.obj)({ ':pk': (0, graphql_mapping_template_1.ref)('util.parseJson($util.dynamodb.toDynamoDBJson($ctx.args.get($pk)))') })),
     ])));
     return (0, graphql_mapping_template_1.block)('Set Primary Key initialization @key', expressions);
-}
-function setSyncKeyExpressionForRangeKey(queryExprReference) {
+};
+const setSyncKeyExpressionForRangeKey = (queryExprReference) => {
     return (0, graphql_mapping_template_1.block)('Applying Key Condition', [
         (0, graphql_mapping_template_1.iff)((0, graphql_mapping_template_1.raw)('!$util.isNull($ctx.args.get($sk)) && !$util.isNull($ctx.args.get($sk).beginsWith)'), (0, graphql_mapping_template_1.compoundExpression)([
             (0, graphql_mapping_template_1.set)((0, graphql_mapping_template_1.ref)(`${queryExprReference}.expression`), (0, graphql_mapping_template_1.raw)(`"$${queryExprReference}.expression AND begins_with(#sortKey, :sortKey)"`)),
@@ -508,8 +543,8 @@ function setSyncKeyExpressionForRangeKey(queryExprReference) {
             (0, graphql_mapping_template_1.qref)(`$${queryExprReference}.expressionValues.put(":sortKey", $util.parseJson($util.dynamodb.toDynamoDBJson($ctx.args.get($sk).ge)))`),
         ])),
     ]);
-}
-function makeSyncQueryResolver() {
+};
+const makeSyncQueryResolver = () => {
     const requestVariable = 'ctx.stash.QueryRequest';
     const queryRequestVariables = 'ctx.stash.QueryRequestVariables';
     const expressions = [];
@@ -528,15 +563,15 @@ function makeSyncQueryResolver() {
         (0, graphql_mapping_template_1.iff)((0, graphql_mapping_template_1.ref)('context.args.nextToken'), (0, graphql_mapping_template_1.set)((0, graphql_mapping_template_1.ref)(`${requestVariable}.nextToken`), (0, graphql_mapping_template_1.ref)('context.args.nextToken')), true),
         (0, graphql_mapping_template_1.iff)((0, graphql_mapping_template_1.and)([(0, graphql_mapping_template_1.raw)('!$util.isNullOrEmpty($filterMap)'), (0, graphql_mapping_template_1.notEquals)((0, graphql_mapping_template_1.toJson)((0, graphql_mapping_template_1.ref)('filterMap')), (0, graphql_mapping_template_1.toJson)((0, graphql_mapping_template_1.obj)({})))]), (0, graphql_mapping_template_1.set)((0, graphql_mapping_template_1.ref)(`${requestVariable}.filter`), (0, graphql_mapping_template_1.ref)('filterMap'))),
         (0, graphql_mapping_template_1.iff)((0, graphql_mapping_template_1.raw)('$index != "dbTable"'), (0, graphql_mapping_template_1.set)((0, graphql_mapping_template_1.ref)(`${requestVariable}.index`), (0, graphql_mapping_template_1.ref)('index'))),
-    ])), (0, graphql_mapping_template_1.raw)(`$util.toJson({})`));
+    ])), (0, graphql_mapping_template_1.raw)('$util.toJson({})'));
     return (0, graphql_mapping_template_1.block)(' Set query expression for @key', expressions);
-}
-function generateSyncResolverInit() {
+};
+const generateSyncResolverInit = () => {
     const expressions = [];
     const requestVariable = 'ctx.stash.QueryRequest';
     expressions.push((0, graphql_mapping_template_1.set)((0, graphql_mapping_template_1.ref)('index'), (0, graphql_mapping_template_1.str)('')), (0, graphql_mapping_template_1.set)((0, graphql_mapping_template_1.ref)('scan'), (0, graphql_mapping_template_1.bool)(true)), (0, graphql_mapping_template_1.set)((0, graphql_mapping_template_1.ref)('filterMap'), (0, graphql_mapping_template_1.obj)({})), (0, graphql_mapping_template_1.set)((0, graphql_mapping_template_1.ref)('QueryMap'), (0, graphql_mapping_template_1.obj)({})), (0, graphql_mapping_template_1.set)((0, graphql_mapping_template_1.ref)('PkMap'), (0, graphql_mapping_template_1.obj)({})), (0, graphql_mapping_template_1.set)((0, graphql_mapping_template_1.ref)('SkMap'), (0, graphql_mapping_template_1.obj)({})), (0, graphql_mapping_template_1.set)((0, graphql_mapping_template_1.ref)('filterArgsMap'), (0, graphql_mapping_template_1.obj)({})), (0, graphql_mapping_template_1.iff)((0, graphql_mapping_template_1.ref)(requestVariable), (0, graphql_mapping_template_1.raw)('#return')), (0, graphql_mapping_template_1.set)((0, graphql_mapping_template_1.ref)('queryRequestVariables'), (0, graphql_mapping_template_1.obj)({})));
     return (0, graphql_mapping_template_1.block)('Set map initialization for @key', expressions);
-}
+};
 const generateAuthExpressionForSandboxMode = (enabled) => {
     let exp;
     if (enabled)
@@ -546,24 +581,30 @@ const generateAuthExpressionForSandboxMode = (enabled) => {
     return (0, graphql_mapping_template_1.printBlock)(`Sandbox Mode ${enabled ? 'Enabled' : 'Disabled'}`)((0, graphql_mapping_template_1.compoundExpression)([(0, graphql_mapping_template_1.iff)((0, graphql_mapping_template_1.not)((0, graphql_mapping_template_1.ref)('ctx.stash.get("hasAuth")')), exp), (0, graphql_mapping_template_1.toJson)((0, graphql_mapping_template_1.obj)({}))]));
 };
 exports.generateAuthExpressionForSandboxMode = generateAuthExpressionForSandboxMode;
-function getDBInfo(ctx, modelName) {
+const getDBInfo = (ctx, modelName) => {
     const dbInfo = ctx.modelToDatasourceMap.get(modelName);
-    const result = dbInfo !== null && dbInfo !== void 0 ? dbInfo : { dbType: 'DDB', provisionDB: true };
+    const result = dbInfo !== null && dbInfo !== void 0 ? dbInfo : { dbType: graphql_transformer_core_1.DDB_DB_TYPE, provisionDB: true, provisionStrategy: "DEFAULT" };
     return result;
-}
+};
 exports.getDBInfo = getDBInfo;
-function getDBType(ctx, modelName) {
-    const dbInfo = getDBInfo(ctx, modelName);
-    const dbType = dbInfo ? dbInfo.dbType : 'DDB';
+const getDBType = (ctx, modelName) => {
+    const dbInfo = (0, exports.getDBInfo)(ctx, modelName);
+    const dbType = dbInfo ? dbInfo.dbType : graphql_transformer_core_1.DDB_DB_TYPE;
     return dbType;
-}
+};
 exports.getDBType = getDBType;
 const getVTLGenerator = (dbInfo) => {
-    const dbType = dbInfo ? dbInfo.dbType : 'DDB';
-    if (dbType === 'MySQL') {
-        return new generators_1.RDSIndexVTLGenerator();
+    const dbType = dbInfo ? dbInfo.dbType : graphql_transformer_core_1.DDB_DB_TYPE;
+    switch (dbType) {
+        case graphql_transformer_core_1.DDB_DB_TYPE:
+            return new generators_1.DynamoDBIndexVTLGenerator();
+        case graphql_transformer_core_1.MYSQL_DB_TYPE:
+            return new generators_1.RDSIndexVTLGenerator();
+        case graphql_transformer_core_1.POSTGRES_DB_TYPE:
+            return new generators_1.RDSIndexVTLGenerator();
+        default:
+            throw new Error(`Unknown database type ${dbType}`);
     }
-    return new generators_1.DynamoDBIndexVTLGenerator();
 };
 exports.getVTLGenerator = getVTLGenerator;
 //# sourceMappingURL=resolvers.js.map