Silverfish/proton-bridge
1
0
Fork 1
mirror of https://github.com/ProtonMail/proton-bridge.git synced 2025-12-17 15:46:44 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

docs: add docstring for buildBodies

Browse Source
This commit is contained in:
James Houlahan
2020-08-18 10:55:26 +02:00
parent 642666fa59
commit fa5f4acdac
1 changed files with 9 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
pkg/message/parser.go
View File

@ -140,6 +140,13 @@ func collectAttachments(p *parser.Parser) ([]*pmapi.Attachment, []io.Reader, err
return atts, data, nil return atts, data, nil
} }
// buildBodies collects all text/html and text/plain parts and returns two bodies,
// - a rich text body (in which html is allowed), and
// - a plaintext body (in which html is converted to plaintext).
//
// text/html parts are converted to plaintext in order to build the plaintext body,
// unless there is already a plaintext part provided via multipart/alternative,
// in which case the provided alternative is chosen.
func buildBodies(p *parser.Parser) (richBody, plainBody string, err error) { func buildBodies(p *parser.Parser) (richBody, plainBody string, err error) {
richParts, err := collectBodyParts(p, "text/html") richParts, err := collectBodyParts(p, "text/html")
if err != nil { if err != nil {
@ -294,6 +301,8 @@ func determineMIMEType(p *parser.Parser) (string, error) {
return "text/plain", nil return "text/plain", nil
} }
// getPlainBody returns the body of the given part, converting html to
// plaintext where possible.
func getPlainBody(part *parser.Part) []byte { func getPlainBody(part *parser.Part) []byte {
contentType, _, err := part.Header.ContentType() contentType, _, err := part.Header.ContentType()
if err != nil { if err != nil {
@ -301,9 +310,6 @@ func getPlainBody(part *parser.Part) []byte {
} }
switch contentType { switch contentType {
case "text/plain":
return part.Body
case "text/html": case "text/html":
text, err := html2text.FromReader(bytes.NewReader(part.Body)) text, err := html2text.FromReader(bytes.NewReader(part.Body))
if err != nil { if err != nil {