From 30800f7383df6e00ab19d76657fa0d07ddfe2889 Mon Sep 17 00:00:00 2001
From: ppacher <ppacher@users.noreply.github.com>
Date: Mon, 6 Apr 2020 12:35:29 +0200
Subject: [PATCH] Allow to disable updates schedule

---
 updates/main.go | 31 ++++++++++++++++++++++++++-----
 1 file changed, 26 insertions(+), 5 deletions(-)

diff --git a/updates/main.go b/updates/main.go
index 9f6a07e2..240924fc 100644
--- a/updates/main.go
+++ b/updates/main.go
@@ -40,10 +40,11 @@ const (
 )
 
 var (
-	module     *modules.Module
-	registry   *updater.ResourceRegistry
-	updateTask *modules.Task
-	updateASAP = abool.New()
+	module              *modules.Module
+	registry            *updater.ResourceRegistry
+	updateTask          *modules.Task
+	updateASAP          = abool.New()
+	disableTaskSchedule bool
 )
 
 func init() {
@@ -111,7 +112,14 @@ func start() error {
 	// start updater task
 	updateTask = module.NewTask("updater", func(ctx context.Context, task *modules.Task) error {
 		return checkForUpdates(ctx)
-	}).Repeat(24 * time.Hour).MaxDelay(1 * time.Hour).Schedule(time.Now().Add(10 * time.Second))
+	})
+
+	if !disableTaskSchedule {
+		updateTask.
+			Repeat(24 * time.Hour).
+			MaxDelay(1 * time.Hour).
+			Schedule(time.Now().Add(10 * time.Second))
+	}
 
 	if updateASAP.IsSet() {
 		updateTask.StartASAP()
@@ -136,6 +144,19 @@ func TriggerUpdate() error {
 	return nil
 }
 
+// DisableUpdateSchedule disables the update schedule.
+// If called, updates are only checked when TriggerUpdate()
+// is called.
+func DisableUpdateSchedule() error {
+	if module.Online() {
+		return fmt.Errorf("module already online")
+	}
+
+	disableTaskSchedule = true
+
+	return nil
+}
+
 func checkForUpdates(ctx context.Context) error {
 	err := registry.DownloadUpdates(ctx)
 	if err != nil {