add cell,eq,hasprefix,hassuffix template functions

bin-templater/template/templating.go

@@ -3,6 +3,7 @@ package template
 import (
 	"bufio"
 	"encoding/json"
+	"fmt"
 	"github.com/leekchan/gtf"
 	"gopkg.in/yaml.v2"
 	"io"
@@ -269,6 +270,45 @@ func sub(x, y int) int {
 	return x - y
 }
 
+func eq(args ...interface{}) bool {
+	if len(args) == 0 {
+		return false
+	}
+	x := args[0]
+	switch x := x.(type) {
+	case string, int, int64, byte, float32, float64:
+		for _, y := range args[1:] {
+			if x == y {
+				return true
+			}
+		}
+		return false
+	}
+
+	for _, y := range args[1:] {
+		if reflect.DeepEqual(x, y) {
+			return true
+		}
+	}
+	return false
+}
+
+type Cell struct{ v interface{} }
+
+func NewCell(v ...interface{}) (*Cell, error) {
+	switch len(v) {
+	case 0:
+		return new(Cell), nil
+	case 1:
+		return &Cell{v[0]}, nil
+	default:
+		return nil, fmt.Errorf("wrong number of args: want 0 or 1, got %v", len(v))
+	}
+}
+
+func (c *Cell) Set(v interface{}) *Cell { c.v = v; return c }
+func (c *Cell) Get() interface{}        { return c.v }
+
 func init() {
 	TemplateFunctions = make(map[string]interface{})
 	TemplateFunctions["base"] = path.Base
@@ -284,6 +324,8 @@ func init() {
 	TemplateFunctions["contains"] = strings.Contains
 	TemplateFunctions["replace"] = strings.Replace
 	TemplateFunctions["repeat"] = strings.Repeat
+	TemplateFunctions["hasPrefix"] = strings.HasPrefix
+	TemplateFunctions["hasSuffix"] = strings.HasSuffix
 	TemplateFunctions["orDef"] = orDef
 	TemplateFunctions["orDefs"] = orDefs
 	TemplateFunctions["ifOrDef"] = ifOrDef
@@ -303,6 +345,8 @@ func init() {
 	TemplateFunctions["mod"] = mod
 	TemplateFunctions["toJson"] = toJson
 	TemplateFunctions["toYaml"] = toYaml
+	TemplateFunctions["cell"] = NewCell
+	TemplateFunctions["eq"] = eq
 
 	TemplateFunctions["IsMapFirst"] = IsMapFirst
 	TemplateFunctions["IsMapLast"] = IsMapLast