from machine import Pin, SPI, I2C
import utime
import sys
import ntptime
import framebuf
from pcd8544 import PCD8544
from music import Music
import wlan

if(sys.platform=="esp32"):
  led = Pin(2, Pin.OUT)
  disp_sck  = 18 # default SCK of SPI(1)
  disp_mosi = 23 # default MOSI of SPI(1)
  disp_miso = 19  # not use
  disp_res = 4
  disp_dc  = 2
  disp_cs  = 5
  disp_blk = 22
  # 初始化 SPI
  disp_spi = SPI(1, baudrate=20000000, polarity=1, phase=0, sck=Pin(disp_sck), mosi=Pin(disp_mosi), miso=Pin(disp_miso))

  I2C_SDA =21 # GPIO21(I2C SDA)
  I2C_SCL =22 # GPIO22(I2C SCL)
  # Initialize I2C
  i2c = I2C(0, scl=Pin(I2C_SCL), sda=Pin(I2C_SDA), freq=400000)     #initializing the I2C method for ESP32
  buzzer_pin = 5
  button_pin = 35

elif(sys.platform=="rp2"):
  led = Pin("LED", Pin.OUT)
  disp_sck  = 6 # default SCK of SPI(0)
  disp_mosi = 7 # default MOSI of SPI(0)
  disp_miso = 4  # not use
  disp_res = 2
  disp_dc  = 1
  disp_cs  = 5
  disp_blk = 3

  # 初始化 SPI
  disp_spi = SPI(0, baudrate=2000000, polarity=0, phase=0, sck=Pin(disp_sck), mosi=Pin(disp_mosi))

  I2C_SDA =16 # GPIO16(I2C SDA)
  I2C_SCL =17 # GPIO17(I2C SCL)
  i2c = I2C(0, scl=Pin(I2C_SCL), sda=Pin(I2C_SDA), freq=400000)
  buzzer_pin = 15
  button_pin = 21

INTERVAL = 1000  # millisecond

