emdb/cmd/terminal-client/tui/tabtmdb.go

package tui

import (
	"fmt"

	"ewintr.nl/emdb/client"
	"ewintr.nl/emdb/cmd/api-service/job"
	"github.com/charmbracelet/bubbles/list"
	"github.com/charmbracelet/bubbles/textinput"
	tea "github.com/charmbracelet/bubbletea"
)

type tabTMDB struct {
	emdb          *client.EMDB
	tmdb          *client.TMDB
	initialized   bool
	focused       string
	searchInput   textinput.Model
	searchResults list.Model
	logger        *Logger
}

func NewTabTMDB(emdb *client.EMDB, tmdb *client.TMDB, logger *Logger) (tea.Model, tea.Cmd) {
	m := tabTMDB{
		emdb:   emdb,
		tmdb:   tmdb,
		logger: logger,
	}

	return m, nil
}

func (m tabTMDB) Init() tea.Cmd {
	return nil
}

func (m tabTMDB) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
	var cmd tea.Cmd
	var cmds []tea.Cmd

	switch msg := msg.(type) {
	case TabSizeMsg:
		if !m.initialized {
			m.initialModel(msg.Width, msg.Height-2)
		}
		m.initialized = true
		m.searchResults.SetSize(msg.Width, msg.Height-2)
	case TabResetMsg:
		m.searchInput.SetValue("")
		m.searchResults.SetItems([]list.Item{})
		m.searchInput.Focus()
		m.focused = "search"
	case tea.KeyMsg:
		switch msg.String() {
		case "ctrl+c", "esc":
			return m, tea.Quit
		case "q":
			if m.focused == "result" {
				return m, tea.Quit
			}
		case "right", "tab":
			cmds = append(cmds, SelectNextTab(), m.ResetCmd())
		case "left", "shift+tab":
			cmds = append(cmds, SelectPrevTab(), m.ResetCmd())
		case "enter":
			if m.focused == "search" {
				cmds = append(cmds, m.SearchTMDBCmd(m.searchInput.Value()))
				m.searchInput.Blur()
				m.Log("search tmdb...")
			}
		case "i":
			if m.focused == "result" {
				movie := m.searchResults.SelectedItem().(Movie)
				cmds = append(cmds, m.ImportMovieCmd(movie), m.ResetCmd())
				m.Log(fmt.Sprintf("imported movie %s", movie.Title()))
			}
		}
	case Movies:
		m.Log(fmt.Sprintf("found %d movies in in tmdb", len(msg)))
		m.searchResults.SetItems(msg.listItems())
		m.focused = "result"
	}

	switch m.focused {
	case "search":
		m.searchInput, cmd = m.searchInput.Update(msg)
	case "result":
		m.searchResults, cmd = m.searchResults.Update(msg)
	}
	cmds = append(cmds, cmd)

	return m, tea.Batch(cmds...)
}

func (m tabTMDB) View() string {
	return fmt.Sprintf("%s\n%s\n", m.searchInput.View(), m.searchResults.View())
}

func (m *tabTMDB) Log(s string) {
	m.logger.Log(s)
}

func (m *tabTMDB) initialModel(width, height int) {
	si := textinput.New()
	si.Placeholder = "title"
	si.CharLimit = 156
	si.Width = 20
	m.searchInput = si
	m.searchInput.Focus()

	m.searchResults = list.New([]list.Item{}, list.NewDefaultDelegate(), width, height-1)
	m.searchResults.Title = "Search results"
	m.searchResults.SetShowHelp(false)

	m.focused = "search"
}

func (m *tabTMDB) SearchTMDBCmd(query string) tea.Cmd {
	return func() tea.Msg {
		tms, err := m.tmdb.Search(query)
		if err != nil {
			return err
		}
		return Movies(tms)
	}
}

func (m *tabTMDB) ImportMovieCmd(movie Movie) tea.Cmd {
	return func() tea.Msg {
		newMovie, err := m.emdb.CreateMovie(movie.m)
		if err != nil {
			return err
		}
		if err := m.emdb.CreateJob(newMovie.ID, string(job.ActionRefreshIMDBReviews)); err != nil {
			return err
		}

		return NewMovie(Movie{m: newMovie})
	}
}

func (m *tabTMDB) ResetCmd() tea.Cmd {
	return func() tea.Msg {
		return TabResetMsg("tmdb")
	}
}