forked from Silverfish/proton-bridge
135 lines
4.2 KiB
Go
135 lines
4.2 KiB
Go
// Copyright (c) 2022 Proton AG
|
|
//
|
|
// This file is part of Proton Mail Bridge.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 context
|
|
|
|
import (
|
|
"strings"
|
|
|
|
"github.com/ProtonMail/proton-bridge/v2/internal/users/credentials"
|
|
"github.com/pkg/errors"
|
|
)
|
|
|
|
// bridgePassword is password to be used for IMAP or SMTP under tests.
|
|
const bridgePassword = "bridgepassword"
|
|
|
|
type fakeCredStore struct {
|
|
credentials map[string]*credentials.Credentials
|
|
|
|
failOnWrite bool
|
|
}
|
|
|
|
// newFakeCredStore returns a fake credentials store (optionally configured with the given credentials).
|
|
func newFakeCredStore(initCreds ...*credentials.Credentials) (c *fakeCredStore) {
|
|
c = &fakeCredStore{credentials: map[string]*credentials.Credentials{}, failOnWrite: false}
|
|
for _, creds := range initCreds {
|
|
if creds == nil {
|
|
continue
|
|
}
|
|
c.credentials[creds.UserID] = &credentials.Credentials{}
|
|
*c.credentials[creds.UserID] = *creds
|
|
}
|
|
return
|
|
}
|
|
|
|
func (c *fakeCredStore) List() (userIDs []string, err error) {
|
|
keys := []string{}
|
|
for key := range c.credentials {
|
|
keys = append(keys, key)
|
|
}
|
|
return keys, nil
|
|
}
|
|
|
|
func (c *fakeCredStore) Add(userID, userName, uid, ref string, mailboxPassword []byte, emails []string) (*credentials.Credentials, error) {
|
|
if c.failOnWrite {
|
|
return nil, errors.New("An invalid attempt to change the owner of an item. (-25244)")
|
|
}
|
|
bridgePassword := bridgePassword
|
|
if c, ok := c.credentials[userID]; ok {
|
|
bridgePassword = c.BridgePassword
|
|
}
|
|
c.credentials[userID] = &credentials.Credentials{
|
|
UserID: userID,
|
|
Name: userName,
|
|
Emails: strings.Join(emails, ";"),
|
|
APIToken: uid + ":" + ref,
|
|
MailboxPassword: mailboxPassword,
|
|
BridgePassword: bridgePassword,
|
|
IsCombinedAddressMode: true, // otherwise by default starts in split mode
|
|
}
|
|
return c.Get(userID)
|
|
}
|
|
|
|
func (c *fakeCredStore) Get(userID string) (*credentials.Credentials, error) {
|
|
return c.credentials[userID], nil
|
|
}
|
|
|
|
func (c *fakeCredStore) SwitchAddressMode(userID string) (*credentials.Credentials, error) {
|
|
if c.failOnWrite {
|
|
return nil, errors.New("An invalid attempt to change the owner of an item. (-25244)")
|
|
}
|
|
return c.credentials[userID], nil
|
|
}
|
|
|
|
func (c *fakeCredStore) UpdateEmails(userID string, emails []string) (*credentials.Credentials, error) {
|
|
if c.failOnWrite {
|
|
return nil, errors.New("An invalid attempt to change the owner of an item. (-25244)")
|
|
}
|
|
return c.credentials[userID], nil
|
|
}
|
|
|
|
func (c *fakeCredStore) UpdatePassword(userID string, password []byte) (*credentials.Credentials, error) {
|
|
if c.failOnWrite {
|
|
return nil, errors.New("An invalid attempt to change the owner of an item. (-25244)")
|
|
}
|
|
creds, err := c.Get(userID)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
creds.MailboxPassword = password
|
|
return creds, nil
|
|
}
|
|
|
|
func (c *fakeCredStore) UpdateToken(userID, uid, ref string) (*credentials.Credentials, error) {
|
|
if c.failOnWrite {
|
|
return nil, errors.New("An invalid attempt to change the owner of an item. (-25244)")
|
|
}
|
|
creds, err := c.Get(userID)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
creds.APIToken = uid + ":" + ref
|
|
return creds, nil
|
|
}
|
|
|
|
func (c *fakeCredStore) Logout(userID string) (*credentials.Credentials, error) {
|
|
if c.failOnWrite {
|
|
return nil, errors.New("An invalid attempt to change the owner of an item. (-25244)")
|
|
}
|
|
c.credentials[userID].APIToken = ""
|
|
c.credentials[userID].MailboxPassword = []byte{}
|
|
return c.credentials[userID], nil
|
|
}
|
|
|
|
func (c *fakeCredStore) Delete(userID string) error {
|
|
if c.failOnWrite {
|
|
return errors.New("An invalid attempt to change the owner of an item. (-25244)")
|
|
}
|
|
delete(c.credentials, userID)
|
|
return nil
|
|
}
|