48 lines
768 B
Go
48 lines
768 B
Go
package plugin
|
|
|
|
import (
|
|
"log"
|
|
|
|
"github.com/hashicorp/packer/packer"
|
|
)
|
|
|
|
type cmdBuilder struct {
|
|
builder packer.Builder
|
|
client *Client
|
|
}
|
|
|
|
func (b *cmdBuilder) Prepare(config ...interface{}) ([]string, error) {
|
|
defer func() {
|
|
r := recover()
|
|
b.checkExit(r, nil)
|
|
}()
|
|
|
|
return b.builder.Prepare(config...)
|
|
}
|
|
|
|
func (b *cmdBuilder) Run(ui packer.Ui, hook packer.Hook) (packer.Artifact, error) {
|
|
defer func() {
|
|
r := recover()
|
|
b.checkExit(r, nil)
|
|
}()
|
|
|
|
return b.builder.Run(ui, hook)
|
|
}
|
|
|
|
func (b *cmdBuilder) Cancel() {
|
|
defer func() {
|
|
r := recover()
|
|
b.checkExit(r, nil)
|
|
}()
|
|
|
|
b.builder.Cancel()
|
|
}
|
|
|
|
func (c *cmdBuilder) checkExit(p interface{}, cb func()) {
|
|
if c.client.Exited() && cb != nil {
|
|
cb()
|
|
} else if p != nil && !Killed {
|
|
log.Panic(p)
|
|
}
|
|
}
|