templatefile/pkg/templater/services.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
	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 for\n")
	err = processConfigFiles(s.ConfigFiles, s.Vars, templateDir)
	if err != nil {
		return fmt.Errorf("ProcessingTemplatesFailed with error: %v", err)
	}

	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)
	}

	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 extention %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)
		}

	}
	return nil
}