We build too many walls and not enough bridges

test/liveapi/labels.go

@@ -0,0 +1,105 @@
+// 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 liveapi
+
+import (
+	"fmt"
+	"strings"
+
+	"github.com/ProtonMail/proton-bridge/pkg/pmapi"
+	"github.com/pkg/errors"
+)
+
+var systemLabelNameToID = map[string]string{ //nolint[gochecknoglobals]
+	"INBOX":    pmapi.InboxLabel,
+	"Trash":    pmapi.TrashLabel,
+	"Spam":     pmapi.SpamLabel,
+	"All Mail": pmapi.AllMailLabel,
+	"Archive":  pmapi.ArchiveLabel,
+	"Sent":     pmapi.SentLabel,
+	"Drafts":   pmapi.DraftLabel,
+}
+
+func (cntrl *Controller) AddUserLabel(username string, label *pmapi.Label) error {
+	client, ok := cntrl.pmapiByUsername[username]
+	if !ok {
+		return fmt.Errorf("user %s does not exist", username)
+	}
+
+	label.Exclusive = getLabelExclusive(label.Name)
+	label.Name = getLabelNameWithoutPrefix(label.Name)
+	label.Color = pmapi.LabelColors[0]
+	if _, err := client.CreateLabel(label); err != nil {
+		return errors.Wrap(err, "failed to create label")
+	}
+	return nil
+}
+
+func (cntrl *Controller) GetLabelIDs(username string, labelNames []string) ([]string, error) {
+	labelIDs := []string{}
+	for _, labelName := range labelNames {
+		labelID, err := cntrl.getLabelID(username, labelName)
+		if err != nil {
+			return nil, err
+		}
+		labelIDs = append(labelIDs, labelID)
+	}
+	return labelIDs, nil
+}
+
+func (cntrl *Controller) getLabelID(username, labelName string) (string, error) {
+	if labelID, ok := systemLabelNameToID[labelName]; ok {
+		return labelID, nil
+	}
+
+	client, ok := cntrl.pmapiByUsername[username]
+	if !ok {
+		return "", fmt.Errorf("user %s does not exist", username)
+	}
+
+	labels, err := client.ListLabels()
+	if err != nil {
+		return "", errors.Wrap(err, "failed to list labels")
+	}
+
+	exclusive := getLabelExclusive(labelName)
+	labelName = getLabelNameWithoutPrefix(labelName)
+	for _, label := range labels {
+		if label.Exclusive == exclusive && label.Name == labelName {
+			return label.ID, nil
+		}
+	}
+	return "", fmt.Errorf("label %s:%s does not exist", username, labelName)
+}
+
+func getLabelNameWithoutPrefix(name string) string {
+	if strings.HasPrefix(name, "Folders/") {
+		return strings.TrimPrefix(name, "Folders/")
+	}
+	if strings.HasPrefix(name, "Labels/") {
+		return strings.TrimPrefix(name, "Labels/")
+	}
+	return name
+}
+
+func getLabelExclusive(name string) int {
+	if strings.HasPrefix(name, "Folders/") {
+		return 1
+	}
+	return 0
+}