import jsonschema import re import os import yaml import validators import openai import sys import discord import logging import requests from colorlog import ColoredFormatter log_level = logging.DEBUG log_format = ( " %(log_color)s%(levelname)-8s%(reset)s | %(log_color)s%(message)s%(reset)s" ) logging.root.setLevel(log_level) formatter = ColoredFormatter(log_format) stream = logging.StreamHandler() stream.setLevel(log_level) stream.setFormatter(formatter) LOG = logging.getLogger("pythonConfig") LOG.setLevel(log_level) LOG.addHandler(stream) TOKEN = None BOT_COLOR = None BOT_INVITE_LINK = None FEEDBACK_CHANNEL_ID = None BUG_CHANNEL_ID = None SPOTIFY_CLIENT_ID = None SPOTIFY_CLIENT_SECRET = None GENIUS_CLIENT_ID = None GENIUS_CLIENT_SECRET = None OPENAI_API_KEY = None LAVALINK_HOST = None LAVALINK_PORT = None LAVALINK_PASSWORD = None schema = { "type": "object", "properties": { "bot_info": { "type": "object", "properties": { "token": {"type": "string"}, "bot_color": {"type": "string"}, "bot_invite_link": {"type": "string"}, "feedback_channel_id": {"type": "integer"}, "bug_channel_id": {"type": "integer"}, }, "required": ["token"], }, "spotify": { "type": "object", "properties": { "spotify_client_id": {"type": "string"}, "spotify_client_secret": {"type": "string"}, }, "required": ["spotify_client_id", "spotify_client_secret"], }, "genius": { "type": "object", "properties": { "genius_client_id": {"type": "string"}, "genius_client_secret": {"type": "string"}, }, "required": ["genius_client_id", "genius_client_secret"], }, "openai": { "type": "object", "properties": { "openai_api_key": {"type": "string"}, }, "required": ["openai_api_key"], }, "lavalink": { "type": "object", "properties": { "host": {"type": "string"}, "port": {"type": "integer"}, "password": {"type": "string"}, }, "required": ["host", "port", "password"], }, }, "required": ["bot_info", "lavalink"], } # Attempt to load the config file, otherwise create a new template def load_config(): if os.path.exists("/.dockerenv"): file_path = "/config/config.yaml" else: file_path = "config.yaml" try: with open(file_path, "r") as f: file_contents = f.read() validate_config(file_contents) except FileNotFoundError: # Create a new config.yaml file with the template with open("config.yaml", "w") as f: f.write( """ bot_info: token: "" bot_color: "" bot_invite_link: "" feedback_channel_id: "" bug_channel_id: "" spotify: spotify_client_id: "" spotify_client_secret: "" genius: genius_client_id: "" genius_client_secret: "" openai: openai_api_key: "" lavalink: host: "" port: "" password: "" """ ) sys.exit( LOG.critical( "Configuration file `config.yaml` has been generated. Please fill out all of the necessary information. Refer to the docs for information on what a specific configuration option is." ) ) # Thouroughly validate all of the options in the config.yaml file def validate_config(file_contents): global TOKEN, BOT_COLOR, BOT_INVITE_LINK, FEEDBACK_CHANNEL_ID, BUG_CHANNEL_ID, SPOTIFY_CLIENT_ID, SPOTIFY_CLIENT_SECRET, GENIUS_CLIENT_ID, GENIUS_CLIENT_SECRET, OPENAI_API_KEY, LAVALINK_HOST, LAVALINK_PORT, LAVALINK_PASSWORD config = yaml.safe_load(file_contents) try: jsonschema.validate(config, schema) except jsonschema.ValidationError as e: sys.exit(LOG.critical(f"Error in config.yaml file: {e.message}")) # # Begin validation for optional BOT_INFO values # # If there is a "bot_invite_link" option, make sure it's a valid URL if "bot_invite_link" in config["bot_info"]: if not validators.url(config["bot_info"]["bot_invite_link"]): LOG.critical( "Error in config.yaml file: bot_invite_link is not a valid URL" ) else: BOT_INVITE_LINK = config["bot_info"]["bot_invite_link"] # Make sure "bot_color" is a valid hex color hex_pattern_one = "^#([A-Fa-f0-9]{6}|[A-Fa-f0-9]{3})$" hex_pattern_two = "^([A-Fa-f0-9]{6}|[A-Fa-f0-9]{3})$" if "bot_color" in config["bot_info"]: if not bool( re.match(hex_pattern_one, config["bot_info"]["bot_color"]) ) and not bool(re.match(hex_pattern_two, config["bot_info"]["bot_color"])): LOG.critical( "Error in config.yaml file: bot_color is not a valid hex color" ) else: BOT_COLOR = discord.Color( int((config["bot_info"]["bot_color"]).replace("#", ""), 16) ) # Make sure "feedback_channel_id" and "bug_channel_id" are exactly 19 characters long if "feedback_channel_id" in config["bot_info"]: if len(str(config["bot_info"]["feedback_channel_id"])) != 0: if len(str(config["bot_info"]["feedback_channel_id"])) != 19: LOG.critical( "Error in config.yaml file: feedback_channel_id is not a valid Discord channel ID" ) else: FEEDBACK_CHANNEL_ID = config["bot_info"]["feedback_channel_id"] if "bug_channel_id" in config["bot_info"]: if len(str(config["bot_info"]["bug_channel_id"])) != 0: if len(str(config["bot_info"]["bug_channel_id"])) != 19: LOG.critical( "Error in config.yaml file: bug_channel_id is not a valid Discord channel ID" ) else: BUG_CHANNEL_ID = config["bot_info"]["bug_channel_id"] # # If the SPOTIFY section is present, make sure the client ID and secret are valid # if "spotify" in config: auth_url = "https://accounts.spotify.com/api/token" data = { "grant_type": "client_credentials", "client_id": config["spotify"]["spotify_client_id"], "client_secret": config["spotify"]["spotify_client_secret"], } response = requests.post(auth_url, data=data) if response.status_code == 200: SPOTIFY_CLIENT_ID = config["spotify"]["spotify_client_id"] SPOTIFY_CLIENT_SECRET = config["spotify"]["spotify_client_secret"] else: LOG.critical( "Error in config.yaml file: Spotify client ID or secret is invalid" ) # # If the GENIUS section is present, make sure the client ID and secret are valid # if "genius" in config: auth_url = "https://api.genius.com/oauth/token" data = { "grant_type": "client_credentials", "client_id": config["genius"]["genius_client_id"], "client_secret": config["genius"]["genius_client_secret"], } response = requests.post(auth_url, data=data) if response.status_code == 200: GENIUS_CLIENT_ID = config["genius"]["genius_client_id"] GENIUS_CLIENT_SECRET = config["genius"]["genius_client_secret"] else: LOG.critical( "Error in config.yaml file: Genius client ID or secret is invalid" ) # # If the OPENAI section is present, make sure the API key is valid # if "openai" in config: client = openai.OpenAI(api_key=config["openai"]["openai_api_key"]) try: client.models.list() OPENAI_API_KEY = config["openai"]["openai_api_key"] except openai.AuthenticationError: LOG.critical("Error in config.yaml file: OpenAI API key is invalid") # Set appropriate values for all non-optional variables TOKEN = config["bot_info"]["token"] LAVALINK_HOST = config["lavalink"]["host"] LAVALINK_PORT = config["lavalink"]["port"] LAVALINK_PASSWORD = config["lavalink"]["password"]