from module.logger import logger
from module.ocr.ocr import DigitCounter
from module.ui.switch import Switch
from tasks.base.ui import UI
from tasks.tactical_challenge.assets.assets_tactical_challenge import *

SWITCH_SKIP = Switch('Skip_switch')
SWITCH_SKIP.add_state('on', SKIP_ON)
SWITCH_SKIP.add_state('off', SKIP_OFF)


class TacticalChallengeUI(UI):
    def get_ticket(self):
        """
        Page:
            in: page_tactical_challenge
        """
        ocr = DigitCounter(OCR_TICKET).ocr_single_line(self.device.image)
        # number of tickets remaining
        ticket, _, total = ocr
        if total == 0:
            logger.warning('Invalid ticket')
            return False, 5
        logger.attr('Ticket', ticket)

        return True, ticket

    def get_reward(self):
        if self.match_color(GET_REWARD_DAILY):
            self.device.click(GET_REWARD_DAILY)
            logger.info('Get tc daily reward')
            return True
        if self.match_color(GET_REWARD_CREDIT):
            self.device.click(GET_REWARD_CREDIT)
            logger.info('Get tc credit reward')
            return True
        if self.match_color(GOT_REWARD_DAILY) and self.match_color(GOT_REWARD_CREDIT):
            logger.info('Both tc reward got')
            return True

        return False

    def set_skip(self):
        """
        Set skip switch to on
        Returns:
            True if switch is set, False if switch not found
        """
        if not SWITCH_SKIP.appear(main=self):
            logger.info('Skip switch not found')
            return False
        SWITCH_SKIP.set('on', main=self)

        return True