#include <sstream>
#include <string>
#include <cmath>

#include "consoleCommand.h"
#include "unitData.h"
#include "aircraftCarrier.h"
#include "cruiser.h"
#include "destroyer.h"
#include "submarine.h"
#include "demoJet.h"
#include "missileJet.h"

using namespace irr::core;
using namespace game::core;
using namespace game::content;
using namespace std;

namespace game{
    namespace gui{
        ConsoleCommand::ConsoleCommand(GameManager *gM, Listbox *l, vector<Player*> players, stringw name, vector<stringw> args) {
            gameManager = gM;
            listbox = l;
            this->players = players;
            this->name = name;
            arguments = args;
            execute();
        }

        ConsoleCommand::~ConsoleCommand() {
        }

        void ConsoleCommand::execute() {
            if (name == commandList[0])
                addUnit();
            else if (name == commandList[1])
                debugUnits();
            else
                printConsoleMessage("No such command");
        }

        void ConsoleCommand::addUnit() {
            int unitId = 0, playerId = 0;
            vector3df pos(0, 0, 0);
            float angle = 0;
            for (int i = 0; i < arguments.size(); i++) {
                arguments[i].erase(0);
                for (int i2 = 0; i2 < arguments[i].size(); i2++) {
                    if (i == 0)
                        playerId += (arguments[i][i2] - 48) * pow(10, arguments[i].size() - (i2 + 1));
                    else
                        unitId += (arguments[i][i2] - 48) * pow(10, arguments[i].size() - (i2 + 1));
                }
            }
            if ((playerId == 0 || playerId == 1) && (unitId >= 0 && unitId <= numberOfUnits)) {
                Unit *u = nullptr;
                switch (unitType[unitId]) {
                    case UNIT_TYPE::VESSEL:
                        u = new Vessel(gameManager, players[playerId],pos, unitId);
                        break;
                    case UNIT_TYPE::DESTROYER:
                        u = new Destroyer(gameManager, players[playerId], pos, unitId);
                        break;
                    case UNIT_TYPE::CRUISER:
                        u = new Cruiser(gameManager, players[playerId],pos, unitId);
                        break;
                    case UNIT_TYPE::SUBMARINE:
                        u = new Submarine(gameManager, players[playerId], pos, unitId);
                        break;
                    case UNIT_TYPE::AIRCRAFT_CARRIER:
                    {
                        u = new AircraftCarrier(gameManager, players[playerId],pos, unitId);
                        break;
                    }
                    case UNIT_TYPE::MISSILE_JET:
                        u = new MissileJet(gameManager, players[playerId],pos, unitId, false);
                        break;
                    case UNIT_TYPE::DEMO_JET:
                        u = new DemoJet(gameManager, players[playerId],pos, unitId, false);
                        break;
                    default:
                        break;
                }
                players[playerId]->addUnit(u);
            } else if (playerId != 0 && playerId != 1)
                printConsoleMessage("Invalid player id");
            else if (unitId < 0 || unitId > numberOfUnits)
                printConsoleMessage("Invalid unit id");
        }

        void ConsoleCommand::debugUnits() {
            if (arguments.size() == 0)
                for (Player *pl : players){
                    vector<Unit*> &units=pl->getUnits();
                    for (Unit *u : units){
                        u->toggleDebugging(true);
                        vector<Projectile*> projectiles=u->getProjectiles();
                        for(Projectile *pr : projectiles)
                            pr->debug();
                    }
                }
            else
                printConsoleMessage("Too many arguments");
        }

        void ConsoleCommand::printConsoleMessage(stringw message) {
            int messageId = 0;
            bool foundEmptySlot = false;
            for (int i = 0; i < listbox->getContents().size() && !foundEmptySlot; i++)
                if (listbox->getContents()[i] == "") {
                    messageId = i;
                    foundEmptySlot = true;
                }
            if (foundEmptySlot)
                listbox->changeLine(messageId, message);
            else
                listbox->addLine(message);
        }
    }
}