aerc/widgets/dirlist.go

208 lines
4.7 KiB
Go
Raw Normal View History

2019-01-13 20:25:46 +01:00
package widgets
import (
"log"
"regexp"
2019-01-13 20:25:46 +01:00
"sort"
"github.com/gdamore/tcell"
2019-05-18 02:57:10 +02:00
"git.sr.ht/~sircmpwn/aerc/config"
"git.sr.ht/~sircmpwn/aerc/lib"
2019-05-18 02:57:10 +02:00
"git.sr.ht/~sircmpwn/aerc/lib/ui"
"git.sr.ht/~sircmpwn/aerc/worker/types"
2019-01-13 20:25:46 +01:00
)
type DirectoryList struct {
ui.Invalidatable
acctConf *config.AccountConfig
uiConf *config.UIConfig
store *lib.DirStore
dirs []string
logger *log.Logger
selecting string
selected string
spinner *Spinner
worker *types.Worker
2019-01-13 20:25:46 +01:00
}
func NewDirectoryList(acctConf *config.AccountConfig, uiConf *config.UIConfig,
logger *log.Logger, worker *types.Worker) *DirectoryList {
2019-01-14 02:02:21 +01:00
dirlist := &DirectoryList{
acctConf: acctConf,
uiConf: uiConf,
logger: logger,
spinner: NewSpinner(),
store: lib.NewDirStore(),
worker: worker,
2019-01-14 02:02:21 +01:00
}
dirlist.spinner.OnInvalidate(func(_ ui.Drawable) {
dirlist.Invalidate()
})
dirlist.spinner.Start()
return dirlist
2019-01-13 20:25:46 +01:00
}
func (dirlist *DirectoryList) List() []string {
return dirlist.store.List()
}
func (dirlist *DirectoryList) UpdateList(done func(dirs []string)) {
2019-07-04 18:31:27 +02:00
// TODO: move this logic into dirstore
var dirs []string
2019-01-13 20:25:46 +01:00
dirlist.worker.PostAction(
&types.ListDirectories{}, func(msg types.WorkerMessage) {
switch msg := msg.(type) {
case *types.Directory:
dirs = append(dirs, msg.Dir.Name)
2019-01-13 20:25:46 +01:00
case *types.Done:
sort.Strings(dirs)
dirlist.store.Update(dirs)
dirlist.filterDirsByFoldersConfig()
2019-01-14 02:02:21 +01:00
dirlist.spinner.Stop()
dirlist.Invalidate()
if done != nil {
done(dirs)
}
2019-01-13 20:25:46 +01:00
}
})
}
func (dirlist *DirectoryList) Select(name string) {
2019-03-15 03:34:34 +01:00
dirlist.selecting = name
dirlist.worker.PostAction(&types.OpenDirectory{Directory: name},
func(msg types.WorkerMessage) {
switch msg.(type) {
case *types.Error:
dirlist.selecting = ""
case *types.Done:
2019-03-15 03:34:34 +01:00
dirlist.selected = dirlist.selecting
2019-06-12 08:31:51 +02:00
dirlist.filterDirsByFoldersConfig()
hasSelected := false
for _, d := range dirlist.dirs {
2019-06-12 08:31:51 +02:00
if d == dirlist.selected {
hasSelected = true
break
}
}
if !hasSelected && dirlist.selected != "" {
dirlist.dirs = append(dirlist.dirs, dirlist.selected)
2019-06-12 08:31:51 +02:00
}
sort.Strings(dirlist.dirs)
2019-03-15 03:34:34 +01:00
}
dirlist.Invalidate()
})
dirlist.Invalidate()
}
2019-03-15 03:34:34 +01:00
func (dirlist *DirectoryList) Selected() string {
return dirlist.selected
}
2019-01-13 20:25:46 +01:00
func (dirlist *DirectoryList) Invalidate() {
dirlist.DoInvalidate(dirlist)
2019-01-13 20:25:46 +01:00
}
func (dirlist *DirectoryList) Draw(ctx *ui.Context) {
ctx.Fill(0, 0, ctx.Width(), ctx.Height(), ' ', tcell.StyleDefault)
2019-01-14 02:02:21 +01:00
if dirlist.spinner.IsRunning() {
dirlist.spinner.Draw(ctx)
return
}
if len(dirlist.dirs) == 0 {
style := tcell.StyleDefault
ctx.Printf(0, 0, style, dirlist.uiConf.EmptyDirlist)
return
}
row := 0
for _, name := range dirlist.dirs {
if row >= ctx.Height() {
break
}
style := tcell.StyleDefault
if name == dirlist.selected {
style = style.Reverse(true)
} else if name == dirlist.selecting {
style = style.Reverse(true)
style = style.Foreground(tcell.ColorGray)
}
ctx.Fill(0, row, ctx.Width(), 1, ' ', style)
ctx.Printf(0, row, style, "%s", name)
row++
}
2019-01-13 20:25:46 +01:00
}
2019-03-11 02:15:24 +01:00
func (dirlist *DirectoryList) NextPrev(delta int) {
curIdx := sort.SearchStrings(dirlist.dirs, dirlist.selected)
if curIdx == len(dirlist.dirs) {
return
}
newIdx := curIdx + delta
ndirs := len(dirlist.dirs)
if newIdx < 0 {
newIdx = ndirs - 1
} else if newIdx >= ndirs {
newIdx = 0
2019-03-11 02:15:24 +01:00
}
dirlist.Select(dirlist.dirs[newIdx])
2019-03-11 02:15:24 +01:00
}
func (dirlist *DirectoryList) Next() {
dirlist.NextPrev(1)
2019-03-11 02:15:24 +01:00
}
func (dirlist *DirectoryList) Prev() {
dirlist.NextPrev(-1)
2019-03-11 02:15:24 +01:00
}
2019-06-12 08:31:51 +02:00
func folderMatches(folder string, pattern string) bool {
if len(pattern) == 0 {
return false
}
if pattern[0] == '~' {
r, err := regexp.Compile(pattern[1:])
if err != nil {
return false
}
return r.Match([]byte(folder))
}
return pattern == folder
}
// filterDirsByFoldersConfig sets dirlist.dirs to the filtered subset of the
// dirstore, based on the AccountConfig.Folders option
2019-06-12 08:31:51 +02:00
func (dirlist *DirectoryList) filterDirsByFoldersConfig() {
dirlist.dirs = dirlist.store.List()
2019-06-12 08:31:51 +02:00
// config option defaults to show all if unset
if len(dirlist.acctConf.Folders) == 0 {
2019-06-12 08:31:51 +02:00
return
}
var filtered []string
for _, folder := range dirlist.dirs {
for _, cfgfolder := range dirlist.acctConf.Folders {
if folderMatches(folder, cfgfolder) {
2019-06-12 08:31:51 +02:00
filtered = append(filtered, folder)
break
}
}
}
dirlist.dirs = filtered
2019-06-12 08:31:51 +02:00
}
func (dirlist *DirectoryList) SelectedMsgStore() (*lib.MessageStore, bool) {
return dirlist.store.MessageStore(dirlist.selected)
}
func (dirlist *DirectoryList) MsgStore(name string) (*lib.MessageStore, bool) {
return dirlist.store.MessageStore(name)
}
func (dirlist *DirectoryList) SetMsgStore(name string, msgStore *lib.MessageStore) {
dirlist.store.SetMessageStore(name, msgStore)
}