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

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

// Wi-Fi credentials
const char* ssid = "Wokwi-GUEST";
const char* password = "";

// Spotify API credentials
String accessToken = "BQA_mbX6RrQqChnpz45vfe89bLND5t8CQgAYtsEXxZwdWpg-vT6HRiJU_yxEPU6JiuASkevq8e-QSd2nzUWaf_WDkKeoUOwfVsYGTn4vlZ_3N6CEmXT6Mkj-YME4dQrH8sPK5xlIlYP1AKkRVu6ieOmMpUWrYO-5E6aD0CVTng43OGvPLQ2O0SYpekdix5gG62jHfNNI9RioiY1tx_w"; // Replace with your token
String apiURL = "https://api.spotify.com/v1/me/player/currently-playing";

void setup() {
  // Initialize Serial for debugging
  Serial.begin(115200);

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

  // Connect to Wi-Fi
  connectToWiFi();

  // Initial display message
  display.setTextSize(1);
  display.setTextColor(SSD1306_WHITE);
  display.setCursor(0, 0);
  display.println("Connected to WiFi");
  display.display();
  delay(2000);
}

void loop() {
  if (WiFi.status() == WL_CONNECTED) {
    // Fetch song data from Spotify API
    String songData = getSpotifyData();
    if (!songData.isEmpty()) {
      displaySongInfo(songData);
    } else {
      displayError("No song data!");
    }
  } else {
    displayError("Wi-Fi disconnected");
  }

  delay(5000); // Refresh every 5 seconds
}

void connectToWiFi() {
  display.clearDisplay();
  display.setCursor(0, 0);
  display.println("Connecting to WiFi...");
  display.display();

  WiFi.begin(ssid, password);
  while (WiFi.status() != WL_CONNECTED) {
    delay(500);
    Serial.print(".");
    display.print(".");
    display.display();
  }

  Serial.println("\nWiFi connected!");
  display.clearDisplay();
  display.setCursor(0, 0);
  display.println("WiFi connected!");
  display.display();
}

String getSpotifyData() {
  HTTPClient http;
  http.begin(apiURL);
  http.addHeader("Authorization", "Bearer " + accessToken);

  int httpResponseCode = http.GET();
  if (httpResponseCode == 200) {
    String payload = http.getString();
    http.end();
    return payload;
  } else {
    Serial.print("Error code: ");
    Serial.println(httpResponseCode);
    http.end();
    return "";
  }
}

void displaySongInfo(String payload) {
  // Extract song title and artist (use JSON parsing library like ArduinoJson for more robust handling)
  String songTitle = extractJsonValue(payload, "name");
  String artistName = extractJsonValue(payload, "artists[0].name");

  display.clearDisplay();
  display.setCursor(0, 0);
  display.println("Now Playing:");
  display.setCursor(0, 10);
  display.println(songTitle);
  display.setCursor(0, 30);
  display.println("Artist:");
  display.setCursor(0, 40);
  display.println(artistName);
  display.display();
}

void displayError(String message) {
  display.clearDisplay();
  display.setCursor(0, 0);
  display.println("Error:");
  display.setCursor(0, 10);
  display.println(message);
  display.display();
}

String extractJsonValue(String json, String key) {
  int start = json.indexOf("\"" + key + "\"") + key.length() + 3;
  int end = json.indexOf("\"", start);
  return json.substring(start, end);
}