fix: support x-kubernetes-int-or-string type flag (#23)

Author: amyXia1994

examples/kube_resource/complex/muti_models/models/core_oam_dev_v1alpha2_containerized_workload.k

@@ -430,14 +430,14 @@ schema CoreOamDevV1alpha2ContainerizedWorkloadSpecContainersItems0ResourcesExten
     ----------
     name : str, default is Undefined, required
          Name of the external resource. Resource names are specified in kind.group/version format, e.g. motionsensor.ext.example.com/v1.
-    required : int, default is Undefined, required
+    required : int | str, default is Undefined, required
          Required extended resource(s), e.g. 8 or "very-cool-widget"
     """
 
 
     name: str
 
-    required: int
+    required: int | str
 
 
 schema CoreOamDevV1alpha2ContainerizedWorkloadSpecContainersItems0ResourcesGpu:

pkg/kube_resource/generator/testdata/muti_models/models/core_oam_dev_v1alpha2_containerized_workload.k

@@ -430,14 +430,14 @@ schema CoreOamDevV1alpha2ContainerizedWorkloadSpecContainersItems0ResourcesExten
     ----------
     name : str, default is Undefined, required
          Name of the external resource. Resource names are specified in kind.group/version format, e.g. motionsensor.ext.example.com/v1.
-    required : int, default is Undefined, required
+    required : int | str, default is Undefined, required
          Required extended resource(s), e.g. 8 or "very-cool-widget"
     """
 
 
     name: str
 
-    required: int
+    required: int | str
 
 
 schema CoreOamDevV1alpha2ContainerizedWorkloadSpecContainersItems0ResourcesGpu:

…am_dev_v1alpha2_containerized_workload.k …am_dev_v1alpha2_containerized_workload.kpkg/kube_resource/generator/testdata/fix_me_union_type/models/core_oam_dev_v1alpha2_containerized_workload.k renamed to pkg/kube_resource/generator/testdata/union_type/models/core_oam_dev_v1alpha2_containerized_workload.k pkg/kube_resource/generator/testdata/fix_me_union_type/models/core_oam_dev_v1alpha2_containerized_workload.k renamed to pkg/kube_resource/generator/testdata/union_type/models/core_oam_dev_v1alpha2_containerized_workload.k

@@ -21,9 +21,9 @@ schema ContainerizedWorkload:
     """
 
 
-    final apiVersion: str = "core.oam.dev/v1alpha2"
+    apiVersion: "core.oam.dev/v1alpha2" = "core.oam.dev/v1alpha2"
 
-    final kind: str = "ContainerizedWorkload"
+    kind: "ContainerizedWorkload" = "ContainerizedWorkload"
 
     metadata?: apis.ObjectMeta
 

…apimachinery/apis/managed_fields_entry.k …apimachinery/apis/managed_fields_entry.kpkg/kube_resource/generator/testdata/fix_me_union_type/models/kusion_kubernetes/apimachinery/apis/managed_fields_entry.k renamed to pkg/kube_resource/generator/testdata/union_type/models/kusion_kubernetes/apimachinery/apis/managed_fields_entry.k pkg/kube_resource/generator/testdata/fix_me_union_type/models/kusion_kubernetes/apimachinery/apis/managed_fields_entry.k renamed to pkg/kube_resource/generator/testdata/union_type/models/kusion_kubernetes/apimachinery/apis/managed_fields_entry.k

@@ -37,7 +37,8 @@ const (
 )
 
 const (
-	intOrStr = "intorstring"
+	intOrStr        = "intorstring"
+	k8sIntOrStrFlag = "x-kubernetes-int-or-string"
 )
 
 // Extensions supported by go-swagger
@@ -187,7 +188,7 @@ func (t *typeResolver) resolveSchemaRef(schema *spec.Schema, isRequired bool) (r
 	return
 }
 
-func (t *typeResolver) resolveFormat(schema *spec.Schema, isAnonymous bool, isRequired bool) (returns bool, result resolvedType, err error) {
+func (t *typeResolver) resolveFormat(schema *spec.Schema, isAnonymous, isRequired bool) (returns bool, result resolvedType, err error) {
 	if schema.Format != "" {
 		// defaults to string
 		result.SwaggerType = str
@@ -215,6 +216,24 @@ func (t *typeResolver) resolveFormat(schema *spec.Schema, isAnonymous bool, isRe
 	return
 }
 
+func (t *typeResolver) resolveExtensions(schema *spec.Schema, isAnonymous, isRequired bool) (returns bool, result resolvedType, err error) {
+	if schema.VendorExtensible.Extensions != nil {
+		if value, ok := schema.VendorExtensible.Extensions.GetBool(k8sIntOrStrFlag); value && ok {
+			// the schema has {"x-kubernetes-int-or-string": "true"} flag
+			debugLog("resolving x-kubernetes-int-or-string type flag (anon: %t, req: %t)", isAnonymous, isRequired)
+			returns = true
+			result.SwaggerType = str
+			if len(schema.Type) > 0 {
+				result.SwaggerType = schema.Type[0]
+			}
+			result.KclType = typeMapping[intOrStr]
+			// propagate extensions in resolvedType
+			result.Extensions = schema.Extensions
+		}
+	}
+	return
+}
+
 func (t *typeResolver) firstType(schema *spec.Schema) string {
 	if len(schema.Type) == 0 || schema.Type[0] == "" {
 		return object
@@ -368,6 +387,12 @@ func (t *typeResolver) ResolveSchema(schema *spec.Schema, isAnonymous, isRequire
 		return
 	}
 
+	returns, result, err = t.resolveExtensions(schema, isAnonymous, isRequired)
+	if returns || err != nil {
+		debugLog("returning after resolve vendor extensions: %s", pretty.Sprint(result))
+		return
+	}
+
 	switch tpe {
 	case array:
 		result, err = t.resolveArray(schema, isAnonymous, false)