from module.base.base import ModuleBase
from module.base.timer import Timer
from module.base.utils import color_bar_percentage
from module.logger import logger


ATTACKER_HP_AREA = (271, 43, 586, 58)
DEFENDER_HP_AREA = (691, 43, 1005, 58)


class HpDaemon(ModuleBase):
    attacker_hp = 1.0
    defender_hp = 1.0
    # _last_secure_time = 0
    low_hp_confirm_timer: Timer

    @staticmethod
    def _calculate_hp(image, area, reverse=False, starter=2, prev_color=(239, 32, 33)):
        """
        Args:
            image:
            area:
            reverse: True if HP is left align.
            starter:
            prev_color:

        Returns:
            float: HP. 0 to 1.
        """
        # bar = np.array(image.crop(area))
        # length = bar.shape[1]
        # bar = np.swapaxes(bar, 0, 1)
        # bar = bar[::-1, :, :] if reverse else bar
        # prev_index = 0
        # for index, color in enumerate(bar):
        #     if index < starter:
        #         continue
        #     mask = color_similar_1d(color, prev_color, threshold=30)
        #     if np.any(mask):
        #         prev_color = color[mask].mean(axis=0)
        #         prev_index = index
        #
        # return prev_index / length
        return color_bar_percentage(image, area, prev_color=prev_color, starter=starter, reverse=reverse)

    def _show_hp(self, low_hp_time=0):
        """
        Examples:
            [ 80% - 70%]
            [ 80% - 70%]
            [ 80% - 70%] - Low HP: 3.154s
        """
        text = '[%s - %s]' % (
            str(int(self.attacker_hp * 100)).rjust(2, '0') + '%',
            str(int(self.defender_hp * 100)).rjust(2, '0') + '%')
        if low_hp_time:
            text += ' - Low HP: %ss' % str(round(low_hp_time, 3)).ljust(5, '0')
        logger.info(text)

    def _at_low_hp(self, image):
        self.attacker_hp = self._calculate_hp(image, area=ATTACKER_HP_AREA, reverse=True)
        self.defender_hp = self._calculate_hp(image, area=DEFENDER_HP_AREA, reverse=False)
        if 0.01 < self.attacker_hp <= self.config.LOW_HP_THRESHOLD:
            if self.low_hp_confirm_timer.reached() and self.low_hp_confirm_timer.current() < 300:
                self._show_hp(self.low_hp_confirm_timer.current())
                return True
            else:
                return False
        else:
            self.low_hp_confirm_timer.reset()
            return False