#include "Piece.hpp"

#include <ostream>

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

Piece::Optional Piece::fromSymbol(char symbol) {
    PieceColor color = std::isupper(symbol) ? PieceColor::White : PieceColor::Black;
    auto type = pieceTypeFromSymbol(symbol);
    if (!type.has_value()) {
        return std::nullopt;
    }

    return Piece(color, type.value());
}

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

PieceType Piece::type() const {
    return mType;
}
std::optional<PieceType> Piece::pieceTypeFromSymbol(char symbol) {
    auto tryType = static_cast<PieceType>(toupper(symbol));

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

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);
}