#pragma once
// The current month as a grid of days, weeks from Monday, today in a filled
// soft box, marked days (the alarm's) underlined. For the room a tall e-ink
// screen has under its clock (the Clock page, the lock screen), and Clock
// tools > Calendar. `tight`: rows a line apart, for a 64-pixel screen.

#include <time.h>
#include <helpers/ui/DisplayDriver.h>
#include "InfoKit.h"
#include "../ui-core/CalendarMath.h"

namespace calendar {

using calmath::daysIn;
using calmath::firstCol;
using calmath::weeks;

// Its height: the day names, a rule, then a row a week.
inline int height(DisplayDriver& d, const struct tm& t, bool tight = false) {
  if (tight) return (1 + weeks(t)) * d.getLineHeight() + 1;
  return (1 + weeks(t)) * (d.getLineHeight() + 2) + 3;
}

// `marked`: bit (day - 1) set for each day to underline. `today`: box
// t.tm_mday (off for another month than the current).
inline void draw(DisplayDriver& d, int x, int y, int w, const struct tm& t, uint32_t marked = 0,
                 bool tight = false, bool today = true) {
  static const char* const NAMES[] = { "Mo", "Tu", "We", "Th", "Fr", "Sa", "Su" };
  static const char* const INITIALS[] = { "M", "T", "W", "T", "F", "S", "S" };
  const int lh = d.getLineHeight(), step = tight ? lh : lh + 2, cw = w / 7;
  const int x0 = x + (w - 7 * cw) / 2;
  d.setColor(UIColor::primary_txt);
  for (int c = 0; c < 7; c++) d.drawTextCentered(x0 + c * cw + cw / 2, y, (tight ? INITIALS : NAMES)[c]);
  info::rule(d, x0, y + (tight ? lh - 1 : lh + 1), 7 * cw);
  const int top = y + (tight ? lh + 1 : step + 3), col0 = firstCol(t), n = daysIn(t.tm_year, t.tm_mon);
  char buf[3];
  for (int day = 1; day <= n; day++) {
    const int k = col0 + day - 1, cx = x0 + (k % 7) * cw, cy = top + (k / 7) * step;
    snprintf(buf, sizeof(buf), "%d", day);
    if (today && day == t.tm_mday) {
      if (tight) d.fillSoftRect(cx, cy - 1, cw, lh);
      else d.fillSoftRect(cx + 1, cy - 1, cw - 2, lh + 2);
      d.setColor(UIColor::window_bkg);
      d.drawTextCentered(cx + cw / 2, cy, buf);
      d.setColor(UIColor::primary_txt);
    } else {
      d.drawTextCentered(cx + cw / 2, cy, buf);
    }
    if (marked & (1UL << (day - 1))) {   // under the number, in the gap below it
      const int uw = d.getTextWidth(buf);
      d.setColor(today && day == t.tm_mday ? UIColor::window_bkg : UIColor::primary_txt);
      d.fillRect(cx + (cw - uw) / 2, cy + (tight ? lh - 2 : lh), uw, d.sepH());
      d.setColor(UIColor::primary_txt);
    }
  }
}

}  // namespace calendar