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

Split code

Browse Source
This commit is contained in:
Mike Farah 2015-10-02 09:05:13 +10:00
parent 58bdb3ed21
commit 01845ea923
3 changed files with 43 additions and 37 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

43
data_navigator.go Normal file
View File

@ -0,0 +1,43 @@
package main
import (
"log"
"strconv"
)
func write(context map[interface{}]interface{}, head string, tail []string, value interface{}) {
// e.g. if updating a.b.c, we need to get the 'b' map...
toUpdate := readMap(context, head, tail[0:len(tail)-1]).(map[interface{}]interface{})
// and then set the 'c' key.
key := (tail[len(tail)-1])
toUpdate[key] = value
}
func readMap(context map[interface{}]interface{}, head string, tail []string) interface{} {
value := context[head]
if len(tail) > 0 {
return recurse(value, tail[0], tail[1:len(tail)])
}
return value
}
func recurse(value interface{}, head string, tail []string) interface{} {
switch value.(type) {
case []interface{}:
index, err := strconv.ParseInt(head, 10, 64)
if err != nil {
log.Fatalf("Error accessing array: %v", err)
}
return readArray(value.([]interface{}), index, tail)
default:
return readMap(value.(map[interface{}]interface{}), head, tail)
}
}
func readArray(array []interface{}, head int64, tail []string) interface{} {
value := array[head]
if len(tail) > 0 {
return recurse(value, tail[0], tail[1:len(tail)])
}
return value
}

0
yaml_test.go → data_navigator_test.go
View File

37
yaml.go
View File

@ -126,40 +126,3 @@ func readFile(filename string) []byte {
}
return rawData
}
func write(context map[interface{}]interface{}, head string, tail []string, value interface{}) {
// e.g. if updating a.b.c, we need to get the 'b' map...
toUpdate := readMap(context, head, tail[0:len(tail)-1]).(map[interface{}]interface{})
// and then set the 'c' key.
key := (tail[len(tail)-1])
toUpdate[key] = value
}
func readMap(context map[interface{}]interface{}, head string, tail []string) interface{} {
value := context[head]
if len(tail) > 0 {
return recurse(value, tail[0], tail[1:len(tail)])
}
return value
}
func recurse(value interface{}, head string, tail []string) interface{} {
switch value.(type) {
case []interface{}:
index, err := strconv.ParseInt(head, 10, 64)
if err != nil {
log.Fatalf("Error accessing array: %v", err)
}
return readArray(value.([]interface{}), index, tail)
default:
return readMap(value.(map[interface{}]interface{}), head, tail)
}
}
func readArray(array []interface{}, head int64, tail []string) interface{} {
value := array[head]
if len(tail) > 0 {
return recurse(value, tail[0], tail[1:len(tail)])
}
return value
}