import time
from job_list import Barbarian, Cleric, Wizard
from player import *
from race_list import Elf, Dwarf, Human


def wait():
    input("Press enter to continue...")


def startGame():
    print("Welcome, stranger.")
    while True:  # Global check to see if Name, Age, and Gender are correct
        # Initialise age
        age = None
        # Name
        name = input(f'\nWhat is your name?\n >> ').title()
        while len(name) < 2:
            name = input(f'Input name was too short. Try again. \nWhat is your name?\n >> ').title()
        # Gender
        while True:
            gender = None
            gender_select = input("What is your gender?\n"
                                  "1) Male\n2) Female\n3) Other\n >> ").lower()
            if gender_select not in ("male", "female", "other", "1", "2", "3"):
                print("Sorry I didn't recognise that gender. Please try again.")
                continue
            else:
                break
        if gender_select in ("1", "male"):
            gender = "Male"
        elif gender_select in ("2", "female"):
            gender = "Female"
        elif gender_select in ("3", "other"):
            gender = "Other"
        while True:
            # Age
            try:
                age = input(f'How old are you?\n >> ')
                age = int(age)
                break
            except ValueError:
                print("Sorry I didn't recognise that age. Please type in a whole number.")
                continue
        # Double check responses
        correct = input(f"Hello, {name}. You are a {age} year old {gender}.\nIs this correct? Y/N\n >> ").lower()
        if correct in ['y', 'yes']:
            pass
        elif correct in ['n', 'no']:
            continue
        else:
            print("Sorry, I didn't catch that.\n")
            correct = input(f"You are a {age} year old {gender}. \nIs this correct? Y/N\n >> ").lower()
            if correct in ['y', 'yes']:
                break
            elif correct in ['n', 'no']:
                continue
        break

    while True:  # Global check to see if Race and Job are correct.
        # Race
        while True:
            race = input("Please select a race to learn more about it: Elf, Dwarf, or Human.\n >> ").title()
            if race not in ("Elf", "Dwarf", "Human", None):
                race = input("Sorry I didn't recognise that race. "
                             "Please select 'Elf', 'Dwarf', or 'Human'.\n >> ").title()
            elif race is None:
                race = input("Please select a race to learn more about it: Elf, Dwarf, or Human.\n >> ").title()
                continue
            else:
                if race == "Elf":
                    print(Elf)
                elif race == "Dwarf":
                    print(Dwarf)
                elif race == "Human":
                    print(Human)
            print('Would you like to proceed with this race or view another?')
            proceed = input(f'1) Proceed\n2) View another race\n >> ').lower()
            if proceed in ('1', 'proceed'):
                break
            elif proceed in ('2', 'view', 'view another', 'view another race'):
                race = None
                continue
        # Job
        while True:
            job = input("Please select a job to learn more about it: Barbarian, Cleric, or Wizard.\n >> ").title()
            if job not in ("Barbarian", "Cleric", "Wizard", None):
                job = input("Sorry I didn't recognise that job. "
                            "Please select 'Barbarian', 'Cleric', or 'Wizard'.\n >> ").title()
            elif job is None:
                job = input("Please select a job to learn more about it: Barbarian, Cleric, or Wizard.\n >> ").title()
                continue
            else:
                if job == "Barbarian":
                    print(Barbarian)
                elif job == "Cleric":
                    print(Cleric)
                elif job == "Wizard":
                    print(Wizard)
            print('Would you like to proceed with this job or view another?')
            proceed = input(f'1) Proceed\n2) View another job\n >> ').lower()
            if proceed in ('1', 'proceed'):
                break
            elif proceed in ('2', 'view', 'view another', 'view another job'):
                job = None
                continue
        while True:
            # Final check
            correct = input(f"{name}, you are a {race} {job}.\nIs this correct? Y/N\n >> ").lower()
            if correct in ['y', 'yes']:
                break
            elif correct in ['n', 'no']:
                continue
            else:
                print("Sorry, I didn't catch that. Please try again.\n")
                continue
        return name, gender, age, race, job


def startGame_PreGen():
    """Use this for when the player is familiar with class races and jobs."""


def query_equip():
    while True:
        print("Do you want to:\n"
              "1) View your equipped items\n"
              "2) Change your equipped items\n"
              "3) Return")
        response = input(">> ")
        if response == "1":
            current_equipment()
        elif response == "2":
            change_equip()
        elif response == "3":
            break


