mirror of
https://github.com/containrrr/watchtower.git
synced 2025-09-22 05:40:50 +02:00
fix merge conflicts and do some refactoring
This commit is contained in:
Simon Aronsson
commit
a2158c2bdc
5 changed files with 239 additions and 180 deletions
239
main.go
239
main.go
|
|
@ -8,12 +8,13 @@ import (
|
|||
|
||||
"strconv"
|
||||
|
||||
"github.com/containrrr/watchtower/actions"
|
||||
cliApp "github.com/containrrr/watchtower/app"
|
||||
"github.com/containrrr/watchtower/container"
|
||||
"github.com/containrrr/watchtower/notifications"
|
||||
"github.com/robfig/cron"
|
||||
log "github.com/sirupsen/logrus"
|
||||
"github.com/urfave/cli"
|
||||
"github.com/containrrr/watchtower/actions"
|
||||
"github.com/containrrr/watchtower/container"
|
||||
"github.com/containrrr/watchtower/notifications"
|
||||
)
|
||||
|
||||
// DockerAPIMinVersion is the version of the docker API, which is minimally required by
|
||||
|
|
@ -32,7 +33,7 @@ var (
|
|||
monitorOnly bool
|
||||
enableLabel bool
|
||||
notifier *notifications.Notifier
|
||||
timeout time.Duration
|
||||
timeout time.Duration
|
||||
)
|
||||
|
||||
func init() {
|
||||
|
|
@ -41,162 +42,21 @@ func init() {
|
|||
|
||||
func main() {
|
||||
app := cli.NewApp()
|
||||
InitApp(app)
|
||||
cliApp.SetupCliFlags(app)
|
||||
|
||||
if err := app.Run(os.Args); err != nil {
|
||||
log.Fatal(err)
|
||||
}
|
||||
}
|
||||
|
||||
// InitApp initializes urfave app metadata and sets up entrypoints
|
||||
func InitApp(app *cli.App) {
|
||||
app.Name = "watchtower"
|
||||
app.Version = version + " - " + commit + " - " + date
|
||||
app.Usage = "Automatically update running Docker containers"
|
||||
app.Before = before
|
||||
app.Action = start
|
||||
app.Flags = []cli.Flag{
|
||||
cli.StringFlag{
|
||||
Name: "host, H",
|
||||
Usage: "daemon socket to connect to",
|
||||
Value: "unix:///var/run/docker.sock",
|
||||
EnvVar: "DOCKER_HOST",
|
||||
},
|
||||
cli.IntFlag{
|
||||
Name: "interval, i",
|
||||
Usage: "poll interval (in seconds)",
|
||||
Value: 300,
|
||||
EnvVar: "WATCHTOWER_POLL_INTERVAL",
|
||||
},
|
||||
cli.StringFlag{
|
||||
Name: "schedule, s",
|
||||
Usage: "the cron expression which defines when to update",
|
||||
EnvVar: "WATCHTOWER_SCHEDULE",
|
||||
},
|
||||
cli.BoolFlag{
|
||||
Name: "no-pull",
|
||||
Usage: "do not pull new images",
|
||||
EnvVar: "WATCHTOWER_NO_PULL",
|
||||
},
|
||||
cli.BoolFlag{
|
||||
Name: "no-restart",
|
||||
Usage: "do not restart containers",
|
||||
EnvVar: "WATCHTOWER_NO_RESTART",
|
||||
},
|
||||
cli.BoolFlag{
|
||||
Name: "cleanup",
|
||||
Usage: "remove old images after updating",
|
||||
EnvVar: "WATCHTOWER_CLEANUP",
|
||||
},
|
||||
cli.BoolFlag{
|
||||
Name: "tlsverify",
|
||||
Usage: "use TLS and verify the remote",
|
||||
EnvVar: "DOCKER_TLS_VERIFY",
|
||||
},
|
||||
cli.DurationFlag{
|
||||
Name: "stop-timeout",
|
||||
Usage: "timeout before container is forcefully stopped",
|
||||
Value: time.Second * 10,
|
||||
EnvVar: "WATCHTOWER_TIMEOUT",
|
||||
},
|
||||
cli.BoolFlag{
|
||||
Name: "label-enable",
|
||||
Usage: "watch containers where the com.centurylinklabs.watchtower.enable label is true",
|
||||
EnvVar: "WATCHTOWER_LABEL_ENABLE",
|
||||
},
|
||||
cli.BoolFlag{
|
||||
Name: "debug",
|
||||
Usage: "enable debug mode with verbose logging",
|
||||
},
|
||||
cli.StringSliceFlag{
|
||||
Name: "notifications",
|
||||
Value: &cli.StringSlice{},
|
||||
Usage: "notification types to send (valid: email, slack, msteams)",
|
||||
EnvVar: "WATCHTOWER_NOTIFICATIONS",
|
||||
},
|
||||
cli.StringFlag{
|
||||
Name: "notifications-level",
|
||||
Usage: "The log level used for sending notifications. Possible values: \"panic\", \"fatal\", \"error\", \"warn\", \"info\" or \"debug\"",
|
||||
EnvVar: "WATCHTOWER_NOTIFICATIONS_LEVEL",
|
||||
Value: "info",
|
||||
},
|
||||
cli.StringFlag{
|
||||
Name: "notification-email-from",
|
||||
Usage: "Address to send notification e-mails from",
|
||||
EnvVar: "WATCHTOWER_NOTIFICATION_EMAIL_FROM",
|
||||
},
|
||||
cli.StringFlag{
|
||||
Name: "notification-email-to",
|
||||
Usage: "Address to send notification e-mails to",
|
||||
EnvVar: "WATCHTOWER_NOTIFICATION_EMAIL_TO",
|
||||
},
|
||||
cli.StringFlag{
|
||||
Name: "notification-email-server",
|
||||
Usage: "SMTP server to send notification e-mails through",
|
||||
EnvVar: "WATCHTOWER_NOTIFICATION_EMAIL_SERVER",
|
||||
},
|
||||
cli.IntFlag{
|
||||
Name: "notification-email-server-port",
|
||||
Usage: "SMTP server port to send notification e-mails through",
|
||||
Value: 25,
|
||||
EnvVar: "WATCHTOWER_NOTIFICATION_EMAIL_SERVER_PORT",
|
||||
},
|
||||
cli.BoolFlag{
|
||||
Name: "notification-email-server-tls-skip-verify",
|
||||
Usage: "Controls whether watchtower verifies the SMTP server's certificate chain and host name. " +
|
||||
"If set, TLS accepts any certificate " +
|
||||
"presented by the server and any host name in that certificate. " +
|
||||
"In this mode, TLS is susceptible to man-in-the-middle attacks. " +
|
||||
"This should be used only for testing.",
|
||||
EnvVar: "WATCHTOWER_NOTIFICATION_EMAIL_SERVER_TLS_SKIP_VERIFY",
|
||||
},
|
||||
cli.StringFlag{
|
||||
Name: "notification-email-server-user",
|
||||
Usage: "SMTP server user for sending notifications",
|
||||
EnvVar: "WATCHTOWER_NOTIFICATION_EMAIL_SERVER_USER",
|
||||
},
|
||||
cli.StringFlag{
|
||||
Name: "notification-email-server-password",
|
||||
Usage: "SMTP server password for sending notifications",
|
||||
EnvVar: "WATCHTOWER_NOTIFICATION_EMAIL_SERVER_PASSWORD",
|
||||
},
|
||||
cli.StringFlag{
|
||||
Name: "notification-slack-hook-url",
|
||||
Usage: "The Slack Hook URL to send notifications to",
|
||||
EnvVar: "WATCHTOWER_NOTIFICATION_SLACK_HOOK_URL",
|
||||
},
|
||||
cli.StringFlag{
|
||||
Name: "notification-slack-identifier",
|
||||
Usage: "A string which will be used to identify the messages coming from this watchtower instance. Default if omitted is \"watchtower\"",
|
||||
EnvVar: "WATCHTOWER_NOTIFICATION_SLACK_IDENTIFIER",
|
||||
Value: "watchtower",
|
||||
},
|
||||
cli.StringFlag{
|
||||
Name: "notification-slack-channel",
|
||||
Usage: "A string which overrides the webhook's default channel. Example: #my-custom-channel",
|
||||
EnvVar: "WATCHTOWER_NOTIFICATION_SLACK_CHANNEL",
|
||||
},
|
||||
cli.StringFlag{
|
||||
Name: "notification-slack-icon-emoji",
|
||||
Usage: "An emoji code string to use in place of the default icon",
|
||||
EnvVar: "WATCHTOWER_NOTIFICATION_SLACK_ICON_EMOJI",
|
||||
},
|
||||
cli.StringFlag{
|
||||
Name: "notification-slack-icon-url",
|
||||
Usage: "An icon image URL string to use in place of the default icon",
|
||||
EnvVar: "WATCHTOWER_NOTIFICATION_SLACK_ICON_URL",
|
||||
},
|
||||
cli.StringFlag{
|
||||
Name: "notification-msteams-hook",
|
||||
Usage: "The MSTeams WebHook URL to send notifications to",
|
||||
EnvVar: "WATCHTOWER_NOTIFICATION_MSTEAMS_HOOK_URL",
|
||||
},
|
||||
cli.BoolFlag{
|
||||
Name: "notification-msteams-data",
|
||||
Usage: "The MSTeams notifier will try to extract log entry fields as MSTeams message facts",
|
||||
EnvVar: "WATCHTOWER_NOTIFICATION_MSTEAMS_USE_LOG_DATA",
|
||||
},
|
||||
cli.BoolFlag{
|
||||
Name: "monitor-only",
|
||||
Usage: "Will only monitor for new images, not update the containers",
|
||||
EnvVar: "WATCHTOWER_MONITOR_ONLY",
|
||||
},
|
||||
}
|
||||
|
||||
if err := app.Run(os.Args); err != nil {
|
||||
log.Fatal(err)
|
||||
}
|
||||
}
|
||||
|
||||
func before(c *cli.Context) error {
|
||||
|
|
@ -215,10 +75,8 @@ func before(c *cli.Context) error {
|
|||
scheduleSpec = "@every " + strconv.Itoa(c.Int("interval")) + "s"
|
||||
}
|
||||
|
||||
cleanup = c.GlobalBool("cleanup")
|
||||
noRestart = c.GlobalBool("no-restart")
|
||||
monitorOnly = c.GlobalBool("monitor-only")
|
||||
timeout = c.GlobalDuration("stop-timeout")
|
||||
readFlags(c)
|
||||
|
||||
if timeout < 0 {
|
||||
log.Fatal("Please specify a positive value for timeout value.")
|
||||
}
|
||||
|
|
@ -238,13 +96,25 @@ func before(c *cli.Context) error {
|
|||
|
||||
func start(c *cli.Context) error {
|
||||
names := c.Args()
|
||||
filter := container.BuildFilter(names, enableLabel)
|
||||
|
||||
if c.GlobalBool("run-once") {
|
||||
log.Info("Running a one time update.")
|
||||
runUpdatesWithNotifications(filter)
|
||||
os.Exit(1)
|
||||
return nil
|
||||
}
|
||||
|
||||
if err := actions.CheckPrereqs(client, cleanup); err != nil {
|
||||
log.Fatal(err)
|
||||
}
|
||||
|
||||
filter := container.BuildFilter(names, enableLabel)
|
||||
runUpgradesOnSchedule(filter)
|
||||
os.Exit(1)
|
||||
return nil
|
||||
}
|
||||
|
||||
func runUpgradesOnSchedule(filter container.Filter) error {
|
||||
tryLockSem := make(chan bool, 1)
|
||||
tryLockSem <- true
|
||||
|
||||
|
|
@ -255,18 +125,7 @@ func start(c *cli.Context) error {
|
|||
select {
|
||||
case v := <-tryLockSem:
|
||||
defer func() { tryLockSem <- v }()
|
||||
notifier.StartNotification()
|
||||
updateParams := actions.UpdateParams{
|
||||
Filter: filter,
|
||||
Cleanup: cleanup,
|
||||
NoRestart: noRestart,
|
||||
Timeout: timeout,
|
||||
MonitorOnly: monitorOnly,
|
||||
}
|
||||
if err := actions.Update(client, updateParams); err != nil {
|
||||
log.Println(err)
|
||||
}
|
||||
notifier.SendNotification()
|
||||
runUpdatesWithNotifications(filter)
|
||||
default:
|
||||
log.Debug("Skipped another update already running.")
|
||||
}
|
||||
|
|
@ -281,7 +140,7 @@ func start(c *cli.Context) error {
|
|||
return err
|
||||
}
|
||||
|
||||
log.Info("First run: " + cron.Entries()[0].Schedule.Next(time.Now()).String())
|
||||
log.Debug("Starting Watchtower and scheduling first run: " + cron.Entries()[0].Schedule.Next(time.Now()).String())
|
||||
cron.Start()
|
||||
|
||||
// Graceful shut-down on SIGINT/SIGTERM
|
||||
|
|
@ -293,16 +152,31 @@ func start(c *cli.Context) error {
|
|||
cron.Stop()
|
||||
log.Info("Waiting for running update to be finished...")
|
||||
<-tryLockSem
|
||||
os.Exit(1)
|
||||
return nil
|
||||
}
|
||||
|
||||
func runUpdatesWithNotifications(filter container.Filter) {
|
||||
notifier.StartNotification()
|
||||
updateParams := actions.UpdateParams{
|
||||
Filter: filter,
|
||||
Cleanup: cleanup,
|
||||
NoRestart: noRestart,
|
||||
Timeout: timeout,
|
||||
MonitorOnly: monitorOnly,
|
||||
}
|
||||
err := actions.Update(client, updateParams)
|
||||
if err != nil {
|
||||
log.Println(err)
|
||||
}
|
||||
}
|
||||
|
||||
func setEnvOptStr(env string, opt string) error {
|
||||
if opt != "" && opt != os.Getenv(env) {
|
||||
err := os.Setenv(env, opt)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
if opt == "" || opt == os.Getenv(env) {
|
||||
return nil
|
||||
}
|
||||
err := os.Setenv(env, opt)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
|
@ -325,3 +199,10 @@ func envConfig(c *cli.Context) error {
|
|||
|
||||
return err
|
||||
}
|
||||
|
||||
func readFlags(c *cli.Context) {
|
||||
cleanup = c.GlobalBool("cleanup")
|
||||
noRestart = c.GlobalBool("no-restart")
|
||||
monitorOnly = c.GlobalBool("monitor-only")
|
||||
timeout = c.GlobalDuration("stop-timeout")
|
||||
}
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue