library/cmd/manage/main.go

230 lines
4.6 KiB
Go
Raw Normal View History

2021-06-26 18:08:35 +00:00
package main
2021-07-03 00:46:58 +00:00
import (
2021-07-07 01:40:12 +00:00
"context"
2021-07-03 01:44:45 +00:00
"fmt"
2021-07-03 00:46:58 +00:00
"log"
2021-07-07 01:40:12 +00:00
"os"
"sync"
2021-07-03 00:46:58 +00:00
2021-07-07 01:40:12 +00:00
"git.yetaga.in/alazyreader/library/book"
2021-07-03 00:46:58 +00:00
"git.yetaga.in/alazyreader/library/config"
2021-07-07 01:40:12 +00:00
"git.yetaga.in/alazyreader/library/database"
2021-07-03 17:30:08 +00:00
"git.yetaga.in/alazyreader/library/ui"
2021-07-03 00:46:58 +00:00
"github.com/gdamore/tcell"
"github.com/kelseyhightower/envconfig"
)
2021-07-07 01:40:12 +00:00
// State holds the UI state keys=>value map and manages access to the map with a mutex
type State struct {
m sync.Mutex
stateMap map[string]interface{}
}
// key, present
func (s *State) Get(key string) interface{} {
s.m.Lock()
defer s.m.Unlock()
if s.stateMap == nil {
s.stateMap = make(map[string]interface{})
}
k, ok := s.stateMap[key]
if !ok {
return nil
}
return k
}
// key, value
func (s *State) Set(key string, value interface{}) {
s.m.Lock()
defer s.m.Unlock()
if s.stateMap == nil {
s.stateMap = make(map[string]interface{})
}
s.stateMap[key] = value
}
const (
IN_MENU = iota
IN_BOOK
)
2021-07-07 01:40:12 +00:00
type EventBookUpdate struct {
tcell.EventTime
book *book.Book
}
func NewEventBookUpdate(b *book.Book) *EventBookUpdate {
e := &EventBookUpdate{book: b}
e.SetEventNow()
return e
}
func (e *EventBookUpdate) Book() *book.Book {
return e.book
}
type EventLoadBook struct {
tcell.EventTime
ID int
}
func NewEventLoadBook(id int) *EventLoadBook {
e := &EventLoadBook{ID: id}
e.SetEventNow()
return e
}
2021-06-26 18:08:35 +00:00
func main() {
2021-07-03 00:46:58 +00:00
var c config.Config
err := envconfig.Process("library", &c)
if err != nil {
log.Fatalln(err)
}
2021-07-07 01:40:12 +00:00
// create state
state := State{}
// set up DB connection
if c.DBUser == "" || c.DBPass == "" || c.DBHost == "" || c.DBPort == "" || c.DBName == "" {
log.Fatalf("vars: %+v", c)
}
lib, err := database.NewMySQLConnection(c.DBUser, c.DBPass, c.DBHost, c.DBPort, c.DBName)
if err != nil {
log.Fatalln(err)
}
err = lib.PrepareDatabase(context.Background())
if err != nil {
log.Fatalln(err)
}
_, _, err = lib.RunMigrations(context.Background())
if err != nil {
log.Fatalln(err)
}
books, err := lib.GetAllBooks(context.Background())
if err != nil {
log.Fatalln(err)
}
state.Set("library", books)
2021-07-03 00:46:58 +00:00
screen, err := tcell.NewScreen()
if err != nil {
log.Fatalln(err)
}
err = screen.Init()
if err != nil {
log.Fatalln(err)
}
2021-07-03 14:58:45 +00:00
2021-07-07 01:40:12 +00:00
l := ui.NewList(Titles(state.Get("library").([]book.Book)), 0)
2021-07-03 17:30:08 +00:00
menu := ui.NewBox(
2021-07-03 14:58:45 +00:00
"library",
2021-07-07 01:40:12 +00:00
[]string{"(n)ew", "(i)mport", "(q)uit"},
2021-07-03 17:30:08 +00:00
ui.Contents{{
Offsets: ui.Offsets{Top: 1, Left: 2, Bottom: -2, Right: -2},
Container: l,
2021-07-03 14:58:45 +00:00
}},
)
2021-07-07 01:40:12 +00:00
book := ui.NewBookDetails(&book.Book{
Title: "test title",
})
2021-07-03 17:30:08 +00:00
activeBook := ui.NewBox(
"book",
2021-07-07 01:40:12 +00:00
nil,
ui.Contents{{
2021-07-07 01:40:12 +00:00
Offsets: ui.Offsets{Top: 1, Left: 2, Bottom: 0, Right: 0},
Container: book,
}},
2021-07-03 17:30:08 +00:00
)
container := ui.NewContainer(
ui.Contents{
{Container: menu, Offsets: ui.Offsets{Percent: 1}},
{Container: activeBook, Offsets: ui.Offsets{Percent: 2}},
},
ui.LayoutHorizontalPercent,
2021-07-03 17:30:08 +00:00
)
2021-07-03 14:58:45 +00:00
// init
2021-07-03 00:46:58 +00:00
screen.Clear()
2021-07-03 01:44:45 +00:00
w, h := screen.Size()
2021-07-03 17:30:08 +00:00
container.SetSize(0, 0, h, w)
container.Draw(screen)
2021-07-03 14:58:45 +00:00
screen.Sync()
2021-07-07 01:40:12 +00:00
// init UI state
state.Set("ui_state", IN_MENU)
2021-07-03 14:58:45 +00:00
// UI loop
2021-07-03 00:46:58 +00:00
for {
e := screen.PollEvent()
switch v := e.(type) {
case *tcell.EventError:
2021-07-07 01:40:12 +00:00
fmt.Fprintf(os.Stderr, "%v", v)
2021-07-03 00:46:58 +00:00
screen.Beep()
2021-07-03 14:58:45 +00:00
case *tcell.EventKey: // input handling
2021-07-07 01:40:12 +00:00
curr := state.Get("ui_state").(int)
if curr == IN_MENU {
if v.Key() == tcell.KeyUp && l.Selected() > 0 {
l.SetSelected(l.Selected() - 1)
}
if v.Key() == tcell.KeyDown && l.Selected() < len(l.ListMembers())-1 {
l.SetSelected(l.Selected() + 1)
}
if v.Rune() == 'q' {
screen.Fini()
fmt.Printf("Thank you for playing Wing Commander!\n\n")
return
}
if v.Key() == tcell.KeyRight {
2021-07-07 01:40:12 +00:00
screen.PostEvent(NewEventLoadBook(l.SelectedID()))
state.Set("ui_state", IN_BOOK)
}
2021-07-07 01:40:12 +00:00
} else if curr == IN_BOOK {
if v.Key() == tcell.KeyLeft {
2021-07-07 01:40:12 +00:00
state.Set("ui_state", IN_MENU)
}
2021-07-03 00:46:58 +00:00
}
2021-07-03 01:44:45 +00:00
screen.Clear()
2021-07-03 17:30:08 +00:00
container.Draw(screen)
2021-07-03 14:58:45 +00:00
case *tcell.EventResize: // screen redraw
2021-07-03 01:44:45 +00:00
w, h := screen.Size()
2021-07-03 17:30:08 +00:00
container.SetSize(0, 0, h, w)
2021-07-03 14:58:45 +00:00
screen.Clear()
2021-07-03 17:30:08 +00:00
container.Draw(screen)
2021-07-07 01:40:12 +00:00
case *EventBookUpdate:
// TK
case *EventLoadBook:
book.SetBook(GetBookByID(v.ID, books))
screen.Clear()
container.Draw(screen)
2021-07-03 01:44:45 +00:00
case *tcell.EventInterrupt:
case *tcell.EventMouse:
2021-07-03 00:46:58 +00:00
case *tcell.EventTime:
2021-07-03 01:44:45 +00:00
default:
2021-07-03 00:46:58 +00:00
}
2021-07-03 17:30:08 +00:00
screen.Show() // repaint
2021-07-03 01:44:45 +00:00
}
2021-06-26 18:08:35 +00:00
}
2021-07-07 01:40:12 +00:00
func Titles(lb []book.Book) []ui.ListKeyValue {
r := []ui.ListKeyValue{}
for i := range lb {
r = append(r, ui.ListKeyValue{
Key: lb[i].ID,
Value: lb[i].Title,
})
}
return r
}
func GetBookByID(id int, lb []book.Book) *book.Book {
for i := range lb {
if lb[i].ID == id {
return &lb[i]
}
}
return &book.Book{}
}