#include <iostream>

#include "Dungeon.h"

using namespace std;

Dungeon::Dungeon(Player p) {
    player = p;
}

void Dungeon::printActions(int numActions, string actions[]) {
    cout << "\nChoose an action:\n";

    for (int i = 0; i < numActions; i++) {
        cout << actions[i] << "\n";
    }
}

void Dungeon::handleLootActions(Room *room) {
    vector<Item> items = room->items;
    int size = room->items.size();
    
    player.lootRoom(room);
    room->clearLoot();
    
    for (int i = 0; i < size; i++) {
        cout << "You open the chest and find a " << items[i].name << "\n";
        cout << "Your health is now " << player.currentHealth << ", your attack is now " << player.attack << ", your defense is now " << player.defense << "\n";
    }
}

void Dungeon::handleFightActions(GameCharacter *enemy) {
    string actions[] = {"a. Attack", "b. Retreat"};

    while (true) {
        printActions(2, actions);
        
        string input;
        cin >> input;

        if (input == "a") {
            int damage = enemy->takeDamage(player.attack);
            cout << "Your attack does " << damage << " damage.\n";
        } else if (input == "b") {
            player.changeRooms(player.previousRoom);
            enterRoom(player.currentRoom);
            return;
        } else {
            cout << "Incorrect choice. Please try again.\n";
        }

        if (enemy->isDead()) {
            cout << "You win! You have defeated the enemy " << enemy->name << ".\n";
            player.increaseStats(10, 5, 5);
            player.currentRoom->clearEnemies();
            return;
        }

        int damage = player.takeDamage(enemy->attack);
        cout << "The " << enemy->name << " attacked and you took " << damage << " damage.\n";
        cout << "You now have " << player.currentHealth << " health.\n";

        if (player.isDead()) {
            cout << "You have been defeated...";
            return;
        }
    }
}


void Dungeon::handleEmptyRoom(Room *room) {
    cout << "You enter the room but it is empty.\n";
    string actions[] = {"a. Move to another room"};
    while (true) {
        printActions(1, actions);
        
        string input;
        cin >> input;

        if (input == "a") {
            return;
        } else {
            cout << "Incorrect choice. Please try again.\n";
        }
    }
}

void Dungeon::handleRoomWithChest(Room *room) {
    cout << "You enter the room and see a large chest in the middle.\n";
    string actions[] = {"a. Loot the chest", "b. Move to another room"};
    while (true) {
        printActions(2, actions);
        
        string input;
        cin >> input;

        if (input == "a") {
            handleLootActions(room);
            return;
        } else if (input == "b") {
            return;
        } else {
            cout << "Incorrect choice. Please try again.\n";
        }
    }
}

void Dungeon::handleRoomWithEnemy(Room *room) {
    GameCharacter enemy = room->enemies.front();
    cout << "You enter the room and see a " << enemy.name << " in the middle.\n";
    string actions[] = {"a. Fight the " + enemy.name, "b. Go back to previous room"};
    while (true) {
        printActions(2, actions);
        
        string input;
        cin >> input;

        if (input == "a") {
            handleFightActions(&enemy);
            return;
        } else if (input == "b") {
            player.changeRooms(player.previousRoom);
            enterRoom(player.currentRoom);
            return;
        } else {
            cout << "Incorrect choice. Please try again.\n";
        }
    }
}

void Dungeon::enterRoom(Room *room) {
    if (room->enemies.size() > 0) {
        handleRoomWithEnemy(room);
    } else if (room->items.size() > 0) {
        handleRoomWithChest(room);
    } else {
        handleEmptyRoom(room);
    }
}

void Dungeon::handleMovementActions(Room *room) {
    while (true){
        if (room->position == 0) {
            string actions[] = {"a. Move right", "b. Move down"};
            printActions(2, actions);

            string input;
            cin >> input;

            if (input == "a") {
                player.changeRooms(&rooms[1]);
                return;
            } else if (input == "b") {
                player.changeRooms(&rooms[2]);
                return;
            } else {
                cout << "Incorrect choice. Please try again.\n";
            }
        }
        else if (room->position == 1) {
            string actions[] = {"a. Move left"};
            printActions(1, actions);

            string input;
            cin >> input;

            if (input == "a") {
                player.changeRooms(&rooms[0]);
                return;
            } else {
                cout << "Incorrect choice. Please try again.\n";
            }
        }
        else if (room->position == 2) {
            string actions[] = {"a. Move up", "b. Move right"};
            printActions(2, actions);

            string input;
            cin >> input;

            if (input == "a") {
                player.changeRooms(&rooms[0]);
                return;
            } else if (input == "b") {
                player.changeRooms(&rooms[3]);
                return;
            } else {
                cout << "Incorrect choice. Please try again.\n";
            }
        }
        else  {
            string actions[] = {"a. Move left"};
            printActions(1, actions);

            string input;
            cin >> input;

            if (input == "a") {
                player.changeRooms(&rooms[2]);
                return;
            } else {
                cout << "Incorrect choice. Please try again.\n";
            }
        }
    }
}

int Dungeon::performEndGameLogic() {
    while (true) {
        string actions[] = {"a. Yes", "b. No"};
        printActions(2, actions);

        string input;
        cin >> input;

        if (input == "a") {
            return 1;
        } else if (input == "b") {
            return 0;
        } else {
            cout << "Incorrect choice. Please try again.\n";
        }
    }
}

int Dungeon::runDungeon() {
    cout << "Welcome to the dungeon! Inside you will find treasure but also enemies. Enter at your peril!\n";

    player.currentRoom = &rooms[0];
    player.previousRoom = &rooms[0];
    
    while (true) {
        enterRoom(player.currentRoom);

        if (player.isDead()) {
            cout << "Game over! Try again?\n";
            return performEndGameLogic();
        } else {
            if (player.currentRoom->isExit) {
                if (player.currentRoom->enemies.size() <= 0) {
                    cout << "You win! Play again?\n";
                    return performEndGameLogic();
                }
            }
        }

        handleMovementActions(player.currentRoom);
    }
}