2648f208da
* bot: add member count to system card * bot: fix potential crash on edge case Fixes a potential crash if a user queries the system card of a system with no members * bot: tidy up code * bot: move member count to embed title/Member field Displays the member count (between parentheses) in the embed title if the system is named, otherwise shows it in the Member field title (again between parentheses) * bot: move member count to Member field title entirely Removed the member count from the embed title and display it in the Member field title within parentheses in all cases
289 lines
10 KiB
Python
289 lines
10 KiB
Python
import discord
|
|
import math
|
|
import humanize
|
|
from typing import Tuple, List
|
|
|
|
from pluralkit import db
|
|
from pluralkit.bot.utils import escape
|
|
from pluralkit.member import Member
|
|
from pluralkit.switch import Switch
|
|
from pluralkit.system import System
|
|
from pluralkit.utils import get_fronters, display_relative
|
|
|
|
|
|
def truncate_field_name(s: str) -> str:
|
|
return s[:256]
|
|
|
|
|
|
def truncate_field_body(s: str) -> str:
|
|
if len(s) > 1024:
|
|
return s[:1024-3] + "..."
|
|
return s
|
|
|
|
|
|
def truncate_description(s: str) -> str:
|
|
return s[:2048]
|
|
|
|
|
|
def truncate_description_list(s: str) -> str:
|
|
if len(s) > 512:
|
|
return s[:512-45] + "..."
|
|
return s
|
|
|
|
|
|
def truncate_title(s: str) -> str:
|
|
return s[:256]
|
|
|
|
|
|
def success(text: str) -> discord.Embed:
|
|
embed = discord.Embed()
|
|
embed.description = truncate_description(text)
|
|
embed.colour = discord.Colour.green()
|
|
return embed
|
|
|
|
|
|
def error(text: str, help: Tuple[str, str] = None) -> discord.Embed:
|
|
embed = discord.Embed()
|
|
embed.description = truncate_description(text)
|
|
embed.colour = discord.Colour.dark_red()
|
|
|
|
if help:
|
|
help_title, help_text = help
|
|
embed.add_field(name=truncate_field_name(help_title), value=truncate_field_body(help_text))
|
|
|
|
return embed
|
|
|
|
|
|
def status(text: str) -> discord.Embed:
|
|
embed = discord.Embed()
|
|
embed.description = truncate_description(text)
|
|
embed.colour = discord.Colour.blue()
|
|
return embed
|
|
|
|
|
|
def exception_log(message_content, author_name, author_discriminator, author_id, server_id,
|
|
channel_id) -> discord.Embed:
|
|
embed = discord.Embed()
|
|
embed.colour = discord.Colour.dark_red()
|
|
embed.title = truncate_title(message_content)
|
|
|
|
embed.set_footer(text="Sender: {}#{} ({}) | Server: {} | Channel: {}".format(
|
|
author_name, author_discriminator, author_id,
|
|
server_id if server_id else "(DMs)",
|
|
channel_id
|
|
))
|
|
return embed
|
|
|
|
|
|
async def system_card(conn, client: discord.Client, system: System, is_own_system: bool = True) -> discord.Embed:
|
|
card = discord.Embed()
|
|
card.colour = discord.Colour.blue()
|
|
|
|
all_members = await system.get_members(conn)
|
|
member_count = str(len(all_members))
|
|
|
|
if system.name:
|
|
card.title = truncate_title(system.name)
|
|
|
|
if system.avatar_url:
|
|
card.set_thumbnail(url=system.avatar_url)
|
|
|
|
if system.tag:
|
|
card.add_field(name="Tag", value=truncate_field_body(system.tag))
|
|
|
|
fronters, switch_time = await get_fronters(conn, system.id)
|
|
if fronters:
|
|
names = ", ".join([member.name for member in fronters])
|
|
fronter_val = "{} (for {})".format(names, humanize.naturaldelta(switch_time))
|
|
card.add_field(name="Current fronter" if len(fronters) == 1 else "Current fronters",
|
|
value=truncate_field_body(fronter_val))
|
|
|
|
account_names = []
|
|
for account_id in await system.get_linked_account_ids(conn):
|
|
try:
|
|
account = await client.get_user_info(account_id)
|
|
account_names.append("<@{}> ({}#{})".format(account_id, account.name, account.discriminator))
|
|
except discord.NotFound:
|
|
account_names.append("(deleted account {})".format(account_id))
|
|
|
|
card.add_field(name="Linked accounts", value=truncate_field_body("\n".join(account_names)))
|
|
|
|
if system.description:
|
|
card.add_field(name="Description",
|
|
value=truncate_field_body(system.description), inline=False)
|
|
|
|
card.add_field(name="Members ({})".format(member_count), value="*See `pk;system {0} list`for the short list, or `pk;system {0} list full` for the detailed list*".format(system.hid) if not is_own_system else "*See `pk;system list` for the short list, or `pk;system list full` for the detailed list*")
|
|
card.set_footer(text="System ID: {}".format(system.hid))
|
|
return card
|
|
|
|
|
|
async def member_card(conn, member: Member) -> discord.Embed:
|
|
system = await member.fetch_system(conn)
|
|
|
|
card = discord.Embed()
|
|
card.colour = discord.Colour.blue()
|
|
|
|
name_and_system = member.name
|
|
if system.name:
|
|
name_and_system += " ({})".format(system.name)
|
|
|
|
card.set_author(name=truncate_field_name(name_and_system), icon_url=member.avatar_url or discord.Embed.Empty)
|
|
if member.avatar_url:
|
|
card.set_thumbnail(url=member.avatar_url)
|
|
|
|
if member.color:
|
|
card.colour = int(member.color, 16)
|
|
|
|
if member.birthday:
|
|
card.add_field(name="Birthdate", value=member.birthday_string())
|
|
|
|
if member.pronouns:
|
|
card.add_field(name="Pronouns", value=truncate_field_body(member.pronouns))
|
|
|
|
message_count = await member.message_count(conn)
|
|
if message_count > 0:
|
|
card.add_field(name="Message Count", value=str(message_count), inline=True)
|
|
|
|
if member.prefix or member.suffix:
|
|
prefix = member.prefix or ""
|
|
suffix = member.suffix or ""
|
|
card.add_field(name="Proxy Tags",
|
|
value=truncate_field_body("{}text{}".format(prefix, suffix)))
|
|
|
|
if member.description:
|
|
card.add_field(name="Description",
|
|
value=truncate_field_body(member.description), inline=False)
|
|
|
|
card.set_footer(text="System ID: {} | Member ID: {}".format(system.hid, member.hid))
|
|
return card
|
|
|
|
|
|
async def front_status(ctx: "CommandContext", switch: Switch) -> discord.Embed:
|
|
if switch:
|
|
embed = status("")
|
|
fronter_names = [member.name for member in await switch.fetch_members(ctx.conn)]
|
|
|
|
if len(fronter_names) == 0:
|
|
embed.add_field(name="Current fronter", value="(no fronter)")
|
|
elif len(fronter_names) == 1:
|
|
embed.add_field(name="Current fronter", value=truncate_field_body(fronter_names[0]))
|
|
else:
|
|
embed.add_field(name="Current fronters", value=truncate_field_body(", ".join(fronter_names)))
|
|
|
|
if switch.timestamp:
|
|
embed.add_field(name="Since",
|
|
value="{} ({})".format(ctx.format_time(switch.timestamp),
|
|
display_relative(switch.timestamp)))
|
|
else:
|
|
embed = error("No switches logged.")
|
|
return embed
|
|
|
|
|
|
async def get_message_contents(client: discord.Client, channel_id: int, message_id: int):
|
|
channel = client.get_channel(channel_id)
|
|
if channel:
|
|
try:
|
|
original_message = await channel.get_message(message_id)
|
|
return original_message.content or None
|
|
except (discord.errors.Forbidden, discord.errors.NotFound):
|
|
pass
|
|
|
|
return None
|
|
|
|
|
|
async def message_card(client: discord.Client, message: db.MessageInfo, include_pronouns: bool = False):
|
|
# Get the original sender of the messages
|
|
try:
|
|
original_sender = await client.get_user_info(message.sender)
|
|
except discord.NotFound:
|
|
# Account was since deleted - rare but we're handling it anyway
|
|
original_sender = None
|
|
|
|
embed = discord.Embed()
|
|
embed.timestamp = discord.utils.snowflake_time(message.mid)
|
|
embed.colour = discord.Colour.blue()
|
|
|
|
if message.system_name:
|
|
system_value = "{} (`{}`)".format(message.system_name, message.system_hid)
|
|
else:
|
|
system_value = "`{}`".format(message.system_hid)
|
|
embed.add_field(name="System", value=system_value)
|
|
|
|
if include_pronouns and message.pronouns:
|
|
embed.add_field(name="Member", value="{} (`{}`)\n*(pronouns: **{}**)*".format(message.name, message.hid, message.pronouns))
|
|
else:
|
|
embed.add_field(name="Member", value="{} (`{}`)".format(message.name, message.hid))
|
|
|
|
if original_sender:
|
|
sender_name = "<@{}> ({}#{})".format(message.sender, original_sender.name, original_sender.discriminator)
|
|
else:
|
|
sender_name = "(deleted account {})".format(message.sender)
|
|
|
|
embed.add_field(name="Sent by", value=sender_name)
|
|
|
|
message_content = await get_message_contents(client, message.channel, message.mid)
|
|
embed.description = message_content or "(unknown, message deleted)"
|
|
|
|
embed.set_author(name=message.name, icon_url=message.avatar_url or discord.Embed.Empty)
|
|
return embed
|
|
|
|
|
|
def help_footer_embed() -> discord.Embed:
|
|
embed = discord.Embed()
|
|
embed.set_footer(text="By @Ske#6201 | GitHub: https://github.com/xSke/PluralKit/")
|
|
return embed
|
|
|
|
# TODO: merge these somehow, they're very similar
|
|
def member_list_short(system: System, all_members: List[Member], current_page: int, page_size: int):
|
|
page_count = int(math.ceil(len(all_members) / page_size))
|
|
|
|
title = ""
|
|
if len(all_members) > page_size:
|
|
title += "[{}/{}] ".format(current_page + 1, page_count)
|
|
|
|
if system.name:
|
|
title += "Members of {} (`{}`)".format(system.name, system.hid)
|
|
else:
|
|
title += "Members of `{}`".format(system.hid)
|
|
|
|
embed = discord.Embed()
|
|
embed.title = title
|
|
|
|
desc = ""
|
|
for member in all_members[current_page*page_size:current_page*page_size+page_size]:
|
|
if member.prefix or member.suffix:
|
|
desc += "[`{}`] **{}** *({}text{})*\n".format(member.hid, member.name, member.prefix or "", member.suffix or "")
|
|
else:
|
|
desc += "[`{}`] **{}**\n".format(member.hid, member.name)
|
|
embed.description = desc
|
|
return embed
|
|
|
|
def member_list_full(system: System, all_members: List[Member], current_page: int, page_size: int):
|
|
page_count = int(math.ceil(len(all_members) / page_size))
|
|
|
|
title = ""
|
|
if len(all_members) > page_size:
|
|
title += "[{}/{}] ".format(current_page + 1, page_count)
|
|
|
|
if system.name:
|
|
title += "Members of {} (`{}`)".format(system.name, system.hid)
|
|
else:
|
|
title += "Members of `{}`".format(system.hid)
|
|
|
|
embed = discord.Embed()
|
|
embed.title = title
|
|
for member in all_members[current_page*page_size:current_page*page_size+page_size]:
|
|
member_description = "**ID**: {}\n".format(member.hid)
|
|
if member.birthday:
|
|
member_description += "**Birthday:** {}\n".format(member.birthday_string())
|
|
if member.pronouns:
|
|
member_description += "**Pronouns:** {}\n".format(member.pronouns)
|
|
if member.description:
|
|
if len(member.description) > 512:
|
|
member_description += "\n" + truncate_description_list(member.description) + "\n" + "Type `pk;member {}` for full description.".format(member.hid)
|
|
else:
|
|
member_description += "\n" + member.description
|
|
|
|
embed.add_field(name=member.name, value=truncate_field_body(member_description) or "\u200B", inline=False)
|
|
return embed
|