77 lines
1.9 KiB
Go
77 lines
1.9 KiB
Go
|
package fix
|
||
|
|
|
||
|
|
import (
|
||
|
|
"strings"
|
||
|
|
|
||
|
|
"github.com/mitchellh/mapstructure"
|
||
|
|
)
|
||
|
|
|
||
|
|
// FixerGalaxyCommand removes the escape character from user
|
||
|
|
// environment variables and replace galaxycommand with galaxy_command
|
||
|
|
type FixerGalaxyCommand struct{}
|
||
|
|
|
||
|
|
func (FixerGalaxyCommand) Fix(input map[string]interface{}) (map[string]interface{}, error) {
|
||
|
|
type template struct {
|
||
|
|
Provisioners []interface{}
|
||
|
|
}
|
||
|
|
|
||
|
|
var gcUnescape = strings.NewReplacer(
|
||
|
|
"`$", "$",
|
||
|
|
"`\"", "\"",
|
||
|
|
"``", "`",
|
||
|
|
"`'", "'",
|
||
|
|
)
|
||
|
|
|
||
|
|
// Decode the input into our structure, if we can
|
||
|
|
var tpl template
|
||
|
|
if err := mapstructure.WeakDecode(input, &tpl); err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
|
||
|
|
for i, raw := range tpl.Provisioners {
|
||
|
|
var provisioners map[string]interface{}
|
||
|
|
if err := mapstructure.Decode(raw, &provisioners); err != nil {
|
||
|
|
// Ignore errors, could be a non-map
|
||
|
|
continue
|
||
|
|
}
|
||
|
|
|
||
|
|
if ok := provisioners["type"] == "ansible*" || provisioners["type"] == "ansible-local"; !ok {
|
||
|
|
continue
|
||
|
|
}
|
||
|
|
|
||
|
|
if _, ok := provisioners["galaxy_command"]; ok {
|
||
|
|
provisioners["galaxy_command"] = gcUnescape.Replace(provisioners["galaxy_command"].(string))
|
||
|
|
|
||
|
|
// drop galaxycommand if it is also included
|
||
|
|
if _, galaxyCommandIncluded := provisioners["galaxycommand"]; galaxyCommandIncluded {
|
||
|
|
delete(provisioners, "galaxycommand")
|
||
|
|
}
|
||
|
|
|
||
|
|
} else {
|
||
|
|
|
||
|
|
// replace galaxycommand with galaxy_command if it exists
|
||
|
|
galaxyCommandRaw, ok := provisioners["galaxycommand"]
|
||
|
|
if !ok {
|
||
|
|
continue
|
||
|
|
}
|
||
|
|
galaxyCommandString := gcUnescape.Replace(galaxyCommandRaw.(string))
|
||
|
|
|
||
|
|
delete(provisioners, "galaxycommand")
|
||
|
|
provisioners["galaxy_command"] = galaxyCommandString
|
||
|
|
}
|
||
|
|
|
||
|
|
// Write all changes back to template
|
||
|
|
tpl.Provisioners[i] = provisioners
|
||
|
|
}
|
||
|
|
|
||
|
|
if len(tpl.Provisioners) > 0 {
|
||
|
|
input["provisioners"] = tpl.Provisioners
|
||
|
|
}
|
||
|
|
|
||
|
|
return input, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (FixerGalaxyCommand) Synopsis() string {
|
||
|
|
return `Removes escapes from user env vars and updates provisioners to use "galaxy_command" rather than "galaxycommand"`
|
||
|
|
}
|