fix: CustomType for SingleTypeOf attribute (#60)

* fix: CustomType for SingleTypeOf attribute * ci(generate): changes by go generate * chore: add changelog --------- Co-authored-by: azrod <[email protected]>
FrangipaneTeam · Nov 6, 2023 · df52cf5 · df52cf5
1 parent 9b08c28
commit df52cf5
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 4 deletions.
diff --git a/.changelog/60.txt b/.changelog/60.txt
@@ -0,0 +1,3 @@
+```release-note:bug
+`SingleNestedAttributeOf` - Fix custom type for Single Nested Attribute of
+```
diff --git a/supersingle_nested_attribute.go b/supersingle_nested_attribute.go
@@ -244,7 +244,7 @@ func (s SuperSingleNestedAttributeOf[T]) GetResource(ctx context.Context) schema
 	}
 	// * If user has not provided a custom type, we will use the default supertypes
 	if a.CustomType == nil {
-		a.CustomType = supertypes.NewSingleNestedObjectTypeOf[T](ctx).ObjectType
+		a.CustomType = supertypes.NewSingleNestedObjectTypeOf[T](ctx)
 	}
 
 	deprecationMessage := ""
@@ -299,7 +299,7 @@ func (s SuperSingleNestedAttributeOf[T]) GetDataSource(ctx context.Context) sche
 	}
 	// * If user has not provided a custom type, we will use the default supertypes
 	if a.CustomType == nil {
-		a.CustomType = supertypes.NewSingleNestedObjectTypeOf[T](ctx).ObjectType
+		a.CustomType = supertypes.NewSingleNestedObjectTypeOf[T](ctx)
 	}
 
 	deprecationMessage := ""

diff --git a/supertype_attribute.go.tmpl b/supertype_attribute.go.tmpl
@@ -489,7 +489,7 @@ func (s Super{{ .TypeName }}AttributeOf[T]) GetResource(ctx context.Context) sch
 	{{- if or (eq .TypeName "SingleNested") }}
 	// * If user has not provided a custom type, we will use the default supertypes
 	if a.CustomType == nil {
-		a.CustomType = supertypes.NewSingleNestedObjectTypeOf[T](ctx).ObjectType
+		a.CustomType = supertypes.NewSingleNestedObjectTypeOf[T](ctx)
 	}
     {{- end }}
 
@@ -593,7 +593,7 @@ func (s Super{{ .TypeName }}AttributeOf[T]) GetDataSource(ctx context.Context) s
 	{{- if or (eq .TypeName "SingleNested") }}
 	// * If user has not provided a custom type, we will use the default supertypes
 	if a.CustomType == nil {
-		a.CustomType = supertypes.NewSingleNestedObjectTypeOf[T](ctx).ObjectType
+		a.CustomType = supertypes.NewSingleNestedObjectTypeOf[T](ctx)
 	}
     {{- end }}