115 lines
2.7 KiB
Go
115 lines
2.7 KiB
Go
package templater
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"log"
|
|
"path/filepath"
|
|
)
|
|
|
|
type Service struct {
|
|
ConfigFiles []ConfigFile `json:"ConfigFiles"`
|
|
Vars map[string]interface{} `json:"Vars"`
|
|
Daemons map[string]SystemService `json:"Daemons"`
|
|
Users map[string]SystemUser `json:"Users"`
|
|
Repos map[string]Repository `json:"Repositories"`
|
|
Packages map[string]SystemPackage `json:"Packages"`
|
|
}
|
|
|
|
func (s *Service) manageRepos(repos map[string]Repository) error {
|
|
for _, repo := range s.Repos {
|
|
if repo.Type == "helm" {
|
|
rp := HelmRepository{repo}
|
|
if err := rp.Manage(); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
if repo.Type == "apk" {
|
|
rp := APKRepository{repo}
|
|
if err := rp.Manage(); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
if repo.Type == "deb" {
|
|
rp := DebRepository{}
|
|
if err := rp.Manage(); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (s *Service) Manage(templateDir string) error {
|
|
|
|
// Manage packages repositories
|
|
log.Printf("Managing package repositories")
|
|
err := s.manageRepos(s.Repos)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
// Manage system packages
|
|
log.Printf("Installing packages")
|
|
for _, pack := range s.Packages {
|
|
err := pack.Manage()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
log.Printf("\tPackage %s installed\n", pack.Name)
|
|
}
|
|
|
|
log.Printf("Generating configuration files\n")
|
|
err = processConfigFiles(s.ConfigFiles, s.Vars, templateDir)
|
|
if err != nil {
|
|
return fmt.Errorf("ProcessingTemplatesFailed with error: %v", err)
|
|
}
|
|
|
|
log.Printf("Managing services:\n")
|
|
for _, daemon := range s.Daemons {
|
|
err = daemon.Manage()
|
|
if err != nil {
|
|
return fmt.Errorf("Error managing service daemons: %v", err)
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func processConfigFiles(tpls []ConfigFile, variables map[string]interface{}, templateDir string) error {
|
|
values, err := json.Marshal(variables)
|
|
if err != nil {
|
|
return fmt.Errorf("Error unmarshaling values on template process; %v", err)
|
|
}
|
|
|
|
var servicesToRestart []string
|
|
for _, tpl := range tpls {
|
|
fileExt := filepath.Ext(tpl.Source)
|
|
if fileExt == ".hcl" {
|
|
tpl.TemplateType = "hcl"
|
|
} else if fileExt == ".tpl" {
|
|
tpl.TemplateType = "go"
|
|
} else {
|
|
return fmt.Errorf("Unsupported file type %s, templates extensions have to be '.hcl' or '.tpl'", fileExt)
|
|
}
|
|
if err := tpl.Generate("/tmp/test", templateDir, values); err != nil {
|
|
return fmt.Errorf("Template %s generation failed with error %v", tpl.Source, err)
|
|
}
|
|
|
|
if len(tpl.Service) != 0 {
|
|
servicesToRestart = append(servicesToRestart, tpl.Service)
|
|
}
|
|
}
|
|
|
|
for _, srv := range servicesToRestart {
|
|
sv := SystemService{
|
|
Name: srv,
|
|
Enabled: true,
|
|
Type: "",
|
|
ToStart: true,
|
|
}
|
|
return sv.Restart()
|
|
}
|
|
return nil
|
|
}
|