from module.campaign.assets import SWITCH_1_HARD_ALCHEMIST
from module.campaign.campaign_base import CampaignBase as CampaignBase_
from module.campaign.campaign_ui import MODE_SWITCH_1
from module.handler.fast_forward import auto_search
from module.logger import logger
from module.map.map_grids import SelectedGrids

for state in MODE_SWITCH_1.status_list:
    if state['status'] == 'hard':
        state['check_button'] = SWITCH_1_HARD_ALCHEMIST
        state['click_button'] = SWITCH_1_HARD_ALCHEMIST


class CampaignBase(CampaignBase_):
    STAGE_INCREASE = [
        'T1 > T2 > T3 > TS1 > T4 > T5',
        'TH1 > TH2 > TH3 > TH4 > TH5',
    ]

    def campaign_set_chapter_event(self, chapter, mode='normal'):
        if chapter.startswith('t'):
            self.ui_goto_event()
            self.campaign_ensure_chapter(index=chapter)
            return True

        return super().campaign_set_chapter_event(chapter, mode=mode)

    @staticmethod
    def _campaign_separate_name(name):
        # T, TH, ASP, EX
        if name == 'ex':
            return 't4', '1'
        if name == 'asp':
            return 't3', '1'
        if name == 'sp':
            return 't3', '1'
        if name == 'ts1':
            return 't1', name[-1]
        if name.startswith('th'):
            return 't2', name[-1]
        if name.startswith('t'):
            return 't1', name[-1]

        return CampaignBase_._campaign_separate_name(name)

    @staticmethod
    def _campaign_get_chapter_index(name):
        if name == 't4':
            return 4
        if name == 't3':
            return 3
        if name == 't2':
            return 2
        if name == 't1':
            return 1

        return CampaignBase_._campaign_get_chapter_index(name)

    def campaign_get_entrance(self, name):
        if name == 'sp':
            name = 'asp'
        return super().campaign_get_entrance(name)

    def map_get_info(self):
        name = str(self.config.Campaign_Name).lower()
        super().map_get_info()

        # Chapter TH has no map_percentage and no 3_stars
        if name.startswith('th') or name.startswith('ht'):
            appear = auto_search.appear(main=self)
            self.map_is_100_percent_clear = self.map_is_3_stars = self.map_is_threat_safe = appear
            self.map_has_clear_mode = appear
            self.map_show_info()

    def clear_map_items(self, grids):
        """

        Args:
            grids (GridInfo, list[GridInfo]): Grid object or a list of them

        Returns:

        """
        if not isinstance(grids, list):
            grids = [grids]
        grids = SelectedGrids(grids).sort('cost')
        for grid in grids:
            logger.hr('Clear map item')
            logger.info(f'Clear map item on {grid}')
            self.goto(grid)