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)
	}
}