From cd05ec424e17476d29bcae8b2c2fab61e5b2768d Mon Sep 17 00:00:00 2001
From: Gorka Lerchundi Osa <glertxundi@gmail.com>
Date: Wed, 11 Nov 2015 09:51:30 +0100
Subject: [PATCH] implements gzip interpolation func

---
 config/interpolate_funcs.go      | 28 ++++++++++++++++++++++++++++
 config/interpolate_funcs_test.go | 12 ++++++++++++
 2 files changed, 40 insertions(+)

diff --git a/config/interpolate_funcs.go b/config/interpolate_funcs.go
index 3ae127221d3b..7edf2358eba0 100644
--- a/config/interpolate_funcs.go
+++ b/config/interpolate_funcs.go
@@ -1,6 +1,8 @@
 package config
 
 import (
+	"bytes"
+	"compress/gzip"
 	"crypto/md5"
 	"crypto/sha1"
 	"crypto/sha256"
@@ -76,6 +78,7 @@ func Funcs() map[string]ast.Function {
 		"floor":        interpolationFuncFloor(),
 		"format":       interpolationFuncFormat(),
 		"formatlist":   interpolationFuncFormatList(),
+		"gzip":         interpolationFuncGzip(),
 		"index":        interpolationFuncIndex(),
 		"join":         interpolationFuncJoin(),
 		"jsonencode":   interpolationFuncJSONEncode(),
@@ -650,6 +653,31 @@ func interpolationFuncFormatList() ast.Function {
 	}
 }
 
+// interpolationFuncGzip implements the "gzip" function that allows gzip compression.
+func interpolationFuncGzip() ast.Function {
+	return ast.Function{
+		ArgTypes:   []ast.Type{ast.TypeString},
+		ReturnType: ast.TypeString,
+		Callback: func(args []interface{}) (interface{}, error) {
+			s := args[0].(string)
+
+			var b bytes.Buffer
+			gz := gzip.NewWriter(&b)
+			if _, err := gz.Write([]byte(s)); err != nil {
+				return "", fmt.Errorf("failed to gzip raw data: '%s'", s)
+			}
+			if err := gz.Flush(); err != nil {
+				return "", fmt.Errorf("failed to gzip raw data: '%s'", s)
+			}
+			if err := gz.Close(); err != nil {
+				return "", fmt.Errorf("failed to gzip raw data: '%s'", s)
+			}
+
+			return b.String(), nil
+		},
+	}
+}
+
 // interpolationFuncIndex implements the "index" function that allows one to
 // find the index of a specific element in a list
 func interpolationFuncIndex() ast.Function {
diff --git a/config/interpolate_funcs_test.go b/config/interpolate_funcs_test.go
index fe68f38c485d..deb9c65f87f2 100644
--- a/config/interpolate_funcs_test.go
+++ b/config/interpolate_funcs_test.go
@@ -1281,6 +1281,18 @@ func TestInterpolateFuncFormatList(t *testing.T) {
 	})
 }
 
+func TestInterpolateFuncGzip(t *testing.T) {
+	testFunction(t, testFunctionConfig{
+		Cases: []testFunctionCase{
+			{
+				`${base64encode(gzip("test"))}`,
+				"H4sIAAAJbogA/ypJLS4BAAAA//8BAAD//wx+f9gEAAAA",
+				false,
+			},
+		},
+	})
+}
+
 func TestInterpolateFuncIndex(t *testing.T) {
 	testFunction(t, testFunctionConfig{
 		Vars: map[string]ast.Variable{