mox/webops/export.go

package webops

import (
	"archive/tar"
	"archive/zip"
	"compress/gzip"
	"fmt"
	"mime"
	"net/http"
	"strconv"
	"strings"
	"time"

	"github.com/mjl-/mox/mlog"
	"github.com/mjl-/mox/store"
)

// Export is used by webmail and webaccount to export messages of one or
// multiple mailboxes, in maildir or mbox format, in a tar/tgz/zip archive or
// direct mbox.
func Export(log mlog.Log, accName string, w http.ResponseWriter, r *http.Request) {
	if r.Method != "POST" {
		http.Error(w, "405 - method not allowed - use post", http.StatusMethodNotAllowed)
		return
	}

	// We
	mailbox := r.FormValue("mailbox") // Empty means all.
	messageIDstr := r.FormValue("messageids")
	var messageIDs []int64
	if messageIDstr != "" {
		for _, s := range strings.Split(messageIDstr, ",") {
			id, err := strconv.ParseInt(s, 10, 64)
			if err != nil {
				http.Error(w, fmt.Sprintf("400 - bad request - bad message id %q: %v", s, err), http.StatusBadRequest)
				return
			}
			messageIDs = append(messageIDs, id)
		}
	}
	if mailbox != "" && len(messageIDs) > 0 {
		http.Error(w, "400 - bad request - cannot specify both mailbox and message ids", http.StatusBadRequest)
		return
	}

	format := r.FormValue("format")
	archive := r.FormValue("archive")
	recursive := r.FormValue("recursive") != ""
	switch format {
	case "maildir", "mbox":
	default:
		http.Error(w, "400 - bad request - unknown format", http.StatusBadRequest)
		return
	}
	switch archive {
	case "none", "tar", "tgz", "zip":
	default:
		http.Error(w, "400 - bad request - unknown archive", http.StatusBadRequest)
		return
	}
	if archive == "none" && (format != "mbox" || recursive) {
		http.Error(w, "400 - bad request - archive none can only be used with non-recursive mbox", http.StatusBadRequest)
		return
	}
	if len(messageIDs) > 0 && recursive {
		http.Error(w, "400 - bad request - cannot export message ids recursively", http.StatusBadRequest)
		return
	}

	acc, err := store.OpenAccount(log, accName, false)
	if err != nil {
		log.Errorx("open account for export", err)
		http.Error(w, "500 - internal server error", http.StatusInternalServerError)
		return
	}
	defer func() {
		err := acc.Close()
		log.Check(err, "closing account")
	}()

	var name string
	if mailbox != "" {
		name = "-" + strings.ReplaceAll(mailbox, "/", "-")
	} else if len(messageIDs) > 1 {
		name = "-selection"
	} else if len(messageIDs) == 0 {
		name = "-all"
	}
	filename := fmt.Sprintf("mailexport%s-%s", name, time.Now().Format("20060102-150405"))
	filename += "." + format
	var archiver store.Archiver
	if archive == "none" {
		w.Header().Set("Content-Type", "application/mbox")
		archiver = &store.MboxArchiver{Writer: w}
	} else if archive == "tar" {
		// Don't tempt browsers to "helpfully" decompress.
		w.Header().Set("Content-Type", "application/x-tar")
		archiver = store.TarArchiver{Writer: tar.NewWriter(w)}
		filename += ".tar"
	} else if archive == "tgz" {
		// Don't tempt browsers to "helpfully" decompress.
		w.Header().Set("Content-Type", "application/octet-stream")

		gzw := gzip.NewWriter(w)
		defer func() {
			_ = gzw.Close()
		}()
		archiver = store.TarArchiver{Writer: tar.NewWriter(gzw)}
		filename += ".tgz"
	} else {
		w.Header().Set("Content-Type", "application/zip")
		archiver = store.ZipArchiver{Writer: zip.NewWriter(w)}
		filename += ".zip"
	}
	defer func() {
		err := archiver.Close()
		log.Check(err, "exporting mail close")
	}()
	w.Header().Set("Content-Disposition", mime.FormatMediaType("attachment", map[string]string{"filename": filename}))
	if err := store.ExportMessages(r.Context(), log, acc.DB, acc.Dir, archiver, format == "maildir", mailbox, messageIDs, recursive); err != nil {
		log.Errorx("exporting mail", err)
	}
}