imap: add support for section.{Specifier,Partial} when fetching messages
This commit is contained in:
parent
44f2e11a69
commit
451ac7a8f6
137
imap/mailbox.go
137
imap/mailbox.go
|
@ -239,39 +239,48 @@ func (mbox *mailbox) getBodyStructure(msg *protonmail.Message, extended bool) (*
|
||||||
}, nil
|
}, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (mbox *mailbox) getInlineSection(msg *protonmail.Message) (message.Header, io.Reader, error) {
|
func (mbox *mailbox) inlineBody(msg *protonmail.Message) (io.Reader, error) {
|
||||||
h := mail.NewTextHeader()
|
h := mail.NewTextHeader()
|
||||||
h.SetContentType(msg.MIMEType, nil)
|
h.SetContentType(msg.MIMEType, nil)
|
||||||
|
|
||||||
md, err := msg.Read(mbox.u.privateKeys, nil)
|
md, err := msg.Read(mbox.u.privateKeys, nil)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
|
|
||||||
// TODO: check signature
|
// TODO: check signature
|
||||||
return h.Header, md.UnverifiedBody, nil
|
return md.UnverifiedBody, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (mbox *mailbox) getAttachmentSection(att *protonmail.Attachment) (message.Header, io.Reader, error) {
|
func (mbox *mailbox) attachmentBody(att *protonmail.Attachment) (io.Reader, error) {
|
||||||
|
rc, err := mbox.u.c.GetAttachment(att.ID)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
md, err := att.Read(rc, mbox.u.privateKeys, nil)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
// TODO: check signature
|
||||||
|
return md.UnverifiedBody, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func inlineHeader(msg *protonmail.Message) message.Header {
|
||||||
|
h := mail.NewTextHeader()
|
||||||
|
h.SetContentType(msg.MIMEType, nil)
|
||||||
|
return h.Header
|
||||||
|
}
|
||||||
|
|
||||||
|
func attachmentHeader(att *protonmail.Attachment) message.Header {
|
||||||
h := mail.NewAttachmentHeader()
|
h := mail.NewAttachmentHeader()
|
||||||
h.SetContentType(att.MIMEType, nil)
|
h.SetContentType(att.MIMEType, nil)
|
||||||
h.SetFilename(att.Name)
|
h.SetFilename(att.Name)
|
||||||
if att.ContentID != "" {
|
if att.ContentID != "" {
|
||||||
h.Set("Content-Id", att.ContentID)
|
h.Set("Content-Id", att.ContentID)
|
||||||
}
|
}
|
||||||
|
return h.Header
|
||||||
rc, err := mbox.u.c.GetAttachment(att.ID)
|
|
||||||
if err != nil {
|
|
||||||
return nil, nil, err
|
|
||||||
}
|
|
||||||
|
|
||||||
md, err := att.Read(rc, mbox.u.privateKeys, nil)
|
|
||||||
if err != nil {
|
|
||||||
return nil, nil, err
|
|
||||||
}
|
|
||||||
|
|
||||||
// TODO: check signature
|
|
||||||
return h.Header, md.UnverifiedBody, nil
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func mailAddress(addr *protonmail.MessageAddress) *mail.Address {
|
func mailAddress(addr *protonmail.MessageAddress) *mail.Address {
|
||||||
|
@ -289,20 +298,7 @@ func mailAddressList(addresses []*protonmail.MessageAddress) []*mail.Address {
|
||||||
return l
|
return l
|
||||||
}
|
}
|
||||||
|
|
||||||
func (mbox *mailbox) getBodySection(msg *protonmail.Message, section *imap.BodySectionName) (imap.Literal, error) {
|
func messageHeader(msg *protonmail.Message) message.Header {
|
||||||
// TODO: section.Peek
|
|
||||||
// TODO: section.Partial
|
|
||||||
// TODO: section.Specifier
|
|
||||||
|
|
||||||
// TODO: only fetch if needed
|
|
||||||
msg, err := mbox.u.c.GetMessage(msg.ID)
|
|
||||||
if err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
|
|
||||||
b := new(bytes.Buffer)
|
|
||||||
|
|
||||||
if len(section.Path) == 0 {
|
|
||||||
h := mail.NewHeader()
|
h := mail.NewHeader()
|
||||||
h.SetDate(time.Unix(msg.Time, 0))
|
h.SetDate(time.Unix(msg.Time, 0))
|
||||||
h.SetSubject(msg.Subject)
|
h.SetSubject(msg.Subject)
|
||||||
|
@ -315,17 +311,36 @@ func (mbox *mailbox) getBodySection(msg *protonmail.Message, section *imap.BodyS
|
||||||
h.SetAddressList("Bcc", mailAddressList(msg.BCCList))
|
h.SetAddressList("Bcc", mailAddressList(msg.BCCList))
|
||||||
// TODO: In-Reply-To
|
// TODO: In-Reply-To
|
||||||
h.Set("Message-Id", getMessageID(msg))
|
h.Set("Message-Id", getMessageID(msg))
|
||||||
|
return h.Header
|
||||||
|
}
|
||||||
|
|
||||||
w, err := message.CreateWriter(b, h.Header)
|
func (mbox *mailbox) getBodySection(msg *protonmail.Message, section *imap.BodySectionName) (imap.Literal, error) {
|
||||||
|
// TODO: section.Peek
|
||||||
|
|
||||||
|
b := new(bytes.Buffer)
|
||||||
|
|
||||||
|
if len(section.Path) == 0 {
|
||||||
|
w, err := message.CreateWriter(b, messageHeader(msg))
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
|
|
||||||
ph, pr, err := mbox.getInlineSection(msg)
|
if section.Specifier == imap.TextSpecifier {
|
||||||
|
b.Reset()
|
||||||
|
}
|
||||||
|
|
||||||
|
switch section.Specifier {
|
||||||
|
case imap.EntireSpecifier, imap.TextSpecifier:
|
||||||
|
msg, err := mbox.u.c.GetMessage(msg.ID)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
pw, err := w.CreatePart(ph)
|
|
||||||
|
pw, err := w.CreatePart(inlineHeader(msg))
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
pr, err := mbox.inlineBody(msg)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
|
@ -335,12 +350,11 @@ func (mbox *mailbox) getBodySection(msg *protonmail.Message, section *imap.BodyS
|
||||||
pw.Close()
|
pw.Close()
|
||||||
|
|
||||||
for _, att := range msg.Attachments {
|
for _, att := range msg.Attachments {
|
||||||
ph, pr, err := mbox.getAttachmentSection(att)
|
pw, err := w.CreatePart(attachmentHeader(att))
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
|
pr, err := mbox.attachmentBody(att)
|
||||||
pw, err := w.CreatePart(ph)
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
|
@ -349,6 +363,7 @@ func (mbox *mailbox) getBodySection(msg *protonmail.Message, section *imap.BodyS
|
||||||
}
|
}
|
||||||
pw.Close()
|
pw.Close()
|
||||||
}
|
}
|
||||||
|
}
|
||||||
|
|
||||||
w.Close()
|
w.Close()
|
||||||
} else {
|
} else {
|
||||||
|
@ -357,33 +372,65 @@ func (mbox *mailbox) getBodySection(msg *protonmail.Message, section *imap.BodyS
|
||||||
}
|
}
|
||||||
|
|
||||||
var h message.Header
|
var h message.Header
|
||||||
var r io.Reader
|
var getBody func() (io.Reader, error)
|
||||||
var err error
|
if part := section.Path[0]; part == 1 {
|
||||||
part := section.Path[0]
|
h = inlineHeader(msg)
|
||||||
if part == 1 {
|
getBody = func() (io.Reader, error) {
|
||||||
h, r, err = mbox.getInlineSection(msg)
|
msg, err := mbox.u.c.GetMessage(msg.ID)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
return mbox.inlineBody(msg)
|
||||||
|
}
|
||||||
} else {
|
} else {
|
||||||
i := part - 2
|
i := part - 2
|
||||||
if i >= msg.NumAttachments {
|
if i >= msg.NumAttachments {
|
||||||
return nil, errors.New("invalid attachment section path")
|
return nil, errors.New("invalid attachment section path")
|
||||||
}
|
}
|
||||||
h, r, err = mbox.getAttachmentSection(msg.Attachments[i])
|
|
||||||
}
|
msg, err := mbox.u.c.GetMessage(msg.ID)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
|
|
||||||
|
att := msg.Attachments[i]
|
||||||
|
h = attachmentHeader(att)
|
||||||
|
getBody = func() (io.Reader, error) {
|
||||||
|
return mbox.attachmentBody(att)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
w, err := message.CreateWriter(b, h)
|
w, err := message.CreateWriter(b, h)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
|
|
||||||
|
if section.Specifier == imap.TextSpecifier {
|
||||||
|
b.Reset()
|
||||||
|
}
|
||||||
|
|
||||||
|
switch section.Specifier {
|
||||||
|
case imap.EntireSpecifier, imap.TextSpecifier:
|
||||||
|
r, err := getBody()
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
if _, err := io.Copy(w, r); err != nil {
|
if _, err := io.Copy(w, r); err != nil {
|
||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
|
}
|
||||||
|
|
||||||
w.Close()
|
w.Close()
|
||||||
}
|
}
|
||||||
|
|
||||||
return b, nil
|
var l imap.Literal = b
|
||||||
|
if section.Partial != nil {
|
||||||
|
l = bytes.NewReader(section.ExtractPartial(b.Bytes()))
|
||||||
|
}
|
||||||
|
|
||||||
|
return l, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (mbox *mailbox) getMessage(isUid bool, id uint32, items []imap.FetchItem) (*imap.Message, error) {
|
func (mbox *mailbox) getMessage(isUid bool, id uint32, items []imap.FetchItem) (*imap.Message, error) {
|
||||||
|
|
Loading…
Reference in New Issue