2021-02-02 18:05:04 +01:00
|
|
|
package packer
|
2013-05-23 21:37:16 -07:00
|
|
|
|
|
|
|
import (
|
2019-03-19 18:11:19 +01:00
|
|
|
"context"
|
2013-05-23 21:37:16 -07:00
|
|
|
"log"
|
2018-01-22 17:21:10 -08:00
|
|
|
|
2019-12-17 11:25:56 +01:00
|
|
|
"github.com/hashicorp/hcl/v2/hcldec"
|
2020-12-17 13:29:25 -08:00
|
|
|
packersdk "github.com/hashicorp/packer-plugin-sdk/packer"
|
2013-05-23 21:37:16 -07:00
|
|
|
)
|
|
|
|
|
|
|
|
type cmdProvisioner struct {
|
2020-12-01 14:25:14 -08:00
|
|
|
p packersdk.Provisioner
|
2021-02-02 18:05:04 +01:00
|
|
|
client *PluginClient
|
2013-05-23 21:37:16 -07:00
|
|
|
}
|
|
|
|
|
2019-12-17 11:25:56 +01:00
|
|
|
func (p *cmdProvisioner) ConfigSpec() hcldec.ObjectSpec {
|
|
|
|
defer func() {
|
|
|
|
r := recover()
|
|
|
|
p.checkExit(r, nil)
|
|
|
|
}()
|
|
|
|
|
|
|
|
return p.p.ConfigSpec()
|
|
|
|
}
|
|
|
|
|
2013-06-06 17:07:42 -07:00
|
|
|
func (c *cmdProvisioner) Prepare(configs ...interface{}) error {
|
2013-05-23 21:37:16 -07:00
|
|
|
defer func() {
|
|
|
|
r := recover()
|
|
|
|
c.checkExit(r, nil)
|
|
|
|
}()
|
|
|
|
|
2013-06-06 17:07:42 -07:00
|
|
|
return c.p.Prepare(configs...)
|
2013-05-23 21:37:16 -07:00
|
|
|
}
|
|
|
|
|
2020-11-19 15:10:00 -08:00
|
|
|
func (c *cmdProvisioner) Provision(ctx context.Context, ui packersdk.Ui, comm packersdk.Communicator, generatedData map[string]interface{}) error {
|
2013-05-23 21:37:16 -07:00
|
|
|
defer func() {
|
|
|
|
r := recover()
|
|
|
|
c.checkExit(r, nil)
|
|
|
|
}()
|
|
|
|
|
2019-07-11 09:37:59 -07:00
|
|
|
return c.p.Provision(ctx, ui, comm, generatedData)
|
2013-08-30 23:21:15 -07:00
|
|
|
}
|
|
|
|
|
2013-05-23 21:37:16 -07:00
|
|
|
func (c *cmdProvisioner) checkExit(p interface{}, cb func()) {
|
2013-08-28 11:21:02 -07:00
|
|
|
if c.client.Exited() && cb != nil {
|
2013-05-23 21:37:16 -07:00
|
|
|
cb()
|
2013-08-19 23:38:22 -07:00
|
|
|
} else if p != nil && !Killed {
|
2013-05-23 21:37:16 -07:00
|
|
|
log.Panic(p)
|
|
|
|
}
|
|
|
|
}
|