aerc/widgets/account.go

235 lines
5.4 KiB
Go
Raw Normal View History

package widgets
import (
"errors"
2019-01-13 19:33:43 +01:00
"fmt"
2019-01-13 19:03:28 +01:00
"log"
"github.com/gdamore/tcell"
2019-01-13 19:03:28 +01:00
2019-05-18 02:57:10 +02:00
"git.sr.ht/~sircmpwn/aerc/config"
"git.sr.ht/~sircmpwn/aerc/lib"
"git.sr.ht/~sircmpwn/aerc/lib/ui"
"git.sr.ht/~sircmpwn/aerc/models"
2019-05-18 02:57:10 +02:00
"git.sr.ht/~sircmpwn/aerc/worker"
"git.sr.ht/~sircmpwn/aerc/worker/types"
)
type AccountView struct {
acct *config.AccountConfig
conf *config.AercConfig
dirlist *DirectoryList
grid *ui.Grid
host TabHost
logger *log.Logger
msglist *MessageList
msgStores map[string]*lib.MessageStore
worker *types.Worker
}
2019-03-15 06:46:14 +01:00
func NewAccountView(conf *config.AercConfig, acct *config.AccountConfig,
logger *log.Logger, host TabHost) *AccountView {
grid := ui.NewGrid().Rows([]ui.GridSpec{
{ui.SIZE_WEIGHT, 1},
}).Columns([]ui.GridSpec{
2019-03-16 01:40:28 +01:00
{ui.SIZE_EXACT, conf.Ui.SidebarWidth},
{ui.SIZE_WEIGHT, 1},
})
2019-03-15 06:46:14 +01:00
worker, err := worker.NewWorker(acct.Source, logger)
if err != nil {
2019-05-21 01:20:20 +02:00
host.SetStatus(fmt.Sprintf("%s: %s", acct.Name, err)).
Color(tcell.ColorDefault, tcell.ColorRed)
2019-01-13 19:33:43 +01:00
return &AccountView{
acct: acct,
grid: grid,
host: host,
logger: logger,
2019-01-13 19:33:43 +01:00
}
}
2019-01-13 19:03:28 +01:00
dirlist := NewDirectoryList(acct, &conf.Ui, logger, worker)
2019-06-07 16:08:09 +02:00
if conf.Ui.SidebarWidth > 0 {
grid.AddChild(ui.NewBordered(dirlist, ui.BORDER_RIGHT))
}
2019-01-13 20:25:46 +01:00
msglist := NewMessageList(conf, logger)
grid.AddChild(msglist).At(0, 1)
2019-03-15 06:46:14 +01:00
view := &AccountView{
acct: acct,
conf: conf,
dirlist: dirlist,
grid: grid,
host: host,
logger: logger,
msglist: msglist,
msgStores: make(map[string]*lib.MessageStore),
worker: worker,
2019-01-13 19:03:28 +01:00
}
go worker.Backend.Run()
2019-03-15 06:46:14 +01:00
worker.PostAction(&types.Configure{Config: acct}, nil)
worker.PostAction(&types.Connect{}, view.connected)
host.SetStatus("Connecting...")
2019-01-13 19:03:28 +01:00
2019-03-15 06:46:14 +01:00
return view
2019-01-13 19:03:28 +01:00
}
func (acct *AccountView) Tick() bool {
2019-05-21 01:20:20 +02:00
if acct.worker == nil {
return false
}
select {
case msg := <-acct.worker.Messages:
msg = acct.worker.ProcessMessage(msg)
acct.onMessage(msg)
return true
default:
return false
}
}
func (acct *AccountView) AccountConfig() *config.AccountConfig {
return acct.acct
}
2019-05-16 18:15:34 +02:00
func (acct *AccountView) Worker() *types.Worker {
return acct.worker
}
func (acct *AccountView) Logger() *log.Logger {
return acct.logger
}
2019-03-15 03:34:34 +01:00
func (acct *AccountView) Name() string {
2019-03-15 06:46:14 +01:00
return acct.acct.Name
2019-03-15 03:34:34 +01:00
}
2019-02-10 22:46:13 +01:00
func (acct *AccountView) Children() []ui.Drawable {
return acct.grid.Children()
}
func (acct *AccountView) OnInvalidate(onInvalidate func(d ui.Drawable)) {
acct.grid.OnInvalidate(func(_ ui.Drawable) {
onInvalidate(acct)
})
}
func (acct *AccountView) Invalidate() {
acct.grid.Invalidate()
}
func (acct *AccountView) Draw(ctx *ui.Context) {
acct.grid.Draw(ctx)
}
2019-01-13 19:33:43 +01:00
func (acct *AccountView) Focus(focus bool) {
// TODO: Unfocus children I guess
2019-01-13 19:33:43 +01:00
}
2019-01-13 20:25:46 +01:00
func (acct *AccountView) connected(msg types.WorkerMessage) {
switch msg.(type) {
2019-01-13 20:25:46 +01:00
case *types.Done:
acct.host.SetStatus("Listing mailboxes...")
acct.logger.Println("Listing mailboxes...")
acct.dirlist.UpdateList(func(dirs []string) {
var dir string
for _, _dir := range dirs {
2019-03-16 02:33:08 +01:00
if _dir == acct.acct.Default {
dir = _dir
break
}
}
2019-06-12 08:31:51 +02:00
if dir == "" && len(dirs) > 0 {
dir = dirs[0]
}
if dir != "" {
acct.dirlist.Select(dir)
}
acct.logger.Println("Connected.")
acct.host.SetStatus("Connected.")
})
2019-01-13 20:25:46 +01:00
}
}
2019-03-11 02:15:24 +01:00
func (acct *AccountView) Directories() *DirectoryList {
return acct.dirlist
}
2019-03-15 04:41:25 +01:00
func (acct *AccountView) Messages() *MessageList {
return acct.msglist
}
func (acct *AccountView) Store() *lib.MessageStore {
return acct.msglist.Store()
}
2019-07-05 18:21:12 +02:00
func (acct *AccountView) SelectedAccount() *AccountView {
return acct
}
func (acct *AccountView) SelectedMessage() (*models.MessageInfo, error) {
if len(acct.msglist.Store().Uids()) == 0 {
return nil, errors.New("no message selected")
}
return acct.msglist.Selected(), nil
}
2019-07-05 18:21:12 +02:00
func (acct *AccountView) SelectedMessagePart() *PartInfo {
return nil
}
2019-03-11 02:15:24 +01:00
func (acct *AccountView) onMessage(msg types.WorkerMessage) {
switch msg := msg.(type) {
case *types.Done:
switch msg.InResponseTo().(type) {
case *types.OpenDirectory:
if store, ok := acct.msgStores[acct.dirlist.selected]; ok {
// If we've opened this dir before, we can re-render it from
// memory while we wait for the update and the UI feels
// snappier. If not, we'll unset the store and show the spinner
// while we download the UID list.
acct.msglist.SetStore(store)
} else {
acct.msglist.SetStore(nil)
}
case *types.CreateDirectory:
acct.dirlist.UpdateList(nil)
}
case *types.DirectoryInfo:
if store, ok := acct.msgStores[msg.Info.Name]; ok {
store.Update(msg)
} else {
store = lib.NewMessageStore(acct.worker, msg.Info)
acct.msgStores[msg.Info.Name] = store
2019-05-14 02:16:55 +02:00
store.OnUpdate(func(_ *lib.MessageStore) {
store.OnUpdate(nil)
acct.msglist.SetStore(store)
})
}
case *types.DirectoryContents:
if store, ok := acct.msgStores[acct.dirlist.selected]; ok {
store.Update(msg)
}
2019-03-31 18:35:51 +02:00
case *types.FullMessage:
if store, ok := acct.msgStores[acct.dirlist.selected]; ok {
store.Update(msg)
}
case *types.MessageInfo:
if store, ok := acct.msgStores[acct.dirlist.selected]; ok {
store.Update(msg)
}
2019-03-21 04:23:38 +01:00
case *types.MessagesDeleted:
if store, ok := acct.msgStores[acct.dirlist.selected]; ok {
store.Update(msg)
}
case *types.Error:
acct.logger.Printf("%v", msg.Error)
acct.host.SetStatus(fmt.Sprintf("%v", msg.Error)).
2019-03-30 18:05:00 +01:00
Color(tcell.ColorDefault, tcell.ColorRed)
}
2019-03-11 02:15:24 +01:00
}