# nokia logo bitmap (84x48/8=504 bytes)
image = bytearray([
    0x80,0x00,0x00,0x80,0x00,0x00,0x80,0x00,0x00,0x80,0x00,0x00,0x80,0x00,0x00,0x80,0x00,0x00,0x80,0x80,0x40,0x40,0x40,0x80,0x80,0xC0,0xC0,0x40,0xC0,0xA0,0xE0,0xC0,0xE0,0xE0,0xF0,0xF0,0xF8,0xF8,0xF8,0xFC,0xFC,0xFE,0xEE,0xF4,0xF0,0xF0,0x70,0x30,0x00,0x80,0x00,0x00,0x80,0x00,0x0C,0x9C,0x1C,0x38,0xB8,0x38,0x38,0xB8,0xF8,0xF0,0xF0,0xF0,0xF0,0xF0,0xF0,0xF0,0xF0,0xF0,0xF0,0xF0,0xF0,0xF0,0xF0,0xF0,0xF0,0xF0,0xF8,0xF8,0xF8,0xF8,
    0x88,0x20,0x8A,0x20,0x08,0x22,0x08,0x00,0x0A,0x00,0x00,0x02,0x80,0x71,0xBA,0xDA,0xFD,0xDD,0xED,0xDE,0xEE,0xF7,0xFF,0xFB,0xFD,0xFD,0xFE,0xFF,0x7F,0x3F,0x1F,0x9F,0x3F,0x7F,0x6F,0x0F,0xAF,0x1F,0xBF,0x3E,0x3C,0x7A,0x78,0x70,0x22,0x88,0xA0,0x2A,0x80,0x08,0x62,0xE0,0xE0,0xF2,0xF0,0x58,0xDA,0xF8,0xFC,0x92,0xFE,0xFF,0xFF,0xD3,0xFF,0xFD,0xF3,0xE1,0xF0,0xF9,0x7F,0xBF,0x3F,0x8F,0x2F,0x4F,0xAF,0x0F,0x4F,0xA7,0x0F,0xAF,0x87,0x2F,
    0x82,0x80,0x20,0xC0,0x80,0x80,0x50,0x40,0xC4,0xD0,0xA0,0xE8,0xE4,0xEA,0xFF,0xFB,0xFD,0xFF,0xFF,0xFF,0xFF,0xFF,0xEF,0x4F,0x27,0x53,0xA8,0x54,0x29,0x4A,0xB5,0x82,0xAC,0xA1,0x8A,0xB6,0x50,0x4D,0x32,0xA4,0x4A,0xB4,0xA9,0x4A,0x52,0xB4,0xAA,0x45,0xA8,0xDA,0x22,0xAC,0xD2,0x2A,0x52,0xA8,0x52,0x4C,0xB0,0xAD,0x43,0x5B,0xB3,0x45,0xA8,0x5B,0xA3,0xAB,0x55,0xA8,0x52,0x54,0xA9,0x56,0xA8,0x45,0xBA,0xA4,0x49,0x5A,0xA2,0x54,0xAA,0x52,
    0xFE,0xFF,0xFF,0xFE,0xFD,0xFF,0xFF,0xFF,0xFE,0xFF,0xFF,0xFF,0xFF,0xFF,0xFF,0xFF,0xFF,0xFF,0xFF,0xFF,0xFF,0x7F,0xFF,0xFE,0xBF,0x7F,0xBF,0xBF,0xFF,0xDF,0xBF,0x5F,0xDF,0x7F,0xDF,0x7F,0xDF,0xAF,0x7F,0xEE,0x8E,0xF1,0x6E,0x99,0xF7,0x6A,0xDD,0xB2,0x6E,0xD5,0x7A,0xD7,0xAC,0x75,0xDB,0x6D,0xD5,0x7A,0xD7,0xAC,0x7B,0xE5,0xDE,0xA9,0x77,0xDA,0xB5,0xEE,0x59,0xB6,0xEB,0xDD,0xB6,0x69,0xD6,0xBF,0xE8,0x55,0xEF,0xB9,0xD6,0xED,0xB5,0x5B,
    0xAB,0xFF,0xFD,0xF7,0xFF,0x01,0x01,0x01,0x01,0xE1,0xC1,0x81,0x03,0x05,0x0F,0x1D,0x2F,0x7E,0x01,0x00,0x01,0x01,0xFF,0xFE,0x03,0x01,0x01,0x00,0xF1,0xF0,0xF1,0x71,0xF1,0xF1,0xB1,0xF1,0x01,0x01,0x01,0x03,0xFE,0xFF,0x01,0x01,0x01,0x01,0xBE,0x1B,0x0D,0x07,0x03,0x41,0xE1,0xF1,0xF9,0x6D,0xFF,0xFF,0x00,0x01,0x01,0x01,0xFF,0xFF,0xEB,0x3E,0x0D,0x03,0x01,0x41,0x71,0x70,0x41,0x01,0x03,0x0E,0x3B,0xEF,0xFE,0xFB,0xEE,0x7D,0xF7,0xFF,
    0xFF,0xFF,0xFF,0xFE,0xFF,0xF0,0xF0,0xF0,0xF0,0xFF,0xFF,0xFF,0xFF,0xFE,0xFC,0xF8,0xF0,0xF0,0xF0,0xF0,0xF0,0xF0,0xFF,0xFF,0xF8,0xF0,0xF0,0xF0,0xF1,0xF1,0xF1,0xF1,0xF1,0xF1,0xF1,0xF1,0xF0,0xF0,0xF0,0xF8,0xFF,0xFF,0xF0,0xF0,0xF0,0xF0,0xFF,0xFF,0xFE,0xFC,0xF8,0xF0,0xF0,0xF1,0xF3,0xF7,0xFF,0xFF,0xF0,0xF0,0xF0,0xF0,0xFF,0xF3,0xF0,0xF0,0xF0,0xFC,0xFC,0xFC,0xFC,0xFC,0xFC,0xFC,0xFC,0xF0,0xF0,0xF0,0xF3,0xFF,0xFF,0xFF,0xFF,0xFF
])

'''
 Source available at https://ithelp.ithome.com.tw/articles/10285468
 用Raspberry Pi Pico發出久違的Nokia之歌
'''
Nokia = [
  'E5:1', 'D5:1', 'F#4:2', 'G#4:2', 'C#5:1', 'B4:1',
  'D4:2', 'E4:2', 'B4:1', 'A4:1', 'C#4:2', 'E4:2','A4:4'
]

"""
  Unix time starts on Jan 1 1970. In seconds, that's 2208988800:
  (date(1970, 1, 1) - date(1900, 1, 1)).days * 24*60*60
  (70*365+17)*24*60*60 = 2208988800 =>UTC0
  28800 = 8*60*60
  2208960000 = 2208988800-28800
"""
# Optional UTC+8 offset time (seconds), if not set, UTC0
# Timezone offset (India = UTC +8:00)
UTC_OFFSET = 28800 # 8 * 60 * 60

