aerc/widgets/directories.go

72 lines
1.6 KiB
Go

package widgets
import (
"log"
"sort"
"github.com/gdamore/tcell"
"git.sr.ht/~sircmpwn/aerc2/config"
"git.sr.ht/~sircmpwn/aerc2/lib/ui"
"git.sr.ht/~sircmpwn/aerc2/worker/types"
)
type DirectoryList struct {
conf *config.AccountConfig
dirs []string
logger *log.Logger
onInvalidate func(d ui.Drawable)
worker *types.Worker
}
func NewDirectoryList(conf *config.AccountConfig,
logger *log.Logger, worker *types.Worker) *DirectoryList {
return &DirectoryList{conf: conf, logger: logger, worker: worker}
}
func (dirlist *DirectoryList) UpdateList() {
var dirs []string
dirlist.worker.PostAction(
&types.ListDirectories{}, func(msg types.WorkerMessage) {
switch msg := msg.(type) {
case *types.Directory:
dirs = append(dirs, msg.Name)
case *types.Done:
sort.Strings(dirs)
dirlist.dirs = dirs
dirlist.Invalidate()
}
})
}
func (dirlist *DirectoryList) OnInvalidate(onInvalidate func(d ui.Drawable)) {
dirlist.onInvalidate = onInvalidate
}
func (dirlist *DirectoryList) Invalidate() {
if dirlist.onInvalidate != nil {
dirlist.onInvalidate(dirlist)
}
}
func (dirlist *DirectoryList) Draw(ctx *ui.Context) {
ctx.Fill(0, 0, ctx.Width(), ctx.Height(), ' ', tcell.StyleDefault)
row := 0
for _, name := range dirlist.dirs {
if row >= ctx.Height() {
break
}
if len(dirlist.conf.Folders) > 1 {
idx := sort.SearchStrings(dirlist.conf.Folders, name)
if idx == len(dirlist.conf.Folders) ||
dirlist.conf.Folders[idx] != name {
continue
}
}
ctx.Printf(0, row, tcell.StyleDefault, "%s", name)
row++
}
}