package main

import (
	"fmt"
	"time"
	"strings"
	"github.com/charmbracelet/lipgloss"
)

func (m model) View() string {
	switch m.mode {
	case monthView:
		return m.viewMonth()
	case hourlyView:
		return m.viewHourly()
	default:
		return m.viewMonth()
	}
}


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, [up]/[down]/[left]/[right] keys to move, [enter] to select a day, [q] to quit."
	return out
}

func (m model) viewHourly() string {
	var b strings.Builder

	date := time.Date(m.year, time.Month(m.monthIndex+1), m.selectedDay, 0, 0, 0, 0, time.UTC)
	header := headerStyle.Render(date.Format("Monday, Jan 2, 2006"))
	b.WriteString(header + "\n\n")

	events, _ := GetEventsForDay(int32(m.year), int32(m.monthIndex+1), int32(m.selectedDay))

	for hour := 0; hour < 24; hour++ {
		label := fmt.Sprintf("%02d:00", hour)
		styled := false

		for _, e := range events {
			if hour >= int(e.StartHour) && hour < int(e.EndHour) {
				style := hourCellStyle.Copy().Background(lipgloss.Color(e.Color))

				if hour == m.hourCursor {
					if hour == int(e.StartHour) {
						label = style.Copy().Bold(true).Render("▶ " + fmt.Sprintf("%02d:00 - %s", hour, e.Title))
					} else {
						label = style.Copy().Bold(true).Render("▶ " + label)
					}
				} else {
					if hour == int(e.StartHour) {
						label = style.Render("  " + fmt.Sprintf("%02d:00 - %s", hour, e.Title))
					} else {
						label = style.Render("  " + label)
					}
				}

				styled = true
				break
			}
		}

		if !styled {
			if hour == m.hourCursor {
				label = hourSelectedStyle.Render("▶ " + label)
			} else if hour == time.Now().Hour() {
				label = currentHourStyle.Render("  " + label)
			} else {
				label = hourCellStyle.Render("  " + label)
			}
		}

		b.WriteString(label + "\n")
	}

	b.WriteString("[n] to add event, [up]/[down] keys to move, [Esc] to go back to month view, [q] to quit.")
	return b.String()
}