feat(GODT-2716): Make Configuration Statistics persistent.

internal/user/config_status.go

@@ -1,127 +0,0 @@
-// Copyright (c) 2023 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 user
-
-import (
-	"encoding/json"
-	"os"
-	"time"
-)
-
-type ConfigurationStatusData struct {
-	PendingSince   time.Time `json:"auto_update"`
-	LastProgress   time.Time `json:"last_progress"`
-	Autoconf       string    `json:"auto_conf"`
-	ClickedLink    uint64    `json:"clicked_link"`
-	ReportSent     bool      `json:"report_sent"`
-	ReportClick    bool      `json:"report_click"`
-	FailureDetails string    `json:"failure_details"`
-}
-
-type ConfigurationStatus struct {
-	FilePath string
-	Data     ConfigurationStatusData
-}
-
-func LoadConfigurationStatus(filepath string) (*ConfigurationStatus, error) {
-	status := ConfigurationStatus{
-		FilePath: filepath,
-		Data:     ConfigurationStatusData{},
-	}
-	if _, err := os.Stat(filepath); err == nil {
-		if err := status.Data.load(filepath); err == nil {
-			return &status, nil
-		}
-	} else {
-		status.Data.init()
-		if err := status.save(); err == nil {
-			return &status, nil
-		}
-	}
-	return &status, nil
-}
-
-func (status *ConfigurationStatus) Success() error {
-	status.Data.init()
-	status.Data.PendingSince = time.Time{}
-	return status.save()
-}
-
-func (status *ConfigurationStatus) Failure(err string) error {
-	status.Data.init()
-	status.Data.FailureDetails = err
-	return status.save()
-}
-
-func (status *ConfigurationStatus) Progress() error {
-	status.Data.LastProgress = time.Now()
-	return status.save()
-}
-
-func (status *ConfigurationStatus) RecordLinkClicked(link uint) error {
-	if !status.Data.hasLinkClicked(link) {
-		status.Data.setClickedLink(link)
-		return status.save()
-	}
-	return nil
-}
-
-func (status *ConfigurationStatus) save() error {
-	return status.Data.save(status.FilePath)
-}
-
-func (data *ConfigurationStatusData) init() {
-	data.PendingSince = time.Now()
-	data.LastProgress = time.Time{}
-	data.Autoconf = ""
-	data.ClickedLink = 0
-	data.ReportSent = false
-	data.ReportClick = false
-	data.FailureDetails = ""
-}
-
-func (data *ConfigurationStatusData) load(filepath string) error {
-	f, err := os.Open(filepath) // nolint: gosec
-	if err != nil {
-		return err
-	}
-
-	defer func() { _ = f.Close() }()
-
-	return json.NewDecoder(f).Decode(data)
-}
-
-func (data *ConfigurationStatusData) save(filepath string) error {
-	f, err := os.Create(filepath) // nolint: gosec
-	if err != nil {
-		return err
-	}
-
-	defer func() { _ = f.Close() }()
-
-	return json.NewEncoder(f).Encode(data)
-}
-
-func (data *ConfigurationStatusData) setClickedLink(pos uint) {
-	data.ClickedLink |= 1 << pos
-}
-
-func (data *ConfigurationStatusData) hasLinkClicked(pos uint) bool {
-	val := data.ClickedLink & (1 << pos)
-	return val > 0
-}

internal/user/user.go

@@ -36,6 +36,7 @@ import (
 	"github.com/ProtonMail/gluon/reporter"
 	"github.com/ProtonMail/go-proton-api"
 	"github.com/ProtonMail/proton-bridge/v3/internal"
+	"github.com/ProtonMail/proton-bridge/v3/internal/configstatus"
 	"github.com/ProtonMail/proton-bridge/v3/internal/events"
 	"github.com/ProtonMail/proton-bridge/v3/internal/logging"
 	"github.com/ProtonMail/proton-bridge/v3/internal/safe"
@@ -94,7 +95,7 @@ type User struct {
 
 	panicHandler async.PanicHandler
 
-	configStatus *ConfigurationStatus
+	configStatus *configstatus.ConfigurationStatus
 }
 
 // New returns a new user.
@@ -130,7 +131,7 @@ func New(
 	}).Info("Creating user object")
 
 	configStatusFile := filepath.Join(cacheDir, apiUser.ID+".json")
-	configStatus, err := LoadConfigurationStatus(configStatusFile)
+	configStatus, err := configstatus.LoadConfigurationStatus(configStatusFile)
 	if err != nil {
 		return nil, fmt.Errorf("failed to init configuration status file: %w", err)
 	}