#include <Wire.h>
#include <Adafruit_GFX.h>
#include <Adafruit_SSD1306.h>
#include <IRremote.hpp>
#include <DHT.h>

// ✅ OLED Display Configuration
#define SCREEN_WIDTH 128
#define SCREEN_HEIGHT 64
#define OLED_RESET    -1
Adafruit_SSD1306 display(SCREEN_WIDTH, SCREEN_HEIGHT, &Wire, OLED_RESET);

// ✅ Pin Definitions
#define LIGHT_BUTTON 15
#define FAN_BUTTON 2
#define ALARM_BUTTON 4
#define PIR_SENSOR 13        // Motion Detector
#define DHT_PIN 27           // Temperature & Humidity Sensor
#define DHT_TYPE DHT22
#define IR_RECEIVER 14       // IR Remote Receiver

#define LIGHT_LED 5
#define FAN_LED 18
#define ALARM_LED 19
#define BUZZER 23

// ✅ Initialize Sensors
DHT dht(DHT_PIN, DHT_TYPE);
bool lightState = false;
bool fanState = false;
bool alarmState = false;

void setup() {
    Serial.begin(115200);

    pinMode(LIGHT_BUTTON, INPUT_PULLUP);
    pinMode(FAN_BUTTON, INPUT_PULLUP);
    pinMode(ALARM_BUTTON, INPUT_PULLUP);
    pinMode(PIR_SENSOR, INPUT);

    pinMode(LIGHT_LED, OUTPUT);
    pinMode(FAN_LED, OUTPUT);
    pinMode(ALARM_LED, OUTPUT);
    pinMode(BUZZER, OUTPUT);

    dht.begin();
    IrReceiver.begin(IR_RECEIVER, ENABLE_LED_FEEDBACK);

    // ✅ Initialize OLED Display
    if (!display.begin(SSD1306_SWITCHCAPVCC, 0x3C)) {
        Serial.println("❌ OLED Display Failed!");
        for (;;);
    }
    
    displayWelcome();
}

void loop() {
    float temperature = dht.readTemperature();
    float humidity = dht.readHumidity();
    int motionDetected = digitalRead(PIR_SENSOR);

    // ✅ Update OLED with Live Status
    updateOLED(temperature, humidity, motionDetected);

    // ✅ Button Control
    handleButtonPress();

    // ✅ IR Remote Control
    handleIRRemote();

    // ✅ Automated Fan Control Based on Temperature
    if (temperature > 30.0) {
        fanState = true;
    } else {
        fanState = false;
    }
    digitalWrite(FAN_LED, fanState);

    // ✅ Security Alert for Motion Detection
    if (motionDetected) {
        alarmState = true;
        digitalWrite(ALARM_LED, HIGH);
        digitalWrite(BUZZER, HIGH);
        Serial.println("🚨 Intrusion Detected! Alarm Activated!");
        delay(2000);
        digitalWrite(BUZZER, LOW);
    }

    delay(500);
}

// ✅ Display Welcome Message on OLED
void displayWelcome() {
    display.clearDisplay();
    display.setTextSize(1);
    display.setTextColor(WHITE);
    display.setCursor(10, 10);
    display.println("Smart Home System");
    display.display();
    delay(2000);
}

// ✅ Update OLED Display with Real-Time Data
void updateOLED(float temperature, float humidity, int motion) {
    display.clearDisplay();
    display.setCursor(0, 0);
    display.print("Temp: "); display.print(temperature); display.print(" C");
    display.setCursor(0, 10);
    display.print("Humidity: "); display.print(humidity); display.print("%");
    display.setCursor(0, 20);
    display.print("Lights: "); display.print(lightState ? "ON" : "OFF");
    display.setCursor(0, 30);
    display.print("Fan: "); display.print(fanState ? "ON" : "OFF");
    display.setCursor(0, 40);
    display.print("Security: "); display.print(alarmState ? "ON" : "OFF");
    display.setCursor(0, 50);
    display.print("Motion: "); display.print(motion ? "DETECTED" : "None");
    display.display();
}

// ✅ Handle Button Presses for Manual Control
void handleButtonPress() {
    if (digitalRead(LIGHT_BUTTON) == LOW) {
        lightState = !lightState;
        digitalWrite(LIGHT_LED, lightState);
        Serial.println(lightState ? "💡 Light ON" : "💡 Light OFF");
        delay(300);
    }
    if (digitalRead(FAN_BUTTON) == LOW) {
        fanState = !fanState;
        digitalWrite(FAN_LED, fanState);
        Serial.println(fanState ? "🌀 Fan ON" : "🌀 Fan OFF");
        delay(300);
    }
    if (digitalRead(ALARM_BUTTON) == LOW) {
        alarmState = !alarmState;
        digitalWrite(ALARM_LED, alarmState);
        digitalWrite(BUZZER, alarmState);
        Serial.println(alarmState ? "🚨 Alarm Activated!" : "🔕 Alarm Deactivated");
        delay(300);
    }
}

// ✅ Handle IR Remote Control
void handleIRRemote() {
    if (IrReceiver.decode()) {
        Serial.print("IR Code: ");
        Serial.println(IrReceiver.decodedIRData.command, HEX);

        switch (IrReceiver.decodedIRData.command) {
            case 0x1:  // Example IR Code for Light
                lightState = !lightState;
                digitalWrite(LIGHT_LED, lightState);
                Serial.println(lightState ? "💡 Light ON via Remote" : "💡 Light OFF via Remote");
                break;
            case 0x2:  // Example IR Code for Fan
                fanState = !fanState;
                digitalWrite(FAN_LED, fanState);
                Serial.println(fanState ? "🌀 Fan ON via Remote" : "🌀 Fan OFF via Remote");
                break;
            case 0x3:  // Example IR Code for Alarm
                alarmState = !alarmState;
                digitalWrite(ALARM_LED, alarmState);
                digitalWrite(BUZZER, alarmState);
                Serial.println(alarmState ? "🚨 Alarm Activated via Remote" : "🔕 Alarm Deactivated via Remote");
                break;
        }
        IrReceiver.resume(); // Receive the next IR signal
    }
}