mirror of
https://git.vectorsigma.ru/public/atlantis.git
synced 2026-07-29 12:48:18 +00:00
This is two changes: Replacing all usages of SimpleLogger with it's interface SimpleLogging which makes it easier to swap out loggers going forward Nukes SimpleLogger in favor of StructuredLogger which allows us to better analyze logs in our log management system we choose.
796 lines
27 KiB
Go
796 lines
27 KiB
Go
// Copyright 2017 HootSuite Media Inc.
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the License);
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an AS IS BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
// Modified hereafter by contributors to runatlantis/atlantis.
|
|
|
|
// Package server handles the web server and executing commands that come in
|
|
// via webhooks.
|
|
package server
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"flag"
|
|
"fmt"
|
|
"log"
|
|
"net/http"
|
|
"net/url"
|
|
"os"
|
|
"os/signal"
|
|
"path/filepath"
|
|
"sort"
|
|
"strings"
|
|
"syscall"
|
|
"time"
|
|
|
|
"github.com/mitchellh/go-homedir"
|
|
"github.com/runatlantis/atlantis/server/events/db"
|
|
"github.com/runatlantis/atlantis/server/events/yaml/valid"
|
|
|
|
assetfs "github.com/elazarl/go-bindata-assetfs"
|
|
"github.com/gorilla/mux"
|
|
"github.com/pkg/errors"
|
|
"github.com/runatlantis/atlantis/server/events"
|
|
"github.com/runatlantis/atlantis/server/events/locking"
|
|
"github.com/runatlantis/atlantis/server/events/models"
|
|
"github.com/runatlantis/atlantis/server/events/runtime"
|
|
"github.com/runatlantis/atlantis/server/events/runtime/policy"
|
|
"github.com/runatlantis/atlantis/server/events/terraform"
|
|
"github.com/runatlantis/atlantis/server/events/vcs"
|
|
"github.com/runatlantis/atlantis/server/events/vcs/bitbucketcloud"
|
|
"github.com/runatlantis/atlantis/server/events/vcs/bitbucketserver"
|
|
"github.com/runatlantis/atlantis/server/events/webhooks"
|
|
"github.com/runatlantis/atlantis/server/events/yaml"
|
|
"github.com/runatlantis/atlantis/server/logging"
|
|
"github.com/runatlantis/atlantis/server/static"
|
|
"github.com/urfave/cli"
|
|
"github.com/urfave/negroni"
|
|
)
|
|
|
|
const (
|
|
// LockViewRouteName is the named route in mux.Router for the lock view.
|
|
// The route can be retrieved by this name, ex:
|
|
// mux.Router.Get(LockViewRouteName)
|
|
LockViewRouteName = "lock-detail"
|
|
// LockViewRouteIDQueryParam is the query parameter needed to construct the lock view
|
|
// route. ex:
|
|
// mux.Router.Get(LockViewRouteName).URL(LockViewRouteIDQueryParam, "my id")
|
|
LockViewRouteIDQueryParam = "id"
|
|
|
|
// binDirName is the name of the directory inside our data dir where
|
|
// we download binaries.
|
|
BinDirName = "bin"
|
|
|
|
// terraformPluginCacheDir is the name of the dir inside our data dir
|
|
// where we tell terraform to cache plugins and modules.
|
|
TerraformPluginCacheDirName = "plugin-cache"
|
|
)
|
|
|
|
// Server runs the Atlantis web server.
|
|
type Server struct {
|
|
AtlantisVersion string
|
|
AtlantisURL *url.URL
|
|
Router *mux.Router
|
|
Port int
|
|
PreWorkflowHooksCommandRunner *events.DefaultPreWorkflowHooksCommandRunner
|
|
CommandRunner *events.DefaultCommandRunner
|
|
Logger logging.SimpleLogging
|
|
Locker locking.Locker
|
|
ApplyLocker locking.ApplyLocker
|
|
EventsController *EventsController
|
|
GithubAppController *GithubAppController
|
|
LocksController *LocksController
|
|
StatusController *StatusController
|
|
IndexTemplate TemplateWriter
|
|
LockDetailTemplate TemplateWriter
|
|
SSLCertFile string
|
|
SSLKeyFile string
|
|
Drainer *events.Drainer
|
|
}
|
|
|
|
// Config holds config for server that isn't passed in by the user.
|
|
type Config struct {
|
|
AllowForkPRsFlag string
|
|
AtlantisURLFlag string
|
|
AtlantisVersion string
|
|
DefaultTFVersionFlag string
|
|
RepoConfigJSONFlag string
|
|
SilenceForkPRErrorsFlag string
|
|
}
|
|
|
|
// WebhookConfig is nested within UserConfig. It's used to configure webhooks.
|
|
type WebhookConfig struct {
|
|
// Event is the type of event we should send this webhook for, ex. apply.
|
|
Event string `mapstructure:"event"`
|
|
// WorkspaceRegex is a regex that is used to match against the workspace
|
|
// that is being modified for this event. If the regex matches, we'll
|
|
// send the webhook, ex. "production.*".
|
|
WorkspaceRegex string `mapstructure:"workspace-regex"`
|
|
// Kind is the type of webhook we should send, ex. slack.
|
|
Kind string `mapstructure:"kind"`
|
|
// Channel is the channel to send this webhook to. It only applies to
|
|
// slack webhooks. Should be without '#'.
|
|
Channel string `mapstructure:"channel"`
|
|
}
|
|
|
|
// NewServer returns a new server. If there are issues starting the server or
|
|
// its dependencies an error will be returned. This is like the main() function
|
|
// for the server CLI command because it injects all the dependencies.
|
|
func NewServer(userConfig UserConfig, config Config) (*Server, error) {
|
|
logger, err := logging.NewStructuredLoggerFromLevel(userConfig.ToLogLevel())
|
|
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
var supportedVCSHosts []models.VCSHostType
|
|
var githubClient *vcs.GithubClient
|
|
var githubAppEnabled bool
|
|
var githubCredentials vcs.GithubCredentials
|
|
var gitlabClient *vcs.GitlabClient
|
|
var bitbucketCloudClient *bitbucketcloud.Client
|
|
var bitbucketServerClient *bitbucketserver.Client
|
|
var azuredevopsClient *vcs.AzureDevopsClient
|
|
|
|
policyChecksEnabled := false
|
|
if userConfig.EnablePolicyChecksFlag {
|
|
logger.Info("Policy Checks are enabled")
|
|
policyChecksEnabled = true
|
|
}
|
|
|
|
if userConfig.GithubUser != "" || userConfig.GithubAppID != 0 {
|
|
supportedVCSHosts = append(supportedVCSHosts, models.Github)
|
|
if userConfig.GithubUser != "" {
|
|
githubCredentials = &vcs.GithubUserCredentials{
|
|
User: userConfig.GithubUser,
|
|
Token: userConfig.GithubToken,
|
|
}
|
|
} else if userConfig.GithubAppID != 0 {
|
|
githubCredentials = &vcs.GithubAppCredentials{
|
|
AppID: userConfig.GithubAppID,
|
|
KeyPath: userConfig.GithubAppKey,
|
|
Hostname: userConfig.GithubHostname,
|
|
AppSlug: userConfig.GithubAppSlug,
|
|
}
|
|
githubAppEnabled = true
|
|
}
|
|
|
|
var err error
|
|
githubClient, err = vcs.NewGithubClient(userConfig.GithubHostname, githubCredentials, logger)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
}
|
|
if userConfig.GitlabUser != "" {
|
|
supportedVCSHosts = append(supportedVCSHosts, models.Gitlab)
|
|
var err error
|
|
gitlabClient, err = vcs.NewGitlabClient(userConfig.GitlabHostname, userConfig.GitlabToken, logger)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
}
|
|
if userConfig.BitbucketUser != "" {
|
|
if userConfig.BitbucketBaseURL == bitbucketcloud.BaseURL {
|
|
supportedVCSHosts = append(supportedVCSHosts, models.BitbucketCloud)
|
|
bitbucketCloudClient = bitbucketcloud.NewClient(
|
|
http.DefaultClient,
|
|
userConfig.BitbucketUser,
|
|
userConfig.BitbucketToken,
|
|
userConfig.AtlantisURL)
|
|
} else {
|
|
supportedVCSHosts = append(supportedVCSHosts, models.BitbucketServer)
|
|
var err error
|
|
bitbucketServerClient, err = bitbucketserver.NewClient(
|
|
http.DefaultClient,
|
|
userConfig.BitbucketUser,
|
|
userConfig.BitbucketToken,
|
|
userConfig.BitbucketBaseURL,
|
|
userConfig.AtlantisURL)
|
|
if err != nil {
|
|
return nil, errors.Wrapf(err, "setting up Bitbucket Server client")
|
|
}
|
|
}
|
|
}
|
|
if userConfig.AzureDevopsUser != "" {
|
|
supportedVCSHosts = append(supportedVCSHosts, models.AzureDevops)
|
|
var err error
|
|
azuredevopsClient, err = vcs.NewAzureDevopsClient("dev.azure.com", userConfig.AzureDevopsUser, userConfig.AzureDevopsToken)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
}
|
|
|
|
if userConfig.WriteGitCreds {
|
|
home, err := homedir.Dir()
|
|
if err != nil {
|
|
return nil, errors.Wrap(err, "getting home dir to write ~/.git-credentials file")
|
|
}
|
|
if userConfig.GithubUser != "" {
|
|
if err := events.WriteGitCreds(userConfig.GithubUser, userConfig.GithubToken, userConfig.GithubHostname, home, logger, false); err != nil {
|
|
return nil, err
|
|
}
|
|
}
|
|
if userConfig.GitlabUser != "" {
|
|
if err := events.WriteGitCreds(userConfig.GitlabUser, userConfig.GitlabToken, userConfig.GitlabHostname, home, logger, false); err != nil {
|
|
return nil, err
|
|
}
|
|
}
|
|
if userConfig.BitbucketUser != "" {
|
|
// The default BitbucketBaseURL is https://api.bitbucket.org which can't actually be used for git
|
|
// so we override it here only if it's that to be bitbucket.org
|
|
bitbucketBaseURL := userConfig.BitbucketBaseURL
|
|
if bitbucketBaseURL == "https://api.bitbucket.org" {
|
|
bitbucketBaseURL = "bitbucket.org"
|
|
}
|
|
if err := events.WriteGitCreds(userConfig.BitbucketUser, userConfig.BitbucketToken, bitbucketBaseURL, home, logger, false); err != nil {
|
|
return nil, err
|
|
}
|
|
}
|
|
if userConfig.AzureDevopsUser != "" {
|
|
if err := events.WriteGitCreds(userConfig.AzureDevopsUser, userConfig.AzureDevopsToken, "dev.azure.com", home, logger, false); err != nil {
|
|
return nil, err
|
|
}
|
|
}
|
|
}
|
|
|
|
var webhooksConfig []webhooks.Config
|
|
for _, c := range userConfig.Webhooks {
|
|
config := webhooks.Config{
|
|
Channel: c.Channel,
|
|
Event: c.Event,
|
|
Kind: c.Kind,
|
|
WorkspaceRegex: c.WorkspaceRegex,
|
|
}
|
|
webhooksConfig = append(webhooksConfig, config)
|
|
}
|
|
webhooksManager, err := webhooks.NewMultiWebhookSender(webhooksConfig, webhooks.NewSlackClient(userConfig.SlackToken))
|
|
if err != nil {
|
|
return nil, errors.Wrap(err, "initializing webhooks")
|
|
}
|
|
vcsClient := vcs.NewClientProxy(githubClient, gitlabClient, bitbucketCloudClient, bitbucketServerClient, azuredevopsClient)
|
|
commitStatusUpdater := &events.DefaultCommitStatusUpdater{Client: vcsClient, StatusName: userConfig.VCSStatusName}
|
|
|
|
binDir, err := mkSubDir(userConfig.DataDir, BinDirName)
|
|
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
cacheDir, err := mkSubDir(userConfig.DataDir, TerraformPluginCacheDirName)
|
|
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
terraformClient, err := terraform.NewClient(
|
|
logger,
|
|
binDir,
|
|
cacheDir,
|
|
userConfig.TFEToken,
|
|
userConfig.TFEHostname,
|
|
userConfig.DefaultTFVersion,
|
|
config.DefaultTFVersionFlag,
|
|
userConfig.TFDownloadURL,
|
|
&terraform.DefaultDownloader{},
|
|
true)
|
|
// The flag.Lookup call is to detect if we're running in a unit test. If we
|
|
// are, then we don't error out because we don't have/want terraform
|
|
// installed on our CI system where the unit tests run.
|
|
if err != nil && flag.Lookup("test.v") == nil {
|
|
return nil, errors.Wrap(err, "initializing terraform")
|
|
}
|
|
markdownRenderer := &events.MarkdownRenderer{
|
|
GitlabSupportsCommonMark: gitlabClient.SupportsCommonMark(),
|
|
DisableApplyAll: userConfig.DisableApplyAll,
|
|
DisableMarkdownFolding: userConfig.DisableMarkdownFolding,
|
|
DisableApply: userConfig.DisableApply,
|
|
DisableRepoLocking: userConfig.DisableRepoLocking,
|
|
}
|
|
|
|
boltdb, err := db.New(userConfig.DataDir)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
var lockingClient locking.Locker
|
|
var applyLockingClient locking.ApplyLocker
|
|
if userConfig.DisableRepoLocking {
|
|
lockingClient = locking.NewNoOpLocker()
|
|
} else {
|
|
lockingClient = locking.NewClient(boltdb)
|
|
applyLockingClient = locking.NewApplyClient(boltdb, userConfig.DisableApply)
|
|
}
|
|
workingDirLocker := events.NewDefaultWorkingDirLocker()
|
|
|
|
var workingDir events.WorkingDir = &events.FileWorkspace{
|
|
DataDir: userConfig.DataDir,
|
|
CheckoutMerge: userConfig.CheckoutStrategy == "merge",
|
|
}
|
|
// provide fresh tokens before clone from the GitHub Apps integration, proxy workingDir
|
|
if githubAppEnabled {
|
|
if !userConfig.WriteGitCreds {
|
|
return nil, errors.New("Github App requires --write-git-creds to support cloning")
|
|
}
|
|
workingDir = &events.GithubAppWorkingDir{
|
|
WorkingDir: workingDir,
|
|
Credentials: githubCredentials,
|
|
GithubHostname: userConfig.GithubHostname,
|
|
}
|
|
}
|
|
|
|
projectLocker := &events.DefaultProjectLocker{
|
|
Locker: lockingClient,
|
|
VCSClient: vcsClient,
|
|
}
|
|
deleteLockCommand := &events.DefaultDeleteLockCommand{
|
|
Locker: lockingClient,
|
|
Logger: logger,
|
|
WorkingDir: workingDir,
|
|
WorkingDirLocker: workingDirLocker,
|
|
DB: boltdb,
|
|
}
|
|
|
|
parsedURL, err := ParseAtlantisURL(userConfig.AtlantisURL)
|
|
if err != nil {
|
|
return nil, errors.Wrapf(err,
|
|
"parsing --%s flag %q", config.AtlantisURLFlag, userConfig.AtlantisURL)
|
|
}
|
|
validator := &yaml.ParserValidator{}
|
|
|
|
globalCfg := valid.NewGlobalCfg(userConfig.AllowRepoConfig, userConfig.RequireMergeable, userConfig.RequireApproval)
|
|
if userConfig.RepoConfig != "" {
|
|
globalCfg, err = validator.ParseGlobalCfg(userConfig.RepoConfig, globalCfg)
|
|
if err != nil {
|
|
return nil, errors.Wrapf(err, "parsing %s file", userConfig.RepoConfig)
|
|
}
|
|
} else if userConfig.RepoConfigJSON != "" {
|
|
globalCfg, err = validator.ParseGlobalCfgJSON(userConfig.RepoConfigJSON, globalCfg)
|
|
if err != nil {
|
|
return nil, errors.Wrapf(err, "parsing --%s", config.RepoConfigJSONFlag)
|
|
}
|
|
}
|
|
|
|
underlyingRouter := mux.NewRouter()
|
|
router := &Router{
|
|
AtlantisURL: parsedURL,
|
|
LockViewRouteIDQueryParam: LockViewRouteIDQueryParam,
|
|
LockViewRouteName: LockViewRouteName,
|
|
Underlying: underlyingRouter,
|
|
}
|
|
pullClosedExecutor := &events.PullClosedExecutor{
|
|
VCSClient: vcsClient,
|
|
Locker: lockingClient,
|
|
WorkingDir: workingDir,
|
|
Logger: logger,
|
|
DB: boltdb,
|
|
}
|
|
eventParser := &events.EventParser{
|
|
GithubUser: userConfig.GithubUser,
|
|
GithubToken: userConfig.GithubToken,
|
|
GitlabUser: userConfig.GitlabUser,
|
|
GitlabToken: userConfig.GitlabToken,
|
|
AllowDraftPRs: userConfig.PlanDrafts,
|
|
BitbucketUser: userConfig.BitbucketUser,
|
|
BitbucketToken: userConfig.BitbucketToken,
|
|
BitbucketServerURL: userConfig.BitbucketBaseURL,
|
|
AzureDevopsUser: userConfig.AzureDevopsUser,
|
|
AzureDevopsToken: userConfig.AzureDevopsToken,
|
|
}
|
|
commentParser := &events.CommentParser{
|
|
GithubUser: userConfig.GithubUser,
|
|
GitlabUser: userConfig.GitlabUser,
|
|
BitbucketUser: userConfig.BitbucketUser,
|
|
AzureDevopsUser: userConfig.AzureDevopsUser,
|
|
ApplyDisabled: userConfig.DisableApply,
|
|
}
|
|
defaultTfVersion := terraformClient.DefaultVersion()
|
|
pendingPlanFinder := &events.DefaultPendingPlanFinder{}
|
|
runStepRunner := &runtime.RunStepRunner{
|
|
TerraformExecutor: terraformClient,
|
|
DefaultTFVersion: defaultTfVersion,
|
|
TerraformBinDir: terraformClient.TerraformBinDir(),
|
|
}
|
|
drainer := &events.Drainer{}
|
|
statusController := &StatusController{
|
|
Logger: logger,
|
|
Drainer: drainer,
|
|
}
|
|
preWorkflowHooksCommandRunner := &events.DefaultPreWorkflowHooksCommandRunner{
|
|
VCSClient: vcsClient,
|
|
GlobalCfg: globalCfg,
|
|
WorkingDirLocker: workingDirLocker,
|
|
WorkingDir: workingDir,
|
|
PreWorkflowHookRunner: runtime.DefaultPreWorkflowHookRunner{},
|
|
}
|
|
projectCommandBuilder := events.NewProjectCommandBuilder(
|
|
policyChecksEnabled,
|
|
validator,
|
|
&events.DefaultProjectFinder{},
|
|
vcsClient,
|
|
workingDir,
|
|
workingDirLocker,
|
|
globalCfg,
|
|
pendingPlanFinder,
|
|
commentParser,
|
|
userConfig.SkipCloneNoChanges,
|
|
userConfig.EnableRegExpCmd,
|
|
)
|
|
|
|
showStepRunner, err := runtime.NewShowStepRunner(terraformClient, defaultTfVersion)
|
|
|
|
if err != nil {
|
|
return nil, errors.Wrap(err, "initializing show step runner")
|
|
}
|
|
|
|
policyCheckRunner, err := runtime.NewPolicyCheckStepRunner(
|
|
defaultTfVersion,
|
|
policy.NewConfTestExecutorWorkflow(logger, binDir, &terraform.DefaultDownloader{}),
|
|
)
|
|
|
|
if err != nil {
|
|
return nil, errors.Wrap(err, "initializing policy check runner")
|
|
}
|
|
|
|
projectCommandRunner := &events.DefaultProjectCommandRunner{
|
|
Locker: projectLocker,
|
|
LockURLGenerator: router,
|
|
InitStepRunner: &runtime.InitStepRunner{
|
|
TerraformExecutor: terraformClient,
|
|
DefaultTFVersion: defaultTfVersion,
|
|
},
|
|
PlanStepRunner: &runtime.PlanStepRunner{
|
|
TerraformExecutor: terraformClient,
|
|
DefaultTFVersion: defaultTfVersion,
|
|
CommitStatusUpdater: commitStatusUpdater,
|
|
AsyncTFExec: terraformClient,
|
|
},
|
|
ShowStepRunner: showStepRunner,
|
|
PolicyCheckStepRunner: policyCheckRunner,
|
|
ApplyStepRunner: &runtime.ApplyStepRunner{
|
|
TerraformExecutor: terraformClient,
|
|
CommitStatusUpdater: commitStatusUpdater,
|
|
AsyncTFExec: terraformClient,
|
|
},
|
|
RunStepRunner: runStepRunner,
|
|
EnvStepRunner: &runtime.EnvStepRunner{
|
|
RunStepRunner: runStepRunner,
|
|
},
|
|
PullApprovedChecker: vcsClient,
|
|
WorkingDir: workingDir,
|
|
Webhooks: webhooksManager,
|
|
WorkingDirLocker: workingDirLocker,
|
|
}
|
|
|
|
dbUpdater := &events.DBUpdater{
|
|
DB: boltdb,
|
|
}
|
|
|
|
pullUpdater := &events.PullUpdater{
|
|
HidePrevPlanComments: userConfig.HidePrevPlanComments,
|
|
VCSClient: vcsClient,
|
|
MarkdownRenderer: markdownRenderer,
|
|
}
|
|
|
|
autoMerger := &events.AutoMerger{
|
|
VCSClient: vcsClient,
|
|
GlobalAutomerge: userConfig.Automerge,
|
|
}
|
|
|
|
policyCheckCommandRunner := events.NewPolicyCheckCommandRunner(
|
|
dbUpdater,
|
|
pullUpdater,
|
|
commitStatusUpdater,
|
|
projectCommandRunner,
|
|
userConfig.ParallelPoolSize,
|
|
)
|
|
|
|
planCommandRunner := events.NewPlanCommandRunner(
|
|
userConfig.SilenceVCSStatusNoPlans,
|
|
vcsClient,
|
|
pendingPlanFinder,
|
|
workingDir,
|
|
commitStatusUpdater,
|
|
projectCommandBuilder,
|
|
projectCommandRunner,
|
|
dbUpdater,
|
|
pullUpdater,
|
|
policyCheckCommandRunner,
|
|
autoMerger,
|
|
userConfig.ParallelPoolSize,
|
|
)
|
|
|
|
applyCommandRunner := events.NewApplyCommandRunner(
|
|
vcsClient,
|
|
userConfig.DisableApplyAll,
|
|
applyLockingClient,
|
|
commitStatusUpdater,
|
|
projectCommandBuilder,
|
|
projectCommandRunner,
|
|
autoMerger,
|
|
pullUpdater,
|
|
dbUpdater,
|
|
boltdb,
|
|
userConfig.ParallelPoolSize,
|
|
)
|
|
|
|
approvePoliciesCommandRunner := events.NewApprovePoliciesCommandRunner(
|
|
commitStatusUpdater,
|
|
projectCommandBuilder,
|
|
projectCommandRunner,
|
|
pullUpdater,
|
|
dbUpdater,
|
|
)
|
|
|
|
unlockCommandRunner := events.NewUnlockCommandRunner(
|
|
deleteLockCommand,
|
|
vcsClient,
|
|
)
|
|
|
|
commentCommandRunnerByCmd := map[models.CommandName]events.CommentCommandRunner{
|
|
models.PlanCommand: planCommandRunner,
|
|
models.ApplyCommand: applyCommandRunner,
|
|
models.ApprovePoliciesCommand: approvePoliciesCommandRunner,
|
|
models.UnlockCommand: unlockCommandRunner,
|
|
}
|
|
|
|
commandRunner := &events.DefaultCommandRunner{
|
|
VCSClient: vcsClient,
|
|
GithubPullGetter: githubClient,
|
|
GitlabMergeRequestGetter: gitlabClient,
|
|
AzureDevopsPullGetter: azuredevopsClient,
|
|
CommentCommandRunnerByCmd: commentCommandRunnerByCmd,
|
|
EventParser: eventParser,
|
|
Logger: logger,
|
|
AllowForkPRs: userConfig.AllowForkPRs,
|
|
AllowForkPRsFlag: config.AllowForkPRsFlag,
|
|
SilenceForkPRErrors: userConfig.SilenceForkPRErrors,
|
|
SilenceForkPRErrorsFlag: config.SilenceForkPRErrorsFlag,
|
|
DisableAutoplan: userConfig.DisableAutoplan,
|
|
Drainer: drainer,
|
|
PreWorkflowHooksCommandRunner: preWorkflowHooksCommandRunner,
|
|
}
|
|
repoAllowlist, err := events.NewRepoAllowlistChecker(userConfig.RepoAllowlist)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
locksController := &LocksController{
|
|
AtlantisVersion: config.AtlantisVersion,
|
|
AtlantisURL: parsedURL,
|
|
Locker: lockingClient,
|
|
ApplyLocker: applyLockingClient,
|
|
Logger: logger,
|
|
VCSClient: vcsClient,
|
|
LockDetailTemplate: lockTemplate,
|
|
WorkingDir: workingDir,
|
|
WorkingDirLocker: workingDirLocker,
|
|
DB: boltdb,
|
|
DeleteLockCommand: deleteLockCommand,
|
|
}
|
|
eventsController := &EventsController{
|
|
CommandRunner: commandRunner,
|
|
PullCleaner: pullClosedExecutor,
|
|
Parser: eventParser,
|
|
CommentParser: commentParser,
|
|
Logger: logger,
|
|
ApplyDisabled: userConfig.DisableApply,
|
|
GithubWebhookSecret: []byte(userConfig.GithubWebhookSecret),
|
|
GithubRequestValidator: &DefaultGithubRequestValidator{},
|
|
GitlabRequestParserValidator: &DefaultGitlabRequestParserValidator{},
|
|
GitlabWebhookSecret: []byte(userConfig.GitlabWebhookSecret),
|
|
RepoAllowlistChecker: repoAllowlist,
|
|
SilenceAllowlistErrors: userConfig.SilenceAllowlistErrors,
|
|
SupportedVCSHosts: supportedVCSHosts,
|
|
VCSClient: vcsClient,
|
|
BitbucketWebhookSecret: []byte(userConfig.BitbucketWebhookSecret),
|
|
AzureDevopsWebhookBasicUser: []byte(userConfig.AzureDevopsWebhookUser),
|
|
AzureDevopsWebhookBasicPassword: []byte(userConfig.AzureDevopsWebhookPassword),
|
|
AzureDevopsRequestValidator: &DefaultAzureDevopsRequestValidator{},
|
|
}
|
|
githubAppController := &GithubAppController{
|
|
AtlantisURL: parsedURL,
|
|
Logger: logger,
|
|
GithubSetupComplete: githubAppEnabled,
|
|
GithubHostname: userConfig.GithubHostname,
|
|
GithubOrg: userConfig.GithubOrg,
|
|
}
|
|
|
|
return &Server{
|
|
AtlantisVersion: config.AtlantisVersion,
|
|
AtlantisURL: parsedURL,
|
|
Router: underlyingRouter,
|
|
Port: userConfig.Port,
|
|
PreWorkflowHooksCommandRunner: preWorkflowHooksCommandRunner,
|
|
CommandRunner: commandRunner,
|
|
Logger: logger,
|
|
Locker: lockingClient,
|
|
ApplyLocker: applyLockingClient,
|
|
EventsController: eventsController,
|
|
GithubAppController: githubAppController,
|
|
LocksController: locksController,
|
|
StatusController: statusController,
|
|
IndexTemplate: indexTemplate,
|
|
LockDetailTemplate: lockTemplate,
|
|
SSLKeyFile: userConfig.SSLKeyFile,
|
|
SSLCertFile: userConfig.SSLCertFile,
|
|
Drainer: drainer,
|
|
}, nil
|
|
}
|
|
|
|
// Start creates the routes and starts serving traffic.
|
|
func (s *Server) Start() error {
|
|
s.Router.HandleFunc("/", s.Index).Methods("GET").MatcherFunc(func(r *http.Request, rm *mux.RouteMatch) bool {
|
|
return r.URL.Path == "/" || r.URL.Path == "/index.html"
|
|
})
|
|
s.Router.HandleFunc("/healthz", s.Healthz).Methods("GET")
|
|
s.Router.HandleFunc("/status", s.StatusController.Get).Methods("GET")
|
|
s.Router.PathPrefix("/static/").Handler(http.FileServer(&assetfs.AssetFS{Asset: static.Asset, AssetDir: static.AssetDir, AssetInfo: static.AssetInfo}))
|
|
s.Router.HandleFunc("/events", s.EventsController.Post).Methods("POST")
|
|
s.Router.HandleFunc("/github-app/exchange-code", s.GithubAppController.ExchangeCode).Methods("GET")
|
|
s.Router.HandleFunc("/github-app/setup", s.GithubAppController.New).Methods("GET")
|
|
s.Router.HandleFunc("/apply/lock", s.LocksController.LockApply).Methods("POST").Queries()
|
|
s.Router.HandleFunc("/apply/unlock", s.LocksController.UnlockApply).Methods("DELETE").Queries()
|
|
s.Router.HandleFunc("/locks", s.LocksController.DeleteLock).Methods("DELETE").Queries("id", "{id:.*}")
|
|
s.Router.HandleFunc("/lock", s.LocksController.GetLock).Methods("GET").
|
|
Queries(LockViewRouteIDQueryParam, fmt.Sprintf("{%s}", LockViewRouteIDQueryParam)).Name(LockViewRouteName)
|
|
n := negroni.New(&negroni.Recovery{
|
|
Logger: log.New(os.Stdout, "", log.LstdFlags),
|
|
PrintStack: false,
|
|
StackAll: false,
|
|
StackSize: 1024 * 8,
|
|
}, NewRequestLogger(s.Logger))
|
|
n.UseHandler(s.Router)
|
|
|
|
defer s.Logger.Flush()
|
|
|
|
// Ensure server gracefully drains connections when stopped.
|
|
stop := make(chan os.Signal, 1)
|
|
// Stop on SIGINTs and SIGTERMs.
|
|
signal.Notify(stop, os.Interrupt, syscall.SIGTERM)
|
|
|
|
server := &http.Server{Addr: fmt.Sprintf(":%d", s.Port), Handler: n}
|
|
go func() {
|
|
s.Logger.Info("Atlantis started - listening on port %v", s.Port)
|
|
|
|
var err error
|
|
if s.SSLCertFile != "" && s.SSLKeyFile != "" {
|
|
err = server.ListenAndServeTLS(s.SSLCertFile, s.SSLKeyFile)
|
|
} else {
|
|
err = server.ListenAndServe()
|
|
}
|
|
|
|
if err != nil && err != http.ErrServerClosed {
|
|
s.Logger.Err(err.Error())
|
|
}
|
|
}()
|
|
<-stop
|
|
|
|
s.Logger.Warn("Received interrupt. Waiting for in-progress operations to complete")
|
|
s.waitForDrain()
|
|
ctx, _ := context.WithTimeout(context.Background(), 5*time.Second) // nolint: vet
|
|
if err := server.Shutdown(ctx); err != nil {
|
|
return cli.NewExitError(fmt.Sprintf("while shutting down: %s", err), 1)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// waitForDrain blocks until draining is complete.
|
|
func (s *Server) waitForDrain() {
|
|
drainComplete := make(chan bool, 1)
|
|
go func() {
|
|
s.Drainer.ShutdownBlocking()
|
|
drainComplete <- true
|
|
}()
|
|
ticker := time.NewTicker(5 * time.Second)
|
|
for {
|
|
select {
|
|
case <-drainComplete:
|
|
s.Logger.Info("All in-progress operations complete, shutting down")
|
|
return
|
|
case <-ticker.C:
|
|
s.Logger.Info("Waiting for in-progress operations to complete, current in-progress ops: %d", s.Drainer.GetStatus().InProgressOps)
|
|
}
|
|
}
|
|
}
|
|
|
|
// Index is the / route.
|
|
func (s *Server) Index(w http.ResponseWriter, _ *http.Request) {
|
|
locks, err := s.Locker.List()
|
|
if err != nil {
|
|
w.WriteHeader(http.StatusServiceUnavailable)
|
|
fmt.Fprintf(w, "Could not retrieve locks: %s", err)
|
|
return
|
|
}
|
|
|
|
var lockResults []LockIndexData
|
|
for id, v := range locks {
|
|
lockURL, _ := s.Router.Get(LockViewRouteName).URL("id", url.QueryEscape(id))
|
|
lockResults = append(lockResults, LockIndexData{
|
|
// NOTE: must use .String() instead of .Path because we need the
|
|
// query params as part of the lock URL.
|
|
LockPath: lockURL.String(),
|
|
RepoFullName: v.Project.RepoFullName,
|
|
PullNum: v.Pull.Num,
|
|
Path: v.Project.Path,
|
|
Workspace: v.Workspace,
|
|
Time: v.Time,
|
|
TimeFormatted: v.Time.Format("02-01-2006 15:04:05"),
|
|
})
|
|
}
|
|
|
|
applyCmdLock, err := s.ApplyLocker.CheckApplyLock()
|
|
s.Logger.Info("Apply Lock: %v", applyCmdLock)
|
|
if err != nil {
|
|
w.WriteHeader(http.StatusServiceUnavailable)
|
|
fmt.Fprintf(w, "Could not retrieve global apply lock: %s", err)
|
|
return
|
|
}
|
|
|
|
applyLockData := ApplyLockData{
|
|
Time: applyCmdLock.Time,
|
|
Locked: applyCmdLock.Locked,
|
|
TimeFormatted: applyCmdLock.Time.Format("02-01-2006 15:04:05"),
|
|
}
|
|
//Sort by date - newest to oldest.
|
|
sort.SliceStable(lockResults, func(i, j int) bool { return lockResults[i].Time.After(lockResults[j].Time) })
|
|
|
|
err = s.IndexTemplate.Execute(w, IndexData{
|
|
Locks: lockResults,
|
|
ApplyLock: applyLockData,
|
|
AtlantisVersion: s.AtlantisVersion,
|
|
CleanedBasePath: s.AtlantisURL.Path,
|
|
})
|
|
if err != nil {
|
|
s.Logger.Err(err.Error())
|
|
}
|
|
}
|
|
|
|
func mkSubDir(parentDir string, subDir string) (string, error) {
|
|
fullDir := filepath.Join(parentDir, subDir)
|
|
if err := os.MkdirAll(fullDir, 0700); err != nil {
|
|
return "", errors.Wrapf(err, "unable to creare dir %q", fullDir)
|
|
}
|
|
|
|
return fullDir, nil
|
|
}
|
|
|
|
// Healthz returns the health check response. It always returns a 200 currently.
|
|
func (s *Server) Healthz(w http.ResponseWriter, _ *http.Request) {
|
|
data, err := json.MarshalIndent(&struct {
|
|
Status string `json:"status"`
|
|
}{
|
|
Status: "ok",
|
|
}, "", " ")
|
|
if err != nil {
|
|
w.WriteHeader(http.StatusInternalServerError)
|
|
fmt.Fprintf(w, "Error creating status json response: %s", err)
|
|
return
|
|
}
|
|
w.Header().Set("Content-Type", "application/json")
|
|
w.Write(data) // nolint: errcheck
|
|
}
|
|
|
|
// ParseAtlantisURL parses the user-passed atlantis URL to ensure it is valid
|
|
// and we can use it in our templates.
|
|
// It removes any trailing slashes from the path so we can concatenate it
|
|
// with other paths without checking.
|
|
func ParseAtlantisURL(u string) (*url.URL, error) {
|
|
parsed, err := url.Parse(u)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if !(parsed.Scheme == "http" || parsed.Scheme == "https") {
|
|
return nil, errors.New("http or https must be specified")
|
|
}
|
|
// We want the path to end without a trailing slash so we know how to
|
|
// use it in the rest of the program.
|
|
parsed.Path = strings.TrimSuffix(parsed.Path, "/")
|
|
return parsed, nil
|
|
}
|