Курнат Андрей
55 lines
1.1 KiB
Go
55 lines
1.1 KiB
Go
// Copyright 2021 The OPA Authors. All rights reserved.
|
|
// Use of this source code is governed by an Apache2
|
|
// license that can be found in the LICENSE file.
|
|
|
|
package ast
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"github.com/open-policy-agent/opa/v1/types"
|
|
"github.com/open-policy-agent/opa/v1/util"
|
|
)
|
|
|
|
// SchemaSet holds a map from a path to a schema.
|
|
type SchemaSet struct {
|
|
m *util.HasherMap[Ref, any]
|
|
}
|
|
|
|
// NewSchemaSet returns an empty SchemaSet.
|
|
func NewSchemaSet() *SchemaSet {
|
|
return &SchemaSet{
|
|
m: util.NewHasherMap[Ref, any](RefEqual),
|
|
}
|
|
}
|
|
|
|
// Put inserts a raw schema into the set.
|
|
func (ss *SchemaSet) Put(path Ref, raw any) {
|
|
ss.m.Put(path, raw)
|
|
}
|
|
|
|
// Get returns the raw schema identified by the path.
|
|
func (ss *SchemaSet) Get(path Ref) any {
|
|
if ss != nil {
|
|
if x, ok := ss.m.Get(path); ok {
|
|
return x
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func loadSchema(raw any, allowNet []string) (types.Type, error) {
|
|
|
|
jsonSchema, err := compileSchema(raw, allowNet)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
tpe, err := newSchemaParser().parseSchema(jsonSchema.RootSchema)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("type checking: %w", err)
|
|
}
|
|
|
|
return tpe, nil
|
|
}
|