54 lines
1.4 KiB
Go
54 lines
1.4 KiB
Go
|
package proxmox
|
||
|
|
||
|
import (
|
||
|
"context"
|
||
|
"fmt"
|
||
|
|
||
|
"github.com/Telmate/proxmox-api-go/proxmox"
|
||
|
"github.com/hashicorp/packer/helper/multistep"
|
||
|
"github.com/hashicorp/packer/packer"
|
||
|
)
|
||
|
|
||
|
// stepConvertToTemplate takes the running VM configured in earlier steps, stops it, and
|
||
|
// converts it into a Proxmox template.
|
||
|
//
|
||
|
// It sets the template_id state which is used for Artifact lookup.
|
||
|
type stepConvertToTemplate struct{}
|
||
|
|
||
|
type templateConverter interface {
|
||
|
ShutdownVm(*proxmox.VmRef) (string, error)
|
||
|
CreateTemplate(*proxmox.VmRef) error
|
||
|
}
|
||
|
|
||
|
var _ templateConverter = &proxmox.Client{}
|
||
|
|
||
|
func (s *stepConvertToTemplate) Run(ctx context.Context, state multistep.StateBag) multistep.StepAction {
|
||
|
ui := state.Get("ui").(packer.Ui)
|
||
|
client := state.Get("proxmoxClient").(templateConverter)
|
||
|
vmRef := state.Get("vmRef").(*proxmox.VmRef)
|
||
|
|
||
|
ui.Say("Stopping VM")
|
||
|
_, err := client.ShutdownVm(vmRef)
|
||
|
if err != nil {
|
||
|
err := fmt.Errorf("Error converting VM to template, could not stop: %s", err)
|
||
|
state.Put("error", err)
|
||
|
ui.Error(err.Error())
|
||
|
return multistep.ActionHalt
|
||
|
}
|
||
|
|
||
|
ui.Say("Converting VM to template")
|
||
|
err = client.CreateTemplate(vmRef)
|
||
|
if err != nil {
|
||
|
err := fmt.Errorf("Error converting VM to template: %s", err)
|
||
|
state.Put("error", err)
|
||
|
ui.Error(err.Error())
|
||
|
return multistep.ActionHalt
|
||
|
}
|
||
|
|
||
|
state.Put("template_id", vmRef.VmId())
|
||
|
|
||
|
return multistep.ActionContinue
|
||
|
}
|
||
|
|
||
|
func (s *stepConvertToTemplate) Cleanup(state multistep.StateBag) {}
|