From 2a1b998bbbb58f6990443ba885b3ee84dbcd742e Mon Sep 17 00:00:00 2001
From: emersion <contact@emersion.fr>
Date: Fri, 12 Jan 2018 19:55:08 +0100
Subject: [PATCH] imap: add support for COPY and MOVE

---
 imap/mailbox.go | 52 ++++++++++++++++++++++++++++++++++++++++++++++---
 1 file changed, 49 insertions(+), 3 deletions(-)

diff --git a/imap/mailbox.go b/imap/mailbox.go
index f207d8e..9c54dfa 100644
--- a/imap/mailbox.go
+++ b/imap/mailbox.go
@@ -8,6 +8,7 @@ import (
 	"time"
 
 	"github.com/emersion/go-imap"
+	imapbackend "github.com/emersion/go-imap/backend"
 
 	"github.com/emersion/hydroxide/imap/database"
 	"github.com/emersion/hydroxide/protonmail"
@@ -343,7 +344,11 @@ func (mbox *mailbox) SearchMessages(isUID bool, c *imap.SearchCriteria) ([]uint3
 }
 
 func (mbox *mailbox) CreateMessage(flags []string, date time.Time, body imap.Literal) error {
-	return errNotYetImplemented // TODO
+	if mbox.label != protonmail.LabelDraft {
+		return errors.New("cannot create messages outside the Drafts mailbox")
+	}
+
+	return errNotYetImplemented
 }
 
 func (mbox *mailbox) fromSeqSet(isUID bool, seqSet *imap.SeqSet) ([]string, error) {
@@ -404,8 +409,49 @@ func (mbox *mailbox) UpdateMessagesFlags(uid bool, seqSet *imap.SeqSet, op imap.
 	return nil
 }
 
-func (mbox *mailbox) CopyMessages(uid bool, seqSet *imap.SeqSet, dest string) error {
-	return errNotYetImplemented // TODO
+func (mbox *mailbox) CopyMessages(uid bool, seqSet *imap.SeqSet, destName string) error {
+	if err := mbox.init(); err != nil {
+		return err
+	}
+
+	apiIDs, err := mbox.fromSeqSet(uid, seqSet)
+	if err != nil {
+		return err
+	}
+
+	dest := mbox.u.getMailboxByLabel(destName)
+	if dest == nil {
+		return imapbackend.ErrNoSuchMailbox
+	}
+
+	if err := mbox.u.c.LabelMessages(dest.label, apiIDs); err != nil {
+		return err
+	}
+	return nil
+}
+
+func (mbox *mailbox) MoveMessages(uid bool, seqSet *imap.SeqSet, destName string) error {
+	if err := mbox.init(); err != nil {
+		return err
+	}
+
+	apiIDs, err := mbox.fromSeqSet(uid, seqSet)
+	if err != nil {
+		return err
+	}
+
+	dest := mbox.u.getMailboxByLabel(destName)
+	if dest == nil {
+		return imapbackend.ErrNoSuchMailbox
+	}
+
+	if err := mbox.u.c.LabelMessages(dest.label, apiIDs); err != nil {
+		return err
+	}
+	if err := mbox.u.c.UnlabelMessages(mbox.label, apiIDs); err != nil {
+		return err
+	}
+	return nil
 }
 
 func (mbox *mailbox) Expunge() error {