"""
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
import utime
from machine import Pin, I2C
import dht
import ujson
from umqtt.simple import MQTTClient
from i2c_lcd import I2cLcd

last_interrupt_time = 0

# LCD setup
I2C_ADDR = 0x27
I2C_NUM_ROWS = 2
I2C_NUM_COLS = 16
i2c = I2C(scl=Pin(22), sda=Pin(21), freq=400000)
lcd = I2cLcd(i2c, I2C_ADDR, I2C_NUM_ROWS, I2C_NUM_COLS)

# function definitions
def togleLights(pin):
    global last_interrupt_time
    if time.time() > last_interrupt_time + 0.2:
        print("lights")
        message = ujson.dumps({
                "type": "device",
                "device": "lamp"
            })
        client.publish(MQTT_TOPIC, message)
        last_interrupt_time = time.time()

def togleFan(pin):
    global last_interrupt_time
    global fan
    if time.time() > last_interrupt_time + 0.2:
        print("fan")
        fan = not(fan)
        relayPin.value(fan)
        last_interrupt_time = time.time()

def turnOffLight(pin):
    global last_interrupt_time
    if time.time() > last_interrupt_time + 0.2:
        print("turn off light")
        message = ujson.dumps({
                "type": "device",
                "device": "lamp",
                "action": "off"
            })
        client.publish(MQTT_TOPIC, message)
        last_interrupt_time = time.time()

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

# HW setup
lightButtonPin = Pin(34, Pin.IN)
fanButtonPin = Pin(36, Pin.IN)
offLightButtonPin = Pin(33, Pin.IN)  # New pin for turning off the light
relayPin = Pin(2, Pin.OUT)
lightButtonPin.irq(trigger=Pin.IRQ_FALLING, handler=togleLights)
fanButtonPin.irq(trigger=Pin.IRQ_FALLING, handler=togleFan)
offLightButtonPin.irq(trigger=Pin.IRQ_FALLING, handler=turnOffLight)  # Assign the new function to the new pin
sensor = dht.DHT22(Pin(15))

# internals
nextWeatherCheck = 0
weatherCheckDelay = 1 # in seconds

fan = False
light = False

# setup
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!")

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

print("Connected!")

# main
prev_weather = ""
nextWeatherCheck = time.time()
nextButtonCheck = time.time()
while True:
    # MQTT
    if(time.time() >= nextWeatherCheck):
        nextWeatherCheck += weatherCheckDelay
        print("Measuring weather conditions... ", end="")
        sensor.measure()
        temp = sensor.temperature()
        humidity = sensor.humidity()
        message = ujson.dumps({
            "type": "weather",
            "temp": temp,
            "humidity": humidity,
        })
        if message != prev_weather:
            print("Updated!")
            print("Reporting to MQTT topic {}: {}".format(MQTT_TOPIC, message))
            client.publish(MQTT_TOPIC, message)
            prev_weather = message
            
            # Update LCD display
            lcd.clear()
            lcd.putstr("Temp: {}C".format(temp))
            lcd.move_to(0, 1)
            lcd.putstr("Humidity: {}%".format(humidity))
        else:
            print("No change")