feat: mission/commissions/event

tasks/mission/ui.py

@@ -0,0 +1,147 @@
+from module.base.timer import Timer
+from module.logger import logger
+from module.ui.switch import Switch
+from module.ocr.ocr import Digit
+from tasks.base.assets.assets_base_page import BACK, MISSION_CHECK
+from tasks.base.page import page_mission, page_commissions
+from tasks.base.ui import UI
+from tasks.mission.assets.assets_mission import *
+from tasks.stage.ap import AP
+from tasks.stage.mission_list import StageList
+from tasks.stage.sweep import StageSweep
+
+
+SHARED_LIST = StageList('SharedList')
+MISSION_SWEEP = StageSweep('MissionSweep', 60)
+MISSION_SWEEP.set_button(button_check=CHECK_MISSION_SWEEP) # Check sweep is different for mission
+SHARED_SWEEP = StageSweep('SharedSweep', 60)
+
+SWITCH_NORMAL = Switch("Normal_switch")
+SWITCH_NORMAL.add_state("on", NORMAL_ON)
+SWITCH_NORMAL.add_state("off", NORMAL_OFF)
+
+SWITCH_HARD = Switch("HARD_switch")
+SWITCH_HARD.add_state("on", HARD_ON)
+SWITCH_HARD.add_state("off", HARD_OFF)
+
+SWITCH_QUEST = None
+
+"""
+A dictionary that maps the mode to a tuple where 
+the first element is an argument to go_back and second is for ui_ensure
+Missing for "E" because there are no event in Global and no page_event
+"""
+MODE_TO_PAGE = {
+    "N": (MISSION_CHECK, page_mission),
+    "H": (MISSION_CHECK, page_mission),
+    "BD": (CHECK_BD, page_commissions),
+    "IR": (CHECK_IR, page_commissions),
+    "E" : ()
+}
+
+
+class MissionUI(UI, AP):
+    def select_mission(self, mode, stage):
+        area = int(stage.split("-")[0])
+        if not self.select_area(area):
+            logger.warning("Area not found")
+            return False
+
+        to_switch = {
+            "N": SWITCH_NORMAL,
+            "H": SWITCH_HARD
+        }
+        switch = to_switch[mode]
+        if not self.select_mode(switch) and not self.select_area(area):
+            return False
+        return True
+    
+    def select_area(self, num):
+        """"
+        May require further error handling for these cases. 
+        1. Fails to ocr area number
+        2. May trigger too many click exception when clicking left or right too many times
+        3. Area not unlocked. Simplest way if left or right button are still present 
+        but problem is it is expensive to check every time and they always keep moving.
+        """
+        tries = 0
+        ocr_area = Digit(OCR_AREA)
+        while 1:
+            try:
+                self.device.screenshot()
+                current_area = int(ocr_area.ocr_single_line(self.device.image))
+                if current_area == num:
+                    return True
+                elif current_area > num:
+                    [self.click_with_interval(LEFT, interval=1) for x in range(abs(current_area-num))]
+                elif current_area < num:
+                    [self.click_with_interval(RIGHT, interval=1) for x in range(abs(current_area-num))]
+            except:
+                tries += 1
+                if tries > 5:
+                    return False
+
+    def select_mode(self, switch):
+        """
+        Set switch to on. 
+        Returns:
+            True if switch is set, False if switch not found
+        """
+        if not switch.appear(main=self):
+            logger.info(f'{switch.name} not found')
+            return False
+        switch.set('on', main=self)
+        return True
+
+    def enter_stage(self, index: int) -> bool:
+        if not index:
+            index = SHARED_LIST.insight_max_sweepable_index(self)
+        if SHARED_LIST.select_index_enter(self, index):
+            return True
+        return False
+
+    def do_sweep(self, mode, num: int) -> bool:
+        if mode in ["N", "H"]:
+            return MISSION_SWEEP.do_sweep(self, num=num)
+        else:
+            return SHARED_SWEEP.do_sweep(self, num=num)
+
+    def navigate(self, prev, next):
+        """
+        go_back is called when the previous stage and next stage are in
+        the same game mode. 
+        For example, "N" and "H" are in Mission so we call go_back.
+        If different, ui_ensure is called for example, "N" and "IR".
+        """
+        if prev==next or (prev in ["N", "H"] and next in ["N", "H"]):
+            self.go_back(MODE_TO_PAGE[next][0])
+        elif prev in ["BD", "IR"] and next in ["BD", "IR"]:
+            self.go_back(CHECK_COMMISSIONS)
+        else:
+            self.ui_ensure(MODE_TO_PAGE[next][1])
+
+    def go_back(self, check):
+        while 1:
+            self.device.screenshot()
+            if self.match_color(check) and self.appear(check):
+                return True
+            self.click_with_interval(BACK, interval=2)
+
+class CommissionsUI(UI, AP):
+    """Works the same way as select_bounty"""
+    def select_commission(self, mode):
+        to_button = {
+            "IR": (SELECT_IR, CHECK_IR),
+            "BD": (SELECT_BD, CHECK_BD)
+        }
+        dest_enter, dest_check = to_button[mode]
+        timer = Timer(5, 10).start()
+        while 1:
+            self.device.screenshot()
+            self.appear_then_click(dest_enter, interval=1)
+            if self.appear(dest_check):
+                return True
+            if timer.reached():
+                return False
+
+