102 lines
2.2 KiB
Go
102 lines
2.2 KiB
Go
package goldsmith
|
|
|
|
import (
|
|
"os"
|
|
"path/filepath"
|
|
"reflect"
|
|
"strings"
|
|
)
|
|
|
|
type fileInfo struct {
|
|
os.FileInfo
|
|
path string
|
|
}
|
|
|
|
func cleanPath(path string) string {
|
|
if filepath.IsAbs(path) {
|
|
var err error
|
|
if path, err = filepath.Rel("/", path); err != nil {
|
|
panic(err)
|
|
}
|
|
}
|
|
|
|
return filepath.Clean(path)
|
|
}
|
|
|
|
func scanDir(root string, infos chan fileInfo) {
|
|
defer close(infos)
|
|
|
|
filepath.Walk(root, func(path string, info os.FileInfo, err error) error {
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
infos <- fileInfo{FileInfo: info, path: path}
|
|
return nil
|
|
})
|
|
}
|
|
|
|
func setDelimValue(container interface{}, path string, data interface{}) bool {
|
|
containerVal := reflect.Indirect(reflect.ValueOf(container))
|
|
|
|
segments := strings.Split(path, ".")
|
|
segmentHead := segments[0]
|
|
|
|
if len(segments) > 1 {
|
|
var fieldVal reflect.Value
|
|
switch containerVal.Kind() {
|
|
case reflect.Map:
|
|
fieldVal = containerVal.MapIndex(reflect.ValueOf(segmentHead))
|
|
case reflect.Struct:
|
|
fieldVal = containerVal.FieldByName(segmentHead)
|
|
if fieldVal.CanAddr() {
|
|
fieldVal = fieldVal.Addr()
|
|
}
|
|
}
|
|
|
|
if fieldVal.IsValid() && fieldVal.CanInterface() {
|
|
pathRest := strings.Join(segments[1:], ".")
|
|
return setDelimValue(fieldVal.Interface(), pathRest, data)
|
|
}
|
|
} else {
|
|
switch containerVal.Kind() {
|
|
case reflect.Map:
|
|
containerVal.SetMapIndex(reflect.ValueOf(segmentHead), reflect.ValueOf(data))
|
|
return true
|
|
case reflect.Struct:
|
|
fieldVal := containerVal.FieldByName(segmentHead)
|
|
if fieldVal.CanSet() {
|
|
fieldVal.Set(reflect.ValueOf(data))
|
|
return true
|
|
}
|
|
}
|
|
}
|
|
|
|
return false
|
|
}
|
|
|
|
func getDelimValue(container interface{}, path string) (interface{}, bool) {
|
|
containerVal := reflect.Indirect(reflect.ValueOf(container))
|
|
|
|
segments := strings.Split(path, ".")
|
|
segmentHead := segments[0]
|
|
|
|
var fieldVal reflect.Value
|
|
switch containerVal.Kind() {
|
|
case reflect.Map:
|
|
fieldVal = containerVal.MapIndex(reflect.ValueOf(segmentHead))
|
|
case reflect.Struct:
|
|
fieldVal = containerVal.FieldByName(segmentHead)
|
|
}
|
|
|
|
if fieldVal.IsValid() && fieldVal.CanInterface() {
|
|
if len(segments) > 1 {
|
|
return getDelimValue(fieldVal.Interface(), strings.Join(segments[1:], "."))
|
|
}
|
|
|
|
return fieldVal.Interface(), true
|
|
}
|
|
|
|
return nil, false
|
|
}
|