use cobra/fang

internal/app/table_data.go

@@ -0,0 +1,205 @@
+package app
+
+import (
+	"fmt"
+	"strings"
+
+	tea "github.com/charmbracelet/bubbletea"
+)
+
+// Table Data Model
+type TableDataModel struct {
+	Shared      *SharedData
+	selectedRow int
+	searchInput string
+	searching   bool
+}
+
+func NewTableDataModel(shared *SharedData) *TableDataModel {
+	return &TableDataModel{
+		Shared:      shared,
+		selectedRow: 0,
+	}
+}
+
+func (m *TableDataModel) Init() tea.Cmd {
+	return nil
+}
+
+func (m *TableDataModel) 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 *TableDataModel) handleSearchInput(msg tea.KeyMsg) (tea.Model, tea.Cmd) {
+	switch msg.String() {
+	case "esc", "enter":
+		m.searching = false
+		m.filterData()
+	case "backspace":
+		if len(m.searchInput) > 0 {
+			m.searchInput = m.searchInput[:len(m.searchInput)-1]
+			m.filterData()
+		}
+	default:
+		if len(msg.String()) == 1 {
+			m.searchInput += msg.String()
+			m.filterData()
+		}
+	}
+	return m, nil
+}
+
+func (m *TableDataModel) handleNavigation(msg tea.KeyMsg) (tea.Model, tea.Cmd) {
+	switch msg.String() {
+	case "q", "esc":
+		return m, func() tea.Msg { return SwitchToTableListMsg{} }
+
+	case "enter":
+		if len(m.Shared.FilteredData) > 0 {
+			return m, func() tea.Msg {
+				return SwitchToRowDetailMsg{RowIndex: m.selectedRow}
+			}
+		}
+
+	case "/":
+		m.searching = true
+		m.searchInput = ""
+		return m, nil
+
+	case "s":
+		return m, func() tea.Msg { return SwitchToQueryMsg{} }
+
+	case "r":
+		if err := m.Shared.LoadTableData(); err == nil {
+			m.filterData()
+		}
+
+	case "up", "k":
+		if m.selectedRow > 0 {
+			m.selectedRow--
+		}
+
+	case "down", "j":
+		if m.selectedRow < len(m.Shared.FilteredData)-1 {
+			m.selectedRow++
+		}
+
+	case "left", "h":
+		if m.Shared.CurrentPage > 0 {
+			m.Shared.CurrentPage--
+			m.Shared.LoadTableData()
+			m.selectedRow = 0
+		}
+
+	case "right", "l":
+		maxPage := (m.Shared.TotalRows - 1) / PageSize
+		if m.Shared.CurrentPage < maxPage {
+			m.Shared.CurrentPage++
+			m.Shared.LoadTableData()
+			m.selectedRow = 0
+		}
+	}
+	return m, nil
+}
+
+func (m *TableDataModel) filterData() {
+	if m.searchInput == "" {
+		m.Shared.FilteredData = make([][]string, len(m.Shared.TableData))
+		copy(m.Shared.FilteredData, m.Shared.TableData)
+	} else {
+		m.Shared.FilteredData = [][]string{}
+		searchLower := strings.ToLower(m.searchInput)
+		for _, row := range m.Shared.TableData {
+			for _, cell := range row {
+				if strings.Contains(strings.ToLower(cell), searchLower) {
+					m.Shared.FilteredData = append(m.Shared.FilteredData, row)
+					break
+				}
+			}
+		}
+	}
+
+	if m.selectedRow >= len(m.Shared.FilteredData) {
+		m.selectedRow = 0
+	}
+}
+
+func (m *TableDataModel) View() string {
+	var content strings.Builder
+
+	tableName := ""
+	if m.Shared.SelectedTable < len(m.Shared.FilteredTables) {
+		tableName = m.Shared.FilteredTables[m.Shared.SelectedTable]
+	}
+
+	content.WriteString(TitleStyle.Render(fmt.Sprintf("Table: %s", tableName)))
+	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 rows)",
+			m.searchInput, len(m.Shared.FilteredData), len(m.Shared.TableData)))
+		content.WriteString("\n")
+	}
+
+	// Show pagination info
+	totalPages := (m.Shared.TotalRows-1)/PageSize + 1
+	content.WriteString(fmt.Sprintf("Page %d/%d (%d total rows)\n\n",
+		m.Shared.CurrentPage+1, totalPages, m.Shared.TotalRows))
+
+	if len(m.Shared.FilteredData) == 0 {
+		content.WriteString("No data found")
+	} else {
+		// Show column headers
+		headerRow := ""
+		for i, col := range m.Shared.Columns {
+			if i > 0 {
+				headerRow += " | "
+			}
+			headerRow += TruncateString(col, 15)
+		}
+		content.WriteString(TitleStyle.Render(headerRow))
+		content.WriteString("\n")
+
+		// Show data rows
+		visibleCount := Max(1, m.Shared.Height-10)
+		startIdx := 0
+		endIdx := Min(len(m.Shared.FilteredData), visibleCount)
+
+		for i := startIdx; i < endIdx; i++ {
+			row := m.Shared.FilteredData[i]
+			rowStr := ""
+			for j, cell := range row {
+				if j > 0 {
+					rowStr += " | "
+				}
+				rowStr += TruncateString(cell, 15)
+			}
+
+			if i == m.selectedRow {
+				content.WriteString(SelectedStyle.Render("> " + rowStr))
+			} else {
+				content.WriteString(NormalStyle.Render("  " + rowStr))
+			}
+			content.WriteString("\n")
+		}
+	}
+
+	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: details • s: SQL • r: refresh • q: back"))
+	}
+
+	return content.String()
+}