import csv
import db
import io
import json
import os

from glob import glob


def update_missions(dir_in, old_path, languages):
    qdb = db.QuestDB(dir_in)
    quest_mst(qdb, old_path, languages)
    scene_mst(qdb, old_path, dir_in, languages)


def scene_mst(qdb, old_path, dir_in, languages):
    scenes = qdb.get_scenes()

    locations = load_locations(dir_in)

    for id, scene in scenes.items():
        i = 0
        while True:
            try:
                scene["Folder"] = locations[scene["Parts"][i]]
            except KeyError as exc:
                i += 1
                if i == len(scene["Parts"]):
                    raise KeyError(f"Could not find folder for scene {id}") from exc
                continue
            break

    with io.open(
        os.path.join(old_path, "XduScene.json"), "w", newline="\n"
    ) as json_file:
        json.dump(scenes, json_file, ensure_ascii=False, indent="\t", sort_keys=True)

    for lang in languages:
        out_dict = {}
        if lang == "jpn":
            for key, value in scenes.items():
                out_dict[key] = {
                    "Name": value["Name"],
                    "SummaryText": value["SummaryText"],
                    "Credits": "POKELABO",
                    "Enabled": False,
                }
        else:
            for key in scenes.keys():
                out_dict[key] = {
                    "Name": "",
                    "SummaryText": "",
                    "Credits": "",
                    "Enabled": False,
                }

        langfile = os.path.join(old_path, f"XduSceneNames_{lang}.json")
        if os.path.isfile(langfile):
            with open(langfile, "r") as lang_file:
                lang_dict = json.load(lang_file)
                out_dict.update(lang_dict)

        with io.open(langfile, "w", newline="\n") as lang_file:
            json.dump(
                out_dict, lang_file, ensure_ascii=False, indent="\t", sort_keys=True,
            )


def quest_mst(qdb, old_path, languages):
    quests = qdb.get_quests()

    with io.open(
        os.path.join(old_path, "XduQuest.json"), "w", newline="\n"
    ) as json_file:
        json.dump(quests, json_file, ensure_ascii=False, indent="\t", sort_keys=True)

    for lang in languages:
        out_dict = {}
        if lang == "jpn":
            for key, value in quests.items():
                out_dict[key] = {"Name": value["Name"], "Enabled": False}
        else:
            for key, value in quests.items():
                out_dict[key] = {"Name": "", "Enabled": False}

        langfile = os.path.join(old_path, f"XduQuestNames_{lang}.json")
        if os.path.isfile(langfile):
            with open(langfile, "r") as lang_file:
                lang_dict = json.load(lang_file)
                out_dict.update(lang_dict)

        with io.open(langfile, "w", newline="\n") as lang_file:
            json.dump(
                out_dict, lang_file, ensure_ascii=False, indent="\t", sort_keys=True,
            )

    return quests


def load_locations(dir_in):
    locations = {}
    settings = glob(
        os.path.join(dir_in, "Common/Asset/Utage/**/Settings/Scenario.tsv"),
        recursive=True,
    )

    if len(settings) == 0:
        raise FileNotFoundError("Ya didn't decrypt the files dummy")

    for scenario in settings:
        with open(scenario, "r") as tsv_file:
            tsv = csv.DictReader(tsv_file, delimiter="\t", quotechar='"')
            for row in tsv:
                tokens = row["FileName"].split("/")
                if len(tokens) < 3:
                    continue
                if tokens[2] in locations:
                    raise ValueError("Conflicting paths for file {tokens[2]}")
                locations[tokens[2]] = tokens[0]

    return locations