HidekoBot/src/main/java/wtf/beatrice/hidekobot/slashcommands/AvatarCommand.java

88 lines
2.8 KiB
Java
Raw Normal View History

2022-11-20 17:19:40 +01:00
package wtf.beatrice.hidekobot.slashcommands;
import net.dv8tion.jda.api.EmbedBuilder;
2022-11-20 17:19:40 +01:00
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.*;
2022-11-20 17:19:40 +01:00
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
public static final int[] acceptedSizes = { 16, 32, 64, 128, 256, 512, 1024 };
2022-11-20 17:19:40 +01:00
public AvatarCommand(@NotNull SlashCommandInteractionEvent event)
{
event.deferReply().queue();
User user;
int resolution;
2022-11-20 17:19:40 +01:00
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();
2022-11-20 17:19:40 +01:00
// method to find closest value to accepted values
int distance = Math.abs(acceptedSizes[0] - resolution);
2022-11-20 17:19:40 +01:00
int idx = 0;
for(int c = 1; c < acceptedSizes.length; c++){
int cdistance = Math.abs(acceptedSizes[c] - resolution);
2022-11-20 17:19:40 +01:00
if(cdistance < distance){
idx = c;
distance = cdistance;
}
}
resolution = acceptedSizes[idx];
2022-11-20 17:19:40 +01:00
} else {
resolution = 512;
}
EmbedBuilder embedBuilder = new EmbedBuilder();
2022-11-20 18:11:00 +01:00
// embed processing
{
embedBuilder.setColor(Color.PINK);
2022-11-20 18:22:32 +01:00
embedBuilder.setTitle("Profile picture");
embedBuilder.addField("User", "<@" + user.getId() + ">", false);
embedBuilder.addField("Current resolution", resolution + " × " + resolution, false);
2022-11-20 18:11:00 +01:00
// 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(")");
2022-11-20 18:11:00 +01:00
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));
2022-11-20 17:19:40 +01:00
}
event.getHook().editOriginalEmbeds(embedBuilder.build()).queue();
2022-11-20 17:19:40 +01:00
}
}