102 lines
2.5 KiB
Go
102 lines
2.5 KiB
Go
package templater
|
|
|
|
import (
|
|
"fmt"
|
|
"io/ioutil"
|
|
"log"
|
|
"os"
|
|
|
|
"github.com/imdario/mergo"
|
|
)
|
|
|
|
var CacheFilePath = "/var/cache/templater.db"
|
|
|
|
type TemplaterConfig struct {
|
|
Name string `json:"Name"`
|
|
TemplateDirectory string `json:"TemplateDirectory"`
|
|
RootDirectory string `json:"RootDirectory"`
|
|
Services map[string]Service `json:"Services"`
|
|
GlobalService Service `json:"Globals"`
|
|
}
|
|
|
|
func (tc *TemplaterConfig) loadCache() error {
|
|
// Load globals from cache
|
|
var cache Service
|
|
err := Load(CacheFilePath, &cache)
|
|
if err != nil {
|
|
fmt.Printf("Warning: No globals to load\n")
|
|
}
|
|
|
|
err = mergo.Merge(&tc.GlobalService, cache)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// Create new configuration "object"
|
|
func (tc *TemplaterConfig) New(confpath string, templateDir string, rootDir string) error {
|
|
// Load stored globals if needed
|
|
lerr := tc.loadCache()
|
|
if lerr != nil {
|
|
return lerr
|
|
}
|
|
// Check if the configuration path is a Directory or a file
|
|
fileInfo, err := os.Stat(confpath)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if fileInfo.IsDir() {
|
|
// The conf path is a directory we load all the files and merge data
|
|
files, err := ioutil.ReadDir(confpath)
|
|
if err != nil {
|
|
return fmt.Errorf("Templater configuration load failed with error: %v", err)
|
|
}
|
|
for _, file := range files {
|
|
fname := fmt.Sprintf("%s/%s", confpath, file.Name())
|
|
var ntc TemplaterConfig
|
|
err := Load(fname, &ntc)
|
|
if err != nil {
|
|
return fmt.Errorf("Templater configuration load failed with error: %v", err)
|
|
}
|
|
err = mergo.Merge(tc, ntc)
|
|
if err != nil {
|
|
return fmt.Errorf("Templater configuration load failed with error: %v", err)
|
|
}
|
|
}
|
|
} else {
|
|
// The conf path is a file we only load this file (of course)
|
|
err = Load(confpath, tc)
|
|
if err != nil {
|
|
return fmt.Errorf("Confiuration read failed with error: %v", err)
|
|
}
|
|
}
|
|
|
|
tc.TemplateDirectory = templateDir
|
|
tc.RootDirectory = rootDir
|
|
|
|
return nil
|
|
}
|
|
|
|
// Process the services contained in the configuration "object"
|
|
func (tc *TemplaterConfig) ManageServices() error {
|
|
// Get global vars to add on each service
|
|
gbls := tc.GlobalService.Vars
|
|
|
|
for name, svr := range tc.Services {
|
|
err := mergo.Merge(&svr.Vars, gbls)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
log.Printf("=== Working on service %s", name)
|
|
if err := svr.Manage(tc.TemplateDirectory, tc.RootDirectory); err != nil {
|
|
return err
|
|
}
|
|
log.Printf("=== Service %s processed", name)
|
|
log.Printf("")
|
|
}
|
|
return nil
|
|
}
|