packer-cn/packer/plugin/hook.go

29 lines
480 B
Go
Raw Normal View History

2013-05-11 13:46:17 -04:00
package plugin
import (
"github.com/mitchellh/packer/packer"
"log"
)
type cmdHook struct {
2013-05-20 19:50:35 -04:00
hook packer.Hook
client *Client
2013-05-11 13:46:17 -04:00
}
func (c *cmdHook) Run(name string, ui packer.Ui, comm packer.Communicator, data interface{}) error {
2013-05-11 13:46:17 -04:00
defer func() {
r := recover()
c.checkExit(r, nil)
}()
return c.hook.Run(name, ui, comm, data)
2013-05-11 13:46:17 -04:00
}
func (c *cmdHook) checkExit(p interface{}, cb func()) {
if c.client.Exited() {
cb()
} else if p != nil && !Killed {
2013-05-11 13:46:17 -04:00
log.Panic(p)
}
}