func Modify()

in cmd/generator/cloudspec/spec_modifier.go [32:171]
115 lines of code
48 McCabe index (conditional complexity)

func Modify(cloudSpec *spec.Swagger) {
	// Iterate over the spec's definitions.
	for k, def := range cloudSpec.Definitions {
		// This hack is required to remove all required fields from `ProxiesSettings` and allow clients
		// to use this model as structured json on Patch API request body.
		if k == "ProxiesSettings" {
			schema := cloudSpec.Definitions[k]
			schema.WithRequired([]string{}...)
			cloudSpec.Definitions[k] = schema
		}

		for kk, prop := range def.Properties {
			if prop.Type.Contains(boolType) {
				prop.AddExtension(nullableKey, true)
				cloudSpec.Definitions[k].Properties[kk] = prop
			}
			// Sets the deleted_on fields as a nullable since it should be.
			if kk == "deleted_on" && prop.Format == "date-time" {
				prop.AddExtension(nullableKey, true)
				cloudSpec.Definitions[k].Properties[kk] = prop
			}

			if k == "DeploymentTemplateInfo" && kk == "instance_configurations" {
				prop.AddExtension(omitEmptyKey, true)
				cloudSpec.Definitions[k].Properties[kk] = prop
			}

			if k == "DeploymentTemplateInfo" && kk == "order" {
				prop.AddExtension(nullableKey, true)
				cloudSpec.Definitions[k].Properties[kk] = prop
			}

			if k == "DeploymentTemplateInfoV2" && kk == "order" {
				prop.AddExtension(nullableKey, true)
				cloudSpec.Definitions[k].Properties[kk] = prop
			}
			if k == "DeploymentTemplateRequestBody" && kk == "order" {
				prop.AddExtension(nullableKey, true)
				cloudSpec.Definitions[k].Properties[kk] = prop
			}

			if k == "ApmTopologyElement" && kk == "instance_configuration_version" {
				prop.AddExtension(nullableKey, true)
				cloudSpec.Definitions[k].Properties[kk] = prop
			}

			if k == "AppSearchTopologyElement" && kk == "instance_configuration_version" {
				prop.AddExtension(nullableKey, true)
				cloudSpec.Definitions[k].Properties[kk] = prop
			}

			if k == "ElasticsearchClusterTopologyElement" && kk == "instance_configuration_version" {
				prop.AddExtension(nullableKey, true)
				cloudSpec.Definitions[k].Properties[kk] = prop
			}

			if k == "EnterpriseSearchTopologyElement" && kk == "instance_configuration_version" {
				prop.AddExtension(nullableKey, true)
				cloudSpec.Definitions[k].Properties[kk] = prop
			}

			if k == "IntegrationsServerTopologyElement" && kk == "instance_configuration_version" {
				prop.AddExtension(nullableKey, true)
				cloudSpec.Definitions[k].Properties[kk] = prop
			}

			if k == "KibanaClusterTopologyElement" && kk == "instance_configuration_version" {
				prop.AddExtension(nullableKey, true)
				cloudSpec.Definitions[k].Properties[kk] = prop
			}

			addExtension := func(model string, field string, key string, value interface{}) {
				if k == model && kk == field {
					prop.AddExtension(key, value)
					cloudSpec.Definitions[k].Properties[kk] = prop
				}
			}
			addExtension("DeploymentCreateRequest", "alias", nullableKey, true)
			addExtension("DeploymentUpdateRequest", "alias", nullableKey, true)

			if k == "ElasticsearchConfiguration" {
				if kk == "enabled_built_in_plugins" ||
					kk == "user_bundles" ||
					kk == "user_plugins" {
					prop.AddExtension(omitEmptyKey, true)
					cloudSpec.Definitions[k].Properties[kk] = prop
				}
			}

			if k == "InstanceConfiguration" {
				if kk == "node_types" {
					prop.AddExtension(omitEmptyKey, true)
					cloudSpec.Definitions[k].Properties[kk] = prop
				}
			}

			if k == "UserSecurity" {
				if kk == "permissions" ||
					kk == "roles" {
					prop.AddExtension(omitEmptyKey, true)
					cloudSpec.Definitions[k].Properties[kk] = prop
				}
			}

			if k == "BoolQuery" {
				if kk == "filter" ||
					kk == "must" ||
					kk == "must_not" ||
					kk == "should" {
					prop.AddExtension(omitEmptyKey, true)
					cloudSpec.Definitions[k].Properties[kk] = prop
				}
			}

			// This hack will make json marshaling to omit all empty fields
			if k == "ProxiesSettings" {
				prop.AddExtension(omitEmptyKey, true)
				prop.AddExtension(nullableKey, true)
				cloudSpec.Definitions[k].Properties[kk] = prop
			}
		}
	}

	// Iterate over the paths and parameters as well.
	for k, path := range cloudSpec.Paths.Paths {
		// This hack changes the request body of the PUT container endpoint operation from string
		// to structured model of models.Container
		if k == "/platform/infrastructure/proxies/settings" {
			pathItemProps := path.PathItemProps
			pathItemProps.Patch.Parameters[1].ParamProps.Schema.SchemaProps.Type = nil
			pathItemProps.Patch.Parameters[1].ParamProps.Schema.SchemaProps.Ref = spec.MustCreateRef("#/definitions/ProxiesSettings")
		}
		for kk, param := range path.Parameters {
			if param.Type == boolType {
				param.AddExtension(nullableKey, true)
				cloudSpec.Paths.Paths[k].Parameters[kk] = param
			}
		}
	}
}