312 lines
9.1 KiB
Python
312 lines
9.1 KiB
Python
from datetime import datetime, timedelta, timezone
|
||
from typing import TypedDict, Any, List, Dict
|
||
import db_utils as db
|
||
from add_card import add_card
|
||
from config import GACHA_ROLL_INTERVAL
|
||
from custom_types import BotResponse, ParsedNotification
|
||
|
||
|
||
def do_roll(author: str) -> BotResponse:
|
||
'''Determines whether the user can roll, then pulls a random card'''
|
||
user_id = db.get_player(author)
|
||
if not user_id:
|
||
return {
|
||
'message': f'{author} 🛑 You haven’t signed up yet! Use the \
|
||
`signup` command to start playing.',
|
||
'attachment_urls': None
|
||
}
|
||
# Get date of user's last roll
|
||
date = db.get_last_rolled_at(user_id)
|
||
|
||
# No date means it's users first roll
|
||
if date:
|
||
# SQLite timestamps returned by the DB are always in UTC
|
||
# Below timestamps are to be converted to UTC
|
||
prev = datetime.strptime(str(date) + '+0000', '%Y-%m-%d %H:%M:%S%z')
|
||
now = datetime.now(timezone.utc)
|
||
|
||
time_since_last_roll = now - prev
|
||
roll_interval = timedelta(seconds=GACHA_ROLL_INTERVAL)
|
||
duration = roll_interval - time_since_last_roll
|
||
|
||
# User needs to wait before they can roll again
|
||
if time_since_last_roll < roll_interval:
|
||
remaining_duration = None
|
||
if duration.seconds > 3600:
|
||
remaining_duration = f'{-(duration.seconds // -3600)} hours'
|
||
elif duration.seconds > 60:
|
||
remaining_duration = f'{-(duration.seconds // -60)} minutes'
|
||
else:
|
||
remaining_duration = f'{duration.seconds} seconds'
|
||
|
||
return {
|
||
'message': f'{author} ⏱️ Please wait another \
|
||
{remaining_duration} before rolling again.',
|
||
'attachment_urls': None
|
||
}
|
||
|
||
card = db.get_random_card()
|
||
|
||
if not card:
|
||
return {
|
||
'message': f'{author} Uwaaa... something went wrong! No \
|
||
cards found. 😿',
|
||
'attachment_urls': None
|
||
}
|
||
|
||
db.insert_pull(user_id, card['id'])
|
||
stars = '⭐️' * card['rarity']
|
||
return {
|
||
'message': f'{author} 🎲 Congrats! You rolled {stars} \
|
||
**{card['name']}**\nShe\'s all yours now~ 💖✨',
|
||
'attachment_urls': [card['image_url']]
|
||
}
|
||
|
||
|
||
def do_signup(author: str) -> BotResponse:
|
||
'''Registers a new user if they haven’t signed up yet.'''
|
||
user_id = db.get_player(author)
|
||
|
||
if user_id:
|
||
return {
|
||
'message': f'{author} 👀 You’re already signed up! Let the rolling \
|
||
begin~ 🎲',
|
||
'attachment_urls': None
|
||
}
|
||
|
||
new_user_id = db.insert_player(author)
|
||
return {
|
||
'message': f'{author} ✅ Signed up successfully! Your gacha \
|
||
destiny begins now... ✨ Use the roll command to start!',
|
||
'attachment_urls': None
|
||
}
|
||
|
||
|
||
def is_float(val: Any) -> bool:
|
||
'''Returns true if `val` can be converted to a float'''
|
||
try:
|
||
float(val)
|
||
return True
|
||
except ValueError:
|
||
return False
|
||
|
||
|
||
def do_create(
|
||
author: str,
|
||
arguments: List[str],
|
||
note_obj: Dict[str, Any]) -> BotResponse:
|
||
'''Creates a card'''
|
||
# Example call from bot logic
|
||
image_url = note_obj.get('files', [{}])[0].get('url') \
|
||
if note_obj.get('files') else None
|
||
|
||
if not image_url:
|
||
return {
|
||
'message': f'{author} You need an image to create a card, \
|
||
dumbass.',
|
||
'attachment_urls': None
|
||
}
|
||
|
||
if len(arguments) != 2:
|
||
return {
|
||
'message': f'{author} Please specify the following attributes \
|
||
in order: name, rarity',
|
||
'attachment_urls': None
|
||
}
|
||
|
||
if not (arguments[1].isnumeric() and 1 <= int(arguments[1]) <= 5):
|
||
return {
|
||
'message': f'{author} Invalid rarity: \'{arguments[1]}\' must \
|
||
be a number between 1 and 5',
|
||
'attachment_urls': None
|
||
}
|
||
|
||
card_id, file_id = add_card(
|
||
name=arguments[0],
|
||
rarity=int(arguments[1]),
|
||
image_url=image_url
|
||
)
|
||
return {
|
||
'message': f'{author} Added {arguments[0]}, ID {card_id}.',
|
||
'attachment_urls': [file_id]
|
||
}
|
||
|
||
|
||
def do_help(author: str) -> BotResponse:
|
||
'''Provides a list of commands that the bot can do.'''
|
||
return {
|
||
'message': f'{author} Here\'s what I can do:\n\
|
||
- `roll` Pulls a random card.\n\
|
||
- `create <name> <rarity>` Creates a card using a given image.\n\
|
||
- `signup` Registers your account.\n\
|
||
- `delete_account` Deletes your account.\n\
|
||
- `help` Shows this message',
|
||
'attachment_urls': None
|
||
}
|
||
|
||
|
||
def delete_account(author: str) -> BotResponse:
|
||
return {
|
||
'message': f'{author} ⚠️ This will permanently delete your account \
|
||
and all your cards.\n'
|
||
'If you\'re sure, reply with `confirm_delete_account` to proceed.\n\n'
|
||
'**There is no undo.** Your gacha luck will be lost to the void... 💀✨',
|
||
'attachment_urls': None
|
||
|
||
}
|
||
|
||
|
||
def confirm_delete(author: str) -> BotResponse:
|
||
db.delete_player(author)
|
||
|
||
return {
|
||
'message': f'{author} 🧼 Your account and all your cards have been \
|
||
deleted. RIP your gacha history 🕊️✨',
|
||
'attachment_urls': None
|
||
}
|
||
|
||
|
||
def do_whitelist(author: str, args: list[str]) -> BotResponse:
|
||
if len(args) == 0:
|
||
return {
|
||
'message': f'{author} Please specify an instance to whitelist',
|
||
'attachment_urls': None
|
||
}
|
||
|
||
if db.add_to_whitelist(args[0]):
|
||
return {
|
||
'message': f'{author} Whitelisted instance: {args[0]}',
|
||
'attachment_urls': None
|
||
}
|
||
else:
|
||
return {
|
||
'message': f'{author} Instance already whitelisted: {args[0]}',
|
||
'attachment_urls': None
|
||
}
|
||
|
||
|
||
def do_unwhitelist(author: str, args: list[str]) -> BotResponse:
|
||
if len(args) == 0:
|
||
return {
|
||
'message': f'{author} Please specify an instance to remove from \
|
||
the whitelist',
|
||
'attachment_urls': None
|
||
}
|
||
|
||
if db.remove_from_whitelist(args[0]):
|
||
return {
|
||
'message': f'{author} Unwhitelisted instance: {args[0]}',
|
||
'attachment_urls': None
|
||
}
|
||
else:
|
||
return {
|
||
'message': f'{author} Instance not whitelisted: {args[0]}',
|
||
'attachment_urls': None
|
||
}
|
||
|
||
|
||
def do_ban(author: str, args: list[str]) -> BotResponse:
|
||
if len(args) == 0:
|
||
return {
|
||
'message': f'{author} Please specify a user to ban',
|
||
'attachment_urls': None
|
||
}
|
||
|
||
if db.is_player_administrator(args[0]):
|
||
return {
|
||
'message': f'{author} Cannot ban other administrators.',
|
||
'attachment_urls': None
|
||
}
|
||
|
||
if db.ban_player(args[0]):
|
||
# Delete banned player's account
|
||
db.delete_player(args[0])
|
||
return {
|
||
'message': f'{author} 🔨 **BONK!** Get banned, {args[0]}!',
|
||
'attachment_urls': None
|
||
}
|
||
else:
|
||
return {
|
||
'message': f'{author} Player is already banned: {args[0]}',
|
||
'attachment_urls': None
|
||
}
|
||
|
||
|
||
def do_unban(author: str, args: list[str]) -> BotResponse:
|
||
if len(args) == 0:
|
||
return {
|
||
'message': f'{author} Please specify a user to unban',
|
||
'attachment_urls': None
|
||
}
|
||
|
||
if db.unban_player(args[0]):
|
||
return {
|
||
'message': f'{author} Player unbanned: {args[0]}!',
|
||
'attachment_urls': None
|
||
}
|
||
else:
|
||
return {
|
||
'message': f'{author} Player was not banned: {args[0]}',
|
||
'attachment_urls': None
|
||
}
|
||
|
||
|
||
def generate_response(notification: ParsedNotification) -> BotResponse | None:
|
||
'''Given a command with arguments, processes the game state and
|
||
returns a response'''
|
||
|
||
# Temporary response variable
|
||
res: BotResponse | None = None
|
||
author = notification['author']
|
||
player_id = db.get_player(author)
|
||
command = notification['command']
|
||
|
||
# Unrestricted commands
|
||
match command:
|
||
case 'roll':
|
||
res = do_roll(author)
|
||
case 'signup':
|
||
res = do_signup(author)
|
||
case 'help':
|
||
res = do_help(author)
|
||
case _:
|
||
pass
|
||
|
||
# Commands beyond this point require the user to have an account
|
||
if not player_id:
|
||
return res
|
||
|
||
# User commands
|
||
match command:
|
||
case 'create':
|
||
res = do_create(
|
||
author,
|
||
notification['arguments'],
|
||
notification['note_obj']
|
||
)
|
||
case 'delete_account':
|
||
res = delete_account(author)
|
||
case 'confirm_delete_account':
|
||
res = confirm_delete(author)
|
||
case _:
|
||
pass
|
||
|
||
# Commands beyond this point require the user to be an administrator
|
||
if not db.is_player_administrator(author):
|
||
return res
|
||
|
||
# Admin commands
|
||
match command:
|
||
case 'whitelist':
|
||
res = do_whitelist(author, notification['arguments'])
|
||
case 'unwhitelist':
|
||
res = do_unwhitelist(author, notification['arguments'])
|
||
case 'ban':
|
||
res = do_ban(author, notification['arguments'])
|
||
case 'unban':
|
||
res = do_unban(author, notification['arguments'])
|
||
case _:
|
||
pass
|
||
|
||
# Administrator commands go here
|
||
return res
|