38 lines
625 B
Go
38 lines
625 B
Go
|
package plugin
|
||
|
|
|
||
|
|
import (
|
||
|
|
"github.com/mitchellh/packer/packer"
|
||
|
|
"log"
|
||
|
|
)
|
||
|
|
|
||
|
|
type cmdPostProcessor struct {
|
||
|
|
p packer.PostProcessor
|
||
|
|
client *Client
|
||
|
|
}
|
||
|
|
|
||
|
|
func (c *cmdPostProcessor) Configure(config interface{}) error {
|
||
|
|
defer func() {
|
||
|
|
r := recover()
|
||
|
|
c.checkExit(r, nil)
|
||
|
|
}()
|
||
|
|
|
||
|
|
return c.p.Configure(config)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (c *cmdPostProcessor) PostProcess(a packer.Artifact) (packer.Artifact, error) {
|
||
|
|
defer func() {
|
||
|
|
r := recover()
|
||
|
|
c.checkExit(r, nil)
|
||
|
|
}()
|
||
|
|
|
||
|
|
return c.p.PostProcess(a)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (c *cmdPostProcessor) checkExit(p interface{}, cb func()) {
|
||
|
|
if c.client.Exited() {
|
||
|
|
cb()
|
||
|
|
} else if p != nil {
|
||
|
|
log.Panic(p)
|
||
|
|
}
|
||
|
|
}
|