@aws-amplify/graphql-model-transformer 0.13.4 → 0.13.5-beta.2

package/lib/graphql-model-transformer.js

@@ -18,11 +18,8 @@ var __importStar = (this && this.__importStar) || function (mod) {
     __setModuleDefault(result, mod);
     return result;
 };
-var __importDefault = (this && this.__importDefault) || function (mod) {
-    return (mod && mod.__esModule) ? mod : { "default": mod };
-};
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.ModelTransformer = exports.directiveDefinition = exports.SubscriptionLevel = void 0;
+exports.ModelTransformer = exports.directiveDefinition = void 0;
 const graphql_transformer_core_1 = require("@aws-amplify/graphql-transformer-core");
 const graphql_transformer_interfaces_1 = require("@aws-amplify/graphql-transformer-interfaces");
 const aws_dynamodb_1 = require("@aws-cdk/aws-dynamodb");
@@ -32,14 +29,8 @@ const graphql_transformer_common_1 = require("graphql-transformer-common");
 const graphql_types_1 = require("./graphql-types");
 const resolvers_1 = require("./resolvers");
 const query_1 = require("./resolvers/query");
-const md5_1 = __importDefault(require("md5"));
 const definitions_1 = require("./definitions");
-var SubscriptionLevel;
-(function (SubscriptionLevel) {
-    SubscriptionLevel["off"] = "off";
-    SubscriptionLevel["public"] = "public";
-    SubscriptionLevel["on"] = "on";
-})(SubscriptionLevel = exports.SubscriptionLevel || (exports.SubscriptionLevel = {}));
+const directive_1 = require("./directive");
 exports.directiveDefinition = `
   directive @model(
     queries: ModelQueryMap
@@ -111,9 +102,9 @@ class ModelTransformer extends graphql_transformer_core_1.TransformerModelBase {
         };
         this.object = (definition, directive, ctx) => {
             var _a, _b, _c;
-            const isTypeNameReserved = definition.name.value === ctx.output.getQueryTypeName() ||
-                definition.name.value === ctx.output.getMutationTypeName() ||
-                definition.name.value === ctx.output.getSubscriptionTypeName();
+            const isTypeNameReserved = definition.name.value === ctx.output.getQueryTypeName()
+                || definition.name.value === ctx.output.getMutationTypeName()
+                || definition.name.value === ctx.output.getSubscriptionTypeName();
             if (isTypeNameReserved) {
                 throw new graphql_transformer_core_1.InvalidDirectiveError(`'${definition.name.value}' is a reserved type name and currently in use within the default schema element.`);
             }
@@ -134,7 +125,7 @@ class ModelTransformer extends graphql_transformer_core_1.TransformerModelBase {
                     delete: (0, graphql_transformer_core_1.getFieldNameFor)('delete', typeName),
                 },
                 subscriptions: {
-                    level: SubscriptionLevel.on,
+                    level: directive_1.SubscriptionLevel.on,
                     onCreate: [(0, graphql_transformer_core_1.getFieldNameFor)('onCreate', typeName)],
                     onDelete: [(0, graphql_transformer_core_1.getFieldNameFor)('onDelete', typeName)],
                     onUpdate: [(0, graphql_transformer_core_1.getFieldNameFor)('onUpdate', typeName)],
@@ -156,17 +147,16 @@ class ModelTransformer extends graphql_transformer_core_1.TransformerModelBase {
             this.modelDirectiveConfig.set(typeName, options);
             this.typesWithModelDirective.add(typeName);
         };
-        this.validate = () => { };
         this.prepare = (context) => {
-            for (const modelTypeName of this.typesWithModelDirective) {
+            this.typesWithModelDirective.forEach(modelTypeName => {
                 const type = context.output.getObject(modelTypeName);
                 context.providerRegistry.registerDataSourceProvider(type, this);
-            }
+            });
         };
         this.transformSchema = (ctx) => {
             (0, graphql_types_1.addModelConditionInputs)(ctx);
             this.ensureModelSortDirectionEnum(ctx);
-            for (const type of this.typesWithModelDirective) {
+            this.typesWithModelDirective.forEach(type => {
                 const def = ctx.output.getObject(type);
                 const hasAuth = def.directives.some(dir => dir.name.value === 'auth');
                 this.createNonModelInputs(ctx, def);
@@ -184,31 +174,31 @@ class ModelTransformer extends graphql_transformer_core_1.TransformerModelBase {
                     const apiKeyDirArray = [(0, graphql_transformer_common_1.makeDirective)(definitions_1.API_KEY_DIRECTIVE, [])];
                     (0, graphql_types_1.extendTypeWithDirectives)(ctx, def.name.value, apiKeyDirArray);
                     (0, graphql_types_1.propagateApiKeyToNestedTypes)(ctx, def, new Set());
-                    for (let operationField of queryFields) {
+                    queryFields.forEach(operationField => {
                         const operationName = operationField.name.value;
                         (0, graphql_types_1.addDirectivesToOperation)(ctx, ctx.output.getQueryTypeName(), operationName, apiKeyDirArray);
-                    }
-                    for (let operationField of mutationFields) {
+                    });
+                    mutationFields.forEach(operationField => {
                         const operationName = operationField.name.value;
                         (0, graphql_types_1.addDirectivesToOperation)(ctx, ctx.output.getMutationTypeName(), operationName, apiKeyDirArray);
-                    }
-                    for (let operationField of subscriptionsFields) {
+                    });
+                    subscriptionsFields.forEach(operationField => {
                         const operationName = operationField.name.value;
                         (0, graphql_types_1.addDirectivesToOperation)(ctx, ctx.output.getSubscriptionTypeName(), operationName, apiKeyDirArray);
-                    }
+                    });
                 }
-            }
+            });
         };
         this.generateResolvers = (context) => {
-            var _a, _b;
-            for (let type of this.typesWithModelDirective) {
+            this.typesWithModelDirective.forEach(type => {
+                var _a, _b;
                 const def = context.output.getObject(type);
                 const tableBaseName = context.resourceHelper.getModelNameMapping(def.name.value);
                 const tableLogicalName = graphql_transformer_common_1.ModelResourceIDs.ModelTableResourceID(tableBaseName);
                 const stack = context.stackManager.getStackFor(tableLogicalName, tableBaseName);
                 this.createModelTable(stack, def, context);
-                const queryFields = this.getQueryFieldNames(context, def);
-                for (let query of queryFields.values()) {
+                const queryFields = this.getQueryFieldNames(def);
+                queryFields.forEach(query => {
                     let resolver;
                     switch (query.type) {
                         case graphql_transformer_interfaces_1.QueryFieldType.GET:
@@ -226,9 +216,9 @@ class ModelTransformer extends graphql_transformer_core_1.TransformerModelBase {
                     resolver.addToSlot('postAuth', graphql_transformer_core_1.MappingTemplate.s3MappingTemplateFromString((0, resolvers_1.generateAuthExpressionForSandboxMode)(context.sandboxModeEnabled), `${query.typeName}.${query.fieldName}.{slotName}.{slotIndex}.req.vtl`));
                     resolver.mapToStack(context.stackManager.getStackFor(query.resolverLogicalId, def.name.value));
                     context.resolvers.addResolver(query.typeName, query.fieldName, resolver);
-                }
-                const mutationFields = this.getMutationFieldNames(context, def);
-                for (let mutation of mutationFields.values()) {
+                });
+                const mutationFields = this.getMutationFieldNames(def);
+                mutationFields.forEach(mutation => {
                     let resolver;
                     switch (mutation.type) {
                         case graphql_transformer_interfaces_1.MutationFieldType.CREATE:
@@ -246,21 +236,21 @@ class ModelTransformer extends graphql_transformer_core_1.TransformerModelBase {
                     resolver.addToSlot('postAuth', graphql_transformer_core_1.MappingTemplate.s3MappingTemplateFromString((0, resolvers_1.generateAuthExpressionForSandboxMode)(context.sandboxModeEnabled), `${mutation.typeName}.${mutation.fieldName}.{slotName}.{slotIndex}.req.vtl`));
                     resolver.mapToStack(context.stackManager.getStackFor(mutation.resolverLogicalId, def.name.value));
                     context.resolvers.addResolver(mutation.typeName, mutation.fieldName, resolver);
-                }
+                });
                 const subscriptionLevel = (_b = (_a = this.modelDirectiveConfig.get(def.name.value)) === null || _a === void 0 ? void 0 : _a.subscriptions) === null || _b === void 0 ? void 0 : _b.level;
-                if (subscriptionLevel === SubscriptionLevel.on) {
-                    const subscriptionFields = this.getSubscriptionFieldNames(context, def);
-                    for (let subscription of subscriptionFields.values()) {
+                if (subscriptionLevel === directive_1.SubscriptionLevel.on) {
+                    const subscriptionFields = this.getSubscriptionFieldNames(def);
+                    subscriptionFields.forEach(subscription => {
                         let resolver;
                         switch (subscription.type) {
                             case graphql_transformer_interfaces_1.SubscriptionFieldType.ON_CREATE:
-                                resolver = this.generateOnCreateResolver(context, def, subscription.typeName, subscription.fieldName, subscription.resolverLogicalId);
+                                resolver = this.generateOnCreateResolver(context, subscription.typeName, subscription.fieldName, subscription.resolverLogicalId);
                                 break;
                             case graphql_transformer_interfaces_1.SubscriptionFieldType.ON_UPDATE:
-                                resolver = this.generateOnUpdateResolver(context, def, subscription.typeName, subscription.fieldName, subscription.resolverLogicalId);
+                                resolver = this.generateOnUpdateResolver(context, subscription.typeName, subscription.fieldName, subscription.resolverLogicalId);
                                 break;
                             case graphql_transformer_interfaces_1.SubscriptionFieldType.ON_DELETE:
-                                resolver = this.generateOnDeleteResolver(context, def, subscription.typeName, subscription.fieldName, subscription.resolverLogicalId);
+                                resolver = this.generateOnDeleteResolver(context, subscription.typeName, subscription.fieldName, subscription.resolverLogicalId);
                                 break;
                             default:
                                 throw new Error('Unknown subscription field type');
@@ -268,9 +258,9 @@ class ModelTransformer extends graphql_transformer_core_1.TransformerModelBase {
                         resolver.addToSlot('postAuth', graphql_transformer_core_1.MappingTemplate.s3MappingTemplateFromString((0, resolvers_1.generateAuthExpressionForSandboxMode)(context.sandboxModeEnabled), `${subscription.typeName}.${subscription.fieldName}.{slotName}.{slotIndex}.req.vtl`));
                         resolver.mapToStack(context.stackManager.getStackFor(subscription.resolverLogicalId, def.name.value));
                         context.resolvers.addResolver(subscription.typeName, subscription.fieldName, resolver);
-                    }
+                    });
                 }
-            }
+            });
         };
         this.generateGetResolver = (ctx, type, typeName, fieldName, resolverLogicalId) => {
             const isSyncEnabled = ctx.isProjectUsingDataStore();
@@ -296,7 +286,7 @@ class ModelTransformer extends graphql_transformer_core_1.TransformerModelBase {
             const resolverKey = `Update${(0, resolvers_1.generateResolverKey)(typeName, fieldName)}`;
             if (!this.resolverMap[resolverKey]) {
                 const resolver = ctx.resolvers.generateMutationResolver(typeName, fieldName, resolverLogicalId, dataSource, graphql_transformer_core_1.MappingTemplate.s3MappingTemplateFromString((0, resolvers_1.generateUpdateRequestTemplate)(typeName, isSyncEnabled), `${typeName}.${fieldName}.req.vtl`), graphql_transformer_core_1.MappingTemplate.s3MappingTemplateFromString((0, resolvers_1.generateDefaultResponseMappingTemplate)(isSyncEnabled, true), `${typeName}.${fieldName}.res.vtl`));
-                resolver.addToSlot('init', graphql_transformer_core_1.MappingTemplate.s3MappingTemplateFromString((0, resolvers_1.generateUpdateInitSlotTemplate)(type.name.value, this.modelDirectiveConfig.get(type.name.value)), `${typeName}.${fieldName}.{slotName}.{slotIndex}.req.vtl`));
+                resolver.addToSlot('init', graphql_transformer_core_1.MappingTemplate.s3MappingTemplateFromString((0, resolvers_1.generateUpdateInitSlotTemplate)(this.modelDirectiveConfig.get(type.name.value)), `${typeName}.${fieldName}.{slotName}.{slotIndex}.req.vtl`));
                 this.resolverMap[resolverKey] = resolver;
             }
             return this.resolverMap[resolverKey];
@@ -310,21 +300,21 @@ class ModelTransformer extends graphql_transformer_core_1.TransformerModelBase {
             }
             return this.resolverMap[resolverKey];
         };
-        this.generateOnCreateResolver = (ctx, type, typeName, fieldName, resolverLogicalId) => {
+        this.generateOnCreateResolver = (ctx, typeName, fieldName, resolverLogicalId) => {
             const resolverKey = `OnCreate${(0, resolvers_1.generateResolverKey)(typeName, fieldName)}`;
             if (!this.resolverMap[resolverKey]) {
                 this.resolverMap[resolverKey] = ctx.resolvers.generateSubscriptionResolver(typeName, fieldName, resolverLogicalId, graphql_transformer_core_1.MappingTemplate.s3MappingTemplateFromString((0, resolvers_1.generateSubscriptionRequestTemplate)(), `${typeName}.${fieldName}.req.vtl`), graphql_transformer_core_1.MappingTemplate.s3MappingTemplateFromString((0, resolvers_1.generateSubscriptionResponseTemplate)(), `${typeName}.${fieldName}.res.vtl`));
             }
             return this.resolverMap[resolverKey];
         };
-        this.generateOnUpdateResolver = (ctx, type, typeName, fieldName, resolverLogicalId) => {
+        this.generateOnUpdateResolver = (ctx, typeName, fieldName, resolverLogicalId) => {
             const resolverKey = `OnUpdate${(0, resolvers_1.generateResolverKey)(typeName, fieldName)}`;
             if (!this.resolverMap[resolverKey]) {
                 this.resolverMap[resolverKey] = ctx.resolvers.generateSubscriptionResolver(typeName, fieldName, resolverLogicalId, graphql_transformer_core_1.MappingTemplate.s3MappingTemplateFromString((0, resolvers_1.generateSubscriptionRequestTemplate)(), `${typeName}.${fieldName}.req.vtl`), graphql_transformer_core_1.MappingTemplate.s3MappingTemplateFromString((0, resolvers_1.generateSubscriptionResponseTemplate)(), `${typeName}.${fieldName}.res.vtl`));
             }
             return this.resolverMap[resolverKey];
         };
-        this.generateOnDeleteResolver = (ctx, type, typeName, fieldName, resolverLogicalId) => {
+        this.generateOnDeleteResolver = (ctx, typeName, fieldName, resolverLogicalId) => {
             const resolverKey = `OnDelete${(0, resolvers_1.generateResolverKey)(typeName, fieldName)}`;
             if (!this.resolverMap[resolverKey]) {
                 this.resolverMap[resolverKey] = ctx.resolvers.generateSubscriptionResolver(typeName, fieldName, resolverLogicalId, graphql_transformer_core_1.MappingTemplate.s3MappingTemplateFromString((0, resolvers_1.generateSubscriptionRequestTemplate)(), `${typeName}.${fieldName}.req.vtl`), graphql_transformer_core_1.MappingTemplate.s3MappingTemplateFromString((0, resolvers_1.generateSubscriptionResponseTemplate)(), `${typeName}.${fieldName}.res.vtl`));
@@ -340,7 +330,7 @@ class ModelTransformer extends graphql_transformer_core_1.TransformerModelBase {
             }
             return this.resolverMap[resolverKey];
         };
-        this.getQueryFieldNames = (ctx, type) => {
+        this.getQueryFieldNames = (type) => {
             var _a, _b, _c;
             const typeName = type.name.value;
             const fields = new Set();
@@ -371,11 +361,11 @@ class ModelTransformer extends graphql_transformer_core_1.TransformerModelBase {
             }
             return fields;
         };
-        this.getMutationFieldNames = (ctx, type) => {
+        this.getMutationFieldNames = (type) => {
             const typeName = type.name.value;
             const modelDirectiveConfig = this.modelDirectiveConfig.get(typeName);
-            const getMutationType = (type) => {
-                switch (type) {
+            const getMutationType = (mutationType) => {
+                switch (mutationType) {
                     case 'create':
                         return graphql_transformer_interfaces_1.MutationFieldType.CREATE;
                     case 'update':
@@ -386,8 +376,8 @@ class ModelTransformer extends graphql_transformer_core_1.TransformerModelBase {
                         throw new Error('Unknown mutation type');
                 }
             };
-            const getMutationResolverLogicalId = (type) => {
-                switch (type) {
+            const getMutationResolverLogicalId = (mutationType) => {
+                switch (mutationType) {
                     case 'create':
                         return graphql_transformer_common_1.ResolverResourceIDs.DynamoDBCreateResolverResourceID(typeName);
                     case 'update':
@@ -399,7 +389,7 @@ class ModelTransformer extends graphql_transformer_core_1.TransformerModelBase {
                 }
             };
             const fieldNames = new Set();
-            for (let [mutationType, mutationName] of Object.entries((modelDirectiveConfig === null || modelDirectiveConfig === void 0 ? void 0 : modelDirectiveConfig.mutations) || {})) {
+            Object.entries((modelDirectiveConfig === null || modelDirectiveConfig === void 0 ? void 0 : modelDirectiveConfig.mutations) || {}).forEach(([mutationType, mutationName]) => {
                 if (mutationName) {
                     fieldNames.add({
                         typeName: 'Mutation',
@@ -408,7 +398,7 @@ class ModelTransformer extends graphql_transformer_core_1.TransformerModelBase {
                         resolverLogicalId: getMutationResolverLogicalId(mutationType),
                     });
                 }
-            }
+            });
             return fieldNames;
         };
         this.getMutationName = (subscriptionType, mutationMap) => {
@@ -417,7 +407,7 @@ class ModelTransformer extends graphql_transformer_core_1.TransformerModelBase {
                 [graphql_transformer_interfaces_1.SubscriptionFieldType.ON_UPDATE]: graphql_transformer_interfaces_1.MutationFieldType.UPDATE,
                 [graphql_transformer_interfaces_1.SubscriptionFieldType.ON_DELETE]: graphql_transformer_interfaces_1.MutationFieldType.DELETE,
             };
-            const mutation = Array.from(mutationMap).find(m => m.type == mutationToSubscriptionTypeMap[subscriptionType]);
+            const mutation = Array.from(mutationMap).find(m => m.type === mutationToSubscriptionTypeMap[subscriptionType]);
             if (mutation) {
                 return mutation.fieldName;
             }
@@ -425,8 +415,8 @@ class ModelTransformer extends graphql_transformer_core_1.TransformerModelBase {
         };
         this.createQueryFields = (ctx, def) => {
             const queryFields = [];
-            const queryFieldNames = this.getQueryFieldNames(ctx, def);
-            for (const queryField of queryFieldNames.values()) {
+            const queryFieldNames = this.getQueryFieldNames(def);
+            queryFieldNames.forEach(queryField => {
                 const outputType = this.getOutputType(ctx, def, queryField);
                 const args = this.getInputs(ctx, def, {
                     fieldName: queryField.fieldName,
@@ -434,27 +424,27 @@ class ModelTransformer extends graphql_transformer_core_1.TransformerModelBase {
                     type: queryField.type,
                 });
                 queryFields.push((0, graphql_transformer_common_1.makeField)(queryField.fieldName, args, (0, graphql_transformer_common_1.makeNamedType)(outputType.name.value)));
-            }
+            });
             return queryFields;
         };
         this.createMutationFields = (ctx, def) => {
             const mutationFields = [];
-            const mutationFieldNames = this.getMutationFieldNames(ctx, def);
-            for (const mutationField of mutationFieldNames) {
+            const mutationFieldNames = this.getMutationFieldNames(def);
+            mutationFieldNames.forEach(mutationField => {
                 const args = this.getInputs(ctx, def, {
                     fieldName: mutationField.fieldName,
                     typeName: mutationField.typeName,
                     type: mutationField.type,
                 });
                 mutationFields.push((0, graphql_transformer_common_1.makeField)(mutationField.fieldName, args, (0, graphql_transformer_common_1.makeNamedType)(def.name.value)));
-            }
+            });
             return mutationFields;
         };
         this.createSubscriptionFields = (ctx, def) => {
-            const subscriptionToMutationsMap = this.getSubscriptionToMutationsReverseMap(ctx, def);
-            const mutationFields = this.getMutationFieldNames(ctx, def);
+            const subscriptionToMutationsMap = this.getSubscriptionToMutationsReverseMap(def);
+            const mutationFields = this.getMutationFieldNames(def);
             const subscriptionFields = [];
-            for (const subscriptionFieldName of Object.keys(subscriptionToMutationsMap)) {
+            Object.keys(subscriptionToMutationsMap).forEach(subscriptionFieldName => {
                 const maps = subscriptionToMutationsMap[subscriptionFieldName];
                 const args = [];
                 maps.map(it => args.concat(this.getInputs(ctx, def, {
@@ -466,61 +456,58 @@ class ModelTransformer extends graphql_transformer_core_1.TransformerModelBase {
                 const directive = (0, graphql_transformer_common_1.makeDirective)('aws_subscribe', [(0, graphql_transformer_common_1.makeArgument)('mutations', (0, graphql_transformer_common_1.makeValueNode)(mutationNames))]);
                 const field = (0, graphql_transformer_common_1.makeField)(subscriptionFieldName, args, (0, graphql_transformer_common_1.makeNamedType)(def.name.value), [directive]);
                 subscriptionFields.push(field);
-            }
+            });
             return subscriptionFields;
         };
-        this.getSubscriptionFieldNames = (ctx, type) => {
+        this.getSubscriptionFieldNames = (type) => {
             var _a, _b, _c, _d, _e, _f, _g;
             const fields = new Set();
             const modelDirectiveConfig = this.modelDirectiveConfig.get(type.name.value);
-            if (((_a = modelDirectiveConfig === null || modelDirectiveConfig === void 0 ? void 0 : modelDirectiveConfig.subscriptions) === null || _a === void 0 ? void 0 : _a.level) !== SubscriptionLevel.off) {
+            if (((_a = modelDirectiveConfig === null || modelDirectiveConfig === void 0 ? void 0 : modelDirectiveConfig.subscriptions) === null || _a === void 0 ? void 0 : _a.level) !== directive_1.SubscriptionLevel.off) {
                 if (((_b = modelDirectiveConfig === null || modelDirectiveConfig === void 0 ? void 0 : modelDirectiveConfig.subscriptions) === null || _b === void 0 ? void 0 : _b.onCreate) && ((_c = modelDirectiveConfig.mutations) === null || _c === void 0 ? void 0 : _c.create)) {
-                    for (const fieldName of modelDirectiveConfig.subscriptions.onCreate) {
+                    modelDirectiveConfig.subscriptions.onCreate.forEach((fieldName) => {
                         fields.add({
                             typeName: 'Subscription',
-                            fieldName: fieldName,
+                            fieldName,
                             type: graphql_transformer_interfaces_1.SubscriptionFieldType.ON_CREATE,
                             resolverLogicalId: graphql_transformer_common_1.ResolverResourceIDs.ResolverResourceID('Subscription', fieldName),
                         });
-                    }
+                    });
                 }
                 if (((_d = modelDirectiveConfig === null || modelDirectiveConfig === void 0 ? void 0 : modelDirectiveConfig.subscriptions) === null || _d === void 0 ? void 0 : _d.onUpdate) && ((_e = modelDirectiveConfig.mutations) === null || _e === void 0 ? void 0 : _e.update)) {
-                    for (const fieldName of modelDirectiveConfig.subscriptions.onUpdate) {
+                    modelDirectiveConfig.subscriptions.onUpdate.forEach((fieldName) => {
                         fields.add({
                             typeName: 'Subscription',
-                            fieldName: fieldName,
+                            fieldName,
                             type: graphql_transformer_interfaces_1.SubscriptionFieldType.ON_UPDATE,
                             resolverLogicalId: graphql_transformer_common_1.ResolverResourceIDs.ResolverResourceID('Subscription', fieldName),
                         });
-                    }
+                    });
                 }
                 if (((_f = modelDirectiveConfig === null || modelDirectiveConfig === void 0 ? void 0 : modelDirectiveConfig.subscriptions) === null || _f === void 0 ? void 0 : _f.onDelete) && ((_g = modelDirectiveConfig.mutations) === null || _g === void 0 ? void 0 : _g.delete)) {
-                    for (const fieldName of modelDirectiveConfig.subscriptions.onDelete) {
+                    modelDirectiveConfig.subscriptions.onDelete.forEach((fieldName) => {
                         fields.add({
                             typeName: 'Subscription',
-                            fieldName: fieldName,
+                            fieldName,
                             type: graphql_transformer_interfaces_1.SubscriptionFieldType.ON_DELETE,
                             resolverLogicalId: graphql_transformer_common_1.ResolverResourceIDs.ResolverResourceID('Subscription', fieldName),
                         });
-                    }
+                    });
                 }
             }
             return fields;
         };
-        this.getDataSourceResource = (ctx, type) => {
-            return this.ddbTableMap[type.name.value];
-        };
-        this.getDataSourceType = () => {
-            return graphql_transformer_interfaces_1.AppSyncDataSourceType.AMAZON_DYNAMODB;
-        };
+        this.getDataSourceResource = (type) => this.ddbTableMap[type.name.value];
+        this.getDataSourceType = () => graphql_transformer_interfaces_1.AppSyncDataSourceType.AMAZON_DYNAMODB;
         this.generateCreateResolver = (ctx, type, typeName, fieldName, resolverLogicalId) => {
             const isSyncEnabled = ctx.isProjectUsingDataStore();
             const dataSource = this.datasourceMap[type.name.value];
             const resolverKey = `Create${(0, resolvers_1.generateResolverKey)(typeName, fieldName)}`;
+            const modelIndexFields = type.fields.filter(field => { var _a; return (_a = field.directives) === null || _a === void 0 ? void 0 : _a.some(it => it.name.value === 'index'); }).map(it => it.name.value);
             if (!this.resolverMap[resolverKey]) {
-                const resolver = ctx.resolvers.generateMutationResolver(typeName, fieldName, resolverLogicalId, dataSource, graphql_transformer_core_1.MappingTemplate.s3MappingTemplateFromString((0, resolvers_1.generateCreateRequestTemplate)(type.name.value), `${typeName}.${fieldName}.req.vtl`), graphql_transformer_core_1.MappingTemplate.s3MappingTemplateFromString((0, resolvers_1.generateDefaultResponseMappingTemplate)(isSyncEnabled, true), `${typeName}.${fieldName}.res.vtl`));
+                const resolver = ctx.resolvers.generateMutationResolver(typeName, fieldName, resolverLogicalId, dataSource, graphql_transformer_core_1.MappingTemplate.s3MappingTemplateFromString((0, resolvers_1.generateCreateRequestTemplate)(type.name.value, modelIndexFields), `${typeName}.${fieldName}.req.vtl`), graphql_transformer_core_1.MappingTemplate.s3MappingTemplateFromString((0, resolvers_1.generateDefaultResponseMappingTemplate)(isSyncEnabled, true), `${typeName}.${fieldName}.res.vtl`));
                 this.resolverMap[resolverKey] = resolver;
-                resolver.addToSlot('init', graphql_transformer_core_1.MappingTemplate.s3MappingTemplateFromString((0, resolvers_1.generateCreateInitSlotTemplate)(type.name.value, this.modelDirectiveConfig.get(type.name.value)), `${typeName}.${fieldName}.{slotName}.{slotIndex}.req.vtl`));
+                resolver.addToSlot('init', graphql_transformer_core_1.MappingTemplate.s3MappingTemplateFromString((0, resolvers_1.generateCreateInitSlotTemplate)(this.modelDirectiveConfig.get(type.name.value)), `${typeName}.${fieldName}.{slotName}.{slotIndex}.req.vtl`));
             }
             return this.resolverMap[resolverKey];
         };
@@ -533,32 +520,33 @@ class ModelTransformer extends graphql_transformer_core_1.TransformerModelBase {
                 const filterInputs = (0, graphql_types_1.createEnumModelFilters)(ctx, type);
                 conditionInput = (0, graphql_types_1.makeMutationConditionInput)(ctx, conditionTypeName, type);
                 filterInputs.push(conditionInput);
-                for (let input of filterInputs) {
+                filterInputs.forEach(input => {
                     const conditionInputName = input.name.value;
                     if (!ctx.output.getType(conditionInputName)) {
                         ctx.output.addInput(input);
                     }
-                }
+                });
             }
             switch (operation.type) {
                 case graphql_transformer_interfaces_1.QueryFieldType.GET:
                     return [(0, graphql_transformer_common_1.makeInputValueDefinition)('id', (0, graphql_transformer_common_1.makeNonNullType)((0, graphql_transformer_common_1.makeNamedType)('ID')))];
-                case graphql_transformer_interfaces_1.QueryFieldType.LIST:
+                case graphql_transformer_interfaces_1.QueryFieldType.LIST: {
                     const filterInputName = (0, graphql_transformer_common_1.toPascalCase)(['Model', type.name.value, 'FilterInput']);
                     const filterInputs = (0, graphql_types_1.createEnumModelFilters)(ctx, type);
                     filterInputs.push((0, graphql_types_1.makeListQueryFilterInput)(ctx, filterInputName, type));
-                    for (let input of filterInputs) {
+                    filterInputs.forEach(input => {
                         const conditionInputName = input.name.value;
                         if (!ctx.output.getType(conditionInputName)) {
                             ctx.output.addInput(input);
                         }
-                    }
+                    });
                     return [
                         (0, graphql_transformer_common_1.makeInputValueDefinition)('filter', (0, graphql_transformer_common_1.makeNamedType)(filterInputName)),
                         (0, graphql_transformer_common_1.makeInputValueDefinition)('limit', (0, graphql_transformer_common_1.makeNamedType)('Int')),
                         (0, graphql_transformer_common_1.makeInputValueDefinition)('nextToken', (0, graphql_transformer_common_1.makeNamedType)('String')),
                     ];
-                case graphql_transformer_interfaces_1.QueryFieldType.SYNC:
+                }
+                case graphql_transformer_interfaces_1.QueryFieldType.SYNC: {
                     const syncFilterInputName = (0, graphql_transformer_common_1.toPascalCase)(['Model', type.name.value, 'FilterInput']);
                     const syncFilterInputs = (0, graphql_types_1.makeListQueryFilterInput)(ctx, syncFilterInputName, type);
                     const conditionInputName = syncFilterInputs.name.value;
@@ -571,7 +559,8 @@ class ModelTransformer extends graphql_transformer_core_1.TransformerModelBase {
                         (0, graphql_transformer_common_1.makeInputValueDefinition)('nextToken', (0, graphql_transformer_common_1.makeNamedType)('String')),
                         (0, graphql_transformer_common_1.makeInputValueDefinition)('lastSync', (0, graphql_transformer_common_1.makeNamedType)('AWSTimestamp')),
                     ];
-                case graphql_transformer_interfaces_1.MutationFieldType.CREATE:
+                }
+                case graphql_transformer_interfaces_1.MutationFieldType.CREATE: {
                     const createInputField = (0, graphql_types_1.makeCreateInputField)(type, this.modelDirectiveConfig.get(type.name.value), knownModels, ctx.inputDocument, isSyncEnabled);
                     const createInputTypeName = createInputField.name.value;
                     if (!ctx.output.getType(createInputField.name.value)) {
@@ -581,7 +570,8 @@ class ModelTransformer extends graphql_transformer_core_1.TransformerModelBase {
                         (0, graphql_transformer_common_1.makeInputValueDefinition)('input', (0, graphql_transformer_common_1.makeNonNullType)((0, graphql_transformer_common_1.makeNamedType)(createInputTypeName))),
                         (0, graphql_transformer_common_1.makeInputValueDefinition)('condition', (0, graphql_transformer_common_1.makeNamedType)(conditionInput.name.value)),
                     ];
-                case graphql_transformer_interfaces_1.MutationFieldType.DELETE:
+                }
+                case graphql_transformer_interfaces_1.MutationFieldType.DELETE: {
                     const deleteInputField = (0, graphql_types_1.makeDeleteInputField)(type, isSyncEnabled);
                     const deleteInputTypeName = deleteInputField.name.value;
                     if (!ctx.output.getType(deleteInputField.name.value)) {
@@ -591,7 +581,8 @@ class ModelTransformer extends graphql_transformer_core_1.TransformerModelBase {
                         (0, graphql_transformer_common_1.makeInputValueDefinition)('input', (0, graphql_transformer_common_1.makeNonNullType)((0, graphql_transformer_common_1.makeNamedType)(deleteInputTypeName))),
                         (0, graphql_transformer_common_1.makeInputValueDefinition)('condition', (0, graphql_transformer_common_1.makeNamedType)(conditionInput.name.value)),
                     ];
-                case graphql_transformer_interfaces_1.MutationFieldType.UPDATE:
+                }
+                case graphql_transformer_interfaces_1.MutationFieldType.UPDATE: {
                     const updateInputField = (0, graphql_types_1.makeUpdateInputField)(type, this.modelDirectiveConfig.get(type.name.value), knownModels, ctx.inputDocument, isSyncEnabled);
                     const updateInputTypeName = updateInputField.name.value;
                     if (!ctx.output.getType(updateInputField.name.value)) {
@@ -601,15 +592,14 @@ class ModelTransformer extends graphql_transformer_core_1.TransformerModelBase {
                         (0, graphql_transformer_common_1.makeInputValueDefinition)('input', (0, graphql_transformer_common_1.makeNonNullType)((0, graphql_transformer_common_1.makeNamedType)(updateInputTypeName))),
                         (0, graphql_transformer_common_1.makeInputValueDefinition)('condition', (0, graphql_transformer_common_1.makeNamedType)(conditionInput.name.value)),
                     ];
+                }
                 case graphql_transformer_interfaces_1.SubscriptionFieldType.ON_CREATE:
                 case graphql_transformer_interfaces_1.SubscriptionFieldType.ON_DELETE:
                 case graphql_transformer_interfaces_1.SubscriptionFieldType.ON_UPDATE:
                     return [];
-                    break;
                 default:
                     throw new Error('Unknown operation type');
             }
-            return [];
         };
         this.getOutputType = (ctx, type, operation) => {
             let outputType;
@@ -624,11 +614,12 @@ class ModelTransformer extends graphql_transformer_core_1.TransformerModelBase {
                     outputType = type;
                     break;
                 case graphql_transformer_interfaces_1.QueryFieldType.SYNC:
-                case graphql_transformer_interfaces_1.QueryFieldType.LIST:
+                case graphql_transformer_interfaces_1.QueryFieldType.LIST: {
                     const isSyncEnabled = ctx.isProjectUsingDataStore();
                     const connectionFieldName = (0, graphql_transformer_common_1.toPascalCase)(['Model', type.name.value, 'Connection']);
                     outputType = (0, graphql_types_1.makeListQueryModel)(type, connectionFieldName, isSyncEnabled);
                     break;
+                }
                 default:
                     throw new Error(`${operation.type} not supported for ${type.name.value}`);
             }
@@ -638,10 +629,11 @@ class ModelTransformer extends graphql_transformer_core_1.TransformerModelBase {
             return outputType;
         };
         this.createNonModelInputs = (ctx, obj) => {
-            for (let field of obj.fields || []) {
+            var _a;
+            ((_a = obj.fields) !== null && _a !== void 0 ? _a : []).forEach(field => {
                 if (!(0, graphql_transformer_common_1.isScalar)(field.type)) {
                     const def = ctx.output.getType((0, graphql_transformer_common_1.getBaseType)(field.type));
-                    if (def && def.kind == 'ObjectTypeDefinition' && !this.isModelField(def.name.value)) {
+                    if (def && def.kind === 'ObjectTypeDefinition' && !this.isModelField(def.name.value)) {
                         const name = this.getNonModelInputObjectName(def.name.value);
                         if (!ctx.output.getType(name)) {
                             const inputObj = graphql_transformer_core_1.InputObjectDefinitionWrapper.fromObject(name, def, ctx.inputDocument);
@@ -650,14 +642,10 @@ class ModelTransformer extends graphql_transformer_core_1.TransformerModelBase {
                         }
                     }
                 }
-            }
-        };
-        this.isModelField = (name) => {
-            return this.typesWithModelDirective.has(name) ? true : false;
-        };
-        this.getNonModelInputObjectName = (name) => {
-            return `${name}Input`;
+            });
         };
+        this.isModelField = (name) => (!!this.typesWithModelDirective.has(name));
+        this.getNonModelInputObjectName = (name) => `${name}Input`;
         this.addAutoGeneratableFields = (ctx, name) => {
             var _a, _b;
             const modelDirectiveConfig = this.modelDirectiveConfig.get(name);
@@ -679,7 +667,7 @@ class ModelTransformer extends graphql_transformer_core_1.TransformerModelBase {
                     timestamps.push((_b = modelDirectiveConfig.timestamps.updatedAt) !== null && _b !== void 0 ? _b : 'updatedAt');
                 }
             }
-            for (let fieldName of timestamps) {
+            timestamps.forEach(fieldName => {
                 if (typeWrapper.hasField(fieldName)) {
                     const field = typeWrapper.getField(fieldName);
                     if (!['String', 'AWSDateTime'].includes(field.getTypeName())) {
@@ -690,7 +678,7 @@ class ModelTransformer extends graphql_transformer_core_1.TransformerModelBase {
                     const field = graphql_transformer_core_1.FieldWrapper.create(fieldName, 'AWSDateTime');
                     typeWrapper.addField(field);
                 }
-            }
+            });
             ctx.output.updateObject(typeWrapper.serialize());
         };
         this.addModelSyncFields = (ctx, name) => {
@@ -704,28 +692,75 @@ class ModelTransformer extends graphql_transformer_core_1.TransformerModelBase {
             typeWrapper.addField(graphql_transformer_core_1.FieldWrapper.create('_lastChangedAt', 'AWSTimestamp'));
             ctx.output.updateObject(typeWrapper.serialize());
         };
-        this.getSubscriptionToMutationsReverseMap = (ctx, def) => {
+        this.getSubscriptionToMutationsReverseMap = (def) => {
             const subscriptionToMutationsMap = {};
-            const subscriptionFieldNames = this.getSubscriptionFieldNames(ctx, def);
-            for (const subscriptionFieldName of subscriptionFieldNames) {
+            const subscriptionFieldNames = this.getSubscriptionFieldNames(def);
+            subscriptionFieldNames.forEach(subscriptionFieldName => {
                 if (!subscriptionToMutationsMap[subscriptionFieldName.fieldName]) {
                     subscriptionToMutationsMap[subscriptionFieldName.fieldName] = [];
                 }
                 subscriptionToMutationsMap[subscriptionFieldName.fieldName].push(subscriptionFieldName);
-            }
+            });
             return subscriptionToMutationsMap;
         };
-        this.getOptions = (options) => {
-            return {
-                EnableDeletionProtection: false,
-                ...options,
-            };
-        };
-        this.ensureValidSubscriptionName = (name) => {
-            if (name.length <= 50)
-                return name;
-            return name.slice(0, 45) + (0, md5_1.default)(name).slice(0, 5);
-        };
+        this.createIAMRole = (context, def, stack, tableName) => {
+            const roleName = context.resourceHelper.generateIAMRoleName(graphql_transformer_common_1.ModelResourceIDs.ModelTableIAMRoleID(def.name.value));
+            const role = new iam.Role(stack, graphql_transformer_common_1.ModelResourceIDs.ModelTableIAMRoleID(def.name.value), {
+                roleName,
+                assumedBy: new iam.ServicePrincipal('appsync.amazonaws.com'),
+            });
+            const amplifyDataStoreTableName = context.resourceHelper.generateTableName(graphql_transformer_common_1.SyncResourceIDs.syncTableName);
+            role.attachInlinePolicy(new iam.Policy(stack, 'DynamoDBAccess', {
+                statements: [
+                    new iam.PolicyStatement({
+                        effect: iam.Effect.ALLOW,
+                        actions: [
+                            'dynamodb:BatchGetItem',
+                            'dynamodb:BatchWriteItem',
+                            'dynamodb:PutItem',
+                            'dynamodb:DeleteItem',
+                            'dynamodb:GetItem',
+                            'dynamodb:Scan',
+                            'dynamodb:Query',
+                            'dynamodb:UpdateItem',
+                        ],
+                        resources: [
+                            cdk.Fn.sub('arn:aws:dynamodb:${AWS::Region}:${AWS::AccountId}:table/${tablename}', {
+                                tablename: tableName,
+                            }),
+                            cdk.Fn.sub('arn:aws:dynamodb:${AWS::Region}:${AWS::AccountId}:table/${tablename}/*', {
+                                tablename: tableName,
+                            }),
+                            ...(context.isProjectUsingDataStore()
+                                ? [
+                                    cdk.Fn.sub('arn:aws:dynamodb:${AWS::Region}:${AWS::AccountId}:table/${tablename}', {
+                                        tablename: amplifyDataStoreTableName,
+                                    }),
+                                    cdk.Fn.sub('arn:aws:dynamodb:${AWS::Region}:${AWS::AccountId}:table/${tablename}/*', {
+                                        tablename: amplifyDataStoreTableName,
+                                    }),
+                                ]
+                                : []),
+                        ],
+                    }),
+                ],
+            }));
+            const syncConfig = graphql_transformer_core_1.SyncUtils.getSyncConfig(context, def.name.value);
+            if (syncConfig && graphql_transformer_core_1.SyncUtils.isLambdaSyncConfig(syncConfig)) {
+                role.attachInlinePolicy(graphql_transformer_core_1.SyncUtils.createSyncLambdaIAMPolicy(stack, syncConfig.LambdaConflictHandler.name, syncConfig.LambdaConflictHandler.region));
+            }
+            return role;
+        };
+        this.ensureModelSortDirectionEnum = (ctx) => {
+            if (!ctx.output.hasType('ModelSortDirection')) {
+                const modelSortDirection = (0, graphql_types_1.makeModelSortDirectionEnumObject)();
+                ctx.output.addEnum(modelSortDirection);
+            }
+        };
+        this.getOptions = (options) => ({
+            EnableDeletionProtection: false,
+            ...options,
+        });
         this.options = this.getOptions(options);
     }
     createModelTable(stack, def, context) {
@@ -786,7 +821,7 @@ class ModelTransformer extends graphql_transformer_core_1.TransformerModelBase {
             },
             stream: aws_dynamodb_1.StreamViewType.NEW_AND_OLD_IMAGES,
             encryption: aws_dynamodb_1.TableEncryption.DEFAULT,
-            removalPolicy: removalPolicy,
+            removalPolicy,
             ...(context.isProjectUsingDataStore() ? { timeToLiveAttribute: '_ttl' } : undefined),
         });
         const cfnTable = table.node.defaultChild;
@@ -838,60 +873,6 @@ class ModelTransformer extends graphql_transformer_core_1.TransformerModelBase {
         context.dataSources.add(def, dataSource);
         this.datasourceMap[def.name.value] = dataSource;
     }
-    createIAMRole(context, def, stack, tableName) {
-        const roleName = context.resourceHelper.generateIAMRoleName(graphql_transformer_common_1.ModelResourceIDs.ModelTableIAMRoleID(def.name.value));
-        const role = new iam.Role(stack, graphql_transformer_common_1.ModelResourceIDs.ModelTableIAMRoleID(def.name.value), {
-            roleName: roleName,
-            assumedBy: new iam.ServicePrincipal('appsync.amazonaws.com'),
-        });
-        const amplifyDataStoreTableName = context.resourceHelper.generateTableName(graphql_transformer_common_1.SyncResourceIDs.syncTableName);
-        role.attachInlinePolicy(new iam.Policy(stack, 'DynamoDBAccess', {
-            statements: [
-                new iam.PolicyStatement({
-                    effect: iam.Effect.ALLOW,
-                    actions: [
-                        'dynamodb:BatchGetItem',
-                        'dynamodb:BatchWriteItem',
-                        'dynamodb:PutItem',
-                        'dynamodb:DeleteItem',
-                        'dynamodb:GetItem',
-                        'dynamodb:Scan',
-                        'dynamodb:Query',
-                        'dynamodb:UpdateItem',
-                    ],
-                    resources: [
-                        cdk.Fn.sub('arn:aws:dynamodb:${AWS::Region}:${AWS::AccountId}:table/${tablename}', {
-                            tablename: tableName,
-                        }),
-                        cdk.Fn.sub('arn:aws:dynamodb:${AWS::Region}:${AWS::AccountId}:table/${tablename}/*', {
-                            tablename: tableName,
-                        }),
-                        ...(context.isProjectUsingDataStore()
-                            ? [
-                                cdk.Fn.sub('arn:aws:dynamodb:${AWS::Region}:${AWS::AccountId}:table/${tablename}', {
-                                    tablename: amplifyDataStoreTableName,
-                                }),
-                                cdk.Fn.sub('arn:aws:dynamodb:${AWS::Region}:${AWS::AccountId}:table/${tablename}/*', {
-                                    tablename: amplifyDataStoreTableName,
-                                }),
-                            ]
-                            : []),
-                    ],
-                }),
-            ],
-        }));
-        const syncConfig = graphql_transformer_core_1.SyncUtils.getSyncConfig(context, def.name.value);
-        if (syncConfig && graphql_transformer_core_1.SyncUtils.isLambdaSyncConfig(syncConfig)) {
-            role.attachInlinePolicy(graphql_transformer_core_1.SyncUtils.createSyncLambdaIAMPolicy(stack, syncConfig.LambdaConflictHandler.name, syncConfig.LambdaConflictHandler.region));
-        }
-        return role;
-    }
-    ensureModelSortDirectionEnum(ctx) {
-        if (!ctx.output.hasType('ModelSortDirection')) {
-            const modelSortDirection = (0, graphql_types_1.makeModelSortDirectionEnumObject)();
-            ctx.output.addEnum(modelSortDirection);
-        }
-    }
 }
 exports.ModelTransformer = ModelTransformer;
 //# sourceMappingURL=graphql-model-transformer.js.map