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.
111 lines
3.1 KiB
111 lines
3.1 KiB
// 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 context |
|
|
|
import ( |
|
"strings" |
|
|
|
"github.com/ProtonMail/proton-bridge/internal/bridge/credentials" |
|
) |
|
|
|
// bridgePassword is password to be used for IMAP or SMTP under tests. |
|
const bridgePassword = "bridgepassword" |
|
|
|
type fakeCredStore struct { |
|
credentials map[string]*credentials.Credentials |
|
} |
|
|
|
// 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{}} |
|
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, apiToken, mailboxPassword string, emails []string) (*credentials.Credentials, error) { |
|
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: apiToken, |
|
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) error { |
|
return nil |
|
} |
|
|
|
func (c *fakeCredStore) UpdateEmails(userID string, emails []string) error { |
|
return nil |
|
} |
|
|
|
func (c *fakeCredStore) UpdatePassword(userID, password string) error { |
|
creds, err := c.Get(userID) |
|
if err != nil { |
|
return err |
|
} |
|
creds.MailboxPassword = password |
|
return nil |
|
} |
|
|
|
func (c *fakeCredStore) UpdateToken(userID, apiToken string) error { |
|
creds, err := c.Get(userID) |
|
if err != nil { |
|
return err |
|
} |
|
creds.APIToken = apiToken |
|
return nil |
|
} |
|
|
|
func (c *fakeCredStore) Logout(userID string) error { |
|
c.credentials[userID].APIToken = "" |
|
c.credentials[userID].MailboxPassword = "" |
|
return nil |
|
} |
|
|
|
func (c *fakeCredStore) Delete(userID string) error { |
|
delete(c.credentials, userID) |
|
return nil |
|
}
|
|
|