mirror of
https://git.vectorsigma.ru/public/atlantis.git
synced 2026-08-07 13:08:34 +00:00
e45bc33330
Since we're including the command output inside the error, we don't need to also return the output. This was resulting in the error output being printed twice.
58 lines
1.8 KiB
Go
58 lines
1.8 KiB
Go
package runtime
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"os/exec"
|
|
"path/filepath"
|
|
|
|
version "github.com/hashicorp/go-version"
|
|
"github.com/runatlantis/atlantis/server/events/models"
|
|
)
|
|
|
|
// RunStepRunner runs custom commands.
|
|
type RunStepRunner struct {
|
|
DefaultTFVersion *version.Version
|
|
}
|
|
|
|
func (r *RunStepRunner) Run(ctx models.ProjectCommandContext, command string, path string) (string, error) {
|
|
cmd := exec.Command("sh", "-c", command) // #nosec
|
|
cmd.Dir = path
|
|
tfVersion := r.DefaultTFVersion.String()
|
|
if ctx.TerraformVersion != nil {
|
|
tfVersion = ctx.TerraformVersion.String()
|
|
}
|
|
baseEnvVars := os.Environ()
|
|
customEnvVars := map[string]string{
|
|
"ATLANTIS_TERRAFORM_VERSION": tfVersion,
|
|
"BASE_BRANCH_NAME": ctx.Pull.BaseBranch,
|
|
"BASE_REPO_NAME": ctx.BaseRepo.Name,
|
|
"BASE_REPO_OWNER": ctx.BaseRepo.Owner,
|
|
"DIR": path,
|
|
"HEAD_BRANCH_NAME": ctx.Pull.HeadBranch,
|
|
"HEAD_REPO_NAME": ctx.HeadRepo.Name,
|
|
"HEAD_REPO_OWNER": ctx.HeadRepo.Owner,
|
|
"PLANFILE": filepath.Join(path, GetPlanFilename(ctx.Workspace, ctx.ProjectName)),
|
|
"PROJECT_NAME": ctx.ProjectName,
|
|
"PULL_AUTHOR": ctx.Pull.Author,
|
|
"PULL_NUM": fmt.Sprintf("%d", ctx.Pull.Num),
|
|
"USER_NAME": ctx.User.Username,
|
|
"WORKSPACE": ctx.Workspace,
|
|
}
|
|
|
|
finalEnvVars := baseEnvVars
|
|
for key, val := range customEnvVars {
|
|
finalEnvVars = append(finalEnvVars, fmt.Sprintf("%s=%s", key, val))
|
|
}
|
|
cmd.Env = finalEnvVars
|
|
out, err := cmd.CombinedOutput()
|
|
|
|
if err != nil {
|
|
err = fmt.Errorf("%s: running %q in %q: \n%s", err, command, path, out)
|
|
ctx.Log.Debug("error: %s", err)
|
|
return "", err
|
|
}
|
|
ctx.Log.Info("successfully ran %q in %q", command, path)
|
|
return string(out), nil
|
|
}
|