feat: updated cache implementation and fixed bugs

internal/cache/table.go

@@ -1,22 +1,77 @@
 package cache
 
 import (
+	"fmt"
+	"time"
+
 	"github.com/charmbracelet/bubbles/table"
+	"github.com/charmbracelet/bubbles/textinput"
 	tea "github.com/charmbracelet/bubbletea"
 	"github.com/charmbracelet/lipgloss"
+	"github.com/cznic/mathutil"
 )
 
-var baseStyle = lipgloss.NewStyle().
-	BorderStyle(lipgloss.NormalBorder()).
-	BorderForeground(lipgloss.Color("240"))
+var (
+	baseStyle = lipgloss.NewStyle().
+			BorderStyle(lipgloss.NormalBorder()).
+			BorderForeground(lipgloss.Color("240"))
+	footerStyle         = lipgloss.NewStyle().Foreground(lipgloss.Color("241"))
+	focusedStyle        = lipgloss.NewStyle().Foreground(lipgloss.Color("205"))
+	blurredStyle        = lipgloss.NewStyle().Foreground(lipgloss.Color("240"))
+	cursorStyle         = focusedStyle
+	noStyle             = lipgloss.NewStyle()
+	helpStyle           = blurredStyle
+	cursorModeHelpStyle = lipgloss.NewStyle().Foreground(lipgloss.Color("244"))
+
+	focusedButton = focusedStyle.Render("[ Update ]")
+	blurredButton = fmt.Sprintf("[ %s ]", blurredStyle.Render("Update"))
+)
 
 type Model struct {
-	selected int
-	Table    table.Model
+	CachePath string
+	Editor    Editor
+	Table     table.Model
+
+	// privates
+	inputs      []textinput.Model
+	focusIndex  int
+	statusMsg   string
+	statusTimer time.Duration
 }
 
-func NewModel() Model {
-	return Model{}
+func NewModel(cachePath string, table *table.Model) Model {
+	m := Model{
+		CachePath:  cachePath,
+		Editor:     NewEditor(),
+		Table:      *table,
+		inputs:     make([]textinput.Model, 4),
+		focusIndex: 0,
+	}
+	for i := range len(m.inputs) {
+		input := textinput.New()
+		input.Cursor.Style = cursorStyle
+		input.CharLimit = 256
+		switch i {
+		case 0:
+			input.Placeholder = "host"
+			input.Focus()
+			input.PromptStyle = focusedStyle
+			input.TextStyle = focusedStyle
+		case 1:
+			input.Placeholder = "ports"
+			input.CharLimit = 5
+		case 2:
+			input.Placeholder = "protocol"
+			input.CharLimit = 5
+
+		case 3:
+			input.Placeholder = "timestamp"
+			input.CharLimit = 10
+		}
+		m.inputs[i] = input
+	}
+
+	return m
 }
 
 func (m Model) Init() tea.Cmd { return nil }
@@ -25,28 +80,115 @@ func (m Model) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
 	var cmd tea.Cmd
 	switch msg := msg.(type) {
 	case tea.WindowSizeMsg:
-		// m.Table = m.Table.Width(msg.Width)
-		// m.Table = m.Table.Height(msg.Height)
+		// m.Table.SetWidth(msg.Width)
+		// m.Table.SetHeight(msg.Height)
 	case tea.KeyMsg:
+		// do always regardless of state
 		switch msg.String() {
-		case "esc":
-			if m.Table.Focused() {
-				m.Table.Blur()
-			} else {
-				m.Table.Focus()
-			}
-		case "q", "ctrl+c":
+		case "ctrl+c":
 			return m, tea.Quit
-		case "enter":
-			return m, tea.Batch(
-				tea.Printf("Selected host '%s'", m.Table.SelectedRow()[0]),
-			)
 		}
+
+		// do while editting table row
+		if m.Editor.IsEditting() {
+			switch msg.String() {
+			case "esc":
+				return m, m.editSelectedRow(false)
+			case "w", "enter":
+				return m, m.updateRow()
+			// case "enter":
+			// 	// Did the user press enter while the submit button was focused?
+			// 	// If so, exit.
+			// 	if m.focusIndex == len(m.inputs) {
+			// 		return m, tea.Quit
+			// 	}
+
+			case "tab", "shift+tab", "up", "down":
+				s := msg.String()
+				switch s {
+				case "up", "shift+tab":
+					m.focusIndex = mathutil.Clamp(m.focusIndex-1, 0, len(m.inputs))
+				case "down", "tab":
+					m.focusIndex = mathutil.Clamp(m.focusIndex+1, 0, len(m.inputs))
+				}
+
+				cmds := make([]tea.Cmd, len(m.inputs))
+				for i := range m.inputs {
+					if i == m.focusIndex {
+						// Set focused state
+						cmds[i] = m.inputs[i].Focus()
+						m.inputs[i].PromptStyle = focusedStyle
+						m.inputs[i].TextStyle = focusedStyle
+						continue
+					}
+					// Remove focused state
+					m.inputs[i].Blur()
+					m.inputs[i].PromptStyle = noStyle
+					m.inputs[i].TextStyle = noStyle
+				}
+				return m, tea.Batch(cmds...)
+			}
+			cmd = m.updateInputs(msg)
+			return m, tea.Batch(cmd)
+			// do while making selection from table
+		} else if m.Editor.IsSelecting() {
+			switch msg.String() {
+			case "enter":
+				return m, m.editSelectedRow(true)
+			case "esc":
+				return m, tea.Quit
+			case "ctrl+w":
+				m.updateCacheData()
+				return m, tea.Quit
+			case "w", "ctrl+s":
+				return m, m.updateCacheData()
+			case "d":
+				return m, m.deleteSelectedRow()
+			case "a":
+				return m, m.addRowAfterCursor()
+			case "ctrl+a":
+				return m, m.addRowAtEnd()
+			case "q":
+				return m, tea.Quit
+			}
+			m.Table, cmd = m.Table.Update(msg)
+		}
+
 	}
-	m.Table, cmd = m.Table.Update(msg)
 	return m, cmd
 }
 
 func (m Model) View() string {
-	return baseStyle.Render(m.Table.View()) + "\n"
+	if m.Editor.IsEditting() {
+		return m.editRowView()
+	} else if m.Editor.IsSelecting() {
+		display := m.Table.View() + "\n"
+		display += footerStyle.Render(fmt.Sprintf(`
+ j/k, ⬇/⬆ : move cursor;  enter: choose;  
+ a: add row after cursor;  ctrl+a: add row at end
+ d: delete row at cursor; 
+ w, ctrl+s: save data;  
+ ctrl+w: save and quit;  q, esc: quit w/o saving;
+----------------------------------------------------------
+ cache: %s
+ cursor: %d, rows: %d
+ %s
+`, m.CachePath, m.Table.Cursor(), len(m.Table.Rows()), m.statusMsg))
+
+		return baseStyle.Render(display + "\n")
+	}
+	return "Something went wrong..."
+}
+
+func (m *Model) displayMessage(message string, timeout time.Duration) {
+
+	// show the message arg for alloted time
+	m.statusMsg = message
+	reset := make(chan string)
+	go func() {
+		time.Sleep(timeout)
+		reset <- ""
+	}()
+	m.statusMsg = <-reset
+
 }