packer-cn/fix/fixer_galaxy_command.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"`
}