proton-bridge/internal/frontend/qt/frontend_settings.go

// 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/>.

//go:build build_qt
// +build build_qt

package qt

import (
	"time"

	"github.com/ProtonMail/proton-bridge/internal/config/settings"
	"github.com/ProtonMail/proton-bridge/internal/frontend/clientconfig"
	"github.com/ProtonMail/proton-bridge/pkg/keychain"
	"github.com/ProtonMail/proton-bridge/pkg/ports"
)

func (f *FrontendQt) setIsDiskCacheEnabled() {
	f.qml.SetIsDiskCacheEnabled(f.settings.GetBool(settings.CacheEnabledKey))
}

func (f *FrontendQt) setDiskCachePath() {
	f.qml.SetDiskCachePath(f.settings.Get(settings.CacheLocationKey))
}

func (f *FrontendQt) changeLocalCache(enableDiskCache bool, diskCachePath string) {
	defer f.qml.ChangeLocalCacheFinished()
	defer f.setIsDiskCacheEnabled()
	defer f.setDiskCachePath()

	if enableDiskCache != f.settings.GetBool(settings.CacheEnabledKey) {
		if enableDiskCache {
			if err := f.bridge.EnableCache(); err != nil {
				f.log.WithError(err).Error("Cannot enable disk cache")
			}
		} else {
			if err := f.bridge.DisableCache(); err != nil {
				f.log.WithError(err).Error("Cannot disable disk cache")
			}
		}
	}

	// If disk cache not enabled, or path not changed then no need to change location
	if !enableDiskCache || diskCachePath == f.settings.Get(settings.CacheLocationKey) {
		return
	}

	if err := f.bridge.MigrateCache(f.settings.Get(settings.CacheLocationKey), diskCachePath); err != nil {
		f.log.WithError(err).Error("The local cache location could not be changed.")
		f.qml.CacheCantMove()
		return
	}

	f.settings.Set(settings.CacheLocationKey, diskCachePath)
	f.qml.CacheLocationChangeSuccess()
	f.restart()
}

func (f *FrontendQt) setIsAutostartOn() {
	f.qml.SetIsAutostartOn(f.autostart.IsEnabled())
}

func (f *FrontendQt) toggleAutostart(makeItEnabled bool) {
	defer f.qml.ToggleAutostartFinished()
	if makeItEnabled == f.autostart.IsEnabled() {
		f.setIsAutostartOn()
		return
	}

	var err error
	if makeItEnabled {
		err = f.autostart.Enable()
	} else {
		err = f.autostart.Disable()
	}
	f.setIsAutostartOn()

	if err != nil {
		f.log.
			WithField("makeItEnabled", makeItEnabled).
			WithField("isEnabled", f.qml.IsAutostartOn()).
			WithError(err).
			Error("Autostart change failed")
	}
}

func (f *FrontendQt) toggleDoH(makeItEnabled bool) {
	if f.settings.GetBool(settings.AllowProxyKey) == makeItEnabled {
		f.qml.SetIsDoHEnabled(makeItEnabled)
		return
	}
	f.settings.SetBool(settings.AllowProxyKey, makeItEnabled)
	f.restart()
}

func (f *FrontendQt) toggleUseSSLforSMTP(makeItEnabled bool) {
	if f.settings.GetBool(settings.SMTPSSLKey) == makeItEnabled {
		f.qml.SetUseSSLforSMTP(makeItEnabled)
		return
	}
	f.settings.SetBool(settings.SMTPPortKey, makeItEnabled)
	f.restart()
}

func (f *FrontendQt) changePorts(imapPort, smtpPort int) {
	defer f.qml.ChangePortFinished()
	f.settings.SetInt(settings.IMAPPortKey, imapPort)
	f.settings.SetInt(settings.SMTPPortKey, smtpPort)
	f.restart()
}

func (f *FrontendQt) isPortFree(port int) bool {
	return ports.IsPortFree(port)
}

func (f *FrontendQt) configureAppleMail(userID, address string) {
	user, err := f.bridge.GetUser(userID)
	if err != nil {
		f.log.WithField("userID", userID).Error("Cannot configure AppleMail for user")
		return
	}

	needRestart, err := clientconfig.ConfigureAppleMail(user, address, f.settings)
	if err != nil {
		f.log.WithError(err).Error("Apple Mail config failed")
	}

	if needRestart {
		// There is delay needed for external window to open
		time.Sleep(2 * time.Second)
		f.restart()
	}
}

func (f *FrontendQt) triggerReset() {
	defer f.qml.ResetFinished()
	f.bridge.FactoryReset()
	f.restart()
}

func (f *FrontendQt) setKeychain() {
	availableKeychain := []string{}
	for chain := range keychain.Helpers {
		availableKeychain = append(availableKeychain, chain)
	}
	f.qml.SetAvailableKeychain(availableKeychain)
	f.qml.SetSelectedKeychain(f.bridge.GetKeychainApp())
}

func (f *FrontendQt) selectKeychain(wantKeychain string) {
	if f.bridge.GetKeychainApp() == wantKeychain {
		return
	}

	f.bridge.SetKeychainApp(wantKeychain)
	f.restart()
}

func (f *FrontendQt) restart() {
	f.log.Info("Restarting bridge")
	f.restarter.SetToRestart()
	f.app.Exit(0)
}

func (f *FrontendQt) quit() {
	f.log.Warn("Your wish is my command.. I quit!")
	f.app.Exit(0)
}

func (f *FrontendQt) guiReady() {
	f.initializationDone.Do(f.initializing.Done)
}