provisioner/shell: can specify multiple scripts to provision with

provisioner/shell/provisioner.go

@@ -28,6 +28,9 @@ type config struct {
 	// The local path of the shell script to upload and execute.
 	Path string
 
+	// An array of multiple scripts to run.
+	Scripts []string
+
 	// The remote path where the local shell script will be uploaded to.
 	// This should be set to a writable file that is in a pre-existing directory.
 	RemotePath string `mapstructure:"remote_path"`
@@ -64,17 +67,29 @@ func (p *Provisioner) Prepare(raws ...interface{}) error {
 		p.config.RemotePath = DefaultRemotePath
 	}
 
+	if p.config.Scripts == nil {
+		p.config.Scripts = make([]string, 0)
+	}
+
 	errs := make([]error, 0)
 
-	if p.config.Path == "" && p.config.Inline == nil {
-		errs = append(errs, errors.New("Either a path or inline script must be specified."))
-	} else if p.config.Path != "" && p.config.Inline != nil {
-		errs = append(errs, errors.New("Only a path or an inline script can be specified, not both."))
+	if p.config.Path != "" && len(p.config.Scripts) > 0 {
+		errs = append(errs, errors.New("Only one of path or scripts can be specified."))
 	}
 
 	if p.config.Path != "" {
-		if _, err := os.Stat(p.config.Path); err != nil {
-			errs = append(errs, fmt.Errorf("Bad script path: %s", err))
+		p.config.Scripts = []string{p.config.Path}
+	}
+
+	if len(p.config.Scripts) == 0 && p.config.Inline == nil {
+		errs = append(errs, errors.New("Either a script file or inline script must be specified."))
+	} else if len(p.config.Scripts) > 0 && p.config.Inline != nil {
+		errs = append(errs, errors.New("Only a script file or an inline script can be specified, not both."))
+	}
+
+	for _, path := range p.config.Scripts {
+		if _, err := os.Stat(path); err != nil {
+			errs = append(errs, fmt.Errorf("Bad script '%s': %s", path, err))
 		}
 	}
 
@@ -86,7 +101,7 @@ func (p *Provisioner) Prepare(raws ...interface{}) error {
 }
 
 func (p *Provisioner) Provision(ui packer.Ui, comm packer.Communicator) {
-	path := p.config.Path
+	scripts := make([]string, len(p.config.Scripts))
 
 	// If we have an inline script, then turn that into a temporary
 	// shell script and use that.
@@ -99,7 +114,7 @@ func (p *Provisioner) Provision(ui packer.Ui, comm packer.Communicator) {
 		defer os.Remove(tf.Name())
 
 		// Set the path to the temporary file
-		path = tf.Name()
+		scripts = append(scripts, tf.Name())
 
 		// Write our contents to it
 		writer := bufio.NewWriter(tf)
@@ -118,6 +133,7 @@ func (p *Provisioner) Provision(ui packer.Ui, comm packer.Communicator) {
 		tf.Close()
 	}
 
+	for _, path := range scripts {
 		ui.Say(fmt.Sprintf("Provisioning with shell script: %s", path))
 
 		log.Printf("Opening %s for reading", path)
@@ -167,7 +183,7 @@ func (p *Provisioner) Provision(ui packer.Ui, comm packer.Communicator) {
 			exitChan <- cmd.ExitStatus
 		}()
 
-OutputLoop:
+	OutputLoop:
 		for {
 			select {
 			case output := <-stderrChan:
@@ -189,4 +205,5 @@ OutputLoop:
 		for output := range stderrChan {
 			ui.Message(output)
 		}
+	}
 }

provisioner/shell/provisioner_test.go

@@ -36,11 +36,11 @@ func TestProvisionerPrepare_Defaults(t *testing.T) {
 }
 
 func TestProvisionerPrepare_Path(t *testing.T) {
-	var p Provisioner
 	config := testConfig()
 	delete(config, "inline")
 
 	config["path"] = "/this/should/not/exist"
+	p := new(Provisioner)
 	err := p.Prepare(config)
 	if err == nil {
 		t.Fatal("should have error")
@@ -54,6 +54,7 @@ func TestProvisionerPrepare_Path(t *testing.T) {
 	defer os.Remove(tf.Name())
 
 	config["path"] = tf.Name()
+	p = new(Provisioner)
 	err = p.Prepare(config)
 	if err != nil {
 		t.Fatalf("should not have error: %s", err)
@@ -85,3 +86,48 @@ func TestProvisionerPrepare_PathAndInline(t *testing.T) {
 		t.Fatal("should have error")
 	}
 }
+
+func TestProvisionerPrepare_PathAndScripts(t *testing.T) {
+	var p Provisioner
+	config := testConfig()
+
+	// Test with both
+	tf, err := ioutil.TempFile("", "packer")
+	if err != nil {
+		t.Fatalf("error tempfile: %s", err)
+	}
+	defer os.Remove(tf.Name())
+
+	config["inline"] = []interface{}{"foo"}
+	config["scripts"] = []string{tf.Name()}
+	err = p.Prepare(config)
+	if err == nil {
+		t.Fatal("should have error")
+	}
+}
+
+func TestProvisionerPrepare_Scripts(t *testing.T) {
+	config := testConfig()
+	delete(config, "inline")
+
+	config["scripts"] = []string{}
+	p := new(Provisioner)
+	err := p.Prepare(config)
+	if err == nil {
+		t.Fatal("should have error")
+	}
+
+	// Test with a good one
+	tf, err := ioutil.TempFile("", "packer")
+	if err != nil {
+		t.Fatalf("error tempfile: %s", err)
+	}
+	defer os.Remove(tf.Name())
+
+	config["scripts"] = []string{tf.Name()}
+	p = new(Provisioner)
+	err = p.Prepare(config)
+	if err != nil {
+		t.Fatalf("should not have error: %s", err)
+	}
+}