suyiiyii/yq
1
0
Fork 0
mirror of https://github.com/mikefarah/yq.git synced 2024-12-19 20:19:04 +00:00
Code Issues Packages Projects Releases Wiki Activity

gofmt'ed

Browse Source
This commit is contained in:
mfarah 2015-09-28 12:00:38 +10:00
parent ccd95146fe
commit 9869a41122
2 changed files with 85 additions and 86 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
yaml.go
View File

@ -2,13 +2,13 @@ package main
import ( import (
"fmt" "fmt"
"gopkg.in/yaml.v2"
"log"
"io/ioutil"
"os"
"github.com/codegangsta/cli" "github.com/codegangsta/cli"
"strings" "gopkg.in/yaml.v2"
"io/ioutil"
"log"
"os"
"strconv" "strconv"
"strings"
) )
func main() { func main() {
@ -64,7 +64,7 @@ func read_file(filename string) []byte {
func read_map(context map[interface{}]interface{}, head string, tail []string) interface{} { func read_map(context map[interface{}]interface{}, head string, tail []string) interface{} {
value := context[head] value := context[head]
if (len(tail) > 0) { if len(tail) > 0 {
return recurse(value, tail[0], tail[1:len(tail)]) return recurse(value, tail[0], tail[1:len(tail)])
} else { } else {
return value return value
@ -73,23 +73,22 @@ func read_map(context map[interface{}]interface{}, head string, tail []string) i
func recurse(value interface{}, head string, tail []string) interface{} { func recurse(value interface{}, head string, tail []string) interface{} {
switch value.(type) { switch value.(type) {
case []interface {}: case []interface{}:
index, err := strconv.ParseInt(head, 10, 64) index, err := strconv.ParseInt(head, 10, 64)
if err != nil { if err != nil {
log.Fatalf("Error accessing array: %v", err) log.Fatalf("Error accessing array: %v", err)
} }
return read_array(value.([]interface {}), index, tail) return read_array(value.([]interface{}), index, tail)
default: default:
return read_map(value.(map[interface{}]interface{}), head, tail) return read_map(value.(map[interface{}]interface{}), head, tail)
} }
} }
func read_array(array []interface {}, head int64, tail[]string) interface{} { func read_array(array []interface{}, head int64, tail []string) interface{} {
value := array[head] value := array[head]
if (len(tail) > 0) { if len(tail) > 0 {
return recurse(value, tail[0], tail[1:len(tail)]) return recurse(value, tail[0], tail[1:len(tail)])
} else { } else {
return value return value
} }
} }

10
yaml_test.go
View File

@ -1,13 +1,13 @@
package main package main
import ( import (
"testing"
"gopkg.in/yaml.v2"
"fmt" "fmt"
"gopkg.in/yaml.v2"
"os" "os"
"testing"
) )
var raw_data = ` var raw_data = `
a: Easy! a: Easy!
b: b:
c: 2 c: 2
@ -28,14 +28,14 @@ func TestMain(m *testing.M) {
func TestRead_map_simple(t *testing.T) { func TestRead_map_simple(t *testing.T) {
result := read_map(parsed_data, "b", []string{"c"}) result := read_map(parsed_data, "b", []string{"c"})
if( result != 2) { if result != 2 {
t.Error("Excpted 2 but got ", result) t.Error("Excpted 2 but got ", result)
} }
} }
func TestRead_map_array(t *testing.T) { func TestRead_map_array(t *testing.T) {
result := read_map(parsed_data, "b", []string{"d", "1"}) result := read_map(parsed_data, "b", []string{"d", "1"})
if( result != 4) { if result != 4 {
t.Error("Excpted 4 but got ", result) t.Error("Excpted 4 but got ", result)
} }
} }