import time
import sys
sys.path.append('.')

from player import Player # Only import what's necessary
from enum_list import ItemType, Slots
from race_list import Elf, Dwarf, Human
from job_list import Barbarian, Cleric, Wizard


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

def startGame():
    print("Welcome stranger, to the world of Kanjin!.")
    # Loop for initial choice: Create or Pre-generated
    while True:
        start = input("Would you like to create your own character or use pre-generated stats?"
                      "\n 1) Create a new character\n 2) Pre-generated\n >> ")

        if start in ("1", "Create"):
            # --- Character Basic Info (Name, Gender, Age) ---
            name, gender, age = None, None, None # Initialize
            # Loop until basic info is confirmed
            while True:
                # Name
                name = input('What is your name?\n >> ').title()
                while len(name) < 2:
                    name = input('Name must be longer than one character. Try again.\nWhat is your name?\n >> ')
                # Gender
                while True:
                    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"
                # Age
                while True:
                    try:
                        age = int(input('How old are you?\n >> '))
                        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}.\nYou are a {age} year old {gender.lower()}.\nIs this correct? Y/N\n >> ")
                if correct.lower() in ['y', 'yes']:
                    break # Exit basic info confirmation loop
                elif correct.lower() in ['n', 'no']:
                    continue # Restart basic info input
                else:
                    print("Sorry, I didn't catch that. Please try again.\n")
                    continue

            # --- Race and Job Selection ---
            race = None
            job = None
            # Loop until Race and Job are confirmed
            while True:
                # Race Selection
                selected_race_obj = None
                # Loop for selecting and viewing race
                while True:
                    race_choice = input("Please select a race to learn more about it:\n"
                                        "1) Elf\n2) Dwarf\n3) Human\n >> ").title()
                    if race_choice in ['Elf', '1']:
                        selected_race_obj = Elf
                    elif race_choice in ['Dwarf', '2']:
                        selected_race_obj = Dwarf
                    elif race_choice in ['Human', '3']:
                        selected_race_obj = Human
                    else:
                        print("Sorry I didn't recognise that race. Please select 'Elf', 'Dwarf', or 'Human'.\n")
                        continue

                    if selected_race_obj:
                        print(f"\n--- {selected_race_obj.name} ---")
                        print(selected_race_obj.__repr__())
                    print('Would you like to proceed with this race or view another?')
                    proceed = input('1) Proceed\n2) View another race\n >> ')
                    if proceed.lower() in ('1', 'proceed'):
                        race = selected_race_obj # Assign race object
                        break # Exit race selection loop
                    elif proceed.lower() in ('2', 'view', 'view another', 'view another race'):
                        continue # Restart race selection
                    else:
                        print("Sorry, I didn't catch that. Please try again.\n")
                        continue

                # Job Selection
                selected_job_obj = None
                # Loop for selecting and viewing job
                while True:
                    job_choice = input("Please select a job to learn more about it:\n"
                                        "1) Barbarian\n2) Cleric\n3) Wizard\n >> ").title()
                    if job_choice in ['Barbarian', '1']:
                        selected_job_obj = Barbarian
                    elif job_choice in ['Cleric', '2']:
                        selected_job_obj = Cleric
                    elif job_choice in ['Wizard', '3']:
                        selected_job_obj = Wizard
                    else:
                        print("Sorry I didn't recognise that job. Please select 'Barbarian', 'Cleric', or 'Wizard'.\n")
                        continue

                    if selected_job_obj:
                        print(f"\n--- {selected_job_obj.name} ---")
                        print(selected_job_obj.__repr__())

                    print('Would you like to proceed with this job or view another?')
                    proceed = input('1) Proceed\n2) View another job\n >> ')
                    if proceed.lower() in ('1', 'proceed'):
                        job = selected_job_obj # Assign job object
                        break # Exit job selection loop
                    elif proceed.lower() in ('2', 'view', 'view another', 'view another job'):
                        continue # Restart job selection
                    else:
                        print("Sorry, I didn't catch that. Please try again.\n")
                        continue

                # Final confirmation for both Race and Job
                while True:
                    # This safeguard should ideally not be hit if inner loops work correctly
                    if race is None or job is None:
                        print("Error: Race or Job not selected. Restarting Race/Job selection.")
                        break # Break out of this inner loop to restart the outer race/job loop
                    correct = input(f"\n{name}, you are a {race.name_adjective} {job.name}.\nIs this correct? Y/N\n >> ")
                    if correct.lower() in ['y', 'yes']:
                        return name, gender, age, race, job # All confirmed, return values
                    elif correct.lower() in ['n', 'no']:
                        # If not correct, break this loop to re-enter race/job selection
                        break
                    else:
                        print("Sorry, I didn't catch that. Please try again.\n")
                        continue

        elif start in ("2", "pre-generated"):
            print("Please be prepared to enter a name, gender, age, race, and class, from those available in the game."
                  "\nIf you are unsure what the options are, please go back and create a new character.")
            create = input("Do you wish to continue? Y/N\n >> ").lower()
            if create in ("y", "yes"):
                name = input("Name: ")
                gender = input("Gender: ")
                age = int(input("Age: "))
                race_str = input("Race (Elf, Dwarf, Human): ")
                job_str = input("Job (Barbarian, Cleric, Wizard): ")

                # Map string inputs to actual Race/Job objects for consistency
                race_map = {'elf': Elf, 'dwarf': Dwarf, 'human': Human}
                job_map = {'barbarian': Barbarian, 'cleric': Cleric, 'wizard': Wizard}

                actual_race = race_map.get(race_str.lower())
                actual_job = job_map.get(job_str.lower())

                if actual_race and actual_job:
                    return name, gender, age, actual_race, actual_job
                else:
                    print("Invalid race or job entered for pre-generated character. Please try again.")
                    # Force restart to character creation choice by setting 'start' to '1'
                    # and then continue the outermost loop.
                    start = "1"
                    continue
            elif create in ("n", "no"):
                # If 'n', re-prompt the initial choice.
                start = input("Would you like to create your own character or use pre-generated stats?"
                              "\n 1) Create a new character\n 2) Pre-generated\n >> ")
                continue
            else:
                print("Invalid input. Please enter Y or N.")
                # If invalid, re-prompt the initial choice.
                start = input("Would you like to create your own character or use pre-generated stats?"
                              "\n 1) Create a new character\n 2) Pre-generated\n >> ")
                continue
        else:
            print("Invalid input. Please select '1' or '2'.")
            # Re-prompt the initial choice.
            start = input("Would you like to create your own character or use pre-generated stats?"
                          "\n 1) Create a new character\n 2) Pre-generated\n >> ")
            continue

