fix non-null type support in generated mutations (#7147)

* fix non-null type support in generated mutations * remove console log
firebase · May 9, 2024 · d30d96c · d30d96c
1 parent de9cbde
commit d30d96c
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/firebase-vscode/src/data-connect/ad-hoc-mutations.ts b/firebase-vscode/src/data-connect/ad-hoc-mutations.ts
@@ -160,11 +160,14 @@ query {
     mutation.push(`${functionSpacing}${name}_insert(data: {`); // insert function
     for (const field of ast.fields) {
       // necessary to avoid type error
-      const fieldType: any = field.type;
-      let fieldTypeName: string = fieldType.type.name.value;
+      let fieldType: any = field.type;
+      // We unwrap NonNullType to obtain the actual type
+      if (fieldType.kind === Kind.NON_NULL_TYPE) {
+        fieldType = fieldType.type;
+      }
+      let fieldTypeName: string = fieldType.name.value;
       let fieldName: string = field.name.value;
       let defaultValue = defaultScalarValues[fieldTypeName] as string;
-
       if (!isDataConnectScalarType(fieldTypeName)) {
         fieldTypeName += "Id";
         fieldName += "Id";