101 lines
3.4 KiB
Python
101 lines
3.4 KiB
Python
import discord
|
|
from discord import app_commands
|
|
from discord.ext import commands, tasks
|
|
|
|
from utils.database import Session
|
|
from utils.jellyfin_create import create_jellyfin_account
|
|
from utils.jellyfin_delete import delete_accounts
|
|
from utils.models import JellyfinAccounts
|
|
from utils.config import (
|
|
JELLYFIN_PUBLIC_URL,
|
|
JELLYFIN_ENABLED,
|
|
ACCOUNT_TIME,
|
|
)
|
|
|
|
|
|
class NewAccount(commands.Cog):
|
|
def __init__(self, bot):
|
|
self.bot = bot
|
|
|
|
def cog_load(self):
|
|
self.delete_accounts_loop.start()
|
|
|
|
@app_commands.command()
|
|
@app_commands.check(lambda inter: JELLYFIN_ENABLED)
|
|
async def newaccount(self, interaction: discord.Interaction) -> None:
|
|
"""Create a new temporary Jellyfin account"""
|
|
# Make sure the user doesn't already have an account
|
|
with Session() as session:
|
|
account = (
|
|
session.query(JellyfinAccounts)
|
|
.filter(JellyfinAccounts.user_id == interaction.user.id)
|
|
.first()
|
|
)
|
|
# Account already allocated
|
|
if account:
|
|
embed = discord.Embed(
|
|
title="Account Already Exists",
|
|
description=(
|
|
"Look at your previous DMs with me to find your account"
|
|
" information. You will be permitted to create a new"
|
|
" account after your current one expires."
|
|
),
|
|
color=0xD01B86,
|
|
)
|
|
return await interaction.response.send_message(
|
|
embed=embed, ephemeral=True
|
|
)
|
|
|
|
# Create a new Jellyfin account for the user
|
|
response = create_jellyfin_account(interaction.user.id)
|
|
if response:
|
|
embed = discord.Embed(
|
|
title="Account Created",
|
|
description=(
|
|
"Your account has been successfully created. Check your"
|
|
" DMs for your account information."
|
|
),
|
|
color=0xD01B86,
|
|
)
|
|
await interaction.response.send_message(
|
|
embed=embed, ephemeral=True
|
|
)
|
|
|
|
# Send the user their account information
|
|
embed = discord.Embed(
|
|
title="Jellyfin Account Information",
|
|
description=(
|
|
# fmt: off
|
|
"Here is your temporary account information.\n\n"
|
|
f"**Server URL:** `{JELLYFIN_PUBLIC_URL}`\n"
|
|
f"**Username:** `{response[0]}`\n"
|
|
f"**Password:** `{response[1]}`\n\n"
|
|
"Your account will be automatically deleted in"
|
|
f" {ACCOUNT_TIME} hours."
|
|
# fmt: on
|
|
),
|
|
color=0xD01B86,
|
|
)
|
|
await interaction.user.send(embed=embed)
|
|
# If account not created for some reason
|
|
else:
|
|
embed = discord.Embed(
|
|
title="Unknown Error Occured",
|
|
description=(
|
|
"Error creating Jellyfin account. Please try again. If the"
|
|
" error persists, contact an administrator."
|
|
),
|
|
color=0xD01B86,
|
|
)
|
|
return await interaction.response.send_message(
|
|
embed=embed, ephemeral=True
|
|
)
|
|
|
|
@tasks.loop(minutes=1)
|
|
async def delete_accounts_loop(self):
|
|
delete_accounts()
|
|
|
|
|
|
async def setup(bot):
|
|
await bot.add_cog(NewAccount(bot))
|