152 lines
3.2 KiB
Go
152 lines
3.2 KiB
Go
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)
|
|
}
|
|
}
|