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" "git.sr.ht/~sircmpwn/getopt" ) type Recover struct{} func init() { register(Recover{}) } func (Recover) Aliases() []string { return []string{"recover"} } func (Recover) Options() string { return "feE" } func (r Recover) Complete(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 nil } return commands.CompletionFromList(files, commands.Operands(args, r.Options())) } func (r Recover) Execute(args []string) error { // Complete() expects to be passed only the arguments, not including the command name if len(Recover{}.Complete(args[1:])) == 0 { return errors.New("No messages to recover.") } force := false editHeaders := config.Compose.EditHeaders opts, optind, err := getopt.Getopts(args, r.Options()) if err != nil { return err } for _, opt := range opts { switch opt.Option { case 'f': force = true case 'e': editHeaders = true case 'E': editHeaders = false } } if len(args) <= optind { return errors.New("Usage: recover [-f] [-E|-e] ") } acct := app.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 := io.ReadAll(recoverFile) if err != nil { return nil, err } return data, nil } data, err := readData() if err != nil { return err } 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 force { err = os.Remove(args[optind]) if err != nil { return err } } return nil }