import json
import socket
import _thread
import time
import random

BEVERAGES = {"beer": 2.5, "wine": 3, "whisky": 2, "lemonade": 0}

class Alcogotchi:
    def __init__(self, name):
        self.name = name
        self.last_drunkentime = time.time()
        self.drunk = 0
        self.health = 100
        self.alco_coin = 100
        self.items = {"food": 0, "beer": 0}

    def get_alcogotchi(self):
        self.last_drunkentime = time.time()
        return self.__dict__

    def double_or_nothing(self, data):
        self.last_drunkentime = time.time()
        bet = dict(data)["bet"]
        if self.alco_coin > bet:
            self.alco_coin += random.choice([-bet, bet])
            return self.get_alcogotchi()
        raise Exception("You got no coin")
        
    def buy_item(self, data):
        self.last_drunkentime = time.time()
        item = dict(data)["item"]
        if item in self.items:
            self.alco_coin -= 3
            self.items[item] += 1
        raise Exception("Item doesn't exist")
        
    def drink(self, data):
        drink_choice = data["drink"]
        self.drunk += BEVERAGES[drink_choice]
        if self.last_drunkentime + 60 < time.time():
            self.last_drunkentime = time.time()
        self.drunk += 10
        self.drunk = max(100, self.drunk)
        return self.get_alcogotchi()
    
alcogotchi = Alcogotchi("Brian")

class Server:
    HTTP_CODES = {
        200: "OK",
        404: "Not Found",
        500: "Internal Server Error",
    }

    def __init__(self, port=80):
        self.port = port
        self.connection = None
        self._sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
        self._sock.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
        self._sock.bind(("", port))
        self._sock.listen(1)
        self._routes = {}

    def add_route(self, path, callback, nowait=False):
        if nowait:
            self._routes[path] = lambda: (_thread.start_new_thread(callback, ()), None)[1]
        else:
            self._routes[path] = callback

    def _send(self, data, code=200):
        if self.connection is not None:
            content = json.dumps({"data" if code == 200 else "error": data})
            self.connection.send(
                "HTTP/1.1 {0} {2}\r\nContent-Type: application/json\r\n\r\n{2}".format(code, self.HTTP_CODES[code], content).encode()
            )

    def start(self):
        print("Server running on port {self.port}...")
        while True:
            self.connection, client_address = self._sock.accept()
            try:
                request = self.connection.recv(1024).decode().split("\r\n")
                method, route, _ = request[0].split(" ")
                print("{0} {1} on :{2} from {3}".format(method, route, self.port, client_address[0]))

                if route in self._routes:
                    if method == "GET":
                        self._send(self._routes[route]())
                    if method == "POST":
                        data = json.loads(request[-1])
                        # data as argument
                        self._send(self._routes[route](data))
                else:
                    self._send("Route is not defined", 404)

            except Exception as e:
                print("Error: {0}".format(e))
                self._send(str(e), 500)

            finally:
                self.connection.close()

ap = network.WLAN(network.AP_IF)
ap.active(True)
ap.config(essid="Cool Server", password="password123")
print("Connection avalible on {0}".format(ap.ifconfig()))
    

server = Server()
    
server.add_route("/", alcogotchi.get_alcogotchi)
server.add_route("/drink", alcogotchi.drink)
server.add_route("/gamble", alcogotchi.double_or_nothing)
server.add_route("/buy", alcogotchi.buy_item)
server.start()