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
1 changed files with 19 additions and 4 deletions

23
yaml.go
View File

@ -7,6 +7,7 @@ import (
"io/ioutil"
"os"
"github.com/codegangsta/cli"
"strings"
)
func main() {
@ -17,7 +18,7 @@ func main() {
{
Name: "read",
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,
},
}
@ -31,17 +32,31 @@ func read_file(c *cli.Context) {
}
var filename = c.Args()[0]
var raw_data, read_error = ioutil.ReadFile(filename)
if read_error != nil {
log.Fatalf("error: %v", read_error)
}
var parsed_data interface{}
var parsed_data map[interface{}]interface{}
err := yaml.Unmarshal([]byte(raw_data), &parsed_data)
if err != nil {
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)])
}
}