#include <WiFi.h>
#include <WebServer.h>
#include <OneWire.h>
#include <DallasTemperature.h>

// Pin configuration
const int oneWirePin = 5;       // GPIO5 for DS18B20
const int ledPin = 2;           // Built-in LED
const int buttonTempPin = 15;   // GPIO15 for temperature button
const int buttonLedPin = 16;    // GPIO16 for LED control button
const int pin1 = 17;            // Additional pin 1
const int pin2 = 18;            // Additional pin 2

// WiFi client settings
const char* ssid = "Black hole";
const char* password = "753oe762";

// Access Point settings
const char* apSSID = "ESP32_Direct";
const char* apPassword = "12345678"; // Minimum 8 characters

// Web server on port 80
WebServer server(80);

// OneWire and DallasTemperature setup
OneWire oneWire(oneWirePin);
DallasTemperature sensors(&oneWire);

// Variables
float tempC = 0;
bool ledState = false;
bool pin1State = false;
bool pin2State = false;
unsigned long lastTempUpdate = 0;
const long tempUpdateInterval = 10000;

// Button states
int lastButtonTempState = HIGH;
int lastButtonLedState = HIGH;
unsigned long lastDebounceTime = 0;
const long debounceDelay = 50;

void setup() {
  Serial.begin(115200);
  
  // Initialize pins
  pinMode(ledPin, OUTPUT);
  pinMode(pin1, OUTPUT);
  pinMode(pin2, OUTPUT);
  pinMode(buttonTempPin, INPUT_PULLUP);
  pinMode(buttonLedPin, INPUT_PULLUP);
  digitalWrite(ledPin, LOW);
  digitalWrite(pin1, LOW);
  digitalWrite(pin2, LOW);
  
  // Initialize temperature sensor
  sensors.begin();
  
  // Start both modes simultaneously
  initWiFi();
  
  // Configure web server
  server.on("/", handleRoot);
  server.on("/temperature", handleTemperature);
  server.on("/led", handleLed);
  server.on("/pin1", handlePin1);
  server.on("/pin2", handlePin2);
  server.begin();
  Serial.println("HTTP server started");
}

void initWiFi() {
  // Start Access Point mode
  WiFi.softAP(apSSID, apPassword);
  Serial.println();
  Serial.println("AP Mode Activated");
  Serial.print("AP SSID: ");
  Serial.println(apSSID);
  Serial.print("AP Password: ");
  Serial.println(apPassword);
  Serial.print("AP IP address: ");
  Serial.println(WiFi.softAPIP());

  // Simultaneously try to connect to router
  WiFi.begin(ssid, password);
  Serial.print("Connecting to WiFi network...");
  
  // Don't wait for connection to avoid blocking
  // Connection will happen in background
}

void loop() {
  server.handleClient();
  
  // Update temperature
  if (millis() - lastTempUpdate >= tempUpdateInterval) {
    updateTemperature();
  }
  
  // Handle buttons
  handleButtons();
  
  delay(10);
}

void updateTemperature() {
  sensors.requestTemperatures();
  tempC = sensors.getTempCByIndex(0);
  lastTempUpdate = millis();
  Serial.print("Temperature updated: ");
  Serial.print(tempC);
  Serial.println(" °C");
}

void handleButtons() {
  // Read button states with debounce
  int currentButtonTempState = digitalRead(buttonTempPin);
  int currentButtonLedState = digitalRead(buttonLedPin);
  
  if ((millis() - lastDebounceTime) > debounceDelay) {
    // Temperature button handling
    if (currentButtonTempState != lastButtonTempState) {
      lastButtonTempState = currentButtonTempState;
      if (currentButtonTempState == LOW) {
        Serial.println("Temperature button pressed");
        updateTemperature();
        Serial.print("Current temperature: ");
        Serial.print(tempC);
        Serial.println(" °C");
      }
    }
    
    // LED button handling
    if (currentButtonLedState != lastButtonLedState) {
      lastButtonLedState = currentButtonLedState;
      if (currentButtonLedState == LOW) {
        ledState = !ledState;
        digitalWrite(ledPin, ledState ? HIGH : LOW);
        Serial.print("LED turned ");
        Serial.println(ledState ? "ON" : "OFF");
      }
    }
  }
  
  if (currentButtonTempState != lastButtonTempState || currentButtonLedState != lastButtonLedState) {
    lastDebounceTime = millis();
  }
}

