#include <IRremote.h>
#include <EEPROM.h>

const int irReceiverPin = 2;   // Pin connected to the IR receiver module
const int priceAddress = 0;    // EEPROM address to store the price
float price = 100.0;            // Initial price value
bool saveData = false;          // Flag to indicate when to save data
bool inMenu = false;            // Flag to indicate when in the menu
bool changePriceOption = false; // Flag to indicate price change option selected

IRrecv IR(irReceiverPin);
decode_results results;

void setup() {
  Serial.begin(9600);
  IR.enableIRIn();              // Start the IR receiver
  EEPROM.begin(512);            // Initialize EEPROM
  loadPriceFromEEPROM();        // Load the price from EEPROM on startup
}

void loadPriceFromEEPROM() {
  price = EEPROM.read(priceAddress);
  Serial.println(price);
}

void savePriceToEEPROM() {
  EEPROM.write(priceAddress, price);
  EEPROM.commit();
  Serial.println("Price saved");
}

void showMenu() {
  Serial.println("Menu:");
  Serial.println("1. Change Price");
  Serial.println("2. Show Saved Price");
}

void loop() {
  if (IR.decode()) {
    long long key = IR.decodedIRData.decodedRawData;

    if (!inMenu) {
      // Not in the menu, check for menu button
      if (key == 0x57A8FF00) { // Replace with the actual IR code for the "Menu" button
        inMenu = true;
        showMenu();
      }
    } else {
      // In the menu, check for menu options
      switch (key) {
        case 0xCF30FF00: // Replace with the actual IR code for the "1" button
          // Change Price
          Serial.println("Enter new price:");
          // Implement code to receive and set the new price here
          changePriceOption = true;
          inMenu = true;
          break;

        case 0xE718FF00: // Replace with the actual IR code for the "2" button
          // Show Saved Price
          Serial.println("Saved Price: " + String(price));
          inMenu = false;
          break;
      }
    }

    if (changePriceOption) {
      // Handle price change or save based on your existing code
      switch (key) {
        case 0xFD02FF00: // Increase price
          Serial.println("+");
          price += 0.01;
          Serial.println(price);
          saveData = true; // Set the flag to save data
          break;

        case 0x6798FF00: // Decrease price
          Serial.println("-");
          price -= 0.01;
          Serial.println(price);
          saveData = true; // Set the flag to save data
          break;

        case 0x6F90FF00: // Replace with the actual IR code for the "OK" button
          if (saveData) {
            savePriceToEEPROM(); // Save the updated price to EEPROM when "OK" is pressed
            saveData = false;   // Reset the flag
            changePriceOption = false;
            inMenu = false;

          }
          break;
      }
    }

    delay(100);
    IR.resume();
  }
}