example.go

package oas

import "github.com/getkin/kin-openapi/openapi3"

// ExampleExtractor returns an example payload according to the openapi3.SchemaRef object.
func ExampleExtractor(schema *openapi3.SchemaRef) interface{} {
	val := schema.Value
	if val.Example != nil {
		return val.Example
	}

	switch {
	case val.Type == openapi3.TypeObject:
		obj := make(map[string]interface{})
		for name, prop := range schema.Value.Properties {
			obj[name] = ExampleExtractor(prop)
		}

		return obj
	case val.Type == openapi3.TypeArray:
		items := ExampleExtractor(val.Items)
		return []interface{}{items}
	default:
		if len(val.Enum) > 0 {
			return val.Enum[0]
		}
		return emptyExampleVal(val)
	}
}

func emptyExampleVal(schema *openapi3.Schema) interface{} {
	switch schema.Type {
	case openapi3.TypeString:
		return "string"
	case openapi3.TypeInteger, openapi3.TypeNumber:
		return 0
	case openapi3.TypeBoolean:
		return true
	default:
		return nil
	}
}