from module.combat.assets import EXP_INFO_C, EXP_INFO_D, COMBAT_AUTO_SWITCH, COMBAT_AUTO_133, COMBAT_AUTO_150
from module.daemon.daemon_base import DaemonBase
from module.exception import CampaignEnd
from module.logger import logger
from module.os.config import OSConfig
from module.os.fleet import OSFleet
from module.os_combat.combat import ContinuousCombat
from module.os_handler.assets import AUTO_SEARCH_REWARD
from module.os_handler.port import PORT_ENTER, PortHandler


class AzurLaneDaemon(DaemonBase, OSFleet, PortHandler):
    def _os_combat_expected_end(self):
        if self.appear_then_click(AUTO_SEARCH_REWARD, offset=(50, 50), interval=2):
            return False

        return super()._os_combat_expected_end()

    def run(self):
        self.handle_cheats()
        self.config.merge(OSConfig())
        self.config.override(HOMO_EDGE_DETECT=False)
        while 1:
            self.device.screenshot()

            # If is running a combat, do nothing.
            if self.is_combat_executing():
                self.device.screenshot()
                if (self.config.OpsiDaemon_FleetMode == 'auto_search' and
                    (self.appear(COMBAT_AUTO_150, offset=(50, 50)) or
                     self.appear_then_click(COMBAT_AUTO_133, offset=(50, 50))
                    )
                ):
                    self.device.click(COMBAT_AUTO_SWITCH)
                    continue
                elif self.appear_then_click(COMBAT_AUTO_SWITCH, offset=(50, 50)):
                    continue

            # Combat
            if self.combat_appear():
                self.combat_preparation(auto=self.config.OpsiDaemon_FleetMode)
            try:
                if self.handle_battle_status():
                    self.combat_status(expected_end='no_searching')
                    continue
            except (CampaignEnd, ContinuousCombat):
                continue
            if self.appear_then_click(EXP_INFO_C, interval=2):
                continue
            if self.appear_then_click(EXP_INFO_D, interval=2):
                continue

            # Map events
            if self.handle_map_event():
                self._nearest_object_click_timer.clear()
                continue
            if self.appear_then_click(AUTO_SEARCH_REWARD, offset=(50, 50), interval=2):
                continue

            # Port repair
            if self.config.OpsiDaemon_RepairShip:
                if self.appear(PORT_ENTER, offset=(20, 20), interval=30):
                    self.port_enter()
                    self.port_dock_repair()
                    self.port_quit()
                    self.interval_reset(PORT_ENTER)
                    logger.info('Port repair finished, '
                                'please move your fleet out of the port in 30s to avoid repairing again')

            if self.config.OpsiDaemon_SelectEnemy:
                if self.click_nearest_object():
                    continue

            # End
            # No end condition, stop it manually.

        return True


if __name__ == '__main__':
    b = AzurLaneDaemon('alas', task='OpsiDaemon')
    b.run()