aerc/commands/msg/archive.go

114 lines
2.3 KiB
Go
Raw Normal View History

package msg
import (
"errors"
"fmt"
"path"
2019-12-18 06:34:02 +01:00
"sync"
"time"
2019-12-18 06:34:02 +01:00
"git.sr.ht/~sircmpwn/aerc/commands"
"git.sr.ht/~sircmpwn/aerc/models"
"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 {
valid := []string{"flat", "year", "month"}
return commands.CompletionFromList(valid, args)
}
func (Archive) Execute(aerc *widgets.Aerc, args []string) error {
if len(args) != 2 {
return errors.New("Usage: archive <flat|year|month>")
}
2019-12-18 06:34:02 +01:00
h := newHelper(aerc)
acct, err := h.account()
if err != nil {
return err
}
2019-12-18 06:34:02 +01:00
store, err := h.store()
if err != nil {
return err
}
2019-12-18 06:34:02 +01:00
msgs, err := h.messages()
if err != nil {
return err
}
archiveDir := acct.AccountConfig().Archive
store.Next()
acct.Messages().Invalidate()
2019-12-18 06:34:02 +01:00
var uidMap map[string][]uint32
switch args[1] {
case ARCHIVE_MONTH:
2019-12-18 06:34:02 +01:00
uidMap = groupBy(msgs, func(msg *models.MessageInfo) string {
dir := path.Join(archiveDir,
fmt.Sprintf("%d", msg.Envelope.Date.Year()),
fmt.Sprintf("%02d", msg.Envelope.Date.Month()))
return dir
})
case ARCHIVE_YEAR:
2019-12-18 06:34:02 +01:00
uidMap = groupBy(msgs, func(msg *models.MessageInfo) string {
dir := path.Join(archiveDir, fmt.Sprintf("%v",
msg.Envelope.Date.Year()))
return dir
})
case ARCHIVE_FLAT:
2019-12-18 06:34:02 +01:00
uidMap = make(map[string][]uint32)
uidMap[archiveDir] = commands.UidsFromMessageInfos(msgs)
}
2019-12-18 06:34:02 +01:00
var wg sync.WaitGroup
wg.Add(len(uidMap))
success := true
2019-12-18 06:34:02 +01:00
for dir, uids := range uidMap {
store.Move(uids, dir, true, func(
msg types.WorkerMessage) {
switch msg := msg.(type) {
case *types.Done:
wg.Done()
case *types.Error:
aerc.PushError(" " + msg.Error.Error())
2019-12-18 06:34:02 +01:00
success = false
wg.Done()
}
})
}
// we need to do that in the background, else we block the main thread
go func() {
wg.Wait()
if success {
aerc.PushStatus("Messages archived.", 10*time.Second)
}
2019-12-18 06:34:02 +01:00
}()
return nil
}
2019-12-18 06:34:02 +01:00
func groupBy(msgs []*models.MessageInfo,
grouper func(*models.MessageInfo) string) map[string][]uint32 {
m := make(map[string][]uint32)
for _, msg := range msgs {
group := grouper(msg)
m[group] = append(m[group], msg.Uid)
}
return m
}