packer-cn/post-processor/docker/post-processor.go

57 lines
1.1 KiB
Go

package docker
import (
"bytes"
"errors"
"github.com/mitchellh/mapstructure"
"github.com/mitchellh/packer/packer"
"os/exec"
)
type Config struct {
Registry string
Username string
Password string
Email string
}
type PostProcessor struct {
config Config
}
func (p *PostProcessor) Configure(raw ...interface{}) error {
if err := mapstructure.Decode(raw, &p.config); err != nil {
return err
}
if p.config.Registry == "" {
p.config.Registry = "registry.docker.io"
}
if p.config.Username == "" {
return errors.New("Username is required to push docker image")
}
if p.config.Password == "" {
return errors.New("Password is required to push docker image")
}
return nil
}
func (p *PostProcessor) PostProcess(ui packer.Ui, artifact packer.Artifact) (packer.Artifact, bool, error) {
id := artifact.Id()
ui.Say("Pushing imgage: " + id)
// TODO: docker login
stdout := new(bytes.Buffer)
cmd := exec.Command("docker", "push", id)
cmd.Stdout = stdout
if err := cmd.Run(); err != nil {
ui.Say("Failed to push image: " + id)
return nil, true, err
}
return nil, true, nil
}