mirror of
https://github.com/LukeHagar/libopenapi.git
synced 2025-12-09 12:37:49 +00:00
124 lines
3.5 KiB
Go
124 lines
3.5 KiB
Go
// Copyright 2022 Princess B33f Heavy Industries / Dave Shanley
|
|
// SPDX-License-Identifier: MIT
|
|
|
|
package v2
|
|
|
|
import (
|
|
"context"
|
|
"crypto/sha256"
|
|
"fmt"
|
|
"sort"
|
|
"strings"
|
|
|
|
"github.com/pb33f/libopenapi/datamodel/low"
|
|
"github.com/pb33f/libopenapi/index"
|
|
"github.com/pb33f/libopenapi/orderedmap"
|
|
"github.com/pb33f/libopenapi/utils"
|
|
"gopkg.in/yaml.v3"
|
|
)
|
|
|
|
// Responses is a low-level representation of a Swagger / OpenAPI 2 Responses object.
|
|
type Responses struct {
|
|
Codes orderedmap.Map[low.KeyReference[string], low.ValueReference[*Response]]
|
|
Default low.NodeReference[*Response]
|
|
Extensions map[low.KeyReference[string]]low.ValueReference[any]
|
|
}
|
|
|
|
// GetExtensions returns all Responses extensions and satisfies the low.HasExtensions interface.
|
|
func (r *Responses) GetExtensions() map[low.KeyReference[string]]low.ValueReference[any] {
|
|
return r.Extensions
|
|
}
|
|
|
|
// Build will extract default value and extensions from node.
|
|
func (r *Responses) Build(ctx context.Context, _, root *yaml.Node, idx *index.SpecIndex) error {
|
|
root = utils.NodeAlias(root)
|
|
utils.CheckForMergeNodes(root)
|
|
r.Extensions = low.ExtractExtensions(root)
|
|
|
|
if utils.IsNodeMap(root) {
|
|
codes, err := low.ExtractMapNoLookup[*Response](ctx, root, idx)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if codes != nil {
|
|
r.Codes = codes
|
|
}
|
|
def := r.getDefault()
|
|
if def != nil {
|
|
// default is bundled into codes, pull it out
|
|
r.Default = *def
|
|
// remove default from codes
|
|
r.deleteCode(DefaultLabel)
|
|
}
|
|
} else {
|
|
return fmt.Errorf("responses build failed: vn node is not a map! line %d, col %d",
|
|
root.Line, root.Column)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (r *Responses) getDefault() *low.NodeReference[*Response] {
|
|
for pair := orderedmap.First(r.Codes); pair != nil; pair = pair.Next() {
|
|
if strings.ToLower(pair.Key().Value) == DefaultLabel {
|
|
return &low.NodeReference[*Response]{
|
|
ValueNode: pair.Value().ValueNode,
|
|
KeyNode: pair.Key().KeyNode,
|
|
Value: pair.Value().Value,
|
|
}
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// used to remove default from codes extracted by Build()
|
|
func (r *Responses) deleteCode(code string) {
|
|
var key *low.KeyReference[string]
|
|
if r.Codes != nil {
|
|
for pair := orderedmap.First(r.Codes); pair != nil; pair = pair.Next() {
|
|
if pair.Key().Value == code {
|
|
key = pair.KeyPtr()
|
|
break
|
|
}
|
|
}
|
|
}
|
|
// should never be nil, but, you never know... science and all that!
|
|
if key != nil {
|
|
r.Codes.Delete(*key)
|
|
}
|
|
}
|
|
|
|
// FindResponseByCode will attempt to locate a Response instance using an HTTP response code string.
|
|
func (r *Responses) FindResponseByCode(code string) *low.ValueReference[*Response] {
|
|
return low.FindItemInOrderedMap[*Response](code, r.Codes)
|
|
}
|
|
|
|
// Hash will return a consistent SHA256 Hash of the Examples object
|
|
func (r *Responses) Hash() [32]byte {
|
|
var f []string
|
|
var keys []string
|
|
keys = make([]string, orderedmap.Len(r.Codes))
|
|
cmap := make(map[string]*Response, len(keys))
|
|
z := 0
|
|
for pair := orderedmap.First(r.Codes); pair != nil; pair = pair.Next() {
|
|
keys[z] = pair.Key().Value
|
|
cmap[pair.Key().Value] = pair.Value().Value
|
|
z++
|
|
}
|
|
sort.Strings(keys)
|
|
for k := range keys {
|
|
f = append(f, fmt.Sprintf("%s-%s", keys[k], low.GenerateHashString(cmap[keys[k]])))
|
|
}
|
|
if !r.Default.IsEmpty() {
|
|
f = append(f, low.GenerateHashString(r.Default.Value))
|
|
}
|
|
keys = make([]string, len(r.Extensions))
|
|
z = 0
|
|
for k := range r.Extensions {
|
|
keys[z] = fmt.Sprintf("%s-%x", k.Value, sha256.Sum256([]byte(fmt.Sprint(r.Extensions[k].Value))))
|
|
z++
|
|
}
|
|
sort.Strings(keys)
|
|
f = append(f, keys...)
|
|
return sha256.Sum256([]byte(strings.Join(f, "|")))
|
|
}
|