feat: add offer reward claim workflow

automation/app.py

@@ -0,0 +1,170 @@
+import json
+import queue
+import threading
+import tkinter as tk
+from dataclasses import asdict
+from pathlib import Path
+from tkinter import filedialog, messagebox, ttk
+
+from PIL import Image, ImageTk
+
+from .engine import Desktop, Matcher, Region, Runner, load_flow
+
+ROOT = Path(__file__).resolve().parent.parent
+LOCAL = ROOT / 'local'
+
+
+class Selector(tk.Toplevel):
+    def __init__(self, parent, screenshot, on_select):
+        super().__init__(parent)
+        self.title('拖动框选游戏内容：不包含微信标题栏和窗口边框')
+        self.original = screenshot
+        self.on_select = on_select
+        ratio = min(1000 / screenshot.width, 700 / screenshot.height, 1)
+        self.preview_size = (round(screenshot.width * ratio), round(screenshot.height * ratio))
+        self.photo = ImageTk.PhotoImage(screenshot.resize(self.preview_size))
+        self.canvas = tk.Canvas(self, width=self.preview_size[0], height=self.preview_size[1], highlightthickness=0)
+        self.canvas.pack()
+        self.canvas.create_image(0, 0, image=self.photo, anchor='nw')
+        self.start = None
+        self.rect = None
+        self.canvas.bind('<ButtonPress-1>', self.press)
+        self.canvas.bind('<B1-Motion>', self.drag)
+        self.canvas.bind('<ButtonRelease-1>', self.release)
+
+    def press(self, event):
+        self.start = (event.x, event.y)
+        if self.rect:
+            self.canvas.delete(self.rect)
+        self.rect = self.canvas.create_rectangle(event.x, event.y, event.x, event.y, outline='#ff3333', width=3)
+
+    def drag(self, event):
+        if self.start:
+            self.canvas.coords(self.rect, *self.start, event.x, event.y)
+
+    def release(self, event):
+        if not self.start:
+            return
+        x1, x2 = sorted((self.start[0], event.x))
+        y1, y2 = sorted((self.start[1], event.y))
+        sx = self.original.width / self.preview_size[0]
+        sy = self.original.height / self.preview_size[1]
+        region = Region(round(x1 * sx), round(y1 * sy), round((x2 - x1) * sx), round((y2 - y1) * sy))
+        try:
+            region.validate(self.original.size)
+        except Exception as exc:
+            messagebox.showerror('重新框选', str(exc), parent=self)
+            return
+        self.on_select(region)
+        self.destroy()
+
+
+class App(tk.Tk):
+    def __init__(self):
+        super().__init__()
+        self.title('蔚蓝星球 · 本地自动化')
+        self.geometry('720x520')
+        self.region = None
+        self.events = queue.Queue()
+        self.stop = threading.Event()
+        self.busy = False
+        self.flow = load_flow(ROOT / 'workflows' / 'pig_claim.json')
+        self.matcher = Matcher(ROOT / 'assets')
+        self.status = tk.StringVar(value='请先将游戏放在主屏幕，打开首页并校准游戏内容区域。')
+        ttk.Label(self, text='第一条流程：小金猪挂机奖励领取', font=('', 18)).pack(pady=16)
+        ttk.Label(self, text='开始前请保持游戏窗口可见且位置不变。鼠标移到主屏幕左上角可紧急停止。').pack()
+        bar = ttk.Frame(self)
+        bar.pack(pady=14)
+        self.buttons = []
+        for name, cmd in [('校准游戏区域', self.capture), ('导入截图校准', self.import_image), ('预览流程', lambda: self.launch(True)), ('执行领取', lambda: self.launch(False))]:
+            button = ttk.Button(bar, text=name, command=cmd)
+            button.pack(side='left', padx=4)
+            self.buttons.append(button)
+        ttk.Button(self, text='停止', command=self.stop.set).pack()
+        ttk.Label(self, textvariable=self.status, wraplength=680).pack(pady=10)
+        self.output = tk.Text(self, height=16, wrap='word', state='disabled')
+        self.output.pack(fill='both', expand=True, padx=16, pady=12)
+        self.protocol('WM_DELETE_WINDOW', self.close)
+        config = LOCAL / 'region.json'
+        if config.exists():
+            try:
+                self.region = Region(**json.loads(config.read_text()))
+                self.status.set('已加载游戏区域；移动游戏窗口或更换屏幕后请重新校准。')
+            except (ValueError, TypeError):
+                pass
+        self.after(100, self.poll)
+
+    def log(self, text):
+        self.events.put(('log', text))
+
+    def set_region(self, region):
+        LOCAL.mkdir(exist_ok=True)
+        (LOCAL / 'region.json').write_text(json.dumps(asdict(region)), encoding='utf-8')
+        self.region = region
+        self.status.set('校准完成。先预览，确认首页识别通过后再执行。')
+
+    def capture(self):
+        self.iconify()
+        self.after(1500, self.take_capture)
+
+    def take_capture(self):
+        try:
+            screenshot = Desktop().screenshot()
+        except Exception as exc:
+            self.deiconify()
+            messagebox.showerror('截图失败', '请检查系统屏幕录制权限。\n' + str(exc))
+            return
+        self.deiconify()
+        Selector(self, screenshot, self.set_region)
+
+    def import_image(self):
+        name = filedialog.askopenfilename(filetypes=[('主屏幕完整截图', '*.png *.jpg *.jpeg')])
+        if name:
+            with Image.open(name) as image:
+                Selector(self, image.convert('RGB').copy(), self.set_region)
+
+    def launch(self, dry_run):
+        if self.busy:
+            return
+        if self.region is None:
+            messagebox.showinfo('需要校准', '请先校准游戏内容区域。')
+            return
+        self.busy = True
+        self.stop.clear()
+        for button in self.buttons:
+            button.configure(state='disabled')
+        self.status.set('正在预览…' if dry_run else '3 秒后执行，请将游戏放到前台。')
+        region = self.region
+        def work():
+            try:
+                if self.stop.wait(3):
+                    return
+                Runner(Desktop(), self.matcher, region, self.log, self.stop).run(self.flow, dry_run)
+            except Exception as exc:
+                self.log('流程停止：' + str(exc))
+            finally:
+                self.events.put(('done', None))
+        threading.Thread(target=work, daemon=True).start()
+
+    def poll(self):
+        while not self.events.empty():
+            kind, value = self.events.get_nowait()
+            if kind == 'log':
+                self.output.configure(state='normal')
+                self.output.insert('end', value + '\n')
+                self.output.see('end')
+                self.output.configure(state='disabled')
+            elif kind == 'done':
+                self.busy = False
+                for button in self.buttons:
+                    button.configure(state='normal')
+                self.status.set('本轮已结束，请查看日志中的实际结果。')
+        self.after(100, self.poll)
+
+    def close(self):
+        self.stop.set()
+        self.destroy()
+
+
+def main():
+    App().mainloop()