index.go

package templates

import (
	"bytes"
	"fmt"
	"io"
	"io/ioutil"
	"net/http"
	"os"
	"path/filepath"
	"strings"
	"text/template"

	"github.com/bmatcuk/doublestar"
	"github.com/mat007/go-msi/manifest"
)

var funcMap = template.FuncMap{
	"dec": func(i int) int {
		return i - 1
	},
	"inc": func(i int) int {
		return i + 1
	},
	"cat": func(filename string) string {
		out, err := ioutil.ReadFile(filename)
		if err != nil {
			panic(fmt.Errorf("failed to read file %q", filename))
		}
		return string(out)
	},
	"download": func(url string) string {
		response, err := http.Get(url)
		if err != nil {
			panic(fmt.Errorf("failed to download url %q", url))
		}
		defer response.Body.Close()
		var b bytes.Buffer
		if _, err := io.Copy(&b, response.Body); err != nil {
			panic(fmt.Errorf("failed to download url %q", url))
		}
		return b.String()
	},
	"upper": strings.ToUpper,
}

// Find all wxs files in given directory
func Find(srcDir string, pattern string) ([]string, error) {
	glob := filepath.Join(srcDir, pattern)
	return doublestar.Glob(glob)
}

// GenerateTemplate generates given src template to out file using given manifest
func GenerateTemplate(wixFile *manifest.WixManifest, src string, out string) error {
	tpl, err := template.New("").Funcs(funcMap).ParseFiles(src)
	if err != nil {
		return err
	}

	fileWriter, err := os.Create(out)
	if err != nil {
		return err
	}
	defer fileWriter.Close()
	err = tpl.ExecuteTemplate(fileWriter, filepath.Base(src), wixFile)
	if err != nil {
		return err
	}
	return nil
}