plugins/base: replace MessageRenderData.PartPath with Part

3 files changed, 41 insertions, 14 deletions

diff --git a/plugins/base/imap.go b/plugins/base/imap.go
index eadb098..177fefc 100755
--- a/plugins/base/imap.go
+++ b/plugins/base/imap.go
@@ -180,6 +180,33 @@ func (msg *IMAPMessage) Attachments() []IMAPPartNode {
 	return attachments
 }
 
+func pathsEqual(a, b []int) bool {
+	if len(a) != len(b) {
+		return false
+	}
+	for i := range a {
+		if a[i] != b[i] {
+			return false
+		}
+	}
+	return true
+}
+
+func (msg *IMAPMessage) PartByPath(path []int) *IMAPPartNode {
+	if msg.BodyStructure == nil {
+		return nil
+	}
+
+	var result *IMAPPartNode
+	msg.BodyStructure.Walk(func(p []int, part *imap.BodyStructure) bool {
+		if result == nil && pathsEqual(path, p) {
+			result = newIMAPPartNode(msg, p, part)
+		}
+		return result == nil
+	})
+	return result
+}
+
 func (msg *IMAPMessage) PartByID(id string) *IMAPPartNode {
 	if msg.BodyStructure == nil || id == "" {
 		return nil
diff --git a/plugins/base/public/message.html b/plugins/base/public/message.html
index 5457627..f019acf 100644
--- a/plugins/base/public/message.html
+++ b/plugins/base/public/message.html
@@ -16,7 +16,7 @@
   {{end}}
 </h2>
 
-<form method="post" action="{{.Message.Uid}}/move">
+<form method="post" action="{{.Message.URL}}/move">
   <label for="move-to">Move to:</label>
   <select name="to" id="move-to">
     {{range .Mailboxes}}
@@ -26,12 +26,12 @@
   <input type="submit" value="Move">
 </form>
 
-<form method="post" action="{{.Message.Uid}}/delete">
+<form method="post" action="{{.Message.URL}}/delete">
   <input type="submit" value="Delete">
 </form>
 
 {{if .Flags}}
-  <form method="post" action="{{.Message.Uid}}/flag">
+  <form method="post" action="{{.Message.URL}}/flag">
     <p>Flags:</p>
     {{range $name, $has := .Flags}}
       {{if ismutableflag $name}}
@@ -85,14 +85,14 @@
   {{with index . 1}}
     <a
       {{if .IsText}}
-        href="{{$.Message.Uid}}?part={{.PathString}}"
+        href="{{$.Message.URL}}?part={{.PathString}}"
       {{else}}
-        href="{{$.Message.Uid}}/raw?part={{.PathString}}"
+        href="{{$.Message.URL}}/raw?part={{.PathString}}"
       {{end}}
     >
-      {{if eq $.PartPath .PathString}}<strong>{{end}}
+      {{if eq $.Part.PathString .PathString}}<strong>{{end}}
       {{.String}}
-      {{if eq $.PartPath .PathString}}</strong>{{end}}
+      {{if eq $.Part.PathString .PathString}}</strong>{{end}}
     </a>
     {{if .Children}}
     <ul>
@@ -113,15 +113,15 @@
 {{if .View}}
   <p>
     {{if .Message.HasFlag "\\Draft"}}
-      <a href="{{.Message.Uid}}/edit?part={{.PartPath}}">Edit draft</a>
+      <a href="{{.Message.URL}}/edit?part={{.Part.PathString}}">Edit draft</a>
     {{else}}
-      <a href="{{.Message.Uid}}/reply?part={{.PartPath}}">Reply</a>
+      <a href="{{.Message.URL}}/reply?part={{.Part.PathString}}">Reply</a>
     {{end}}
   </p>
   {{.View}}
 {{else}}
   <p>Can't preview this message part.</p>
-  <a href="{{.Message.Uid}}/raw?part={{.PartPath}}">Download</a>
+  <a href="{{.Message.URL}}/raw?part={{.Part.PathString}}">Download</a>
 {{end}}
 
 {{template "foot.html"}}
diff --git a/plugins/base/routes.go b/plugins/base/routes.go
index ad4d121..4503fb9 100644
--- a/plugins/base/routes.go
+++ b/plugins/base/routes.go
@@ -176,8 +176,8 @@ type MessageRenderData struct {
 	Mailboxes   []*imap.MailboxInfo
 	Mailbox     *imap.MailboxStatus
 	Message     *IMAPMessage
+	Part        *IMAPPartNode
 	View        interface{}
-	PartPath    string
 	MailboxPage int
 	Flags       map[string]bool
 }
@@ -187,8 +187,7 @@ func handleGetPart(ctx *koushin.Context, raw bool) error {
 	if err != nil {
 		return echo.NewHTTPError(http.StatusBadRequest, err)
 	}
-	partPathString := ctx.QueryParam("part")
-	partPath, err := parsePartPath(partPathString)
+	partPath, err := parsePartPath(ctx.QueryParam("part"))
 	if err != nil {
 		return echo.NewHTTPError(http.StatusBadRequest, err)
 	}
@@ -273,8 +272,8 @@ func handleGetPart(ctx *koushin.Context, raw bool) error {
 		Mailboxes:      mailboxes,
 		Mailbox:        mbox,
 		Message:        msg,
+		Part:           msg.PartByPath(partPath),
 		View:           view,
-		PartPath:       partPathString,
 		MailboxPage:    int(mbox.Messages-msg.SeqNum) / messagesPerPage,
 		Flags:          flags,
 	})
@@ -431,6 +430,7 @@ func handleCompose(ctx *koushin.Context, msg *OutgoingMessage, draft *messagePat
 
 func handleComposeNew(ctx *koushin.Context) error {
 	// These are common mailto URL query parameters
+	// TODO: cc, bcc
 	return handleCompose(ctx, &OutgoingMessage{
 		To:        strings.Split(ctx.QueryParam("to"), ","),
 		Subject:   ctx.QueryParam("subject"),