menu_options.py

from colorama import init, Fore, Back   # for colouring boolean options
init()  # initializes colorama library


def toggle_option(name):
    # There's absolutely a better way to do this but I can't be bothered coming up with it at the moment lol
    if name == MenuNames.FILTER: Menu.IS_FILTER_RUNNING = not Menu.IS_FILTER_RUNNING
    if name == MenuNames.LOG_BLOCKED_ACTIVITY: Menu.IS_LOGGING = not Menu.IS_LOGGING
    if name == MenuNames.DROP_INC_80:
        Menu.IS_DROP_INC_80 = not Menu.IS_DROP_INC_80
        if Menu.DROP_OPTIONS_MUTUALLY_EXCLUSIVE:
            Menu.IS_DROP_CLIENT_POST = False
            Menu.IS_DROP_LENGTH = False
    if name == MenuNames.DROP_CLIENT_POST:
        Menu.IS_DROP_CLIENT_POST = not Menu.IS_DROP_CLIENT_POST
        if Menu.DROP_OPTIONS_MUTUALLY_EXCLUSIVE:
            Menu.IS_DROP_INC_80 = False
            Menu.IS_DROP_LENGTH = False
    if name == MenuNames.DROP_LENGTHS:
        Menu.IS_DROP_LENGTH = not Menu.IS_DROP_LENGTH
        if Menu.DROP_OPTIONS_MUTUALLY_EXCLUSIVE:
            Menu.IS_DROP_INC_80 = False
            Menu.IS_DROP_CLIENT_POST = False


class MenuNames:
    FILTER = "FILTER:"
    ADJUST_FILTER = "ADJUST FILTER SETTINGS"
    LOG_BLOCKED_ACTIVITY = "LOG BLOCKED ACTIVITY:"
    OPEN_DONATION_URL = "Open Donation URL"
    EXIT_PROGRAM = "EXIT PROGRAM"
    DROP_INC_80 = "DROP INC 80:"
    DROP_CLIENT_POST = "DROP CLIENT POST [RECOMMENDED]:"
    DROP_LENGTHS = "DROP LENGTHS [EXPERIMENTAL]:"
    FILTER_SETTINGS_GO_BACK = "Go back"


