Fix!: The ability to get Content-Type value from attachments (Inbound Parse)

helpers/inbound/README.md

@@ -24,9 +24,9 @@ func inboundHandler(response http.ResponseWriter, request *http.Request) {
 
 	fmt.Print(parsedEmail.Headers["From"])
 
-    for filename, contents := range parsedEmail.Attachments {
+    for _, file := range parsedEmail.Attachments {
         // Do something with an attachment
-        handleAttachment(filename, contents)
+        handleAttachment(file.Filename, file.Content)
     }
 
     for section, body := range parsedEmail.Body {

helpers/inbound/inbound.go

@@ -7,22 +7,29 @@ import (
 	"mime"
 	"mime/multipart"
 	"net/http"
+	"net/textproto"
 	"strings"
 )
 
+type ParsedAttachment struct {
+	Headers		textproto.MIMEHeader
+	ContentType	string
+	Filename	string
+	Content		[]byte
+}
 
 type ParsedEmail struct {
 	Headers     map[string]string
 	Body        map[string]string
-	Attachments map[string][]byte
+	Attachments []ParsedAttachment
 	rawRequest  *http.Request
 }
 
 func Parse(request *http.Request) *ParsedEmail {
 	result := ParsedEmail{
 		Headers:     make(map[string]string),
 		Body:        make(map[string]string),
-		Attachments: make(map[string][]byte),
+		Attachments: []ParsedAttachment{},
 		rawRequest:  request,
 	}
 	result.parse()
@@ -65,7 +72,16 @@ func (email *ParsedEmail) parseRawEmail(rawEmail string) {
 			}
 
 		} else if emailPart.FileName() != "" {
-			email.Attachments[emailPart.FileName()] = readBody(emailPart)
+			content := readBody(emailPart)
+
+			attachment := ParsedAttachment{
+				Filename: emailPart.FileName(),
+				ContentType: strings.Split(emailPart.Header.Get("Content-Type"), ";")[0],
+				Content: content,
+				Headers: emailPart.Header,
+			}
+
+			email.Attachments = append(email.Attachments, attachment)
 		} else {
 			header := emailPart.Header.Get("Content-Type")
 			email.Body[header] = string(readBody(emailPart))

helpers/inbound/inbound_test.go

@@ -49,6 +49,14 @@ func TestParse(t *testing.T) {
 	}
 }
 
+func TestAttachments(t *testing.T) {
+	req := createRequest("./sample_data/raw_data_with_attachments.txt")
+	email := Parse(req)
+	contentType := "image/jpeg"
+
+	assert.Equalf(t, contentType, email.Attachments[0].ContentType,"Expected From: %s, Got: %s", contentType, email.Attachments[0].ContentType)
+}
+
 func ExampleParsedEmail_parseHeaders() {
 	headers := `
 Foo: foo
@@ -57,7 +65,7 @@ Bar: baz
 	email := ParsedEmail{
 		Headers:     make(map[string]string),
 		Body:        make(map[string]string),
-		Attachments: make(map[string][]byte),
+		Attachments: []ParsedAttachment{},
 		rawRequest:  nil,
 	}
 	email.parseHeaders(headers)
@@ -89,7 +97,7 @@ Content-Transfer-Encoding: quoted-printable
 	email := ParsedEmail{
 		Headers:     make(map[string]string),
 		Body:        make(map[string]string),
-		Attachments: make(map[string][]byte),
+		Attachments: []ParsedAttachment{},
 		rawRequest:  nil,
 	}
 	email.parseRawEmail(rawEmail)
@@ -103,4 +111,4 @@ Content-Transfer-Encoding: quoted-printable
 	// Subject Test Email
 	// Content-Type multipart/mixed; boundary=TwiLIo
 	// Hello Twilio SendGrid!
-}
+}