config: add file() function for reading file contents
This commit is contained in:
parent
625fb65526
commit
80385c7682
|
@ -2,6 +2,7 @@ package config
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"fmt"
|
"fmt"
|
||||||
|
"io/ioutil"
|
||||||
"strings"
|
"strings"
|
||||||
)
|
)
|
||||||
|
|
||||||
|
@ -10,10 +11,28 @@ var Funcs map[string]InterpolationFunc
|
||||||
|
|
||||||
func init() {
|
func init() {
|
||||||
Funcs = map[string]InterpolationFunc{
|
Funcs = map[string]InterpolationFunc{
|
||||||
|
"file": interpolationFuncFile,
|
||||||
"lookup": interpolationFuncLookup,
|
"lookup": interpolationFuncLookup,
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// interpolationFuncFile implements the "file" function that allows
|
||||||
|
// loading contents from a file.
|
||||||
|
func interpolationFuncFile(
|
||||||
|
vs map[string]string, args ...string) (string, error) {
|
||||||
|
if len(args) != 1 {
|
||||||
|
return "", fmt.Errorf(
|
||||||
|
"file expects 1 arguments, got %d", len(args))
|
||||||
|
}
|
||||||
|
|
||||||
|
data, err := ioutil.ReadFile(args[0])
|
||||||
|
if err != nil {
|
||||||
|
return "", err
|
||||||
|
}
|
||||||
|
|
||||||
|
return string(data), nil
|
||||||
|
}
|
||||||
|
|
||||||
// interpolationFuncLookup implements the "lookup" function that allows
|
// interpolationFuncLookup implements the "lookup" function that allows
|
||||||
// dynamic lookups of map types within a Terraform configuration.
|
// dynamic lookups of map types within a Terraform configuration.
|
||||||
func interpolationFuncLookup(
|
func interpolationFuncLookup(
|
||||||
|
|
|
@ -1,9 +1,58 @@
|
||||||
package config
|
package config
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"io/ioutil"
|
||||||
|
"os"
|
||||||
"testing"
|
"testing"
|
||||||
)
|
)
|
||||||
|
|
||||||
|
func TestInterpolateFuncFile(t *testing.T) {
|
||||||
|
tf, err := ioutil.TempFile("", "tf")
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("err: %s", err)
|
||||||
|
}
|
||||||
|
path := tf.Name()
|
||||||
|
tf.Write([]byte("foo"))
|
||||||
|
tf.Close()
|
||||||
|
defer os.Remove(path)
|
||||||
|
|
||||||
|
cases := []struct {
|
||||||
|
Args []string
|
||||||
|
Result string
|
||||||
|
Error bool
|
||||||
|
}{
|
||||||
|
{
|
||||||
|
[]string{path},
|
||||||
|
"foo",
|
||||||
|
false,
|
||||||
|
},
|
||||||
|
|
||||||
|
// Invalid path
|
||||||
|
{
|
||||||
|
[]string{"/i/dont/exist"},
|
||||||
|
"",
|
||||||
|
true,
|
||||||
|
},
|
||||||
|
|
||||||
|
// Too many args
|
||||||
|
{
|
||||||
|
[]string{"foo", "bar"},
|
||||||
|
"",
|
||||||
|
true,
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
for i, tc := range cases {
|
||||||
|
actual, err := interpolationFuncFile(nil, tc.Args...)
|
||||||
|
if (err != nil) != tc.Error {
|
||||||
|
t.Fatalf("%d: err: %s", i, err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if actual != tc.Result {
|
||||||
|
t.Fatalf("%d: bad: %#v", i, actual)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
func TestInterpolateFuncLookup(t *testing.T) {
|
func TestInterpolateFuncLookup(t *testing.T) {
|
||||||
cases := []struct {
|
cases := []struct {
|
||||||
M map[string]string
|
M map[string]string
|
||||||
|
|
Loading…
Reference in New Issue