init: pristine aerc 0.20.0 source

commands/account/recover.go

@@ -0,0 +1,88 @@
+package account
+
+import (
+	"bytes"
+	"errors"
+	"io"
+	"os"
+	"path/filepath"
+
+	"git.sr.ht/~rjarry/aerc/app"
+	"git.sr.ht/~rjarry/aerc/commands"
+	"git.sr.ht/~rjarry/aerc/config"
+)
+
+type Recover struct {
+	Force  bool   `opt:"-f" desc:"Delete recovered file after opening the composer."`
+	Edit   bool   `opt:"-e" desc:"Force [compose].edit-headers = true."`
+	NoEdit bool   `opt:"-E" desc:"Force [compose].edit-headers = false."`
+	File   string `opt:"file" complete:"CompleteFile" desc:"Recover file path."`
+}
+
+func init() {
+	commands.Register(Recover{})
+}
+
+func (Recover) Description() string {
+	return "Resume composing a message that was not sent nor postponed."
+}
+
+func (Recover) Context() commands.CommandContext {
+	return commands.MESSAGE_LIST
+}
+
+func (Recover) Aliases() []string {
+	return []string{"recover"}
+}
+
+func (Recover) Options() string {
+	return "feE"
+}
+
+func (*Recover) CompleteFile(arg 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 nil
+	}
+	return commands.FilterList(files, arg, nil)
+}
+
+func (r Recover) Execute(args []string) error {
+	file, err := os.Open(r.File)
+	if err != nil {
+		return err
+	}
+	defer file.Close()
+	data, err := io.ReadAll(file)
+	if err != nil {
+		return err
+	}
+
+	acct := app.SelectedAccount()
+	if acct == nil {
+		return errors.New("No account selected")
+	}
+
+	editHeaders := (config.Compose.EditHeaders || r.Edit) && !r.NoEdit
+
+	composer, err := app.NewComposer(acct,
+		acct.AccountConfig(), acct.Worker(), editHeaders,
+		"", nil, nil, bytes.NewReader(data))
+	if err != nil {
+		return err
+	}
+	composer.Tab = app.NewTab(composer, "Recovered")
+
+	// remove file if force flag is set
+	if r.Force {
+		err = os.Remove(r.File)
+		if err != nil {
+			return err
+		}
+	}
+
+	return nil
+}