qcs-python/main.py

import os
import json
import logging
import toml
import getpass
import textwrap
import readchar

from colorama import Fore, Back, Style, init as colorama_init

import contentapi
import myutils

CONFIGFILE="config.toml"

# The entire config object with all defaults
config = {
    "api" : "https://oboy.smilebasicsource.com/api",
    "default_loglevel" : "WARNING",
    "default_room" : 0, # Zero means it will ask you for a room
    "expire_seconds" : 31536000, # 365 days in seconds, expiration for token
    "appear_in_global" : False,
    "tokenfile" : ".qcstoken"
}

class WebsocketContext:
    def __init__(self, api_context, user_info):
        self.api_context = api_context
        self.current_room = 0
        self.user_info = user_info
        self.connected = False # Should this be a var? IDK
        self.pause

def main():
    print("Program start")
    colorama_init() # colorama init
    load_or_create_global_config()
    logging.info("Config: " + json.dumps(config, indent = 2))
    context = contentapi.ApiContext(config["api"], logging)
    logging.info("Testing connection to API at " + config["api"])
    logging.debug(json.dumps(context.api_status(), indent = 2))
    authenticate(config, context)

    ws_context = WebsocketContext(context, context.user_me())
    ws_context.current_room = config["default_room"]

    # - Connect to websocket, be ready to receive junk
    # - Alert user they're not connected to any room, maybe have a status line that lists controls + room
    # - Enter input loop, but check room number on "input" mode, don't let messages send in room 0
    #   - h to help
    #   - s to search rooms, enter #1234 to connect directly, empty string to quit
    #   - g to list global users
    #   - u to list users in room
    #   - i to input
    #   - q to quit entirely

    printstatus = True

    # The infinite input loop! Or something!
    while True:
        if printstatus:
            print_statusline(ws_context)
        printstatus = True
        key = readchar.readkey()
        if key == "h":
            print("not yet")
        elif key == "s":
            print("not yet")
        elif key == "g":
            print("not yet")
        elif key == "u":
            print("not yet")
        elif key == "i":
            print("not yet")
        elif key == "q":
            break
        else:
            printstatus = False

    
    # TODO: Will the websocket end if the program just ends?

    print("Program end")

# Loads the config from file into the global config var. If the file
# doesn't exist, the file is created from the defaults in config.
# The function returns nothing
def load_or_create_global_config():
    global config
    # Check if the config file exists
    if os.path.isfile(CONFIGFILE):
        # Read and deserialize the config file
        with open(CONFIGFILE, 'r', encoding='utf-8') as f:
            temp_config = toml.load(f)
            myutils.merge_dictionary(temp_config, config)
    else:
        # Serialize and write the config dictionary to the config file
        logging.warn("No config found at " + CONFIGFILE + ", creating now")
        with open(CONFIGFILE, 'w', encoding='utf-8') as f:
            toml.dump(config, f)

    myutils.set_logging_level(config["default_loglevel"])


# Either pull the token from a file, or get the login from the command
# line if that doesn't work. WILL test your token against the real API
# even if it's pulled from file!
def authenticate(config, context: contentapi.ApiContext):
    message = "No token file found"
    if os.path.isfile(config["tokenfile"]):
        with open(config["tokenfile"], 'r') as f:
            token = f.read()
        logging.debug("Token from file: " + token)
        context.token = token
        if context.is_token_valid():
            logging.info("Logged in using token file " + config["tokenfile"])
            return
        else:
            message = "Token file expired"
    
    message += ", Please enter login for " + config["api"]

    while True:
        print(message)
        username = input("Username: ")
        password = getpass.getpass("Password: ")
        try:
            token = context.login(username, password, config["expire_seconds"])
            with open(config["tokenfile"], 'w') as f:
                f.write(token)
            logging.info("Token accepted, written to " + config["tokenfile"])
            context.token = token
            return
        except Exception as ex:
            print("ERROR: %s" % ex)
            message = "Please try logging in again:"

def print_statusline(ws_context: WebsocketContext):
    if ws_context.connected:
        bg = Back.GREEN
    else:
        bg = Back.RED
    print(bg + Fore.BLACK + "\n User: " + ws_context.user_info["username"] + "  CTRL: h s g u i q  " + Style.RESET_ALL)

# Because python reasons
if __name__ == "__main__":
    main()