You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
97 lines
3.0 KiB
97 lines
3.0 KiB
// Copyright (c) 2021 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 ( |
|
"context" |
|
"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 bool(label.Exclusive) |
|
} |
|
} |
|
return labelID == pmapi.InboxLabel || labelID == pmapi.ArchiveLabel || labelID == pmapi.SentLabel |
|
} |
|
|
|
func (api *FakePMAPI) ListLabels(context.Context) ([]*pmapi.Label, error) { |
|
if err := api.checkAndRecordCall(GET, "/labels/1", nil); err != nil { |
|
return nil, err |
|
} |
|
return api.labels, nil |
|
} |
|
|
|
func (api *FakePMAPI) CreateLabel(_ context.Context, 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 { |
|
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(_ context.Context, 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(_ context.Context, 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) |
|
}
|
|
|