package main

import (
	"log"
	"os"
	"strconv"

	engine "git.pablu.de/pablu/sqv-engine"
	"github.com/charmbracelet/bubbles/table"
	tea "github.com/charmbracelet/bubbletea"
	"github.com/charmbracelet/lipgloss"
)

var baseStyle = lipgloss.NewStyle().
	BorderStyle(lipgloss.NormalBorder()).
	BorderForeground(lipgloss.Color("240"))

type model struct {
	table table.Model

	currTableName string

	manager *engine.Manager
}

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

func (m model) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
	var cmd tea.Cmd
	switch msg := msg.(type) {
	case tea.KeyMsg:
		switch msg.String() {
		case "esc":
			if m.table.Focused() {
				m.table.Blur()
			} else {
				m.table.Focus()
			}
		case "ctrl+c", "q":
			return m, tea.Quit
		case "enter":
			return m, tea.Batch(
				tea.Printf("Let's go to %s!", m.table.SelectedRow()[1]),
			)
		case "n":
			return m, updateTable(m.manager, m.currTableName)
		}
	case newTableMsg:
		m.table.SetRows(msg.rows)
		m.table.SetColumns(msg.columns)
	}
	m.table, cmd = m.table.Update(msg)

	return m, cmd
}

type newTableMsg struct {
	columns []table.Column
	rows    []table.Row
}

func updateTable(m *engine.Manager, tableName string) tea.Cmd {
	return func() tea.Msg {
		t, _ := m.GetTable(tableName)

		columns := make([]table.Column, len(t.Columns))
		for i, column := range t.Columns {
			columns[i] = table.Column{
				Title: column.Name,
				Width: 10,
			}
		}

		for range 5 {
			row := engine.Row{Values: make([]string, len(t.Columns))}
			for i := range t.Columns {
				row.Values[i] = "test " + strconv.Itoa(i)
			}
			t.Rows = append(t.Rows, row)
		}

		rows := make([]table.Row, len(t.Rows))
		for i, row := range t.Rows {
			rows[i] = row.Values
		}

		return newTableMsg{columns: columns, rows: rows}
	}
}

func (m model) View() string {
	return baseStyle.Render(m.table.View()) + "\n"
}

func main() {
	file, err := os.ReadFile("test.sql")
	if err != nil {
		log.Fatal(err)
	}

	m := engine.NewManagerFromFile(string(file))
	m.Start()
	lobby, _ := m.GetTable("lobbys")
	lobby.Rows = []engine.Row{
		{Values: []string{"1", "me@pablu.de", "1"}},
		{Values: []string{"2", "pablu23@gmx.de", "2"}},
		{Values: []string{"5", "max@mustermann.org", "5"}},
		{Values: []string{"8", "eins@zwei.de", "8"}},
		{Values: []string{"19", "heinz.bongo@katzen.us", "19"}},
		{Values: []string{"22", "thessa@katze.de", "22"}},
	}

	columns := make([]table.Column, len(lobby.Columns))
	for i, column := range lobby.Columns {
		columns[i] = table.Column{
			Title: column.Name,
			Width: 10,
		}
	}

	rows := make([]table.Row, len(lobby.Rows))
	for i, row := range lobby.Rows {
		rows[i] = row.Values
	}

	t := table.New(
		table.WithColumns(columns),
		table.WithRows(rows),
		table.WithFocused(true),
		table.WithHeight(7),
	)

	s := table.DefaultStyles()
	s.Header = s.Header.
		BorderStyle(lipgloss.NormalBorder()).
		BorderForeground(lipgloss.Color("240")).
		BorderBottom(true).
		Bold(false)
	s.Selected = s.Selected.
		Foreground(lipgloss.Color("229")).
		Background(lipgloss.Color("57")).
		Bold(false)
	t.SetStyles(s)

	p := tea.NewProgram(model{table: t, manager: m, currTableName: lobby.Name})
	if _, err := p.Run(); err != nil {
		log.Fatalf("Alas, theres been an error: %v", err)
	}
}