import base64

from classes.Message import Message


class ShuffleMessage(Message):
	def __init__(self, name: str, cards: list[bytes], stage: str):
		super().__init__()
		self.message_fields['type'] = 'shuffle'
		self.set_name(name)
		self.set_cards(cards)
		self.set_stage(stage)

	def set_name(self, name: str):
		self.message_fields['name'] = name

	def get_name(self) -> str:
		return self.message_fields['name']

	def set_cards(self, cards: list[bytes]):
		self.message_fields['cards'] = []
		for card in cards:
			self.message_fields['cards'].append(base64.b64encode(card).decode('utf-8'))

	def get_cards(self) -> list[bytes]:
		cards = []
		for card in self.message_fields['cards']:
			cards.append(base64.b64decode(card))
		return cards

	def set_stage(self, stage: str):
		self.message_fields['stage'] = stage

	def get_stage(self) -> str:
		return self.message_fields['stage']