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

# MQTT Server Parameters
MQTT_CLIENT_ID = "renaldiendrawansic6"
MQTT_BROKER    = "broker.emqx.io"
MQTT_USER      = ""
MQTT_PASSWORD  = ""

# Topic untuk kontrol LED (subscribe)
MQTT_TOPIC_SUB = "/BRAINWAVE/renaldi_endrawan/aktuasi_led"

# Fungsi koneksi WiFi
def do_connect():
    sta_if = network.WLAN(network.STA_IF)
    if not sta_if.isconnected():
        print('Connecting to network...')
        sta_if.active(True)
        sta_if.connect('Testhotspot', '12345678')
        while not sta_if.isconnected():
            pass
    print('Network config:', sta_if.ifconfig())

# Fungsi callback untuk menangani pesan yang diterima
def message_callback(topic, msg):
    print("Received message on topic {}: {}".format(topic, msg))
    if msg == b"ON":  
        led.on()
        print("LED ON")
    elif msg == b"OFF":  
        led.off()
        print("LED OFF")

# Koneksi ke WiFi
do_connect()

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

# Set callback untuk menangani pesan masuk
client.set_callback(message_callback)
client.connect()
client.subscribe(MQTT_TOPIC_SUB)  # Subscribe ke topik kontrol LED

# Inisialisasi LED
led = Pin(4, Pin.OUT)

# Loop utama untuk menerima pesan
while True:
    try:
        # Cek pesan masuk (harus dipanggil secara berkala)
        client.check_msg()
        time.sleep(1)
    except Exception as e:
        print("Error:", e)