logging in tab submodel

This commit is contained in:
Erik Winter 2023-12-23 18:43:38 +01:00
parent c523577469
commit 2adf65d5bf
5 changed files with 274 additions and 188 deletions

View File

@ -8,15 +8,18 @@ import (
) )
func main() { func main() {
logger := tui.NewLogger()
p, err := tui.New(tui.Config{ p, err := tui.New(tui.Config{
TMDBAPIKey: os.Getenv("TMDB_API_KEY"), TMDBAPIKey: os.Getenv("TMDB_API_KEY"),
EMDBAPIKey: os.Getenv("EMDB_API_KEY"), EMDBAPIKey: os.Getenv("EMDB_API_KEY"),
EMDBBaseURL: "https://emdb.ewintr.nl", EMDBBaseURL: "https://emdb.ewintr.nl",
}) }, logger)
if err != nil { if err != nil {
fmt.Println(err) fmt.Println(err)
os.Exit(1) os.Exit(1)
} }
logger.SetProgram(p)
if _, err := p.Run(); err != nil { if _, err := p.Run(); err != nil {
fmt.Println(err) fmt.Println(err)
os.Exit(1) os.Exit(1)

View File

@ -0,0 +1,136 @@
package tui
import (
"fmt"
"strings"
"ewintr.nl/emdb/client"
"github.com/charmbracelet/bubbles/viewport"
tea "github.com/charmbracelet/bubbletea"
"github.com/charmbracelet/lipgloss"
)
type baseModel struct {
config Config
emdb *client.EMDB
tmdb *client.TMDB
Tabs []string
TabContent tea.Model
activeTab int
ready bool
logger *Logger
logViewport viewport.Model
windowSize tea.WindowSizeMsg
contentSize tea.WindowSizeMsg
tabSize tea.WindowSizeMsg
}
func (m baseModel) Init() tea.Cmd {
return nil
}
func (m baseModel) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
var cmd tea.Cmd
var cmds []tea.Cmd
switch msg := msg.(type) {
case tea.KeyMsg:
switch msg.String() {
case "ctrl+c", "q", "esc":
return m, tea.Quit
case "right", "tab":
m.activeTab = min(m.activeTab+1, len(m.Tabs)-1)
return m, nil
case "left", "shift+tab":
m.activeTab = max(m.activeTab-1, 0)
return m, nil
}
case tea.WindowSizeMsg:
m.windowSize = msg
if !m.ready {
m.initialModel()
}
m.Log(fmt.Sprintf("new window size: %dx%d", msg.Width, msg.Height))
m.setSize()
tabSize := TabSizeMsgType{
Width: m.contentSize.Width,
Height: m.contentSize.Height,
}
m.TabContent, cmd = m.TabContent.Update(tabSize)
cmds = append(cmds, cmd)
m.Log("done with resize")
}
m.TabContent, cmd = m.TabContent.Update(msg)
cmds = append(cmds, cmd)
m.logViewport.SetContent(strings.Join(m.logger.Lines, "\n"))
m.logViewport.GotoBottom()
m.logViewport, cmd = m.logViewport.Update(msg)
return m, tea.Batch(cmds...)
}
func (m *baseModel) Log(msg string) {
m.logger.Log(msg)
}
func (m baseModel) View() string {
if !m.ready {
return "\n Initializing..."
}
doc := strings.Builder{}
doc.WriteString(m.renderMenu())
doc.WriteString("\n")
doc.WriteString(m.renderTabContent())
doc.WriteString("\n")
doc.WriteString(m.renderLog())
return docStyle.Render(doc.String())
}
func (m *baseModel) renderMenu() string {
var items []string
for i, t := range m.Tabs {
if i == m.activeTab {
items = append(items, lipgloss.NewStyle().
Foreground(colorHighLightForeGround).
Render(fmt.Sprintf(" * %s ", t)))
continue
}
items = append(items, lipgloss.NewStyle().
Foreground(colorNormalForeground).
Render(fmt.Sprintf(" %s ", t)))
}
return lipgloss.PlaceHorizontal(m.contentSize.Width, lipgloss.Left, lipgloss.JoinHorizontal(lipgloss.Top, items...))
}
func (m *baseModel) renderTabContent() string {
content := m.TabContent.View()
return windowStyle.Width(m.contentSize.Width).Height(m.contentSize.Height).Render(content)
}
func (m *baseModel) renderLog() string {
return windowStyle.Width(m.contentSize.Width).Height(logLineCount).Render(m.logViewport.View())
}
func (m *baseModel) initialModel() {
m.logViewport = viewport.New(0, 0)
m.logViewport.KeyMap = viewport.KeyMap{}
m.setSize()
m.ready = true
}
func (m *baseModel) setSize() {
logHeight := logLineCount + docStyle.GetVerticalFrameSize()
menuHeight := 1
m.contentSize.Width = m.windowSize.Width - windowStyle.GetHorizontalFrameSize() - docStyle.GetHorizontalFrameSize()
m.contentSize.Height = m.windowSize.Height - windowStyle.GetVerticalFrameSize() - docStyle.GetVerticalFrameSize() - logHeight - menuHeight
m.logViewport.Width = m.contentSize.Width
m.logViewport.Height = logLineCount
}

View File

@ -0,0 +1,76 @@
package tui
import (
"fmt"
"github.com/charmbracelet/bubbles/list"
tea "github.com/charmbracelet/bubbletea"
)
type emdbTab struct {
ready bool
list list.Model
parent *baseModel
logger *Logger
}
func NewEMDBTab(parent *baseModel, logger *Logger) tea.Model {
m := emdbTab{}
m.parent = parent
m.logger = logger
return m
}
func (m emdbTab) Init() tea.Cmd {
return nil
}
func (m emdbTab) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
var cmd tea.Cmd
var cmds []tea.Cmd
switch msg := msg.(type) {
case TabSizeMsgType:
if !m.ready {
m.initialModel()
}
m.list.SetSize(msg.Width, msg.Height)
}
m.list, cmd = m.list.Update(msg)
cmds = append(cmds, cmd)
return m, tea.Batch(cmds...)
}
func (m emdbTab) View() string {
return m.list.View()
}
func (m *emdbTab) initialModel() {
m.list = list.New([]list.Item{}, list.NewDefaultDelegate(), 0, 0)
m.list.Title = "Movies"
m.list.SetShowHelp(false)
m.refreshMovieList()
m.ready = true
}
func (m *emdbTab) Log(s string) {
m.logger.Log(s)
}
func (m *emdbTab) refreshMovieList() {
m.Log("fetch emdb movies...")
ems, err := m.parent.emdb.GetMovies()
if err != nil {
m.Log(err.Error())
}
items := make([]list.Item, len(ems))
for i, em := range ems {
items[i] = list.Item(Movie{m: em})
}
m.list.SetItems(items)
m.Log(fmt.Sprintf("found %d movies in in emdb", len(items)))
}

View File

@ -0,0 +1,24 @@
package tui
//focused string
//searchInput textinput.Model
//searchResults list.Model
//movieList list.Model
//func (m *model) Search() {
// m.Log("start search")
// movies, err := m.tmdb.Search(m.searchInput.Value())
// if err != nil {
// m.Log(fmt.Sprintf("error: %v", err))
// return
// }
//
// m.Log(fmt.Sprintf("found %d results", len(movies)))
// items := []list.Item{}
// for _, res := range movies {
// items = append(items, Movie{m: res})
// }
//
// m.searchResults.SetItems(items)
// m.focused = "result"
//}

View File

@ -1,12 +1,7 @@
package tui package tui
import ( import (
"fmt"
"strings"
"ewintr.nl/emdb/client" "ewintr.nl/emdb/client"
"github.com/charmbracelet/bubbles/list"
"github.com/charmbracelet/bubbles/viewport"
tea "github.com/charmbracelet/bubbletea" tea "github.com/charmbracelet/bubbletea"
"github.com/charmbracelet/lipgloss" "github.com/charmbracelet/lipgloss"
"github.com/muesli/termenv" "github.com/muesli/termenv"
@ -24,10 +19,31 @@ var (
logLineCount = 5 logLineCount = 5
) )
func New(conf Config) (*tea.Program, error) { type Logger struct {
tabs := []string{"Erik's movie database", "The movie database"} p *tea.Program
tabContent := []string{"Emdb", "TMDB"} Lines []string
}
func NewLogger() *Logger {
return &Logger{
Lines: make([]string, 0),
}
}
type LogMsg tea.Msg
func (l *Logger) SetProgram(p *tea.Program) {
l.p = p
}
func (l *Logger) Log(s string) {
l.Lines = append(l.Lines, s)
}
type TabSizeMsgType tea.WindowSizeMsg
func New(conf Config, logger *Logger) (*tea.Program, error) {
tabs := []string{"Erik's movie database", "The movie database"}
tmdb, err := client.NewTMDB(conf.TMDBAPIKey) tmdb, err := client.NewTMDB(conf.TMDBAPIKey)
if err != nil { if err != nil {
return nil, err return nil, err
@ -37,180 +53,11 @@ func New(conf Config) (*tea.Program, error) {
emdb: client.NewEMDB(conf.EMDBBaseURL, conf.EMDBAPIKey), emdb: client.NewEMDB(conf.EMDBBaseURL, conf.EMDBAPIKey),
tmdb: tmdb, tmdb: tmdb,
Tabs: tabs, Tabs: tabs,
TabContent: tabContent, logger: logger,
} }
return tea.NewProgram(m, tea.WithAltScreen()), nil m.TabContent = NewEMDBTab(&m, m.logger)
}
p := tea.NewProgram(m, tea.WithAltScreen())
type baseModel struct {
config Config return p, nil
emdb *client.EMDB
tmdb *client.TMDB
Tabs []string
TabContent []string
activeTab int
//focused string
//searchInput textinput.Model
//searchResults list.Model
movieList list.Model
logContent string
ready bool
logViewport viewport.Model
windowSize tea.WindowSizeMsg
contentSize tea.WindowSizeMsg
tabSize tea.WindowSizeMsg
}
func (m baseModel) Init() tea.Cmd {
return nil
}
func (m baseModel) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
var cmd tea.Cmd
var cmds []tea.Cmd
switch msg := msg.(type) {
case tea.KeyMsg:
switch msg.String() {
case "ctrl+c", "q", "esc":
return m, tea.Quit
case "right", "tab":
m.activeTab = min(m.activeTab+1, len(m.Tabs)-1)
return m, nil
case "left", "shift+tab":
m.activeTab = max(m.activeTab-1, 0)
return m, nil
}
case tea.WindowSizeMsg:
m.windowSize = msg
if !m.ready {
m.initialModel()
}
//m.Log(fmt.Sprintf("new window size: %dx%d", msg.Width, msg.Height))
m.setSizes()
}
//switch m.focused {
//case "search":
// m.searchInput, cmd = m.searchInput.Update(msg)
//case "result":
// m.searchResults, cmd = m.searchResults.Update(msg)
//}
m.logViewport, cmd = m.logViewport.Update(msg)
cmds = append(cmds, cmd)
return m, tea.Batch(cmds...)
}
func (m *baseModel) Log(msg string) {
m.logContent = fmt.Sprintf("%s\n%s", m.logContent, msg)
m.logViewport.SetContent(m.logContent)
m.logViewport.GotoBottom()
}
//func (m *model) Search() {
// m.Log("start search")
// movies, err := m.tmdb.Search(m.searchInput.Value())
// if err != nil {
// m.Log(fmt.Sprintf("error: %v", err))
// return
// }
//
// m.Log(fmt.Sprintf("found %d results", len(movies)))
// items := []list.Item{}
// for _, res := range movies {
// items = append(items, Movie{m: res})
// }
//
// m.searchResults.SetItems(items)
// m.focused = "result"
//}
func (m baseModel) View() string {
if !m.ready {
return "\n Initializing..."
}
doc := strings.Builder{}
doc.WriteString(m.renderMenu())
doc.WriteString("\n")
doc.WriteString(m.renderTabContent())
doc.WriteString("\n")
doc.WriteString(m.renderLog())
return docStyle.Render(doc.String())
}
func (m *baseModel) renderMenu() string {
var items []string
for i, t := range m.Tabs {
if i == m.activeTab {
items = append(items, lipgloss.NewStyle().
Foreground(colorHighLightForeGround).
Render(fmt.Sprintf(" * %s ", t)))
continue
}
items = append(items, lipgloss.NewStyle().
Foreground(colorNormalForeground).
Render(fmt.Sprintf(" %s ", t)))
}
return lipgloss.PlaceHorizontal(m.contentSize.Width, lipgloss.Left, lipgloss.JoinHorizontal(lipgloss.Top, items...))
}
func (m *baseModel) renderTabContent() string {
content := m.TabContent[m.activeTab]
switch m.activeTab {
case 0:
content = m.movieList.View()
case 1:
content = "tmdb"
}
return windowStyle.Width(m.contentSize.Width).Height(m.contentSize.Height).Render(content)
}
func (m *baseModel) renderLog() string {
return windowStyle.Width(m.contentSize.Width).Height(logLineCount).Render(m.logViewport.View())
}
func (m *baseModel) initialModel() {
m.movieList = list.New([]list.Item{}, list.NewDefaultDelegate(), 0, 0)
m.movieList.Title = "Movies"
m.movieList.SetShowHelp(false)
m.logViewport = viewport.New(0, 0)
m.logViewport.KeyMap = viewport.KeyMap{}
m.setSizes()
m.refreshMovieList()
m.ready = true
}
func (m *baseModel) setSizes() {
logHeight := logLineCount + docStyle.GetVerticalFrameSize()
menuHeight := 1
m.contentSize.Width = m.windowSize.Width - windowStyle.GetHorizontalFrameSize() - docStyle.GetHorizontalFrameSize()
m.contentSize.Height = m.windowSize.Height - windowStyle.GetVerticalFrameSize() - docStyle.GetVerticalFrameSize() - logHeight - menuHeight
m.movieList.SetSize(m.contentSize.Width, m.contentSize.Height)
m.logViewport.Width = m.contentSize.Width
m.logViewport.Height = logLineCount
}
func (m *baseModel) refreshMovieList() {
m.Log("fetch emdb movies...")
ems, err := m.emdb.GetMovies()
if err != nil {
m.Log(err.Error())
}
items := make([]list.Item, len(ems))
for i, em := range ems {
items[i] = list.Item(Movie{m: em})
}
m.movieList.SetItems(items)
m.Log(fmt.Sprintf("found %d movies in in emdb", len(items)))
} }