feat(GODT-2799): SMTP service interacts directly with Server Manager

Bridge no longer needs to manually add and remove accounts from the
service.
This commit is contained in:
Leander Beernaert
2023-08-08 14:14:52 +02:00
parent ded4f370dc
commit 20b188368a
6 changed files with 88 additions and 30 deletions

View File

@ -0,0 +1,41 @@
// Copyright (c) 2023 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 smtp
import "context"
type ServerManager interface {
AddSMTPAccount(ctx context.Context, service *Service) error
RemoveSMTPAccount(ctx context.Context, service *Service) error
}
type NullServerManager struct{}
func NewNullServerManager() *NullServerManager {
return &NullServerManager{}
}
func (n NullServerManager) AddSMTPAccount(_ context.Context, _ *Service) error {
// Does nothing.
return nil
}
func (n NullServerManager) RemoveSMTPAccount(_ context.Context, _ *Service) error {
// Does nothing.
return nil
}

View File

@ -62,7 +62,8 @@ type Service struct {
addressSubscriber *userevents.AddressChanneledSubscriber
userSubscriber *userevents.UserChanneledSubscriber
addressMode usertypes.AddressMode
addressMode usertypes.AddressMode
serverManager ServerManager
}
func NewService(
@ -77,6 +78,7 @@ func NewService(
eventService userevents.Subscribable,
mode usertypes.AddressMode,
identityState *useridentity.State,
serverManager ServerManager,
) *Service {
subscriberName := fmt.Sprintf("smpt-%v", userID)
@ -102,7 +104,8 @@ func NewService(
userSubscriber: userevents.NewUserSubscriber(subscriberName),
addressSubscriber: userevents.NewAddressSubscriber(subscriberName),
addressMode: mode,
addressMode: mode,
serverManager: serverManager,
}
}
@ -129,6 +132,12 @@ func (s *Service) Resync(ctx context.Context) error {
return err
}
func (s *Service) OnLogout(ctx context.Context) error {
_, err := s.cpc.Send(ctx, &onLogoutReq{})
return err
}
func (s *Service) checkAuth(ctx context.Context, email string, password []byte) (string, error) {
return cpc.SendTyped[string](ctx, s.cpc, &checkAuthReq{
email: email,
@ -136,8 +145,13 @@ func (s *Service) checkAuth(ctx context.Context, email string, password []byte)
})
}
func (s *Service) Start(ctx context.Context, group *orderedtasks.OrderedCancelGroup) {
func (s *Service) Start(ctx context.Context, group *orderedtasks.OrderedCancelGroup) error {
s.log.Debug("Starting service")
if err := s.serverManager.AddSMTPAccount(ctx, s); err != nil {
return fmt.Errorf("failed to add SMTP account to server: %w", err)
}
group.Go(ctx, s.userID, "smtp-service", func(ctx context.Context) {
logging.DoAnnotated(ctx, func(ctx context.Context) {
s.run(ctx)
@ -146,6 +160,8 @@ func (s *Service) Start(ctx context.Context, group *orderedtasks.OrderedCancelGr
"service": "smtp",
})
})
return nil
}
func (s *Service) UserID() string {
@ -196,6 +212,10 @@ func (s *Service) run(ctx context.Context) {
err := s.identityState.OnRefreshEvent(ctx)
request.Reply(ctx, nil, err)
case *onLogoutReq:
err := s.serverManager.RemoveSMTPAccount(ctx, s)
request.Reply(ctx, nil, err)
default:
s.log.Error("Received unknown request")
}
@ -262,3 +282,5 @@ type checkAuthReq struct {
}
type resyncReq struct{}
type onLogoutReq struct{}