# Picozero as a library import module is included as a python file 
# As Woki was having issues importing it by default
# AS part of the pico hardware steps this is imported without needing to include the file
#The hardware version of this script had to be modified for a DHT22 as Wokwi does not have a DHT11
#Code adapted from https://wokwi.com/projects/374636085248881665

print("Hello World")

from simple import MQTTClient
from picozero import pico_led
from picozero import LED
from picozero import Speaker
from picozero import Button
from picozero import Switch
from picozero import Pot # Pot is short for Potentiometer
from machine import Pin, I2C
import dht
from math import trunc
import network
import config
import time 
import network
import utime
import sys
import ujson 
print("Libraries Declared")

#Declare GPIO usage  
#Buttons
red_button = Button(14)
green_button = Button(15)

#LEDs 
blue_led = LED(21)
red_led = LED(20)
yellow_led = LED(19)
green_led = LED(18)

#Potentiometers
blue_varistor = Pot(28) # Connected to pin A1 (GP_27)
red_varistor = Pot(27) # Connected to pin A1 (GP_27)
white_varistor = Pot(26) # Connected to pin A1 (GP_27)

#Humidity & Temperature Sensor
dhtpin = machine.Pin(16)

#Range Finder Sensor
trigger = Pin(12, Pin.OUT)
echo = Pin(13, Pin.IN)

print("GPIO Pins Declared")

# Constants for MQTT Topics
MQTT_TOPIC_TEST = 'ashification/picow'
MQTT_TOPIC_RedPot = 'picow/redpot'
MQTT_TOPIC_BluePot = 'picow/bluepot'
MQTT_TOPIC_WhitePot = 'picow/whitepot'
MQTT_TOPIC_LED = 'picow/led'

# MQTT Parameters
MQTT_SERVER = config.mqtt_server
MQTT_PORT = config.mqtt_port
MQTT_USER = config.mqtt_username
MQTT_PASSWORD = config.mqtt_password
MQTT_CLIENT_ID = b"raspberrypi_picow"
MQTT_KEEPALIVE = 7200
MQTT_SSL = False   # set to False if using local Mosquitto MQTT broker
MQTT_SSL_PARAMS = {'server_hostname': MQTT_SERVER}


def is_json(myjson):
  try:
    ujson.loads(myjson)
  except ValueError as e:
    return False
  return True



def ultra():
   trigger.low()
   utime.sleep_us(2)
   trigger.high()
   utime.sleep_us(5)
   trigger.low()
   while echo.value() == 0:
       signaloff = utime.ticks_us()
   while echo.value() == 1:
       signalon = utime.ticks_us()
   timepassed = signalon - signaloff
   distance = (timepassed * 0.0343) / 2
   print("The distance from object is ",distance,"cm")
   return distance




def initialize_wifi(ssid, password):
    wlan = network.WLAN(network.STA_IF)
    wlan.active(True)

    # Connect to the network
    wlan.connect(ssid, password)

    # Wait for Wi-Fi connection
    connection_timeout = 10
    while connection_timeout > 0:
        if wlan.status() >= 3:
            break
        connection_timeout -= 1
        print('Waiting for Wi-Fi connection...')
        time.sleep(1)

    # Check if connection is successful
    if wlan.status() != 3:
        return False
    else:
        print('Connection successful!')
        network_info = wlan.ifconfig()
        print('IP address:', network_info[0])
        return True

def connect_mqtt():
    try:
        client = MQTTClient(client_id=MQTT_CLIENT_ID,
                            server=MQTT_SERVER,
                            port=MQTT_PORT,
                            )
        client.connect()
        return client
    except Exception as e:
        print('Error connecting to MQTT:', e)
        raise  # Re-raise the exception to see the full traceback

def publish_mqtt(topic, value):
    client.publish(topic, value)
    print(topic)
    print(value)
    print("Publish Done")

# Subcribe to MQTT topics
def subscribe(client, topic):
    client.subscribe(topic)
    print('Subscribe to topic:', topic)
    
# Callback function that runs when you receive a message on subscribed topic
def my_callback(topic, message):
    # Perform desired actions based on the subscribed topic and response
    print('Received message on topic:', topic)
    print('Response:', message)  
        
    # Check the content of the received message
    if message == b'RED ON' :
        #print('Turning LED ON')
        red_led.on()  # Turn LED ON
    elif message == b'RED OFF' :
        #print('Turning LED OFF')
        red_led.off()  # Turn LED OFF
    else:
        print('Unknown command')

try:
    if not initialize_wifi(config.wifi_ssid, config.wifi_password):
        print('Error connecting to the network... exiting program')
    else:
        # Connect to MQTT broker, start MQTT client
        client = connect_mqtt()
        client.set_callback(my_callback)
        subscribe(client, MQTT_TOPIC_LED)
        while True:
            
            #Blink LEDs to show pwr and tx
            pico_led.blink() #Pico board LED
            blue_led.blink()
            
            #Read in Button press data
            if red_button.is_pressed == True:
               print('red button pressed', red_button.is_pressed)
               publish_mqtt(MQTT_TOPIC_LED, str("RED ON"))
               client.check_msg()
               
            if green_button.is_pressed == True:
               print('green button pressed', green_button.is_pressed)
               publish_mqtt(MQTT_TOPIC_LED, str("RED OFF"))
               client.check_msg()
                      
            #Read temp & humidity data         
            data_dht = dht.DHT11(dhtpin)
            data_dht.measure()
    
            # Read potentiometer sensor data         
            blue_varistor_value = int(100 - (blue_varistor.value*100))
            #red_varistor_value = int(100 - (red_varistor.value*100))
            #white_varistor_value = int(100 - (white_varistor.value*100))

            
            #Create JSON Message
            json_data = {
                    "BluePotValue": blue_varistor_value,
                    #"RedPotValue": red_varistor_value,
                    #"WhitePotValue": white_varistor_value,
                    "TemperatureValue": data_dht.temperature(),
                    "HumidityValue": data_dht.humidity(),
                    "DistanceValue": ultra()
                    }
                
           
            # Publish as MQTT payload
            publish_mqtt(MQTT_TOPIC_TEST, ujson.dumps(json_data))

            client.check_msg()
            

            # Delay 1 seconds
            time.sleep(3)
            

except Exception as e:
    print('Error:', e)