def query_equip(player: Player):
    """Allows the player to view or change equipped items."""
    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":
            player.inventory.current_equipment()
        elif response == "2":
            change_equip(player)
        elif response == "3":
            break

def change_equip(player: Player):
    """Handles the process of changing equipped items."""
    player.inventory.update_attunement() # Ensure attunement is up-to-date
    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 in ("1", "Weapon"):
            # Filter weapons from inventory that are not currently equipped in weapon slots
            itemlist = [item_obj for item_obj, item_data in player.inventory.items.items()
                        if item_data["object"] == ItemType.Weapon and
                        item_obj not in [player.inventory.equipped_items[Slots.MainHand],
                                         player.inventory.equipped_items[Slots.OffHand],
                                         player.inventory.equipped_items[Slots.TwoHanded]]]
            itemlist.append("Return") # Add return option last

            if not itemlist or (len(itemlist) == 1 and itemlist[0] == "Return"):
                print("You have no unequipped weapons available in your inventory.\n")
            else:
                print("What would you like to equip?\n")
                for i, item_obj in enumerate(itemlist, start=1):
                    if item_obj == "Return":
                        print(f'{i}) {item_obj}')
                    else:
                        print(f'{i}) {item_obj.name}')

                try:
                    selection = int(input(">> "))
                    if 1 <= selection <= len(itemlist):
                        chosen_item = itemlist[selection - 1]
                        if chosen_item == "Return":
                            break
                        else:
                            # Determine the correct slot based on weapon type
                            if chosen_item.slot == Slots.TwoHanded:
                                player.inventory.equip_item(chosen_item, Slots.TwoHanded)
                            elif chosen_item.versatile: # For versatile, equip to main hand
                                player.inventory.equip_item(chosen_item, Slots.MainHand)
                            else: # Default to main hand for 1H weapons
                                player.inventory.equip_item(chosen_item, Slots.MainHand)
                            player.inventory.current_equipment()
                    else:
                        print("Invalid selection.")
                except ValueError:
                    print("Invalid input. Please enter a number.")

        elif response in ("2", "Armor"):
            itemlist = [item_obj for item_obj, item_data in player.inventory.items.items()
                        if item_data["object"] == ItemType.Armor and
                        item_obj not in [player.inventory.equipped_items[Slots.Helm],
                                         player.inventory.equipped_items[Slots.Chest],
                                         player.inventory.equipped_items[Slots.Wrists],
                                         player.inventory.equipped_items[Slots.Feet]]]
            itemlist.append("Return")

            if not itemlist or (len(itemlist) == 1 and itemlist[0] == "Return"):
                print("You have no unequipped armor available in your inventory.\n")
            else:
                print("What would you like to equip?\n")
                for i, item_obj in enumerate(itemlist, start=1):
                    if item_obj == "Return":
                        print(f'{i}) {item_obj}')
                    else:
                        print(f'{i}) {item_obj.name}')

                try:
                    selection = int(input(">> "))
                    if 1 <= selection <= len(itemlist):
                        chosen_item = itemlist[selection - 1]
                        if chosen_item == "Return":
                            break
                        else:
                            # Logic for equipping armor based on its intended slot
                            if chosen_item.slot == Slots.Helm:
                                player.inventory.equip_item(chosen_item, Slots.Helm)
                            elif chosen_item.slot == Slots.Chest:
                                player.inventory.equip_item(chosen_item, Slots.Chest)
                            elif chosen_item.slot == Slots.Wrists:
                                player.inventory.equip_item(chosen_item, Slots.Wrists)
                            elif chosen_item.slot == Slots.Feet:
                                player.inventory.equip_item(chosen_item, Slots.Feet)
                            else:
                                print(f"This armor ({chosen_item.name}) doesn't seem to fit a standard slot.")
                            player.inventory.current_equipment()
                    else:
                        print("Invalid selection.")
                except ValueError:
                    print("Invalid input. Please enter a number.")

        elif response in ("3", "Other"):
            itemlist = [item_obj for item_obj, item_data in player.inventory.items.items()
                        if item_data["object"] == ItemType.Item and
                        item_obj not in [player.inventory.equipped_items[Slots.Neck],
                                         player.inventory.equipped_items[Slots.Cloak],
                                         player.inventory.equipped_items[Slots.LeftRing],
                                         player.inventory.equipped_items[Slots.RightRing],
                                         player.inventory.equipped_items[Slots.Other]]]
            itemlist.append("Return")

            if not itemlist or (len(itemlist) == 1 and itemlist[0] == "Return"):
                print("You have no other unequipped items available in your inventory.\n")
            else:
                print("What would you like to equip?\n")
                for i, item_obj in enumerate(itemlist, start=1):
                    if item_obj == "Return":
                        print(f'{i}) {item_obj}')
                    else:
                        print(f'{i}) {item_obj.name}')

                try:
                    selection = int(input(">> "))
                    if 1 <= selection <= len(itemlist):
                        chosen_item = itemlist[selection - 1]
                        if chosen_item == "Return":
                            break
                        else:
                            # Logic for equipping other items based on their intended slot
                            if chosen_item.slot == Slots.Neck:
                                player.inventory.equip_item(chosen_item, Slots.Neck)
                            elif chosen_item.slot == Slots.Cloak:
                                player.inventory.equip_item(chosen_item, Slots.Cloak)
                            elif chosen_item.slot == Slots.LeftRing:
                                player.inventory.equip_item(chosen_item, Slots.LeftRing)
                            elif chosen_item.slot == Slots.RightRing:
                                player.inventory.equip_item(chosen_item, Slots.RightRing)
                            elif chosen_item.slot == Slots.Other:
                                player.inventory.equip_item(chosen_item, Slots.Other)
                            else:
                                print(f"This item ({chosen_item.name}) doesn't seem to fit a standard 'other' slot.")
                            player.inventory.current_equipment()
                    else:
                        print("Invalid selection.")
                except ValueError:
                    print("Invalid input. Please enter a number.")

        elif response in ("4", "Return"):
            break
        else:
            print("That wasn't an option.")


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(1) # Reduced sleep for faster testing
    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
    words = input_text.split()

    if not words: # Handle empty input
        return "error", None

    # Check for multi-word commands first to avoid partial matches
    if len(words) >= 2:
        if words[0] == "check" and words[1] in ("inventory", "bag", "backpack"):
            command = "inventory"
            return command, object1
        elif words[0] == "check" and words[1] in ("equipment", "equip", "items"):
            command = "equipment"
            return command, object1
        elif words[0] == "check" and words[1] == "stats":
            command = "stats"
            return command, object1
        elif words[0] == "check" and words[1] in ("hp", "hitpoints", "health"):
            command = "hp"
            return command, object1
        elif words[0] == "go":
            command = "go"
            object1 = " ".join(words[1:]) # The rest of the words are the direction
            return command, object1
        elif words[0] == "pick" and words[1] == "up" and len(words) > 2:
            command = "take"
            object1 = " ".join(words[2:])
            return command, object1

    # Single-word commands
    if words[0] == "help":
        command = "help"
    elif words[0] == "scene" or words[0] == "location": # Added 'location' as an alias
        command = "scene"
    elif words[0] in ("inventory", "bag", "backpack"):
        command = "inventory"
    elif words[0] in ("equip", "equipment"):
        command = "equipment"
    elif words[0] == "stats":
        command = "stats"
    elif words[0] in ("hp", "hitpoints", "health"):
        command = "hp"
    elif words[0] == "examine":
        if len(words) > 1:
            command = "examine"
            object1 = " ".join(words[1:])
        else:
            command = "examine" # User needs to specify what to examine
            object1 = None
    elif words[0] == "take":
        if len(words) > 1:
            command = "take"
            object1 = " ".join(words[1:])
        else:
            command = "take" # User needs to specify what to take
            object1 = None
    elif words[0] == "loot":
        if len(words) > 1:
            command = "loot"
            object1 = " ".join(words[1:])
        else:
            command = "loot" # User needs to specify what to loot
            object1 = None
    elif words[0] == "drop":
        if len(words) > 1:
            command = "drop"
            object1 = " ".join(words[1:])
        else:
            command = "drop" # User needs to specify what to drop
            object1 = None
    elif words[0] == "quit":
        command = "quit"
    else:
        command = "error" # Default for unrecognized commands

    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 calc_weight():
    pass # Placeholder for future weight calculation