Live Bus Arrival Display

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

// Troque estas duas strings pelos dados da sua rede Wi-Fi antes de usar.

struct BusInfo {
  const char *line;
  const char *destination;
  int minutes;
};


// Forward declarations
bool wifiIsConnected();
void connectWiFi();
void loadSimulatedBusData();
void setStatusLEDs(int minutes);
void setDisplayBrightnessIfNeeded();
void drawBusScreen();
void drawOfflineScreen();
void updateArrivalAlert();
void readButton();
void refreshVisibleState();

const char *WIFI_SSID = "SEU_WIFI";
const char *WIFI_PASSWORD = "SUA_SENHA";

const int OLED_SDA = 23;
const int OLED_SCL = 22;
const int LED_GREEN_PIN = 25;
const int LED_YELLOW_PIN = 26;
const int LED_RED_PIN = 27;
const int BUTTON_PIN = 33;
const int BUZZER_PIN = 18;
const int LDR_PIN = 34;

const unsigned long UPDATE_INTERVAL_MS = 30000UL;
const unsigned long BUTTON_DEBOUNCE_MS = 250UL;
const unsigned long RECONNECT_INTERVAL_MS = 10000UL;
const int NIGHT_LIGHT_THRESHOLD = 1400;
const uint8_t OLED_BRIGHTNESS_DAY = 0xCF;
const uint8_t OLED_BRIGHTNESS_NIGHT = 0x20;



BusInfo buses[3] = {
  {"431", "Centro", 3},
  {"702", "Terminal Norte", 9},
  {"815", "Rodoviaria", 18}
};

Adafruit_SSD1306 display(128, 64, &Wire, -1);

int selectedBus = 0;
unsigned long lastDataUpdateMs = 0;
unsigned long lastButtonMs = 0;
unsigned long lastReconnectMs = 0;
bool lastButtonState = HIGH;
bool alertAlreadyPlayed = false;
bool screenNeedsRedraw = true;
bool lastWifiState = false;
int lastBrightnessMode = -1;

bool wifiIsConnected() {
  return WiFi.status() == WL_CONNECTED;
}

void connectWiFi() {
  WiFi.mode(WIFI_STA);
  WiFi.begin(WIFI_SSID, WIFI_PASSWORD);
}

void loadSimulatedBusData() {
  // Ponto de troca para uma API real: após ler HTTP/JSON, atualize
  // buses[i].line, buses[i].destination e buses[i].minutes nesta função.
  static int cycle = 0;
  const int sampleMinutes[3][3] = {
    {3, 9, 18},
    {2, 8, 16},
    {1, 6, 14}
  };

  for (int i = 0; i < 3; i++) {
    buses[i].minutes = sampleMinutes[cycle][i];
  }
  cycle = (cycle + 1) % 3;
}

void setStatusLEDs(int minutes) {
  digitalWrite(LED_GREEN_PIN, minutes <= 5 ? HIGH : LOW);
  digitalWrite(LED_YELLOW_PIN, (minutes >= 6 && minutes <= 15) ? HIGH : LOW);
  digitalWrite(LED_RED_PIN, minutes > 15 ? HIGH : LOW);
}

void setDisplayBrightnessIfNeeded() {
  int brightnessMode = analogRead(LDR_PIN) < NIGHT_LIGHT_THRESHOLD ? 0 : 1;
  if (brightnessMode == lastBrightnessMode) {
    return;
  }

  // O OLED físico aceita contraste por comando; o simulador mantém o brilho padrão.
#ifndef SCHEMATIK_SIMULATOR
  display.ssd1306_command(0x81); // comando SSD1306: ajustar contraste
  display.ssd1306_command(brightnessMode == 0 ? OLED_BRIGHTNESS_NIGHT : OLED_BRIGHTNESS_DAY);
#endif
  lastBrightnessMode = brightnessMode;
}

