#include <WiFi.h>
#include <ArduinoWebsockets.h>

using namespace websockets;

const int relayPin = 23;
const int ledPin = 22;

WebsocketsClient webSocket;

void setup() {
  Serial.begin(115200);
  pinMode(relayPin, OUTPUT);
  pinMode(ledPin, OUTPUT);
  digitalWrite(relayPin, LOW);
  digitalWrite(ledPin, LOW);

  // Connect to Wi-Fi
  WiFi.begin("HUAWEI-B315-2B2D", "R7AH4Q16ALH");
  while (WiFi.status() != WL_CONNECTED) {
    delay(500);
    Serial.print(".");
  }
  Serial.println("Connected to WiFi");

  // Set up WebSocket
  webSocket.onMessage(onMessage);
  webSocket.connect("ws://192.168.129.1:8080"); // Ensure this matches your WebSocket server setup
}

void loop() {
  webSocket.poll();
}

void onMessage(WebsocketsMessage message) {
  String msg = message.data();
  if (msg == "on") {
    digitalWrite(relayPin, HIGH);
    digitalWrite(ledPin, HIGH);
  } else if (msg == "off") {
    digitalWrite(relayPin, LOW);
    digitalWrite(ledPin, LOW);
  }
}