from machine import Pin, I2C, RTC
from neopixel import NeoPixel

import network
import time
import ntptime
import urequests
import localtime_fr # TODO : Lire info HP.. HC.. sur HA > sensor.compteur_linky_ptec
import webrepl

Broche = [0, 1, 2, 21, 22, 23, 16, 17, 19, 20, 18]

class Bouton:
  DEBOUNCE_MS = 500

  def __init__(self, nom, pin, todo = None, pin_led = None, led_on = False):
    self.dernier_appui = 0
    self.nom = nom
    self.pin = Pin(pin, Pin.IN, Pin.PULL_UP)
    self.pin.irq(trigger = Pin.IRQ_FALLING, handler = self.bouton_presse)
    self.todo = todo
    if pin_led is None:
      self.avec_led = False
    else:
      self.avec_led = True
      self.pin_led = Pin(pin_led, Pin.OUT)
      self.led_on() if led_on else self.led_off()
  
  def led_on(self):
    if self.avec_led:
      self.pin_led.on()

  def led_off(self):
    if self.avec_led:
      self.pin_led.off()

  def bouton_presse(self, pin):
    self.maintenant = time.ticks_ms()  
    if time.ticks_diff(self.maintenant, self.dernier_appui) > self.DEBOUNCE_MS:
      self.dernier_appui = self.maintenant
      print("Bouton " + self.nom + " pressé !")
      if self.todo is not None:
        self.todo()

class Relais:
  i2c = I2C(scl=Pin(Broche[5]), sda=Pin(Broche[4]), freq=20000)
  I2C_ADDR = 0x11

  def __init__(self) -> None:
    self.solaire()

  def off(self):
    self.i2c.writeto_mem(self.I2C_ADDR, 0x10, bytes([0b0000]))

  def solaire(self):
    self.i2c.writeto_mem(self.I2C_ADDR, 0x10, bytes([0b0011]))

  def reseau(self):
    self.i2c.writeto_mem(self.I2C_ADDR, 0x10, bytes([0b1100]))

class Cloud:
  token_ha = "eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJpc3MiOiJkNGY4MDdmYWYzNDQ0NTc0ODY4MmFmNzA4NDdmMTE0MyIsImlhdCI6MTc2NDQ1Mzk0NSwiZXhwIjoyMDc5ODEzOTQ1fQ.DJgSqeTKPHWbKEFH3HuFih4QKt3CSqLqot34_vhCOQU"
  headers = {"Authorization": f"Bearer {token_ha}", "Content-Type": "application/json"}  
  wifi = network.WLAN(network.STA_IF)
  wifi.active(True)
  wifi.connect("Livebox-BDC6", "KFQSn7PDCMSgPpM2Ws")

  def zoe(self) -> int:
    try:
      response = urequests.get("https://ha-demo.arbi.fr/api/states/sensor.zoe_batterie", headers=self.headers)
      data = response.json()
      response.close()
      return int(data["state"])
    except Exception as e:
      print("Erreur :", e)
    return -1

# Capter exception pour éviter les sorties intempestives (comme timeout sur ntp)
# Regarder https://docs.micropython.org/en/latest/esp32/quickref.html#timers

# led_jaune = Pin(0, Pin.OUT)  # D0
# button_jaune = Pin(1, Pin.IN, Pin.PULL_UP)  # D1 

relais = Relais()

cloud = Cloud()

def action_jaune():
  bouton_jaune.led_on()
  bouton_bleu.led_off()
  bouton_rouge.led_off()
  relais.solaire()

bouton_jaune = Bouton("jaune", Broche[1], action_jaune, Broche[0], True)

def action_bleu():
  bouton_jaune.led_off()
  bouton_bleu.led_on()
  bouton_rouge.led_off()    
  relais.reseau()

bouton_bleu = Bouton("bleu", Broche[3], action_bleu, Broche[2])

def action_rouge():
  bouton_jaune.led_off()
  bouton_bleu.led_off()
  bouton_rouge.led_on()    
  relais.reseau()

bouton_rouge = Bouton("bleu", Broche[10], action_rouge, Broche[9])

button_moins = Pin(16, Pin.IN, Pin.PULL_UP)  # D6 (et masse)
button_plus = Pin(17, Pin.IN, Pin.PULL_UP)  # D7 (et masse)

bouton_moins = Bouton("moins", Broche[7])

bouton_plus = Bouton("plus", Broche[6])

#######################################################""

brightness = 0.01  # 1 %

def set_pixel(i, r, g, b):
  np[i] = (
    int(r * brightness),
    int(g * brightness),
    int(b * brightness)
  )

pin_leds = Pin(19, Pin.OUT) # D8
np = NeoPixel(pin_leds, 10)
set_pixel(0, 0, 255, 0)
set_pixel(7, 0, 255, 0)
np.write()
# np.brightness(50)

pas_temps = 10


# def HC(date_heure) -> bool:
#   heure, minute = date_heure[4], date_heure[5]
#   nb_minutes = heure * 60 + minute
#   _21h30 = 21 * 60 + 30 # TODO : Ne pas faire le calcul à chaque appel
#   _23h30 = 23 * 60 + 30
#   if (nb_minutes > _21h30  and nb_minutes < _23h30):
#     return True
#   return False

# def ntp():
#   global ntp_ok, ntp_ttl, ntp_attente
#   if (wifi.isconnected() and (not ntp_ok or ntp_attente < 0)):
#     print("WIFI OK, call NTP")
#     ntptime.settime()
#     ntp_ok = True
#     ntp_attente = ntp_ttl
#     webrepl.start()
#   if (ntp_ok):
#     date_heure = localtime_fr.localtime()
#     print(f"Date et heure : {date_heure[3]}:{date_heure[4]}:{date_heure[5]} ({ntp_attente}, {HC(date_heure)})")
#     ntp_attente = ntp_attente - pas_temps


while True:
  # now = datetime.now()
  # print("Date et heure :", now)
  print(f"Charge : {cloud.zoe()}")
  time.sleep(pas_temps)
  # set_pixel(0, 0, 0, 0) # OFF
  # np.write()