import uiautomator2 as u2
import time


class Automator:
    def __init__(self, device: str):
        """
        device: 如果是 USB 连接，则为 adb devices 的返回结果；如果是模拟器，则为模拟器的控制 URL 。
        """
        self.d = u2.connect(device)
    def _autoupdate(self,indx):
        self.d.click(self._get_position(11)[0], self._get_position(11)[1])
        time.sleep(1)
        self.d.click(self._get_position(indx)[0], self._get_position(indx)[1])
        time.sleep(1)
        self.d.click(self._get_position(10)[0], self._get_position(10)[1]) #升级
        time.sleep(.5)
        self.d.click(self._get_position(11)[0], self._get_position(11)[1])

    def _match(self):

        for i in range(13,16):
            for k in range(4):
                for j in range(1,10):
                    xp = self._get_position(j)[0]
                    yp = self._get_position(j)[1]
                    self.d.swipe(self._get_position(i)[0],self._get_position(i)[1],xp,yp)

    def start(self):
        """
        启动脚本，请确保已进入游戏页面。
        """
        while True:

            self._match()


            # 简单粗暴的方式，处理 “XX之光” 的荣誉显示。
            # 当然，也可以使用图像探测的模式。
            self.d.click(self._get_position(12)[0], self._get_position(12)[1])  #防止错误
            time.sleep(.5)
            self.d.click(550, 1650)
            k = [7,8,9]  #要升级建筑的标号
            for i in k:
                self._autoupdate(i)
                time.sleep(1)
            # 滑动屏幕，收割金币。
            self._swipe()

    def _swipe(self):
        """
        滑动屏幕，收割金币。
        """
        for i in range(3):
            # 横向滑动，共 3 次。
            sx, sy = self._get_position(i * 3 + 1)
            ex, ey = self._get_position(i * 3 + 3)
            self.d.swipe(sx, sy, ex, ey)

    @staticmethod
    def _get_position(key):
        """
        获取指定建筑的屏幕位置。
        """
        positions = {
            1: (294, 1184),
            2: (551, 1061),
            3: (807, 961),
            4: (275, 935),
            5: (535, 810),
            6: (799, 687),
            7: (304, 681),
            8: (541, 568),
            9: (787, 447),
            10: (857, 1749),
            11: (977, 1145),
            12: (193,1809),
            13: (675,1645),#货物1位置
            14:(830,1571), #货物2位置
            15:(971,1477) #货物3位置
        }
        return positions.get(key)

    # def _get_target_position(self, target: TargetType):
    #     """
    #     获取货物要移动到的屏幕位置。
    #     """
    #     return self._get_position(self.targets.get(target))

    # def _match_target(self, target: TargetType):
    #     """
    #     探测货物，并搬运货物。
    #     """
    #     # 获取当前屏幕快照
    #     screen = self.d.screenshot(format="opencv")
    #
    #     # 由于 OpenCV 的模板匹配有时会智障，故我们探测次数实现冗余。
    #     counter = 8
    #     while counter != 0:
    #         counter = counter - 1
    #
    #         # 使用 OpenCV 探测货物。
    #         result = UIMatcher.match(screen, target)
    #
    #         # 若无探测到，终止对该货物的探测。
    #         # 实现冗余的原因：返回的货物屏幕位置与实际位置存在偏差，导致移动失效
    #         if result is None:
    #             break
    #
    #         sx, sy = result
    #         # 获取货物目的地的屏幕位置。
    #         ex, ey = self._get_target_position(target)
    #
    #         # 搬运货物。
    #         self.d.swipe(sx, sy, ex, ey)