import os
import network
import socket
import time
from machine import Pin

import constants
import globals
import helper
import components.layout as layout
import components.main as main
from secrets import secrets
from type import Setting, Time

# while True:
#     time.sleep_ms(200)

globals.init()

global day_mode_filename
day_mode_filename = "day_mode.txt"

led = Pin('LED', Pin.OUT)
def blink_onboard_led(num_blinks):
    for i in range(num_blinks):
        led.value(1)
        time.sleep_ms(200)
        led.value(0)
        time.sleep_ms(200)

blink_onboard_led(1)

if not helper.file_or_dir_exists(day_mode_filename):
    print('File does not exist.')
    with open(day_mode_filename, "w") as f:
        f.write("byweek")
        f.close()
else:
    with open(day_mode_filename) as f:
        content = f.readline()

    if content == "byweek":
        print("byweek")
        globals.day_mode = constants.BY_WEEK
    else:
        print("byday")
        globals.day_mode = constants.BY_DAY

ssid = secrets['ssid']
password = secrets['pw']

wlan = network.WLAN(network.STA_IF)
wlan.active(True)
wlan.connect(ssid, password)

max_wait = 10
while max_wait > 0:
    if wlan.status() < 0 or wlan.status() >= 3:
        break
    max_wait -= 1
    print('waiting for connection...')
    time.sleep(1)


if wlan.status() != 3:
    blink_onboard_led(3)
    raise RuntimeError('network connection failed')
else:
    print('connected')
    status = wlan.ifconfig()
    print( 'ip = ' + status[0] )

blink_onboard_led(2)

addr = socket.getaddrinfo('0.0.0.0', 80)[0][-1]

s = socket.socket()
s.bind(addr)
s.listen(1)

print('listening on', addr)

def result_ok(cl, response = None):
    cl.send('HTTP/1.0 200 OK\r\nContent-type: text/html\r\n\r\n')
    cl.send(response if response is not None else "Ok")
    cl.close()

def result_notfound(cl, response = None):
    cl.send('HTTP/1.0 404 NotFound\r\nContent-type: text/html\r\n\r\n')
    cl.send(response if response is not None else "Not Found")
    cl.close()

def check_header_button_endpoints(cl, request):
    if request.find('/set_by_week') == 6:
        set_day_mode(constants.BY_WEEK)
        result_ok(cl, main.get_header_buttons(day_mode = globals.day_mode) + main.get_settings_with_id(day_mode = globals.day_mode, oob = True))
        return True
    elif request.find('/set_by_day') == 6:
        set_day_mode(constants.BY_DAY)
        result_ok(cl, main.get_header_buttons(day_mode = globals.day_mode) + main.get_settings_with_id(day_mode = globals.day_mode, oob = True))
        return True
    
    return False

def check_by_week_endpoints(cl, request):
    if request.find('/add_by_week') == 6:
        globals.by_week_settings.append(Setting(Time(10,0), 22))
        result_ok(cl, main.get_byweek_settings_content())
        return True

def set_day_mode(mode):
    global day_mode_filename
    if mode == constants.BY_WEEK:
        globals.day_mode = constants.BY_WEEK
    elif mode == constants.BY_DAY:
        globals.day_mode = constants.BY_DAY
    else:
        return
    
    with open(day_mode_filename, "w") as f:
        value = "byday" if mode == constants.BY_DAY else "byweek"
        print("Setting day mode to " + value)
        f.write(value)
        f.close()

# Listen for connections
while True:
    cl, addr = s.accept()
    try:
        print('client connected from', addr)
        request = cl.recv(1024)
        print(request)

        request = str(request)

        if request.find('/ ') == 6:
            response = layout.html % main.get_main(day_mode = globals.day_mode)

            result_ok(cl, response)
        elif request.find('/light/on') == 6:
            print("led on")
            led.value(1)
            
            response = layout.html % main.get_main(day_mode = globals.day_mode)
            result_ok(cl, response)
        elif request.find('/light/off') == 6:
            print("led off")
            led.value(0)

            response = layout.html % main.get_main(day_mode = globals.day_mode)
            result_ok(cl, response)
        elif request.find('/exit') == 6:
            print("exiting")
            blink_onboard_led(5)
            s.close()
            break
        elif check_header_button_endpoints(cl, request):
            pass
        elif check_by_week_endpoints(cl, request):
            pass
        else:
            result_notfound(cl)


    except OSError as e:
        cl.close()
        print('connection closed')