From b36972ce711bf2822d039f3ab54d48f6ced4c2a8 Mon Sep 17 00:00:00 2001
From: Leander Beernaert <leander.beernaert@proton.ch>
Date: Mon, 17 Oct 2022 16:01:27 +0200
Subject: [PATCH] GODT-1650: Implement Connector.CreateMessage

---
 internal/user/imap.go | 54 ++++++++++++++++++++++++++++++++++++++++---
 internal/user/sync.go |  3 ++-
 2 files changed, 53 insertions(+), 4 deletions(-)

diff --git a/internal/user/imap.go b/internal/user/imap.go
index 04e19c08..8d937c23 100644
--- a/internal/user/imap.go
+++ b/internal/user/imap.go
@@ -20,6 +20,9 @@ package user
 import (
 	"context"
 	"fmt"
+	"github.com/ProtonMail/gopenpgp/v2/crypto"
+	"github.com/google/uuid"
+	"github.com/sirupsen/logrus"
 	"time"
 
 	"github.com/ProtonMail/gluon/imap"
@@ -206,11 +209,56 @@ func (conn *imapConnector) CreateMessage(
 	ctx context.Context,
 	labelID imap.LabelID,
 	literal []byte,
-	parsedMessage *imap.ParsedMessage,
 	flags imap.FlagSet,
 	date time.Time,
-) (imap.Message, error) {
-	return imap.Message{}, ErrNotImplemented
+) (imap.Message, []byte, error) {
+
+	var msgFlags liteapi.MessageFlag
+
+	switch labelID {
+	case liteapi.SentLabel:
+		msgFlags |= liteapi.MessageFlagSent
+
+	default:
+		msgFlags |= liteapi.MessageFlagReceived
+	}
+
+	var importResult liteapi.ImportRes
+	if err := conn.withAddrKR(conn.addrID, func(ring *crypto.KeyRing) error {
+		requestName := uuid.NewString()
+
+		importReq := []liteapi.ImportReq{{
+			Name: requestName,
+			Metadata: liteapi.ImportMetadata{
+				AddressID: conn.addrID,
+				LabelIDs:  []string{string(labelID)},
+				Flags:     msgFlags,
+			},
+			Message: literal,
+		}}
+
+		r, err := conn.client.ImportMessages(ctx, ring, importReq)
+		if err != nil {
+			return err
+		}
+
+		importResult = r[requestName]
+
+		return nil
+	}); err != nil {
+		return imap.Message{}, nil, err
+	}
+
+	if importResult.Code != liteapi.SuccessCode {
+		logrus.Errorf("Failed to import message: %v", importResult.Message)
+		return imap.Message{}, nil, fmt.Errorf("failed to create message: %08x", importResult.Code)
+	}
+
+	return imap.Message{
+		ID:    imap.MessageID(importResult.MessageID),
+		Flags: flags,
+		Date:  date,
+	}, literal, nil
 }
 
 // LabelMessages labels the given messages with the given label ID.
diff --git a/internal/user/sync.go b/internal/user/sync.go
index c6549b8d..e4257c3f 100644
--- a/internal/user/sync.go
+++ b/internal/user/sync.go
@@ -21,6 +21,7 @@ import (
 	"context"
 	"errors"
 	"fmt"
+	"runtime"
 	"strings"
 	"time"
 
@@ -163,7 +164,7 @@ func syncMessages(
 
 	// Fetch and build each message.
 	buildCh := stream.Map(
-		client.GetFullMessages(ctx, messageIDs...),
+		client.GetFullMessages(ctx, runtime.NumCPU(), runtime.NumCPU(), messageIDs...),
 		func(ctx context.Context, full liteapi.FullMessage) (*buildRes, error) {
 			return buildRFC822(ctx, full, addrKRs[full.AddressID])
 		},