from json import loads
from time import sleep

from pymem import Pymem
from requests import get
from win32api import GetAsyncKeyState


class Hack():
    def __init__(self) -> None:
        # Loading offsets
        self.offsets = self._find_offsets()

        self.cheats_list = [func for func in dir(self)
                            # Function
                            if callable(getattr(self, func))
                            # User defined
                            if not (func.startswith("__") and func.endswith("__"))
                            # Blacklist
                            if func not in ["_find_offsets",
                                            "_find_process",
                                            "find_module"]]

        self.pm = self._find_process(True)

    def _find_offsets(self) -> dict[str, int]:
        hazedumper_data = get(
            "https://raw.githubusercontent.com/frk1/hazedumper/master/csgo.min.json")
        serial_data = loads(hazedumper_data.text)
        return serial_data["signatures"] | serial_data["netvars"]

    def _find_process(self, verbose: bool = False) -> Pymem:
        """Find game process"""
        process_found = False
        print("Looking for process...") if verbose else None

        pm = None
        while not process_found:
            try:
                pm = Pymem("csgo.exe")
            except:
                # Timeout
                sleep(.5)
            else:
                print("Process found!") if verbose else None
                process_found = True

        if pm:
            return pm
        exit(1)

    def find_module(self, module: str):
        found = None
        for internal_module in list(self.pm.list_modules()):
            if internal_module.name == module + ".dll":
                found = internal_module.lpBaseOfDll

        if found:
            return found
        else:
            raise MemoryError

    def bhop(self) -> None:
        # Aliases
        mem = self.pm
        offset = self.offsets

        # Get module address
        client = self.find_module("client")

        # Hack loop
        while True:
            # Reduce CPU usage
            sleep(0.01)

            # Space bar detection
            if not GetAsyncKeyState(ord(" ")):
                continue

            # Get local player
            local_player = mem.read_uint(client + offset["dwLocalPlayer"])
            if not local_player:
                continue

            # Check if player is alive
            if not mem.read_int(local_player + offset["m_iHealth"]):
                continue

            # Check if player on ground
            if mem.read_uint(local_player + offset["m_fFlags"]) & (1 << 0):
                mem.write_uint(client + offset["dwForceJump"], 5)
                sleep(0.01)
                mem.write_uint(client + offset["dwForceJump"], 4)


if __name__ == "__main__":
    # Cheat
    c = Hack()

    # Cheat list
    print("Enter 0 to exit.")
    print("Available cheats:")
    for idx, cheat in enumerate(c.cheats_list):
        print(f"#{idx + 1} - {cheat}")

    # Select cheat
    c_id = None
    while c_id == None:
        try:
            match int(input("Enter ID: #")):
                case 0:
                    exit(0)
                case i if i > len(c.cheats_list):
                    raise IndexError
                case _ as i:
                    c_id = i - 1
        except:
            print("Invalid ID.")
            pass

    # Run cheat
    getattr(c, c.cheats_list[c_id])()