213 lines
3.9 KiB
Go
213 lines
3.9 KiB
Go
package widgets
|
|
|
|
import (
|
|
"log"
|
|
"sync/atomic"
|
|
|
|
"github.com/gdamore/tcell"
|
|
|
|
"git.sr.ht/~sircmpwn/aerc2/config"
|
|
"git.sr.ht/~sircmpwn/aerc2/lib"
|
|
"git.sr.ht/~sircmpwn/aerc2/lib/ui"
|
|
"git.sr.ht/~sircmpwn/aerc2/worker/types"
|
|
)
|
|
|
|
type MessageList struct {
|
|
ui.Invalidatable
|
|
conf *config.AercConfig
|
|
logger *log.Logger
|
|
height int
|
|
scroll int
|
|
selected int
|
|
spinner *Spinner
|
|
store atomic.Value // *lib.MessageStore
|
|
}
|
|
|
|
// TODO: fish in config
|
|
func NewMessageList(logger *log.Logger) *MessageList {
|
|
ml := &MessageList{
|
|
logger: logger,
|
|
selected: 0,
|
|
spinner: NewSpinner(),
|
|
}
|
|
ml.store.Store((*lib.MessageStore)(nil))
|
|
ml.spinner.OnInvalidate(func(_ ui.Drawable) {
|
|
ml.Invalidate()
|
|
})
|
|
// TODO: stop spinner, probably
|
|
ml.spinner.Start()
|
|
return ml
|
|
}
|
|
|
|
func (ml *MessageList) Invalidate() {
|
|
ml.DoInvalidate(ml)
|
|
}
|
|
|
|
func (ml *MessageList) Draw(ctx *ui.Context) {
|
|
ml.height = ctx.Height()
|
|
ctx.Fill(0, 0, ctx.Width(), ctx.Height(), ' ', tcell.StyleDefault)
|
|
|
|
store := ml.Store()
|
|
if store == nil {
|
|
ml.spinner.Draw(ctx)
|
|
return
|
|
}
|
|
|
|
store.Lock()
|
|
|
|
var (
|
|
needsHeaders []uint32
|
|
row int = 0
|
|
)
|
|
|
|
for i := len(store.Uids) - 1 - ml.scroll; i >= 0; i-- {
|
|
uid := store.Uids[i]
|
|
msg := store.Messages[uid]
|
|
|
|
if row >= ctx.Height() {
|
|
break
|
|
}
|
|
|
|
if msg == nil {
|
|
needsHeaders = append(needsHeaders, uid)
|
|
ml.spinner.Draw(ctx.Subcontext(0, row, ctx.Width(), 1))
|
|
row += 1
|
|
continue
|
|
}
|
|
|
|
style := tcell.StyleDefault
|
|
if row == ml.selected-ml.scroll {
|
|
style = style.Reverse(true)
|
|
}
|
|
if _, ok := store.Deleted[msg.Uid]; ok {
|
|
style = style.Foreground(tcell.ColorGray)
|
|
}
|
|
ctx.Fill(0, row, ctx.Width(), 1, ' ', style)
|
|
ctx.Printf(0, row, style, "%s", msg.Envelope.Subject)
|
|
|
|
row += 1
|
|
}
|
|
|
|
if len(store.Uids) == 0 {
|
|
msg := "(no messages)"
|
|
ctx.Printf((ctx.Width()/2)-(len(msg)/2), 0,
|
|
tcell.StyleDefault, "%s", msg)
|
|
}
|
|
|
|
store.Unlock()
|
|
|
|
if len(needsHeaders) != 0 {
|
|
store.FetchHeaders(needsHeaders, nil)
|
|
ml.spinner.Start()
|
|
} else {
|
|
ml.spinner.Stop()
|
|
}
|
|
}
|
|
|
|
func (ml *MessageList) Height() int {
|
|
return ml.height
|
|
}
|
|
|
|
func (ml *MessageList) storeUpdate(store *lib.MessageStore) {
|
|
if ml.Store() != store {
|
|
return
|
|
}
|
|
|
|
store.Lock()
|
|
if len(store.Uids) > 0 {
|
|
for ml.selected >= len(store.Uids) {
|
|
ml.Prev()
|
|
}
|
|
}
|
|
store.Unlock()
|
|
|
|
ml.Invalidate()
|
|
}
|
|
|
|
func (ml *MessageList) SetStore(store *lib.MessageStore) {
|
|
if ml.Store() == store {
|
|
ml.scroll = 0
|
|
ml.selected = 0
|
|
}
|
|
ml.store.Store(store)
|
|
if store != nil {
|
|
ml.spinner.Stop()
|
|
store.OnUpdate(ml.storeUpdate)
|
|
} else {
|
|
ml.spinner.Start()
|
|
}
|
|
ml.Invalidate()
|
|
}
|
|
|
|
func (ml *MessageList) Store() *lib.MessageStore {
|
|
return ml.store.Load().(*lib.MessageStore)
|
|
}
|
|
|
|
func (ml *MessageList) Empty() bool {
|
|
store := ml.Store()
|
|
store.Lock()
|
|
defer store.Unlock()
|
|
|
|
return store == nil || len(store.Uids) == 0
|
|
}
|
|
|
|
func (ml *MessageList) Selected() *types.MessageInfo {
|
|
store := ml.Store()
|
|
store.Lock()
|
|
defer store.Unlock()
|
|
|
|
return store.Messages[store.Uids[len(store.Uids)-ml.selected-1]]
|
|
}
|
|
|
|
func (ml *MessageList) Select(index int) {
|
|
store := ml.Store()
|
|
store.Lock()
|
|
defer store.Unlock()
|
|
|
|
ml.selected = index
|
|
for ; ml.selected < 0; ml.selected = len(store.Uids) + ml.selected {
|
|
}
|
|
if ml.selected > len(store.Uids) {
|
|
ml.selected = len(store.Uids)
|
|
}
|
|
// I'm too lazy to do the math right now
|
|
for ml.selected-ml.scroll >= ml.Height() {
|
|
ml.scroll += 1
|
|
}
|
|
for ml.selected-ml.scroll < 0 {
|
|
ml.scroll -= 1
|
|
}
|
|
}
|
|
|
|
func (ml *MessageList) nextPrev(delta int) {
|
|
store := ml.Store()
|
|
store.Lock()
|
|
defer store.Unlock()
|
|
|
|
if store == nil || len(store.Uids) == 0 {
|
|
return
|
|
}
|
|
ml.selected += delta
|
|
if ml.selected < 0 {
|
|
ml.selected = 0
|
|
}
|
|
if ml.selected >= len(store.Uids) {
|
|
ml.selected = len(store.Uids) - 1
|
|
}
|
|
if ml.Height() != 0 {
|
|
if ml.selected-ml.scroll >= ml.Height() {
|
|
ml.scroll += 1
|
|
} else if ml.selected-ml.scroll < 0 {
|
|
ml.scroll -= 1
|
|
}
|
|
}
|
|
ml.Invalidate()
|
|
}
|
|
|
|
func (ml *MessageList) Next() {
|
|
ml.nextPrev(1)
|
|
}
|
|
|
|
func (ml *MessageList) Prev() {
|
|
ml.nextPrev(-1)
|
|
}
|