2019-05-16 20:24:53 -04:00
|
|
|
package chroot
|
|
|
|
|
|
|
|
import (
|
|
|
|
"context"
|
|
|
|
"errors"
|
2019-05-27 02:20:11 -04:00
|
|
|
"fmt"
|
|
|
|
"log"
|
2019-05-16 20:24:53 -04:00
|
|
|
"runtime"
|
|
|
|
|
2019-05-27 02:20:11 -04:00
|
|
|
"github.com/Azure/azure-sdk-for-go/profiles/latest/compute/mgmt/compute"
|
2019-05-31 14:39:43 -04:00
|
|
|
"github.com/hashicorp/packer/builder/amazon/chroot"
|
2019-05-16 20:24:53 -04:00
|
|
|
azcommon "github.com/hashicorp/packer/builder/azure/common"
|
2019-05-27 02:20:11 -04:00
|
|
|
"github.com/hashicorp/packer/builder/azure/common/client"
|
2019-05-16 20:24:53 -04:00
|
|
|
"github.com/hashicorp/packer/common"
|
|
|
|
"github.com/hashicorp/packer/helper/config"
|
|
|
|
"github.com/hashicorp/packer/helper/multistep"
|
|
|
|
"github.com/hashicorp/packer/packer"
|
|
|
|
"github.com/hashicorp/packer/template/interpolate"
|
|
|
|
)
|
|
|
|
|
|
|
|
type Config struct {
|
|
|
|
common.PackerConfig `mapstructure:",squash"`
|
|
|
|
|
2019-05-27 02:20:11 -04:00
|
|
|
FromScratch bool `mapstructure:"from_scratch"`
|
|
|
|
|
2019-05-31 14:39:43 -04:00
|
|
|
CommandWrapper string `mapstructure:"command_wrapper"`
|
|
|
|
MountOptions []string `mapstructure:"mount_options"`
|
|
|
|
MountPartition string `mapstructure:"mount_partition"`
|
|
|
|
MountPath string `mapstructure:"mount_path"`
|
|
|
|
PreMountCommands []string `mapstructure:"pre_mount_commands"`
|
|
|
|
PostMountCommands []string `mapstructure:"post_mount_commands"`
|
2019-05-28 20:40:44 -04:00
|
|
|
|
2019-05-27 02:20:11 -04:00
|
|
|
OSDiskSizeGB int32 `mapstructure:"osdisk_size_gb"`
|
|
|
|
OSDiskStorageAccountType string `mapstructure:"osdisk_storageaccounttype"`
|
|
|
|
|
2019-05-16 20:24:53 -04:00
|
|
|
ctx interpolate.Context
|
|
|
|
}
|
|
|
|
|
|
|
|
type Builder struct {
|
|
|
|
config Config
|
|
|
|
runner multistep.Runner
|
|
|
|
}
|
|
|
|
|
|
|
|
func (b *Builder) Prepare(raws ...interface{}) ([]string, error) {
|
|
|
|
b.config.ctx.Funcs = azcommon.TemplateFuncs
|
|
|
|
err := config.Decode(&b.config, &config.DecodeOpts{
|
|
|
|
Interpolate: true,
|
|
|
|
InterpolateContext: &b.config.ctx,
|
|
|
|
InterpolateFilter: &interpolate.RenderFilter{
|
|
|
|
Exclude: []string{
|
|
|
|
// fields to exclude from interpolation
|
|
|
|
},
|
|
|
|
},
|
|
|
|
}, raws...)
|
|
|
|
|
2019-05-27 02:20:11 -04:00
|
|
|
// defaults
|
|
|
|
if b.config.OSDiskStorageAccountType == "" {
|
|
|
|
b.config.OSDiskStorageAccountType = string(compute.PremiumLRS)
|
|
|
|
}
|
|
|
|
|
2019-05-16 20:24:53 -04:00
|
|
|
// checks, accumulate any errors or warnings
|
|
|
|
var errs *packer.MultiError
|
|
|
|
var warns []string
|
|
|
|
|
2019-05-27 02:20:11 -04:00
|
|
|
if b.config.FromScratch {
|
|
|
|
if b.config.OSDiskSizeGB == 0 {
|
|
|
|
errs = packer.MultiErrorAppend(
|
|
|
|
errs, errors.New("osdisk_size_gb is required with from_scratch."))
|
|
|
|
}
|
2019-05-28 20:40:44 -04:00
|
|
|
if len(b.config.PreMountCommands) == 0 {
|
|
|
|
errs = packer.MultiErrorAppend(
|
|
|
|
errs, errors.New("pre_mount_commands is required with from_scratch."))
|
|
|
|
}
|
2019-05-27 02:20:11 -04:00
|
|
|
}
|
|
|
|
|
2019-05-16 20:24:53 -04:00
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
return warns, errs
|
|
|
|
}
|
|
|
|
|
|
|
|
func (b *Builder) Run(ctx context.Context, ui packer.Ui, hook packer.Hook) (packer.Artifact, error) {
|
|
|
|
if runtime.GOOS != "linux" {
|
|
|
|
return nil, errors.New("The azure-chroot builder only works on Linux environments.")
|
|
|
|
}
|
|
|
|
|
2019-05-27 02:20:11 -04:00
|
|
|
var azcli client.AzureClientSet
|
|
|
|
|
2019-05-28 20:40:44 -04:00
|
|
|
wrappedCommand := func(command string) (string, error) {
|
|
|
|
ictx := b.config.ctx
|
|
|
|
ictx.Data = &struct{ Command string }{Command: command}
|
|
|
|
return interpolate.Render(b.config.CommandWrapper, &ictx)
|
|
|
|
}
|
|
|
|
|
2019-05-16 20:24:53 -04:00
|
|
|
// Setup the state bag and initial state for the steps
|
|
|
|
state := new(multistep.BasicStateBag)
|
|
|
|
state.Put("config", &b.config)
|
|
|
|
state.Put("hook", hook)
|
|
|
|
state.Put("ui", ui)
|
2019-05-31 14:39:43 -04:00
|
|
|
state.Put("wrappedCommand", chroot.CommandWrapper(wrappedCommand))
|
2019-05-16 20:24:53 -04:00
|
|
|
|
2019-05-27 02:20:11 -04:00
|
|
|
info, err := azcli.MetadataClient().GetComputeInfo()
|
|
|
|
if err != nil {
|
|
|
|
log.Printf("MetadataClient().GetComputeInfo(): error: %+v", err)
|
|
|
|
err := fmt.Errorf(
|
|
|
|
"Error retrieving information ARM resource ID and location" +
|
|
|
|
"of the VM that Packer is running on.\n" +
|
|
|
|
"Please verify that Packer is running on a proper Azure VM.")
|
|
|
|
ui.Error(err.Error())
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
|
|
|
|
osDiskName := "PackerBuiltOsDisk"
|
|
|
|
|
|
|
|
state.Put("instance", info)
|
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
|
2019-05-16 20:24:53 -04:00
|
|
|
// Build the steps
|
|
|
|
var steps []multistep.Step
|
|
|
|
|
2019-05-27 02:20:11 -04:00
|
|
|
if !b.config.FromScratch {
|
|
|
|
panic("Only from_scratch is currently implemented")
|
|
|
|
// create disk from PIR / managed image (warn on non-linux images)
|
|
|
|
} else {
|
|
|
|
steps = append(steps,
|
|
|
|
&StepCreateNewDisk{
|
|
|
|
SubscriptionID: info.SubscriptionID,
|
|
|
|
ResourceGroup: info.ResourceGroupName,
|
|
|
|
DiskName: osDiskName,
|
|
|
|
DiskSizeGB: b.config.OSDiskSizeGB,
|
|
|
|
DiskStorageAccountType: b.config.OSDiskStorageAccountType,
|
|
|
|
})
|
|
|
|
}
|
|
|
|
|
2019-05-28 20:40:44 -04:00
|
|
|
steps = append(steps,
|
2019-05-29 14:00:04 -04:00
|
|
|
&StepAttachDisk{ // sets 'device' in stateBag
|
|
|
|
SubscriptionID: info.SubscriptionID,
|
|
|
|
ResourceGroup: info.ResourceGroupName,
|
|
|
|
DiskName: osDiskName,
|
|
|
|
},
|
2019-05-31 14:39:43 -04:00
|
|
|
&chroot.StepPreMountCommands{
|
2019-05-28 20:40:44 -04:00
|
|
|
Commands: b.config.PreMountCommands,
|
|
|
|
},
|
2019-05-31 14:39:43 -04:00
|
|
|
&StepMountDevice{
|
|
|
|
MountOptions: b.config.MountOptions,
|
|
|
|
MountPartition: b.config.MountPartition,
|
|
|
|
MountPath: b.config.MountPath,
|
|
|
|
},
|
|
|
|
&chroot.StepPostMountCommands{
|
|
|
|
Commands: b.config.PostMountCommands,
|
|
|
|
},
|
2019-05-28 20:40:44 -04:00
|
|
|
)
|
|
|
|
|
2019-05-16 20:24:53 -04:00
|
|
|
// Run!
|
|
|
|
b.runner = common.NewRunner(steps, b.config.PackerConfig, ui)
|
|
|
|
b.runner.Run(ctx, state)
|
|
|
|
|
|
|
|
// If there was an error, return that
|
|
|
|
if rawErr, ok := state.GetOk("error"); ok {
|
|
|
|
return nil, rawErr.(error)
|
|
|
|
}
|
|
|
|
|
|
|
|
return nil, nil
|
|
|
|
}
|
|
|
|
|
|
|
|
var _ packer.Builder = &Builder{}
|