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
	initialized bool
	logger      *Logger
	logViewport viewport.Model
	windowSize  tea.WindowSizeMsg
	contentSize 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) 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
}