Files
libopenapi/datamodel/low/v2/responses.go
2024-08-26 20:42:40 +01:00

104 lines
3.0 KiB
Go

// Copyright 2022 Princess B33f Heavy Industries / Dave Shanley
// SPDX-License-Identifier: MIT
package v2
import (
"context"
"crypto/sha256"
"fmt"
"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 *orderedmap.Map[low.KeyReference[string], low.ValueReference[*yaml.Node]]
}
// GetExtensions returns all Responses extensions and satisfies the low.HasExtensions interface.
func (r *Responses) GetExtensions() *orderedmap.Map[low.KeyReference[string], low.ValueReference[*yaml.Node]] {
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 code, resp := range r.Codes.FromOldest() {
if strings.ToLower(code.Value) == DefaultLabel {
return &low.NodeReference[*Response]{
ValueNode: resp.ValueNode,
KeyNode: code.KeyNode,
Value: resp.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
f = low.AppendMapHashes(f, orderedmap.SortAlpha(r.Codes))
if !r.Default.IsEmpty() {
f = append(f, low.GenerateHashString(r.Default.Value))
}
f = append(f, low.HashExtensions(r.Extensions)...)
return sha256.Sum256([]byte(strings.Join(f, "|")))
}