#include "Piece.hpp"

#include <ostream>

Piece::Piece(PieceColor color, PieceType type) : mColor(color), mType(type) {
}

Piece::Optional Piece::fromSymbol(char symbol) {
    PieceType type;
    PieceColor color = std::isupper(symbol) ? PieceColor::White : PieceColor::Black;

    auto tryType = static_cast<PieceType>(toupper(symbol));

    switch (tryType) {
    case PieceType::Pawn:type = PieceType::Pawn;
        break;
    case PieceType::Knight:type = PieceType::Knight;
        break;
    case PieceType::Bishop:type = PieceType::Bishop;
        break;
    case PieceType::Rook:type = PieceType::Rook;
        break;
    case PieceType::Queen:type = PieceType::Queen;
        break;
    case PieceType::King:type = PieceType::King;
        break;
    default:return std::nullopt;
    }

    return Piece(color, type);
}

PieceColor Piece::color() const {
    return mColor;
}

PieceType Piece::type() const {
    return mType;
}

bool operator==(const Piece &lhs, const Piece &rhs) {
    return lhs.color() == lhs.color() && lhs.type() == rhs.type();
}

std::ostream &operator<<(std::ostream &os, const Piece &piece) {
    int typeNum = static_cast<int>(piece.type());
    if (piece.color() == PieceColor::Black) {
        typeNum = std::tolower(typeNum);
    }

    os << static_cast<char>(typeNum);
    return os;
}

PieceColor operator!(PieceColor color) {
    return static_cast<PieceColor>((static_cast<int>(color) + 1) % 2);
}