void drawBusScreen() {
  display.clearDisplay();
  display.setTextSize(1);
  display.setTextColor(SSD1306_WHITE);
  display.setCursor(0, 0);
  display.print("PONTO VIVO  Wi-Fi OK");
  display.drawLine(0, 10, 127, 10, SSD1306_WHITE);

  for (int i = 0; i < 3; i++) {
    const int y = 14 + i * 16;
    if (i == selectedBus) {
      display.fillRect(0, y - 1, 128, 14, SSD1306_WHITE);
      display.setTextColor(SSD1306_BLACK);
    } else {
      display.setTextColor(SSD1306_WHITE);
    }

    display.setCursor(2, y);
    display.print(buses[i].line);
    display.print(" ");
    display.print(buses[i].destination);
    display.setCursor(104, y);
    display.print(buses[i].minutes);
    display.print("m");
  }

  display.setTextColor(SSD1306_WHITE);
  display.setCursor(0, 56);
  display.print("Botao: proxima linha");
  display.display();
}

void drawOfflineScreen() {
  display.clearDisplay();
  display.setTextSize(1);
  display.setTextColor(SSD1306_WHITE);
  display.setCursor(15, 12);
  display.print("SEM INTERNET");
  display.setCursor(4, 30);
  display.print("Verifique o Wi-Fi");
  display.setCursor(6, 46);
  display.print("Tentando reconectar");
  display.display();
}

void updateArrivalAlert() {
  bool busIsClose = buses[selectedBus].minutes < 2;
  if (busIsClose && !alertAlreadyPlayed) {
    ledcWriteTone(0, 2200);
    delay(180);
    ledcWriteTone(0, 0);
    alertAlreadyPlayed = true;
  }
  if (!busIsClose) {
    alertAlreadyPlayed = false;
  }
}

void readButton() {
  bool buttonState = digitalRead(BUTTON_PIN);
  if (lastButtonState == HIGH && buttonState == LOW && millis() - lastButtonMs >= BUTTON_DEBOUNCE_MS) {
    selectedBus = (selectedBus + 1) % 3;
    lastButtonMs = millis();
    alertAlreadyPlayed = false;
    screenNeedsRedraw = true;
  }
  lastButtonState = buttonState;
}

void refreshVisibleState() {
  bool wifiNow = wifiIsConnected();
  setDisplayBrightnessIfNeeded();

  if (wifiNow != lastWifiState) {
    screenNeedsRedraw = true;
    lastWifiState = wifiNow;
  }

  if (wifiNow) {
    setStatusLEDs(buses[selectedBus].minutes);
    updateArrivalAlert();
    if (screenNeedsRedraw) {
      drawBusScreen();
      screenNeedsRedraw = false;
    }
  } else {
    setStatusLEDs(99);
    if (screenNeedsRedraw) {
      drawOfflineScreen();
      screenNeedsRedraw = false;
    }
  }
}

void setup() {
  pinMode(LED_GREEN_PIN, OUTPUT);
  pinMode(LED_YELLOW_PIN, OUTPUT);
  pinMode(LED_RED_PIN, OUTPUT);
  pinMode(BUTTON_PIN, INPUT_PULLUP);
  pinMode(LDR_PIN, INPUT);

  ledcSetup(0, 2000, 8);
  ledcAttachPin(BUZZER_PIN, 0);

  Wire.begin(OLED_SDA, OLED_SCL);
  if (!display.begin(SSD1306_SWITCHCAPVCC, 0x3C)) {
    while (true) {
      delay(1000);
    }
  }

  connectWiFi();
  loadSimulatedBusData();
  lastDataUpdateMs = millis();
  refreshVisibleState();
}

void loop() {
  readButton();

  if (!wifiIsConnected() && millis() - lastReconnectMs >= RECONNECT_INTERVAL_MS) {
    WiFi.reconnect();
    lastReconnectMs = millis();
  }

  if (wifiIsConnected() && millis() - lastDataUpdateMs >= UPDATE_INTERVAL_MS) {
    loadSimulatedBusData();
    lastDataUpdateMs = millis();
    screenNeedsRedraw = true;
  }

  refreshVisibleState();
  delay(20);
}