87 lines
2.4 KiB
Python
87 lines
2.4 KiB
Python
from typing import Tuple
|
|
|
|
import discord
|
|
|
|
from payphone.db import models
|
|
|
|
|
|
async def get_channel(channel: discord.TextChannel) -> models.Channel:
|
|
"""Returns a channel from the database, creating or updating it if it already exists.
|
|
|
|
Args:
|
|
channel (discord.TextChannel): The channel to fetch.
|
|
|
|
Returns:
|
|
models.Channel: The channel row.
|
|
"""
|
|
|
|
row, _ = await models.Channel.update_or_create(
|
|
id=channel.id, server=channel.guild.id
|
|
)
|
|
|
|
channel_name = channel.name.replace("discord", "d").replace("wumpus", "w")
|
|
webhook_name = f"Payphone #{channel_name}"
|
|
|
|
if row.webhook_id is None:
|
|
webhook = await channel.create_webhook(
|
|
name=webhook_name, reason="First call started in channel."
|
|
)
|
|
|
|
row.webhook_id = webhook.id
|
|
row.webhook_url = webhook.url
|
|
|
|
await row.save(update_fields=["webhook_id", "webhook_url"])
|
|
|
|
else:
|
|
found = False
|
|
|
|
for webhook in await channel.webhooks():
|
|
if webhook.id != row.webhook_id:
|
|
if webhook.token is None:
|
|
await webhook.delete(reason="Webhook is unaccessible by Payphone.")
|
|
break
|
|
|
|
else:
|
|
found = True
|
|
|
|
if not found:
|
|
webhook = await channel.create_webhook(
|
|
name=webhook_name, reason="Previous webhook was deleted."
|
|
)
|
|
|
|
row.webhook_id = webhook.id
|
|
row.webhook_url = webhook.url
|
|
|
|
await row.save(update_fields=["webhook_id", "webhook_url"])
|
|
|
|
return row
|
|
|
|
|
|
async def get_user(
|
|
user: discord.User | discord.Member,
|
|
) -> Tuple[models.User, models.Profile]:
|
|
"""Returns a user from the database, creating or updating it if it already exists.
|
|
|
|
Args:
|
|
user (discord.User | discord.Member): The user or member to fetch.
|
|
|
|
Returns:
|
|
Tuple[models.User, models.Profile]: The user and profile.
|
|
"""
|
|
|
|
user_row, _ = await models.User.update_or_create(
|
|
{"username": user.name}, id=user.id
|
|
)
|
|
|
|
# TODO: Make per-server profiles a premium feature.
|
|
profile_row, _ = await models.Profile.update_or_create(
|
|
{
|
|
"avatar_url": user.display_avatar.url,
|
|
"name": user.global_name or user.name,
|
|
},
|
|
user=user_row,
|
|
server_id=None, # user.guild.id if isinstance(user, discord.Member) else None,
|
|
)
|
|
|
|
return (user_row, profile_row)
|