169 lines
6.2 KiB
Go
169 lines
6.2 KiB
Go
/*
|
|
Copyright 2016 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 common
|
|
|
|
import (
|
|
"net/http"
|
|
"strings"
|
|
|
|
"github.com/emicklei/go-restful"
|
|
"github.com/go-openapi/spec"
|
|
)
|
|
|
|
// OpenAPIDefinition describes single type. Normally these definitions are auto-generated using gen-openapi.
|
|
type OpenAPIDefinition struct {
|
|
Schema spec.Schema
|
|
Dependencies []string
|
|
}
|
|
|
|
type ReferenceCallback func(path string) spec.Ref
|
|
|
|
// OpenAPIDefinitions is collection of all definitions.
|
|
type GetOpenAPIDefinitions func(ReferenceCallback) map[string]OpenAPIDefinition
|
|
|
|
// OpenAPIDefinitionGetter gets openAPI definitions for a given type. If a type implements this interface,
|
|
// the definition returned by it will be used, otherwise the auto-generated definitions will be used. See
|
|
// GetOpenAPITypeFormat for more information about trade-offs of using this interface or GetOpenAPITypeFormat method when
|
|
// possible.
|
|
type OpenAPIDefinitionGetter interface {
|
|
OpenAPIDefinition() *OpenAPIDefinition
|
|
}
|
|
|
|
type PathHandler interface {
|
|
Handle(path string, handler http.Handler)
|
|
}
|
|
|
|
// Config is set of configuration for openAPI spec generation.
|
|
type Config struct {
|
|
// List of supported protocols such as https, http, etc.
|
|
ProtocolList []string
|
|
|
|
// Info is general information about the API.
|
|
Info *spec.Info
|
|
|
|
// DefaultResponse will be used if an operation does not have any responses listed. It
|
|
// will show up as ... "responses" : {"default" : $DefaultResponse} in the spec.
|
|
DefaultResponse *spec.Response
|
|
|
|
// CommonResponses will be added as a response to all operation specs. This is a good place to add common
|
|
// responses such as authorization failed.
|
|
CommonResponses map[int]spec.Response
|
|
|
|
// List of webservice's path prefixes to ignore
|
|
IgnorePrefixes []string
|
|
|
|
// OpenAPIDefinitions should provide definition for all models used by routes. Failure to provide this map
|
|
// or any of the models will result in spec generation failure.
|
|
GetDefinitions GetOpenAPIDefinitions
|
|
|
|
// GetOperationIDAndTags returns operation id and tags for a restful route. It is an optional function to customize operation IDs.
|
|
GetOperationIDAndTags func(r *restful.Route) (string, []string, error)
|
|
|
|
// GetDefinitionName returns a friendly name for a definition base on the serving path. parameter `name` is the full name of the definition.
|
|
// It is an optional function to customize model names.
|
|
GetDefinitionName func(name string) (string, spec.Extensions)
|
|
|
|
// PostProcessSpec runs after the spec is ready to serve. It allows a final modification to the spec before serving.
|
|
PostProcessSpec func(*spec.Swagger) (*spec.Swagger, error)
|
|
|
|
// SecurityDefinitions is list of all security definitions for OpenAPI service. If this is not nil, the user of config
|
|
// is responsible to provide DefaultSecurity and (maybe) add unauthorized response to CommonResponses.
|
|
SecurityDefinitions *spec.SecurityDefinitions
|
|
|
|
// DefaultSecurity for all operations. This will pass as spec.SwaggerProps.Security to OpenAPI.
|
|
// For most cases, this will be list of acceptable definitions in SecurityDefinitions.
|
|
DefaultSecurity []map[string][]string
|
|
}
|
|
|
|
var schemaTypeFormatMap = map[string][]string{
|
|
"uint": {"integer", "int32"},
|
|
"uint8": {"integer", "byte"},
|
|
"uint16": {"integer", "int32"},
|
|
"uint32": {"integer", "int64"},
|
|
"uint64": {"integer", "int64"},
|
|
"int": {"integer", "int32"},
|
|
"int8": {"integer", "byte"},
|
|
"int16": {"integer", "int32"},
|
|
"int32": {"integer", "int32"},
|
|
"int64": {"integer", "int64"},
|
|
"byte": {"integer", "byte"},
|
|
"float64": {"number", "double"},
|
|
"float32": {"number", "float"},
|
|
"bool": {"boolean", ""},
|
|
"time.Time": {"string", "date-time"},
|
|
"string": {"string", ""},
|
|
"integer": {"integer", ""},
|
|
"number": {"number", ""},
|
|
"boolean": {"boolean", ""},
|
|
"[]byte": {"string", "byte"}, // base64 encoded characters
|
|
"interface{}": {"object", ""},
|
|
}
|
|
|
|
// This function is a reference for converting go (or any custom type) to a simple open API type,format pair. There are
|
|
// two ways to customize spec for a type. If you add it here, a type will be converted to a simple type and the type
|
|
// comment (the comment that is added before type definition) will be lost. The spec will still have the property
|
|
// comment. The second way is to implement OpenAPIDefinitionGetter interface. That function can customize the spec (so
|
|
// the spec does not need to be simple type,format) or can even return a simple type,format (e.g. IntOrString). For simple
|
|
// type formats, the benefit of adding OpenAPIDefinitionGetter interface is to keep both type and property documentation.
|
|
// Example:
|
|
// type Sample struct {
|
|
// ...
|
|
// // port of the server
|
|
// port IntOrString
|
|
// ...
|
|
// }
|
|
// // IntOrString documentation...
|
|
// type IntOrString { ... }
|
|
//
|
|
// Adding IntOrString to this function:
|
|
// "port" : {
|
|
// format: "string",
|
|
// type: "int-or-string",
|
|
// Description: "port of the server"
|
|
// }
|
|
//
|
|
// Implement OpenAPIDefinitionGetter for IntOrString:
|
|
//
|
|
// "port" : {
|
|
// $Ref: "#/definitions/IntOrString"
|
|
// Description: "port of the server"
|
|
// }
|
|
// ...
|
|
// definitions:
|
|
// {
|
|
// "IntOrString": {
|
|
// format: "string",
|
|
// type: "int-or-string",
|
|
// Description: "IntOrString documentation..." // new
|
|
// }
|
|
// }
|
|
//
|
|
func GetOpenAPITypeFormat(typeName string) (string, string) {
|
|
mapped, ok := schemaTypeFormatMap[typeName]
|
|
if !ok {
|
|
return "", ""
|
|
}
|
|
return mapped[0], mapped[1]
|
|
}
|
|
|
|
func EscapeJsonPointer(p string) string {
|
|
// Escaping reference name using rfc6901
|
|
p = strings.Replace(p, "~", "~0", -1)
|
|
p = strings.Replace(p, "/", "~1", -1)
|
|
return p
|
|
}
|