implements base64{enc,dec} interpolation funcs
fixes #3320 Signed-off-by: Gorka Lerchundi Osa <glertxundi@gmail.com>
This commit is contained in:
parent
140b3377f4
commit
70522fb770
|
@ -2,6 +2,7 @@ package config
|
|||
|
||||
import (
|
||||
"bytes"
|
||||
"encoding/base64"
|
||||
"errors"
|
||||
"fmt"
|
||||
"io/ioutil"
|
||||
|
@ -29,6 +30,8 @@ func init() {
|
|||
"length": interpolationFuncLength(),
|
||||
"replace": interpolationFuncReplace(),
|
||||
"split": interpolationFuncSplit(),
|
||||
"base64enc": interpolationFuncBase64Encode(),
|
||||
"base64dec": interpolationFuncBase64Decode(),
|
||||
}
|
||||
}
|
||||
|
||||
|
@ -392,3 +395,33 @@ func interpolationFuncValues(vs map[string]ast.Variable) ast.Function {
|
|||
},
|
||||
}
|
||||
}
|
||||
|
||||
// interpolationFuncBase64Encode implements the "base64enc" function that allows
|
||||
// Base64 encoding.
|
||||
func interpolationFuncBase64Encode() ast.Function {
|
||||
return ast.Function{
|
||||
ArgTypes: []ast.Type{ast.TypeString},
|
||||
ReturnType: ast.TypeString,
|
||||
Callback: func(args []interface{}) (interface{}, error) {
|
||||
s := args[0].(string)
|
||||
return base64.StdEncoding.EncodeToString([]byte(s)), nil
|
||||
},
|
||||
}
|
||||
}
|
||||
|
||||
// interpolationFuncBase64Decode implements the "base64dec" function that allows
|
||||
// Base64 decoding.
|
||||
func interpolationFuncBase64Decode() ast.Function {
|
||||
return ast.Function{
|
||||
ArgTypes: []ast.Type{ast.TypeString},
|
||||
ReturnType: ast.TypeString,
|
||||
Callback: func(args []interface{}) (interface{}, error) {
|
||||
s := args[0].(string)
|
||||
sDec, err := base64.StdEncoding.DecodeString(s)
|
||||
if err != nil {
|
||||
return "", fmt.Errorf("failed to decode base64 data '%s'", s)
|
||||
}
|
||||
return string(sDec), nil
|
||||
},
|
||||
}
|
||||
}
|
||||
|
|
|
@ -584,6 +584,39 @@ func TestInterpolateFuncElement(t *testing.T) {
|
|||
})
|
||||
}
|
||||
|
||||
func TestInterpolateFuncBase64Encode(t *testing.T) {
|
||||
testFunction(t, testFunctionConfig{
|
||||
Cases: []testFunctionCase{
|
||||
// Regular base64 encoding
|
||||
{
|
||||
`${base64enc("abc123!?$*&()'-=@~")}`,
|
||||
"YWJjMTIzIT8kKiYoKSctPUB+",
|
||||
false,
|
||||
},
|
||||
},
|
||||
})
|
||||
}
|
||||
|
||||
func TestInterpolateFuncBase64Decode(t *testing.T) {
|
||||
testFunction(t, testFunctionConfig{
|
||||
Cases: []testFunctionCase{
|
||||
// Regular base64 decoding
|
||||
{
|
||||
`${base64dec("YWJjMTIzIT8kKiYoKSctPUB+")}`,
|
||||
"abc123!?$*&()'-=@~",
|
||||
false,
|
||||
},
|
||||
|
||||
// Invalid base64 data decoding
|
||||
{
|
||||
`${base64dec("this-is-an-invalid-base64-data")}`,
|
||||
nil,
|
||||
true,
|
||||
},
|
||||
},
|
||||
})
|
||||
}
|
||||
|
||||
type testFunctionConfig struct {
|
||||
Cases []testFunctionCase
|
||||
Vars map[string]ast.Variable
|
||||
|
|
Loading…
Reference in New Issue