diff --git a/main.go b/main.go index 3ba217b..283ac15 100644 --- a/main.go +++ b/main.go @@ -10,25 +10,44 @@ import ( ) const ( - numRows = 6 // Max possible rows in a calendar month view - numCols = 7 // Days in a week + numRows = 6 + numCols = 7 cellW = 10 cellH = 1 ) var ( cellStyle = lipgloss.NewStyle().Width(cellW).Height(cellH).Align(lipgloss.Center) - selectedStyle = cellStyle.Copy().Bold(true).Background(lipgloss.Color("12")).Foreground(lipgloss.Color("15")) + 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) + + 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")) +) + +type viewMode int + +const ( + monthView viewMode = iota + hourlyView ) type model struct { - cursorRow int - cursorCol int - monthIndex int - year int + 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 { @@ -36,11 +55,50 @@ func (m model) Init() tea.Cmd { } 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.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.cursorRow = offset / numCols + m.cursorCol = offset % numCols + return m, nil case "up": if m.cursorRow > 0 { m.cursorRow-- @@ -57,19 +115,44 @@ func (m model) Update(msg tea.Msg) (tea.Model, tea.Cmd) { if m.cursorCol < numCols-1 { m.cursorCol++ } - case "a": // Previous month - if m.monthIndex == 0 { - m.monthIndex = 11 - m.year-- - } else { - m.monthIndex-- + 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 } - case "d": // Next month - if m.monthIndex == 11 { - m.monthIndex = 0 - m.year++ - } else { - m.monthIndex++ + } + + // 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++ } } } @@ -77,59 +160,95 @@ func (m model) Update(msg tea.Msg) (tea.Model, tea.Cmd) { } 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 - // Header 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" - // Days of the week header weekdays := []string{"Sun", "Mon", "Tue", "Wed", "Thu", "Fri", "Sat"} for _, day := range weekdays { out += daysOfWeekStyle.Render(day) } out += "\n" - // First weekday and number of days - firstDay := time.Date(m.year, time.Month(m.monthIndex+1), 1, 0, 0, 0, 0, time.UTC) - startWeekday := int(firstDay.Weekday()) - daysInMonth := time.Date(m.year, time.Month(m.monthIndex+2), 0, 0, 0, 0, 0, time.UTC).Day() - - // Calendar grid day := 1 for r := 0; r < numRows; r++ { for c := 0; c < numCols; c++ { cellNum := r*numCols + c var content string - if cellNum >= startWeekday && day <= daysInMonth { + if cellNum >= m.startOffset && day <= m.daysInMonth { content = fmt.Sprintf("%2d", day) if r == m.cursorRow && c == m.cursorCol { out += selectedStyle.Render(content) } else { - out += unselectedStyle.Render(content) + 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("") // Empty cell + out += cellStyle.Render("") } } out += "\n" } - out += "\n[a]/[d] to change month, arrows to move, q to quit." + 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" + + for hour := 0; hour < 24; hour++ { + label := fmt.Sprintf("%02d:00 - %02d:00", hour, hour+1) + if hour == m.hourCursor { + out += hourSelectedStyle.Render(label) + "\n" + } else { + out += hourCellStyle.Render(label) + "\n" + } + } + + out += "\nPress ESC to return to month view, q to quit." return out } func main() { now := time.Now() + start := time.Date(now.Year(), now.Month(), 1, 0, 0, 0, 0, time.UTC) + offset := int(start.Weekday()) p := tea.NewProgram(model{ - year: now.Year(), - monthIndex: int(now.Month()) - 1, + year: now.Year(), + monthIndex: int(now.Month()) - 1, + cursorRow: offset / numCols, + cursorCol: offset % numCols, + startOffset: offset, + mode: monthView, }) if err := p.Start(); err != nil { - fmt.Fprintf(os.Stderr, "error: %v", err) + fmt.Fprintf(os.Stderr, "error: %v\n", err) os.Exit(1) } }