136 lines
3.2 KiB
Go
136 lines
3.2 KiB
Go
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 {
|
|
emdb *client.EMDB
|
|
tmdb *client.TMDB
|
|
tabs *TabSet
|
|
activeTab int
|
|
initialized bool
|
|
logger *Logger
|
|
logViewport viewport.Model
|
|
windowSize tea.WindowSizeMsg
|
|
contentSize tea.WindowSizeMsg
|
|
tabSize tea.WindowSizeMsg
|
|
}
|
|
|
|
func NewBaseModel(emdb *client.EMDB, tmdb *client.TMDB, logger *Logger) (tea.Model, tea.Cmd) {
|
|
logViewport := viewport.New(0, 0)
|
|
logViewport.KeyMap = viewport.KeyMap{}
|
|
|
|
m := baseModel{
|
|
emdb: emdb,
|
|
tmdb: tmdb,
|
|
tabs: NewTabSet(),
|
|
logViewport: logViewport,
|
|
logger: logger,
|
|
}
|
|
m.setSize()
|
|
|
|
return m, nil
|
|
}
|
|
|
|
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.tabs.Next()
|
|
case "left", "shift+tab":
|
|
m.tabs.Previous()
|
|
default:
|
|
cmds = append(cmds, m.tabs.Update(msg))
|
|
}
|
|
case tea.WindowSizeMsg:
|
|
m.windowSize = msg
|
|
if !m.initialized {
|
|
var emdbTab, tmdbTab tea.Model
|
|
emdbTab, cmd = NewTabEMDB(m.emdb, m.logger)
|
|
cmds = append(cmds, cmd)
|
|
tmdbTab, cmd = NewTabTMDB(m.tmdb, m.logger)
|
|
cmds = append(cmds, cmd)
|
|
m.tabs.AddTab("emdb", "EMDB", emdbTab)
|
|
m.tabs.AddTab("tmdb", "TMDB", tmdbTab)
|
|
m.initialized = true
|
|
}
|
|
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,
|
|
}
|
|
cmds = append(cmds, m.tabs.Update(tabSize))
|
|
m.Log("done with resize")
|
|
default:
|
|
cmds = append(cmds, m.tabs.Update(msg))
|
|
}
|
|
|
|
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.initialized {
|
|
return "\n Initializing..."
|
|
}
|
|
|
|
doc := strings.Builder{}
|
|
doc.WriteString(lipgloss.PlaceHorizontal(m.contentSize.Width, lipgloss.Left, m.tabs.ViewMenu()))
|
|
doc.WriteString("\n")
|
|
doc.WriteString(m.tabs.ViewContent())
|
|
doc.WriteString("\n")
|
|
doc.WriteString(m.renderLog())
|
|
return docStyle.Render(doc.String())
|
|
}
|
|
|
|
func (m *baseModel) newTab() (tea.Model, tea.Cmd) {
|
|
switch m.activeTab {
|
|
case 0:
|
|
return NewTabEMDB(m.emdb, m.logger)
|
|
case 1:
|
|
return NewTabTMDB(m.tmdb, m.logger)
|
|
default:
|
|
return nil, nil
|
|
}
|
|
}
|
|
|
|
func (m *baseModel) renderLog() string {
|
|
return windowStyle.Width(m.contentSize.Width).Height(logLineCount).Render(m.logViewport.View())
|
|
}
|
|
|
|
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
|
|
}
|