mirror of
https://git.vectorsigma.ru/public/atlantis.git
synced 2026-07-29 14:28:22 +00:00
95 lines
2.3 KiB
Go
95 lines
2.3 KiB
Go
package server
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"github.com/hootsuite/atlantis/logging"
|
|
"github.com/hootsuite/atlantis/recovery"
|
|
)
|
|
|
|
type CommandHandler struct {
|
|
planExecutor *PlanExecutor
|
|
applyExecutor *ApplyExecutor
|
|
helpExecutor *HelpExecutor
|
|
githubClient *GithubClient
|
|
eventParser *EventParser
|
|
logger *logging.SimpleLogger
|
|
}
|
|
|
|
type CommandType int
|
|
|
|
const (
|
|
Apply CommandType = iota
|
|
Plan
|
|
Help
|
|
// Adding more? Don't forget to update String() below
|
|
)
|
|
|
|
func (c CommandType) String() string {
|
|
switch c {
|
|
case Apply:
|
|
return "apply"
|
|
case Plan:
|
|
return "plan"
|
|
case Help:
|
|
return "help"
|
|
}
|
|
return ""
|
|
}
|
|
|
|
type Command struct {
|
|
verbose bool
|
|
environment string
|
|
commandType CommandType
|
|
}
|
|
|
|
func (c *CommandHandler) ExecuteCommand(ctx *CommandContext) {
|
|
src := fmt.Sprintf("%s/pull/%d", ctx.Repo.FullName, ctx.Pull.Num)
|
|
// it'e safe to reuse the underlying logger e.logger.Log
|
|
ctx.Log = logging.NewSimpleLogger(src, c.logger.Log, true, c.logger.Level)
|
|
defer c.recover(ctx)
|
|
|
|
// need to get additional data from the PR
|
|
ghPull, _, err := c.githubClient.GetPullRequest(ctx.Repo, ctx.Pull.Num)
|
|
if err != nil {
|
|
ctx.Log.Err("pull request data api call failed: %v", err)
|
|
return
|
|
}
|
|
pull, err := c.eventParser.ExtractPullData(ghPull)
|
|
if err != nil {
|
|
ctx.Log.Err("failed to extract required fields from comment data: %v", err)
|
|
return
|
|
}
|
|
ctx.Pull = pull
|
|
|
|
if ghPull.GetState() != "open" {
|
|
ctx.Log.Info("command run on closed pull request")
|
|
c.githubClient.CreateComment(ctx.Repo, ctx.Pull, "Atlantis commands can't be run on closed pull requests")
|
|
return
|
|
}
|
|
|
|
switch ctx.Command.commandType {
|
|
case Plan:
|
|
c.planExecutor.execute(ctx, c.githubClient)
|
|
case Apply:
|
|
c.applyExecutor.execute(ctx, c.githubClient)
|
|
case Help:
|
|
c.helpExecutor.execute(ctx, c.githubClient)
|
|
default:
|
|
ctx.Log.Err("failed to determine desired command, neither plan nor apply")
|
|
}
|
|
}
|
|
|
|
func (c *CommandHandler) SetLockURL(f func(id string) (url string)) {
|
|
c.planExecutor.LockURL = f
|
|
}
|
|
|
|
// recover logs and creates a comment on the pull request for panics
|
|
func (c *CommandHandler) recover(ctx *CommandContext) {
|
|
if err := recover(); err != nil {
|
|
stack := recovery.Stack(3)
|
|
c.githubClient.CreateComment(ctx.Repo, ctx.Pull, fmt.Sprintf("**Error: goroutine panic. This is a bug.**\n```\n%s\n%s```", err, stack))
|
|
ctx.Log.Err("PANIC: %s\n%s", err, stack)
|
|
}
|
|
}
|