46 lines
1.1 KiB
Go
46 lines
1.1 KiB
Go
|
package chroot
|
||
|
|
|
||
|
|
import (
|
||
|
|
"github.com/mitchellh/multistep"
|
||
|
|
"github.com/mitchellh/packer/packer"
|
||
|
|
)
|
||
|
|
|
||
|
|
type postMountCommandsData struct {
|
||
|
|
Device string
|
||
|
|
MountPath string
|
||
|
|
}
|
||
|
|
|
||
|
|
// StepPostMountCommands allows running arbitrary commands after mounting the
|
||
|
|
// device, but prior to the bind mount and copy steps.
|
||
|
|
type StepPostMountCommands struct {
|
||
|
|
Commands []string
|
||
|
|
}
|
||
|
|
|
||
|
|
func (s *StepPostMountCommands) Run(state multistep.StateBag) multistep.StepAction {
|
||
|
|
config := state.Get("config").(*Config)
|
||
|
|
device := state.Get("device").(string)
|
||
|
|
mountPath := state.Get("mount_path").(string)
|
||
|
|
ui := state.Get("ui").(packer.Ui)
|
||
|
|
wrappedCommand := state.Get("wrappedCommand").(CommandWrapper)
|
||
|
|
|
||
|
|
if len(s.Commands) == 0 {
|
||
|
|
return multistep.ActionContinue
|
||
|
|
}
|
||
|
|
|
||
|
|
ctx := config.ctx
|
||
|
|
ctx.Data = &postMountCommandsData{
|
||
|
|
Device: device,
|
||
|
|
MountPath: mountPath,
|
||
|
|
}
|
||
|
|
|
||
|
|
ui.Say("Running post-mount commands...")
|
||
|
|
if err := RunLocalCommands(s.Commands, wrappedCommand, ctx, ui); err != nil {
|
||
|
|
state.Put("error", err)
|
||
|
|
ui.Error(err.Error())
|
||
|
|
return multistep.ActionHalt
|
||
|
|
}
|
||
|
|
return multistep.ActionContinue
|
||
|
|
}
|
||
|
|
|
||
|
|
func (s *StepPostMountCommands) Cleanup(state multistep.StateBag) {}
|