/*
 * Advanced Vape OLED Features
 * Temperature Monitor, Clock, Menu System
 */

#include <DHT.h>
#include <RTClib.h>

// Sensor configuration
#define DHT_PIN 2
#define DHT_TYPE DHT22
DHT dht(DHT_PIN, DHT_TYPE);
RTC_DS3231 rtc;

// Menu system
int currentMenu = 0;
const int maxMenus = 4;
String menuItems[] = {"Home", "Sensors", "Clock", "Settings"};

// Button pins
#define BUTTON_UP 3
#define BUTTON_DOWN 4
#define BUTTON_SELECT 5

void setup() {
  Serial.begin(9600);
  
  // Initialize sensors
  dht.begin();
  
  if (!rtc.begin()) {
    Serial.println("❌ RTC not found");
  }
  
  // Initialize buttons
  pinMode(BUTTON_UP, INPUT_PULLUP);
  pinMode(BUTTON_DOWN, INPUT_PULLUP);
  pinMode(BUTTON_SELECT, INPUT_PULLUP);
  
  // Initialize display (same as previous code)
  if(!display.begin(SSD1306_SWITCHCAPVCC, SCREEN_ADDRESS)) {
    Serial.println("❌ SSD1306 allocation failed");
    for(;;);
  }
  
  Serial.println("✅ Advanced OLED system ready!");
}

void loop() {
  handleButtons();
  updateDisplay();
  delay(100);
}

void handleButtons() {
  static unsigned long lastButtonPress = 0;
  unsigned long currentTime = millis();
  
  // Debounce buttons
  if (currentTime - lastButtonPress < 200) return;
  
  if (digitalRead(BUTTON_UP) == LOW) {
    currentMenu = (currentMenu - 1 + maxMenus) % maxMenus;
    lastButtonPress = currentTime;
    Serial.println("⬆️ Menu Up: " + menuItems[currentMenu]);
  }
  
  if (digitalRead(BUTTON_DOWN) == LOW) {
    currentMenu = (currentMenu + 1) % maxMenus;
    lastButtonPress = currentTime;
    Serial.println("⬇️ Menu Down: " + menuItems[currentMenu]);
  }
  
  if (digitalRead(BUTTON_SELECT) == LOW) {
    lastButtonPress = currentTime;
    Serial.println("✅ Selected: " + menuItems[currentMenu]);
  }
}

void updateDisplay() {
  display.clearDisplay();
  
  switch(currentMenu) {
    case 0: showHomeScreen(); break;
    case 1: showSensorScreen(); break;
    case 2: showClockScreen(); break;
    case 3: showSettingsScreen(); break;
  }
  
  // Draw menu indicator
  drawMenuIndicator();
  
  display.display();
}

void showHomeScreen() {
  // Header
  display.setTextSize(2);
  display.setTextColor(SSD1306_WHITE);
  display.setCursor(0, 0);
  display.println("HOME");
  
  // Quick info
  display.setTextSize(1);
  display.setCursor(0, 20);
  display.print("Temp: ");
  display.print(dht.readTemperature(), 1);
  display.println("C");
  
  display.setCursor(0, 30);
  display.print("Humidity: ");
  display.print(dht.readHumidity(), 1);
  display.println("%");
  
  // Time
  DateTime now = rtc.now();
  display.setCursor(0, 40);
  display.print("Time: ");
  if (now.hour() < 10) display.print("0");
  display.print(now.hour());
  display.print(":");
  if (now.minute() < 10) display.print("0");
  display.print(now.minute());
  
  // Status
  display.setCursor(0, 50);
  display.print("Status: ");
  display.print(millis() % 2000 < 1000 ? "ACTIVE" : "STANDBY");
}

void showSensorScreen() {
  display.setTextSize(1);
  display.setTextColor(SSD1306_WHITE);
  display.setCursor(0, 0);
  display.println("=== SENSORS ===");
  
  // Temperature
  float temp = dht.readTemperature();
  display.setCursor(0, 12);
  display.print("Temperature:");
  display.setCursor(0, 22);
  display.setTextSize(2);
  display.print(temp, 1);
  display.print("C");
  
  // Humidity
  float humidity = dht.readHumidity();
  display.setTextSize(1);
  display.setCursor(0, 40);
  display.print("Humidity:");
  display.setCursor(0, 50);
  display.setTextSize(2);
  display.print(humidity, 1);
  display.print("%");
  
  // Heat index
  float heatIndex = dht.computeHeatIndex(temp, humidity, false);
  display.setTextSize(1);
  display.setCursor(70, 50);
  display.print("HI:");
  display.print(heatIndex, 1);
}

void showClockScreen() {
  DateTime now = rtc.now();
  
  // Large time display
  display.setTextSize(2);
  display.setTextColor(SSD1306_WHITE);
  display.setCursor(10, 10);
  
  if (now.hour() < 10) display.print("0");
  display.print(now.hour());
  display.print(":");
  if (now.minute() < 10) display.print("0");
  display.print(now.minute());
  
  // Seconds
  display.setTextSize(1);
  display.setCursor(100, 20);
  if (now.second() < 10) display.print("0");
  display.print(now.second());
  
  // Date
  display.setCursor(0, 35);
  display.print("Date: ");
  display.print(now.day());
  display.print("/");
  display.print(now.month());
  display.print("/");
  display.print(now.year());
  
  // Day of week
  String days[] = {"Sun", "Mon", "Tue", "Wed", "Thu", "Fri", "Sat"};
  display.setCursor(0, 45);
  display.print("Day: ");
  display.print(days[now.dayOfTheWeek()]);
  
  // Uptime
  display.setCursor(0, 55);
  display.print("Uptime: ");
  display.print(millis() / 1000);
  display.print("s");
}

void showSettingsScreen() {
  display.setTextSize(1);
  display.setTextColor(SSD1306_WHITE);
  display.setCursor(0, 0);
  display.println("=== SETTINGS ===");
  
  // System info
  display.setCursor(0, 12);
  display.print("Display: 128x64");
  
  display.setCursor(0, 22);
  display.print("I2C: 0x");
  display.print(SCREEN_ADDRESS, HEX);
  
  display.setCursor(0, 32);
  display.print("Free RAM: ");
  display.print(getFreeRAM());
  
  display.setCursor(0, 42);
  display.print("Version: 1.0");
  
  display.setCursor(0, 52);
  display.print("Global Byte Shop");
}

void drawMenuIndicator() {
  // Draw menu dots at bottom
  int dotY = 58;
  int startX = 40;
  
  for (int i = 0; i < maxMenus; i++) {
    int dotX = startX + (i * 12);
    
    if (i == currentMenu) {
      display.fillCircle(dotX, dotY, 3, SSD1306_WHITE);
    } else {
      display.drawCircle(dotX, dotY, 2, SSD1306_WHITE);
    }
  }
}