mirror of
https://github.com/ProtonMail/proton-bridge.git
synced 2025-12-10 12:46:46 +00:00
125 lines
3.4 KiB
Go
125 lines
3.4 KiB
Go
// Copyright (c) 2022 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/base64"
|
|
"fmt"
|
|
"reflect"
|
|
"strings"
|
|
|
|
"github.com/ProtonMail/go-proton-api"
|
|
"golang.org/x/exp/maps"
|
|
"golang.org/x/exp/slices"
|
|
)
|
|
|
|
// mapTo converts the slice to the given type.
|
|
// This is not runtime safe, so make sure the slice is of the correct type!
|
|
// (This is a workaround for the fact that slices cannot be converted to other types generically).
|
|
func mapTo[From, To any](from []From) []To {
|
|
to := make([]To, 0, len(from))
|
|
|
|
for _, from := range from {
|
|
val, ok := reflect.ValueOf(from).Convert(reflect.TypeOf(to).Elem()).Interface().(To)
|
|
if !ok {
|
|
panic(fmt.Sprintf("cannot convert %T to %T", from, *new(To))) //nolint:gocritic
|
|
}
|
|
|
|
to = append(to, val)
|
|
}
|
|
|
|
return to
|
|
}
|
|
|
|
// groupBy returns a map of the given slice grouped by the given key.
|
|
// Duplicate keys are overwritten.
|
|
func groupBy[Key comparable, Value any](items []Value, key func(Value) Key) map[Key]Value {
|
|
groups := make(map[Key]Value)
|
|
|
|
for _, item := range items {
|
|
groups[key(item)] = item
|
|
}
|
|
|
|
return groups
|
|
}
|
|
|
|
// b64Encode returns the base64 encoding of the given byte slice.
|
|
func b64Encode(b []byte) []byte {
|
|
enc := make([]byte, base64.StdEncoding.EncodedLen(len(b)))
|
|
|
|
base64.StdEncoding.Encode(enc, b)
|
|
|
|
return enc
|
|
}
|
|
|
|
// b64RawEncode returns the base64 encoding of the given byte slice.
|
|
func b64RawEncode(b []byte) []byte {
|
|
enc := make([]byte, base64.RawURLEncoding.EncodedLen(len(b)))
|
|
|
|
base64.RawURLEncoding.Encode(enc, b)
|
|
|
|
return enc
|
|
}
|
|
|
|
// b64RawDecode returns the bytes represented by the base64 encoding of the given byte slice.
|
|
func b64RawDecode(b []byte) ([]byte, error) {
|
|
dec := make([]byte, base64.RawURLEncoding.DecodedLen(len(b)))
|
|
|
|
n, err := base64.RawURLEncoding.Decode(dec, b)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return dec[:n], nil
|
|
}
|
|
|
|
// getAddrID returns the address ID for the given email address.
|
|
func getAddrID(apiAddrs map[string]proton.Address, email string) (string, error) {
|
|
for _, addr := range apiAddrs {
|
|
if strings.EqualFold(addr.Email, sanitizeEmail(email)) {
|
|
return addr.ID, nil
|
|
}
|
|
}
|
|
|
|
return "", fmt.Errorf("address %s not found", email)
|
|
}
|
|
|
|
// getAddrIdx returns the address with the given index.
|
|
func getAddrIdx(apiAddrs map[string]proton.Address, idx int) (proton.Address, error) {
|
|
sorted := sortSlice(maps.Values(apiAddrs), func(a, b proton.Address) bool {
|
|
return a.Order < b.Order
|
|
})
|
|
|
|
if idx < 0 || idx >= len(sorted) {
|
|
return proton.Address{}, fmt.Errorf("address index %d out of range", idx)
|
|
}
|
|
|
|
return sorted[idx], nil
|
|
}
|
|
|
|
// sortSlice returns the given slice sorted by the given comparator.
|
|
func sortSlice[Item any](items []Item, less func(Item, Item) bool) []Item {
|
|
sorted := make([]Item, len(items))
|
|
|
|
copy(sorted, items)
|
|
|
|
slices.SortFunc(sorted, less)
|
|
|
|
return sorted
|
|
}
|