@aws-amplify/data-schema 1.20.0 → 1.20.1

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@aws-amplify/data-schema",
-  "version": "1.20.0",
+  "version": "1.20.1",
   "license": "Apache-2.0",
   "repository": {
     "type": "git",

package/src/SchemaProcessor.ts

@@ -1444,8 +1444,24 @@ function generateInputTypes(
     if (isRefField(argDef)) {
       const refType = getRefType(argDef.data.link, operationName);
       if (refType.type === 'CustomType') {
+        const { valueRequired, array, arrayRequired } = argDef.data;
+
         const inputTypeName = `${argDef.data.link}Input`;
-        argDefinitions.push(`${argName}: ${inputTypeName}`);
+        let field = inputTypeName;
+
+        if (valueRequired === true) {
+          field += '!';
+        }
+
+        if (array) {
+          field = `[${field}]`;
+
+          if (arrayRequired === true) {
+            field += '!';
+          }
+        }
+
+        argDefinitions.push(`${argName}: ${field}`);
 
         // Process the input type if it hasn't been processed yet
         if (!processedTypes.has(inputTypeName)) {
@@ -1462,7 +1478,23 @@ function generateInputTypes(
           });
         }
       } else if (refType.type === 'Enum') {
-        argDefinitions.push(`${argName}: ${argDef.data.link}`);
+        const { valueRequired, array, arrayRequired } = argDef.data;
+
+        let field = argDef.data.link;
+
+        if (valueRequired === true) {
+          field += '!';
+        }
+
+        if (array) {
+          field = `[${field}]`;
+
+          if (arrayRequired === true) {
+            field += '!';
+          }
+        }
+
+        argDefinitions.push(`${argName}: ${field}`);
       } else {
         throw new Error(
           `Unsupported reference type '${refType.type}' for argument '${argName}' in '${operationName}'. ` +