import_json.py

import json
from datetime import date, time, datetime, timedelta

from data_model import Bénévole, Spectacle, Lieu, Type_de_quête, Quête


def load_spectacles(obj):
    spectacles = obj["shows"]
    for s in spectacles["pages"]:
        id = s["id"]
        props = s["properties"]
        Spectacle(id, props["Nom"]["title"][0]["plain_text"])


def load_lieux(obj):
    places = obj["places"]
    for p in places["pages"]:
        id = p["id"]
        props = p["properties"]
        Lieu(id, props["Name"]["title"][0]["plain_text"])


def load_types_de_quête(obj):
    quest_types = obj["questsTypes"]
    for p in quest_types["pages"]:
        id = p["id"]
        name = p["properties"]["Name"]["title"][0]["plain_text"]
        sécable = p["properties"]["D\u00e9coupable ?"]["checkbox"]
        Type_de_quête(id, name, sécable)


def update_pref_horaire(heure_début, data, prefs, indisponibilités):
    heure_fin = heure_début + 1
    if heure_début == 23:
        heure_fin = 0
    début = f"{heure_début:0=2d}"
    if heure_début == 9:
        début = f"{heure_début}"
    select = data[f"{début}h - {heure_fin:0=2d}h"]["select"]
    pref = "no pref"
    if select:
        pref = select["id"]
    val = 0
    if pref == "E>t^":  # Indisponible
        indisponibilités.append(time(hour=heure_début))
        if heure_début == 0:
            indisponibilités.append(time(hour=(heure_début + 1)))
        return
    if pref == "P;>X":  # Sous la contrainte
        val = -4000000
    if pref == "Sa=z":  # Horaire de prédilection
        val = 1
    prefs[time(hour=heure_début)] = val


def load_bénévoles(obj):
    volunteers = obj["volunteers"]
    for p in volunteers["pages"]:
        props = p["properties"]
        id = p["id"]
        sérénité = props["Team S\u00e9r\u00e9nit\u00e9"]["checkbox"]
        heures_théoriques = float(props["heures th\u00e9oriques par jour"]["number"])
        date_départ = None
        if props["Date d\u00e9part"]["date"]:
            date_départ = datetime.fromisoformat(
                props["Date d\u00e9part"]["date"]["start"]
            )
        types_de_quête_interdits = list(
            map(
                lambda tdq: Type_de_quête.tous[tdq["id"]],
                props["type de Quete interdit"]["relation"],
            )
        )
        if heures_théoriques > 0:
            indisponibilités = []
            pref_horaires = {}
            update_pref_horaire(0, props, pref_horaires, indisponibilités)
            for i in range(9, 23 + 1):
                update_pref_horaire(i, props, pref_horaires, indisponibilités)
            Bénévole(
                id,
                props["Pseudo"]["title"][0]["plain_text"],
                props["Pr\u00e9nom"]["rich_text"][0]["plain_text"],
                props["Nom"]["rich_text"][0]["plain_text"],
                heures_théoriques,
                indisponibilités,
                pref_horaires,
                sérénité,
                types_de_quête_interdits,
                date_départ,
            )


def parse_horaires(horaire: str):
    horaire_groups = re.match(r"(.*) \(.*\) → (.*) \(.*\)", horaire)
    if horaire_groups:
        début = datetime.strptime(horaire_groups.group(1), "%d/%m/%Y %H:%M")
        fin = datetime.strptime(horaire_groups.group(2), "%d/%m/%Y %H:%M")
        return début, fin
    else:
        horaire_groups = re.match(r"(.*) \(.*\) → (.*):(.*)", horaire)
        début = datetime.strptime(horaire_groups.group(1), "%d/%m/%Y %H:%M")
        heure_fin = time(int(horaire_groups.group(2)), int(horaire_groups.group(3)))
        fin = datetime.combine(début, heure_fin)
        return début, fin


def load_quêtes(obj):
    quests = obj["quests"]
    for p in quests["pages"]:
        props = p["properties"]
        id = p["id"]
        name = props["Name"]["title"][0]["plain_text"]
        needed = props["Needed"]["number"]
        places = props["Place"]["relation"]
        place = None
        if len(places) > 0:
            place = Lieu.tous[places[0]["id"]]
        types_de_quête = list(
            map(
                lambda tdq: Type_de_quête.tous[tdq["id"]],
                props["Type de Quete"]["relation"],
            )
        )
        début = datetime.fromisoformat(props["Horaire"]["date"]["start"])
        fin = datetime.fromisoformat(props["Horaire"]["date"]["end"])
        bénévoles = list(
            map(
                lambda b: Bénévole.tous[b["id"]],
                props["B\u00e9n\u00e9voles v\u00e9rouill\u00e9s"]["relation"],
            )
        )
        spectacle = None
        if len(props["Spectacle"]["relation"]) > 0:
            spectacle = Spectacle.tous[props["Spectacle"]["relation"][0]["id"]]

        def new_quête(id, name, début: date, fin):
            q = Quête(
                id,
                name,
                types_de_quête,
                place,
                spectacle,
                needed,
                début,
                fin,
                bénévoles,
            )

            for b in bénévoles:
                b.quêtes_assignées.append(q)

        def new_quêtes(name, début: date, fin):
            dup = False
            for t in types_de_quête:
                if not ("Affichage" in t.nom):
                    dup = True
            new_quête(
                id,
                name,
                début,
                fin,
            )
            if dup:
                new_quête(
                    f"{id}",
                    f"{name}",
                    début + timedelta(days=1),
                    fin + timedelta(days=1),
                )
                new_quête(
                    f"{id}",
                    f"{name}",
                    début + timedelta(days=2),
                    fin + timedelta(days=2),
                )
                new_quête(
                    f"{id}",
                    f"{name}",
                    début + timedelta(days=3),
                    fin + timedelta(days=3),
                )

        if all(t.sécable for t in types_de_quête):
            fin_acc = début
            i = 0
            while fin_acc < fin:
                début_acc = fin_acc
                fin_acc = min(fin_acc + timedelta(minutes=120), fin)
                i = i + 1
                new_quêtes(
                    f"{name} #{i}",
                    début_acc,
                    fin_acc,
                )
        else:
            new_quêtes(
                name,
                début,
                fin,
            )


def load_assignations(obj):
    affectations = obj["affectations"]
    for a in affectations["pages"]:
        props = a["properties"]
        début = datetime.fromisoformat(props["Horaire"]["date"]["start"])
        fin = datetime.fromisoformat(props["Horaire"]["date"]["end"])
        id_quête = props["Qu\u00eate"]["relation"][0]["id"]
        bénévoles = []
        for id_b in props["B\u00e9n\u00e9voles"]["relation"]:
            bénévoles.append(Bénévole.tous[id_b["id"]])
        sécable = p["properties"]["D\u00e9coupable ?"]["checkbox"]
        Type_de_quête(id, name, sécable)


def from_file(src):
    with open(src, "r") as file:
        obj = json.load(file)
        load_spectacles(obj)
        load_lieux(obj)
        load_types_de_quête(obj)
        load_bénévoles(obj)
        load_quêtes(obj)