aboutsummaryrefslogtreecommitdiff
path: root/utils/email/parse.go
blob: 46acc85b41fdb3f97a1fdc2c71a18fbd757d865a (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
package email

import (
	"bytes"
	"net/mail"
	"strings"

	"github.com/jhillyerd/enmime"
)

type ParsedEmail struct {
	MessageID      string
	FromAddress    string
	FromName       string
	ToAddresses    []string
	CcAddresses    []string
	BccAddresses   []string
	ReplyToAddress string
	ReturnPath     string
	Subject        string
	Snippet        string
	Size           int64
	Attachments    []ParsedAttachment
}

type ParsedAttachment struct {
	Filename    string
	ContentType string
	ContentID   string
	Size        int64
	IsInline    bool
}

func Parse(rawMessage []byte) (*ParsedEmail, error) {
	envelope, parseError := enmime.ReadEnvelope(bytes.NewReader(rawMessage))
	if parseError != nil {
		return nil, parseError
	}

	return &ParsedEmail{
		MessageID:      envelope.GetHeader("Message-ID"),
		FromAddress:    extractAddress(envelope.GetHeader("From")),
		FromName:       extractName(envelope.GetHeader("From")),
		ToAddresses:    extractAddressList(envelope.GetHeader("To")),
		CcAddresses:    extractAddressList(envelope.GetHeader("Cc")),
		BccAddresses:   extractAddressList(envelope.GetHeader("Bcc")),
		ReplyToAddress: envelope.GetHeader("Reply-To"),
		ReturnPath:     envelope.GetHeader("Return-Path"),
		Subject:        envelope.GetHeader("Subject"),
		Snippet:        generateSnippet(envelope.Text),
		Size:           int64(len(rawMessage)),
		Attachments:    extractAttachments(envelope),
	}, nil
}

func extractAddress(rawHeader string) string {
	parsed, parseError := mail.ParseAddress(rawHeader)
	if parseError != nil {
		return rawHeader
	}

	return parsed.Address
}

func extractName(rawHeader string) string {
	parsed, parseError := mail.ParseAddress(rawHeader)
	if parseError != nil {
		return ""
	}

	return parsed.Name
}

func extractAddressList(rawHeader string) []string {
	if rawHeader == "" {
		return nil
	}

	addresses, parseError := mail.ParseAddressList(rawHeader)
	if parseError != nil {
		return []string{rawHeader}
	}

	extracted := make([]string, len(addresses))
	for index, address := range addresses {
		extracted[index] = address.Address
	}

	return extracted
}

func generateSnippet(textBody string) string {
	trimmed := strings.TrimSpace(textBody)
	if len(trimmed) <= SnippetLength {
		return trimmed
	}

	return trimmed[:SnippetLength]
}

func extractAttachments(envelope *enmime.Envelope) []ParsedAttachment {
	var attachments []ParsedAttachment

	for _, attachment := range envelope.Attachments {
		attachments = append(attachments, ParsedAttachment{
			Filename:    attachment.FileName,
			ContentType: attachment.ContentType,
			ContentID:   attachment.ContentID,
			Size:        int64(len(attachment.Content)),
			IsInline:    false,
		})
	}

	for _, inline := range envelope.Inlines {
		attachments = append(attachments, ParsedAttachment{
			Filename:    inline.FileName,
			ContentType: inline.ContentType,
			ContentID:   inline.ContentID,
			Size:        int64(len(inline.Content)),
			IsInline:    true,
		})
	}

	return attachments
}