import enum

import gdb.printing


class Square(object):
    """
    Python representation of a 'seer::board::square::Square' raw value.
    """

    FILES = list(map(lambda n: chr(ord("A") + n), range(8)))
    RANKS = list(map(lambda n: str(n + 1), range(8)))

    def __init__(self, val):
        self._val = val

    def __str__(self):
        return self.FILES[self.file] + self.RANKS[self.rank]

    @property
    def rank(self):
        return int(self._val) % 8

    @property
    def file(self):
        return int(self._val) // 8


class Bitboard(object):
    """
    Python representation of a 'seer::board::bitboard::Bitboard' raw value.
    """

    def __init__(self, val):
        self._val = val

    def __str__(self):
        return "[" + ", ".join(map(str, self.squares)) + "]"

    @property
    def squares(self):
        n = self._val
        while n:
            b = n & (~n + 1)
            yield Square(b.bit_length() - 1)
            n ^= b


class Color(enum.IntEnum):
    """
    Python representation of a 'seer::board::color::Color' raw value.
    """

    # Should be kept in sync with the enum in `color.rs`
    WHITE = 0
    BLACK = 1

    def __str__(self):
        return self.name.title()


class File(enum.IntEnum):
    """
    Python representation of a 'seer::board::file::File' raw value.
    """

    # Should be kept in sync with the enum in `file.rs`
    A = 0
    B = 1
    C = 2
    D = 3
    E = 4
    F = 5
    G = 6
    H = 7

    def __str__(self):
        return self.name.title()


class Rank(enum.IntEnum):
    """
    Python representation of a 'seer::board::rank::Rank' raw value.
    """

    # Should be kept in sync with the enum in `rank.rs`
    First = 0
    Second = 1
    Third = 2
    Fourth = 3
    Fifth = 4
    Sixth = 5
    Seventh = 6
    Eighth = 7

    def __str__(self):
        return self.name.title()


class Piece(enum.IntEnum):
    """
    Python representation of a 'seer::board::piece::Piece' raw value.
    """

    # Should be kept in sync with the enum in `piece.rs`
    KING = 0
    QUEEN = 1
    ROOK = 2
    BISHOP = 3
    KNIGHT = 4
    PAWN = 5

    def __str__(self):
        return self.name.title()


class SquarePrinter(object):
    "Print a seer::board::square::Square"

    def __init__(self, val):
        self._val = Square(val)

    def to_string(self):
        return str(self._val)


class BitboardPrinter(object):
    "Print a seer::board::bitboard::Bitboard"

    def __init__(self, val):
        self._val = Bitboard(int(val["__0"]))

    def to_string(self):
        return "Bitboard{" + str(self._val)[1:-1] + "}"


class ColorPrinter(object):
    "Print a seer::board::color::Color"

    def __init__(self, val):
        self._val = Color(int(val))

    def to_string(self):
        return str(self._val)


class FilePrinter(object):
    "Print a seer::board::file::File"

    def __init__(self, val):
        self._val = File(int(val))

    def to_string(self):
        return str(self._val)


class RankPrinter(object):
    "Print a seer::board::rank::Rank"

    def __init__(self, val):
        self._val = Rank(int(val))

    def to_string(self):
        return str(self._val)


class PiecePrinter(object):
    "Print a seer::board::piece::Piece"

    def __init__(self, val):
        self._val = Piece(int(val))

    def to_string(self):
        return str(self._val)


def build_pretty_printer():
    pp = gdb.printing.RegexpCollectionPrettyPrinter('seer')

    pp.add_printer('Square', '^seer::board::square::Square$', SquarePrinter)
    pp.add_printer('Bitboard', '^seer::board::bitboard::Bitboard$', BitboardPrinter)
    pp.add_printer('Color', '^seer::board::color::Color$', ColorPrinter)
    pp.add_printer('File', '^seer::board::file::File$', FilePrinter)
    pp.add_printer('Rank', '^seer::board::rank::Rank$', RankPrinter)
    pp.add_printer('Piece', '^seer::board::piece::Piece$', ColorPrinter)

    return pp

gdb.printing.register_pretty_printer(gdb.current_objfile(), build_pretty_printer(), True)