private async Task AvatarCommandTree(AvatarLocation location, Context ctx, PKMember target,
MemberGuildSettings?guildData)
{
// First, see if we need to *clear*
if (await ctx.MatchClear(location == AvatarLocation.Server
? "this member's server avatar"
: "this member's avatar"))
{
ctx.CheckSystem().CheckOwnMember(target);
await AvatarClear(location, ctx, target, guildData);
return;
}
// Then, parse an image from the command (from various sources...)
var avatarArg = await ctx.MatchImage();
if (avatarArg == null)
{
// If we didn't get any, just show the current avatar
await AvatarShow(location, ctx, target, guildData);
return;
}
ctx.CheckSystem().CheckOwnMember(target);
await AvatarUtils.VerifyAvatarOrThrow(_client, avatarArg.Value.Url);
await UpdateAvatar(location, ctx, target, avatarArg.Value.Url);
await PrintResponse(location, ctx, target, avatarArg.Value, guildData);
}
private async Task AvatarClear(AvatarLocation location, Context ctx, PKMember target, MemberGuildSettings?mgs)
{
await UpdateAvatar(location, ctx, target, null);
if (location == AvatarLocation.Server)
{
if (target.AvatarUrl != null)
{
await ctx.Reply(
$"{Emojis.Success} Member server avatar cleared. This member will now use the global avatar in this server (**{ctx.Guild.Name}**).");
}
else
{
await ctx.Reply($"{Emojis.Success} Member server avatar cleared. This member now has no avatar.");
}
}
else
{
if (mgs?.AvatarUrl != null)
{
await ctx.Reply(
$"{Emojis.Success} Member avatar cleared. Note that this member has a server-specific avatar set here, type `pk;member {target.Reference(ctx)} serveravatar clear` if you wish to clear that too.");
}
else
{
await ctx.Reply($"{Emojis.Success} Member avatar cleared.");
}
}
}
private Task PrintResponse(AvatarLocation location, Context ctx, PKMember target, ParsedImage avatar,
MemberGuildSettings?targetGuildData)
{
var typeFrag = location switch
{
AvatarLocation.Server => "server avatar",
AvatarLocation.Member => "avatar",
_ => throw new ArgumentOutOfRangeException(nameof(location))
};
var serverFrag = location switch
{
AvatarLocation.Server => $" This avatar will now be used when proxying in this server (**{ctx.Guild.Name}**).",
AvatarLocation.Member when targetGuildData?.AvatarUrl != null => $"\n{Emojis.Note} Note that this member *also* has a server-specific avatar set in this server (**{ctx.Guild.Name}**), and thus changing the global avatar will have no effect here.",
_ => ""
};
var msg = avatar.Source switch
{
AvatarSource.User => $"{Emojis.Success} Member {typeFrag} changed to {avatar.SourceUser?.Username}'s avatar!{serverFrag}\n{Emojis.Warn} If {avatar.SourceUser?.Username} changes their avatar, the member's avatar will need to be re-set.",
AvatarSource.Url => $"{Emojis.Success} Member {typeFrag} changed to the image at the given URL.{serverFrag}",
AvatarSource.Attachment => $"{Emojis.Success} Member {typeFrag} changed to attached image.{serverFrag}\n{Emojis.Warn} If you delete the message containing the attachment, the avatar will stop working.",
_ => throw new ArgumentOutOfRangeException()
};
// The attachment's already right there, no need to preview it.
var hasEmbed = avatar.Source != AvatarSource.Attachment;
return(hasEmbed
? ctx.Reply(msg, embed: new DiscordEmbedBuilder().WithImageUrl(avatar.Url).Build())
: ctx.Reply(msg));
}
private async Task AvatarShow(AvatarLocation location, Context ctx, PKMember target, MemberGuildSettings?guildData)
{
var currentValue = location == AvatarLocation.Member ? target.AvatarUrl : guildData?.AvatarUrl;
var canAccess = location != AvatarLocation.Member || target.AvatarPrivacy.CanAccess(ctx.LookupContextFor(target));
if (string.IsNullOrEmpty(currentValue) || !canAccess)
{
if (location == AvatarLocation.Member)
{
if (target.System == ctx.System?.Id)
{
throw new PKSyntaxError("This member does not have an avatar set. Set one by attaching an image to this command, or by passing an image URL or @mention.");
}
throw new PKError("This member does not have an avatar set.");
}
if (location == AvatarLocation.Server)
{
throw new PKError($"This member does not have a server avatar set. Type `pk;member {target.Hid} avatar` to see their global avatar.");
}
}
var field = location == AvatarLocation.Server ? $"server avatar (for {ctx.Guild.Name})" : "avatar";
var cmd = location == AvatarLocation.Server ? "serveravatar" : "avatar";
var eb = new DiscordEmbedBuilder()
.WithTitle($"{target.NameFor(ctx)}'s {field}")
.WithImageUrl(currentValue);
if (target.System == ctx.System?.Id)
{
eb.WithDescription($"To clear, use `pk;member {target.Hid} {cmd} clear`.");
}
await ctx.Reply(embed : eb.Build());
}
private async Task AvatarCommandTree(AvatarLocation location, Context ctx, PKMember target, MemberGuildSettings?guildData)
{
if (ctx.Match("clear", "remove", "reset") || ctx.MatchFlag("c", "clear"))
{
await AvatarClear(location, ctx, target, guildData);
}
else if (ctx.RemainderOrNull() == null && ctx.Message.Attachments.Count == 0)
{
await AvatarShow(location, ctx, target, guildData);
}
else if (await ctx.MatchUser() is {} user)
{
await AvatarFromUser(location, ctx, target, user);
}
private async Task AvatarFromAttachment(AvatarLocation location, Context ctx, PKMember target, DiscordAttachment attachment)
{
ctx.CheckSystem().CheckOwnMember(target);
await AvatarUtils.VerifyAvatarOrThrow(attachment.Url);
await UpdateAvatar(location, ctx, target, attachment.Url);
if (location == AvatarLocation.Server)
{
await ctx.Reply($"{Emojis.Success} Member server avatar changed to attached image. This avatar will now be used when proxying in this server (**{ctx.Guild.Name}**). Please note that if you delete the message containing the attachment, the avatar will stop working.");
}
else if (location == AvatarLocation.Member)
{
await ctx.Reply($"{Emojis.Success} Member avatar changed to attached image. Please note that if you delete the message containing the attachment, the avatar will stop working.");
}
}
private Task UpdateAvatar(AvatarLocation location, Context ctx, PKMember target, string?url)
{
switch (location)
{
case AvatarLocation.Server:
return(ctx.Repository.UpdateMemberGuild(target.Id, ctx.Guild.Id, new MemberGuildPatch {
AvatarUrl = url
}));
case AvatarLocation.Member:
return(ctx.Repository.UpdateMember(target.Id, new MemberPatch {
AvatarUrl = url
}));
default:
throw new ArgumentOutOfRangeException($"Unknown avatar location {location}");
}
}
private async Task AvatarFromArg(AvatarLocation location, Context ctx, PKMember target, string url)
{
ctx.CheckSystem().CheckOwnMember(target);
if (url.Length > Limits.MaxUriLength)
{
throw Errors.InvalidUrl(url);
}
await AvatarUtils.VerifyAvatarOrThrow(url);
await UpdateAvatar(location, ctx, target, url);
var embed = new DiscordEmbedBuilder().WithImageUrl(url).Build();
if (location == AvatarLocation.Server)
{
await ctx.Reply($"{Emojis.Success} Member server avatar changed. This avatar will now be used when proxying in this server (**{ctx.Guild.Name}**).", embed : embed);
}
}
private async Task AvatarShow(AvatarLocation location, Context ctx, PKMember target,
MemberGuildSettings?guildData)
{
// todo: this privacy code is really confusing
// for now, we skip privacy flag/config parsing for this, but it would be good to fix that at some point
var currentValue = location == AvatarLocation.Member ? target.AvatarUrl : guildData?.AvatarUrl;
var canAccess = location != AvatarLocation.Member ||
target.AvatarPrivacy.CanAccess(ctx.DirectLookupContextFor(target.System));
if (string.IsNullOrEmpty(currentValue) || !canAccess)
{
if (location == AvatarLocation.Member)
{
if (target.System == ctx.System?.Id)
{
throw new PKSyntaxError(
"This member does not have an avatar set. Set one by attaching an image to this command, or by passing an image URL or @mention.");
}
throw new PKError("This member does not have an avatar set.");
}
if (location == AvatarLocation.Server)
{
throw new PKError(
$"This member does not have a server avatar set. Type `pk;member {target.Reference(ctx)} avatar` to see their global avatar.");
}
}
var field = location == AvatarLocation.Server ? $"server avatar (for {ctx.Guild.Name})" : "avatar";
var cmd = location == AvatarLocation.Server ? "serveravatar" : "avatar";
var eb = new EmbedBuilder()
.Title($"{target.NameFor(ctx)}'s {field}")
.Image(new Embed.EmbedImage(currentValue?.TryGetCleanCdnUrl()));
if (target.System == ctx.System?.Id)
{
eb.Description($"To clear, use `pk;member {target.Reference(ctx)} {cmd} clear`.");
}
await ctx.Reply(embed : eb.Build());
}
private Task UpdateAvatar(AvatarLocation location, Context ctx, PKMember target, string?url)
{
switch (location)
{
case AvatarLocation.Server:
var serverPatch = new MemberGuildPatch {
AvatarUrl = url
};
return(_db.Execute(c => c.UpsertMemberGuild(target.Id, ctx.Guild.Id, serverPatch)));
case AvatarLocation.Member:
var memberPatch = new MemberPatch {
AvatarUrl = url
};
return(_db.Execute(c => c.UpdateMember(target.Id, memberPatch)));
default:
throw new ArgumentOutOfRangeException($"Unknown avatar location {location}");
}
}
private async Task AvatarFromUser(AvatarLocation location, Context ctx, PKMember target, DiscordUser user)
{
ctx.CheckSystem().CheckOwnMember(target);
if (user.AvatarHash == null)
{
throw Errors.UserHasNoAvatar;
}
var url = user.GetAvatarUrl(ImageFormat.Png, 256);
await UpdateAvatar(location, ctx, target, url);
var embed = new DiscordEmbedBuilder().WithImageUrl(url).Build();
if (location == AvatarLocation.Server)
{
await ctx.Reply($"{Emojis.Success} Member server avatar changed to {user.Username}'s avatar! This avatar will now be used when proxying in this server (**{ctx.Guild.Name}**). {Emojis.Warn} Please note that if {user.Username} changes their avatar, the member's server avatar will need to be re-set.", embed : embed);
}
else if (location == AvatarLocation.Member)
{
await ctx.Reply($"{Emojis.Success} Member avatar changed to {user.Username}'s avatar! {Emojis.Warn} Please note that if {user.Username} changes their avatar, the member's avatar will need to be re-set.", embed : embed);
}
}
