plugins/base: add redirect param to move/delete/flag endpoints

1 files changed, 10 insertions, 1 deletions

diff --git a/plugins/base/routes.go b/plugins/base/routes.go
index 0782159..1ea611d 100644
--- a/plugins/base/routes.go
+++ b/plugins/base/routes.go
@@ -696,6 +696,9 @@ func handleMove(ctx *koushin.Context) error {
 		return err
 	}
 
+	if path := ctx.QueryParam("next"); path != "" {
+		return ctx.Redirect(http.StatusFound, path)
+	}
 	return ctx.Redirect(http.StatusFound, fmt.Sprintf("/mailbox/%v", url.PathEscape(to)))
 }
 
@@ -744,6 +747,9 @@ func handleDelete(ctx *koushin.Context) error {
 		return err
 	}
 
+	if path := ctx.QueryParam("next"); path != "" {
+		return ctx.Redirect(http.StatusFound, path)
+	}
 	return ctx.Redirect(http.StatusFound, fmt.Sprintf("/mailbox/%v", url.PathEscape(mboxName)))
 }
 
@@ -813,7 +819,10 @@ func handleSetFlags(ctx *koushin.Context) error {
 		return err
 	}
 
-	if len(uids) != 1 || (op == imap.RemoveFlags && len(flags) == 1 && flags[0] == "\\Seen") {
+	if path := ctx.QueryParam("next"); path != "" {
+		return ctx.Redirect(http.StatusFound, path)
+	}
+	if len(uids) != 1 || (op == imap.RemoveFlags && len(flags) == 1 && flags[0] == imap.SeenFlag) {
 		// Redirecting to the message view would mark the message as read again
 		return ctx.Redirect(http.StatusFound, fmt.Sprintf("/mailbox/%v", url.PathEscape(mboxName)))
 	}