feat(GODT-3199): log ids for fetched body.

ProtonMail · Feb 26, 2024 · 52774c7 · 52774c7
1 parent 70bd825
commit 52774c7
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 1 deletion.
diff --git a/internal/state/mailbox.go b/internal/state/mailbox.go
@@ -42,6 +42,7 @@ type AppendOnlyMailbox interface {
 
 func newMailbox(mbox *db.Mailbox, state *State, snap *snapshot) *Mailbox {
 	id := db.NewMailboxIDPair(mbox)
+
 	return &Mailbox{
 		id:          id,
 		name:        mbox.Name,
@@ -52,7 +53,9 @@ func newMailbox(mbox *db.Mailbox, state *State, snap *snapshot) *Mailbox {
 		selected: snap != nil,
 		readOnly: state.ro,
 		snap:     snap,
-		log:      logrus.WithField("pkg", "gluon/state/mailbox").WithField("id", id),
+		log: logrus.
+			WithField("pkg", "gluon/state/mailbox").
+			WithField("mboxID", id.String()),
 	}
 }
 

diff --git a/internal/state/mailbox_fetch.go b/internal/state/mailbox_fetch.go
@@ -33,6 +33,7 @@ func (m *Mailbox) Fetch(ctx context.Context, cmd *command.Fetch, ch chan respons
 		needsLiteral bool
 		wantUID      bool
 		setSeen      bool
+		isBodyFetch  bool
 	)
 
 	for _, attribute := range cmd.Attributes {
@@ -53,11 +54,13 @@ func (m *Mailbox) Fetch(ctx context.Context, cmd *command.Fetch, ch chan respons
 		case *command.FetchAttributeRFC822:
 			setSeen = true
 			needsLiteral = true
+			isBodyFetch = true
 
 			operations = append(operations, fetchRFC822)
 		case *command.FetchAttributeRFC822Text:
 			setSeen = true
 			needsLiteral = true
+			isBodyFetch = true
 
 			operations = append(operations, fetchRFC822Text)
 		case *command.FetchAttributeRFC822Header:
@@ -78,6 +81,7 @@ func (m *Mailbox) Fetch(ctx context.Context, cmd *command.Fetch, ch chan respons
 			operations = append(operations, fetchBodyStructure)
 		case *command.FetchAttributeBodySection:
 			needsLiteral = true
+			isBodyFetch = true
 
 			if !attribute.Peek {
 				setSeen = true
@@ -162,6 +166,10 @@ func (m *Mailbox) Fetch(ctx context.Context, cmd *command.Fetch, ch chan respons
 			snapMessages[i].snapMsg = nil
 		}
 
+		if isBodyFetch {
+			m.log.WithField("UID", msg.UID).WithField("messageID", msg.ID.String()).Debug("Fetch Body")
+		}
+
 		ch <- response.Fetch(msg.Seq).WithItems(items...)
 
 		return nil