3
0
Fork 0
mirror of https://github.com/ZeppelinBot/Zeppelin.git synced 2025-07-10 12:37:19 +00:00

feat: per-server avatars & username fixes

This commit is contained in:
rubyowo 2023-07-03 16:46:54 +04:00
parent 5a4e50b19d
commit b5c12770dc
No known key found for this signature in database
11 changed files with 37 additions and 23 deletions

View file

@ -11,7 +11,7 @@ export async function createCase(pluginData: GuildPluginData<CasesPluginType>, a
const userName = renderUserUsername(user); const userName = renderUserUsername(user);
const mod = await resolveUser(pluginData.client, args.modId); const mod = await resolveUser(pluginData.client, args.modId);
const modName = mod.tag; const modName = renderUserUsername(mod);
let ppName: string | null = null; let ppName: string | null = null;
let ppId: Snowflake | null = null; let ppId: Snowflake | null = null;

View file

@ -1,6 +1,6 @@
import { GuildPluginData } from "knub"; import { GuildPluginData } from "knub";
import { ERRORS, RecoverablePluginError } from "../../../RecoverablePluginError"; import { ERRORS, RecoverablePluginError } from "../../../RecoverablePluginError";
import { UnknownUser, resolveUser } from "../../../utils"; import { UnknownUser, renderUserUsername, resolveUser } from "../../../utils";
import { CaseNoteArgs, CasesPluginType } from "../types"; import { CaseNoteArgs, CasesPluginType } from "../types";
import { postCaseToCaseLogChannel } from "./postToCaseLogChannel"; import { postCaseToCaseLogChannel } from "./postToCaseLogChannel";
import { resolveCaseId } from "./resolveCaseId"; import { resolveCaseId } from "./resolveCaseId";
@ -16,7 +16,7 @@ export async function createCaseNote(pluginData: GuildPluginData<CasesPluginType
throw new RecoverablePluginError(ERRORS.INVALID_USER); throw new RecoverablePluginError(ERRORS.INVALID_USER);
} }
const modName = mod.tag; const modName = renderUserUsername(mod);
let body = args.body; let body = args.body;

View file

