forked from Silverfish/proton-bridge
3ef526333a
Refactor code to isolate the SMTP functionality in a dedicated SMTP service for each user as discussed in the Bridge Service Architecture RFC. Some shared types have been moved from `user` to `usertypes` so that they can be shared with Service and User Code. Finally due to lack of recursive imports, the user data SMTP needs access to is hidden behind an interface until the User Identity service is implemented.
87 lines
2.6 KiB
Go
87 lines
2.6 KiB
Go
// 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 (
|
|
"github.com/ProtonMail/gluon/rfc822"
|
|
"github.com/ProtonMail/go-proton-api"
|
|
"github.com/ProtonMail/gopenpgp/v2/crypto"
|
|
"github.com/ProtonMail/proton-bridge/v3/pkg/message"
|
|
"github.com/bradenaw/juniper/xslices"
|
|
"golang.org/x/exp/maps"
|
|
"golang.org/x/exp/slices"
|
|
)
|
|
|
|
func createSendReq(
|
|
kr *crypto.KeyRing,
|
|
mimeBody message.MIMEBody,
|
|
richBody, plainBody message.Body,
|
|
recipients recipients,
|
|
attKeys map[string]*crypto.SessionKey,
|
|
) (proton.SendDraftReq, error) {
|
|
var req proton.SendDraftReq
|
|
|
|
if recs := recipients.scheme(proton.PGPMIMEScheme, proton.ClearMIMEScheme); len(recs) > 0 {
|
|
if err := req.AddMIMEPackage(kr, string(mimeBody), recs); err != nil {
|
|
return proton.SendDraftReq{}, err
|
|
}
|
|
}
|
|
|
|
if recs := recipients.scheme(proton.InternalScheme, proton.ClearScheme, proton.PGPInlineScheme); len(recs) > 0 {
|
|
if recs := recs.content(rfc822.TextHTML); len(recs) > 0 {
|
|
if err := req.AddTextPackage(kr, string(richBody), rfc822.TextHTML, recs, attKeys); err != nil {
|
|
return proton.SendDraftReq{}, err
|
|
}
|
|
}
|
|
|
|
if recs := recs.content(rfc822.TextPlain); len(recs) > 0 {
|
|
if err := req.AddTextPackage(kr, string(plainBody), rfc822.TextPlain, recs, attKeys); err != nil {
|
|
return proton.SendDraftReq{}, err
|
|
}
|
|
}
|
|
}
|
|
|
|
return req, nil
|
|
}
|
|
|
|
type recipients map[string]proton.SendPreferences
|
|
|
|
func (r recipients) scheme(scheme ...proton.EncryptionScheme) recipients {
|
|
res := make(recipients)
|
|
|
|
for _, addr := range xslices.Filter(maps.Keys(r), func(addr string) bool {
|
|
return slices.Contains(scheme, r[addr].EncryptionScheme)
|
|
}) {
|
|
res[addr] = r[addr]
|
|
}
|
|
|
|
return res
|
|
}
|
|
|
|
func (r recipients) content(mimeType ...rfc822.MIMEType) recipients {
|
|
res := make(recipients)
|
|
|
|
for _, addr := range xslices.Filter(maps.Keys(r), func(addr string) bool {
|
|
return slices.Contains(mimeType, r[addr].MIMEType)
|
|
}) {
|
|
res[addr] = r[addr]
|
|
}
|
|
|
|
return res
|
|
}
|