from Deck import Deck

class Solitaire:
    def __init__(self):
        self.deck = Deck()
        self.face_up = []

    def deal_until_four(self):
        while len(self.face_up) < 4 and self.deck.number_of_cards() > 0:
            self.face_up.append(self.deck.deal())

    def remove_four_same_suit(self) -> bool:
        if len(self.face_up) < 4:
            return False
        
        last_four = self.face_up[-4:]
        suit = last_four[0].get_suit()

        if all(card.get_suit() == suit for card in last_four):
            del self.face_up[-4:]
            return True
        
        return False
    
    def remove_inner_two(self) -> bool:
        if len(self.face_up) < 4:
            return False

        last_four = self.face_up[-4:]
        if last_four[0].get_suit() == last_four[3].get_suit():
            del self.face_up[-3:-1]
            return True

        return False

    def remove_all_possible(self):
        removed = True
        while removed and len(self.face_up) >= 4:
            removed = self.remove_four_same_suit()
            if not removed:
                removed = self.remove_inner_two()

    def playGame(self) -> bool:
        self.deck.shuffle()
        self.face_up = []

        self.deal_until_four()

        while self.deck.number_of_cards() > 0:
            self.remove_all_possible()
            self.face_up.append(self.deck.deal())

        self.remove_all_possible()
        return len(self.face_up) == 0
    
if __name__ == "__main__":
    game = Solitaire()
    print("win?", game.playGame())