#include <Wire.h>
#include <RTClib.h>
#include <Adafruit_GFX.h>
#include <Adafruit_ST7789.h>

#define TFT_CS     10
#define TFT_RST    9
#define TFT_DC     8
#define TFT_WIDTH  240
#define TFT_HEIGHT 280

Adafruit_ST7789 tft = Adafruit_ST7789(TFT_CS, TFT_DC, TFT_RST);
RTC_DS3231 rtc;

#define BLACK       0x0000
#define WHITE       0xFFFF
#define RED         0xF800
#define ORANGE      0xFD20
#define YELLOW      0xFFE0
#define GREEN       0x07E0
#define BLUE        0x001F
#define INDIGO      0x4810
#define VIOLET      0x780F

uint16_t rainbowColors[] = {RED, ORANGE, YELLOW, GREEN, BLUE, INDIGO, VIOLET};

void setup() {
  Serial.begin(9600);
  Serial.println("Initializing...");

  if (!rtc.begin()) {
    Serial.println("Couldn't find RTC");
    while (1);
  }

  if (rtc.lostPower()) {
    Serial.println("RTC lost power, setting the time!");
    rtc.adjust(DateTime(F(__DATE__), F(__TIME__)));
  }

  tft.init(TFT_WIDTH, TFT_HEIGHT);
  
  tft.setRotation(2); // Set rotation of the display
 
  tft.fillScreen(BLACK); // Clear the screen
 
  Serial.println("Display initialized.");
}

void loop() {
  DateTime now = rtc.now();

  // Clear the area where the digital clock will be displayed
  tft.fillRect(0, 0, TFT_WIDTH, TFT_HEIGHT, BLACK);

  // Display digital time on TFT with rainbow colors
  tft.setTextSize(4);

  int colorIndex = now.second() % 7;
  tft.setTextColor(rainbowColors[colorIndex]);

  tft.setCursor(40, 130);
  if (now.hour() < 10) tft.print('0'); // Add leading zero
  tft.print(now.hour());
  tft.print(':');
  
  if (now.minute() < 10) tft.print('0'); // Add leading zero
  tft.print(now.minute());
  tft.print(':');
  
  if (now.second() < 10) tft.print('0'); // Add leading zero
  tft.print(now.second());

  // Display "Arvind" in the empty space
  tft.setTextSize(2);
  tft.setTextColor(WHITE);
  tft.setCursor(80, 180);
  tft.print("Arvind");

  delay(3000); // Update every second
}