from machine import Pin, ADC
import time 
from umqtt.simple import MQTTClient
import ujson
import network
import utime as time

# Device Setup
DEVICE_ID = "wokwi001"

# WiFi Setup
WIFI_SSID       = "Wokwi-GUEST"
WIFI_PASSWORD   = ""

# MQTT Setup
MQTT_BROKER             = "broker.mqttdashboard.com"
MQTT_CLIENT             = DEVICE_ID
MQTT_TELEMETRY_TOPIC    = "iot/telemetry"
MQTT_CONTROL_TOPIC      = "iot/control"

led_r = Pin (14, Pin.OUT)
led_v = Pin (26, Pin.OUT)
caudalimetro = Pin (34)
Caudalimetro = ADC (Pin (34))
led_b = (Pin(2, Pin.OUT))
led_r.on()
led_b.on()

Caudalimetro.width(ADC.WIDTH_10BIT)
Caudalimetro.atten(ADC.ATTN_11DB)

lectura = int(Caudalimetro.read())

Agua = 75 / 1023
flujo = Agua * lectura
time.sleep_ms(500)

"""
while True:

    if flujo <= 0:
        print ("No está pasando agua por el caudalímetro")
        led_r.value(1)
        led_v.value(0)

    if flujo > 0 and flujo < 75:
        print (f"se están usando {flujo} Litros por segundo de agua en la casa")
        led_r.value(0)
        led_v.value(1)

    if flujo >= 75:
        print ("conecte el caudalimetro a una tuberia de 25mm o menor")
        led_r.value(1)
        led_v.value(0)

    time.sleep_ms(500)
"""

#receive messages
def received_callback (topic, message):
    print (f"se ha recibido un mensaje del tema {0}, que dice = {1}")
    

#conection with broker mqtt
def mqtt_conection ():
    print (f"conectandose al broker mqtt {MQTT_BROKER}...", end="")
    mqtt_client = MQTTClient(MQTT_CLIENT, MQTT_BROKER, user="", password="")
    mqtt_client.set_callback(received_callback)
    mqtt_client.connect
    print ("Conectado")
    #mqtt_client.subscribe(MQTT_CONTROL_TOPIC)
    print (f"suscrito al tema {MQTT_CONTROL_TOPIC}") 
    #return mqtt_client

def create_json_data (flujo):
    data = ujson.dumps ({
        "flujo": flujo,
    })
    return data

def mqtt_publish (topic, data):
    print("\n Actudalizando el broker mqtt...")
    mqtt_client.publish(topic, data)
    print(data)

#logic


#WiFi Connection
Wifi_client = network.WLAN(network.STA_IF)
Wifi_client.active(True)
print("Conectandose a WiFi...", end="")
Wifi_client.connect(WIFI_SSID, WIFI_PASSWORD)

while not Wifi_client.isconnected():
    print ("Conectandose a la red WiFi")
    time.sleep(0.1)
print("Conectado a la red WiFi")
print(Wifi_client.ifconfig())

#MQTT Broker conection
mqtt_client = mqtt_conection()

sensor = caudalimetro
Sensor = Caudalimetro

led_r.value(0)
led_v.value(0)

telemetry_data_old = ""

while True:
    #mqtt_client.check_msg()
    #print (". ", end="")
    ensor.read()
    telemetry_data_new = create_json_data (sensor())
    if telemetry_data_new != telemetry_data_old:
        mqtt_publish(MQTT_TELEMETRY_TOPIC, telemetry_data_new)
        telemetry_data_old = telemetry_data_new

    time.sleep(1)