#include "CheckController.hpp"

#include "MoveController.hpp"

CheckController::CheckController(Player* whitePlayer, Player* blackPlayer) {

this->whitePlayer = whitePlayer;

this->blackPlayer = blackPlayer;

}

CheckController::~CheckController() {}

Player* CheckController::getWhitePlayer() {

return this->whitePlayer;

}

Player* CheckController::getBlackPlayer() {

return this->blackPlayer;

}

std::vector<Player*>* CheckController::getClonedPositionAfterMove(Move* move) {

Player* clonedWhitePlayer = whitePlayer->clonePlayer();

Player* clonedBlackPlayer = blackPlayer->clonePlayer();

std::vector<Figure*>* clonedWhiteFigures = clonedWhitePlayer->getUncapturedFigures();

std::vector<Figure*>* clonedBlackFigures = clonedBlackPlayer->getUncapturedFigures();

if (move->getFigure()->getIsWhite()) {

Figure* figure = clonedWhitePlayer->getPieceAtPosition(move->getStartHorizontalPosition(), move->getStartVerticalPosition());

figure->setHorizontalPosition(move->getEndHorizontalPosition());

figure->setVerticalPosition(move->getEndVerticalPosition());

if (Capture* capture = dynamic_cast<Capture*>(move)) {

Figure* capturedFigure = clonedBlackPlayer->getPieceAtPosition(capture->getCapturedFigure()->getHorizontalPosition(), capture->getCapturedFigure()->getVerticalPosition());

capturedFigure->setNotCaptured(false);

}

} else {

Figure* figure = clonedBlackPlayer->getPieceAtPosition(move->getStartHorizontalPosition(), move->getStartVerticalPosition());

figure->setHorizontalPosition(move->getEndHorizontalPosition());

figure->setVerticalPosition(move->getEndVerticalPosition());

if (Capture* capture = dynamic_cast<Capture*>(move)) {

Figure* capturedFigure = clonedWhitePlayer->getPieceAtPosition(capture->getCapturedFigure()->getHorizontalPosition(), capture->getCapturedFigure()->getVerticalPosition());

capturedFigure->setNotCaptured(false);

}

}

std::vector<Player*>* clonedPlayers = new std::vector<Player*>();

if (move->getFigure()->getIsWhite()) {

clonedPlayers->push_back(clonedWhitePlayer);

clonedPlayers->push_back(clonedBlackPlayer);

} else {

clonedPlayers->push_back(clonedBlackPlayer);

clonedPlayers->push_back(clonedWhitePlayer);

}

return clonedPlayers;

}

std::vector<Move*>* CheckController::validateMoves(std::vector<Move*>* moves) {

std::vector<Move*>* validMoves = new std::vector<Move*>();

for (Move* move : *moves) {

if (Castle* castle = dynamic_cast<Castle*>(move)) {

std::vector<Player*>* currentPosition;

if (castle->getKing()->getIsWhite()) {

currentPosition->push_back(this->whitePlayer);

currentPosition->push_back(this->blackPlayer);

} else {

currentPosition->push_back(this->blackPlayer);

currentPosition->push_back(this->whitePlayer);

}

if (this->isCheck(currentPosition)) {

break;

}

}

std::vector<Player*>* clonedPosition = this->getClonedPositionAfterMove(move);

if (!this->isCheck(clonedPosition)) {

validMoves->push_back(move);

} else {

delete(move);

}

for (Player* p: *clonedPosition) {

delete(p);

}

clonedPosition->clear();

clonedPosition->shrink_to_fit();

}

return validMoves;

}

bool CheckController::isCheck(std::vector<Player*>* players) {

}