@ -48,7 +48,7 @@ export async function sendMessage(
...content, ...content,
...(pluginData.client.user && { ...(pluginData.client.user && {
username: pluginData.client.user.username, username: pluginData.client.user.username,
avatarURL: pluginData.client.user.avatarURL() || pluginData.client.user.defaultAvatarURL, avatarURL: pluginData.client.user.displayAvatarURL(),
}), }),
}) })
.then((apiMessage) => ({ .then((apiMessage) => ({

View file

@ -1,7 +1,7 @@
import { APIEmbed, User } from "discord.js"; import { APIEmbed, GuildMember, User } from "discord.js";
import { commandTypeHelpers as ct } from "../../../commandTypes"; import { commandTypeHelpers as ct } from "../../../commandTypes";
import { sendErrorMessage } from "../../../pluginUtils"; import { sendErrorMessage } from "../../../pluginUtils";
import { emptyEmbedValue, resolveUser, trimLines } from "../../../utils"; import { emptyEmbedValue, renderUserUsername, resolveMember, resolveUser, trimLines } from "../../../utils";
import { asyncMap } from "../../../utils/async"; import { asyncMap } from "../../../utils/async";
import { createPaginatedMessage } from "../../../utils/createPaginatedMessage"; import { createPaginatedMessage } from "../../../utils/createPaginatedMessage";
import { getChunkedEmbedFields } from "../../../utils/getChunkedEmbedFields"; import { getChunkedEmbedFields } from "../../../utils/getChunkedEmbedFields";
@ -28,8 +28,10 @@ export const CasesModCmd = modActionsCmd({
async run({ pluginData, message: msg, args }) { async run({ pluginData, message: msg, args }) {
const modId = args.mod || msg.author.id; const modId = args.mod || msg.author.id;
const mod = await resolveUser(pluginData.client, modId); const mod =
const modName = mod instanceof User ? mod.tag : modId; (await resolveMember(pluginData.client, pluginData.guild, modId)) ||
(await resolveUser(pluginData.client, modId));
const modName = mod instanceof User || mod instanceof GuildMember ? renderUserUsername(mod) : modId;
const casesPlugin = pluginData.getPlugin(CasesPlugin); const casesPlugin = pluginData.getPlugin(CasesPlugin);
const totalCases = await casesPlugin.getTotalCasesByMod(modId); const totalCases = await casesPlugin.getTotalCasesByMod(modId);
@ -57,7 +59,7 @@ export const CasesModCmd = modActionsCmd({
const embed = { const embed = {
author: { author: {
name: title, name: title,
icon_url: mod instanceof User ? mod.displayAvatarURL() : undefined, icon_url: mod instanceof GuildMember || mod instanceof User ? mod.displayAvatarURL() : undefined,
}, },
fields: [ fields: [
...getChunkedEmbedFields(emptyEmbedValue, lines.join("\n")), ...getChunkedEmbedFields(emptyEmbedValue, lines.join("\n")),

View file

@ -1,9 +1,17 @@
import { APIEmbed, User } from "discord.js"; import { APIEmbed, GuildMember, User } from "discord.js";
import { commandTypeHelpers as ct } from "../../../commandTypes"; import { commandTypeHelpers as ct } from "../../../commandTypes";
import { CaseTypes } from "../../../data/CaseTypes"; import { CaseTypes } from "../../../data/CaseTypes";
import { sendErrorMessage } from "../../../pluginUtils"; import { sendErrorMessage } from "../../../pluginUtils";
import { CasesPlugin } from "../../../plugins/Cases/CasesPlugin"; import { CasesPlugin } from "../../../plugins/Cases/CasesPlugin";
import { UnknownUser, chunkArray, emptyEmbedValue, renderUserUsername, resolveUser, trimLines } from "../../../utils"; import {
UnknownUser,
chunkArray,
emptyEmbedValue,
renderUserUsername,
resolveMember,
resolveUser,
trimLines,
} from "../../../utils";
import { asyncMap } from "../../../utils/async"; import { asyncMap } from "../../../utils/async";
import { getChunkedEmbedFields } from "../../../utils/getChunkedEmbedFields"; import { getChunkedEmbedFields } from "../../../utils/getChunkedEmbedFields";
import { getGuildPrefix } from "../../../utils/getGuildPrefix"; import { getGuildPrefix } from "../../../utils/getGuildPrefix";
@ -35,7 +43,9 @@ export const CasesUserCmd = modActionsCmd({
], ],
async run({ pluginData, message: msg, args }) { async run({ pluginData, message: msg, args }) {
const user = await resolveUser(pluginData.client, args.user); const user =
(await resolveMember(pluginData.client, pluginData.guild, args.user)) ||
(await resolveUser(pluginData.client, args.user));
if (!user.id) { if (!user.id) {
sendErrorMessage(pluginData, msg.channel, `User not found`); sendErrorMessage(pluginData, msg.channel, `User not found`);
return; return;
@ -123,7 +133,7 @@ export const CasesUserCmd = modActionsCmd({
lineChunks.length === 1 lineChunks.length === 1
? `Cases for ${userName} (${lines.length} total)` ? `Cases for ${userName} (${lines.length} total)`
: `Cases ${chunkStart}${chunkEnd} of ${lines.length} for ${userName}`, : `Cases ${chunkStart}${chunkEnd} of ${lines.length} for ${userName}`,
icon_url: user instanceof User ? user.displayAvatarURL() : undefined, icon_url: user instanceof GuildMember || user instanceof User ? user.displayAvatarURL() : undefined,
}, },
fields: [ fields: [
...getChunkedEmbedFields(emptyEmbedValue, linesInChunk.join("\n")), ...getChunkedEmbedFields(emptyEmbedValue, linesInChunk.join("\n")),

View file

@ -1,6 +1,6 @@
import { GuildChannel, Message } from "discord.js"; import { GuildChannel, Message } from "discord.js";
import path from "path"; import path from "path";
import { EMPTY_CHAR, EmbedWith } from "../../../utils"; import { EMPTY_CHAR, EmbedWith, renderUserUsername } from "../../../utils";
const imageAttachmentExtensions = ["jpeg", "jpg", "png", "gif", "webp"]; const imageAttachmentExtensions = ["jpeg", "jpg", "png", "gif", "webp"];
const audioAttachmentExtensions = ["wav", "mp3", "m4a"]; const audioAttachmentExtensions = ["wav", "mp3", "m4a"];
@ -18,7 +18,7 @@ export function createStarboardEmbedFromMessage(
text: `#${(msg.channel as GuildChannel).name}`, text: `#${(msg.channel as GuildChannel).name}`,
}, },
author: { author: {
name: msg.author.tag, name: renderUserUsername(msg.author),
}, },
fields: [], fields: [],
timestamp: msg.createdAt.toISOString(), timestamp: msg.createdAt.toISOString(),
@ -28,7 +28,7 @@ export function createStarboardEmbedFromMessage(
embed.color = color; embed.color = color;
} }
embed.author.icon_url = msg.author.displayAvatarURL(); embed.author.icon_url = (msg.member || msg.author).displayAvatarURL();
// The second condition here checks for messages with only an image link that is then embedded. // The second condition here checks for messages with only an image link that is then embedded.
// The message content in that case is hidden by the Discord client, so we hide it here too. // The message content in that case is hidden by the Discord client, so we hide it here too.

View file

@ -10,11 +10,11 @@ export const AvatarCmd = utilityCmd({
permission: "can_avatar", permission: "can_avatar",
signature: { signature: {
user: ct.resolvedUserLoose({ required: false }), user: ct.resolvedMember({ required: false }) || ct.resolvedUserLoose({ required: false }),
}, },
async run({ message: msg, args, pluginData }) { async run({ message: msg, args, pluginData }) {
const user = args.user || msg.author; const user = args.user || msg.member || msg.author;
if (!(user instanceof UnknownUser)) { if (!(user instanceof UnknownUser)) {
const embed: APIEmbed = { const embed: APIEmbed = {
image: { image: {

View file

@ -46,7 +46,7 @@ export async function getUserInfoEmbed(
name: `${user.bot ? "Bot" : "User"}: ${renderUsername(user.username, user.discriminator)}`, name: `${user.bot ? "Bot" : "User"}: ${renderUsername(user.username, user.discriminator)}`,
}; };
const avatarURL = user.displayAvatarURL(); const avatarURL = (member || user).displayAvatarURL();
embed.author.icon_url = avatarURL; embed.author.icon_url = avatarURL;
if (compact) { if (compact) {

View file

@ -495,7 +495,7 @@ function formatSearchResultList(members: Array<GuildMember | User>): string {
line = `${paddedId} ${renderUserUsername(member.user)}`; line = `${paddedId} ${renderUserUsername(member.user)}`;
if (member.nickname) line += ` (${member.nickname})`; if (member.nickname) line += ` (${member.nickname})`;
} else { } else {
line = `${paddedId} ${member.tag}`; line = `${paddedId} ${renderUserUsername(member)}`;
} }
return line; return line;
}); });

View file

@ -1610,6 +1610,7 @@ export function renderUsername(username: string, discriminator: string): string
return `${username}#${discriminator}`; return `${username}#${discriminator}`;
} }
export function renderUserUsername(user: User | UnknownUser): string { export function renderUserUsername(user: GuildMember | User | UnknownUser): string {
user instanceof GuildMember ? (user = user.user) : null;
return renderUsername(user.username, user.discriminator); return renderUsername(user.username, user.discriminator);
} }

View file

@ -91,7 +91,7 @@ export class TemplateSafeMember extends TemplateSafeUser {
nick: string; nick: string;
roles: TemplateSafeRole[]; roles: TemplateSafeRole[];
joinedAt?: number; joinedAt?: number;
// guildAvatarURL: string, Once DJS supports per-server avatars guildAvatarURL: string;
guildName: string; guildName: string;
constructor(data: InputProps<TemplateSafeMember>) { constructor(data: InputProps<TemplateSafeMember>) {
@ -259,7 +259,7 @@ export function userToTemplateSafeUser(user: User | UnknownUser): TemplateSafeUs
discriminator: user.discriminator, discriminator: user.discriminator,
mention: `<@${user.id}>`, mention: `<@${user.id}>`,
tag: user.tag, tag: user.tag,
avatarURL: user.displayAvatarURL?.(), avatarURL: user.displayAvatarURL(),
bot: user.bot, bot: user.bot,
createdAt: user.createdTimestamp, createdAt: user.createdTimestamp,
renderedUsername: renderUserUsername(user), renderedUsername: renderUserUsername(user),
@ -285,6 +285,7 @@ export function memberToTemplateSafeMember(member: GuildMember | PartialGuildMem
nick: member.nickname ?? "*None*", nick: member.nickname ?? "*None*",
roles: [...member.roles.cache.mapValues((r) => roleToTemplateSafeRole(r)).values()], roles: [...member.roles.cache.mapValues((r) => roleToTemplateSafeRole(r)).values()],
joinedAt: member.joinedTimestamp ?? undefined, joinedAt: member.joinedTimestamp ?? undefined,
guildAvatarURL: member.displayAvatarURL(),
guildName: member.guild.name, guildName: member.guild.name,
}); });
} }