def sync_time_with_retry():
  try:
    print("Syncing with NTP server...")
    ntptime.settime()   # UTC time
    # 獲取並顯示本地時間 (台灣時區 +8)
    # utime.localtime() 取得 UTC 時間，所以要加上 8 小時
    timer = utime.localtime(utime.time() + UTC_OFFSET)
    print("Time synchronized.")
    return timer
  except OSError as e:
    # print("Error syncing time:", e)
    if e.args[0] == 110: # ETIMEDOUT
      print("ETIMEDOUT error during time sync. Retrying in 5 seconds...")
      utime.sleep(5)
      sync_time_with_retry() # Recursive call to retry
    else:
      print(f"An unexpected error occurred: {e}")

# Function to handle button press interrupt (optional)
def button_press(pin):
  global button_down
  button_down = True

# Start Function
if __name__ == '__main__':
  print("Hello, " + sys.platform + "!")

  button = Pin(button_pin, Pin.IN, Pin.PULL_UP)   # 按鍵接在 GPxx, Internal pull-up resistor
  button.irq(trigger=Pin.IRQ_FALLING, handler=button_press)  # Attach interrupt to the push button pin
  button_down = False

  cs = Pin(disp_cs, Pin.OUT)
  dc = Pin(disp_dc, Pin.OUT)
  rst = Pin(disp_res, Pin.OUT)
  lcd = PCD8544(disp_spi, cs, dc, rst)

  #Use a framebuffer to store the 4032 pixels (84x48):
  buffer = bytearray((lcd.height // 8) * lcd.width)
  framebuf = framebuf.FrameBuffer1(buffer, lcd.width, lcd.height)

  #Light every pixel:
  framebuf.fill(1)
  lcd.data(buffer)
  utime.sleep_ms(500)

  #Clear screen:
  framebuf.fill(0)
  lcd.data(buffer)

  # test pattern (50% on)
  lcd.data(bytearray([0x55, 0xAA] * 42 * 6))
  utime.sleep(1)

  framebuf.fill(0)
  framebuf.text("Hello Wokwi", 0, 0, 1)
  framebuf.text("Nokia 5110", 0, 10, 1)
  lcd.data(buffer)

  # draw a 84x48 picture
  lcd.data(image)
  utime.sleep_ms(1000)

  # 引腳定義
  button = Pin(button_pin, Pin.IN, Pin.PULL_UP)   # 按鍵接在 GPxx, Internal pull-up resistor
  music = Music(buzzer_pin)           # 蜂鳴器接在 GPxx
  music.set_volume(1000)

  music.tempo(4,120)
  music.play(Nokia)

  # ========== CONNECT WIFI ==========
  wlan.connect_wifi() # Connecting to WiFi Router
  utime.sleep_ms(1000)

  # ========== NTP TIME SYNC ==========
  led.on()
  timer = sync_time_with_retry()  # Synchronize the time
  hour = "00"
  minute = "00"
  second = "00"
  led.off()

  lastTime = utime.ticks_ms()
  # ========== MAIN LOOP ==========
  while True:
    try:
      currTime = utime.ticks_ms()
      if (currTime - lastTime > INTERVAL):
        lastTime = currTime

        time_now = utime.localtime(utime.time() + UTC_OFFSET)

        year = "{:04d}".format(time_now[0])
        month = "{:02d}".format(time_now[1])
        day = "{:02d}".format(time_now[2])
        hour = "{:02d}".format(time_now[3])
        minute = "{:02d}".format(time_now[4])
        second = "{:02d}".format(time_now[5])

        # SERIAL PRINT
        str_date = year + ":" + month + ":" + day
        str_time = hour + ":" + minute + ":" + second
        print("Date: " + str_date, end='')
        print("; Time: " + str_time)

        # lcd
        framebuf.fill(0)
        framebuf.text("Date: ", 0, 8, 1)
        framebuf.text(year + "/" + month + "/" + day, 0, 16, 1)
        framebuf.text("Time: ", 0, 24, 1)
        framebuf.text(hour + ":" + minute + ":" + second, 0, 32, 1)
        lcd.data(buffer)

      if (hour == "12") and (minute == "00") and (second == "00"):
        # ========== CONNECT WIFI ==========
        wlan.connect_wifi() # Connecting to WiFi Router
        utime.sleep_ms(1000)

        led.on()
        timer = sync_time_with_retry()  # Synchronize the time
        led.off()

      if button_down == True:
        # ========== CONNECT WIFI ==========
        wlan.connect_wifi() # Connecting to WiFi Router
        utime.sleep_ms(1000)

        led.on()
        timer = sync_time_with_retry()  # Synchronize the time
        button_down = False
        led.off()

      utime.sleep_ms(100)
    except KeyboardInterrupt: # 使用者中斷執行(通常是輸入^C)
      print("Stop")
      break