105 lines
2.2 KiB
Go
105 lines
2.2 KiB
Go
package ui
|
|
|
|
import (
|
|
"log"
|
|
|
|
tb "github.com/nsf/termbox-go"
|
|
|
|
"git.sr.ht/~sircmpwn/aerc2/config"
|
|
"git.sr.ht/~sircmpwn/aerc2/worker"
|
|
"git.sr.ht/~sircmpwn/aerc2/worker/types"
|
|
)
|
|
|
|
type AccountTab struct {
|
|
Config *config.AccountConfig
|
|
Worker worker.Worker
|
|
Parent *UIState
|
|
logger *log.Logger
|
|
counter int
|
|
callbacks map[types.WorkerMessage]func(msg types.WorkerMessage)
|
|
}
|
|
|
|
func NewAccountTab(conf *config.AccountConfig,
|
|
logger *log.Logger) (*AccountTab, error) {
|
|
|
|
work, err := worker.NewWorker(conf.Source, logger)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
go work.Run()
|
|
acc := &AccountTab{
|
|
Config: conf,
|
|
Worker: work,
|
|
logger: logger,
|
|
callbacks: make(map[types.WorkerMessage]func(msg types.WorkerMessage)),
|
|
}
|
|
acc.postAction(types.Configure{Config: conf}, nil)
|
|
acc.postAction(types.Connect{}, func(msg types.WorkerMessage) {
|
|
if _, ok := msg.(types.Ack); ok {
|
|
acc.logger.Println("Connected.")
|
|
} else {
|
|
acc.logger.Println("Connection failed.")
|
|
}
|
|
})
|
|
return acc, nil
|
|
}
|
|
|
|
func (acc *AccountTab) Name() string {
|
|
return acc.Config.Name
|
|
}
|
|
|
|
func (acc *AccountTab) SetParent(parent *UIState) {
|
|
acc.Parent = parent
|
|
}
|
|
|
|
func (acc *AccountTab) Render(at Geometry) {
|
|
cell := tb.Cell{
|
|
Ch: ' ',
|
|
Fg: tb.ColorDefault,
|
|
Bg: tb.ColorDefault,
|
|
}
|
|
TFill(at, cell)
|
|
TPrintf(&at, cell, "%s %d\n", acc.Name(), acc.counter)
|
|
acc.counter++
|
|
if acc.counter%10000 == 0 {
|
|
acc.counter = 0
|
|
}
|
|
acc.Parent.InvalidateFrom(acc)
|
|
}
|
|
|
|
func (acc *AccountTab) GetChannel() chan types.WorkerMessage {
|
|
return acc.Worker.GetMessages()
|
|
}
|
|
|
|
func (acc *AccountTab) postAction(msg types.WorkerMessage,
|
|
cb func(msg types.WorkerMessage)) {
|
|
|
|
acc.logger.Printf("-> %T\n", msg)
|
|
acc.Worker.PostAction(msg)
|
|
if cb != nil {
|
|
acc.callbacks[msg] = cb
|
|
delete(acc.callbacks, msg)
|
|
}
|
|
}
|
|
|
|
func (acc *AccountTab) HandleMessage(msg types.WorkerMessage) {
|
|
acc.logger.Printf("<- %T\n", msg)
|
|
if cb, ok := acc.callbacks[msg.InResponseTo()]; ok {
|
|
cb(msg)
|
|
}
|
|
switch msg.(type) {
|
|
case types.Ack:
|
|
// no-op
|
|
case types.ApproveCertificate:
|
|
// TODO: Ask the user
|
|
acc.logger.Println("Approving certificate")
|
|
acc.postAction(types.Ack{
|
|
Message: types.RespondTo(msg),
|
|
}, nil)
|
|
default:
|
|
acc.postAction(types.Unsupported{
|
|
Message: types.RespondTo(msg),
|
|
}, nil)
|
|
}
|
|
}
|