String getConnectionInfo() {
  String info;
  
  if (WiFi.status() == WL_CONNECTED) {
    info += "<div class=\"connection-info router\">";
    info += "<h3>Router Connection</h3>";
    info += "SSID: <strong>" + String(ssid) + "</strong><br>";
    info += "IP address: <strong>" + WiFi.localIP().toString() + "</strong><br>";
    info += "Signal strength: <strong>" + String(WiFi.RSSI()) + " dBm</strong>";
    info += "</div>";
  } else {
    info += "<div class=\"connection-info router disconnected\">";
    info += "<h3>Router Connection</h3>";
    info += "Status: <strong class=\"disconnected\">Not connected</strong>";
    info += "</div>";
  }
  
  info += "<div class=\"connection-info direct\">";
  info += "<h3>Direct Connection</h3>";
  info += "SSID: <strong>" + String(apSSID) + "</strong><br>";
  info += "Password: <strong>" + String(apPassword) + "</strong><br>";
  info += "IP address: <strong>" + WiFi.softAPIP().toString() + "</strong>";
  info += "</div>";
  
  return info;
}

void handleRoot() {
  String html = "<!DOCTYPE html><html><head>";
  html += "<title>ESP32 Dual Mode Control Panel</title>";
  html += "<meta name=\"viewport\" content=\"width=device-width, initial-scale=1\">";
  html += "<style>";
  html += "body {font-family: Arial, sans-serif; margin: 0 auto; max-width: 800px; padding: 20px;}";
  html += "h1 {color: #444; text-align: center;}";
  html += ".btn {";
  html += "  background-color: #4CAF50;";
  html += "  border: none;";
  html += "  color: white;";
  html += "  padding: 15px 32px;";
  html += "  text-align: center;";
  html += "  text-decoration: none;";
  html += "  display: inline-block;";
  html += "  font-size: 16px;";
  html += "  margin: 10px 5px;";
  html += "  cursor: pointer;";
  html += "  border-radius: 5px;";
  html += "  box-shadow: 0 4px 8px 0 rgba(0,0,0,0.2);";
  html += "}";
  html += ".btn:active {";
  html += "  background-color: #3e8e41;";
  html += "  box-shadow: 0 2px 4px 0 rgba(0,0,0,0.2);";
  html += "  transform: translateY(2px);";
  html += "}";
  html += ".btn-on {background-color: #4CAF50;}";
  html += ".btn-off {background-color: #f44336;}";
  html += ".temp {font-size: 24px; margin: 20px; text-align: center;}";
  html += ".status {font-size: 18px; margin: 10px; text-align: center;}";
  html += ".connection-info {";
  html += "  margin: 20px 0;";
  html += "  padding: 15px;";
  html += "  border-radius: 5px;";
  html += "  background-color: #f8f8f8;";
  html += "}";
  html += ".router {border-left: 5px solid #2196F3;}";
  html += ".direct {border-left: 5px solid #FF9800;}";
  html += ".disconnected {color: #f44336;}";
  html += ".controls {text-align: center;}";
  html += "</style>";
  html += "</head><body>";
  
  html += "<h1>ESP32 Dual Mode Control Panel</h1>";
  
  // Connection information
  html += getConnectionInfo();
  
  html += "<div class=\"temp\">Current temperature: " + String(tempC) + " °C</div>";
  
  // Temperature button
  html += "<div class=\"controls\"><a href=\"/temperature\"><button class=\"btn\">Update Temperature</button></a></div>";
  
  // LED button
  html += "<div class=\"status\">LED: " + String(ledState ? "ON" : "OFF") + "</div>";
  html += "<div class=\"controls\"><a href=\"/led\"><button class=\"btn " + String(ledState ? "btn-on" : "btn-off") + "\">";
  html += String(ledState ? "Turn OFF LED" : "Turn ON LED");
  html += "</button></a></div>";
  
  // Pin1 button
  html += "<div class=\"status\">GPIO17: " + String(pin1State ? "ON" : "OFF") + "</div>";
  html += "<div class=\"controls\"><a href=\"/pin1\"><button class=\"btn " + String(pin1State ? "btn-on" : "btn-off") + "\">";
  html += String(pin1State ? "Turn OFF GPIO17" : "Turn ON GPIO17");
  html += "</button></a></div>";
  
  // Pin2 button
  html += "<div class=\"status\">GPIO18: " + String(pin2State ? "ON" : "OFF") + "</div>";
  html += "<div class=\"controls\"><a href=\"/pin2\"><button class=\"btn " + String(pin2State ? "btn-on" : "btn-off") + "\">";
  html += String(pin2State ? "Turn OFF GPIO18" : "Turn ON GPIO18");
  html += "</button></a></div>";
  
  html += "</body></html>";
  
  server.send(200, "text/html", html);
}

void handleTemperature() {
  updateTemperature();
  server.sendHeader("Location", "/");
  server.send(303);
}

void handleLed() {
  ledState = !ledState;
  digitalWrite(ledPin, ledState ? HIGH : LOW);
  server.sendHeader("Location", "/");
  server.send(303);
}

void handlePin1() {
  pin1State = !pin1State;
  digitalWrite(pin1, pin1State ? HIGH : LOW);
  server.sendHeader("Location", "/");
  server.send(303);
}

void handlePin2() {
  pin2State = !pin2State;
  digitalWrite(pin2, pin2State ? HIGH : LOW);
  server.sendHeader("Location", "/");
  server.send(303);
}