vendor/k8s.io/kube-openapi/pkg/util/proto/document.go - cloudstack-kubernetes-provider - Git at Google

 /*
 Copyright 2017 The Kubernetes Authors.

 Licensed under the Apache License, Version 2.0 (the "License");
 you may not use this file except in compliance with the License.
 You may obtain a copy of the License at

     http://www.apache.org/licenses/LICENSE-2.0

 Unless required by applicable law or agreed to in writing, software
 distributed under the License is distributed on an "AS IS" BASIS,
 WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 See the License for the specific language governing permissions and
 limitations under the License.
 */

 package proto

 import (
 	"fmt"
 	"sort"
 	"strings"

 	"github.com/googleapis/gnostic/OpenAPIv2"
 	"gopkg.in/yaml.v2"
 )

 func newSchemaError(path *Path, format string, a ...interface{}) error {
 	err := fmt.Sprintf(format, a...)
 	if path.Len() == 0 {
 		return fmt.Errorf("SchemaError: %v", err)
 	}
 	return fmt.Errorf("SchemaError(%v): %v", path, err)
 }

 // VendorExtensionToMap converts openapi VendorExtension to a map.
 func VendorExtensionToMap(e []*openapi_v2.NamedAny) map[string]interface{} {
 	values := map[string]interface{}{}

 	for _, na := range e {
 		if na.GetName() == "" || na.GetValue() == nil {
 			continue
 		}
 		if na.GetValue().GetYaml() == "" {
 			continue
 		}
 		var value interface{}
 		err := yaml.Unmarshal([]byte(na.GetValue().GetYaml()), &value)
 		if err != nil {
 			continue
 		}

 		values[na.GetName()] = value
 	}

 	return values
 }

 // Definitions is an implementation of `Models`. It looks for
 // models in an openapi Schema.
 type Definitions struct {
 	models map[string]Schema
 }

 var _ Models = &Definitions{}

 // NewOpenAPIData creates a new `Models` out of the openapi document.
 func NewOpenAPIData(doc *openapi_v2.Document) (Models, error) {
 	definitions := Definitions{
 		models: map[string]Schema{},
 	}

 	// Save the list of all models first. This will allow us to
 	// validate that we don't have any dangling reference.
 	for _, namedSchema := range doc.GetDefinitions().GetAdditionalProperties() {
 		definitions.models[namedSchema.GetName()] = nil
 	}

 	// Now, parse each model. We can validate that references exists.
 	for _, namedSchema := range doc.GetDefinitions().GetAdditionalProperties() {
 		path := NewPath(namedSchema.GetName())
 		schema, err := definitions.ParseSchema(namedSchema.GetValue(), &path)
 		if err != nil {
 			return nil, err
 		}
 		definitions.models[namedSchema.GetName()] = schema
 	}

 	return &definitions, nil
 }

 // We believe the schema is a reference, verify that and returns a new
 // Schema
 func (d *Definitions) parseReference(s *openapi_v2.Schema, path *Path) (Schema, error) {
 	if len(s.GetProperties().GetAdditionalProperties()) > 0 {
 		return nil, newSchemaError(path, "unallowed embedded type definition")
 	}
 	if len(s.GetType().GetValue()) > 0 {
 		return nil, newSchemaError(path, "definition reference can't have a type")
 	}

 	if !strings.HasPrefix(s.GetXRef(), "#/definitions/") {
 		return nil, newSchemaError(path, "unallowed reference to non-definition %q", s.GetXRef())
 	}
 	reference := strings.TrimPrefix(s.GetXRef(), "#/definitions/")
 	if _, ok := d.models[reference]; !ok {
 		return nil, newSchemaError(path, "unknown model in reference: %q", reference)
 	}
 	return &Ref{
 		BaseSchema:  d.parseBaseSchema(s, path),
 		reference:   reference,
 		definitions: d,
 	}, nil
 }

 func (d *Definitions) parseBaseSchema(s *openapi_v2.Schema, path *Path) BaseSchema {
 	return BaseSchema{
 		Description: s.GetDescription(),
 		Extensions:  VendorExtensionToMap(s.GetVendorExtension()),
 		Path:        *path,
 	}
 }

 // We believe the schema is a map, verify and return a new schema
 func (d *Definitions) parseMap(s *openapi_v2.Schema, path *Path) (Schema, error) {
 	if len(s.GetType().GetValue()) != 0 && s.GetType().GetValue()[0] != object {
 		return nil, newSchemaError(path, "invalid object type")
 	}
 	var sub Schema
 	if s.GetAdditionalProperties().GetSchema() == nil {
 		sub = &Arbitrary{
 			BaseSchema: d.parseBaseSchema(s, path),
 		}
 	} else {
 		var err error
 		sub, err = d.ParseSchema(s.GetAdditionalProperties().GetSchema(), path)
 		if err != nil {
 			return nil, err
 		}
 	}
 	return &Map{
 		BaseSchema: d.parseBaseSchema(s, path),
 		SubType:    sub,
 	}, nil
 }

 func (d *Definitions) parsePrimitive(s *openapi_v2.Schema, path *Path) (Schema, error) {
 	var t string
 	if len(s.GetType().GetValue()) > 1 {
 		return nil, newSchemaError(path, "primitive can't have more than 1 type")
 	}
 	if len(s.GetType().GetValue()) == 1 {
 		t = s.GetType().GetValue()[0]
 	}
 	switch t {
 	case String: // do nothing
 	case Number: // do nothing
 	case Integer: // do nothing
 	case Boolean: // do nothing
 	default:
 		return nil, newSchemaError(path, "Unknown primitive type: %q", t)
 	}
 	return &Primitive{
 		BaseSchema: d.parseBaseSchema(s, path),
 		Type:       t,
 		Format:     s.GetFormat(),
 	}, nil
 }

 func (d *Definitions) parseArray(s *openapi_v2.Schema, path *Path) (Schema, error) {
 	if len(s.GetType().GetValue()) != 1 {
 		return nil, newSchemaError(path, "array should have exactly one type")
 	}
 	if s.GetType().GetValue()[0] != array {
 		return nil, newSchemaError(path, `array should have type "array"`)
 	}
 	if len(s.GetItems().GetSchema()) != 1 {
 		return nil, newSchemaError(path, "array should have exactly one sub-item")
 	}
 	sub, err := d.ParseSchema(s.GetItems().GetSchema()[0], path)
 	if err != nil {
 		return nil, err
 	}
 	return &Array{
 		BaseSchema: d.parseBaseSchema(s, path),
 		SubType:    sub,
 	}, nil
 }

 func (d *Definitions) parseKind(s *openapi_v2.Schema, path *Path) (Schema, error) {
 	if len(s.GetType().GetValue()) != 0 && s.GetType().GetValue()[0] != object {
 		return nil, newSchemaError(path, "invalid object type")
 	}
 	if s.GetProperties() == nil {
 		return nil, newSchemaError(path, "object doesn't have properties")
 	}

 	fields := map[string]Schema{}
 	fieldOrder := []string{}

 	for _, namedSchema := range s.GetProperties().GetAdditionalProperties() {
 		var err error
 		name := namedSchema.GetName()
 		path := path.FieldPath(name)
 		fields[name], err = d.ParseSchema(namedSchema.GetValue(), &path)
 		if err != nil {
 			return nil, err
 		}
 		fieldOrder = append(fieldOrder, name)
 	}

 	return &Kind{
 		BaseSchema:     d.parseBaseSchema(s, path),
 		RequiredFields: s.GetRequired(),
 		Fields:         fields,
 		FieldOrder:     fieldOrder,
 	}, nil
 }

 func (d *Definitions) parseArbitrary(s *openapi_v2.Schema, path *Path) (Schema, error) {
 	return &Arbitrary{
 		BaseSchema: d.parseBaseSchema(s, path),
 	}, nil
 }

 // ParseSchema creates a walkable Schema from an openapi schema. While
 // this function is public, it doesn't leak through the interface.
 func (d *Definitions) ParseSchema(s *openapi_v2.Schema, path *Path) (Schema, error) {
 	if s.GetXRef() != "" {
 		return d.parseReference(s, path)
 	}
 	objectTypes := s.GetType().GetValue()
 	switch len(objectTypes) {
 	case 0:
 		// in the OpenAPI schema served by older k8s versions, object definitions created from structs did not include
 		// the type:object property (they only included the "properties" property), so we need to handle this case
 		if s.GetProperties() != nil {
 			return d.parseKind(s, path)
 		} else {
 			// Definition has no type and no properties. Treat it as an arbitrary value
 			// TODO: what if it has additionalProperties or patternProperties?
 			return d.parseArbitrary(s, path)
 		}
 	case 1:
 		t := objectTypes[0]
 		switch t {
 		case object:
 			if s.GetProperties() != nil {
 				return d.parseKind(s, path)
 			} else {
 				return d.parseMap(s, path)
 			}
 		case array:
 			return d.parseArray(s, path)
 		}
 		return d.parsePrimitive(s, path)
 	default:
 		// the OpenAPI generator never generates (nor it ever did in the past) OpenAPI type definitions with multiple types
 		return nil, newSchemaError(path, "definitions with multiple types aren't supported")
 	}
 }

 // LookupModel is public through the interface of Models. It
 // returns a visitable schema from the given model name.
 func (d *Definitions) LookupModel(model string) Schema {
 	return d.models[model]
 }

 func (d *Definitions) ListModels() []string {
 	models := []string{}

 	for model := range d.models {
 		models = append(models, model)
 	}

 	sort.Strings(models)
 	return models
 }

 type Ref struct {
 	BaseSchema

 	reference   string
 	definitions *Definitions
 }

 var _ Reference = &Ref{}

 func (r *Ref) Reference() string {
 	return r.reference
 }

 func (r *Ref) SubSchema() Schema {
 	return r.definitions.models[r.reference]
 }

 func (r *Ref) Accept(v SchemaVisitor) {
 	v.VisitReference(r)
 }

 func (r *Ref) GetName() string {
 	return fmt.Sprintf("Reference to %q", r.reference)
 }
	/*
	Copyright 2017 The Kubernetes Authors.

	Licensed under the Apache License, Version 2.0 (the "License");
	you may not use this file except in compliance with the License.
	You may obtain a copy of the License at

	http://www.apache.org/licenses/LICENSE-2.0

	Unless required by applicable law or agreed to in writing, software
	distributed under the License is distributed on an "AS IS" BASIS,
	WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
	See the License for the specific language governing permissions and
	limitations under the License.
	*/

	package proto

	import (
	"fmt"
	"sort"
	"strings"

	"github.com/googleapis/gnostic/OpenAPIv2"
	"gopkg.in/yaml.v2"
	)

	func newSchemaError(path *Path, format string, a ...interface{}) error {
	err := fmt.Sprintf(format, a...)
	if path.Len() == 0 {
	return fmt.Errorf("SchemaError: %v", err)
	}
	return fmt.Errorf("SchemaError(%v): %v", path, err)
	}

	// VendorExtensionToMap converts openapi VendorExtension to a map.
	func VendorExtensionToMap(e []*openapi_v2.NamedAny) map[string]interface{} {
	values := map[string]interface{}{}

	for _, na := range e {
	if na.GetName() == "" \|\| na.GetValue() == nil {
	continue
	}
	if na.GetValue().GetYaml() == "" {
	continue
	}
	var value interface{}
	err := yaml.Unmarshal([]byte(na.GetValue().GetYaml()), &value)
	if err != nil {
	continue
	}

	values[na.GetName()] = value
	}

	return values
	}

	// Definitions is an implementation of `Models`. It looks for
	// models in an openapi Schema.
	type Definitions struct {
	models map[string]Schema
	}

	var _ Models = &Definitions{}

	// NewOpenAPIData creates a new `Models` out of the openapi document.
	func NewOpenAPIData(doc *openapi_v2.Document) (Models, error) {
	definitions := Definitions{
	models: map[string]Schema{},
	}

	// Save the list of all models first. This will allow us to
	// validate that we don't have any dangling reference.
	for _, namedSchema := range doc.GetDefinitions().GetAdditionalProperties() {
	definitions.models[namedSchema.GetName()] = nil
	}

	// Now, parse each model. We can validate that references exists.
	for _, namedSchema := range doc.GetDefinitions().GetAdditionalProperties() {
	path := NewPath(namedSchema.GetName())
	schema, err := definitions.ParseSchema(namedSchema.GetValue(), &path)
	if err != nil {
	return nil, err
	}
	definitions.models[namedSchema.GetName()] = schema
	}

	return &definitions, nil
	}

	// We believe the schema is a reference, verify that and returns a new
	// Schema
	func (d Definitions) parseReference(s openapi_v2.Schema, path *Path) (Schema, error) {
	if len(s.GetProperties().GetAdditionalProperties()) > 0 {
	return nil, newSchemaError(path, "unallowed embedded type definition")
	}
	if len(s.GetType().GetValue()) > 0 {
	return nil, newSchemaError(path, "definition reference can't have a type")
	}

	if !strings.HasPrefix(s.GetXRef(), "#/definitions/") {
	return nil, newSchemaError(path, "unallowed reference to non-definition %q", s.GetXRef())
	}
	reference := strings.TrimPrefix(s.GetXRef(), "#/definitions/")
	if _, ok := d.models[reference]; !ok {
	return nil, newSchemaError(path, "unknown model in reference: %q", reference)
	}
	return &Ref{
	BaseSchema: d.parseBaseSchema(s, path),
	reference: reference,
	definitions: d,
	}, nil
	}

	func (d Definitions) parseBaseSchema(s openapi_v2.Schema, path *Path) BaseSchema {
	return BaseSchema{
	Description: s.GetDescription(),
	Extensions: VendorExtensionToMap(s.GetVendorExtension()),
	Path: *path,
	}
	}

	// We believe the schema is a map, verify and return a new schema
	func (d Definitions) parseMap(s openapi_v2.Schema, path *Path) (Schema, error) {
	if len(s.GetType().GetValue()) != 0 && s.GetType().GetValue()[0] != object {
	return nil, newSchemaError(path, "invalid object type")
	}
	var sub Schema
	if s.GetAdditionalProperties().GetSchema() == nil {
	sub = &Arbitrary{
	BaseSchema: d.parseBaseSchema(s, path),
	}
	} else {
	var err error
	sub, err = d.ParseSchema(s.GetAdditionalProperties().GetSchema(), path)
	if err != nil {
	return nil, err
	}
	}
	return &Map{
	BaseSchema: d.parseBaseSchema(s, path),
	SubType: sub,
	}, nil
	}

	func (d Definitions) parsePrimitive(s openapi_v2.Schema, path *Path) (Schema, error) {
	var t string
	if len(s.GetType().GetValue()) > 1 {
	return nil, newSchemaError(path, "primitive can't have more than 1 type")
	}
	if len(s.GetType().GetValue()) == 1 {
	t = s.GetType().GetValue()[0]
	}
	switch t {
	case String: // do nothing
	case Number: // do nothing
	case Integer: // do nothing
	case Boolean: // do nothing
	default:
	return nil, newSchemaError(path, "Unknown primitive type: %q", t)
	}
	return &Primitive{
	BaseSchema: d.parseBaseSchema(s, path),
	Type: t,
	Format: s.GetFormat(),
	}, nil
	}

	func (d Definitions) parseArray(s openapi_v2.Schema, path *Path) (Schema, error) {
	if len(s.GetType().GetValue()) != 1 {
	return nil, newSchemaError(path, "array should have exactly one type")
	}
	if s.GetType().GetValue()[0] != array {
	return nil, newSchemaError(path, `array should have type "array"`)
	}
	if len(s.GetItems().GetSchema()) != 1 {
	return nil, newSchemaError(path, "array should have exactly one sub-item")
	}
	sub, err := d.ParseSchema(s.GetItems().GetSchema()[0], path)
	if err != nil {
	return nil, err
	}
	return &Array{
	BaseSchema: d.parseBaseSchema(s, path),
	SubType: sub,
	}, nil
	}

	func (d Definitions) parseKind(s openapi_v2.Schema, path *Path) (Schema, error) {
	if len(s.GetType().GetValue()) != 0 && s.GetType().GetValue()[0] != object {
	return nil, newSchemaError(path, "invalid object type")
	}
	if s.GetProperties() == nil {
	return nil, newSchemaError(path, "object doesn't have properties")
	}

	fields := map[string]Schema{}
	fieldOrder := []string{}

	for _, namedSchema := range s.GetProperties().GetAdditionalProperties() {
	var err error
	name := namedSchema.GetName()
	path := path.FieldPath(name)
	fields[name], err = d.ParseSchema(namedSchema.GetValue(), &path)
	if err != nil {
	return nil, err
	}
	fieldOrder = append(fieldOrder, name)
	}

	return &Kind{
	BaseSchema: d.parseBaseSchema(s, path),
	RequiredFields: s.GetRequired(),
	Fields: fields,
	FieldOrder: fieldOrder,
	}, nil
	}

	func (d Definitions) parseArbitrary(s openapi_v2.Schema, path *Path) (Schema, error) {
	return &Arbitrary{
	BaseSchema: d.parseBaseSchema(s, path),
	}, nil
	}

	// ParseSchema creates a walkable Schema from an openapi schema. While
	// this function is public, it doesn't leak through the interface.
	func (d Definitions) ParseSchema(s openapi_v2.Schema, path *Path) (Schema, error) {
	if s.GetXRef() != "" {
	return d.parseReference(s, path)
	}
	objectTypes := s.GetType().GetValue()
	switch len(objectTypes) {
	case 0:
	// in the OpenAPI schema served by older k8s versions, object definitions created from structs did not include
	// the type:object property (they only included the "properties" property), so we need to handle this case
	if s.GetProperties() != nil {
	return d.parseKind(s, path)
	} else {
	// Definition has no type and no properties. Treat it as an arbitrary value
	// TODO: what if it has additionalProperties or patternProperties?
	return d.parseArbitrary(s, path)
	}
	case 1:
	t := objectTypes[0]
	switch t {
	case object:
	if s.GetProperties() != nil {
	return d.parseKind(s, path)
	} else {
	return d.parseMap(s, path)
	}
	case array:
	return d.parseArray(s, path)
	}
	return d.parsePrimitive(s, path)
	default:
	// the OpenAPI generator never generates (nor it ever did in the past) OpenAPI type definitions with multiple types
	return nil, newSchemaError(path, "definitions with multiple types aren't supported")
	}
	}

	// LookupModel is public through the interface of Models. It
	// returns a visitable schema from the given model name.
	func (d *Definitions) LookupModel(model string) Schema {
	return d.models[model]
	}

	func (d *Definitions) ListModels() []string {
	models := []string{}

	for model := range d.models {
	models = append(models, model)
	}

	sort.Strings(models)
	return models
	}

	type Ref struct {
	BaseSchema

	reference string
	definitions *Definitions
	}

	var _ Reference = &Ref{}

	func (r *Ref) Reference() string {
	return r.reference
	}

	func (r *Ref) SubSchema() Schema {
	return r.definitions.models[r.reference]
	}

	func (r *Ref) Accept(v SchemaVisitor) {
	v.VisitReference(r)
	}

	func (r *Ref) GetName() string {
	return fmt.Sprintf("Reference to %q", r.reference)
	}