import network
import time
from machine import Pin, Timer, ADC,SPI,I2C,PWM
import dht
import ujson
from umqtt.simple import MQTTClient
import framebuf
import ssd1306

eye0 = [
  B00000000,B00000000,B00000000,B00000000,
  B00000000,B00000000,B00000000,B00000000,
  B00000000,B00000000,B00000000,B00000000,
  B00000000,B00000000,B00000000,B00000000,
  B00000000,B00000000,B00000000,B00000000,
  B00000000,B00000000,B00000000,B00000000,
  B00000000,B00000000,B00000000,B00000000,
  B00000000,B00000000,B00000000,B00000000,
  B00000000,B00000000,B00000000,B00000000,
  B00000000,B00000000,B00000000,B00000000,
  B00000000,B00000000,B00000000,B00000000,
  B00000000,B00000000,B00000000,B00000000,
  B00000000,B00000000,B00000000,B00000000,
  B00000000,B00000000,B00000000,B00000000,
  B00000000,B00000000,B00000000,B00000000,
  B00000000,B00000000,B00000000,B00000000,
  B00000000,B00000000,B00000000,B00000000,
  B11111111,B11111111,B11111111,B11111111,
  B11111111,B11111111,B11111111,B11111111,
  B11111111,B11111111,B11111111,B11111111,
  B11111111,B11111111,B11111111,B11111111,
  B00000000,B00000000,B00000000,B00000000,
  B00000000,B00000000,B00000000,B00000000,
  B00000000,B00000000,B00000000,B00000000,
  B00000000,B00000000,B00000000,B00000000,
  B00000000,B00000000,B00000000,B00000000,
  B00000000,B00000000,B00000000,B00000000,
  B00000000,B00000000,B00000000,B00000000,
  B00000000,B00000000,B00000000,B00000000,
  B00000000,B00000000,B00000000,B00000000,
  B00000000,B00000000,B00000000,B00000000,
  B00000000,B00000000,B00000000,B00000000,
]


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

sensor = dht.DHT22(Pin(15))
#setup oled
i2c = I2C(0, scl=Pin(22), sda=Pin(21))

oled_width = 128
oled_height = 64
oled = ssd1306.SSD1306_I2C(oled_width, oled_height, i2c)


#setup PIR
pir = Pin(33,Pin.IN)

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

def interrupt_pir(pin):
  global motion
  motion = True
  global interrupt_pin
  interrupt_pin = pin 
  
def update_display(temperature, humidity):
    """Update OLED display with temperature and humidity values"""    
    oled.fill(0)  
    oled.text("Temp:", 0, 0)
    oled.text(f"{temperature:.1f} C", 0, 10)
    oled.text("Humidity:", 0, 30)
    oled.text(f"{humidity:.1f} %", 0, 40)
    oled.show()  

def ShowEyes():
    eye0_bytes = bytearray(eye0)
    fb = framebuf.FrameBuffer(eye0_bytes, bitmap_width, bitmap_height, framebuf.MONO_HLSB)

    # Blit the framebuffer onto the OLED display at position (x, y)
    x = 48  # Horizontal position (adjust as needed)
    y = 16  # Vertical position (adjust as needed)
    oled.blit(fb, x, y)
    oled.show()

motion = False
pir.irq(trigger=Pin.IRQ_RISING, handler=interrupt_pir)

prev_weather = ""
motiontimeout = time.time()
while True:
  print("Measuring weather conditions... ", end="")
  sensor.measure() 
  
  ShowEyes()

  message = ujson.dumps({
    "temp": sensor.temperature(),
    "humidity": sensor.humidity(),
  })
  
  if motion:
    update_display(sensor.temperature(),sensor.humidity())
    motiontimeout = time.time()    
    print (motiontimeout)
    motion = False

  print ("time: "+ str(time.time()))
  print ("timeout: "+ str(motiontimeout))
  print ("diff: " + str(time.time() - motiontimeout))
  if (time.time() - motiontimeout >= 10):
    oled.fill(0)  
    oled.show()

  if message != prev_weather:
    print("Updated!")
    print("Reporting to MQTT topic {}: {}".format(MQTT_TOPIC, message))
    client.publish(MQTT_TOPIC, message)
    prev_weather = message
  else:
    print("No change")
  time.sleep(1)