StarRailCopilot/tasks/dungeon/weekly.py

from module.config.utils import get_server_next_monday_update
from module.logger import logger
from module.ocr.ocr import DigitCounter
from tasks.daily.keywords import KEYWORDS_DAILY_QUEST
from tasks.dungeon.assets.assets_dungeon_ui import OCR_DUNGEON_LIST, OCR_WEEKLY_LIMIT
from tasks.dungeon.dungeon import Dungeon
from tasks.dungeon.keywords import DungeonList, KEYWORDS_DUNGEON_NAV, KEYWORDS_DUNGEON_TAB
from tasks.dungeon.ui import DUNGEON_LIST


class OcrWeeklyLimit(DigitCounter):
    def format_result(self, result) -> tuple[int, int, int]:
        current, remain, total = super().format_result(result)
        # [OCR_WEEKLY_LIMIT format] ３3/3 -> (33, -30, 3)
        if current == 11:
            current = 1
            remain = total - current
        if current == 22:
            current = 2
            remain = total - current
        if current == 33:
            current = 3
            remain = total - current

        return current, remain, total


class WeeklyDungeon(Dungeon):
    def require_compulsory_support(self) -> bool:
        return False

    def _dungeon_run(self, dungeon: DungeonList, team: int = None, wave_limit: int = 0, support_character: str = None,
                     skip_ui_switch: bool = False):
        if team is None:
            team = self.config.Weekly_Team

        skip_ui_switch = True
        return super()._dungeon_run(
            dungeon=dungeon, team=team, wave_limit=wave_limit,
            support_character=support_character, skip_ui_switch=skip_ui_switch)

    def handle_ascension_dungeon_prepare(self):
        # combat_wave_cost==30 in weekly, but no handle_ascension_dungeon_prepare required
        return False

    def get_weekly_remain(self) -> int:
        """
        Pages:
            in: page_guide, Survival_Index, KEYWORDS_DUNGEON_NAV.Echo_of_War
        """
        ocr = OcrWeeklyLimit(OCR_WEEKLY_LIMIT)
        current, _, _ = ocr.ocr_single_line(self.device.image)
        total = self.config.stored.EchoOfWar.FIXED_TOTAL
        if current <= total:
            logger.attr('EchoOfWar', f'{current}/{total}')
            self.config.stored.EchoOfWar.value = current
            return current
        else:
            logger.warning(f'Invalid EchoOfWar limit: {current}/{total}')
            return 0

    def run(self):
        # self.config.update_battle_pass_quests()
        self.config.update_daily_quests()
        self.called_daily_support = False
        self.achieved_daily_quest = False
        self.achieved_weekly_quest = False
        self.daily_quests = self.config.stored.DailyQuest.load_quests()
        self.weekly_quests = self.config.stored.BattlePassWeeklyQuest.load_quests()

        dungeon = DungeonList.find(self.config.Weekly_Name)
        planner = self.planner.get_weekly()
        if planner is not None:
            dungeon = planner
            self.is_doing_planner = True
        logger.attr('DungeonWeekly', dungeon)

        # UI switches
        self.dungeon_tab_goto(KEYWORDS_DUNGEON_TAB.Survival_Index)
        # Equivalent to self.dungeon_goto(dungeon), but check limit remains
        DUNGEON_LIST.search_button = OCR_DUNGEON_LIST
        self._dungeon_nav_goto(KEYWORDS_DUNGEON_NAV.Echo_of_War)
        self._dungeon_wait_until_dungeon_list_loaded()
        monday = get_server_next_monday_update(self.config.Scheduler_ServerUpdate)

        # Check limit
        remain = self.get_weekly_remain()
        if remain <= 0:
            if KEYWORDS_DAILY_QUEST.Complete_Echo_of_War_1_times in self.daily_quests:
                logger.info('Reached the limit to get Echo_of_War rewards, continue cause daily quests require it')
                remain = 1
            else:
                logger.info('Reached the limit to get Echo_of_War rewards, stop')
                self.config.task_delay(target=monday)
                self.config.task_stop()

        self._dungeon_insight(dungeon)
        self._dungeon_enter(dungeon)

        # Combat
        count = self.dungeon_run(dungeon, wave_limit=min(remain, 3))
        self.is_doing_planner = False

        logger.attr('achieved_daily_quest', self.achieved_daily_quest)
        logger.attr('achieved_weekly_quest', self.achieved_weekly_quest)
        with self.config.multi_set():
            if count:
                # Check battle pass
                if self.achieved_weekly_quest:
                    self.config.task_call('BattlePass')
                # Check daily
                if self.achieved_daily_quest:
                    self.config.task_call('DailyQuest')
                # Update dashboard
                self.config.stored.EchoOfWar.add(-count)
                if self.config.stored.EchoOfWar.value < 0:
                    logger.warning(f'Invalid EchoOfWar value, revise to 0')
                    self.config.stored.EchoOfWar.value = 0
            # Finished all remains
            if count >= remain:
                logger.info('All Echo_of_War rewards got')
                self.config.task_delay(target=monday)
                self.config.task_stop()

            logger.warning(f'Unexpected Echo_of_War case, count={count}, remain={remain}')
            self.config.task_delay(server_update=True)
            self.config.task_stop()