Add Substr and Split template functions

Both of these can take any type the cast lib can turn into a string.

diff --git a/tpl/template.go b/tpl/template.go
index c61a91dc7b76d365a62280e054d7e276d407bc52..7e9f41f1f66c622e7d917ae36728e288f243e09c 100644 (file)
--- a/tpl/template.go
+++ b/tpl/template.go
@@ -188,6 +188,22 @@ func compareGetFloat(a interface{}, b interface{}) (float64, float64) {
        return left, right
 }
 
+func Substr(a interface{}, pos, length int) (string, error) {
+       aStr, err := cast.ToStringE(a)
+       if err != nil {
+               return "", err
+       }
+       return aStr[pos:length], nil
+}
+
+func Split(a interface{}, delimiter string) ([]string, error) {
+       aStr, err := cast.ToStringE(a)
+       if err != nil {
+               return []string{}, err
+       }
+       return strings.Split(aStr, delimiter), nil
+}
+
 func Intersect(l1, l2 interface{}) (interface{}, error) {
        if l1 == nil || l2 == nil {
                return make([]interface{}, 0), nil
@@ -1308,6 +1324,8 @@ func init() {
                "lt":          Lt,
                "le":          Le,
                "in":          In,
+               "substr":      Substr,
+               "split":       Split,
                "intersect":   Intersect,
                "isSet":       IsSet,
                "isset":       IsSet,

diff --git a/tpl/template_test.go b/tpl/template_test.go
index 0c9a1ac91393620f0646ebaf9da4987c967898c0..9a8b0a0890e7f1321513b0d7ac3efb2c94d50b36 100644 (file)
--- a/tpl/template_test.go
+++ b/tpl/template_test.go
@@ -276,6 +276,63 @@ func TestIn(t *testing.T) {
        }
 }
 
+func TestSubstr(t *testing.T) {
+       for i, this := range []struct {
+               v1 interface{}
+               v2 int
+               v3 int
+               expect string
+       }{
+               {"abc", 1, 2, "b"},
+               {"abc", 1, 3, "bc"},
+               {"abc", 0, 1, "a"},
+       } {
+               result, err := Substr(this.v1, this.v2, this.v3)
+
+               if err != nil {
+                       t.Errorf("[%d] failed: %s", i, err)
+                       continue
+               }
+
+               if result != this.expect {
+                       t.Errorf("[%d] Got %v but expected %v", i, result, this.expect)
+               }
+       }
+
+       _, err := Substr(tstNoStringer{}, 0, 1)
+       if err == nil {
+               t.Error("Expected error for non-string-convertable variable")
+       }
+}
+
+func TestSplit(t *testing.T) {
+       for i, this := range []struct {
+               v1 interface{}
+               v2 string
+               expect []string
+       }{
+               {"a, b", ", ", []string{"a", "b"}},
+               {"a & b & c", " & ", []string{"a", "b", "c"}},
+               {"http://exmaple.com", "http://", []string{"", "exmaple.com"}},
+       } {
+               result, err := Split(this.v1, this.v2)
+
+               if err != nil {
+                       t.Errorf("[%d] failed: %s", i, err)
+                       continue
+               }
+
+               if !reflect.DeepEqual(result, this.expect) {
+                       t.Errorf("[%d] Got %s but expected %s", i, result, this.expect)
+               }
+       }
+
+       _, err := Split(tstNoStringer{}, ",")
+       if err == nil {
+               t.Error("Expected error for non-string-convertable variable")
+       }
+}
+
 func TestIntersect(t *testing.T) {
        for i, this := range []struct {
                sequence1 interface{}