import badge
import ugfx
import network
import wifi
import time
import math
import dialogs
import usocket as socket
import _thread as thread
import uos as os


badge.init()
ugfx.init()
wifi.init()

ugfx.set_lut(ugfx.LUT_FULL)

ugfx.clear(ugfx.WHITE)
ugfx.flush()
ugfx.clear(ugfx.BLACK)
ugfx.flush()
ugfx.clear(ugfx.WHITE)
ugfx.flush()

lines = []
def render_term():
  ugfx.clear(ugfx.WHITE)
  ugfx.flush()
  cy = 0
  h = ugfx.height()
  count = math.floor(h / 13)
  for line in lines[-count:]:
    ugfx.string(0, cy, line, "pixelade13", ugfx.BLACK)
    cy = cy + 13
    if cy > h:
      break
  ugfx.flush()


def print_ln(text):
  lines.append(text)
  render_term()



def send_message(pressed):
  msgb = bytes(dialogs.prompt_text("Your message:") + "\n")
  s.send(msgb)


def wifi_up():
  while not wifi.sta_if.isconnected():
    time.sleep(0.1)
    pass
  return wifi.sta_if.ifconfig()[0]


def socket_listen(port):
  s = socket.socket()
  ai = socket.getaddrinfo("0.0.0.0", port)
  addr = ai[0][-1]
  # s.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
  s.bind(addr)
  s.listen(5)
  return s


def webserver(s, handler):
  run = True
  while run:
    res = s.accept()
    client_s = res[0]
    client_ip = str(res[1][0])
    req = str(client_s.readline().decode("ascii")).split(" ")
    method = str(req[0]).upper()
    url = str(req[1])
    headers = []
    while True:
      cline = str(client_s.readline().decode("ascii"))
      if cline == "" or cline == "\r\n":
        break
      headers.append(cline)
    body = ""
    if method.upper == "POST":
      body = client_s.recv(8192)
    resp = bytes(handler(client_s, method, url, body, headers, client_ip).encode("utf-8"))
    client_s.send(resp)
    client_s.close()



def files_handler(url, query_parms, headers, ip):
  ret = "<h1>Files</h1>\n<p>"
  for file in os.ilistdir():
    ret += str(file) + "\n"
  ret += "</p>"
  return ret

routes = [
  ("/",          "static", "/lib/badgeweb/index.html.py"),
  ("/files",     "get",    files_handler),
#  ("/files",     "post",    files_post_handler),
#  ("/leds",      "post",   leds_handler),
#  ("/vibration", "post",   vibration_handler),
#  ("/inputs",    "post",   inputs_handler),
#  ("/screen",    "post",   screen_handler)
]

default_reply = """\
HTTP/1.0 200 OK
Server: SHA2017 Badge
Content-Type: text/html

"""

error_reply = """\
HTTP/1.0 404 Not Found/Supported
Server: SHA2017 Badge
Content-Type: text/html

<h1>Error</h1>
<h2>Possibly 404.</h2>
<p>This badge can't satisfy your combination of URL and request method.</p>
"""


def router_handler(s, method, url, body, headers, client_ip):
  print_ln(method + " request from " + client_ip + " to " + url)
  found = False
  reply = default_reply
  query_parms = {}
  if url.find("?") != -1:
    _url = url.split("?")
    url = _url[0]
    if url.find("&") != -1:
      _query_parms = _url[1].split("&")
      for parm in _query_parms:
        _parm = parm.split("=")
        query_parms[_parm[0]] = _parm[1]
  for entry in routes:
    if url == entry[0]:
      if method == "GET":
        if entry[1] == "get":
          found = True
          reply += entry[2](url, query_parms, headers, client_ip)
        elif entry[1] == "static":
          found = True
          f = open(entry[2], "r")
          reply += f.read()
          f.close()
      elif method == "POST" and entry[1] == "post":
        found = True
        reply += entry[2](url, query_parms, headers, client_ip, body)
  if found:
    return reply
  else:
    return error_reply

print_ln("Waiting for wifi...")
cur_ip = wifi_up()
print_ln("Your IP: " + cur_ip)

print_ln("Creating listening socket on port 8080...")
s = socket_listen(8080)
print_ln("Starting webserver...")
webserver(s, router_handler)