fishyboteso/fishy/__main__.py

"""Fishy

Usage:
  fishy.py -h | --help
  fishy.py -v | --version
  fishy.py [--debug] [--ip=<ipv4>] [--collect-r] [--borderless]

Options:
  -h, --help                Show this screen.
  -v, --version             Show version.
  --ip=<ipv4>               Local Ip Address of the android phone.
  --debug                   Start program in debug controls.
  --borderless              Use if the game is in fullscreen or borderless window
"""

import time

import cv2

from docopt import docopt
from pynput.keyboard import Listener

from fishy.systems import *

"""
Start reading from `init.py`
"""


def on_release(key):
    """
    Reads input, finds out the resultant action and performs it

    :param key: key pressed
    """

    c = Control.find(key)
    if c is None:
        return

    if c[0] == Control.Keywords.StartPause:

        if not G.pause:
            print("PAUSED")
            G.pause = True
            return

        if PixelLoc.config():
            print("STARTED")
            G.pause = False
        else:
            print("addon properly not installed, if it is installed try restarting the game.")

    elif c[0] == Control.Keywords.Debug:
        G.debug = not G.debug

    elif c[0] == Control.Keywords.Stop:
        G.stop = True

    elif c[0] == Control.Keywords.SwitchMode:
        Control.nextState()
        Log.ctrl()

    elif c[0] == Control.Keywords.ClearPrintOnce:
        Log.clearPrintIds()


def hsv2rgb(img):
    return cv2.cvtColor(img, cv2.COLOR_HSV2RGB)


def startFishing():
    """
    Starts the fishing
    code explained in comments in detail
    """

    Control.current = 1 if G.arguments["--debug"] else 0

    use_net = G.arguments["--ip"] is not None
    if use_net:
        net.initialize(G.arguments["--ip"])

    sleepFor = 1

    # initializes fishing modes and their callbacks
    FishingMode("hook", 0, HookEvent())
    FishingMode("stick", 1, StickEvent())
    FishingMode("look", 2, LookEvent())
    FishingMode("idle", 3, IdleEvent(use_net))

    Log.ctrl()

    fishPixWindow = Window(color=cv2.COLOR_RGB2HSV)

    # initialize widow
    Window.Init()
    with Listener(on_release=on_release):
        while not G.stop:
            # record the time to calculate time taken to execute one loop
            current_time = time.time()

            # Services to be ran in the start of the main loop
            Window.Loop()
            Log.Loop()

            # get the PixelLoc and find the color values, to give it to `FishingMode.Loop`
            fishPixWindow.crop = PixelLoc.val
            hueValue = fishPixWindow.getCapture()[0][0][0]
            FishingMode.Loop(hueValue, G.pause)

            # if debug is on, show the color on the PixelLoc in a window and print the hue values of it
            if G.debug:
                fishPixWindow.show("pixloc", resize=200, func=hsv2rgb)
                Log.ou(str(FishingMode.CurrentMode.label) + ":" + str(fishPixWindow.getCapture()[0][0]))

            # Services to be ran in the end of the main loop
            Log.LoopEnd()
            Window.LoopEnd()

            # calculate the time it took to execute one loop of code, if it is more than the expected time warn user
            frameTime = time.time() - current_time
            if frameTime < sleepFor:
                time.sleep(sleepFor - frameTime)
            else:
                Log.po(231, "Program taking more time than expected, this might slow your computer try increasing "
                            "\"--check-frequency\".")


def main():
    G.arguments = docopt(__doc__)
    if G.arguments["--version"]:
        quit()

    startFishing()


if __name__ == "__main__":
    main()