From 2478d14955024129dea6f9b5344a35a75bf979c4 Mon Sep 17 00:00:00 2001
From: LmeSzinc <37934724+LmeSzinc@users.noreply.github.com>
Date: Wed, 14 Aug 2024 22:49:59 +0800
Subject: [PATCH] Add: Daemon to do trashbin minigame

---
 tasks/minigame/trashbin.py | 58 ++++++++++++++++++++++++++++++++++++++
 1 file changed, 58 insertions(+)
 create mode 100644 tasks/minigame/trashbin.py

diff --git a/tasks/minigame/trashbin.py b/tasks/minigame/trashbin.py
new file mode 100644
index 000000000..002ecde74
--- /dev/null
+++ b/tasks/minigame/trashbin.py
@@ -0,0 +1,58 @@
+import argparse
+
+from module.base.button import ClickButton
+from module.base.timer import Timer
+from module.base.utils import color_similar, get_color
+from tasks.base.ui import UI
+
+TAKE_PHOTO = ClickButton((1101, 332, 1162, 387), name='TAKE_PHOTO')
+
+
+class TrashBin(UI):
+    def is_camera_active(self):
+        color = get_color(self.device.image, (568, 358, 588, 362))
+        if color_similar(color, (23, 254, 180), threshold=20):
+            return True
+        if color_similar(color, (225, 214, 124), threshold=30):
+            return True
+        return False
+
+    def is_in_camera(self):
+        # Green icon
+        if self.image_color_count(TAKE_PHOTO, color=(134, 209, 187), threshold=221, count=200):
+            # White background
+            if self.image_color_count(TAKE_PHOTO, color=(235, 233, 237), threshold=221, count=1000):
+                return True
+        return False
+
+    def run(self):
+        self.device.disable_stuck_detection()
+        self.device.screenshot_interval_set(0.05)
+        _ = self.device.maatouch_builder
+        interval = Timer(1)
+        while 1:
+            self.device.screenshot()
+
+            if interval.reached():
+                if self.is_in_camera() and self.is_camera_active():
+                    self.device.click(TAKE_PHOTO)
+                    interval.reset()
+
+
+if __name__ == '__main__':
+    """
+    Do minigame in 2.4 (2024.08), taking photos of the trashbins
+    
+    To run this file:
+        python -m tasks.minigame.trashbin
+    Or:
+        python -m tasks.minigame.trashbin <instance>
+    """
+    parser = argparse.ArgumentParser()
+    parser.add_argument('instance', nargs='?', default='src', help='SRC instance name')
+
+    args = parser.parse_args()
+    instance = args.instance
+
+    src = TrashBin(instance)
+    src.run()