mirror of https://github.com/stijndcl/didier
130 lines
3.2 KiB
Python
130 lines
3.2 KiB
Python
from data.database_classes.custom_commands import CustomCommand
|
|
import discord
|
|
from functions.database import utils
|
|
|
|
|
|
def is_custom_command(message: discord.Message) -> CustomCommand:
|
|
"""
|
|
Check if a message triggers a custom command
|
|
These use "?" as a prefix
|
|
"""
|
|
content = message.content
|
|
# Message didn't call a custom command
|
|
if not content.startswith("?"):
|
|
return CustomCommand()
|
|
|
|
# Can't be invoked by bots to prevent spam (@RPS)
|
|
if message.author.bot:
|
|
return CustomCommand()
|
|
|
|
# Ignore capitals & spaces, strip off prefix
|
|
content = content.lower().replace(" ", "")[1:]
|
|
|
|
by_name = _find_by_name(content)
|
|
|
|
# Command was found by its name
|
|
if by_name:
|
|
return CustomCommand(*by_name)
|
|
|
|
# Check if a command exists with this alias instead
|
|
return CustomCommand(*_find_by_alias(content), alias_used=content)
|
|
|
|
|
|
def _find_by_name(message):
|
|
"""
|
|
Find a command by its name
|
|
"""
|
|
connection = utils.connect()
|
|
cursor = connection.cursor()
|
|
|
|
cursor.execute("SELECT * FROM custom_commands WHERE name = %s", (message,))
|
|
|
|
return cursor.fetchone()
|
|
|
|
|
|
def _find_by_alias(message):
|
|
"""
|
|
Find a command by one of its aliases
|
|
"""
|
|
connection = utils.connect()
|
|
cursor = connection.cursor()
|
|
|
|
cursor.execute("SELECT command FROM custom_command_aliases WHERE alias = %s", (message,))
|
|
|
|
res = cursor.fetchone()
|
|
|
|
# No command matched this alias
|
|
if not res:
|
|
return ()
|
|
|
|
cursor.execute("SELECT * FROM custom_commands WHERE id = %s", (res,))
|
|
|
|
return cursor.fetchone()
|
|
|
|
|
|
def is_name_free(name) -> bool:
|
|
"""
|
|
Check if a name is already in use by a command
|
|
Includes aliases
|
|
"""
|
|
connection = utils.connect()
|
|
cursor = connection.cursor()
|
|
|
|
cursor.execute("SELECT id from custom_commands WHERE name = %s", (name,))
|
|
|
|
if cursor.fetchone():
|
|
return False
|
|
|
|
cursor.execute("SELECT id FROM custom_command_aliases WHERE alias = %s", (name,))
|
|
|
|
return cursor.fetchone() is None
|
|
|
|
|
|
def _clean(inp: str):
|
|
"""
|
|
Strip markdown and other stuff out of a command name
|
|
"""
|
|
return "".join(filter(lambda x: x.isalnum(), inp))
|
|
|
|
|
|
def add_command(name: str, response: str):
|
|
"""
|
|
Add a new custom command
|
|
"""
|
|
name = _clean(name.lower())
|
|
|
|
if not is_name_free(name):
|
|
return "Er is al een commando met deze naam."
|
|
|
|
connection = utils.connect()
|
|
cursor = connection.cursor()
|
|
|
|
cursor.execute("INSERT INTO custom_commands(name, response) VALUES (%s, E%s)", (name, response,))
|
|
connection.commit()
|
|
|
|
|
|
def add_alias(command: str, alias: str):
|
|
"""
|
|
Add an alias for a command
|
|
Assumes the command exists
|
|
"""
|
|
command = _clean(command.lower())
|
|
alias = _clean(alias.lower())
|
|
|
|
# Base command doesn't exist
|
|
if is_name_free(command):
|
|
return "Er is geen commando met deze naam."
|
|
|
|
# Alias already exists
|
|
if not is_name_free(alias):
|
|
return "Er is al een commando met deze naam."
|
|
|
|
# Find the id of the base command
|
|
command_id = CustomCommand(*_find_by_name(command)).id
|
|
|
|
connection = utils.connect()
|
|
cursor = connection.cursor()
|
|
|
|
cursor.execute("INSERT INTO custom_command_aliases(command, alias) VALUES(%s, %s)", (command_id, alias,))
|
|
connection.commit()
|