87 lines
2.9 KiB
Java
87 lines
2.9 KiB
Java
package wtf.beatrice.hidekobot.slashcommands;
|
||
|
||
import net.dv8tion.jda.api.EmbedBuilder;
|
||
import net.dv8tion.jda.api.entities.User;
|
||
import net.dv8tion.jda.api.events.interaction.command.SlashCommandInteractionEvent;
|
||
import net.dv8tion.jda.api.interactions.commands.OptionMapping;
|
||
import org.jetbrains.annotations.NotNull;
|
||
|
||
import java.awt.*;
|
||
|
||
public class AvatarCommand
|
||
{
|
||
// discord api returns a broken image if you don't use specific sizes (powers of 2), so we limit it to these
|
||
private final int[] acceptedSizes = { 16, 32, 64, 128, 256, 512, 1024 };
|
||
|
||
public AvatarCommand(@NotNull SlashCommandInteractionEvent event)
|
||
{
|
||
event.deferReply().queue();
|
||
|
||
User user;
|
||
int resolution;
|
||
|
||
OptionMapping userArg = event.getOption("user");
|
||
if(userArg != null)
|
||
{
|
||
user = userArg.getAsUser();
|
||
} else {
|
||
user = event.getUser();
|
||
}
|
||
|
||
OptionMapping sizeArg = event.getOption("size");
|
||
if(sizeArg != null)
|
||
{
|
||
resolution = sizeArg.getAsInt();
|
||
|
||
// method to find closest value to accepted values
|
||
int distance = Math.abs(acceptedSizes[0] - resolution);
|
||
int idx = 0;
|
||
for(int c = 1; c < acceptedSizes.length; c++){
|
||
int cdistance = Math.abs(acceptedSizes[c] - resolution);
|
||
if(cdistance < distance){
|
||
idx = c;
|
||
distance = cdistance;
|
||
}
|
||
}
|
||
resolution = acceptedSizes[idx];
|
||
|
||
} else {
|
||
resolution = 512;
|
||
}
|
||
|
||
EmbedBuilder embedBuilder = new EmbedBuilder();
|
||
|
||
// embed processing
|
||
{
|
||
embedBuilder.setColor(Color.PINK);
|
||
embedBuilder.setAuthor(event.getUser().getAsTag(), null, event.getUser().getEffectiveAvatarUrl());
|
||
embedBuilder.setTitle(user.getAsTag() + "'s profile picture");
|
||
|
||
embedBuilder.addField("Current resolution", resolution + " × " + resolution, false);
|
||
|
||
// string builder to create a string that links to all available resolutions
|
||
StringBuilder links = new StringBuilder();
|
||
for(int pos = 0; pos < acceptedSizes.length; pos++)
|
||
{
|
||
int currSize = acceptedSizes[pos];
|
||
|
||
String currLink = user.getEffectiveAvatar().getUrl(currSize);
|
||
|
||
links.append("[").append(currSize).append("px](").append(currLink).append(")");
|
||
if(pos + 1 != acceptedSizes.length) // don't add a separator on the last iteration
|
||
{
|
||
links.append(" | ");
|
||
}
|
||
}
|
||
|
||
embedBuilder.addField("Available resolutions", links.toString(), false);
|
||
|
||
|
||
|
||
embedBuilder.setImage(user.getEffectiveAvatar().getUrl(resolution));
|
||
}
|
||
|
||
event.getHook().editOriginalEmbeds(embedBuilder.build()).queue();
|
||
}
|
||
}
|