1
0
mirror of https://github.com/danog/gojekyll.git synced 2024-11-27 05:44:38 +01:00
gojekyll/helpers/yaml.go
2017-06-21 09:43:26 -04:00

27 lines
554 B
Go

package helpers
import yaml "gopkg.in/yaml.v2"
// UnmarshalYAMLInterface is a wrapper for yaml.Unmarshall that
// knows how to unmarshal maps and lists.
func UnmarshalYAMLInterface(b []byte, i *interface{}) error {
var m map[interface{}]interface{}
err := yaml.Unmarshal(b, &m)
switch err.(type) {
case *yaml.TypeError:
// Work around https://github.com/go-yaml/yaml/issues/20
var s []interface{}
err = yaml.Unmarshal(b, &s)
if err != nil {
return err
}
*i = s
default:
if err != nil {
return err
}
*i = m
}
return nil
}