"""
MicroPython IoT Weather Station Example for Wokwi.com

To view the data:

1. Go to http://www.hivemq.com/demos/websocket-client/
2. Click "Connect"
3. Under Subscriptions, click "Add New Topic Subscription"
4. In the Topic field, type "wokwi-weather" then click "Subscribe"

Now click on the DHT22 sensor in the simulation,
change the temperature/humidity, and you should see
the message appear on the MQTT Broker, in the "Messages" pane.

Copyright (C) 2022, Uri Shaked

https://wokwi.com/arduino/projects/322577683855704658
"""

import network
import time
from time import sleep
from machine import Pin
import ujson
from umqtt.simple import MQTTClient
from neopixel import NeoPixel

# Received messages from subscriptions will be delivered to this callback
def sub_cb(topic, msg):
  if msg == b"g":
    for i in range(16):
      pixels[i] = rainbow[1]
    pixels.write()
    ledGreen.value(1)
    time.sleep(0.5)
    ledGreen.value(0)
  elif msg == b"b":
    for i in range(16):
      pixels[i] = rainbow[2]
    pixels.write()
    ledGreen.value(1)
    time.sleep(0.5)
    ledGreen.value(0)
  elif msg == b"r":
    for i in range(16):
      pixels[i] = rainbow[0]
    pixels.write()
    ledGreen.value(1)
    time.sleep(0.5)
    ledGreen.value(0)
  else:
    print((topic, msg))
    ledRed.value(1)
    time.sleep(0.5)
    ledRed.value(0)


# MQTT Server Parameters
MQTT_CLIENT_ID = "micropython-weather-demo"
MQTT_BROKER    = "broker.mqttdashboard.com"
MQTT_USER      = ""
MQTT_PASSWORD  = ""
MQTT_TOPIC     = "wokwi-weather"

ledRed = Pin(12, Pin.OUT)
ledGreen = Pin(13, Pin.OUT)
ledConnected = Pin(2, Pin.OUT)


rainbow = [(126 , 0 , 0), (0 , 126 , 0), (0, 0, 126)]
pixels = NeoPixel(Pin(23), 16)

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="")
  ledConnected.value(0)
  time.sleep(0.1)

print(" Connected!")

print("Connecting to MQTT server... ", end="")
client = MQTTClient(MQTT_CLIENT_ID, MQTT_BROKER, user=MQTT_USER, password=MQTT_PASSWORD)
client.set_callback(sub_cb)
client.connect()

ledConnected.value(1)

client.subscribe(b"testPosUni9");
client.publish(b"testPosUni9", b"hello")

print("Connected!")

while True:
  if True:           
    client.wait_msg()
  else:
    client.check_msg()
    time.sleep(1)