import network
import time
from machine import Pin
import dht
import ujson
from umqtt.simple import MQTTClient
from machine import Pin

# WIFI CONFIGURATION
SSID = "Wokwi-GUEST"

# MQTT Server Parameters
MQTT_CLIENT_ID = "Muhammad_Arrafi_Reva_Razqana_Arassy"
MQTT_BROKER    = "broker.emqx.io"
MQTT_USER      = ""
MQTT_PASSWORD  = ""
MQTT_TOPIC     = "/STRIKE!!/Muhammad_Arrafi_Reva_Razqana_Arassy/data_sensor"
MQTT_TOPICLED  = "/STRIKE!!/Muhammad_Arrafi_Reva_Razqana_Arassy/aktuasi_led"

""" Sensor DHT22 & LED & PushButton """
sensor1 = dht.DHT22(Pin(15))
led = Pin(12, Pin.OUT)
pushButton=Pin(32,Pin.IN)

print("Halo, Selamat Datang di ESP32!")
time.sleep(1)

""" Koneksi WiFi """
# WIFI Connection
print("Connecting to WiFi", end="")
sta_if = network.WLAN(network.STA_IF)
sta_if.active(True)
sta_if.connect('Wokwi-GUEST', '')
while not sta_if.isconnected():
  print(".", end="")
  time.sleep(0.1)
print('Connected to ' + SSID)

""" Koneksi MQTT """
# MQTT Server connection
print("Connecting to MQTT server... ", end="")
client = MQTTClient(MQTT_CLIENT_ID, MQTT_BROKER, user=MQTT_USER, password=MQTT_PASSWORD)
client.connect()
print("Connected!")

""" Inisiasi prev variable """
# Build the message in JSON format and send the message only if there is a change
prev_weather = ""
buttonStatus = ""

# Callback funcion for Topic Subscription
def sub_cb(topic, msg):
  print(str(topic,'UTF-8'))
  print(str(msg,'UTF-8'))
  buttonStatus = str(msg,'UTF-8')
  messageNyala = "Lampu LED Dinyalakan"
  messageMati = "Lampu LED Dimatikan"
  if buttonStatus == "On" :
    print("Lampu Led Menyala")
    client.publish(MQTT_TOPICLED, messageNyala)
    led.on()
  elif buttonStatus == "Off" :
    print("Lampu Led Mati")
    client.publish(MQTT_TOPICLED, messageMati)
    led.off()

# MQTT Subcribtion details
client.set_callback(sub_cb)
client.subscribe(MQTT_TOPICLED)

while True:
  """ Welcome For Loops """

  """ Config Sensor DHT22 """
  sensor1.measure() 
  message = ujson.dumps({
    "temperature": sensor1.temperature(),
    "humidity": sensor1.humidity(),
  })

  """ Send MQTT from DHT22 """
  if message != prev_weather:
    print("Updated!")
    print("Reporting to MQTT for DHT 22 topic {}: {}".format(MQTT_TOPIC, message))
    # Send the message
    client.publish(MQTT_TOPIC, message)
    prev_weather = message
  client.check_msg()
  time.sleep(1)