#include <Wire.h>
#include <Adafruit_SSD1306.h>
#include <Adafruit_GFX.h>

#define SCREEN_WIDTH 128
#define SCREEN_HEIGHT 64
#define OLED_RESET -1
Adafruit_SSD1306 display(SCREEN_WIDTH, SCREEN_HEIGHT, &Wire, OLED_RESET);

// Define random test values
float voltageReading = 12.34;
float currentReading = 0.56;
float temperature = 25.0;
unsigned long uptime = 12345;

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

    // Initialize OLED Display
    if (!display.begin(SSD1306_SWITCHCAPVCC, 0x3C)) {
        Serial.println("SSD1306 allocation failed");
        for (;;);
    }
    display.clearDisplay();
    display.setTextSize(1);
    display.setTextColor(WHITE);

    // Initial Display Test
    display.setCursor(0, 0);
    display.print("Cursor Test:");
    display.setCursor(0, 10);
    display.print("Voltage: ");
    display.print(voltageReading);
    display.setCursor(0, 20);
    display.print("Current: ");
    display.print(currentReading);
    display.setCursor(0, 30);
    display.print("Temp: ");
    display.print(temperature);
    display.setCursor(0, 40);
    display.print("Uptime: ");
    display.print(uptime);

    display.display();  // Show the above values
    delay(2000);
}

void loop() {
    // Update the display with random values to test cursor placement
    display.clearDisplay();

    display.setCursor(0, 0);
    display.print("Random Value Test");

    // Set random values
    voltageReading = random(1000) / 100.0;   // Voltage between 0.00 and 10.00
    currentReading = random(100) / 100.0;    // Current between 0.00 and 1.00
    temperature = random(1500) / 50.0;       // Temperature between 0.00 and 30.00
    uptime = random(10000);                   // Random uptime in seconds

    display.setCursor(0, 10);
    display.print("Voltage: ");
    display.print(voltageReading);

    display.setCursor(0, 20);
    display.print("Current: ");
    display.print(currentReading);

    display.setCursor(0, 30);
    display.print("Temp: ");
    display.print(temperature);

    display.setCursor(0, 40);
    display.print("Uptime: ");
    display.print(uptime);

    // Display Relay 1 and Relay 2 state
    display.setCursor(0, 54);
    display.print("L1: ");
    // display.print("ON ");
    display.print("OFF");
    
    display.setCursor(65, 54);
    display.print("L2: ");
    // display.print("ON");
    display.print("OFF");

    display.display();  // Update the OLED display with the new values
    delay(2000);
}