encoding/openapi/orderedmap.go - master - servlamo/cue

encoding/openapi/orderedmap.go
70 строк2 KB
Daniel Martí
encoding/openapi: unexport OrderedMap
30 сен 2024, 16:36
30 сен 2024, 16:369f2c2fe
// Copyright 2019 CUE 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 openapi

import (
	"cuelang.org/go/cue/ast"
)

// An orderedMap is a set of key-value pairs that preserves the order in which
// items were added.
//
// Deprecated: the API now returns an ast.File. This allows OpenAPI to be
// represented as JSON, YAML, or CUE data, in addition to being able to use
// all the ast-related tooling.
type orderedMap ast.StructLit

func (m *orderedMap) len() int {
	return len(m.Elts)
}

func (m *orderedMap) find(key string) *ast.Field {
	for _, v := range m.Elts {
		f, ok := v.(*ast.Field)
		if !ok {
			continue
		}
		s, _, err := ast.LabelName(f.Label)
		if err == nil && s == key {
			return f
		}
	}
	return nil
}

func (m *orderedMap) setExpr(key string, expr ast.Expr) {
	if f := m.find(key); f != nil {
		f.Value = expr
		return
	}
	m.Elts = append(m.Elts, &ast.Field{
		Label: ast.NewString(key),
		Value: expr,
	})
}

// exists reports whether a key-value pair exists for the given key.
func (m *orderedMap) exists(key string) bool {
	return m.find(key) != nil
}

// exists reports whether a key-value pair exists for the given key.
func (m *orderedMap) getMap(key string) *orderedMap {
	f := m.find(key)
	if f == nil {
		return nil
	}
	return (*orderedMap)(f.Value.(*ast.StructLit))
}