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

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

const char* WIFI_SSID = "Wokwi-GUEST";
const char* WIFI_PASSWORD = "";
const char* WEATHER_API_KEY = "9297b3c9acfb1f7f00c8e45bd95a6180";
const char* WEATHER_API_URL = "https://api.openweathermap.org/data/2.5/weather?lat=31.374716196088148&lon=75.5450096440109&appid=9297b3c9acfb1f7f00c8e45bd95a6180";

void setup() {
  Serial.begin(115200);
  delay(1000);
  Serial.println("Connecting to Wi-Fi...");

  WiFi.begin(WIFI_SSID, WIFI_PASSWORD);
  while (WiFi.status() != WL_CONNECTED) {
    delay(1000);
    Serial.println("Connecting...");
  }
  Serial.println("Connected to Wi-Fi.");

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

  String url = String(WEATHER_API_URL) + String(WEATHER_API_KEY);
  Serial.println("Requesting weather data...");
  Serial.print(url);
  HTTPClient http;
  http.begin(url);
  int httpCode = http.GET();
  if (httpCode == HTTP_CODE_OK) {
    String payload = http.getString();
    DynamicJsonDocument doc(1024);
    deserializeJson(doc, payload);
    float temp = doc["main"]["temp"];
    float feelsLike = doc["main"]["feels_like"];
    float humidity = doc["main"]["humidity"];
    float windSpeed = doc["wind"]["speed"];

    display.setTextSize(1);
    display.setTextColor(SSD1306_WHITE);
    display.setCursor(0, 0);
    display.print("Temperature:");
    display.print(temp);
    display.print(" C");

    display.setCursor(0, 12);
    display.print("Feels like:");
    display.print(feelsLike);
    display.print(" C");

    display.setCursor(0, 24);
    display.print("Humidity:");
    display.print(humidity);
    display.print(" %");

    display.setCursor(0, 36);
    display.print("Wind speed:");
    display.print(windSpeed);
    display.print(" m/s");

    display.display();
  } else {
    Serial.printf("Error getting weather data. HTTP code: %d\n", httpCode);
  }
  http.end();
}

void loop() {
  // Nothing to do here
}