66 lines
1.4 KiB
Go
66 lines
1.4 KiB
Go
package fix
|
|
|
|
import (
|
|
"github.com/mitchellh/mapstructure"
|
|
)
|
|
|
|
// FixerSSHTimout replaces ssh_wait_timeout with ssh_timeout
|
|
type FixerSSHTimout struct{}
|
|
|
|
func (FixerSSHTimout) Fix(input map[string]interface{}) (map[string]interface{}, error) {
|
|
type template struct {
|
|
Builders []interface{}
|
|
}
|
|
|
|
// 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.Builders {
|
|
var builders map[string]interface{}
|
|
if err := mapstructure.Decode(raw, &builders); err != nil {
|
|
// Ignore errors, could be a non-map
|
|
continue
|
|
}
|
|
|
|
if _, ok := builders["ssh_timeout"]; ok {
|
|
|
|
// drop ssh_wait_timeout if it is also included
|
|
if _, sshWaitTimeoutIncluded := builders["ssh_wait_timeout"]; sshWaitTimeoutIncluded {
|
|
delete(builders, "ssh_wait_timeout")
|
|
}
|
|
|
|
} else {
|
|
|
|
// replace ssh_wait_timeout with ssh_timeout if it exists
|
|
sshWaitTimeoutRaw, ok := builders["ssh_wait_timeout"]
|
|
if !ok {
|
|
continue
|
|
}
|
|
|
|
sshWaitTimeoutString, ok := sshWaitTimeoutRaw.(string)
|
|
if !ok {
|
|
continue
|
|
}
|
|
|
|
delete(builders, "ssh_wait_timeout")
|
|
builders["ssh_timeout"] = sshWaitTimeoutString
|
|
}
|
|
|
|
// Write all changes back to template
|
|
tpl.Builders[i] = builders
|
|
}
|
|
|
|
if len(tpl.Builders) > 0 {
|
|
input["builders"] = tpl.Builders
|
|
}
|
|
|
|
return input, nil
|
|
}
|
|
|
|
func (FixerSSHTimout) Synopsis() string {
|
|
return `Replaces "ssh_wait_timeout" with "ssh_timeout"`
|
|
}
|