def change_equip():
    while True:
        response = input("Which equipment slot would you like to change?\n"
                         "1) Weapon\n"
                         "2) Armor\n"
                         "3) Other\n"
                         "4) Return\n"
                         ">> ").title()
        if response not in ("1", "2", "3", "4", "Weapon", "Armor", "Other", "Return"):
            print("Sorry I didn't recognise that. Please try again.\n")
            change_equip()
        elif response in ("1", "Weapon"):  # Weapon
            itemlist = ["Return"]  # create blank list for items
            view_equippedweapon()
            for item_key, item_value in Inventory.items():  # item, attribute
                for key, value in item_value.items():  # attribute, stat
                    if key == "object" and value == Weapon:  # if item = class `Weapon`
                        itemlist.append(item_key)
                        if item_key in Equipment.values():
                            itemlist.remove(item_key)
            if len(itemlist) == 1:
                print("You have nothing available to equip.\n")
            else:
                print("What would you like to equip?\n")
                for i, name in enumerate(itemlist, start=1):
                    option = f'{i}) {name}'
                    print(option)
                selection = int(input(">> "))
                for i, name in enumerate(itemlist, start=1):
                    if selection == i:
                        if name == "Return":
                            break
                        elif name.itemType == Weapon:
                            try:
                                if name.attunement:
                                    print("Not coded attunement yet")  # if item has attunement - check availability
                            except AttributeError:  # if no attunement
                                if name.slot == Slots.MainHand:
                                    equip = {"Main Hand": name}
                                    Equipment.update(equip)
                                elif not name.versatile:
                                    equip = {"Off Hand": name}
                                    Equipment.update(equip)
                                else:
                                    equip = {"Main Hand": None, "Off Hand": None, "Two Handed": name}
                                    Equipment.update(equip)
                print(f"You now have equipped: \n"
                      f"{view_equippedweapon()}")
        elif response in ("2", "Armor"):  # Armor
            itemlist = ["Return"]  # create blank list for items
            view_equippedarmor()
            for item_key, item_value in Inventory.items():  # item, attribute
                for key, value in item_value.items():  # attribute, stat
                    if key == "object" and value == Armor:  # if item = class `Weapon`
                        itemlist.append(item_key)  # Add item type to list
                        if item_key in Equipment.values():
                            itemlist.remove(item_key)  # remove from list if currently equipped -
                            # consider if holding multiple objects
            if len(itemlist) == 1:
                print("You have nothing available to equip.\n")
            else:
                print("What would you like to equip?\n")
                for i, name in enumerate(itemlist, start=1):
                    option = f'{i}) {name}'
                    print(option)
                selection = int(input(">> "))
                for i, name in enumerate(itemlist, start=1):
                    if selection == i:
                        if name == "Return":
                            break
                        elif name.itemType == Armor:
                            try:
                                if name.attunment:
                                    print("Not coded attunement yet")  # if item has attunement - check availability
                            except AttributeError:  # if no attunement
                                if name.slot == Slots.Helm:
                                    equip = {"Helm": name}
                                    Equipment.update(equip)
                                elif name.slot == Slots.Chest:
                                    equip = {"Chest": name}
                                    Equipment.update(equip)
                                elif name.slot == Slots.Wrists:
                                    equip = {"Wrists": name}
                                    Equipment.update(equip)
                                elif name.slot == Slots.Feet:
                                    equip = {"Feet": name}
                                    Equipment.update(equip)
                print(f"You now have equipped: \n"
                      f"{view_equippedarmor()}")
        elif response in ("3", "Other"):  # Other
            itemlist = ["Return"]  # create blank list for items
            view_equippedother()
            for item_key, item_value in Inventory.items():  # item, attribute
                for key, value in item_value.items():  # attribute, stat
                    if key == "object" and value == Armor:  # if item = class `Weapon`
                        itemlist.append(item_key)  # Add item type to list
                        if item_key in Equipment.values():
                            itemlist.remove(item_key)  # remove from list if currently equipped -
                            # consider if holding multiple objects
            if len(itemlist) == 1:
                print("You have nothing available to equip.\n")
            else:
                print("What would you like to equip?\n")
                for i, name in enumerate(itemlist, start=1):
                    option = f'{i}) {name}'
                    print(option)
                selection = int(input(">> "))
                for i, name in enumerate(itemlist, start=1):
                    if selection == i:
                        if name == "Return":
                            break
                        elif name.itemType == Armor:
                            try:
                                if name.attunment:
                                    print("Not coded attunement yet")  # if item has attunement - check availability
                            except AttributeError:  # if no attunement
                                if name.slot == Slots.Neck:
                                    equip = {"Neck": name}
                                    Equipment.update(equip)
                                elif name.slot == Slots.Cloak:
                                    equip = {"Cloak": name}
                                    Equipment.update(equip)
                                elif name.slot == Slots.LeftRing:
                                    equip = {"Left Ring": name}
                                    Equipment.update(equip)
                                elif name.slot == Slots.RightRing:
                                    equip = {"Right Ring": name}
                                    Equipment.update(equip)
                print(f"You now have equipped: \n"
                      f"{view_equippedother()}")
        elif response in ("4", "Return"):
            break
        else:
            print("That wasn't an option.")


def view_equippedweapon():
    equipped = ""
    for slot, equipment in Equipment.items():
        if slot in ("Main Hand", "Off Hand", "Two Handed") and equipment is not None:
            equipped = equipped + f'{slot}: {equipment} \n'
            return "You currently have equipped: \n" + equipped
        else:
            return "You have nothing equipped!"


def view_equippedarmor():
    equipped = ""
    for slot, equipment in Equipment.items():
        if slot in ("Helm", "Chest", "Wrists", "Feet") and equipment is not None:
            equipped = equipped + f'{slot}: {equipment} \n'
            return "You currently have equipped: \n" + equipped
        else:
            return "You have nothing equipped!"


