WIP TUI

1 files changed, 538 insertions, 0 deletions

diff --git a/internal/tui/app.go b/internal/tui/app.go
new file mode 100644
index 0000000..98bad4f
--- /dev/null
+++ b/internal/tui/app.go
@@ -0,0 +1,538 @@
+package tui
+
+import (
+	"context"
+	"database/sql"
+	"fmt"
+	"time"
+
+	"punchcard/internal/queries"
+
+	tea "github.com/charmbracelet/bubbletea"
+	"github.com/charmbracelet/lipgloss/v2"
+)
+
+// NewApp creates a new TUI application
+func NewApp(ctx context.Context, q *queries.Queries) *AppModel {
+	return &AppModel{
+		ctx:                  ctx,
+		queries:              q,
+		selectedBox:          TimerBox,
+		timerBoxModel:        NewTimerBoxModel(),
+		clientsProjectsModel: NewClientsProjectsModel(),
+		historyBoxModel:      NewHistoryBoxModel(),
+	}
+}
+
+// Init initializes the app
+func (m AppModel) Init() tea.Cmd {
+	return tea.Batch(
+		m.updateDataCmd(),
+		m.tickCmd(),
+	)
+}
+
+// Update handles messages for the app
+func (m AppModel) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
+	var cmds []tea.Cmd
+
+	switch msg := msg.(type) {
+	case tea.WindowSizeMsg:
+		m.width = msg.Width
+		m.height = msg.Height
+
+	case tea.KeyMsg:
+		if m.showModal {
+			// Handle modal input
+			cmd := m.handleModalInput(msg)
+			if cmd != nil {
+				cmds = append(cmds, cmd)
+			}
+		} else {
+			// Handle normal input
+			action := HandleKeyPress(msg, m.selectedBox, m.historyBoxModel.viewLevel, m.timerBoxModel.timerInfo.IsActive)
+			cmd := m.handleAction(action)
+			if cmd != nil {
+				cmds = append(cmds, cmd)
+			}
+		}
+
+	case TickMsg:
+		// Update timer duration if active using cached start time
+		if m.runningTimerStart != nil {
+			m.timerBoxModel.timerInfo.IsActive = true
+			m.timerBoxModel.timerInfo.Duration = time.Since(*m.runningTimerStart)
+			m.timerBoxModel.timerInfo.StartTime = *m.runningTimerStart
+			// Keep history model in sync
+			m.historyBoxModel.runningTimerStart = m.runningTimerStart
+		} else {
+			m.timerBoxModel.timerInfo.IsActive = false
+			m.timerBoxModel.timerInfo.Duration = 0
+			// Keep history model in sync
+			m.historyBoxModel.runningTimerStart = nil
+		}
+		cmds = append(cmds, m.tickCmd())
+
+	case dataUpdatedMsg:
+		// Update all models with fresh data
+		m.timerBoxModel = m.timerBoxModel.UpdateTimerInfo(msg.timerInfo)
+		m.clientsProjectsModel = m.clientsProjectsModel.UpdateData(msg.clients, msg.projects)
+		m.historyBoxModel = m.historyBoxModel.UpdateData(msg.entries, msg.clients, msg.projects)
+		// Update running timer data in history model too
+		if msg.timerInfo.IsActive {
+			m.historyBoxModel.runningTimerStart = &msg.timerInfo.StartTime
+		} else {
+			m.historyBoxModel.runningTimerStart = nil
+		}
+		// Cache stats and running timer start time
+		m.stats = msg.stats
+		if msg.timerInfo.IsActive {
+			m.runningTimerStart = &msg.timerInfo.StartTime
+		} else {
+			m.runningTimerStart = nil
+		}
+
+		// Schedule next data update in 30 seconds
+		cmds = append(cmds, tea.Tick(30*time.Second, func(t time.Time) tea.Msg {
+			return updateDataCmd{}
+		}))
+
+	case updateDataCmd:
+		cmds = append(cmds, m.updateDataCmd())
+	}
+
+	return m, tea.Batch(cmds...)
+}
+
+// handleAction processes the action returned by key handling
+func (m *AppModel) handleAction(action KeyAction) tea.Cmd {
+	switch action {
+	// Global actions
+	case ActionNextPane:
+		m.selectedBox = (m.selectedBox + 1) % 3
+	case ActionPrevPane:
+		m.selectedBox = (m.selectedBox + 2) % 3 // +2 is like -1 in mod 3
+	case ActionPunchToggle:
+		return m.handlePunchToggle()
+	case ActionSearch:
+		return m.handleSearch()
+	case ActionRefresh:
+		return m.handleRefresh()
+	case ActionQuit:
+		return tea.Quit
+
+	// Timer pane actions
+	case ActionTimerEnter:
+		return m.handleTimerEnter()
+	case ActionTimerDescribe:
+		m.handleTimerDescribe()
+
+	// Projects pane actions
+	case ActionProjectsNext:
+		m.handleProjectsNext()
+	case ActionProjectsPrev:
+		m.handleProjectsPrev()
+	case ActionProjectsEnter:
+		return m.handleProjectsEnter()
+	case ActionProjectsNewProject:
+		return m.handleProjectsNewProject()
+	case ActionProjectsNewClient:
+		return m.handleProjectsNewClient()
+
+	// History pane actions
+	case ActionHistoryNext:
+		m.handleHistoryNext()
+	case ActionHistoryPrev:
+		m.handleHistoryPrev()
+	case ActionHistoryEnter:
+		return m.handleHistoryEnter()
+	case ActionHistoryEdit:
+		return m.handleHistoryEdit()
+	case ActionHistoryDelete:
+		return m.handleHistoryDelete()
+	case ActionHistoryResume:
+		return m.handleHistoryResume()
+	case ActionHistoryBack:
+		m.handleHistoryBack()
+	}
+
+	return nil
+}
+
+// Global action handlers
+func (m *AppModel) handlePunchToggle() tea.Cmd {
+	// TODO: Implement punch in/out toggle
+	return nil
+}
+
+func (m *AppModel) handleSearch() tea.Cmd {
+	// TODO: Implement search modal
+	return nil
+}
+
+func (m *AppModel) handleRefresh() tea.Cmd {
+	// Immediately refresh data from database
+	return m.updateDataCmd()
+}
+
+// Timer pane action handlers
+func (m *AppModel) handleTimerEnter() tea.Cmd {
+	if m.timerBoxModel.timerInfo.IsActive {
+		// TODO: Implement punch out
+		return nil
+	} else {
+		// TODO: Implement resume recent (punch back in to most recent project)
+		return nil
+	}
+}
+
+// Projects pane action handlers
+func (m *AppModel) handleProjectsNext() {
+	m.clientsProjectsModel = m.clientsProjectsModel.NextSelection()
+}
+
+func (m *AppModel) handleProjectsPrev() {
+	m.clientsProjectsModel = m.clientsProjectsModel.PrevSelection()
+}
+
+func (m *AppModel) handleProjectsEnter() tea.Cmd {
+	// TODO: Punch in to selected client/project
+	return nil
+}
+
+func (m *AppModel) handleProjectsNewProject() tea.Cmd {
+	// TODO: Open new project modal
+	return nil
+}
+
+func (m *AppModel) handleProjectsNewClient() tea.Cmd {
+	// TODO: Open new client modal
+	return nil
+}
+
+// History pane action handlers
+func (m *AppModel) handleHistoryNext() {
+	m.historyBoxModel = m.historyBoxModel.NextSelection()
+}
+
+func (m *AppModel) handleHistoryPrev() {
+	m.historyBoxModel = m.historyBoxModel.PrevSelection()
+}
+
+func (m *AppModel) handleHistoryEnter() tea.Cmd {
+	if m.historyBoxModel.viewLevel == HistoryLevelSummary {
+		// Drill down to details view
+		m.historyBoxModel = m.historyBoxModel.DrillDown()
+		return nil
+	} else {
+		// TODO: Resume selected entry (punch in with same details)
+		return nil
+	}
+}
+
+func (m *AppModel) handleHistoryEdit() tea.Cmd {
+	// TODO: Open edit modal for selected entry
+	return nil
+}
+
+func (m *AppModel) handleHistoryDelete() tea.Cmd {
+	// TODO: Delete selected entry
+	return nil
+}
+
+func (m *AppModel) handleHistoryResume() tea.Cmd {
+	// TODO: Resume selected entry (punch in with same details)
+	return nil
+}
+
+func (m *AppModel) handleHistoryBack() {
+	// Switch back to summary view
+	m.historyBoxModel = m.historyBoxModel.GoBack()
+}
+
+// Timer action handlers
+func (m *AppModel) handleTimerDescribe() {
+	// Show modal for timer description
+	m.showModal = true
+	m.modalType = ModalDescribeTimer
+	
+	// Get current description if any
+	currentDesc := ""
+	if m.timerBoxModel.timerInfo.Description != "" {
+		currentDesc = m.timerBoxModel.timerInfo.Description
+	}
+	
+	m.textInputModel = TextInputModel{
+		prompt:      "Enter timer description:",
+		value:       currentDesc,
+		placeholder: "Working on...",
+		cursorPos:   len(currentDesc),
+	}
+}
+
+// Modal handling methods
+func (m *AppModel) handleModalInput(msg tea.KeyMsg) tea.Cmd {
+	key := msg.String()
+	
+	switch key {
+	case "enter":
+		return m.submitModal()
+	case "escape", "ctrl+c":
+		m.closeModal()
+		return nil
+	case "backspace":
+		if m.textInputModel.cursorPos > 0 {
+			// Remove character before cursor
+			value := m.textInputModel.value
+			m.textInputModel.value = value[:m.textInputModel.cursorPos-1] + value[m.textInputModel.cursorPos:]
+			m.textInputModel.cursorPos--
+		}
+	case "left":
+		if m.textInputModel.cursorPos > 0 {
+			m.textInputModel.cursorPos--
+		}
+	case "right":
+		if m.textInputModel.cursorPos < len(m.textInputModel.value) {
+			m.textInputModel.cursorPos++
+		}
+	case "home", "ctrl+a":
+		m.textInputModel.cursorPos = 0
+	case "end", "ctrl+e":
+		m.textInputModel.cursorPos = len(m.textInputModel.value)
+	default:
+		// Handle character input
+		if len(key) == 1 && key[0] >= 32 && key[0] <= 126 {
+			// Insert character at cursor position
+			value := m.textInputModel.value
+			m.textInputModel.value = value[:m.textInputModel.cursorPos] + key + value[m.textInputModel.cursorPos:]
+			m.textInputModel.cursorPos++
+		}
+	}
+	
+	return nil
+}
+
+func (m *AppModel) submitModal() tea.Cmd {
+	var cmd tea.Cmd
+	
+	switch m.modalType {
+	case ModalDescribeTimer:
+		// Update the active timer description
+		cmd = m.updateTimerDescription(m.textInputModel.value)
+	}
+	
+	m.closeModal()
+	return cmd
+}
+
+func (m *AppModel) closeModal() {
+	m.showModal = false
+	m.textInputModel = TextInputModel{}
+}
+
+func (m *AppModel) updateTimerDescription(description string) tea.Cmd {
+	return func() tea.Msg {
+		// Update the active timer's description in the database
+		var desc sql.NullString
+		if description != "" {
+			desc = sql.NullString{String: description, Valid: true}
+		}
+		
+		err := m.queries.UpdateActiveTimerDescription(m.ctx, desc)
+		if err != nil {
+			// Handle error silently for now
+			return nil
+		}
+		
+		// Trigger a data refresh to update the UI
+		return updateDataCmd{}
+	}
+}
+
+// View renders the app
+func (m AppModel) View() string {
+	if m.width == 0 || m.height == 0 {
+		return "Loading..."
+	}
+
+	// Calculate dimensions
+	topBarHeight := 1
+	bottomBarHeight := 1
+	contentHeight := m.height - topBarHeight - bottomBarHeight
+
+	vertBoxOverhead := 6 // 2 border, 4 padding
+	horizBoxOverhead := 4 // 2 border, 2 padding
+
+	// Timer box is in top-left
+	timerBoxWidth := (m.width / 3) - horizBoxOverhead
+	timerBoxHeight := (contentHeight / 2) - vertBoxOverhead
+
+	// Clients/Projects box is in bottom-left
+	clientsProjectsBoxWidth := (m.width / 3) - horizBoxOverhead
+	clientsProjectsBoxHeight := (contentHeight - timerBoxHeight) - vertBoxOverhead
+
+	// History box takes the right side
+	historyBoxWidth := (m.width - (m.width / 3)) - horizBoxOverhead
+	historyBoxHeight := contentHeight - vertBoxOverhead
+
+	// Render top bar with current box info and time stats
+	viewName := fmt.Sprintf("Selected: %s", m.selectedBox.String())
+	// Use cached stats, but add running timer duration to both today's and week's totals if active
+	currentStats := m.stats
+	if m.runningTimerStart != nil {
+		runningDuration := time.Since(*m.runningTimerStart)
+		currentStats.TodayTotal += runningDuration
+		currentStats.WeekTotal += runningDuration
+	}
+	topBar := RenderTopBar(viewName, currentStats, m.width)
+
+	// Render boxes
+	timerBox := m.timerBoxModel.View(timerBoxWidth, timerBoxHeight, m.selectedBox == TimerBox)
+	clientsProjectsBox := m.clientsProjectsModel.View(clientsProjectsBoxWidth, clientsProjectsBoxHeight, m.selectedBox == ClientsProjectsBox)
+	historyBox := m.historyBoxModel.View(historyBoxWidth, historyBoxHeight, m.selectedBox == HistoryBox)
+
+	// Layout: Timer box above Clients/Projects box on the left, History box on the right
+	leftColumn := lipgloss.JoinVertical(lipgloss.Left, timerBox, clientsProjectsBox)
+	mainContent := lipgloss.JoinHorizontal(lipgloss.Top, leftColumn, historyBox)
+
+	// Render bottom bar
+	keyBindings := GetContextualKeyBindings(m.selectedBox, m.historyBoxModel.viewLevel, m.timerBoxModel.timerInfo.IsActive)
+	bottomBar := RenderBottomBar(keyBindings, m.width)
+
+	// Combine everything
+	finalView := topBar + "\n" + mainContent + "\n" + bottomBar
+	
+	// Overlay modal if one is active using lipgloss v2 Layers
+	if m.showModal {
+		modal := m.renderModal()
+		
+		// Create layers for base content and modal
+		baseLayer := lipgloss.NewLayer(finalView)
+		modalLayer := lipgloss.NewLayer(modal).
+			X((m.width - 60) / 2).  // Center horizontally
+			Y((m.height - 8) / 2).  // Center vertically
+			Z(1)                     // Put modal on top
+		
+		// Use lipgloss v2 Canvas to overlay the modal
+		canvas := lipgloss.NewCanvas(baseLayer, modalLayer)
+		finalView = canvas.Render()
+	}
+	
+	return finalView
+}
+
+// renderModal renders the modal content with proper styling
+func (m AppModel) renderModal() string {
+	// Modal dimensions
+	modalWidth := 60
+	
+	// Create modal content based on type
+	var modalContent string
+	switch m.modalType {
+	case ModalDescribeTimer:
+		modalContent = m.renderDescribeTimerModal(modalWidth-8) // Account for border and padding
+	}
+	
+	// Create modal box with border using lipgloss v2
+	modalStyle := lipgloss.NewStyle().
+		Border(lipgloss.RoundedBorder()).
+		BorderForeground(lipgloss.Color("62")).
+		Background(lipgloss.Color("235")).
+		Padding(1, 2).
+		Width(modalWidth-4)
+	
+	return modalStyle.Render(modalContent)
+}
+
+
+// renderDescribeTimerModal renders the timer description modal
+func (m AppModel) renderDescribeTimerModal(width int) string {
+	prompt := m.textInputModel.prompt
+	value := m.textInputModel.value
+	placeholder := m.textInputModel.placeholder
+	cursorPos := m.textInputModel.cursorPos
+	
+	// Show placeholder if value is empty
+	displayValue := value
+	if displayValue == "" {
+		displayValue = placeholder
+		// Style placeholder differently
+		displayValue = lipgloss.NewStyle().Foreground(lipgloss.Color("8")).Render(displayValue)
+	} else {
+		// Insert cursor
+		if cursorPos >= 0 && cursorPos <= len(value) {
+			if cursorPos == len(value) {
+				displayValue = value + "│"
+			} else {
+				displayValue = value[:cursorPos] + "│" + value[cursorPos:]
+			}
+		}
+	}
+	
+	// Input field styling
+	inputStyle := lipgloss.NewStyle().
+		Border(lipgloss.NormalBorder()).
+		BorderForeground(lipgloss.Color("6")).
+		Padding(0, 1).
+		Width(width - 2)
+	
+	inputField := inputStyle.Render(displayValue)
+	
+	instructions := lipgloss.NewStyle().
+		Foreground(lipgloss.Color("8")).
+		Render("Press Enter to save, Escape to cancel")
+	
+	return prompt + "\n\n" + inputField + "\n\n" + instructions
+}
+
+// tickCmd returns a command that sends a tick message every second
+func (m AppModel) tickCmd() tea.Cmd {
+	return tea.Tick(time.Second, func(t time.Time) tea.Msg {
+		return TickMsg(t)
+	})
+}
+
+// updateDataCmd triggers a data update
+type updateDataCmd struct{}
+
+// dataUpdatedMsg is sent when data is updated from the database
+type dataUpdatedMsg struct {
+	timerInfo TimerInfo
+	stats     TimeStats
+	clients   []queries.Client
+	projects  []queries.ListAllProjectsRow
+	entries   []queries.TimeEntry
+}
+
+// updateDataCmd returns a command to update data
+func (m AppModel) updateDataCmd() tea.Cmd {
+	return func() tea.Msg {
+		timerInfo, stats, clients, projects, entries, err := GetAppData(m.ctx, m.queries)
+		if err != nil {
+			// Handle error silently for now - return empty data
+			return dataUpdatedMsg{
+				timerInfo: TimerInfo{},
+				stats:     TimeStats{},
+				clients:   []queries.Client{},
+				projects:  []queries.ListAllProjectsRow{},
+				entries:   []queries.TimeEntry{},
+			}
+		}
+
+		return dataUpdatedMsg{
+			timerInfo: timerInfo,
+			stats:     stats,
+			clients:   clients,
+			projects:  projects,
+			entries:   entries,
+		}
+	}
+}
+
+// Run starts the TUI application
+func Run(ctx context.Context, q *queries.Queries) error {
+	app := NewApp(ctx, q)
+	p := tea.NewProgram(app, tea.WithAltScreen())
+	_, err := p.Run()
+	return err
+}