packer-opennebula/cmd/post-processor/image-template/image_template.go

102 lines
2.7 KiB
Go

package main
import (
"fmt"
"strings"
"github.com/Cadoles/goca"
"github.com/hashicorp/packer/common"
"github.com/hashicorp/packer/helper/config"
"github.com/hashicorp/packer/packer"
"github.com/hashicorp/packer/template/interpolate"
)
// Config handles configuration options of the OpenNebula post-processor
type Config struct {
common.PackerConfig `mapstructure:",squash"`
User string
Password string
Endpoint string
ImageName string `mapstructure:"image_name"`
ImageTemplate []string `mapstructure:"image_template"`
DatastoreName string `mapstructure:"datastore_name"`
MergeTemplate bool `mapstructure:"merge_template"`
ctx interpolate.Context
}
// PostProcessor is an OpenNebula post processor for packer
type PostProcessor struct {
Conf *Config
}
// Configure configures the packer OpenNebula post-processor
func (pp *PostProcessor) Configure(raws ...interface{}) error {
conf := &Config{}
err := config.Decode(conf, &config.DecodeOpts{
Interpolate: true,
InterpolateContext: &conf.ctx,
}, raws...)
if err != nil {
return err
}
pp.Conf = conf
return nil
}
// PostProcess creates/updates your configured image to OpenNebula using the XML-RPC API
func (pp *PostProcessor) PostProcess(ui packer.Ui, a packer.Artifact) (packer.Artifact, bool, error) {
// Initialize OpenNebula XML-RPC API client
config := goca.NewConfig(pp.Conf.User, pp.Conf.Password, pp.Conf.Endpoint)
if err := goca.SetClient(config); err != nil {
return a, true, err
}
// Search image template by its name
img, err := goca.NewImageFromName(pp.Conf.ImageName)
if err != nil && err.Error() != "resource not found" {
return a, true, err
}
ui.Say(fmt.Sprintf("Connecting to OpenNebula RPC endpoint '%s' with provided credentials...", pp.Conf.Endpoint))
// Generate image template
tmplStr := serializeImageTemplate(pp.Conf.ImageTemplate)
// Fi the image template can not be found, we create it
if img == nil {
ui.Say(fmt.Sprintf("The OpenNebula image template '%s' could not be found. Creating it...", pp.Conf.ImageName))
// Search image datastore's ID
datastore, err := goca.NewDatastoreFromName(pp.Conf.DatastoreName)
if err != nil {
return a, true, err
}
// Create image template
if _, err = goca.CreateImage(tmplStr, datastore.ID); err != nil {
return a, true, err
}
} else {
ui.Say(fmt.Sprintf("The OpenNebula image template '%s' has been found. Updating it...", pp.Conf.ImageName))
// Update image template
if err := img.Update(tmplStr, pp.Conf.MergeTemplate); err != nil {
return a, true, err
}
}
ui.Say("Operation completed.")
return a, true, nil
}
func serializeImageTemplate(tmpl []string) string {
return strings.Join(tmpl, "\n")
}