mirror of
https://github.com/ProtonMail/proton-bridge.git
synced 2025-12-10 12:46:46 +00:00
101 lines
3.4 KiB
Go
101 lines
3.4 KiB
Go
// Copyright (c) 2025 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 imapservice
|
|
|
|
import (
|
|
"errors"
|
|
|
|
"github.com/ProtonMail/go-proton-api"
|
|
"github.com/ProtonMail/proton-bridge/v3/internal/usertypes"
|
|
"github.com/ProtonMail/proton-bridge/v3/pkg/message/parser"
|
|
)
|
|
|
|
type connectorInterface interface {
|
|
getSenderProtonAddress(p *parser.Parser) (proton.Address, error)
|
|
getAddress(id string) (proton.Address, bool)
|
|
getPrimaryAddress() (proton.Address, error)
|
|
getAddressMode() usertypes.AddressMode
|
|
logError(err error, errMsg string)
|
|
}
|
|
|
|
func (s *Connector) logError(err error, errMsg string) {
|
|
s.log.WithError(err).Warn(errMsg)
|
|
}
|
|
|
|
func (s *Connector) getAddressMode() usertypes.AddressMode {
|
|
return s.addressMode
|
|
}
|
|
|
|
func (s *Connector) getPrimaryAddress() (proton.Address, error) {
|
|
return s.identityState.GetPrimaryAddress()
|
|
}
|
|
|
|
func (s *Connector) getAddress(id string) (proton.Address, bool) {
|
|
return s.identityState.GetAddress(id)
|
|
}
|
|
|
|
func getImportAddress(p *parser.Parser, isDraft bool, id string, conn connectorInterface) (proton.Address, error) {
|
|
// addr is primary for combined mode or active for split mode
|
|
address, ok := conn.getAddress(id)
|
|
if !ok {
|
|
return proton.Address{}, errors.New("could not find account address")
|
|
}
|
|
|
|
// If the address is external and not BYOE - with sending enabled, then use the primary address as an import target.
|
|
if address.Type == proton.AddressTypeExternal && !address.Send {
|
|
var err error
|
|
address, err = conn.getPrimaryAddress()
|
|
if err != nil {
|
|
return proton.Address{}, errors.New("could not get primary account address")
|
|
}
|
|
}
|
|
|
|
inCombinedMode := conn.getAddressMode() == usertypes.AddressModeCombined
|
|
if !inCombinedMode {
|
|
return address, nil
|
|
}
|
|
|
|
senderAddr, err := conn.getSenderProtonAddress(p)
|
|
if err != nil {
|
|
if !errors.Is(err, errNoSenderAddressMatch) {
|
|
conn.logError(err, "Could not get import address")
|
|
}
|
|
|
|
// We did not find a match, so we use the default address.
|
|
return address, nil
|
|
}
|
|
|
|
if senderAddr.ID == address.ID {
|
|
return address, nil
|
|
}
|
|
|
|
// GODT-3185 / BRIDGE-120 In combined mode, in certain cases we adapt the address used for encryption.
|
|
// - draft with non-default address in combined mode: using sender address
|
|
// - import with non-default address in combined mode: using sender address
|
|
// - import with non-default disabled address in combined mode: using sender address
|
|
isSenderAddressDisabled := (!bool(senderAddr.Send)) || (senderAddr.Status != proton.AddressStatusEnabled)
|
|
isSenderExternalNonBYOE := senderAddr.Type == proton.AddressTypeExternal && !bool(senderAddr.Send)
|
|
|
|
// Forbid drafts/imports for external non-BYOE addresses
|
|
if isSenderExternalNonBYOE || (isDraft && isSenderAddressDisabled) {
|
|
return address, nil
|
|
}
|
|
|
|
return senderAddr, nil
|
|
}
|