From bbbd0d60b695809214912e25908f31e484776447 Mon Sep 17 00:00:00 2001 From: steven carpenter Date: Sat, 28 Jun 2025 22:42:27 -0400 Subject: [PATCH] restructured main --- constants.go | 9 ++++ model.go | 30 +++++++++++++ styles.go | 19 ++++++++ update.go | 125 +++++++++++++++++++++++++++++++++++++++++++++++++++ view.go | 89 ++++++++++++++++++++++++++++++++++++ 5 files changed, 272 insertions(+) create mode 100644 constants.go create mode 100644 model.go create mode 100644 styles.go create mode 100644 update.go create mode 100644 view.go diff --git a/constants.go b/constants.go new file mode 100644 index 0000000..12cf83a --- /dev/null +++ b/constants.go @@ -0,0 +1,9 @@ +package main + +const ( + numRows = 6 + numCols = 7 + cellW = 10 + cellH = 1 +) + diff --git a/model.go b/model.go new file mode 100644 index 0000000..f0c78b7 --- /dev/null +++ b/model.go @@ -0,0 +1,30 @@ +package main + +import tea "github.com/charmbracelet/bubbletea" + +type viewMode int + +const ( + monthView viewMode = iota + hourlyView +) + +type model struct { + cursorRow int + cursorCol int + monthIndex int + year int + startOffset int // weekday offset where day 1 starts + daysInMonth int + + mode viewMode + + // For hourly view + selectedDay int + hourCursor int // which hour (0-23) is selected in hourly view +} + +func (m model) Init() tea.Cmd { + return nil +} + diff --git a/styles.go b/styles.go new file mode 100644 index 0000000..22b219a --- /dev/null +++ b/styles.go @@ -0,0 +1,19 @@ +package main + +import "github.com/charmbracelet/lipgloss" + +// Styles for calendar cells and headers +var ( + cellStyle = lipgloss.NewStyle().Width(cellW).Height(cellH).Align(lipgloss.Center) + selectedStyle = cellStyle.Copy().Bold(true).Background(lipgloss.Color("12")).Foreground(lipgloss.Color("15")) // Blue bg + unselectedStyle = cellStyle.Copy().Background(lipgloss.Color("236")).Foreground(lipgloss.Color("250")) + todayStyle = cellStyle.Copy().Background(lipgloss.Color("99")).Foreground(lipgloss.Color("15")).Bold(true) // Purple bg + white text + headerStyle = lipgloss.NewStyle().Width(numCols * cellW).Align(lipgloss.Center).Bold(true) + daysOfWeekStyle = lipgloss.NewStyle().Width(cellW).Align(lipgloss.Center).Bold(true) + + // Styles for hourly view cells + hourCellStyle = lipgloss.NewStyle().Width(cellW * 3).Height(cellH).Align(lipgloss.Left).PaddingLeft(1) + hourSelectedStyle = hourCellStyle.Copy().Bold(true).Background(lipgloss.Color("12")).Foreground(lipgloss.Color("15")) + currentHourStyle = hourCellStyle.Copy().Background(lipgloss.Color("99")).Foreground(lipgloss.Color("15")).Bold(true) // Purple bg + white text +) + diff --git a/update.go b/update.go new file mode 100644 index 0000000..fe720d3 --- /dev/null +++ b/update.go @@ -0,0 +1,125 @@ +package main + +import ( + "time" + + tea "github.com/charmbracelet/bubbletea" +) + +func (m model) Update(msg tea.Msg) (tea.Model, tea.Cmd) { + switch m.mode { + case monthView: + return m.updateMonthView(msg) + case hourlyView: + return m.updateHourlyView(msg) + default: + return m, nil + } +} + +func (m model) updateMonthView(msg tea.Msg) (tea.Model, tea.Cmd) { + firstDay := time.Date(m.year, time.Month(m.monthIndex+1), 1, 0, 0, 0, 0, time.UTC) + m.startOffset = int(firstDay.Weekday()) + m.daysInMonth = time.Date(m.year, time.Month(m.monthIndex+2), 0, 0, 0, 0, 0, time.UTC).Day() + + switch msg := msg.(type) { + case tea.KeyMsg: + switch msg.String() { + case "ctrl+c", "q": + return m, tea.Quit + case "a": // Previous month + if m.monthIndex == 0 { + m.monthIndex = 11 + m.year-- + } else { + m.monthIndex-- + } + + first := time.Date(m.year, time.Month(m.monthIndex+1), 1, 0, 0, 0, 0, time.UTC) + offset := int(first.Weekday()) + + m.startOffset = offset + m.daysInMonth = time.Date(m.year, time.Month(m.monthIndex+2), 0, 0, 0, 0, 0, time.UTC).Day() + + m.cursorRow = offset / numCols + m.cursorCol = offset % numCols + + return m, nil + case "d": // Next month + if m.monthIndex == 11 { + m.monthIndex = 0 + m.year++ + } else { + m.monthIndex++ + } + + first := time.Date(m.year, time.Month(m.monthIndex+1), 1, 0, 0, 0, 0, time.UTC) + offset := int(first.Weekday()) + + m.startOffset = offset + m.daysInMonth = time.Date(m.year, time.Month(m.monthIndex+2), 0, 0, 0, 0, 0, time.UTC).Day() + + m.cursorRow = offset / numCols + m.cursorCol = offset % numCols + + return m, nil + case "up": + if m.cursorRow > 0 { + m.cursorRow-- + } + case "down": + if m.cursorRow < numRows-1 { + m.cursorRow++ + } + case "left": + if m.cursorCol > 0 { + m.cursorCol-- + } + case "right": + if m.cursorCol < numCols-1 { + m.cursorCol++ + } + case "enter": + // Calculate selected day based on cursor + dayIndex := m.cursorRow*numCols + m.cursorCol + if dayIndex >= m.startOffset && dayIndex < m.startOffset+m.daysInMonth { + m.selectedDay = dayIndex - m.startOffset + 1 + m.hourCursor = 0 + m.mode = hourlyView + } + } + + // Clamp cursor to valid days + dayIndex := m.cursorRow*numCols + m.cursorCol + if dayIndex < m.startOffset || dayIndex >= m.startOffset+m.daysInMonth { + m.cursorRow = m.startOffset / numCols + m.cursorCol = m.startOffset % numCols + } + return m, nil + default: + return m, nil + } +} + +func (m model) updateHourlyView(msg tea.Msg) (tea.Model, tea.Cmd) { + switch msg := msg.(type) { + case tea.KeyMsg: + switch msg.String() { + case "esc": + m.mode = monthView + return m, nil + case "ctrl+c", "q": + return m, tea.Quit + case "up": + if m.hourCursor > 0 { + m.hourCursor-- + } + case "down": + if m.hourCursor < 23 { + m.hourCursor++ + } + } + } + return m, nil +} + diff --git a/view.go b/view.go new file mode 100644 index 0000000..3f91ac4 --- /dev/null +++ b/view.go @@ -0,0 +1,89 @@ +package main + +import ( + "fmt" + "time" +) + +func (m model) View() string { + switch m.mode { + case monthView: + return m.viewMonth() + case hourlyView: + return m.viewHourly() + default: + return "" + } +} + +func (m model) viewMonth() string { + var out string + + monthTime := time.Date(m.year, time.Month(m.monthIndex+1), 1, 0, 0, 0, 0, time.UTC) + header := headerStyle.Render(monthTime.Format("January 2006")) + out += header + "\n" + + weekdays := []string{"Sun", "Mon", "Tue", "Wed", "Thu", "Fri", "Sat"} + for _, day := range weekdays { + out += daysOfWeekStyle.Render(day) + } + out += "\n" + + day := 1 + for r := 0; r < numRows; r++ { + for c := 0; c < numCols; c++ { + cellNum := r*numCols + c + var content string + + if cellNum >= m.startOffset && day <= m.daysInMonth { + content = fmt.Sprintf("%2d", day) + if r == m.cursorRow && c == m.cursorCol { + out += selectedStyle.Render(content) + } else { + today := time.Now() + isToday := m.year == today.Year() && + m.monthIndex == int(today.Month())-1 && + day == today.Day() + if isToday { + out += todayStyle.Render(content) + } else { + out += unselectedStyle.Render(content) + } + } + day++ + } else { + out += cellStyle.Render("") + } + } + out += "\n" + } + + out += "\n[a]/[d] to change month, arrows to move, enter to select a day, q to quit." + return out +} + +func (m model) viewHourly() string { + var out string + + dateHeader := fmt.Sprintf("Schedule for %04d-%02d-%02d", m.year, m.monthIndex+1, m.selectedDay) + out += headerStyle.Render(dateHeader) + "\n\n" + + nowHour := time.Now().Hour() + + for hour := 0; hour < 24; hour++ { + label := fmt.Sprintf("%02d:00 - %02d:00", hour, hour+1) + + switch { + case hour == m.hourCursor: + out += hourSelectedStyle.Render(label) + "\n" + case hour == nowHour: + out += currentHourStyle.Render(label) + "\n" + default: + out += hourCellStyle.Render(label) + "\n" + } + } + + out += "\nPress ESC to return to month view, q to quit." + return out +} +