from piece import Piece
from board import Board
from square import Square
from abc import abstractmethod
class Rook(Piece):
    @abstractmethod
    def straightline_moves(self, piece, row, column, isvertical):
        if isvertical is True:
            self.direction = [
                (0, 1),
                (0, -1),
                (1, 0),
                (-1, 0)]
        else:
            self.direction = [
                (1, 1),
                (-1, -1),
                (1, -1),
                (-1, 1)
            ]
        for dir in self.direction:
            self.left, self.right = dir
            self.possible_row = row + self.left
            self.possible_column = column + self.right
            while True:
                if Square.on_board(self.possible_row, self.possible_column) is True:
                    if self.board.boardlist[self.possible_row][self.possible_column].has_piece() is False:
                        piece.add_moves((self.possible_row, self.possible_column))
                        self.possible_row = self.possible_row + self.left
                        self.possible_column = self.possible_column + self.right
                    elif self.board.boardlist[self.possible_row][self.possible_column].is_rival(piece.color) is True:
                        piece.add_moves((self.possible_row, self.possible_column))
                        break
                    else:
                        break
                else:
                    break