GODT-1779: Remove go-imap

internal/bridge/updates.go

@@ -0,0 +1,72 @@
+package bridge
+
+import (
+	"time"
+
+	"github.com/ProtonMail/proton-bridge/v2/internal/constants"
+	"github.com/ProtonMail/proton-bridge/v2/internal/events"
+	"github.com/ProtonMail/proton-bridge/v2/internal/updater"
+)
+
+func (bridge *Bridge) CheckForUpdates() {
+	bridge.updateCheckCh <- struct{}{}
+}
+
+func (bridge *Bridge) watchForUpdates() error {
+	ticker := time.NewTicker(constants.UpdateCheckInterval)
+
+	go func() {
+		for {
+			select {
+			case <-bridge.updateCheckCh:
+			case <-ticker.C:
+			}
+
+			version, err := bridge.updater.GetVersionInfo(bridge.api, bridge.vault.GetUpdateChannel())
+			if err != nil {
+				continue
+			}
+
+			if err := bridge.handleUpdate(version); err != nil {
+				continue
+			}
+		}
+	}()
+
+	bridge.updateCheckCh <- struct{}{}
+
+	return nil
+}
+
+func (bridge *Bridge) handleUpdate(version updater.VersionInfo) error {
+	switch {
+	case !version.Version.GreaterThan(bridge.curVersion):
+		bridge.publish(events.UpdateNotAvailable{})
+
+	case version.RolloutProportion < bridge.vault.GetUpdateRollout():
+		bridge.publish(events.UpdateNotAvailable{})
+
+	case bridge.curVersion.LessThan(version.MinAuto):
+		bridge.publish(events.UpdateAvailable{
+			Version:    version,
+			CanInstall: false,
+		})
+
+	case !bridge.vault.GetAutoUpdate():
+		bridge.publish(events.UpdateAvailable{
+			Version:    version,
+			CanInstall: true,
+		})
+
+	default:
+		if err := bridge.updater.InstallUpdate(bridge.api, version); err != nil {
+			return err
+		}
+
+		bridge.publish(events.UpdateInstalled{
+			Version: version,
+		})
+	}
+
+	return nil
+}