init: pristine aerc 0.20.0 source

2026-04-07 19:54:54 -04:00
commit 083402a548
502 changed files with 68722 additions and 0 deletions
@@ -0,0 +1,189 @@
+package account
+
+import (
+	"bytes"
+	"compress/gzip"
+	"errors"
+	"fmt"
+	"io"
+	"net/http"
+	"os"
+	"regexp"
+	"sync/atomic"
+	"time"
+
+	"git.sr.ht/~rjarry/aerc/app"
+	"git.sr.ht/~rjarry/aerc/commands"
+	"git.sr.ht/~rjarry/aerc/lib/log"
+	"git.sr.ht/~rjarry/aerc/lib/xdg"
+	"git.sr.ht/~rjarry/aerc/models"
+	mboxer "git.sr.ht/~rjarry/aerc/worker/mbox"
+	"git.sr.ht/~rjarry/aerc/worker/types"
+)
+
+type ImportMbox struct {
+	Path string `opt:"path" complete:"CompleteFilename" desc:"Input file path or URL."`
+}
+
+func init() {
+	commands.Register(ImportMbox{})
+}
+
+func (ImportMbox) Description() string {
+	return "Import all messages from an (gzipped) mbox file to the current folder."
+}
+
+func (ImportMbox) Context() commands.CommandContext {
+	return commands.MESSAGE_LIST
+}
+
+func (ImportMbox) Aliases() []string {
+	return []string{"import-mbox"}
+}
+
+func (*ImportMbox) CompleteFilename(arg string) []string {
+	return commands.CompletePath(arg, false)
+}
+
+func (i ImportMbox) Execute(args []string) error {
+	acct := app.SelectedAccount()
+	if acct == nil {
+		return errors.New("No account selected")
+	}
+	store := acct.Store()
+	if store == nil {
+		return errors.New("No message store selected")
+	}
+
+	folder := acct.SelectedDirectory()
+	if folder == "" {
+		return errors.New("No directory selected")
+	}
+
+	importFolder := func(r io.ReadCloser) {
+		defer log.PanicHandler()
+		defer r.Close()
+
+		messages, err := mboxer.Read(r)
+		if err != nil {
+			app.PushError(err.Error())
+			return
+		}
+
+		var appended uint32
+		for i, m := range messages {
+			done := make(chan bool)
+			var retries int = 4
+			for retries > 0 {
+				var buf bytes.Buffer
+				r, err := m.NewReader()
+				if err != nil {
+					log.Errorf("could not get reader for uid %d", m.UID())
+					break
+				}
+				nbytes, _ := io.Copy(&buf, r)
+				store.Append(
+					folder,
+					models.SeenFlag,
+					time.Now(),
+					&buf,
+					int(nbytes),
+					func(msg types.WorkerMessage) {
+						switch msg := msg.(type) {
+						case *types.Unsupported:
+							errMsg := fmt.Sprintf("%s: AppendMessage is unsupported", args[0])
+							log.Errorf(errMsg)
+							app.PushError(errMsg)
+							return
+						case *types.Error:
+							log.Errorf("AppendMessage failed: %v", msg.Error)
+							done <- false
+						case *types.Done:
+							atomic.AddUint32(&appended, 1)
+							done <- true
+						}
+					},
+				)
+
+				select {
+				case ok := <-done:
+					if ok {
+						retries = 0
+					} else {
+						// error encountered; try to append again after a quick nap
+						retries -= 1
+						sleeping := time.Duration((5 - retries) * 1e9)
+
+						log.Debugf("sleeping for %s before append message %d again", sleeping, i)
+						time.Sleep(sleeping)
+					}
+				case <-time.After(30 * time.Second):
+					log.Warnf("timed-out; appended %d of %d", appended, len(messages))
+					return
+				}
+			}
+		}
+		infoStr := fmt.Sprintf("%s: imported %d of %d successfully.", args[0], appended, len(messages))
+		log.Debugf(infoStr)
+		app.PushSuccess(infoStr)
+	}
+
+	var buf []byte
+
+	path := i.Path
+	if ok, err := regexp.MatchString("^(http[s]\\:|www\\.)", path); ok && err == nil {
+		resp, err := http.Get(path)
+		if err != nil {
+			return err
+		}
+		buf, err = io.ReadAll(resp.Body)
+		_ = resp.Body.Close()
+		if err != nil {
+			return err
+		}
+	} else {
+		path = xdg.ExpandHome(path)
+		buf, err = os.ReadFile(path)
+		if err != nil {
+			return err
+		}
+	}
+
+	var r io.ReadCloser
+
+	// detect gzip format compressed files as specified in RFC 1952
+	if len(buf) >= 2 && buf[0] == 0x1f && buf[1] == 0x8b {
+		var err error
+		r, err = gzip.NewReader(bytes.NewReader(buf))
+		if err != nil {
+			return err
+		}
+	} else {
+		r = io.NopCloser(bytes.NewReader(buf))
+	}
+
+	statusInfo := fmt.Sprintln("Importing", path, "to folder", folder)
+	app.PushStatus(statusInfo, 10*time.Second)
+	log.Debugf(statusInfo)
+
+	if len(store.Uids()) > 0 {
+		confirm := app.NewSelectorDialog(
+			"Selected directory is not empty",
+			fmt.Sprintf("Import mbox file to %s anyways?", folder),
+			[]string{"No", "Yes"}, 0, app.SelectedAccountUiConfig(),
+			func(option string, err error) {
+				app.CloseDialog()
+				if option == "Yes" {
+					go importFolder(r)
+				} else {
+					_ = r.Close()
+				}
+			},
+		)
+		app.AddDialog(confirm)
+	} else {
+		go importFolder(r)
+	}
+
+	return nil
+}