import random
import base64
import json
import os
import datetime


class Model:
    def __init__(self, filename = "tehmodel.json"):
        self.sockets = {}
        self.filename = filename
        self.sessions = None
        if os.path.isfile(filename):
            with open(filename) as f:
                try:
                    self.sessions = json.load(f)
                except:
                    self.sessions = {}
        else:
            self.sessions = {}
    
    def handle_post(self, data):
        print("I have received P O S T data: " + str(data))
        print("let me tell everyone")
        for k, v in self.sockets.items():
            v.send_json({"message": "nudes"})

        # handle post request data

    def __enter__(self):
        return self

    def __exit__(self, exception_type, exception_value, traceback):
        if not (exception_type or exception_value or traceback):
            with open(self.filename, "w") as f:
                json.dump(self.sessions, f)
        if not os.path.isdir("backups"):
            try:
                os.mkdir("backups")
            except FileExistsError:
                print("backups is a file, no directory. Please delete yourself")
        datestring = datetime.datetime.strftime(datetime.datetime.now(), "%Y-%m-%d-%H%M%S")
        with open(f"backups/{datestring}_{self.filename}", "w") as f:
            json.dump(self.sessions, f)

    def exists_client(self, clientid: str) -> bool:
        return clientid in self.sockets

    def create_client(self, sessionid: str) -> str:
        clientname = base64.b32encode(bytearray(random.randint(0, 0xFF) for _ in range(10)))[:16].decode().lower()
        self.sessions[sessionname]["players"].append(clientname)
        return sessionname

    def create_session(self) -> str:
        sessionname = base64.b32encode(bytearray(random.randint(0, 0xFF) for _ in range(10)))[:16].decode().lower()
        self.sessions[sessionname] = {"players": []}
        return sessionname

    def subscribe(self, clientid, socket):
        self.sockets[clientid] = socket
        # todo subscribe socket, match with id
        
    
    def unsubscribe(self, socket):
        for k,v in self.sockets.items():
            if v == socket:
                self.sockets.pop(k)