#include <ESP8266WiFi.h>
#include <ESP8266HTTPClient.h>
#include <WiFiClient.h>

// WiFi Configuration
const char* ssid = "MyWiFi";       // Change this to your WiFi name
const char* password = "**********"; // Change this to your WiFi password

// Webhook Configuration
const char* webhookUrl = "http://example.com/webhook";  // Change this to your webhook URL - must not be SSL as ESP8266 doesn't support SSL

// Configuration for the rotary phone
const int phonePin = A0;           // Analog pin for the rotary phone
const int threshold = 1000;        // Threshold value for pulse detection
const int debounceDelay = 20;      // Debounce time in milliseconds
const int interDigitDelay = 700;   // Time to wait before recognizing a new digit

// Buzzer / Sound Output
const int buzzerPin = D5;          // Pin for the buzzer or speaker
const int pulseBeepDuration = 50;  // Duration of beep tone in milliseconds
const int digitBeepDuration = 200; // Duration of beep tone for recognized digits
const int digitBeepFrequency = 1000; // Pitch for recognized digits

bool pulseState = false;           // Current pulse state
bool lastPulseState = false;       // Previous pulse state
int pulseCount = 0;                // Number of pulses (dial clicks)
unsigned long lastPulseTime = 0;   // Time of last pulse
bool countingDigit = false;        // Whether we're currently counting a digit

String dialedNumber = "";          // Stores the complete dialed number
unsigned long lastDigitTime = 0;   // Time of last recognized digit
const int completeNumberDelay = 3000; // Time after last digit before number is considered complete

// WiFi Management
unsigned long lastWifiCheckTime = 0;
const int wifiCheckInterval = 30000;  // Check WiFi every 30 seconds

void setup() {
  Serial.begin(9600);
  pinMode(phonePin, INPUT);
  pinMode(buzzerPin, OUTPUT);
  
  // Disable WiFi sleep mode for more stable connection
  WiFi.setSleepMode(WIFI_NONE_SLEEP);
  
  // Connect to WiFi with higher timeout value
  connectToWifi();
  
  Serial.println("Rotary phone reader started");
  Serial.println("-------------------------");
  
  // Startup sound
  playSuccessSound();
}

// Simple tone with specified frequency and duration
void playTone(int frequency, int duration) {
  // We use a simple on/off tone for ESP8266
  // Frequency is simulated by delay between switches
  long delayValue = 1000000 / frequency / 2;
  long numCycles = frequency * duration / 1000;
  
  for (long i = 0; i < numCycles; i++) {
    digitalWrite(buzzerPin, HIGH);
    delayMicroseconds(delayValue);
    digitalWrite(buzzerPin, LOW);
    delayMicroseconds(delayValue);
  }
}

// Short tone for detected pulses
void playPulseBeep() {
  digitalWrite(buzzerPin, HIGH);
  delay(pulseBeepDuration);
  digitalWrite(buzzerPin, LOW);
}

// Tone for recognized digit
void playDigitBeep(int digit) {
  // Play a tone with frequency proportional to digit
  int freq = digitBeepFrequency + (digit * 100); // Higher tone for higher digits
  playTone(freq, digitBeepDuration);
}

// Success melody for successful sending
void playSuccessSound() {
  playTone(1000, 100);
  delay(50);
  playTone(1500, 100);
  delay(50);
  playTone(2000, 200);
}

void connectToWifi() {
  Serial.print("Connecting to WiFi ");
  Serial.print(ssid);
  Serial.println("...");
  
  WiFi.mode(WIFI_STA);  // Explicitly set station mode
  WiFi.begin(ssid, password);
  
  // Set higher timeout value for connection (10 seconds)
  int timeout = 20;  // 10 second timeout
  while (WiFi.status() != WL_CONNECTED && timeout > 0) {
    delay(500);
    Serial.print(".");
    timeout--;
  }
  
  if (WiFi.status() == WL_CONNECTED) {
    Serial.println("");
    Serial.println("WiFi connected!");
    Serial.print("IP address: ");
    Serial.println(WiFi.localIP());
    playSuccessSound(); // Tone for successful connection
  } else {
    Serial.println("");
    Serial.println("WiFi connection failed. Will try again later.");
    // Error tone for failed connection
    playTone(300, 500);
  }
}

