Add docker-import post-processor.

Implemented initial working version
of Docker image importing code. #774

config.go

@@ -43,7 +43,8 @@ const defaultConfig = `
 	"post-processors": {
 		"vagrant": "packer-post-processor-vagrant",
 		"vsphere": "packer-post-processor-vsphere",
-                "docker-push": "packer-post-processor-docker-push"
+    "docker-push": "packer-post-processor-docker-push",
+    "docker-import": "packer-post-processor-docker-import"
 	},
 
 	"provisioners": {

plugin/post-processor-docker-import/main.go

+package main
+
+import (
+	"github.com/mitchellh/packer/packer/plugin"
+	"github.com/mitchellh/packer/post-processor/docker-import"
+)
+
+func main() {
+	server, err := plugin.Server()
+	if err != nil {
+		panic(err)
+	}
+	server.RegisterPostProcessor(new(dockerimport.PostProcessor))
+	server.Serve()
+}

plugin/post-processor-docker-import/main_test.go

+package main

post-processor/docker-import/post-processor.go

+package dockerimport
+
+import (
+	"fmt"
+	"github.com/mitchellh/packer/common"
+	"github.com/mitchellh/packer/packer"
+	"io"
+	"os"
+	"os/exec"
+)
+
+type Config struct {
+	common.PackerConfig `mapstructure:",squash"`
+
+	Repository string `mapstructure:"repository"`
+	Tag        string `mapstructure:"tag"`
+	Dockerfile string `mapstructure:"dockerfile"`
+
+	tpl *packer.ConfigTemplate
+}
+
+type PostProcessor struct {
+	config Config
+}
+
+func (p *PostProcessor) Configure(raws ...interface{}) error {
+	_, err := common.DecodeConfig(&p.config, raws...)
+	if err != nil {
+		return err
+	}
+
+	p.config.tpl, err = packer.NewConfigTemplate()
+	if err != nil {
+		return err
+	}
+	p.config.tpl.UserVars = p.config.PackerUserVars
+
+	// Accumulate any errors
+	errs := new(packer.MultiError)
+
+	templates := map[string]*string{
+		"repository": &p.config.Repository,
+	}
+
+	for key, ptr := range templates {
+		if *ptr == "" {
+			errs = packer.MultiErrorAppend(
+				errs, fmt.Errorf("%s must be set", key))
+		}
+
+		*ptr, err = p.config.tpl.Process(*ptr, nil)
+		if err != nil {
+			errs = packer.MultiErrorAppend(
+				errs, fmt.Errorf("Error processing %s: %s", key, err))
+		}
+	}
+
+	if len(errs.Errors) > 0 {
+		return errs
+	}
+
+	return nil
+
+}
+
+func (p *PostProcessor) PostProcess(ui packer.Ui, artifact packer.Artifact) (packer.Artifact, bool, error) {
+	id := artifact.Id()
+	ui.Say("Importing image: " + id)
+
+	if p.config.Tag == "" {
+
+		cmd := exec.Command("docker",
+			"import",
+			"-",
+			p.config.Repository)
+
+		stdin, err := cmd.StdinPipe()
+
+		if err != nil {
+			return nil, false, err
+		}
+
+		// There should be only one artifact of the Docker builder
+		file, err := os.Open(artifact.Files()[0])
+
+		if err != nil {
+			return nil, false, err
+		}
+
+		defer file.Close()
+
+		if err := cmd.Start(); err != nil {
+			ui.Say("Image import failed")
+			return nil, false, err
+		}
+
+		go func() {
+			io.Copy(stdin, file)
+			// close stdin so that program will exit
+			stdin.Close()
+		}()
+
+		cmd.Wait()
+
+	} else {
+
+		cmd := exec.Command("docker",
+			"import",
+			"-",
+			p.config.Repository+":"+p.config.Tag)
+
+		stdin, err := cmd.StdinPipe()
+
+		if err != nil {
+			return nil, false, err
+		}
+
+		// There should be only one artifact of the Docker builder
+		file, err := os.Open(artifact.Files()[0])
+
+		if err != nil {
+			return nil, false, err
+		}
+
+		defer file.Close()
+
+		if err := cmd.Start(); err != nil {
+			ui.Say("Image import failed")
+			return nil, false, err
+		}
+
+		go func() {
+			io.Copy(stdin, file)
+			// close stdin so that program will exit
+			stdin.Close()
+		}()
+
+		cmd.Wait()
+
+	}
+
+	// Process Dockerfile if provided
+	if p.config.Dockerfile != "" {
+
+		cmd := exec.Command("docker", "build", id)
+		if err := cmd.Run(); err != nil {
+			ui.Say("Failed to build image: " + id)
+			return nil, false, err
+		}
+
+		// TODO implement dockerfile provisioning
+
+	}
+	return nil, false, nil
+}

post-processor/docker-push/post-processor.go

@@ -129,5 +129,5 @@ func (p *PostProcessor) PostProcess(ui packer.Ui, artifact packer.Artifact) (pac
 		return nil, false, err
 	}
 
-	return nil, true, nil
+	return nil, false, nil
 }