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{