fix polymorphic uuid struct fields

lolopinto · Nov 13, 2024 · c6296b7 · c6296b7
1 parent 095c305
commit c6296b7
Show file tree

Hide file tree

Showing 15 changed files with 93 additions and 18 deletions.
diff --git a/examples/simple/src/ent/generated/types.ts b/examples/simple/src/ent/generated/types.ts
diff --git a/examples/simple/src/graphql/generated/mutations/contact/contact_create_type.ts b/examples/simple/src/graphql/generated/mutations/contact/contact_create_type.ts
diff --git a/examples/simple/src/graphql/generated/mutations/contact/contact_edit_type.ts b/examples/simple/src/graphql/generated/mutations/contact/contact_edit_type.ts
diff --git a/examples/simple/src/graphql/generated/mutations/event/event_create_type.ts b/examples/simple/src/graphql/generated/mutations/event/event_create_type.ts
diff --git a/examples/simple/src/graphql/generated/mutations/event/event_edit_type.ts b/examples/simple/src/graphql/generated/mutations/event/event_edit_type.ts
diff --git a/examples/simple/src/graphql/generated/mutations/input/attachment_input_type.ts b/examples/simple/src/graphql/generated/mutations/input/attachment_input_type.ts
diff --git a/examples/simple/src/graphql/generated/resolvers/attachment_type.ts b/examples/simple/src/graphql/generated/resolvers/attachment_type.ts
diff --git a/examples/simple/src/graphql/generated/resolvers/contact_info_type.ts b/examples/simple/src/graphql/generated/resolvers/contact_info_type.ts
diff --git a/examples/simple/src/graphql/generated/resolvers/feedback_type.ts b/examples/simple/src/graphql/generated/resolvers/feedback_type.ts
diff --git a/examples/simple/src/graphql/generated/resolvers/with_day_of_week_type.ts b/examples/simple/src/graphql/generated/resolvers/with_day_of_week_type.ts
diff --git a/examples/simple/src/graphql/generated/schema.gql b/examples/simple/src/graphql/generated/schema.gql
diff --git a/examples/simple/src/graphql/generated/schema.ts b/examples/simple/src/graphql/generated/schema.ts
diff --git a/examples/simple/src/graphql/resolvers/internal.ts b/examples/simple/src/graphql/resolvers/internal.ts
@@ -20,7 +20,6 @@ export * from "../generated/resolvers/user_super_nested_object_type";
 export * from "../generated/resolvers/address_type";
 export * from "../generated/resolvers/comment_type";
 export * from "../generated/resolvers/contact_email_type";
-export * from "../generated/resolvers/contact_info_type";
 export * from "../generated/resolvers/contact_phone_number_type";
 export * from "../generated/resolvers/contact_type";
 export * from "../generated/resolvers/event_type";

diff --git a/examples/simple/src/schema/__global__schema.ts b/examples/simple/src/schema/__global__schema.ts
@@ -76,6 +76,19 @@ const glo: GlobalSchema = {
         date: TimestampType(),
         phone_number: PhoneNumberType({ nullable: true }),
         email_address: EmailType({ nullable: true }),
+        creator_id: UUIDType({
+          polymorphic: {
+            types: ["User"],
+          },
+          nullable: true,
+        }),
+        // doesn't seem like we support this...
+        // owner_ids: UUIDListType({
+        //   polymorphic: {
+        //     types: ["User"],
+        //   },
+        //   nullable: true,
+        // }),
       },
     }),
   },

diff --git a/internal/graphql/generate_ts_code.go b/internal/graphql/generate_ts_code.go
@@ -2417,16 +2417,22 @@ func addSingularEdge(edge edge.Edge, obj *objectType) error {
 	return obj.addField(gqlField)
 }
 
