aerc/commands/msg/archive.go
2019-07-10 13:21:38 -04:00

77 lines
1.6 KiB
Go

package msg
import (
"errors"
"fmt"
"path"
"time"
"github.com/gdamore/tcell"
"git.sr.ht/~sircmpwn/aerc/widgets"
"git.sr.ht/~sircmpwn/aerc/worker/types"
)
const (
ARCHIVE_FLAT = "flat"
ARCHIVE_YEAR = "year"
ARCHIVE_MONTH = "month"
)
type Archive struct{}
func init() {
register(Archive{})
}
func (_ Archive) Aliases() []string {
return []string{"archive"}
}
func (_ Archive) Complete(aerc *widgets.Aerc, args []string) []string {
return nil
}
func (_ Archive) Execute(aerc *widgets.Aerc, args []string) error {
if len(args) != 2 {
return errors.New("Usage: archive <flat|year|month>")
}
widget := aerc.SelectedTab().(widgets.ProvidesMessage)
acct := widget.SelectedAccount()
if acct == nil {
return errors.New("No account selected")
}
msg, err := widget.SelectedMessage()
if err != nil {
return err
}
store := widget.Store()
archiveDir := acct.AccountConfig().Archive
store.Next()
acct.Messages().Scroll()
switch args[1] {
case ARCHIVE_MONTH:
archiveDir = path.Join(archiveDir,
fmt.Sprintf("%d", msg.Envelope.Date.Year()),
fmt.Sprintf("%02d", msg.Envelope.Date.Month()))
case ARCHIVE_YEAR:
archiveDir = path.Join(archiveDir, fmt.Sprintf("%v",
msg.Envelope.Date.Year()))
case ARCHIVE_FLAT:
// deliberately left blank
}
store.Move([]uint32{msg.Uid}, archiveDir, true, func(
msg types.WorkerMessage) {
switch msg := msg.(type) {
case *types.Done:
aerc.PushStatus("Messages archived.", 10*time.Second)
case *types.Error:
aerc.PushStatus(" "+msg.Error.Error(), 10*time.Second).
Color(tcell.ColorDefault, tcell.ColorRed)
}
})
return nil
}