From 8bec814b8d65448e2500392a691ab0643bcc4396 Mon Sep 17 00:00:00 2001
From: YoursFunny <admin@yoursfunny.top>
Date: Mon, 20 Nov 2023 16:25:06 +0800
Subject: [PATCH] feat: support bounty

---
 tasks/bounty/bounty.py | 106 +++++++++++++++++++++++++++++++++++++++++
 tasks/bounty/ui.py     |  48 +++++++++++++++++++
 2 files changed, 154 insertions(+)
 create mode 100644 tasks/bounty/bounty.py
 create mode 100644 tasks/bounty/ui.py

diff --git a/tasks/bounty/bounty.py b/tasks/bounty/bounty.py
new file mode 100644
index 0000000..9e85db9
--- /dev/null
+++ b/tasks/bounty/bounty.py
@@ -0,0 +1,106 @@
+from enum import Flag
+
+from module.base.timer import Timer
+from module.exception import RequestHumanTakeover
+from module.logger import logger
+from tasks.base.assets.assets_base_page import BACK
+from tasks.base.page import page_bounty
+from tasks.bounty.assets.assets_bounty import *
+from tasks.bounty.ui import BountyUI
+
+
+class BountyStatus(Flag):
+    OCR = 0
+    SELECT = 1
+    ENTER = 2
+    SWEEP = 3
+    END = 4
+    FINISH = 5
+
+
+class Bounty(BountyUI):
+    @property
+    def bounty_info(self):
+        bounty = (SELECT_HIGHWAY, SELECT_DESERT_RAILROAD, SELECT_SCHOOLHOUSE)
+        check = (CHECK_HIGHWAY, CHECK_DESERT_RAILROAD, CHECK_SCHOOLHOUSE)
+        stage = (self.config.Highway_Stage, self.config.DesertRailroad_Stage, self.config.Schoolhouse_Stage)
+        count = (self.config.Highway_Count, self.config.DesertRailroad_Count, self.config.Schoolhouse_Count)
+        info = zip(bounty, check, stage, count)
+        return filter(lambda x: x[3] > 0, info)
+
+    @property
+    def valid_task(self) -> list:
+        task = list(self.bounty_info)
+        if not task:
+            logger.warning('Bounty enabled but no task set')
+            self.config.task_delay(server_update=True)
+            self.config.task_stop()
+        return task
+
+    @property
+    def is_ticket_enough(self) -> bool:
+        return self.config.stored.BountyTicket.value >= self.current_count
+
+    @property
+    def current_bounty(self):
+        return self.task[0][:2]
+
+    @property
+    def current_stage(self):
+        return self.task[0][2]
+
+    @property
+    def current_count(self):
+        return self.task[0][3]
+
+    def handle_bounty(self, status):
+        match status:
+            case BountyStatus.OCR:
+                if self.get_ticket():
+                    return BountyStatus.SELECT
+            case BountyStatus.SELECT:
+                if not self.is_ticket_enough:
+                    logger.warning('Bounty ticket not enough')
+                    raise RequestHumanTakeover
+                if self.select_bounty(*self.current_bounty):
+                    return BountyStatus.ENTER
+            case BountyStatus.ENTER:
+                if self.enter_stage(self.current_stage):
+                    return BountyStatus.SWEEP
+            case BountyStatus.SWEEP:
+                if self.do_sweep(self.current_count):
+                    self.task.pop(0)
+                    if not self.task:
+                        return BountyStatus.FINISH
+                    return BountyStatus.END
+                return BountyStatus.ENTER
+            case BountyStatus.END:
+                if self.appear(CHECK_BOUNTY):
+                    return BountyStatus.OCR
+                self.click_with_interval(BACK, interval=2)
+            case BountyStatus.FINISH:
+                return status
+            case _:
+                logger.warning(f'Invalid status: {status}')
+        return status
+
+    def run(self):
+        self.ui_ensure(page_bounty)
+        self.task = self.valid_task
+        action_timer = Timer(0.5, 1)
+        status = BountyStatus.OCR
+
+        while 1:
+            self.device.screenshot()
+
+            if self.ui_additional():
+                continue
+
+            if action_timer.reached_and_reset():
+                logger.attr('Status', status)
+                status = self.handle_bounty(status)
+
+            if status == BountyStatus.FINISH:
+                break
+
+        self.config.task_delay(server_update=True)
diff --git a/tasks/bounty/ui.py b/tasks/bounty/ui.py
new file mode 100644
index 0000000..afa0a40
--- /dev/null
+++ b/tasks/bounty/ui.py
@@ -0,0 +1,48 @@
+from module.base.timer import Timer
+from module.logger import logger
+from module.ocr.ocr import DigitCounter
+from tasks.base.ui import UI
+from tasks.bounty.assets.assets_bounty import *
+from tasks.stage.list import StageList
+from tasks.stage.sweep import StageSweep
+
+BOUNTY_LIST = StageList('BountyList')
+BOUNTY_SWEEP = StageSweep('BountySweep', 6)
+
+
+class BountyUI(UI):
+    def select_bounty(self, dest_enter: ButtonWrapper, dest_check: ButtonWrapper):
+        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
+
+    def enter_stage(self, index: int) -> bool:
+        if BOUNTY_LIST.select_index_enter(index, self):
+            return True
+        return False
+
+    def do_sweep(self, num: int) -> bool:
+        if BOUNTY_SWEEP.do_sweep(self, num=num):
+            return True
+        return False
+
+    def get_ticket(self):
+        """
+        Page:
+            in: page_bounty
+        """
+        if not self.appear(CHECK_BOUNTY):
+            logger.warning('OCR failed due to invalid page')
+            return False
+        ticket, _, total = DigitCounter(OCR_TICKET).ocr_single_line(self.device.image)
+        if total == 0:
+            logger.warning('Invalid ticket')
+            return False
+        logger.attr('BountyTicket', ticket)
+        self.config.stored.BountyTicket.set(ticket)
+        return True