use cobra/fang

internal/app/table_list.go

@@ -0,0 +1,198 @@
+package app
+
+import (
+	"fmt"
+	"strings"
+
+	tea "github.com/charmbracelet/bubbletea"
+)
+
+// Table List Model
+type TableListModel struct {
+	Shared        *SharedData
+	searchInput   string
+	searching     bool
+	selectedTable int
+	currentPage   int
+}
+
+func NewTableListModel(shared *SharedData) *TableListModel {
+	return &TableListModel{
+		Shared:        shared,
+		selectedTable: 0,
+		currentPage:   0,
+	}
+}
+
+func (m *TableListModel) Init() tea.Cmd {
+	return nil
+}
+
+func (m *TableListModel) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
+	switch msg := msg.(type) {
+	case tea.KeyMsg:
+		if m.searching {
+			return m.handleSearchInput(msg)
+		}
+		return m.handleNavigation(msg)
+	}
+	return m, nil
+}
+
+func (m *TableListModel) handleSearchInput(msg tea.KeyMsg) (tea.Model, tea.Cmd) {
+	switch msg.String() {
+	case "esc", "enter":
+		m.searching = false
+		m.filterTables()
+	case "backspace":
+		if len(m.searchInput) > 0 {
+			m.searchInput = m.searchInput[:len(m.searchInput)-1]
+			m.filterTables()
+		}
+	default:
+		if len(msg.String()) == 1 {
+			m.searchInput += msg.String()
+			m.filterTables()
+		}
+	}
+	return m, nil
+}
+
+func (m *TableListModel) handleNavigation(msg tea.KeyMsg) (tea.Model, tea.Cmd) {
+	switch msg.String() {
+	case "/":
+		m.searching = true
+		m.searchInput = ""
+		return m, nil
+
+	case "enter":
+		if len(m.Shared.FilteredTables) > 0 {
+			return m, func() tea.Msg {
+				return SwitchToTableDataMsg{TableIndex: m.selectedTable}
+			}
+		}
+
+	case "s":
+		return m, func() tea.Msg { return SwitchToQueryMsg{} }
+
+	case "r":
+		if err := m.Shared.LoadTables(); err == nil {
+			m.filterTables()
+		}
+
+	case "up", "k":
+		if m.selectedTable > 0 {
+			m.selectedTable--
+			m.adjustPage()
+		}
+
+	case "down", "j":
+		if m.selectedTable < len(m.Shared.FilteredTables)-1 {
+			m.selectedTable++
+			m.adjustPage()
+		}
+
+	case "left", "h":
+		if m.currentPage > 0 {
+			m.currentPage--
+			m.selectedTable = m.currentPage * m.getVisibleCount()
+		}
+
+	case "right", "l":
+		maxPage := (len(m.Shared.FilteredTables) - 1) / m.getVisibleCount()
+		if m.currentPage < maxPage {
+			m.currentPage++
+			m.selectedTable = m.currentPage * m.getVisibleCount()
+			if m.selectedTable >= len(m.Shared.FilteredTables) {
+				m.selectedTable = len(m.Shared.FilteredTables) - 1
+			}
+		}
+	}
+	return m, nil
+}
+
+func (m *TableListModel) filterTables() {
+	if m.searchInput == "" {
+		m.Shared.FilteredTables = make([]string, len(m.Shared.Tables))
+		copy(m.Shared.FilteredTables, m.Shared.Tables)
+	} else {
+		m.Shared.FilteredTables = []string{}
+		searchLower := strings.ToLower(m.searchInput)
+		for _, table := range m.Shared.Tables {
+			if strings.Contains(strings.ToLower(table), searchLower) {
+				m.Shared.FilteredTables = append(m.Shared.FilteredTables, table)
+			}
+		}
+	}
+
+	if m.selectedTable >= len(m.Shared.FilteredTables) {
+		m.selectedTable = 0
+		m.currentPage = 0
+	}
+}
+
+func (m *TableListModel) getVisibleCount() int {
+	reservedLines := 8
+	if m.searching {
+		reservedLines += 2
+	}
+	return Max(1, m.Shared.Height-reservedLines)
+}
+
+func (m *TableListModel) adjustPage() {
+	visibleCount := m.getVisibleCount()
+	m.currentPage = m.selectedTable / visibleCount
+}
+
+func (m *TableListModel) View() string {
+	var content strings.Builder
+
+	content.WriteString(TitleStyle.Render("SQLite TUI - Tables"))
+	content.WriteString("\n")
+
+	if m.searching {
+		content.WriteString("\nSearch: " + m.searchInput + "_")
+		content.WriteString("\n")
+	} else if m.searchInput != "" {
+		content.WriteString(fmt.Sprintf("\nFiltered by: %s (%d/%d tables)",
+			m.searchInput, len(m.Shared.FilteredTables), len(m.Shared.Tables)))
+		content.WriteString("\n")
+	}
+	content.WriteString("\n")
+
+	if len(m.Shared.FilteredTables) == 0 {
+		if m.searchInput != "" {
+			content.WriteString("No tables match your search")
+		} else {
+			content.WriteString("No tables found in database")
+		}
+	} else {
+		visibleCount := m.getVisibleCount()
+		startIdx := m.currentPage * visibleCount
+		endIdx := Min(startIdx+visibleCount, len(m.Shared.FilteredTables))
+
+		for i := startIdx; i < endIdx; i++ {
+			table := m.Shared.FilteredTables[i]
+			if i == m.selectedTable {
+				content.WriteString(SelectedStyle.Render(fmt.Sprintf("> %s", table)))
+			} else {
+				content.WriteString(NormalStyle.Render(fmt.Sprintf("  %s", table)))
+			}
+			content.WriteString("\n")
+		}
+
+		if len(m.Shared.FilteredTables) > visibleCount {
+			totalPages := (len(m.Shared.FilteredTables)-1)/visibleCount + 1
+			content.WriteString(fmt.Sprintf("\nPage %d/%d", m.currentPage+1, totalPages))
+		}
+	}
+
+	content.WriteString("\n")
+	if m.searching {
+		content.WriteString(HelpStyle.Render("Type to search • enter/esc: finish search"))
+	} else {
+		content.WriteString(HelpStyle.Render("↑/↓: navigate • ←/→: page • /: search • enter: view • s: SQL • r: refresh • ctrl+c: quit"))
+	}
+
+	return content.String()
+}