terraform/builtin/provisioners/local-exec/resource_provisioner.go

96 lines
2.1 KiB
Go
Raw Normal View History

2014-07-09 22:34:08 +02:00
package localexec
import (
"fmt"
"io"
2014-07-09 22:34:08 +02:00
"os/exec"
"runtime"
"github.com/armon/circbuf"
"github.com/hashicorp/terraform/helper/config"
"github.com/hashicorp/terraform/terraform"
"github.com/mitchellh/go-linereader"
2014-07-09 22:34:08 +02:00
)
const (
// maxBufSize limits how much output we collect from a local
// invocation. This is to prevent TF memory usage from growing
// to an enormous amount due to a faulty process.
maxBufSize = 8 * 1024
)
type ResourceProvisioner struct{}
func (p *ResourceProvisioner) Apply(
2014-10-05 01:29:33 +02:00
o terraform.UIOutput,
s *terraform.InstanceState,
2014-07-22 19:38:39 +02:00
c *terraform.ResourceConfig) error {
2014-07-09 22:34:08 +02:00
// Get the command
commandRaw, ok := c.Config["command"]
2014-07-09 22:34:08 +02:00
if !ok {
2014-07-22 19:38:39 +02:00
return fmt.Errorf("local-exec provisioner missing 'command'")
2014-07-09 22:34:08 +02:00
}
command, ok := commandRaw.(string)
if !ok {
2014-07-22 19:38:39 +02:00
return fmt.Errorf("local-exec provisioner command must be a string")
2014-07-09 22:34:08 +02:00
}
// Execute the command using a shell
var shell, flag string
if runtime.GOOS == "windows" {
shell = "cmd"
flag = "/C"
} else {
shell = "/bin/sh"
flag = "-c"
}
// Setup the reader that will read the lines from the command
pr, pw := io.Pipe()
copyDoneCh := make(chan struct{})
go p.copyOutput(o, pr, copyDoneCh)
2014-07-09 22:34:08 +02:00
// Setup the command
cmd := exec.Command(shell, flag, command)
output, _ := circbuf.NewBuffer(maxBufSize)
cmd.Stderr = io.MultiWriter(output, pw)
cmd.Stdout = io.MultiWriter(output, pw)
// Output what we're about to run
o.Output(fmt.Sprintf(
"Executing: %s %s \"%s\"",
shell, flag, command))
2014-07-09 22:34:08 +02:00
// Run the command to completion
err := cmd.Run()
// Close the write-end of the pipe so that the goroutine mirroring output
// ends properly.
pw.Close()
<-copyDoneCh
if err != nil {
2014-07-22 19:38:39 +02:00
return fmt.Errorf("Error running command '%s': %v. Output: %s",
2014-07-09 22:34:08 +02:00
command, err, output.Bytes())
}
2014-07-22 19:38:39 +02:00
return nil
2014-07-09 22:34:08 +02:00
}
func (p *ResourceProvisioner) Validate(c *terraform.ResourceConfig) ([]string, []error) {
validator := config.Validator{
Required: []string{"command"},
}
return validator.Validate(c)
}
func (p *ResourceProvisioner) copyOutput(
o terraform.UIOutput, r io.Reader, doneCh chan<- struct{}) {
defer close(doneCh)
lr := linereader.New(r)
for line := range lr.Ch {
o.Output(line)
}
}