aboutsummaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
authorKoni Marti <koni.marti@gmail.com>2022-01-28 22:50:02 +0100
committerRobin Jarry <robin@jarry.cc>2022-01-29 22:01:25 +0100
commitcdec23323c644b1805ca9cca138e7fa91be6ab6d (patch)
tree906d3ba137afe22a7eb496a5fab856045279178e
parentfc07495ee889f45e1df9ef928bde5410d01eb90d (diff)
downloadaerc-cdec23323c644b1805ca9cca138e7fa91be6ab6d.tar.gz
recover: recover emails from tempdir after a crash
implements a recover command that searches the local temp dir for aerc emails. If a file is found and selected, a new composer tab will open with the file contents as the message body. No header data is stored in the temp file and thus this information cannot be recovered. Recover will not remove the temporary file unless the force flag (-f) is explicitly used. This recovery method only works when the editor buffer is saved to disk and the Close() function of the composer has not been called yet. Sending, postponing or quitting will call the Close() function which removes the temporary file completely. After Close() is called, no recovery is possible anymore. Signed-off-by: Koni Marti <koni.marti@gmail.com>
-rw-r--r--commands/account/recover.go110
1 files changed, 110 insertions, 0 deletions
diff --git a/commands/account/recover.go b/commands/account/recover.go
new file mode 100644
index 00000000..a167d500
--- /dev/null
+++ b/commands/account/recover.go
@@ -0,0 +1,110 @@
+package account
+
+import (
+ "bytes"
+ "errors"
+ "io/ioutil"
+ "os"
+ "path/filepath"
+ "strings"
+
+ "git.sr.ht/~rjarry/aerc/models"
+ "git.sr.ht/~rjarry/aerc/widgets"
+ "git.sr.ht/~sircmpwn/getopt"
+)
+
+type Recover struct{}
+
+func init() {
+ register(Recover{})
+}
+
+func (Recover) Aliases() []string {
+ return []string{"recover"}
+}
+
+func (Recover) Complete(aerc *widgets.Aerc, args []string) []string {
+ // file name of temp file is hard-coded in the NewComposer() function
+ files, err := filepath.Glob(
+ filepath.Join(os.TempDir(), "aerc-compose-*.eml"),
+ )
+ if err != nil {
+ return []string{}
+ }
+ arg := strings.Join(args, " ")
+ if arg != "" {
+ for i, file := range files {
+ files[i] = strings.Join([]string{arg, file}, " ")
+ }
+ }
+ return files
+}
+
+func (Recover) Execute(aerc *widgets.Aerc, args []string) error {
+ if len(Recover{}.Complete(aerc, args)) == 0 {
+ return errors.New("No messages to recover.")
+ }
+
+ force := false
+
+ opts, optind, err := getopt.Getopts(args, "f")
+ if err != nil {
+ return err
+ }
+ for _, opt := range opts {
+ switch opt.Option {
+ case 'f':
+ force = true
+ }
+ }
+
+ if len(args) <= optind {
+ return errors.New("Usage: recover [-f] <file>")
+ }
+
+ acct := aerc.SelectedAccount()
+ if acct == nil {
+ return errors.New("No account selected")
+ }
+
+ readData := func() ([]byte, error) {
+ recoverFile, err := os.Open(args[optind])
+ if err != nil {
+ return nil, err
+ }
+ defer recoverFile.Close()
+ data, err := ioutil.ReadAll(recoverFile)
+ if err != nil {
+ return nil, err
+ }
+ return data, nil
+ }
+ data, err := readData()
+ if err != nil {
+ return err
+ }
+
+ composer, err := widgets.NewComposer(aerc, acct,
+ aerc.Config(), acct.AccountConfig(), acct.Worker(),
+ "", nil, models.OriginalMail{})
+ if err != nil {
+ return err
+ }
+
+ tab := aerc.NewTab(composer, "Recovered")
+ composer.OnHeaderChange("Subject", func(subject string) {
+ tab.Name = subject
+ tab.Content.Invalidate()
+ })
+ go composer.AppendContents(bytes.NewReader(data))
+
+ // remove file if force flag is set
+ if force {
+ err = os.Remove(args[optind])
+ if err != nil {
+ return err
+ }
+ }
+
+ return nil
+}