mirror of https://github.com/zieckey/gochart.git
105 lines
2.1 KiB
Go
105 lines
2.1 KiB
Go
package main
|
|
|
|
import (
|
|
"errors"
|
|
"io/ioutil"
|
|
"log"
|
|
"os"
|
|
"path/filepath"
|
|
"strings"
|
|
|
|
"github.com/zieckey/goini"
|
|
)
|
|
|
|
const DataPrefix = "Data|"
|
|
const emptyRunes = " \r\t\v"
|
|
|
|
type ChartIf interface {
|
|
Parse(ini *goini.INI, file string) (map[string]string, error)
|
|
Template() string
|
|
}
|
|
|
|
type TemplateArgs struct {
|
|
args map[string]string
|
|
tmpl string
|
|
}
|
|
|
|
func Parse(file string) (tt TemplateArgs, err error) {
|
|
ini := goini.New()
|
|
err = ini.ParseFile(file)
|
|
if err != nil {
|
|
return tt, err
|
|
}
|
|
|
|
t, _ := ini.Get("ChartType")
|
|
if f, ok := ChartHandlers[t]; ok {
|
|
tt.args, err = f.Parse(ini, file)
|
|
tt.tmpl = f.Template()
|
|
}
|
|
|
|
return tt, err
|
|
}
|
|
|
|
func LookupChartFiles(dir string) ([]string, error) {
|
|
var files []string = make([]string, 0, 5)
|
|
|
|
err := filepath.Walk(dir, func(path string, f os.FileInfo, err error) error {
|
|
if f == nil {
|
|
return err
|
|
}
|
|
|
|
if f.IsDir() {
|
|
return nil
|
|
}
|
|
|
|
if ok, err := filepath.Match("*.chart", f.Name()); err != nil {
|
|
return err
|
|
} else if ok {
|
|
log.Printf("Add chart file %v", path)
|
|
files = append(files, path)
|
|
}
|
|
return nil
|
|
})
|
|
|
|
if len(files) == 0 {
|
|
return files, errors.New("Not found any *.chart files")
|
|
}
|
|
|
|
return files, err
|
|
}
|
|
|
|
/**
|
|
* 读取配置文件获取有序map
|
|
*
|
|
* @return order map
|
|
* @param {[type]} ini string [description]
|
|
*/
|
|
func LoadConfGetOrderMap(configFile string) ([]string, map[string]string, error) {
|
|
stream, err := ioutil.ReadFile(configFile)
|
|
if err != nil {
|
|
return nil, nil, errors.New("cannot load config file")
|
|
}
|
|
content := string(stream)
|
|
|
|
confMap := make(map[string]string)
|
|
mapkeys := make([]string, 0)
|
|
|
|
lines := strings.Split(content, "\n")
|
|
for _, line := range lines {
|
|
line = strings.Trim(line, emptyRunes)
|
|
//过滤注释 //过滤非DataPrefix
|
|
if line == "" || line[0] == '#' || !strings.HasPrefix(line, DataPrefix) {
|
|
continue
|
|
}
|
|
//过滤
|
|
parts := strings.SplitN(line, "=", 2)
|
|
if len(parts) == 2 {
|
|
key := strings.Trim(parts[0], emptyRunes)
|
|
value := strings.Trim(parts[1], emptyRunes)
|
|
mapkeys = append(mapkeys, key)
|
|
confMap[key] = value
|
|
}
|
|
}
|
|
return mapkeys, confMap, nil
|
|
}
|