Revert "fix: use related fields for has_one and belongs_to properties…

… when building GraphQL API payloads" This reverts commit e12116a.
aws-amplify · Aug 11, 2023 · c7df29a · c7df29a
1 parent 4fdcb32
commit c7df29a
Show file tree

Hide file tree

Showing 9 changed files with 85 additions and 160 deletions.
diff --git a/...s/codegen-ui-react/lib/__tests__/__snapshots__/studio-ui-codegen-react-forms.test.ts.snap b/...s/codegen-ui-react/lib/__tests__/__snapshots__/studio-ui-codegen-react-forms.test.ts.snap
diff --git a/packages/codegen-ui-react/lib/__tests__/studio-ui-codegen-react-forms.test.ts b/packages/codegen-ui-react/lib/__tests__/studio-ui-codegen-react-forms.test.ts
@@ -1001,9 +1001,9 @@ describe('amplify form renderer tests', () => {
       );
 
       expect(componentText).toContain('postCommentsId');
-      expect(componentText).toContain('postID');
-      expect(componentText).toContain('userCommentsId');
-      expect(componentText).toContain('orgCommentsId');
+      expect(componentText).not.toContain('postID');
+      expect(componentText).not.toContain('userCommentsId');
+      expect(componentText).not.toContain('orgCommentsId');
       expect(componentText).toMatchSnapshot();
       expect(declaration).toMatchSnapshot();
     });

diff --git a/packages/codegen-ui-react/lib/amplify-ui-renderers/form.ts b/packages/codegen-ui-react/lib/amplify-ui-renderers/form.ts
@@ -323,13 +323,7 @@ export default class FormRenderer extends ReactComponentRenderer<BaseComponentPr
                   [],
                 ),
               ),
-              buildModelFieldObject(
-                dataSourceType !== 'DataStore',
-                formMetadata?.fieldConfigs,
-                this.componentMetadata.dataSchemaMetadata?.models,
-                undefined,
-                this.isRenderingGraphQL(),
-              ),
+              buildModelFieldObject(dataSourceType !== 'DataStore', formMetadata?.fieldConfigs),
               ...onSubmitValidationRun(hasModelField),
               ...this.getOnSubmitDSCall(),
             ],

