Initial charm version showing table

cmd/sqv-tea/main.go

@@ -0,0 +1,151 @@
+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)
+	}
+}