func main() {
runtime.GOMAXPROCS(runtime.NumCPU())
var err error
var cfg *config.Config
// create config and try to get b0x file from args
j := new(config.JSON)
err = j.FromArg()
// required info
if err != nil {
log.Fatal(err)
}
// load b0x file
cfg, err = j.Load()
cfg.Defaults()
files := make(map[string]*file.File)
dirs := new(dir.Dir)
// loop through b0x's [custom] objects
for _, c := range cfg.Custom {
err = c.Parse(&files, &dirs)
if err != nil {
log.Fatal(err)
}
}
// create files template and exec it
t := new(template.Template)
t.Set("files")
t.Variables = struct {
Pkg string
Files map[string]*file.File
Spread bool
DirList []string
}{
Pkg: cfg.Pkg,
Files: files,
Spread: cfg.Spread,
DirList: dirs.Clean(),
}
tmpl, err := t.Exec()
if err != nil {
log.Fatal(err)
}
// create dest folder when it doesn't exists
if _, err := os.Stat(cfg.Dest); os.IsNotExist(err) {
err = os.MkdirAll(cfg.Dest, 0777)
if err != nil {
log.Fatal(err)
}
}
// write final execuTed template into the destination file
err = ioutil.WriteFile(cfg.Dest+cfg.Output, tmpl, 0777)
if err != nil {
log.Fatal(err)
}
// write spreaded files
if cfg.Spread {
a := strings.Split(path.Dir(cfg.Dest), "/")
dirName := a[len(a)-1:][0]
for _, f := range files {
a := strings.Split(path.Dir(f.Path), "/")
fileDirName := a[len(a)-1:][0]
if dirName == fileDirName {
continue
}
// transform / to _ and some other chars...
customName := "b0xfile_" + utils.FixName(f.Path) + ".go"
// creates file template and exec it
t := new(template.Template)
t.Set("file")
t.Variables = struct {
Pkg string
Path string
Name string
Dir [][]string
Data string
}{
Pkg: cfg.Pkg,
Path: f.Path,
Name: f.Name,
Dir: dirs.List,
Data: f.Data,
}
tmpl, err := t.Exec()
if err != nil {
log.Fatal(err)
}
// write final execuTed template into the destination file
err = ioutil.WriteFile(cfg.Dest+customName, tmpl, 0777)
if err != nil {
log.Fatal(err)
}
}
}
// success
log.Println("fileb0x:", cfg.Dest+cfg.Output, "writen!")
}
func main() {
runtime.GOMAXPROCS(runtime.NumCPU())
// create config and try to get b0x file from args
f := new(config.File)
err = f.FromArg(true)
if err != nil {
log.Fatal(err)
}
// load b0x file's config
cfg, err = f.Load()
if err != nil {
log.Fatal(err)
}
err = cfg.Defaults()
if err != nil {
log.Fatal(err)
}
// creates a config that can be inserTed into custom
// without causing a import cycle
sharedConfig := new(custom.SharedConfig)
sharedConfig.Output = cfg.Output
sharedConfig.Compression = compression.NewGzip()
sharedConfig.Compression.Options = cfg.Compression
// loop through b0x's [custom] objects
for _, c := range cfg.Custom {
err = c.Parse(&files, &dirs, sharedConfig)
if err != nil {
log.Fatal(err)
}
}
// create files template and exec it
t := new(template.Template)
t.Set("files")
t.Variables = struct {
Pkg string
Files map[string]*file.File
Spread bool
DirList []string
Compression *compression.Options
Debug bool
}{
Pkg: cfg.Pkg,
Files: files,
Spread: cfg.Spread,
DirList: dirs.Clean(),
Compression: cfg.Compression,
Debug: cfg.Debug,
}
tmpl, err := t.Exec()
if err != nil {
log.Fatal(err)
}
// create dest folder when it doesn't exists
if !utils.Exists(cfg.Dest) {
err = os.MkdirAll(cfg.Dest, 0777)
if err != nil {
log.Fatal(err)
}
}
// gofmt
if cfg.Fmt {
tmpl, err = format.Source(tmpl)
if err != nil {
log.Fatal(err)
}
}
// write final execuTed template into the destination file
err = ioutil.WriteFile(cfg.Dest+cfg.Output, tmpl, 0777)
if err != nil {
log.Fatal(err)
}
// write spread files
if cfg.Spread {
a := strings.Split(path.Dir(cfg.Dest), "/")
dirName := a[len(a)-1:][0]
for _, f := range files {
a := strings.Split(path.Dir(f.Path), "/")
fileDirName := a[len(a)-1:][0]
if dirName == fileDirName {
continue
}
// transform / to _ and some other chars...
customName := "b0xfile_" + utils.FixName(f.Path) + ".go"
// creates file template and exec it
t := new(template.Template)
t.Set("file")
t.Variables = struct {
Pkg string
Path string
Name string
Dir [][]string
Data string
Compression *compression.Options
}{
Pkg: cfg.Pkg,
Path: f.Path,
Name: f.Name,
Dir: dirs.List,
Data: f.Data,
Compression: cfg.Compression,
}
tmpl, err := t.Exec()
if err != nil {
log.Fatal(err)
}
// gofmt
if cfg.Fmt {
tmpl, err = format.Source(tmpl)
if err != nil {
log.Fatal(err)
}
}
// write final execuTed template into the destination file
err = ioutil.WriteFile(cfg.Dest+customName, tmpl, 0777)
if err != nil {
log.Fatal(err)
}
}
}
// success
log.Println("fileb0x:", cfg.Dest+cfg.Output, "written!")
}