diff --git a/packages/codegen-ui-react/lib/forms/form-renderer-helper/all-props.ts b/packages/codegen-ui-react/lib/forms/form-renderer-helper/all-props.ts
@@ -113,9 +113,7 @@ export const addFormAttributes = (
     if (formMetadata.formActionType === 'update' && !fieldConfig.isArray && !isControlledComponent(componentType)) {
       attributes.push(renderDefaultValueAttribute(renderedVariableName, fieldConfig, componentType));
     }
-    attributes.push(
-      buildOnChangeStatement(component, formMetadata.fieldConfigs, dataApi, dataSchema?.models, dataApi === 'GraphQL'),
-    );
+    attributes.push(buildOnChangeStatement(component, formMetadata.fieldConfigs, dataApi));
     attributes.push(buildOnBlurStatement(componentName, fieldConfig));
     attributes.push(
       factory.createJsxAttribute(

diff --git a/packages/codegen-ui-react/lib/forms/form-renderer-helper/event-handler-props.ts b/packages/codegen-ui-react/lib/forms/form-renderer-helper/event-handler-props.ts
@@ -22,7 +22,6 @@ import {
   isValidVariableName,
   shouldIncludeCancel,
   InvalidInputError,
-  GenericDataModel,
 } from '@aws-amplify/codegen-ui';
 import {
   BindingElement,
@@ -212,8 +211,6 @@ export const buildOverrideOnChangeStatement = (
   fieldName: string,
   fieldConfigs: Record<string, FieldConfigMetadata>,
   valueNameOverride?: Identifier,
-  models?: Record<string, GenericDataModel>,
-  isRenderingGraphQL = false,
 ): IfStatement => {
   const keyPath = fieldName.split('.');
   const keyName = keyPath[0];
@@ -232,15 +229,9 @@ export const buildOverrideOnChangeStatement = (
     factory.createIdentifier('onChange'),
     factory.createBlock(
       [
-        buildModelFieldObject(
-          true,
-          fieldConfigs,
-          models || {},
-          {
-            [keyName]: keyValueExpression,
-          },
-          isRenderingGraphQL,
-        ),
+        buildModelFieldObject(true, fieldConfigs, {
+          [keyName]: keyValueExpression,
+        }),
         factory.createVariableStatement(
           undefined,
           factory.createVariableDeclarationList(
@@ -295,9 +286,7 @@ function getCallbackVarName(fieldType: string): string {
 export const buildOnChangeStatement = (
   component: StudioComponent | StudioComponentChild,
   fieldConfigs: Record<string, FieldConfigMetadata>,
-  dataApi: DataApiKind | undefined,
-  models: Record<string, GenericDataModel> = {},
-  isRenderingGraphQL = false,
+  dataApi?: DataApiKind,
 ) => {
   const { name: fieldName, componentType: fieldType } = component;
   const fieldConfig = fieldConfigs[fieldName];
@@ -320,9 +309,7 @@ export const buildOnChangeStatement = (
   }
 
   if (!shouldWrapInArrayField(fieldConfig)) {
-    handleChangeStatements.push(
-      buildOverrideOnChangeStatement(fieldName, fieldConfigs, undefined, models, isRenderingGraphQL),
-    );
+    handleChangeStatements.push(buildOverrideOnChangeStatement(fieldName, fieldConfigs));
   }
 
   handleChangeStatements.push(getOnChangeValidationBlock(fieldName));
@@ -528,7 +515,6 @@ export const buildStorageManagerOnChangeStatement = (
   component: StudioComponent | StudioComponentChild,
   fieldConfigs: Record<string, FieldConfigMetadata>,
   handlerName: 'onUploadSuccess' | 'onFileRemove',
-  isRenderingGraphQL: boolean,
 ) => {
   const { name: fieldName } = component;
   const fieldConfig = fieldConfigs[fieldName];
@@ -595,7 +581,7 @@ export const buildStorageManagerOnChangeStatement = (
                           NodeFlags.Let,
                         ),
                       ),
-                      buildOverrideOnChangeStatement(fieldName, fieldConfigs, undefined, undefined, isRenderingGraphQL),
+                      buildOverrideOnChangeStatement(fieldName, fieldConfigs),
                       factory.createReturnStatement(factory.createIdentifier('value')),
                     ],
                     true,

diff --git a/packages/codegen-ui-react/lib/forms/form-renderer-helper/model-fields.ts b/packages/codegen-ui-react/lib/forms/form-renderer-helper/model-fields.ts
@@ -14,7 +14,7 @@
   limitations under the License.
  */
 import { factory, NodeFlags, ObjectLiteralElementLike } from 'typescript';
-import { FieldConfigMetadata, GenericDataModel } from '@aws-amplify/codegen-ui';
+import { FieldConfigMetadata } from '@aws-amplify/codegen-ui';
 
 /**
  * builds modelFields object which is used to validate, onSubmit, onSuccess/onError
@@ -33,49 +33,30 @@ import { FieldConfigMetadata, GenericDataModel } from '@aws-amplify/codegen-ui';
 export const buildModelFieldObject = (
   shouldBeConst: boolean,
   fieldConfigs: Record<string, FieldConfigMetadata> = {},
-  models: Record<string, GenericDataModel> = {},
   nameOverrides: Record<string, ObjectLiteralElementLike> = {},
-  isRenderingGraphQL = false,
 ) => {
   const fieldSet = new Set<string>();
   const fields = Object.keys(fieldConfigs).reduce<ObjectLiteralElementLike[]>((acc, value) => {
     const fieldName = value.split('.')[0];
-    const { sanitizedFieldName, relationship } = fieldConfigs[value];
+    const { sanitizedFieldName } = fieldConfigs[value];
     const renderedFieldName = sanitizedFieldName || fieldName;
     if (!fieldSet.has(renderedFieldName)) {
-      let assignments: ObjectLiteralElementLike[] = [
-        factory.createShorthandPropertyAssignment(factory.createIdentifier(fieldName), undefined),
-      ];
+      let assignment: ObjectLiteralElementLike = factory.createShorthandPropertyAssignment(
+        factory.createIdentifier(fieldName),
+        undefined,
+      );
+
       if (nameOverrides[fieldName]) {
-        assignments = [nameOverrides[fieldName]];
-      } else if (
-        isRenderingGraphQL &&
-        typeof fieldConfigs[value].dataType === 'object' &&
-        relationship &&
-        (relationship.type === 'BELONGS_TO' || relationship.type === 'HAS_ONE')
-      ) {
-        assignments =
-          relationship.associatedFields?.map((associatedField, index) => {
-            return factory.createPropertyAssignment(
-              factory.createStringLiteral(associatedField),
-              factory.createPropertyAccessChain(
-                factory.createIdentifier(fieldName),
-                undefined,
-                models[relationship.relatedModelName].primaryKeys[index],
-              ),
-            );
-          }) || [];
+        assignment = nameOverrides[fieldName];
       } else if (sanitizedFieldName) {
         // if overrides present, ignore sanitizedFieldName
-        assignments = [
-          factory.createPropertyAssignment(
-            factory.createStringLiteral(fieldName),
-            factory.createIdentifier(sanitizedFieldName),
-          ),
-        ];
+        assignment = factory.createPropertyAssignment(
+          factory.createStringLiteral(fieldName),
+          factory.createIdentifier(sanitizedFieldName),
+        );
       }
 
-      acc.push(...assignments);
+      acc.push(assignment);
       fieldSet.add(renderedFieldName);
     }
     return acc;

diff --git a/packages/codegen-ui-react/lib/forms/form-renderer-helper/render-array-field.ts b/packages/codegen-ui-react/lib/forms/form-renderer-helper/render-array-field.ts
@@ -13,7 +13,7 @@
   See the License for the specific language governing permissions and
   limitations under the License.
  */
-import { FieldConfigMetadata, GenericDataModel, LabelDecorator } from '@aws-amplify/codegen-ui';
+import { FieldConfigMetadata, LabelDecorator } from '@aws-amplify/codegen-ui';
 import { Expression, factory, Identifier, JsxAttribute, JsxChild, NodeFlags, SyntaxKind } from 'typescript';
 import {
   buildAccessChain,
@@ -38,16 +38,12 @@ function getOnChangeAttribute({
   renderedFieldName,
   fieldConfigs,
   isLimitedToOneValue,
-  models,
-  isRenderingGraphQL,
 }: {
   setStateName: Identifier;
   fieldName: string;
   renderedFieldName: string;
   fieldConfigs: Record<string, FieldConfigMetadata>;
   isLimitedToOneValue?: boolean;
-  models: Record<string, GenericDataModel> | undefined;
-  isRenderingGraphQL: boolean;
 }): JsxAttribute {
   const fieldConfig = fieldConfigs[fieldName];
   const { dataType, componentType } = fieldConfig;
@@ -101,7 +97,7 @@ function getOnChangeAttribute({
                 NodeFlags.Let,
               ),
             ),
-            buildOverrideOnChangeStatement(fieldName, fieldConfigs, valueName, models, isRenderingGraphQL),
+            buildOverrideOnChangeStatement(fieldName, fieldConfigs, valueName),
             ...setStateStatements,
           ],
           true,
@@ -136,7 +132,6 @@ export const renderArrayFieldComponent = (
   labelDecorator?: LabelDecorator,
   isRequired?: boolean,
   dataApi: DataApiKind = 'DataStore',
-  models: Record<string, GenericDataModel> = {},
 ) => {
   const fieldConfig = fieldConfigs[fieldName];
   const { sanitizedFieldName, dataType, componentType } = fieldConfig;
@@ -177,17 +172,7 @@ export const renderArrayFieldComponent = (
     );
   }
 
-  props.push(
-    getOnChangeAttribute({
-      fieldName,
-      isLimitedToOneValue,
-      fieldConfigs,
-      renderedFieldName,
-      setStateName,
-      models,
-      isRenderingGraphQL: dataApi === 'GraphQL',
-    }),
-  );
+  props.push(getOnChangeAttribute({ fieldName, isLimitedToOneValue, fieldConfigs, renderedFieldName, setStateName }));
   let labelAttribute = factory.createJsxAttribute(
     factory.createIdentifier('label'),
     factory.createJsxExpression(undefined, factory.createStringLiteral(fieldLabel)),

diff --git a/packages/codegen-ui-react/lib/react-component-renderer.ts b/packages/codegen-ui-react/lib/react-component-renderer.ts
@@ -98,7 +98,6 @@ export class ReactComponentRenderer<TPropIn> extends ComponentRendererBase<
           fieldConfigs,
           labelDecorator,
           isRequired,
-          this.importCollection.rendererConfig?.apiConfiguration?.dataApi === 'GraphQL',
         );
       }
 
@@ -119,7 +118,6 @@ export class ReactComponentRenderer<TPropIn> extends ComponentRendererBase<
           labelDecorator,
           isRequired,
           this.importCollection.rendererConfig?.apiConfiguration?.dataApi,
-          this.componentMetadata.dataSchemaMetadata?.models,
         );
       }
     }

diff --git a/packages/codegen-ui-react/lib/utils/forms/storage-field-component.ts b/packages/codegen-ui-react/lib/utils/forms/storage-field-component.ts
@@ -324,7 +324,6 @@ export const renderStorageFieldComponent = (
   fieldConfigs: Record<string, FieldConfigMetadata>,
   labelDecorator?: LabelDecorator,
   isRequired?: boolean,
-  isRenderingGraphQL = false,
 ) => {
   const { name: componentName } = component;
   const dataTypeName = componentMetadata.formMetadata?.dataType.dataTypeName || '';
@@ -419,12 +418,8 @@ export const renderStorageFieldComponent = (
       );
     }
 
-    storageManagerAttributes.push(
-      buildStorageManagerOnChangeStatement(component, fieldConfigs, 'onUploadSuccess', isRenderingGraphQL),
-    );
-    storageManagerAttributes.push(
-      buildStorageManagerOnChangeStatement(component, fieldConfigs, 'onFileRemove', isRenderingGraphQL),
-    );
+    storageManagerAttributes.push(buildStorageManagerOnChangeStatement(component, fieldConfigs, 'onUploadSuccess'));
+    storageManagerAttributes.push(buildStorageManagerOnChangeStatement(component, fieldConfigs, 'onFileRemove'));
     storageManagerAttributes.push(
       factory.createJsxAttribute(
         factory.createIdentifier('processFile'),