void checkWifiConnection() {
  if (millis() - lastWifiCheckTime > wifiCheckInterval) {
    lastWifiCheckTime = millis();
    
    if (WiFi.status() != WL_CONNECTED) {
      Serial.println("WiFi connection lost. Trying to reconnect...");
      WiFi.disconnect();
      delay(1000);
      connectToWifi();
    } else {
      // Optional: Confirmation that WiFi is still connected
      // Serial.println("WiFi connection OK");
    }
  }
}

void loop() {
  // Regularly check WiFi connection
  checkWifiConnection();
  
  // Read input from rotary phone
  int sensorValue = analogRead(phonePin);
  
  // Determine pulse state (high or low)
  pulseState = (sensorValue < threshold);
  
  // Detection of falling edge (pulse start)
  if (pulseState && !lastPulseState) {
    // Debounce
    delay(debounceDelay);
    
    // Read again for confirmation
    sensorValue = analogRead(phonePin);
    if (sensorValue < threshold) {
      // Valid pulse detected
      countingDigit = true;
      pulseCount++;
      lastPulseTime = millis();
      
      Serial.print("Pulse detected! Count: ");
      Serial.println(pulseCount);
      
      // Generate short signal tone for each detected pulse
      playPulseBeep();
    }
  }
  
  // If we're counting a digit and there's a pause, interpret the digit
  if (countingDigit && (millis() - lastPulseTime > interDigitDelay)) {
    int dialedDigit = (pulseCount == 10) ? 0 : pulseCount;
    Serial.print("Dialed digit: ");
    Serial.println(dialedDigit);
    
    // Play a tone for the recognized digit
    playDigitBeep(dialedDigit);
    
    // Add the digit to the dialed number
    dialedNumber += String(dialedDigit);
    lastDigitTime = millis();
    
    // Reset for the next digit
    pulseCount = 0;
    countingDigit = false;
  }
  
  // Check if complete number was dialed (no new digit for X seconds)
  if (dialedNumber.length() > 0 && (millis() - lastDigitTime > completeNumberDelay)) {
    Serial.print("Complete dialed number: ");
    Serial.println(dialedNumber);
    
    // Send number to webhook
    sendNumberToWebhook(dialedNumber);
    
    // Reset for next number
    dialedNumber = "";
  }
  
  // Save current state for next comparison
  lastPulseState = pulseState;
  
  delay(1); // Small delay for stability
}

void sendNumberToWebhook(String number) {
  // Check WiFi connection
  if (WiFi.status() == WL_CONNECTED) {
    WiFiClient client;
    HTTPClient http;
    
    // Create URL with dialed number
    String url = String(webhookUrl) + "?number=" + number;
    
    Serial.print("Sending number to webhook: ");
    Serial.println(url);
    
    // Start HTTP connection with higher timeout
    http.begin(client, url);
    http.setTimeout(10000);  // 10 second timeout for HTTP connection
    
    // Send HTTP GET request
    int httpResponseCode = http.GET();
    
    if (httpResponseCode > 0) {
      Serial.print("HTTP response code: ");
      Serial.println(httpResponseCode);
      String payload = http.getString();
      Serial.println(payload);
      
      // Play success sound
      playSuccessSound();
    } else {
      Serial.print("Error in HTTP request. Error code: ");
      Serial.println(httpResponseCode);
      
      // Play error tone
      playTone(300, 500);
    }
    
    // End HTTP connection
    http.end();
  } else {
    Serial.println("WiFi not connected. Number will be cached and sent later.");
    // Play error tone
    playTone(300, 500);
  }
}