mirror of
https://github.com/goccy/go-yaml.git
synced 2025-10-31 21:31:06 +00:00
951 lines
25 KiB
Go
951 lines
25 KiB
Go
package yaml
|
|
|
|
import (
|
|
"encoding/base64"
|
|
"fmt"
|
|
"io"
|
|
"io/ioutil"
|
|
"math"
|
|
"os"
|
|
"path/filepath"
|
|
"reflect"
|
|
"strconv"
|
|
"time"
|
|
|
|
"github.com/goccy/go-yaml/ast"
|
|
"github.com/goccy/go-yaml/internal/errors"
|
|
"github.com/goccy/go-yaml/parser"
|
|
"github.com/goccy/go-yaml/token"
|
|
"golang.org/x/xerrors"
|
|
)
|
|
|
|
// Decoder reads and decodes YAML values from an input stream.
|
|
type Decoder struct {
|
|
reader io.Reader
|
|
referenceReaders []io.Reader
|
|
anchorMap map[string]ast.Node
|
|
opts []DecodeOption
|
|
referenceFiles []string
|
|
referenceDirs []string
|
|
isRecursiveDir bool
|
|
isResolvedReference bool
|
|
validator StructValidator
|
|
disallowUnknownField bool
|
|
}
|
|
|
|
// NewDecoder returns a new decoder that reads from r.
|
|
func NewDecoder(r io.Reader, opts ...DecodeOption) *Decoder {
|
|
return &Decoder{
|
|
reader: r,
|
|
anchorMap: map[string]ast.Node{},
|
|
opts: opts,
|
|
referenceReaders: []io.Reader{},
|
|
referenceFiles: []string{},
|
|
referenceDirs: []string{},
|
|
isRecursiveDir: false,
|
|
isResolvedReference: false,
|
|
disallowUnknownField: false,
|
|
}
|
|
}
|
|
|
|
func (d *Decoder) castToFloat(v interface{}) interface{} {
|
|
switch vv := v.(type) {
|
|
case int:
|
|
return float64(vv)
|
|
case int8:
|
|
return float64(vv)
|
|
case int16:
|
|
return float64(vv)
|
|
case int32:
|
|
return float64(vv)
|
|
case int64:
|
|
return float64(vv)
|
|
case uint:
|
|
return float64(vv)
|
|
case uint8:
|
|
return float64(vv)
|
|
case uint16:
|
|
return float64(vv)
|
|
case uint32:
|
|
return float64(vv)
|
|
case uint64:
|
|
return float64(vv)
|
|
case float32:
|
|
return float64(vv)
|
|
case float64:
|
|
return vv
|
|
case string:
|
|
// if error occurred, return zero value
|
|
f, _ := strconv.ParseFloat(vv, 64)
|
|
return f
|
|
}
|
|
return 0
|
|
}
|
|
|
|
func (d *Decoder) nodeToValue(node ast.Node) interface{} {
|
|
switch n := node.(type) {
|
|
case *ast.NullNode:
|
|
return nil
|
|
case *ast.StringNode:
|
|
return n.GetValue()
|
|
case *ast.IntegerNode:
|
|
return n.GetValue()
|
|
case *ast.FloatNode:
|
|
return n.GetValue()
|
|
case *ast.BoolNode:
|
|
return n.GetValue()
|
|
case *ast.InfinityNode:
|
|
return n.GetValue()
|
|
case *ast.NanNode:
|
|
return n.GetValue()
|
|
case *ast.TagNode:
|
|
switch n.Start.Value {
|
|
case token.TimestampTag:
|
|
t, _ := d.castToTime(n.Value)
|
|
return t
|
|
case token.FloatTag:
|
|
return d.castToFloat(d.nodeToValue(n.Value))
|
|
case token.NullTag:
|
|
return nil
|
|
case token.BinaryTag:
|
|
b, _ := base64.StdEncoding.DecodeString(d.nodeToValue(n.Value).(string))
|
|
return b
|
|
}
|
|
case *ast.AnchorNode:
|
|
anchorName := n.Name.GetToken().Value
|
|
anchorValue := d.nodeToValue(n.Value)
|
|
d.anchorMap[anchorName] = n.Value
|
|
return anchorValue
|
|
case *ast.AliasNode:
|
|
aliasName := n.Value.GetToken().Value
|
|
return d.nodeToValue(d.anchorMap[aliasName])
|
|
case *ast.LiteralNode:
|
|
return n.Value.GetValue()
|
|
case *ast.MappingValueNode:
|
|
m := map[string]interface{}{}
|
|
if n.Key.Type() == ast.MergeKeyType {
|
|
mapValue := d.nodeToValue(n.Value).(map[string]interface{})
|
|
for k, v := range mapValue {
|
|
m[k] = v
|
|
}
|
|
} else {
|
|
key := n.Key.GetToken().Value
|
|
m[key] = d.nodeToValue(n.Value)
|
|
}
|
|
return m
|
|
case *ast.MappingNode:
|
|
m := map[string]interface{}{}
|
|
for _, value := range n.Values {
|
|
subMap := d.nodeToValue(value).(map[string]interface{})
|
|
for k, v := range subMap {
|
|
m[k] = v
|
|
}
|
|
}
|
|
return m
|
|
case *ast.SequenceNode:
|
|
v := []interface{}{}
|
|
for _, value := range n.Values {
|
|
v = append(v, d.nodeToValue(value))
|
|
}
|
|
return v
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (d *Decoder) getMapNode(node ast.Node) (ast.MapNode, error) {
|
|
if _, ok := node.(*ast.NullNode); ok {
|
|
return nil, nil
|
|
}
|
|
if anchor, ok := node.(*ast.AnchorNode); ok {
|
|
mapNode, ok := anchor.Value.(ast.MapNode)
|
|
if ok {
|
|
return mapNode, nil
|
|
}
|
|
return nil, xerrors.Errorf("%s node doesn't MapNode", anchor.Value.Type())
|
|
}
|
|
if alias, ok := node.(*ast.AliasNode); ok {
|
|
aliasName := alias.Value.GetToken().Value
|
|
anchorNode := d.anchorMap[aliasName]
|
|
mapNode, ok := anchorNode.(ast.MapNode)
|
|
if ok {
|
|
return mapNode, nil
|
|
}
|
|
return nil, xerrors.Errorf("%s node doesn't MapNode", anchorNode.Type())
|
|
}
|
|
mapNode, ok := node.(ast.MapNode)
|
|
if !ok {
|
|
return nil, xerrors.Errorf("%s node doesn't MapNode", node.Type())
|
|
}
|
|
return mapNode, nil
|
|
}
|
|
|
|
func (d *Decoder) getArrayNode(node ast.Node) (ast.ArrayNode, error) {
|
|
if _, ok := node.(*ast.NullNode); ok {
|
|
return nil, nil
|
|
}
|
|
if anchor, ok := node.(*ast.AnchorNode); ok {
|
|
arrayNode, ok := anchor.Value.(ast.ArrayNode)
|
|
if ok {
|
|
return arrayNode, nil
|
|
}
|
|
return nil, xerrors.Errorf("%s node doesn't ArrayNode", anchor.Value.Type())
|
|
}
|
|
if alias, ok := node.(*ast.AliasNode); ok {
|
|
aliasName := alias.Value.GetToken().Value
|
|
anchorNode := d.anchorMap[aliasName]
|
|
arrayNode, ok := anchorNode.(ast.ArrayNode)
|
|
if ok {
|
|
return arrayNode, nil
|
|
}
|
|
return nil, xerrors.Errorf("%s node doesn't ArrayNode", anchorNode.Type())
|
|
}
|
|
arrayNode, ok := node.(ast.ArrayNode)
|
|
if !ok {
|
|
return nil, xerrors.Errorf("%s node doesn't ArrayNode", node.Type())
|
|
}
|
|
return arrayNode, nil
|
|
}
|
|
|
|
func (d *Decoder) fileToNode(f *ast.File) ast.Node {
|
|
for _, doc := range f.Docs {
|
|
if v := d.nodeToValue(doc.Body); v != nil {
|
|
return doc.Body
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (d *Decoder) convertValue(v reflect.Value, typ reflect.Type) (reflect.Value, error) {
|
|
if typ.Kind() != reflect.String {
|
|
if !v.Type().ConvertibleTo(typ) {
|
|
return reflect.Zero(typ), errTypeMismatch(typ, v.Type())
|
|
}
|
|
return v.Convert(typ), nil
|
|
}
|
|
// cast value to string
|
|
switch v.Type().Kind() {
|
|
case reflect.Int, reflect.Int8, reflect.Int16, reflect.Int32, reflect.Int64:
|
|
return reflect.ValueOf(fmt.Sprint(v.Int())), nil
|
|
case reflect.Float32, reflect.Float64:
|
|
return reflect.ValueOf(fmt.Sprint(v.Float())), nil
|
|
case reflect.Uint, reflect.Uint8, reflect.Uint16, reflect.Uint32, reflect.Uint64, reflect.Uintptr:
|
|
return reflect.ValueOf(fmt.Sprint(v.Uint())), nil
|
|
case reflect.Bool:
|
|
return reflect.ValueOf(fmt.Sprint(v.Bool())), nil
|
|
}
|
|
if !v.Type().ConvertibleTo(typ) {
|
|
return reflect.Zero(typ), errTypeMismatch(typ, v.Type())
|
|
}
|
|
return v.Convert(typ), nil
|
|
}
|
|
|
|
type overflowError struct {
|
|
dstType reflect.Type
|
|
srcNum string
|
|
}
|
|
|
|
func (e *overflowError) Error() string {
|
|
return fmt.Sprintf("cannot unmarshal %s into Go value of type %s ( overflow )", e.srcNum, e.dstType)
|
|
}
|
|
|
|
func errOverflow(dstType reflect.Type, num string) *overflowError {
|
|
return &overflowError{dstType: dstType, srcNum: num}
|
|
}
|
|
|
|
type typeError struct {
|
|
dstType reflect.Type
|
|
srcType reflect.Type
|
|
structFieldName *string
|
|
}
|
|
|
|
func (e *typeError) Error() string {
|
|
if e.structFieldName != nil {
|
|
return fmt.Sprintf("cannot unmarshal %s into Go struct field %s of type %s", e.srcType, *e.structFieldName, e.dstType)
|
|
}
|
|
return fmt.Sprintf("cannot unmarshal %s into Go value of type %s", e.srcType, e.dstType)
|
|
}
|
|
|
|
func errTypeMismatch(dstType, srcType reflect.Type) *typeError {
|
|
return &typeError{dstType: dstType, srcType: srcType}
|
|
}
|
|
|
|
type unknownFieldError struct {
|
|
*errors.SyntaxError
|
|
}
|
|
|
|
func errUnknowField(msg string, tk *token.Token) *unknownFieldError {
|
|
return &unknownFieldError{SyntaxError: errors.ErrSyntax(msg, tk)}
|
|
}
|
|
|
|
func (d *Decoder) deleteStructKeys(str reflect.Value, unknownFields map[string]ast.Node) error {
|
|
strType := str.Type()
|
|
structFieldMap, err := structFieldMap(strType)
|
|
if err != nil {
|
|
return errors.Wrapf(err, "failed to create struct field map")
|
|
}
|
|
|
|
for j := 0; j < strType.NumField(); j++ {
|
|
field := str.Type().Field(j)
|
|
if isIgnoredStructField(field) {
|
|
continue
|
|
}
|
|
|
|
structField, ok := structFieldMap[field.Name]
|
|
if ok {
|
|
delete(unknownFields, structField.RenderName)
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (d *Decoder) decodeValue(dst reflect.Value, src ast.Node) error {
|
|
valueType := dst.Type()
|
|
if unmarshaler, ok := dst.Addr().Interface().(BytesUnmarshaler); ok {
|
|
b := fmt.Sprintf("%v", src)
|
|
if err := unmarshaler.UnmarshalYAML([]byte(b)); err != nil {
|
|
return errors.Wrapf(err, "failed to UnmarshalYAML")
|
|
}
|
|
return nil
|
|
} else if unmarshaler, ok := dst.Addr().Interface().(InterfaceUnmarshaler); ok {
|
|
if err := unmarshaler.UnmarshalYAML(func(v interface{}) error {
|
|
rv := reflect.ValueOf(v)
|
|
if rv.Type().Kind() != reflect.Ptr {
|
|
return errors.ErrDecodeRequiredPointerType
|
|
}
|
|
if err := d.decodeValue(rv.Elem(), src); err != nil {
|
|
return errors.Wrapf(err, "failed to decode value")
|
|
}
|
|
return nil
|
|
}); err != nil {
|
|
return errors.Wrapf(err, "failed to UnmarshalYAML")
|
|
}
|
|
return nil
|
|
}
|
|
switch valueType.Kind() {
|
|
case reflect.Ptr:
|
|
if dst.IsNil() {
|
|
return nil
|
|
}
|
|
if src.Type() == ast.NullType {
|
|
// set nil value to pointer
|
|
dst.Set(reflect.Zero(valueType))
|
|
return nil
|
|
}
|
|
v := d.createDecodableValue(dst.Type())
|
|
if err := d.decodeValue(v, src); err != nil {
|
|
return errors.Wrapf(err, "failed to decode ptr value")
|
|
}
|
|
dst.Set(d.castToAssignableValue(v, dst.Type()))
|
|
case reflect.Interface:
|
|
v := reflect.ValueOf(d.nodeToValue(src))
|
|
if v.IsValid() {
|
|
dst.Set(v)
|
|
}
|
|
case reflect.Map:
|
|
return d.decodeMap(dst, src)
|
|
case reflect.Array:
|
|
return d.decodeArray(dst, src)
|
|
case reflect.Slice:
|
|
return d.decodeSlice(dst, src)
|
|
case reflect.Struct:
|
|
if _, ok := dst.Addr().Interface().(*time.Time); ok {
|
|
return d.decodeTime(dst, src)
|
|
}
|
|
return d.decodeStruct(dst, src)
|
|
case reflect.Int, reflect.Int8, reflect.Int16, reflect.Int32, reflect.Int64:
|
|
v := d.nodeToValue(src)
|
|
switch vv := v.(type) {
|
|
case int64:
|
|
if !dst.OverflowInt(vv) {
|
|
dst.SetInt(vv)
|
|
return nil
|
|
}
|
|
case uint64:
|
|
if vv <= math.MaxInt64 && !dst.OverflowInt(int64(vv)) {
|
|
dst.SetInt(int64(vv))
|
|
return nil
|
|
}
|
|
case float64:
|
|
if vv <= math.MaxInt64 && !dst.OverflowInt(int64(vv)) {
|
|
dst.SetInt(int64(vv))
|
|
return nil
|
|
}
|
|
default:
|
|
return errTypeMismatch(valueType, reflect.TypeOf(v))
|
|
}
|
|
return errOverflow(valueType, fmt.Sprint(v))
|
|
case reflect.Uint, reflect.Uint8, reflect.Uint16, reflect.Uint32, reflect.Uint64:
|
|
v := d.nodeToValue(src)
|
|
switch vv := v.(type) {
|
|
case int64:
|
|
if 0 <= vv && !dst.OverflowUint(uint64(vv)) {
|
|
dst.SetUint(uint64(vv))
|
|
return nil
|
|
}
|
|
case uint64:
|
|
if !dst.OverflowUint(vv) {
|
|
dst.SetUint(vv)
|
|
return nil
|
|
}
|
|
case float64:
|
|
if 0 <= vv && vv <= math.MaxUint64 && !dst.OverflowUint(uint64(vv)) {
|
|
dst.SetUint(uint64(vv))
|
|
return nil
|
|
}
|
|
default:
|
|
return errTypeMismatch(valueType, reflect.TypeOf(v))
|
|
}
|
|
return errOverflow(valueType, fmt.Sprint(v))
|
|
}
|
|
v := reflect.ValueOf(d.nodeToValue(src))
|
|
if v.IsValid() {
|
|
convertedValue, err := d.convertValue(v, dst.Type())
|
|
if err != nil {
|
|
return errors.Wrapf(err, "failed to convert value")
|
|
}
|
|
dst.Set(convertedValue)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (d *Decoder) createDecodableValue(typ reflect.Type) reflect.Value {
|
|
for {
|
|
if typ.Kind() == reflect.Ptr {
|
|
typ = typ.Elem()
|
|
continue
|
|
}
|
|
break
|
|
}
|
|
return reflect.New(typ).Elem()
|
|
}
|
|
|
|
func (d *Decoder) castToAssignableValue(value reflect.Value, target reflect.Type) reflect.Value {
|
|
if target.Kind() != reflect.Ptr {
|
|
return value
|
|
}
|
|
maxTryCount := 5
|
|
tryCount := 0
|
|
for {
|
|
if tryCount > maxTryCount {
|
|
return value
|
|
}
|
|
if value.Type().AssignableTo(target) {
|
|
break
|
|
}
|
|
value = value.Addr()
|
|
tryCount++
|
|
}
|
|
return value
|
|
}
|
|
|
|
func (d *Decoder) keyToNodeMap(node ast.Node, getKeyOrValueNode func(*ast.MapNodeIter) ast.Node) (map[string]ast.Node, error) {
|
|
mapNode, err := d.getMapNode(node)
|
|
if err != nil {
|
|
return nil, errors.Wrapf(err, "failed to get map node")
|
|
}
|
|
keyToNodeMap := map[string]ast.Node{}
|
|
if mapNode == nil {
|
|
return keyToNodeMap, nil
|
|
}
|
|
mapIter := mapNode.MapRange()
|
|
for mapIter.Next() {
|
|
keyNode := mapIter.Key()
|
|
if keyNode.Type() == ast.MergeKeyType {
|
|
mergeMap, err := d.keyToNodeMap(mapIter.Value(), getKeyOrValueNode)
|
|
if err != nil {
|
|
return nil, errors.Wrapf(err, "failed to get keyToNodeMap by MergeKey node")
|
|
}
|
|
for k, v := range mergeMap {
|
|
keyToNodeMap[k] = v
|
|
}
|
|
} else {
|
|
key, ok := d.nodeToValue(keyNode).(string)
|
|
if !ok {
|
|
return nil, errors.Wrapf(err, "failed to decode map key")
|
|
}
|
|
keyToNodeMap[key] = getKeyOrValueNode(mapIter)
|
|
}
|
|
}
|
|
return keyToNodeMap, nil
|
|
}
|
|
|
|
func (d *Decoder) keyToKeyNodeMap(node ast.Node) (map[string]ast.Node, error) {
|
|
m, err := d.keyToNodeMap(node, func(nodeMap *ast.MapNodeIter) ast.Node { return nodeMap.Key() })
|
|
if err != nil {
|
|
return nil, errors.Wrapf(err, "failed to get keyToNodeMap")
|
|
}
|
|
return m, nil
|
|
}
|
|
|
|
func (d *Decoder) keyToValueNodeMap(node ast.Node) (map[string]ast.Node, error) {
|
|
m, err := d.keyToNodeMap(node, func(nodeMap *ast.MapNodeIter) ast.Node { return nodeMap.Value() })
|
|
if err != nil {
|
|
return nil, errors.Wrapf(err, "failed to get keyToNodeMap")
|
|
}
|
|
return m, nil
|
|
}
|
|
|
|
func (d *Decoder) setDefaultValueIfConflicted(v reflect.Value, fieldMap StructFieldMap) error {
|
|
typ := v.Type()
|
|
if typ.Kind() != reflect.Struct {
|
|
return nil
|
|
}
|
|
embeddedStructFieldMap, err := structFieldMap(typ)
|
|
if err != nil {
|
|
return errors.Wrapf(err, "failed to get struct field map by embedded type")
|
|
}
|
|
for i := 0; i < typ.NumField(); i++ {
|
|
field := typ.Field(i)
|
|
if isIgnoredStructField(field) {
|
|
continue
|
|
}
|
|
structField := embeddedStructFieldMap[field.Name]
|
|
if !fieldMap.isIncludedRenderName(structField.RenderName) {
|
|
continue
|
|
}
|
|
// if declared same key name, set default value
|
|
fieldValue := v.Field(i)
|
|
if fieldValue.CanSet() {
|
|
fieldValue.Set(reflect.Zero(fieldValue.Type()))
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// This is a subset of the formats allowed by the regular expression
|
|
// defined at http://yaml.org/type/timestamp.html.
|
|
var allowedTimestampFormats = []string{
|
|
"2006-1-2T15:4:5.999999999Z07:00", // RCF3339Nano with short date fields.
|
|
"2006-1-2t15:4:5.999999999Z07:00", // RFC3339Nano with short date fields and lower-case "t".
|
|
"2006-1-2 15:4:5.999999999", // space separated with no time zone
|
|
"2006-1-2", // date only
|
|
}
|
|
|
|
func (d *Decoder) castToTime(src ast.Node) (time.Time, error) {
|
|
if src == nil {
|
|
return time.Time{}, nil
|
|
}
|
|
v := d.nodeToValue(src)
|
|
if t, ok := v.(time.Time); ok {
|
|
return t, nil
|
|
}
|
|
s, ok := v.(string)
|
|
if !ok {
|
|
return time.Time{}, errTypeMismatch(reflect.TypeOf(time.Time{}), reflect.TypeOf(v))
|
|
}
|
|
for _, format := range allowedTimestampFormats {
|
|
t, err := time.Parse(format, s)
|
|
if err != nil {
|
|
// invalid format
|
|
continue
|
|
}
|
|
return t, nil
|
|
}
|
|
return time.Time{}, nil
|
|
}
|
|
|
|
func (d *Decoder) decodeTime(dst reflect.Value, src ast.Node) error {
|
|
t, err := d.castToTime(src)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
dst.Set(reflect.ValueOf(t))
|
|
return nil
|
|
}
|
|
|
|
func (d *Decoder) decodeStruct(dst reflect.Value, src ast.Node) error {
|
|
if src == nil {
|
|
return nil
|
|
}
|
|
structType := dst.Type()
|
|
structValue := reflect.New(structType)
|
|
structFieldMap, err := structFieldMap(structType)
|
|
if err != nil {
|
|
return errors.Wrapf(err, "failed to create struct field map")
|
|
}
|
|
keyToNodeMap, err := d.keyToValueNodeMap(src)
|
|
if err != nil {
|
|
return errors.Wrapf(err, "failed to get keyToValueNodeMap")
|
|
}
|
|
var unknownFields map[string]ast.Node
|
|
if d.disallowUnknownField {
|
|
unknownFields, err = d.keyToKeyNodeMap(src)
|
|
if err != nil {
|
|
return errors.Wrapf(err, "failed to get keyToKeyNodeMap")
|
|
}
|
|
}
|
|
|
|
var foundErr error
|
|
|
|
for i := 0; i < structType.NumField(); i++ {
|
|
field := structType.Field(i)
|
|
if isIgnoredStructField(field) {
|
|
continue
|
|
}
|
|
structField := structFieldMap[field.Name]
|
|
if structField.IsInline {
|
|
fieldValue := structValue.Elem().FieldByName(field.Name)
|
|
if !fieldValue.CanSet() {
|
|
return xerrors.Errorf("cannot set embedded type as unexported field %s.%s", field.PkgPath, field.Name)
|
|
}
|
|
if fieldValue.Type().Kind() == reflect.Ptr && src.Type() == ast.NullType {
|
|
// set nil value to pointer
|
|
fieldValue.Set(reflect.Zero(fieldValue.Type()))
|
|
continue
|
|
}
|
|
newFieldValue := d.createDecodableValue(fieldValue.Type())
|
|
if err := d.decodeValue(newFieldValue, src); err != nil {
|
|
if foundErr != nil {
|
|
continue
|
|
}
|
|
var te *typeError
|
|
if xerrors.As(err, &te) {
|
|
if te.structFieldName != nil {
|
|
fieldName := fmt.Sprintf("%s.%s", structType.Name(), *te.structFieldName)
|
|
te.structFieldName = &fieldName
|
|
} else {
|
|
fieldName := fmt.Sprintf("%s.%s", structType.Name(), field.Name)
|
|
te.structFieldName = &fieldName
|
|
}
|
|
foundErr = te
|
|
continue
|
|
}
|
|
|
|
var ufe *unknownFieldError
|
|
if !xerrors.As(err, &ufe) {
|
|
foundErr = err
|
|
continue
|
|
}
|
|
|
|
err = d.deleteStructKeys(fieldValue, unknownFields)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
d.setDefaultValueIfConflicted(newFieldValue, structFieldMap)
|
|
fieldValue.Set(d.castToAssignableValue(newFieldValue, fieldValue.Type()))
|
|
continue
|
|
}
|
|
v, exists := keyToNodeMap[structField.RenderName]
|
|
if !exists {
|
|
continue
|
|
}
|
|
delete(unknownFields, structField.RenderName)
|
|
fieldValue := structValue.Elem().FieldByName(field.Name)
|
|
if fieldValue.Type().Kind() == reflect.Ptr && src.Type() == ast.NullType {
|
|
// set nil value to pointer
|
|
fieldValue.Set(reflect.Zero(fieldValue.Type()))
|
|
continue
|
|
}
|
|
newFieldValue := d.createDecodableValue(fieldValue.Type())
|
|
if err := d.decodeValue(newFieldValue, v); err != nil {
|
|
if foundErr != nil {
|
|
continue
|
|
}
|
|
var te *typeError
|
|
if xerrors.As(err, &te) {
|
|
fieldName := fmt.Sprintf("%s.%s", structType.Name(), field.Name)
|
|
te.structFieldName = &fieldName
|
|
foundErr = te
|
|
} else {
|
|
foundErr = err
|
|
}
|
|
continue
|
|
}
|
|
fieldValue.Set(d.castToAssignableValue(newFieldValue, fieldValue.Type()))
|
|
}
|
|
if d.validator != nil {
|
|
if err := d.validator.Struct(structValue.Interface()); err != nil {
|
|
ev := reflect.ValueOf(err)
|
|
if ev.Type().Kind() == reflect.Slice {
|
|
for i := 0; i < ev.Len(); i++ {
|
|
fieldErr, ok := ev.Index(i).Interface().(FieldError)
|
|
if !ok {
|
|
continue
|
|
}
|
|
fieldName := fieldErr.StructField()
|
|
structField := structFieldMap[fieldName]
|
|
node, exists := keyToNodeMap[structField.RenderName]
|
|
if exists {
|
|
// TODO: to make FieldError message cutomizable
|
|
return errors.ErrSyntax(fmt.Sprintf("%s", err), node.GetToken())
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
dst.Set(structValue.Elem())
|
|
if len(unknownFields) != 0 && d.disallowUnknownField {
|
|
for key, node := range unknownFields {
|
|
return errUnknowField(fmt.Sprintf(`unknown field "%s"`, key), node.GetToken())
|
|
}
|
|
}
|
|
if foundErr != nil {
|
|
return errors.Wrapf(foundErr, "failed to decode value")
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (d *Decoder) decodeArray(dst reflect.Value, src ast.Node) error {
|
|
arrayNode, err := d.getArrayNode(src)
|
|
if err != nil {
|
|
return errors.Wrapf(err, "failed to get array node")
|
|
}
|
|
if arrayNode == nil {
|
|
return nil
|
|
}
|
|
iter := arrayNode.ArrayRange()
|
|
arrayValue := reflect.New(dst.Type()).Elem()
|
|
arrayType := dst.Type()
|
|
elemType := arrayType.Elem()
|
|
idx := 0
|
|
|
|
var foundErr error
|
|
for iter.Next() {
|
|
v := iter.Value()
|
|
if elemType.Kind() == reflect.Ptr && v.Type() == ast.NullType {
|
|
// set nil value to pointer
|
|
arrayValue.Index(idx).Set(reflect.Zero(elemType))
|
|
} else {
|
|
dstValue := d.createDecodableValue(elemType)
|
|
if err := d.decodeValue(dstValue, v); err != nil {
|
|
if foundErr == nil {
|
|
foundErr = err
|
|
}
|
|
continue
|
|
} else {
|
|
arrayValue.Index(idx).Set(d.castToAssignableValue(dstValue, elemType))
|
|
}
|
|
}
|
|
idx++
|
|
}
|
|
dst.Set(arrayValue)
|
|
if foundErr != nil {
|
|
return errors.Wrapf(foundErr, "failed to decode value")
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (d *Decoder) decodeSlice(dst reflect.Value, src ast.Node) error {
|
|
arrayNode, err := d.getArrayNode(src)
|
|
if err != nil {
|
|
return errors.Wrapf(err, "failed to get array node")
|
|
}
|
|
if arrayNode == nil {
|
|
return nil
|
|
}
|
|
iter := arrayNode.ArrayRange()
|
|
sliceType := dst.Type()
|
|
sliceValue := reflect.MakeSlice(sliceType, 0, iter.Len())
|
|
elemType := sliceType.Elem()
|
|
|
|
var foundErr error
|
|
for iter.Next() {
|
|
v := iter.Value()
|
|
if elemType.Kind() == reflect.Ptr && v.Type() == ast.NullType {
|
|
// set nil value to pointer
|
|
sliceValue = reflect.Append(sliceValue, reflect.Zero(elemType))
|
|
continue
|
|
}
|
|
dstValue := d.createDecodableValue(elemType)
|
|
if err := d.decodeValue(dstValue, v); err != nil {
|
|
if foundErr == nil {
|
|
foundErr = err
|
|
}
|
|
continue
|
|
}
|
|
sliceValue = reflect.Append(sliceValue, d.castToAssignableValue(dstValue, elemType))
|
|
}
|
|
dst.Set(sliceValue)
|
|
if foundErr != nil {
|
|
return errors.Wrapf(foundErr, "failed to decode value")
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (d *Decoder) decodeMap(dst reflect.Value, src ast.Node) error {
|
|
mapNode, err := d.getMapNode(src)
|
|
if err != nil {
|
|
return errors.Wrapf(err, "failed to get map node")
|
|
}
|
|
if mapNode == nil {
|
|
return nil
|
|
}
|
|
mapType := dst.Type()
|
|
mapValue := reflect.MakeMap(mapType)
|
|
keyType := mapValue.Type().Key()
|
|
valueType := mapValue.Type().Elem()
|
|
mapIter := mapNode.MapRange()
|
|
|
|
var foundErr error
|
|
for mapIter.Next() {
|
|
key := mapIter.Key()
|
|
value := mapIter.Value()
|
|
k := reflect.ValueOf(d.nodeToValue(key))
|
|
if k.IsValid() && k.Type().ConvertibleTo(keyType) {
|
|
k = k.Convert(keyType)
|
|
}
|
|
if valueType.Kind() == reflect.Ptr && value.Type() == ast.NullType {
|
|
// set nil value to pointer
|
|
mapValue.SetMapIndex(k, reflect.Zero(valueType))
|
|
continue
|
|
}
|
|
dstValue := d.createDecodableValue(valueType)
|
|
if err := d.decodeValue(dstValue, value); err != nil {
|
|
if foundErr == nil {
|
|
foundErr = err
|
|
}
|
|
}
|
|
if !k.IsValid() {
|
|
// expect nil key
|
|
mapValue.SetMapIndex(d.createDecodableValue(keyType), d.castToAssignableValue(dstValue, valueType))
|
|
continue
|
|
}
|
|
mapValue.SetMapIndex(k, d.castToAssignableValue(dstValue, valueType))
|
|
}
|
|
dst.Set(mapValue)
|
|
if foundErr != nil {
|
|
return errors.Wrapf(foundErr, "failed to decode value")
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (d *Decoder) fileToReader(file string) (io.Reader, error) {
|
|
reader, err := os.Open(file)
|
|
if err != nil {
|
|
return nil, errors.Wrapf(err, "failed to open file")
|
|
}
|
|
return reader, nil
|
|
}
|
|
|
|
func (d *Decoder) isYAMLFile(file string) bool {
|
|
ext := filepath.Ext(file)
|
|
if ext == ".yml" {
|
|
return true
|
|
}
|
|
if ext == ".yaml" {
|
|
return true
|
|
}
|
|
return false
|
|
}
|
|
|
|
func (d *Decoder) readersUnderDir(dir string) ([]io.Reader, error) {
|
|
pattern := fmt.Sprintf("%s/*", dir)
|
|
matches, err := filepath.Glob(pattern)
|
|
if err != nil {
|
|
return nil, errors.Wrapf(err, "failed to get files by %s", pattern)
|
|
}
|
|
readers := []io.Reader{}
|
|
for _, match := range matches {
|
|
if !d.isYAMLFile(match) {
|
|
continue
|
|
}
|
|
reader, err := d.fileToReader(match)
|
|
if err != nil {
|
|
return nil, errors.Wrapf(err, "failed to get reader")
|
|
}
|
|
readers = append(readers, reader)
|
|
}
|
|
return readers, nil
|
|
}
|
|
|
|
func (d *Decoder) readersUnderDirRecursive(dir string) ([]io.Reader, error) {
|
|
readers := []io.Reader{}
|
|
if err := filepath.Walk(dir, func(path string, info os.FileInfo, err error) error {
|
|
if !d.isYAMLFile(path) {
|
|
return nil
|
|
}
|
|
reader, err := d.fileToReader(path)
|
|
if err != nil {
|
|
return errors.Wrapf(err, "failed to get reader")
|
|
}
|
|
readers = append(readers, reader)
|
|
return nil
|
|
}); err != nil {
|
|
return nil, errors.Wrapf(err, "interrupt walk in %s", dir)
|
|
}
|
|
return readers, nil
|
|
}
|
|
|
|
func (d *Decoder) resolveReference() error {
|
|
for _, opt := range d.opts {
|
|
if err := opt(d); err != nil {
|
|
return errors.Wrapf(err, "failed to exec option")
|
|
}
|
|
}
|
|
for _, file := range d.referenceFiles {
|
|
reader, err := d.fileToReader(file)
|
|
if err != nil {
|
|
return errors.Wrapf(err, "failed to get reader")
|
|
}
|
|
d.referenceReaders = append(d.referenceReaders, reader)
|
|
}
|
|
for _, dir := range d.referenceDirs {
|
|
if !d.isRecursiveDir {
|
|
readers, err := d.readersUnderDir(dir)
|
|
if err != nil {
|
|
return errors.Wrapf(err, "failed to get readers from under the %s", dir)
|
|
}
|
|
d.referenceReaders = append(d.referenceReaders, readers...)
|
|
} else {
|
|
readers, err := d.readersUnderDirRecursive(dir)
|
|
if err != nil {
|
|
return errors.Wrapf(err, "failed to get readers from under the %s", dir)
|
|
}
|
|
d.referenceReaders = append(d.referenceReaders, readers...)
|
|
}
|
|
}
|
|
for _, reader := range d.referenceReaders {
|
|
bytes, err := ioutil.ReadAll(reader)
|
|
if err != nil {
|
|
return errors.Wrapf(err, "failed to read buffer")
|
|
}
|
|
|
|
// assign new anchor definition to anchorMap
|
|
if _, err := d.decode(bytes); err != nil {
|
|
return errors.Wrapf(err, "failed to decode")
|
|
}
|
|
}
|
|
d.isResolvedReference = true
|
|
return nil
|
|
}
|
|
|
|
func (d *Decoder) decode(bytes []byte) (ast.Node, error) {
|
|
f, err := parser.ParseBytes(bytes, 0)
|
|
if err != nil {
|
|
return nil, errors.Wrapf(err, "failed to parse yaml")
|
|
}
|
|
return d.fileToNode(f), nil
|
|
}
|
|
|
|
// Decode reads the next YAML-encoded value from its input
|
|
// and stores it in the value pointed to by v.
|
|
//
|
|
// See the documentation for Unmarshal for details about the
|
|
// conversion of YAML into a Go value.
|
|
func (d *Decoder) Decode(v interface{}) error {
|
|
if !d.isResolvedReference {
|
|
if err := d.resolveReference(); err != nil {
|
|
return errors.Wrapf(err, "failed to resolve reference")
|
|
}
|
|
}
|
|
rv := reflect.ValueOf(v)
|
|
if rv.Type().Kind() != reflect.Ptr {
|
|
return errors.ErrDecodeRequiredPointerType
|
|
}
|
|
bytes, err := ioutil.ReadAll(d.reader)
|
|
if err != nil {
|
|
return errors.Wrapf(err, "failed to read buffer")
|
|
}
|
|
node, err := d.decode(bytes)
|
|
if err != nil {
|
|
return errors.Wrapf(err, "failed to decode")
|
|
}
|
|
if node == nil {
|
|
return nil
|
|
}
|
|
if err := d.decodeValue(rv.Elem(), node); err != nil {
|
|
return errors.Wrapf(err, "failed to decode value")
|
|
}
|
|
return nil
|
|
}
|