GODT-1425: Factory reset enables launch on startup

internal/bridge/autostart.go

@@ -0,0 +1,31 @@
+// Copyright (c) 2021 Proton Technologies AG
+//
+// This file is part of ProtonMail Bridge.
+//
+// ProtonMail Bridge is free software: you can redistribute it and/or modify
+// it under the terms of the GNU General Public License as published by
+// the Free Software Foundation, either version 3 of the License, or
+// (at your option) any later version.
+//
+// ProtonMail Bridge is distributed in the hope that it will be useful,
+// but WITHOUT ANY WARRANTY; without even the implied warranty of
+// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+// GNU General Public License for more details.
+//
+// You should have received a copy of the GNU General Public License
+// along with ProtonMail Bridge.  If not, see <https://www.gnu.org/licenses/>.
+
+// Package bridge provides core functionality of Bridge app.
+package bridge
+
+func (b *Bridge) IsAutostartEnabled() bool {
+	return b.autostart.IsEnabled()
+}
+
+func (b *Bridge) EnableAutostart() error {
+	return b.autostart.Enable()
+}
+
+func (b *Bridge) DisableAutostart() error {
+	return b.autostart.Disable()
+}

internal/bridge/bridge.go

@@ -25,6 +25,7 @@ import (
 	"time"
 
 	"github.com/Masterminds/semver/v3"
+	"github.com/ProtonMail/go-autostart"
 	"github.com/ProtonMail/proton-bridge/internal/config/settings"
 	"github.com/ProtonMail/proton-bridge/internal/constants"
 	"github.com/ProtonMail/proton-bridge/internal/metrics"
@@ -52,8 +53,11 @@ type Bridge struct {
 	updater       Updater
 	versioner     Versioner
 	cacheProvider CacheProvider
+	autostart     *autostart.App
 	// Bridge's global errors list.
 	errors []error
+
+	lastVersion string
 }
 
 func New(
@@ -69,6 +73,7 @@ func New(
 	credStorer users.CredentialsStorer,
 	updater Updater,
 	versioner Versioner,
+	autostart *autostart.App,
 ) *Bridge {
 	// Allow DoH before starting the app if the user has previously set this setting.
 	// This allows us to start even if protonmail is blocked.
@@ -94,6 +99,7 @@ func New(
 		updater:       updater,
 		versioner:     versioner,
 		cacheProvider: cacheProvider,
+		autostart:     autostart,
 	}
 
 	if setting.GetBool(settings.FirstStartKey) {
@@ -101,9 +107,17 @@ func New(
 			logrus.WithError(err).Error("Failed to send metric")
 		}
 
+		if err := b.EnableAutostart(); err != nil {
+			log.WithError(err).Error("Failed to enable autostart")
+		}
+
 		setting.SetBool(settings.FirstStartKey, false)
 	}
 
+	// Keep in bridge and update in settings the last used version.
+	b.lastVersion = b.settings.Get(settings.LastVersionKey)
+	b.settings.Set(settings.LastVersionKey, constants.Version)
+
 	go b.heartbeat()
 
 	return b
@@ -279,3 +293,8 @@ func (b *Bridge) HasError(err error) bool {
 
 	return false
 }
+
+// GetLastVersion returns the version which was used in previous execution of Bridge.
+func (b *Bridge) GetLastVersion() string {
+	return b.lastVersion
+}