mirror of
https://github.com/woodpecker-ci/woodpecker.git
synced 2024-06-01 07:08:30 +00:00
104 lines
2.5 KiB
Go
104 lines
2.5 KiB
Go
package yml
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"io"
|
|
"strconv"
|
|
|
|
"gopkg.in/yaml.v3"
|
|
)
|
|
|
|
// toJSON convert gopkg.in/yaml.v3 nodes to object that can be serialized as json
|
|
// fmt.Sprint() with default formatting is used to convert the key to a string key.
|
|
func toJSON(node *yaml.Node) (interface{}, error) {
|
|
switch node.Kind {
|
|
case yaml.DocumentNode:
|
|
return toJSON(node.Content[0])
|
|
|
|
case yaml.SequenceNode:
|
|
val := make([]interface{}, len(node.Content))
|
|
var err error
|
|
for i := range node.Content {
|
|
if val[i], err = toJSON(node.Content[i]); err != nil {
|
|
return nil, err
|
|
}
|
|
}
|
|
return val, nil
|
|
|
|
case yaml.MappingNode:
|
|
if (len(node.Content) % 2) != 0 {
|
|
return nil, fmt.Errorf("broken mapping node")
|
|
}
|
|
val := make(map[string]interface{}, len(node.Content)%2)
|
|
for i := len(node.Content); i > 1; i = i - 2 {
|
|
k, err := toJSON(node.Content[i-2])
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if val[fmt.Sprint(k)], err = toJSON(node.Content[i-1]); err != nil {
|
|
return nil, err
|
|
}
|
|
}
|
|
return val, nil
|
|
|
|
case yaml.ScalarNode:
|
|
switch node.Tag {
|
|
case nullTag:
|
|
return nil, nil
|
|
case boolTag:
|
|
return strconv.ParseBool(node.Value)
|
|
case intTag:
|
|
return strconv.ParseInt(node.Value, 10, 64)
|
|
case floatTag:
|
|
return strconv.ParseFloat(node.Value, 64)
|
|
}
|
|
return node.Value, nil
|
|
}
|
|
|
|
return nil, fmt.Errorf("do not support yaml node kind '%v'", node.Kind)
|
|
}
|
|
|
|
// ToJSON converts YAML bytes to JSON
|
|
func ToJSON(data []byte) ([]byte, error) {
|
|
m := &yaml.Node{}
|
|
if err := yaml.Unmarshal(data, m); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
d, err := toJSON(m)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return json.Marshal(d)
|
|
}
|
|
|
|
// LoadYmlReaderAsJSON reads from an io.Reader containing YAML and converts it to JSON
|
|
func LoadYmlReaderAsJSON(r io.Reader) (j []byte, err error) {
|
|
data, err := io.ReadAll(r)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
j, err = ToJSON(data)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return j, nil
|
|
}
|
|
|
|
// Source: https://github.com/go-yaml/yaml/blob/3e3283e801afc229479d5fc68aa41df1137b8394/resolve.go#L70-L81
|
|
const (
|
|
nullTag = "!!null"
|
|
boolTag = "!!bool"
|
|
intTag = "!!int"
|
|
floatTag = "!!float"
|
|
// strTag = "!!str" // we dont have to parse it
|
|
// timestampTag = "!!timestamp" // TODO: do we have to parse this?
|
|
// seqTag = "!!seq" // TODO: do we have to parse this?
|
|
// mapTag = "!!map" // TODO: do we have to parse this?
|
|
// binaryTag = "!!binary" // TODO: do we have to parse this?
|
|
// mergeTag = "!!merge" // TODO: do we have to parse this?
|
|
)
|