124 lines
2.6 KiB
Go
124 lines
2.6 KiB
Go
package registry
|
|
|
|
import (
|
|
"bytes"
|
|
"fmt"
|
|
"io/ioutil"
|
|
"os"
|
|
"path/filepath"
|
|
"strings"
|
|
"text/template"
|
|
"time"
|
|
|
|
"github.com/mholt/archiver/v3"
|
|
)
|
|
|
|
// Template is a tmpl project
|
|
type Template struct {
|
|
reg *Registry `toml:"-"`
|
|
Name string `toml:"name"`
|
|
Path string `toml:"path"`
|
|
Repository string `toml:"repository"`
|
|
Branch string `toml:"branch"`
|
|
Created time.Time `toml:"created"`
|
|
}
|
|
|
|
// ArchiveName is the name given to the archive for this Template
|
|
func (t *Template) ArchiveName() string {
|
|
return fmt.Sprintf("%s.tar.gz", t.Name)
|
|
}
|
|
|
|
// ArchivePath is the full path to the archive for this Template within the Registry
|
|
func (t *Template) ArchivePath() string {
|
|
return filepath.Join(t.reg.dir, t.ArchiveName())
|
|
}
|
|
|
|
// Execute runs the Template and copies to dest
|
|
func (t *Template) Execute(dest string, defaults, overwrite bool) error {
|
|
tmp, err := ioutil.TempDir(os.TempDir(), "tmpl")
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer os.RemoveAll(tmp)
|
|
|
|
if err := archiver.Unarchive(t.ArchivePath(), tmp); err != nil {
|
|
return err
|
|
}
|
|
|
|
prompts, err := prompt(tmp, defaults)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
funcs := mergeMaps(funcMap, prompts.ToFuncMap())
|
|
|
|
base := filepath.Join(tmp, "template")
|
|
return filepath.Walk(base, func(walkPath string, walkInfo os.FileInfo, walkErr error) error {
|
|
if walkErr != nil {
|
|
return walkErr
|
|
}
|
|
|
|
if walkInfo.IsDir() {
|
|
return nil
|
|
}
|
|
|
|
contents, err := ioutil.ReadFile(walkPath)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
newDest := strings.TrimPrefix(walkPath, base+"/")
|
|
newDest = filepath.Join(dest, newDest)
|
|
|
|
tmplDest, err := template.New("dest").Funcs(funcs).Parse(newDest)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
var buf bytes.Buffer
|
|
if err := tmplDest.Execute(&buf, prompts.ToMap()); err != nil {
|
|
return err
|
|
}
|
|
newDest = buf.String()
|
|
|
|
if err := os.MkdirAll(filepath.Dir(newDest), os.ModePerm); err != nil {
|
|
return err
|
|
}
|
|
|
|
oldFi, err := os.Lstat(walkPath)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
// Check if new file exists. If it does, only skip if not overwriting
|
|
if _, err := os.Lstat(newDest); err == nil && !overwrite {
|
|
return nil
|
|
}
|
|
|
|
newFi, err := os.OpenFile(newDest, os.O_RDWR|os.O_CREATE|os.O_TRUNC, oldFi.Mode())
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
tmplContents, err := template.New("tmpl").Funcs(funcs).Parse(string(contents))
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if err := tmplContents.Execute(newFi, prompts.ToMap()); err != nil {
|
|
return err
|
|
}
|
|
|
|
return newFi.Close()
|
|
})
|
|
}
|
|
|
|
func mergeMaps(maps ...map[string]interface{}) map[string]interface{} {
|
|
m := make(map[string]interface{})
|
|
for _, mm := range maps {
|
|
for k, v := range mm {
|
|
m[k] = v
|
|
}
|
|
}
|
|
return m
|
|
}
|