dati

commit e1d52a86e9fe173e3dbd8ab50de64501c8b55f7d
parent 6a6e5129851da03ba05cb7fe6e361d1772305ed0
Author: gearsix <gearsix@tuta.io>
Date:   Thu, 11 Feb 2021 20:09:08 +0000

suti.go: added data parsing

Diffstat:
M
suti.go
 | 
114
++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++-----------

1 file changed, 98 insertions(+), 16 deletions(-)
diff --git a/suti.go b/suti.go
@@ -1,37 +1,49 @@
 package main
 
 import (
+	"encoding/json"
 	"fmt"
+	"io"
+	"io/ioutil"
 	"os"
+	"path/filepath"
+	"strings"
 )
 
+type data interface{}
 type Options struct {
-	TemplatePaths []string
+	TemplatePaths   []string
 	GlobalDataPaths []string
-	DataPaths []string
-	DataKey string
-	SortData string
-	ConfigFile string
+	DataPaths       []string
+	DataKey         string
+	SortData        string
+	ConfigFile      string
 }
 
-func init () {
+var options Options
+
+func init() {
 	if len(os.Args) <= 1 {
 		print("nothing to do")
 		os.Exit(0)
 	}
 }
 
-func main () {
-	_ = parseArgs(os.Args[1:])
+func main() {
+	options := parseArgs(os.Args[1:])
+	d := LoadDataFiles(options.DataPaths)
+	fmt.Println(d)
+
+	return
 }
 
-func warn (action string, args ...interface{}) {
-	fmt.Println("WARNING", fmt.Sprintf(action, args...))
+func warn(msg string, args ...interface{}) {
+	fmt.Println("WARNING", fmt.Sprintf(msg, args...))
 	return
 }
 
 // custom arg parser because golang.org/pkg/flag doesn't support list args
-func parseArgs (args []string) (o Options) {
+func parseArgs(args []string) (o Options) {
 	var flag string
 	for a := 0; a < len(args); a++ {
 		arg := args[a]
@@ -42,14 +54,14 @@ func parseArgs (args []string) (o Options) {
 				flag = flag[1:]
 				ndelims++
 			}
-			
+
 			if ndelims > 2 {
 				warn("bad flag syntax: '%s'", arg)
 				flag = ""
 			}
-			
+
 			// set valid any flags that don't take arguments here
-			
+
 		} else if flag == "t" || flag == "template" {
 			o.TemplatePaths = append(o.TemplatePaths, arg)
 		} else if flag == "gd" || flag == "globaldata" {
@@ -69,6 +81,77 @@ func parseArgs (args []string) (o Options) {
 			flag = ""
 		}
 	}
+
+	return
+}
+
+// LoadDataFiles TODO
+func LoadDataFiles(paths []string) map[string]data {
+	var err error
+	var stat os.FileInfo
+	var d data
 	
+	loaded := make(map[string]data)
+
+	for _, dpath := range paths {
+		if stat, err = os.Stat(dpath); err != nil {
+			warn("skipping data file '%s' (%s)", dpath, err)
+			continue
+		}
+
+		if stat.IsDir() {
+			_ = filepath.Walk(dpath,
+				func(path string, info os.FileInfo, e error) error {
+					if e == nil && !info.IsDir() {
+						if d, e = LoadDataFile(path); e == nil {
+							loaded[path] = d
+						}
+					}
+
+					if e != nil {
+						warn("unable to load data from file '%s' (%s)", path, e)
+					}
+
+					return e
+				})
+		} else {
+			if d, err = LoadDataFile(dpath); err == nil {
+				loaded[dpath] = d
+			} else {
+				warn("unable to load data from file '%s' (%s)", dpath, err)
+			}
+		}
+	}
+
+	return loaded
+}
+
+// LoadDataFile TODO
+func LoadDataFile(path string) (d data, e error) {
+	var f *os.File
+
+	f, e = os.Open(path)
+	defer f.Close()
+	if e == nil {
+		dtype := strings.TrimPrefix(filepath.Ext(path), ".")
+		d, e = LoadData(dtype, f)
+	}
+
+	return d, e
+}
+
+// LoadData TODO
+func LoadData(lang string, in io.Reader) (d data, e error) {
+	var fbuf []byte
+	if fbuf, e = ioutil.ReadAll(in); e != nil {
+		return
+	}
+
+	if lang == "json" {
+		e = json.Unmarshal(fbuf, &d)
+	} else {
+		e = fmt.Errorf("%s is not a supported data language", lang)
+	}
+
 	return
-}
-\ No newline at end of file
+}