#include <WiFi.h>
#include <MQTT.h>

const char ssid[] = "Wokwi-GUEST";
const char pass[] = "";

WiFiClient net;
MQTTClient client;

unsigned long lastMillis = 0;

int count = 0;

void connect() {
  Serial.print("checking wifi...");
  while (WiFi.status() != WL_CONNECTED) {
    Serial.print(".");
    delay(1000);
  }

  Serial.print("\nconnecting...");
  while (!client.connect("arduino", "public", "public")) {
    Serial.print(".");
    delay(1000);
  }

  Serial.println("\nconnected!");

  client.subscribe("koalawan/iot/kontak1");
  client.subscribe("koalawan/iot/kontak2");
  client.subscribe("koalawan/iot/count");
  client.subscribe("koalawan/iot/temp");
  client.subscribe("koalawan/iot/humd");
}

void messageReceived(String &topic, String &payload) {
  Serial.println(topic + ": " + payload);
  if (topic == "koalawan/iot/kontak1") {
    Serial.println("LED1 " + payload);
    digitalWrite(19, payload.toInt());
  }

  if (topic == "koalawan/iot/kontak2") {
    Serial.println("LED2 " +payload);
    digitalWrite(18, payload.toInt());
  }
}

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

  // start wifi and mqtt
  WiFi.begin(ssid, pass);
  client.begin("public.cloud.shiftr.io", net);
  client.onMessage(messageReceived);

  pinMode(19, OUTPUT);
  pinMode(18, OUTPUT);

  connect();
}

void loop() {
  client.loop();
  delay(10);

  // check if connected
  if (!client.connected()) {
    connect();
  }

  // publish a message roughly every second.
  if (millis() - lastMillis > 2000) {
    lastMillis = millis();
    client.publish("/koalawan/iot/temperature", String(count));
    client.publish("/koalawan/iot/humidity", String(count));
    count++;
  }
}