class Menu:
    IS_FILTER_RUNNING = True
    START_STOP_STR = "Stop" if IS_FILTER_RUNNING else "Start"
    IS_LOGGING = False

    IS_DROP_INC_80 = False
    IS_DROP_CLIENT_POST = True
    IS_DROP_LENGTH = False

    DROP_OPTIONS_MUTUALLY_EXCLUSIVE = True  # Can't have more than one drop option enabled at a time.

    ERR_MSG_NO_FILTERS = "\n ERROR: Attempted to start the network filter but "+\
                         Back.RED+"no filters are enabled"+Back.RESET+".\n        "+\
                         Fore.LIGHTCYAN_EX+"Press 2"+Fore.RESET+" to open the filter settings and "+\
                         Fore.LIGHTCYAN_EX+"enable at least one"+Fore.RESET+" of them!"

    """
    Key is text to change option.
    """
    MAIN_OPTIONS = {"1": {"value": IS_FILTER_RUNNING,
                          "name": MenuNames.FILTER,
                          "desc": "Press 1 to "+START_STOP_STR+" the network filter.",
                          "action": toggle_option},
                    "2": {"value": "",
                          "name": MenuNames.ADJUST_FILTER,
                          "desc": "Change / update which filters you're currently using.",},
                    "3": {"value": IS_LOGGING,
                          "name": MenuNames.LOG_BLOCKED_ACTIVITY,
                          "desc": "Log information to the debug file about what was blocked and why.",
                          "action": toggle_option},
                    "4": {"value": "",
                          "name": MenuNames.OPEN_DONATION_URL,
                          "desc": "If this program helped you, donating would be a great way of saying thanks."},
                    "x": {"value": "",
                          "name": MenuNames.EXIT_PROGRAM,
                          "desc": "Safely exit the program."}}

    FILTER_OPTIONS = {"1": {"value": IS_DROP_INC_80,
                            "name": MenuNames.DROP_INC_80,
                            "desc": "In order to notify your client of an incoming notification, R* Servers\n"
                                    "will send you a specific packet. Dropping this packet will prevent your client\n"
                                    "from being made aware of new notifications entirely. This method is fast and\n"
                                    "simple but may conflict with in-game traffic, and you may be flooded with\n"
                                    "notifications when the filter is turned off.",
                                    "action": toggle_option},
                      "2": {"value": IS_DROP_CLIENT_POST,
                            "name": MenuNames.DROP_CLIENT_POST,
                            "desc": "If your client is informed that it has notifications to fetch\n"
                                    "(i.e. DROP INC 80 is OFF), your client will POST some data to a specific endpoint.\n"
                                    "Enabling this setting will drop packets that are sent to this endpoint.\n"
                                    "Much less likely to interfere with in-game traffic, but computationally\n"
                                    "expensive and will block all notifications.",
                                    "action": toggle_option},
                      "3": {"value": IS_DROP_LENGTH,
                            "name": MenuNames.DROP_LENGTHS,
                            "desc": "All payloads sent and received when playing GTA Online are encrypted, but\n"
                                    "uncompressed. This means that even if we can't see the packet's details, we can\n"
                                    "\"guess\" what type it is based on its' size. This method is surprisingly powerful\n"
                                    "and can allow some notifications through while still blocking session invites.\n"
                                    "However, this filter is a work in progress and may not behave correctly.",
                                    "action": toggle_option},
                      "b": {"value": "",
                            "name": MenuNames.FILTER_SETTINGS_GO_BACK,
                            "desc": "Save your changes and go back to the main menu."}}

    @staticmethod
    def construct_print_options(options):
        ret = ""                    # I'm aware of the O(n^2) complexity of this function. Please don't hurt me.
        for key in options:
            ret += " " + key + ") " + options[key]["name"] + " " + Menu.color_bool(options[key]["value"]) + "\n" +\
                   "\t" + options[key]["desc"].replace("\n", "\n\t") + "\n\n"
        return ret

    @staticmethod
    def update_menu():
        Menu.MAIN_OPTIONS["1"]["value"]   = Menu.IS_FILTER_RUNNING
        Menu.MAIN_OPTIONS["1"]["desc"]    = Menu.filter_instructions()
        Menu.MAIN_OPTIONS["3"]["value"]   = Menu.IS_LOGGING
        Menu.FILTER_OPTIONS["1"]["value"] = Menu.IS_DROP_INC_80
        Menu.FILTER_OPTIONS["2"]["value"] = Menu.IS_DROP_CLIENT_POST
        Menu.FILTER_OPTIONS["3"]["value"] = Menu.IS_DROP_LENGTH

    @staticmethod
    def filter_instructions():
        return "Press 1 to "+(Fore.RED+"Stop" if Menu.IS_FILTER_RUNNING else Fore.CYAN+"Start")+Fore.RESET\
               +" the network filter."

    @staticmethod
    def color_bool(boolean, color_true=Fore.LIGHTGREEN_EX, color_false=Fore.GREEN):
        if not isinstance(boolean, bool): return boolean
        return (color_true + "ON" if boolean else color_false + "OFF") + Fore.RESET

    @staticmethod
    def get_options():
        return Menu.IS_FILTER_RUNNING, Menu.IS_LOGGING, Menu.IS_DROP_INC_80, Menu.IS_DROP_CLIENT_POST, \
               Menu.IS_DROP_LENGTH

    @staticmethod
    def set_options(options):
        if len(options) != len(Menu.get_options()):
            raise ValueError("Length mismatch")
        Menu.IS_FILTER_RUNNING = options[0]
        Menu.IS_LOGGING = options[1]
        Menu.IS_DROP_INC_80 = options[2]
        Menu.IS_DROP_CLIENT_POST = options[3]
        Menu.IS_DROP_LENGTH = options[4]

    @staticmethod
    def emphasize():
        Menu.MAIN_OPTIONS["4"]["desc"] = Fore.WHITE + Menu.MAIN_OPTIONS["4"]["desc"] + Fore.RESET