def view_equippedother():
    equipped = ""
    for slot, equipment in Equipment.items():
        if slot in ("Neck", "Cloak", "Left Ring", "Right Ring") and equipment is not None:
            equipped = equipped + f'{slot}: {equipment} \n'
            return "You currently have equipped: \n" + equipped
        else:
            return "You have nothing equipped!"


def get_instructions():
    descrip1 = ("There are certain commands that will be available almost anytime you are able to type,\n"
                "such as viewing your inventory, checking your equipped items, and also changing them.\n"
                "You can also view your stats including your current and max hp.\n")
    descrip2 = ("Some examples are: 'Check inventory', 'Check equipment', and 'View stats'.\n"
                "To travel to a new area, just type 'Go north' or 'enter cave' etc.\n"
                "To replay the description of the current area, type 'location'.")
    print(descrip1)
    time.sleep(5)
    print(descrip2)


def error_message():
    print("That command was not recognised. Please try again.")


def parse(input_text):
    command = input_text.lower()
    object1 = None
    remaining_words_index = None

    # the .split() function splits the input_text string into a list of individual words
    words = input_text.split()

    if len(words) > 0:
        if words[0] == "help":
            command = "help"

        if words[0] == "scene":
            command = "scene"

        if words[0] in ("view", "check", "show"):
            if words[1] in ("inventory", "bag", "backpack"):
                command = "inventory"
            elif words[1] in ("equipment", "equip", "items"):
                command = "equipment"
            elif words[1] == "stats":
                command = "stats"
            elif words[1] in ("hp", "hitpoints", "health"):
                command = "hp"
            elif words[1] == "hit" and words[2] == "points":
                command = "hp"

        if words[0] in ("equip", "equipment"):
            command = "equipment"

        if words[0] in ("inventory", "bag", "backpack"):
            command = "inventory"

        found_examine_words = False
        if (words[0] == "examine") and len(words) > 1:
            found_examine_words = True
            remaining_words_index = 1

        if found_examine_words:
            remaining_words = ""
            for i in range(remaining_words_index, len(words)):
                remaining_words += words[i]
                if i < len(words) - 1:
                    remaining_words += " "
            command = "examine"
            object1 = remaining_words

        found_take_words = False
        if ((words[0] == "take") and len(words) > 1) or \
                ((words[0] == "pick") and (words[1] == "up") and len(words) > 2):
            found_take_words = True
            remaining_words_index = 1

        if found_take_words:
            remaining_words = ""
            for i in range(remaining_words_index, len(words)):
                remaining_words += words[i]
                if i < len(words) - 1:
                    remaining_words += " "
            command = "take"
            object1 = remaining_words

        found_loot_words = False
        if (words[0] == "loot") and len(words) > 1:
            found_loot_words = True
            remaining_words_index = 1

        if found_loot_words:
            remaining_words = ""
            for i in range(remaining_words_index, len(words)):
                remaining_words += words[i]
                if i < len(words) - 1:
                    remaining_words += " "
            command = "loot"
            object1 = remaining_words

        found_drop_words = False
        if (words[0] == "drop") and len(words) > 1:
            found_drop_words = True
            remaining_words_index = 1

        if found_drop_words:
            remaining_words = ""
            for i in range(remaining_words_index, len(words)):
                remaining_words += words[i]
                if i < len(words) - 1:
                    remaining_words += " "
            command = "drop"
            object1 = remaining_words

        if words[0] not in ("view", "check", "show", "loot", "examine", "take", "equip", "equipment", "inventory",
                            "bag", "backpack", "drop"):
            command = "error"

    return command, object1


def skill_check(skill):
    """Roll a d20 and add modifier
    Return value to compare vs a hardcoded DC"""


def roll_dice(num_dice):
    roll_results = []
    for _ in range(num_dice):
        roll = randint(*user.hitDie)
        roll_results.append(roll)
        return roll_results


def current_inventory():
    """Prints a list of items in your backpack that aren't equipped to your person."""
    print(f'In your rucksack you have:')
    for item_key, item_value in Inventory.items():  # string, dictionary
        if item_key not in Equipment.values():
            values = item_value.values()  # make a variable of all values
            values_list = list(values)  # turn variable into a list
            count_value = values_list[0]  # take the first value from list - why?
            print(f'{item_key.name} x {count_value}\n'
                  f'    {item_key.description}\n')


def current_equipment():
    """Prints a list of items that you have equipped in your slots."""
    print(f'You are currently wearing:')
    for item_key, item_value in Equipment.items():  # string, dictionary
        if item_value is not None:
            if item_value.itemType == ItemType.Weapon:
                print(f'{item_key}: {item_value.name} - {item_value.damage1H}\n'
                      f'    {item_value.description}\n')
            elif item_value.itemType == ItemType.Armor:
                print(f'{item_key}: {item_value.name} - AC {item_value.ac}\n'
                      f'    {item_value.description}\n')
            elif item_value.itemType == ItemType.Item:
                print(f'{item_key}: {item_value.name}\n'
                      f'    {item_value.description}\n')


def calc_weight():
    pass