write fixer

This commit is contained in:
Megan Marsh 2019-03-29 15:18:14 -07:00
parent bcacd331c3
commit 2b57b10284
2 changed files with 60 additions and 0 deletions

View File

@ -39,6 +39,7 @@ func init() {
"powershell-escapes": new(FixerPowerShellEscapes),
"hyperv-deprecations": new(FixerHypervDeprecations),
"hyperv-vmxc-typo": new(FixerHypervVmxcTypo),
"hyperv-cpu-and-ram": new(FizerHypervCPUandRAM),
"vmware-compaction": new(FixerVMwareCompaction),
}
@ -61,5 +62,6 @@ func init() {
"docker-email",
"powershell-escapes",
"vmware-compaction",
"hyperv-cpu-and-ram",
}
}

View File

@ -0,0 +1,58 @@
package fix
import (
"github.com/mitchellh/mapstructure"
)
// FizerHypervCPUandRAM fixes the typo in "clone_from_vmxc_path" replacing
// it with "clone_from_vmcx_path" in Hyper-V VMCX builder templates
type FizerHypervCPUandRAM struct{}
func (FizerHypervCPUandRAM) Fix(input map[string]interface{}) (map[string]interface{}, error) {
// The type we'll decode into; we only care about builders
type template struct {
Builders []map[string]interface{}
}
// Decode the input into our structure, if we can
var tpl template
if err := mapstructure.Decode(input, &tpl); err != nil {
return nil, err
}
for _, builder := range tpl.Builders {
builderTypeRaw, ok := builder["type"]
if !ok {
continue
}
builderType, ok := builderTypeRaw.(string)
if !ok {
continue
}
if builderType != "hyperv-vmcx" && builderType != "hyperv-iso" {
continue
}
ncpus, ok := builder["cpu"]
if ok {
delete(builder, "cpu")
builder["cpus"] = ncpus
}
memory, ok := builder["ram_size"]
if ok {
delete(builder, "ram_size")
builder["memory"] = memory
}
}
input["builders"] = tpl.Builders
return input, nil
}
func (FizerHypervCPUandRAM) Synopsis() string {
return `Replaces "cpu" with "cpus" and "ram_size" with "memory"` +
`in Hyper-V VMCX builder templates`
}