format_utils.go

package utils

import (
	"bufio"
	"encoding/json"
	"errors"
	"html/template"
	"os"
	"strings"
	"text/tabwriter"

	"github.com/golang/glog"
)

var templates = map[string]string{
	"SingleVersionPackageDiff":    SingleVersionDiffOutput,
	"MultiVersionPackageDiff":     MultiVersionDiffOutput,
	"HistDiff":                    HistoryDiffOutput,
	"DirDiff":                     FSDiffOutput,
	"ListAnalyze":                 ListAnalysisOutput,
	"FileAnalyze":                 FileAnalysisOutput,
	"MultiVersionPackageAnalyze":  MultiVersionPackageOutput,
	"SingleVersionPackageAnalyze": SingleVersionPackageOutput,
}

func JSONify(diff interface{}) error {
	diffBytes, err := json.MarshalIndent(diff, "", "  ")
	if err != nil {
		return err
	}
	f := bufio.NewWriter(os.Stdout)
	defer f.Flush()
	f.Write(diffBytes)
	return nil
}

func getTemplate(templateType string) (string, error) {
	if template, ok := templates[templateType]; ok {
		return template, nil
	}
	return "", errors.New("No available template")
}

func TemplateOutput(diff interface{}, templateType string) error {
	outputTmpl, err := getTemplate(templateType)
	if err != nil {
		glog.Error(err)

	}
	funcs := template.FuncMap{"join": strings.Join}
	tmpl, err := template.New("tmpl").Funcs(funcs).Parse(outputTmpl)
	if err != nil {
		glog.Error(err)
		return err
	}
	w := tabwriter.NewWriter(os.Stdout, 8, 8, 8, ' ', 0)
	err = tmpl.Execute(w, diff)
	if err != nil {
		glog.Error(err)
		return err
	}
	w.Flush()
	return nil
}