-func addSingularEdgeFromExternal(processor *codegen.Processor, edge edge.Edge, f *field.Field, obj *objectType) error {
+func addSingularEdgeFromExternal(processor *codegen.Processor, edge *edge.FieldEdge, f *field.Field, obj *objectType) error {
 	gqlField := createFieldTypeForSingularEdge(edge)
 	var contents []string
 
 	tsFieldName := f.TsFieldName(processor.Config)
 	if f.Nullable() {
 		contents = append(contents, fmt.Sprintf("if (obj.%s === null || obj.%s === undefined)	{ return null;}", tsFieldName, tsFieldName))
 	}
-	contents = append(contents, fmt.Sprintf("return %s.load(context.getViewer(), obj.%s);", edge.GetNodeInfo().Node, tsFieldName))
-	gqlField.ExtraImports = append(gqlField.ExtraImports, tsimport.NewLocalEntImportPath(edge.GetNodeInfo().Node))
+	// TODO make this easier to do. shouldn't have to remember this
+	if edge.Polymorphic != nil {
+		contents = append(contents, fmt.Sprintf("return loadEntByType(context.getViewer(), obj.%s as unknown as NodeType, obj.%s);", edge.Polymorphic.NodeTypeField, edge.TSFieldName))
+		gqlField.ExtraImports = append(gqlField.ExtraImports, tsimport.NewTypesEntImportPath("NodeType"), tsimport.NewLocalEntImportPath("loadEntByType"))
+	} else {
+		contents = append(contents, fmt.Sprintf("return %s.load(context.getViewer(), obj.%s);", edge.GetNodeInfo().Node, tsFieldName))
+		gqlField.ExtraImports = append(gqlField.ExtraImports, tsimport.NewLocalEntImportPath(edge.GetNodeInfo().Node))
+	}
 	gqlField.FunctionContents = contents
 	return obj.addField(gqlField)
 }
@@ -2445,7 +2451,7 @@ func addPluralEdge(edge edge.Edge, obj *objectType) error {
 	return obj.addField(gqlField)
 }
 
-func addPluralEdgeFromExternal(processor *codegen.Processor, edge edge.Edge, f *field.Field, obj *objectType) error {
+func addPluralEdgeFromExternal(processor *codegen.Processor, edge *edge.FieldEdge, f *field.Field, obj *objectType) error {
 	gqlField := createFieldTypeForPluralEdge(edge)
 	gqlField.HasAsyncModifier = true
 
@@ -2454,10 +2460,17 @@ func addPluralEdgeFromExternal(processor *codegen.Processor, edge edge.Edge, f *
 	if f.Nullable() {
 		contents = append(contents, fmt.Sprintf("if (obj.%s === null || obj.%s === undefined)	{ return null;}", tsFieldName, tsFieldName))
 	}
-	contents = append(contents, fmt.Sprintf("const objs = await %s.loadMany(context.getViewer(), ...obj.%s);", edge.GetNodeInfo().Node, tsFieldName))
-	contents = append(contents, "return Array.from(objs.values());")
+	if edge.Polymorphic != nil {
+		// PS: doesn't seem like we support this. see examples/simple/src/schema/__global__schema.ts
+		contents = append(contents, fmt.Sprintf("const objs = await Promise.all(obj.%s.map((id, idx) => loadEntByType(context.getViewer(), obj.%s[idx] as unknown as NodeType, id));", edge.Polymorphic.NodeTypeField, edge.TSFieldName))
+		contents = append(contents, "return objs.filter((obj) => obj !== null);")
+		gqlField.ExtraImports = append(gqlField.ExtraImports, tsimport.NewTypesEntImportPath("NodeType"), tsimport.NewLocalEntImportPath("loadEntByType"))
 
-	gqlField.ExtraImports = append(gqlField.ExtraImports, tsimport.NewLocalEntImportPath(edge.GetNodeInfo().Node))
+	} else {
+		contents = append(contents, fmt.Sprintf("const objs = await %s.loadMany(context.getViewer(), ...obj.%s);", edge.GetNodeInfo().Node, tsFieldName))
+		contents = append(contents, "return Array.from(objs.values());")
+		gqlField.ExtraImports = append(gqlField.ExtraImports, tsimport.NewLocalEntImportPath(edge.GetNodeInfo().Node))
+	}
 	gqlField.FunctionContents = contents
 	return obj.addField(gqlField)
 }