Silverfish/proton-bridge
1
0
Fork 1
mirror of https://github.com/ProtonMail/proton-bridge.git synced 2025-12-17 23:56:56 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Use map instead of list as set

Browse Source
This commit is contained in:
Michal Horejsek
2020-10-12 10:13:58 +02:00
parent ef85c8df24
commit e0292fe957
6 changed files with 41 additions and 39 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
internal/transfer/provider_mbox_gmail_labels.go
View File

@ -25,9 +25,11 @@ import (
"strings"
)
type stringSet map[string]bool
const xGmailLabelsHeader = "X-Gmail-Labels"
func getGmailLabelsFromMboxFile(filePath string) ([]string, error) {
func getGmailLabelsFromMboxFile(filePath string) (stringSet, error) {
f, err := os.Open(filePath) //nolint[gosec]
if err != nil {
return nil, err
@ -35,8 +37,8 @@ func getGmailLabelsFromMboxFile(filePath string) ([]string, error) {
return getGmailLabelsFromMboxReader(f)
}
func getGmailLabelsFromMboxReader(f io.Reader) ([]string, error) {
allLabels := []string{}
func getGmailLabelsFromMboxReader(f io.Reader) (stringSet, error) {
allLabels := map[string]bool{}
r := bufio.NewReader(f)
for {
@ -57,8 +59,8 @@ func getGmailLabelsFromMboxReader(f io.Reader) ([]string, error) {
continue
}
if bytes.HasPrefix(b, []byte(xGmailLabelsHeader)) {
for _, label := range getGmailLabelsFromValue(string(b)) {
allLabels = appendIfNew(allLabels, label)
for label := range getGmailLabelsFromValue(string(b)) {
allLabels[label] = true
}
}
}
@ -66,7 +68,7 @@ func getGmailLabelsFromMboxReader(f io.Reader) ([]string, error) {
return allLabels, nil
}
func getGmailLabelsFromMessage(body []byte) ([]string, error) {
func getGmailLabelsFromMessage(body []byte) (stringSet, error) {
header, err := getMessageHeader(body)
if err != nil {
return nil, err
@ -75,15 +77,15 @@ func getGmailLabelsFromMessage(body []byte) ([]string, error) {
return getGmailLabelsFromValue(labels), nil
}
func getGmailLabelsFromValue(value string) []string {
func getGmailLabelsFromValue(value string) stringSet {
value = strings.TrimPrefix(value, xGmailLabelsHeader+":")
labels := []string{}
labels := map[string]bool{}
for _, label := range strings.Split(value, ",") {
label = strings.TrimSpace(label)
if label == "" {
continue
}
labels = appendIfNew(labels, label)
labels[label] = true
}
return labels
}