yesBad/proton-bridge
1
0
Fork 0
forked from Silverfish/proton-bridge
Code Actions Activity

GODT-1779: Remove go-imap

Browse Source
This commit is contained in:
James Houlahan
2022-08-26 17:00:21 +02:00
parent 3b0bc1ca15
commit 39433fe707
593 changed files with 12725 additions and 91626 deletions
Download Patch File Download Diff File Expand all files Collapse all files

139
internal/updater/updater.go
View File

@ -1,91 +1,50 @@
// Copyright (c) 2022 Proton AG
//
// This file is part of Proton Mail Bridge.
//
// Proton Mail 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.
//
// Proton Mail 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 Proton Mail Bridge. If not, see <https://www.gnu.org/licenses/>.
package updater
import (
"bytes"
"context"
"encoding/json"
"fmt"
"io"
"github.com/Masterminds/semver/v3"
"github.com/ProtonMail/gopenpgp/v2/crypto"
"github.com/ProtonMail/proton-bridge/v2/internal/config/settings"
"github.com/ProtonMail/proton-bridge/v2/pkg/pmapi"
"github.com/pkg/errors"
"github.com/sirupsen/logrus"
)
var ErrManualUpdateRequired = errors.New("manual update is required")
var (
ErrDownloadVerify = errors.New("failed to download or verify the update")
ErrInstall = errors.New("failed to install the update")
)
type Downloader interface {
DownloadAndVerify(ctx context.Context, kr *crypto.KeyRing, url, sig string) ([]byte, error)
}
type Installer interface {
InstallUpdate(*semver.Version, io.Reader) error
}
type Settings interface {
Get(settings.Key) string
Set(settings.Key, string)
GetFloat64(settings.Key) float64
}
type Updater struct {
cm pmapi.Manager
installer Installer
settings Settings
kr *crypto.KeyRing
curVer *semver.Version
updateURLName string
platform string
locker *locker
verifier *crypto.KeyRing
product string
platform string
}
func New(
cm pmapi.Manager,
installer Installer,
s Settings,
kr *crypto.KeyRing,
curVer *semver.Version,
updateURLName, platform string,
) *Updater {
// If there's some unexpected value in the preferences, we force it back onto the default channel.
// This prevents users from screwing up silent updates by modifying their prefs.json file.
if channel := UpdateChannel(s.Get(settings.UpdateChannelKey)); !(channel == StableChannel || channel == EarlyChannel) {
s.Set(settings.UpdateChannelKey, string(DefaultUpdateChannel))
}
func NewUpdater(installer Installer, verifier *crypto.KeyRing, product, platform string) *Updater {
return &Updater{
cm: cm,
installer: installer,
settings: s,
kr: kr,
curVer: curVer,
updateURLName: updateURLName,
platform: platform,
locker: newLocker(),
installer: installer,
verifier: verifier,
product: product,
platform: platform,
}
}
func (u *Updater) Check() (VersionInfo, error) {
logrus.Info("Checking for updates")
b, err := u.cm.DownloadAndVerify(
u.kr,
func (u *Updater) GetVersionInfo(downloader Downloader, channel Channel) (VersionInfo, error) {
b, err := downloader.DownloadAndVerify(
context.Background(),
u.verifier,
u.getVersionFileURL(),
u.getVersionFileURL()+".sig",
)
@ -99,7 +58,7 @@ func (u *Updater) Check() (VersionInfo, error) {
return VersionInfo{}, err
}
version, ok := versionMap[u.settings.Get(settings.UpdateChannelKey)]
version, ok := versionMap[channel]
if !ok {
return VersionInfo{}, errors.New("no updates available for this channel")
}
@ -107,45 +66,27 @@ func (u *Updater) Check() (VersionInfo, error) {
return version, nil
}
func (u *Updater) IsUpdateApplicable(version VersionInfo) bool {
if !version.Version.GreaterThan(u.curVer) {
return false
func (u *Updater) InstallUpdate(downloader Downloader, update VersionInfo) error {
b, err := downloader.DownloadAndVerify(
context.Background(),
u.verifier,
update.Package,
update.Package+".sig",
)
if err != nil {
return ErrDownloadVerify
}
if u.settings.GetFloat64(settings.RolloutKey) > version.RolloutProportion {
return false
if err := u.installer.InstallUpdate(update.Version, bytes.NewReader(b)); err != nil {
return ErrInstall
}
return true
return nil
}
func (u *Updater) IsDowngrade(version VersionInfo) bool {
return version.Version.LessThan(u.curVer)
}
func (u *Updater) CanInstall(version VersionInfo) bool {
if version.MinAuto == nil {
return true
}
return !u.curVer.LessThan(version.MinAuto)
}
func (u *Updater) InstallUpdate(update VersionInfo) error {
return u.locker.doOnce(func() error {
logrus.WithField("package", update.Package).Info("Installing update package")
b, err := u.cm.DownloadAndVerify(u.kr, update.Package, update.Package+".sig")
if err != nil {
return errors.Wrap(ErrDownloadVerify, err.Error())
}
if err := u.installer.InstallUpdate(update.Version, bytes.NewReader(b)); err != nil {
return errors.Wrap(ErrInstall, err.Error())
}
u.curVer = update.Version
return nil
})
// getVersionFileURL returns the URL of the version file.
// For example:
// - https://protonmail.com/download/bridge/version_linux.json
func (u *Updater) getVersionFileURL() string {
return fmt.Sprintf("%v/%v/version_%v.json", Host, u.product, u.platform)
}