Working select statements for know tables, with and without semi at the end

cmd/sqv-tview/main.go

@@ -11,18 +11,27 @@ import (
 func main() {
 	app := tview.NewApplication()
 
-	menu := tview.NewList()
-	table := tview.NewTable().SetBorders(true)
-	table.SetBorder(true).SetTitle("TABLE")
+	menuView := tview.NewList()
+	tableView := tview.NewTable().SetBorders(true)
+	sqlEditor := tview.NewTextArea()
 
-	menu.SetTitle("TABLES").SetBorder(true)
-	menu.ShowSecondaryText(false).SetDoneFunc(func() {
-		table.Clear()
+	sqlEditor.SetTitle("SQL Editor").SetBorder(true)
+
+	tableView.SetBorder(true).SetTitle("TABLE")
+
+	menuView.SetTitle("TABLES").SetBorder(true)
+	menuView.ShowSecondaryText(false).SetDoneFunc(func() {
+		tableView.Clear()
 	})
 
-	flex := tview.NewFlex().
-		AddItem(menu, 0, 1, true).
-		AddItem(table, 0, 3, true)
+	verticalFlex := tview.NewFlex().
+		AddItem(menuView, 0, 1, true).
+		AddItem(tableView, 0, 3, false)
+
+	horizontalFlex := tview.NewFlex().
+		SetDirection(tview.FlexRow).
+		AddItem(verticalFlex, 0, 4, true).
+		AddItem(sqlEditor, 0, 1, false)
 
 	m, err := engine.NewManager("db.sqlite")
 	if err != nil {
@@ -33,19 +42,20 @@ func main() {
 	tables := m.GetTables()
 
 	for _, t := range tables {
-		menu.AddItem(t.Name, "", 0, nil)
+		menuView.AddItem(t.Name, "", 0, nil)
 	}
-	menu.SetChangedFunc(func(index int, mainText, secondaryText string, shortcut rune) {
-		table.Clear()
+	menuView.SetChangedFunc(func(index int, mainText, secondaryText string, shortcut rune) {
+		tableView.Clear()
 
 		t, ok := m.GetTable(mainText)
 		if !ok {
 			panic("AHHHHHHH")
 		}
 
+		tableView.SetFixed(1, 0)
 		for i, c := range t.Columns {
 			color := tcell.ColorDarkGreen
-			table.SetCell(0, i, tview.NewTableCell(c.Name).SetTextColor(color).SetAlign(tview.AlignCenter))
+			tableView.SetCell(0, i, tview.NewTableCell(c.Name).SetTextColor(color).SetAlign(tview.AlignCenter))
 		}
 
 		err = m.LoadTable(&t)
@@ -55,25 +65,88 @@ func main() {
 
 		for ri, r := range t.Rows {
 			for rc, c := range r.Values {
-				table.SetCell(ri+1, rc, tview.NewTableCell(c).SetTextColor(tcell.ColorWhite).SetAlign(tview.AlignCenter))
+				tableView.SetCell(ri+1, rc, tview.NewTableCell(c).SetTextColor(tcell.ColorWhite).SetAlign(tview.AlignCenter))
 			}
 		}
 	})
-	menu.SetCurrentItem(1)
 
-	table.Select(0, 0).SetFixed(1, 1).SetDoneFunc(func(key tcell.Key) {
-		switch key {
+	// Idk this shouldnt be needed imo but with only 0 it doesnt work, and with 1, well we are on Table 1 not zero, WHICH WE CANT ALWAYS SAY THERE IS
+	menuView.SetCurrentItem(1)
+	menuView.SetCurrentItem(0)
+
+	menuView.SetInputCapture(func(event *tcell.EventKey) *tcell.EventKey {
+		switch event.Key() {
 		case tcell.KeyEscape:
 			app.Stop()
 		case tcell.KeyEnter:
-			table.SetSelectable(true, true)
+			app.SetFocus(tableView)
+			return nil
+		case tcell.KeyRune:
+			switch event.Rune() {
+			case 'j':
+				return tcell.NewEventKey(tcell.KeyDown, 'j', event.Modifiers())
+			case 'k':
+				return tcell.NewEventKey(tcell.KeyUp, 'k', event.Modifiers())
+			}
 		}
-	}).SetSelectedFunc(func(row, column int) {
-		table.GetCell(row, column).SetTextColor(tcell.ColorRed)
-		table.SetSelectable(false, false)
+
+		return event
 	})
 
-	if err := app.SetRoot(flex, true).EnableMouse(true).Run(); err != nil {
+	sqlEditor.SetInputCapture(func(event *tcell.EventKey) *tcell.EventKey {
+		if event.Key() == tcell.KeyEnter {
+			t, err := m.RunSql(sqlEditor.GetText())
+			if err != nil {
+				sqlEditor.SetText(err.Error(), true)
+			}
+
+			tableView.Clear()
+			tableView.SetFixed(1, 0)
+			for i, c := range t.Columns {
+				color := tcell.ColorDarkGreen
+				tableView.SetCell(0, i, tview.NewTableCell(c.Name).SetTextColor(color).SetAlign(tview.AlignCenter))
+			}
+
+			for ri, r := range t.Rows {
+				for rc, c := range r.Values {
+					tableView.SetCell(ri+1, rc, tview.NewTableCell(c).SetTextColor(tcell.ColorWhite).SetAlign(tview.AlignCenter))
+				}
+			}
+
+			return nil
+		}
+
+		return event
+	})
+
+	tableView.SetInputCapture(func(event *tcell.EventKey) *tcell.EventKey {
+		switch event.Key() {
+		case tcell.KeyEscape:
+			app.SetFocus(menuView)
+			return nil
+		}
+
+		return event
+	})
+
+	menuHidden := false
+	verticalFlex.SetInputCapture(func(event *tcell.EventKey) *tcell.EventKey {
+		switch event.Key() {
+		case tcell.KeyCtrlH:
+			if !menuHidden {
+				verticalFlex.ResizeItem(menuView, 0, 0)
+				app.SetFocus(tableView)
+			} else {
+				verticalFlex.ResizeItem(menuView, 0, 1)
+				app.SetFocus(menuView)
+			}
+			menuHidden = !menuHidden
+		}
+
+		return event
+	})
+
+	if err := app.SetRoot(horizontalFlex, true).EnableMouse(true).Run(); err != nil {
 		panic(err)
 	}