258 lines
8.2 KiB
Python
258 lines
8.2 KiB
Python
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"]
|