added MapDir function for mapping out etcd directories

pinterb · May 28, 2014 · cd99c2a · cd99c2a
1 parent 3a20318
commit cd99c2a
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 0 deletions.
diff --git a/resource/template/resource.go b/resource/template/resource.go
@@ -118,6 +118,7 @@ func (t *TemplateResource) createStageFile() error {
 	tplFuncMap["Base"] = path.Base
 
 	tplFuncMap["GetDir"] = t.Dirs.Get
+    tplFuncMap["MapDir"] = mapNodes
 	tmpl := template.Must(template.New(path.Base(t.Src)).Funcs(tplFuncMap).ParseFiles(t.Src))
 	if err = tmpl.Execute(temp, t.Vars); err != nil {
 		return err

diff --git a/resource/template/util.go b/resource/template/util.go
@@ -14,6 +14,7 @@ import (
 	"syscall"
 
 	"github.com/kelseyhightower/confd/log"
+    "github.com/coreos/go-etcd/etcd"
 )
 
 func appendPrefix(prefix string, keys []string) []string {
@@ -34,6 +35,18 @@ func cleanKeys(vars map[string]interface{}, prefix string) map[string]interface{
 	return clean
 }
 
+func mapNodes(node *etcd.Node) map[string]interface{} {
+	result := make(map[string]interface{})
+    for _, node := range node.Nodes {
+        if len(node.Nodes) > 0 {
+            result[node.Key] = node.Nodes
+        } else {
+            result[node.Key] = node.Value
+        }
+    }
+    return cleanKeys(result, node.Key)
+}
+
 // isFileExist reports whether path exits.
 func isFileExist(fpath string) bool {
 	if _, err := os.Stat(fpath); os.IsNotExist(err) {