-
-
Notifications
You must be signed in to change notification settings - Fork 1.3k
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Merge pull request #733 from alpacamybags118/varfile_extract
adding function to extract terraform vars from var file
- Loading branch information
Showing
5 changed files
with
425 additions
and
0 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,149 @@ | ||
| package terraform | ||
|
|
||
| import ( | ||
| "errors" | ||
| "fmt" | ||
| "io/ioutil" | ||
| "reflect" | ||
| "testing" | ||
|
|
||
| "github.com/hashicorp/hcl" | ||
| "github.com/stretchr/testify/require" | ||
| ) | ||
|
|
||
| // GetVariableAsStringFromVarFile Gets the string represention of a variable from a provided input file found in VarFile | ||
| // For list or map, use GetVariableAsListFromVarFile or GetVariableAsMapFromVarFile, respectively. | ||
| func GetVariableAsStringFromVarFile(t *testing.T, fileName string, key string) string { | ||
| result, err := GetVariableAsStringFromVarFileE(t, fileName, key) | ||
| require.NoError(t, err) | ||
|
|
||
| return result | ||
| } | ||
|
|
||
| // GetVariableAsStringFromVarFileE Gets the string represention of a variable from a provided input file found in VarFile | ||
| // Will return an error if GetAllVariablesFromVarFileE returns an error or the key provided does not exist in the file. | ||
| // For list or map, use GetVariableAsListFromVarFile or GetVariableAsMapFromVarFile, respectively. | ||
| func GetVariableAsStringFromVarFileE(t *testing.T, fileName string, key string) (string, error) { | ||
| var variables map[string]interface{} | ||
|
|
||
| err := GetAllVariablesFromVarFileE(t, fileName, &variables) | ||
|
|
||
| if err != nil { | ||
| return "", err | ||
| } | ||
|
|
||
| variable, exists := variables[key] | ||
|
|
||
| if !exists { | ||
| return "", InputFileKeyNotFound{FilePath: fileName, Key: key} | ||
| } | ||
|
|
||
| return fmt.Sprintf("%v", variable), nil | ||
| } | ||
|
|
||
| // GetVariableAsMapFromVarFile Gets the map represention of a variable from a provided input file found in VarFile | ||
| // Note that this returns a map of strings. For maps containing complex types, use GetAllVariablesFromVarFile. | ||
| func GetVariableAsMapFromVarFile(t *testing.T, fileName string, key string) map[string]string { | ||
| result, err := GetVariableAsMapFromVarFileE(t, fileName, key) | ||
| require.NoError(t, err) | ||
|
|
||
| return result | ||
| } | ||
|
|
||
| // GetVariableAsMapFromVarFileE Gets the map represention of a variable from a provided input file found in VarFile. | ||
| // Note that this returns a map of strings. For maps containing complex types, use GetAllVariablesFromVarFile | ||
| // Returns an error if GetAllVariablesFromVarFileE returns an error, the key provided does not exist, or the value associated with the key is not a map | ||
| func GetVariableAsMapFromVarFileE(t *testing.T, fileName string, key string) (map[string]string, error) { | ||
| var variables map[string]interface{} | ||
|
|
||
| resultMap := make(map[string]string) | ||
| err := GetAllVariablesFromVarFileE(t, fileName, &variables) | ||
|
|
||
| if err != nil { | ||
| return nil, err | ||
| } | ||
|
|
||
| variable, exists := variables[key] | ||
|
|
||
| if !exists { | ||
| return nil, InputFileKeyNotFound{FilePath: fileName, Key: key} | ||
| } | ||
|
|
||
| if reflect.TypeOf(variable).String() != "[]map[string]interface {}" { | ||
| return nil, UnexpectedOutputType{Key: key, ExpectedType: "[]map[string]interface {}", ActualType: reflect.TypeOf(variable).String()} | ||
| } | ||
|
|
||
| mapKeys := variable.([]map[string]interface{}) | ||
|
|
||
| if len(mapKeys) == 0 { | ||
| return nil, errors.New("no map keys could be found for given map") | ||
| } | ||
|
|
||
| for mapKey, mapVal := range mapKeys[0] { | ||
| resultMap[mapKey] = fmt.Sprintf("%v", mapVal) | ||
| } | ||
|
|
||
| return resultMap, nil | ||
| } | ||
|
|
||
| // GetVariableAsListFromVarFile Gets the string list represention of a variable from a provided input file found in VarFile | ||
| // Note that this returns a list of strings. For lists containing complex types, use GetAllVariablesFromVarFile. | ||
| func GetVariableAsListFromVarFile(t *testing.T, fileName string, key string) []string { | ||
| result, err := GetVariableAsListFromVarFileE(t, fileName, key) | ||
| require.NoError(t, err) | ||
|
|
||
| return result | ||
| } | ||
|
|
||
| // GetVariableAsListFromVarFileE Gets the string list represention of a variable from a provided input file found in VarFile | ||
| // Note that this returns a list of strings. For lists containing complex types, use GetAllVariablesFromVarFile. | ||
| // Will return error if GetAllVariablesFromVarFileE returns an error, the key provided does not exist, or the value associated with the key is not a list | ||
| func GetVariableAsListFromVarFileE(t *testing.T, fileName string, key string) ([]string, error) { | ||
| var variables map[string]interface{} | ||
| resultArray := []string{} | ||
| err := GetAllVariablesFromVarFileE(t, fileName, &variables) | ||
|
|
||
| if err != nil { | ||
| return nil, err | ||
| } | ||
|
|
||
| variable, exists := variables[key] | ||
|
|
||
| if !exists { | ||
| return nil, InputFileKeyNotFound{FilePath: fileName, Key: key} | ||
| } | ||
|
|
||
| if reflect.TypeOf(variable).String() != "[]interface {}" { | ||
| return nil, UnexpectedOutputType{Key: key, ExpectedType: "[]interface {}", ActualType: reflect.TypeOf(variable).String()} | ||
| } | ||
|
|
||
| for _, item := range variable.([]interface{}) { | ||
| resultArray = append(resultArray, fmt.Sprintf("%v", item)) | ||
| } | ||
|
|
||
| return resultArray, nil | ||
| } | ||
|
|
||
| // GetAllVariablesFromVarFile Parses all data from a provided input file found in VarFile and stores the result in the value pointed to by out | ||
| func GetAllVariablesFromVarFile(t *testing.T, fileName string, out interface{}) { | ||
| err := GetAllVariablesFromVarFileE(t, fileName, out) | ||
| require.NoError(t, err) | ||
| } | ||
|
|
||
| // GetAllVariablesFromVarFileE Parses all data from a provided input file found ind in VarFile and stores the result in the value pointed to by out | ||
| // Returns an error if the specified file does not exist, the specified file is not readable, or the specified file cannot be decoded from HCL | ||
| func GetAllVariablesFromVarFileE(t *testing.T, fileName string, out interface{}) error { | ||
| fileContents, err := ioutil.ReadFile(fileName) | ||
|
|
||
| if err != nil { | ||
| return err | ||
| } | ||
|
|
||
| err = hcl.Decode(out, string(fileContents)) | ||
|
|
||
| if err != nil { | ||
| return HclDecodeError{FilePath: fileName, ErrorText: err.Error()} | ||
| } | ||
|
|
||
| return nil | ||
| } |
Oops, something went wrong.