event: Use pointers for InboundMessage addresses (#447)

* event: Use pointers for InboundMessage addresses

To ease conversions to/from MessageMetadata

Signed-off-by: James Hillyerd <james@hillyerd.com>

* message: test StoreManager.MailboxForAddress()

Signed-off-by: James Hillyerd <james@hillyerd.com>

---------

Signed-off-by: James Hillyerd <james@hillyerd.com>

pkg/message/manager.go

@@ -57,20 +57,24 @@ func (s *StoreManager) Deliver(
 	if err != nil {
 		return err
 	}
-	fromaddr, err := enmime.ParseAddressList(header.Get("From"))
-	if err != nil || len(fromaddr) == 0 {
-		fromaddr = make([]*mail.Address, 1)
-		fromaddr[0] = &from.Address
+
+	fromAddrs, err := enmime.ParseAddressList(header.Get("From"))
+	if err != nil || len(fromAddrs) == 0 {
+		// Failed to parse From header, use SMTP MAIL FROM instead.
+		fromAddrs = make([]*mail.Address, 1)
+		fromAddrs[0] = &from.Address
 	}
-	toaddr, err := enmime.ParseAddressList(header.Get("To"))
+
+	toAddrs, err := enmime.ParseAddressList(header.Get("To"))
 	if err != nil {
-		toaddr = make([]*mail.Address, len(recipients))
+		// Failed to parse To header, use SMTP RCPT TO instead.
+		toAddrs = make([]*mail.Address, len(recipients))
 		for i, torecip := range recipients {
-			toaddr[i] = &torecip.Address
+			toAddrs[i] = &torecip.Address
 		}
 	}
-	subject := header.Get("Subject")
 
+	subject := header.Get("Subject")
 	now := time.Now()
 	tstamp := now.UTC().Format(recvdTimeFmt)
 
@@ -79,14 +83,11 @@ func (s *StoreManager) Deliver(
 	for i, recip := range recipients {
 		mailboxes[i] = recip.Mailbox
 	}
-	toAddrs := make([]mail.Address, len(toaddr))
-	for i, addr := range toaddr {
-		toAddrs[i] = *addr
-	}
 
+	// Construct InboundMessage event and process through extensions.
 	inbound := &event.InboundMessage{
 		Mailboxes: mailboxes,
-		From:      *fromaddr[0],
+		From:      fromAddrs[0],
 		To:        toAddrs,
 		Subject:   subject,
 		Size:      int64(len(source)),
@@ -105,13 +106,9 @@ func (s *StoreManager) Deliver(
 	} else {
 		// Event response overrides destination mailboxes and address policy.
 		inbound = extResult
-		toaddr = make([]*mail.Address, len(inbound.To))
-		for i := range inbound.To {
-			toaddr[i] = &inbound.To[i]
-		}
 	}
 
-	// Deliver to mailboxes.
+	// Deliver to each mailbox.
 	for _, mb := range inbound.Mailboxes {
 		// Append recipient and timestamp to generated Recieved header.
 		recvd := fmt.Sprintf("%s  for <%s>; %s\r\n", recvdHeader, mb, tstamp)
@@ -121,8 +118,8 @@ func (s *StoreManager) Deliver(
 		delivery := &Delivery{
 			Meta: event.MessageMetadata{
 				Mailbox: mb,
-				From:    &inbound.From,
-				To:      toaddr,
+				From:    inbound.From,
+				To:      inbound.To,
 				Date:    now,
 				Subject: inbound.Subject,
 			},

pkg/message/manager_test.go

@@ -137,8 +137,8 @@ test email`),
 
 	require.NotNil(t, got, "BeforeMessageStored listener did not receive InboundMessage")
 	assert.Equal(t, []string{"u1@example.com", "u2@example.com"}, got.Mailboxes, "Mailboxes not equal")
-	assert.Equal(t, mail.Address{Name: "", Address: "from@example.com"}, got.From, "From not equal")
-	assert.Equal(t, []mail.Address{
+	assert.Equal(t, &mail.Address{Name: "", Address: "from@example.com"}, got.From, "From not equal")
+	assert.Equal(t, []*mail.Address{
 		{Name: "", Address: "u1@example.com"},
 		{Name: "", Address: "u3@external.com"},
 	}, got.To, "To not equal")
@@ -173,8 +173,8 @@ func TestDeliverEmitsBeforeMessageStoredEventRcptTo(t *testing.T) {
 
 	require.NotNil(t, got, "BeforeMessageStored listener did not receive InboundMessage")
 	assert.Equal(t, []string{"u1@example.com", "u2@example.com"}, got.Mailboxes, "Mailboxes not equal")
-	assert.Equal(t, mail.Address{Name: "", Address: "from@example.com"}, got.From, "From not equal")
-	assert.Equal(t, []mail.Address{
+	assert.Equal(t, &mail.Address{Name: "", Address: "from@example.com"}, got.From, "From not equal")
+	assert.Equal(t, []*mail.Address{
 		{Name: "", Address: "u1@example.com"},
 		{Name: "", Address: "u2@example.com"},
 	}, got.To, "To not equal")
@@ -256,10 +256,10 @@ func TestDeliverUsesBeforeMessageStoredEventResponseFields(t *testing.T) {
 		func(msg event.InboundMessage) *event.InboundMessage {
 			// Listener rewrites destination mailboxes.
 			msg.Subject = "event subj"
-			msg.From = mail.Address{Address: "from@event.com", Name: "From Event"}
+			msg.From = &mail.Address{Address: "from@event.com", Name: "From Event"}
 
 			// Changing To does not affect destination mailbox(es).
-			msg.To = []mail.Address{
+			msg.To = []*mail.Address{
 				{Address: "to@event.com", Name: "To Event"},
 				{Address: "to2@event.com", Name: "To 2 Event"},
 			}
@@ -479,6 +479,17 @@ test email`
 	assert.Contains(t, got, msgSource, "Source should contain original message source")
 }
 
+func TestMailboxForAddress(t *testing.T) {
+	// Configured for FullNaming.
+	sm, _ := testStoreManager()
+
+	addr := "u1@example.com"
+	got, err := sm.MailboxForAddress(addr)
+	require.NoError(t, err)
+
+	assert.Equal(t, addr, got, "FullNaming mode should return a full address for mailbox")
+}
+
 // Returns an empty StoreManager and extension Host pair, configured for testing.
 func testStoreManager() (*message.StoreManager, *extension.Host) {
 	extHost := extension.NewHost()