First cut - can recursively read yaml!

This commit is contained in:
Mike Farah 2015-09-27 13:11:10 +10:00
parent da31cbe672
commit b1c670e121

23
yaml.go
View File

@ -7,6 +7,7 @@ import (
"io/ioutil" "io/ioutil"
"os" "os"
"github.com/codegangsta/cli" "github.com/codegangsta/cli"
"strings"
) )
func main() { func main() {
@ -17,7 +18,7 @@ func main() {
{ {
Name: "read", Name: "read",
Aliases: []string{"r"}, Aliases: []string{"r"},
Usage: "(default) reads a property from a given yaml file", Usage: "read <filename> <path>\n\te.g.: yaml read sample.json a.b.c\n\t(default) reads a property from a given yaml file",
Action: read_file, Action: read_file,
}, },
} }
@ -31,17 +32,31 @@ func read_file(c *cli.Context) {
} }
var filename = c.Args()[0] var filename = c.Args()[0]
var raw_data, read_error = ioutil.ReadFile(filename) var raw_data, read_error = ioutil.ReadFile(filename)
if read_error != nil { if read_error != nil {
log.Fatalf("error: %v", read_error) log.Fatalf("error: %v", read_error)
} }
var parsed_data interface{} var parsed_data map[interface{}]interface{}
err := yaml.Unmarshal([]byte(raw_data), &parsed_data) err := yaml.Unmarshal([]byte(raw_data), &parsed_data)
if err != nil { if err != nil {
log.Fatalf("error: %v", err) log.Fatalf("error: %v", err)
} }
fmt.Println(parsed_data) var path = c.Args()[1]
var paths = strings.Split(path, ".")
fmt.Println(read(parsed_data, paths[0], paths[1:len(paths)]))
}
func read(context map[interface{}]interface{}, head string, tail []string) interface{} {
value := context[head]
// fmt.Println("read called")
switch value.(type) {
case bool, int, string, []interface{}:
return value
default: // recurse into map
return read(value.(map[interface{}]interface{}), tail[0], tail[1:len(tail)])
}
} }