mirror of
https://github.com/ProtonMail/proton-bridge.git
synced 2025-12-10 12:46:46 +00:00
97 lines
3.0 KiB
Go
97 lines
3.0 KiB
Go
// Copyright (c) 2020 Proton Technologies AG
|
|
//
|
|
// This file is part of ProtonMail Bridge.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 fakeapi
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"github.com/ProtonMail/proton-bridge/pkg/pmapi"
|
|
)
|
|
|
|
func (api *FakePMAPI) isLabelFolder(labelID string) bool {
|
|
for _, label := range api.labels {
|
|
if label.ID == labelID {
|
|
return label.Exclusive == 1
|
|
}
|
|
}
|
|
return labelID == pmapi.InboxLabel || labelID == pmapi.ArchiveLabel || labelID == pmapi.SentLabel
|
|
}
|
|
|
|
func (api *FakePMAPI) ListLabels() ([]*pmapi.Label, error) {
|
|
if err := api.checkAndRecordCall(GET, "/labels/1", nil); err != nil {
|
|
return nil, err
|
|
}
|
|
return api.labels, nil
|
|
}
|
|
|
|
func (api *FakePMAPI) CreateLabel(label *pmapi.Label) (*pmapi.Label, error) {
|
|
if err := api.checkAndRecordCall(POST, "/labels", &pmapi.LabelReq{Label: label}); err != nil {
|
|
return nil, err
|
|
}
|
|
for _, existingLabel := range api.labels {
|
|
if existingLabel.Name == label.Name {
|
|
return nil, fmt.Errorf("folder or label %s already exists", label.Name)
|
|
}
|
|
}
|
|
prefix := "label"
|
|
if label.Exclusive == 1 {
|
|
prefix = "folder"
|
|
}
|
|
label.ID = api.controller.labelIDGenerator.next(prefix)
|
|
if label.Path == "" {
|
|
label.Path = label.Name
|
|
}
|
|
api.labels = append(api.labels, label)
|
|
api.addEventLabel(pmapi.EventCreate, label)
|
|
return label, nil
|
|
}
|
|
|
|
func (api *FakePMAPI) UpdateLabel(label *pmapi.Label) (*pmapi.Label, error) {
|
|
if err := api.checkAndRecordCall(PUT, "/labels", &pmapi.LabelReq{Label: label}); err != nil {
|
|
return nil, err
|
|
}
|
|
for idx, existingLabel := range api.labels {
|
|
if existingLabel.ID == label.ID {
|
|
// Request doesn't have to include all properties and these have to stay the same.
|
|
label.Type = existingLabel.Type
|
|
label.Exclusive = existingLabel.Exclusive
|
|
if label.Path == "" {
|
|
label.Path = label.Name
|
|
}
|
|
api.labels[idx] = label
|
|
api.addEventLabel(pmapi.EventUpdate, label)
|
|
return label, nil
|
|
}
|
|
}
|
|
return nil, fmt.Errorf("label %s does not exist", label.ID)
|
|
}
|
|
|
|
func (api *FakePMAPI) DeleteLabel(labelID string) error {
|
|
if err := api.checkAndRecordCall(DELETE, "/labels/"+labelID, nil); err != nil {
|
|
return err
|
|
}
|
|
for idx, existingLabel := range api.labels {
|
|
if existingLabel.ID == labelID {
|
|
api.labels = append(api.labels[:idx], api.labels[idx+1:]...)
|
|
api.addEventLabel(pmapi.EventDelete, existingLabel)
|
|
return nil
|
|
}
|
|
}
|
|
return fmt.Errorf("label %s does not exist", labelID)
|
|
}
|