6838c23478
Signed-off-by: Wagner Riffel <wgrriffel@gmail.com>
80 lines
1.7 KiB
Go
80 lines
1.7 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")
|
|
}
|
|
store := widget.Store()
|
|
if store == nil {
|
|
return errors.New("Cannot perform action. Messages still loading")
|
|
}
|
|
msg, err := widget.SelectedMessage()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
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
|
|
}
|