import argparse
import yaml
from dotenv import load_dotenv
load_dotenv()
from sqlalchemy.orm import sessionmaker
from truthinquiry.ext.database.models import *
from truthinquiry.ext.database.sa import engine

Session = sessionmaker(bind=engine)
session = Session()

# full reset .w.
Base.metadata.drop_all(engine)
Base.metadata.create_all(engine)


class LocaleManager():
    def __init__(self):
        self.used_lids = [0]
        [self.used_lids.append(locale.LID) for locale in session.query(Locale).all()]

    def get_unused_lid(self):
        new_lid = max(self.used_lids) + 1
        self.used_lids.append(new_lid)
        return new_lid
    
    def get_used_lids(self):
        return self.used_lids[1:]


def bulk_import(data):

    # Data list that will be commited to the db
    TEXT_LIST = []
    TRAIT_LIST = []
    REACTION_LIST = []
    QUESTIONS_LIST = []
    ANSWER_LIST = []
    NPC_LIST = []
    ROOMS_LIST = []
    
    # helper list to simplify
    trait_names = {}
    lm = LocaleManager()
    getid = lm.get_unused_lid

    # Questions

    # Where type questions

    question_type_zero = QuestionType(0, getid())
    QUESTIONS_LIST.append(question_type_zero)

    question_type_one = QuestionType(1, getid())
    QUESTIONS_LIST.append(question_type_one)

    questions = data["questions"]
    # handle where type quetions
    for question in questions["where"]["text"]:
        lang = list(question.keys())[0]
        text = list(question.values())[0]
        TEXT_LIST.append(Text(0,question_type_zero.TEXT_LID, lang, text))

    # handle with who type quetions
    for question in questions["withwho"]["text"]:
        lang = list(question.keys())[0]
        text = list(question.values())[0]
        TEXT_LIST.append(Text(0,question_type_one.TEXT_LID, lang, text))

    # Traits
    traits = data["traits"]
    for trait in traits.values():
        # create the new trait
        new_trait = Trait(0,getid(), getid())

        for lang in trait["name"]:
            TEXT_LIST.append(Text(0,new_trait.NAME_LID,
                            lang, trait["name"][lang]))
            trait_names[trait["name"][lang]] = new_trait.TRAIT_ID

        for lang in trait["description"]:
            TEXT_LIST.append(Text(0,new_trait.DESC_LID, lang,
                            trait["description"][lang]))

        TRAIT_LIST.append(new_trait)

    # Npcs
    npcs = data["npcs"]
    npcid = 1
    for npc in npcs.values():
        new_npc = Npc(npcid, getid())

        # handle the names
        for lang in npc["name"]:
            TEXT_LIST.append(Text(0,new_npc.NAME_LID, lang, npc["name"][lang]))

        # TODO handle reactions
        """         
        for reaction in npc["reactions"]:
            if reaction in list(trait_names.keys()):
                new_reaction = Reaction(0,new_npc.NPC_ID, trait_names[reaction])
                REACTION_LIST.append(new_reaction) """

        for question_type in npc["answers"]:
            question_type_id = question_type_zero.QUESTION_TYPE_ID if question_type == "where" else question_type_one.QUESTION_TYPE_ID

            new_answer = Answer(question_type_id, new_npc.NPC_ID, getid())
            ANSWER_LIST.append(new_answer)

            for answer in npc["answers"][question_type]:
                lang = list(answer.keys())[0]
                text = list(answer.values())[0]
                TEXT_LIST.append(Text(0,new_answer.TEXT_LID, lang, text))

        NPC_LIST.append(new_npc)
        npcid += 1

    # rooms
    rooms = data["rooms"]
    for room in rooms.values():
        new_room = Place(0,getid())
        for lang in room:
            TEXT_LIST.append(Text(0,new_room.NAME_LID, lang, room[lang]))
        ROOMS_LIST.append(new_room)
    
    for lid in lm.get_used_lids():
        print("lid :"+ str(lid))
        session.add(Locale(lid));
    
    for text in TEXT_LIST:
        print("Text : "+str(text))
        session.add(text)
        session.commit()

    for question in QUESTIONS_LIST:
        print("Question : "+str(question))
        session.add(question)
        session.commit()

    for trait in TRAIT_LIST:
        print("Trait : "+ str(trait))
        session.add(trait)
        session.commit()

    for npc in NPC_LIST:
        print("Npc : "+ str(npc))
        session.add(npc)
        session.commit()

    for reaction in REACTION_LIST:
        print("Reaction : " + str(reaction))
        session.add(reaction)
        session.commit()

    for answer in ANSWER_LIST:
        print("Answer : "+ str(answer))
        session.add(answer)
        session.commit()

    for room in ROOMS_LIST:
        print("Room : "+str(room))
        session.add(room)
        session.commit()

file = open("bulk_data.yml", "r")

bulk_import(yaml.load(file, yaml.Loader))