forked from Silverfish/proton-bridge
128 lines
3.2 KiB
Go
128 lines
3.2 KiB
Go
// 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
|
|
}
|