from machine import I2C, Pin, ADC
from umqtt.simple import MQTTClient
from bmp180 import BMP180
import machine
import network
import ujson
import time
import dht

# ---------------------------
# Actuators
# ---------------------------

# Buzzer Configuration (GPIO 18)
buzzer = Pin(18, Pin.OUT)
buzzer.value(0) # مطفي في البداية

# ---------------------------
# Sensors
# ---------------------------

# DHT22
dht_sensor = dht.DHT22(Pin(15))

# Soil Moisture
soil_adc = ADC(Pin(34))
soil_adc.atten(ADC.ATTN_11DB)
soil_adc.width(ADC.WIDTH_12BIT)

# LDR
ldr_adc = ADC(Pin(35))
ldr_adc.atten(ADC.ATTN_11DB)
ldr_adc.width(ADC.WIDTH_12BIT)

# PIR Motion
motion_sensor = Pin(27, Pin.IN)

# BMP180
i2c = I2C(0, scl=Pin(22), sda=Pin(21))
bmp = BMP180(i2c)

# HC-SR04 Ultrasonic
trig = Pin(13, Pin.OUT)
echo = Pin(12, Pin.IN)

trig.value(0)

def get_distance():
    trig.value(0)
    time.sleep_us(2)

    trig.value(1)
    time.sleep_us(10)

    trig.value(0)

    try:
        pulse_time = machine.time_pulse_us(echo, 1, 30000)
        distance = (pulse_time * 0.0343) / 2
        return round(distance, 1)

    except OSError:
        return None


# ---------------------------
# WiFi
# ---------------------------

wifi = network.WLAN(network.STA_IF)
wifi.active(True)

print("Connecting to WiFi...")

wifi.connect("Wokwi-GUEST", "")

while not wifi.isconnected():
    time.sleep(0.5)

print("WiFi Connected!")
print(wifi.ifconfig())

# ---------------------------
# MQTT Setup & Callback
# ---------------------------

MQTT_BROKER = "mqtt-dashboard.com"
MQTT_CLIENT_ID = "do7amqtt_se"
TOPIC_PUBLISH = "do7a_d7d7"
TOPIC_BUZZER = b"esp32/actuators/buzzer"

def mqtt_callback(topic, msg):
    print("Received MQTT Message:", topic, msg)
    if topic == TOPIC_BUZZER:
        if msg == b'1':
            buzzer.value(1)
            print("Buzzer Activated! (ON)")
        elif msg == b'0':
            buzzer.value(0)
            print("Buzzer Deactivated! (OFF)")

client = MQTTClient(MQTT_CLIENT_ID, MQTT_BROKER)
client.set_callback(mqtt_callback)

print("Connecting to MQTT...")

client.connect()

# الاشتراك في موضوع الـ Buzzer لاستقبال الأوامر من Node-RED
client.subscribe(TOPIC_BUZZER)

print("MQTT Connected & Subscribed to Buzzer topic!")

# ---------------------------
# Main Loop
# ---------------------------

while True:

    try:
        # فحص وجود رسائل MQTT قادمة من Node-RED لتشغيل/إيقاف البزر
        client.check_msg()

        # DHT22
        dht_sensor.measure()
        temp = dht_sensor.temperature()
        humidity = dht_sensor.humidity()

        # Soil Moisture
        soil_raw = soil_adc.read()
        soil = round((soil_raw / 4095) * 100, 1)

        # LDR
        light = ldr_adc.read()

        # PIR
        motion = motion_sensor.value()

        # BMP180
        try:
            pressure = int(bmp.pressure())
        except:
            pressure = int(bmp.pressure)

        # Ultrasonic
        distance = get_distance()

        payload = {
            "temp": temp,
            "humidity": humidity,
            "soil": soil,
            "light": light,
            "motion": motion,
            "pressure": pressure,
            "distance": distance
        }

        client.publish(TOPIC_PUBLISH, ujson.dumps(payload))

        print(payload)

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

    time.sleep(2)