1 files changed, 72 insertions, 0 deletions
diff --git a/funcs/math.go b/funcs/math.go
new file mode 100644
index 00000000..e26e42dc
--- /dev/null
+++ b/funcs/math.go
@@ -0,0 +1,72 @@
+package funcs
+
+import (
+	"fmt"
+	gmath "math"
+	"sync"
+
+	"github.com/hairyhenderson/gomplate/conv"
+
+	"github.com/hairyhenderson/gomplate/math"
+)
+
+var (
+	mathNS     *MathFuncs
+	mathNSInit sync.Once
+)
+
+// MathNS - the math namespace
+func MathNS() *MathFuncs {
+	mathNSInit.Do(func() { mathNS = &MathFuncs{} })
+	return mathNS
+}
+
+// AddMathFuncs -
+func AddMathFuncs(f map[string]interface{}) {
+	f["math"] = MathNS
+
+	f["add"] = MathNS().Add
+	f["sub"] = MathNS().Sub
+	f["mul"] = MathNS().Mul
+	f["div"] = MathNS().Div
+	f["rem"] = MathNS().Rem
+	f["pow"] = MathNS().Pow
+}
+
+// MathFuncs -
+type MathFuncs struct{}
+
+// Add -
+func (f *MathFuncs) Add(n ...interface{}) int64 {
+	return math.AddInt(conv.ToInt64s(n...)...)
+}
+
+// Mul -
+func (f *MathFuncs) Mul(n ...interface{}) int64 {
+	return math.MulInt(conv.ToInt64s(n...)...)
+}
+
+// Sub -
+func (f *MathFuncs) Sub(a, b interface{}) int64 {
+	return conv.ToInt64(a) - conv.ToInt64(b)
+}
+
+// Div -
+func (f *MathFuncs) Div(a, b interface{}) (int64, error) {
+	divisor := conv.ToInt64(a)
+	dividend := conv.ToInt64(b)
+	if dividend == 0 {
+		return 0, fmt.Errorf("Error: division by 0")
+	}
+	return divisor / dividend, nil
+}
+
+// Rem -
+func (f *MathFuncs) Rem(a, b interface{}) int64 {
+	return conv.ToInt64(a) % conv.ToInt64(b)
+}
+
+// Pow -
+func (f *MathFuncs) Pow(a, b interface{}) int64 {
+	return conv.ToInt64(gmath.Pow(conv.ToFloat64(a), conv.ToFloat64(b)))
+}