hammy
/
Enso-Bot
mirror of https://github.com/sgoudham/Enso-Bot.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Start adding Information Commands

Browse Source
java-rewrite
Hammy 3 years ago
parent 7becdc4528
commit c5bf382125
1 changed files with 227 additions and 0 deletions
Show Stats Download Patch File Download Diff File

227
src/main/java/me/goudham/bot/command/info/Info.java
Unescape Escape View File

@ -0,0 +1,227 @@
package me.goudham.bot.command.info;
import io.micronaut.context.annotation.Executable;
import jakarta.inject.Inject;
import java.awt.Color;
import java.util.EnumSet;
import java.util.List;
import java.util.StringJoiner;
import me.goudham.command.annotation.Option;
import me.goudham.command.annotation.SlashCommand;
import me.goudham.command.annotation.SubCommand;
import me.goudham.domain.Constants;
import me.goudham.service.EmbedService;
import net.dv8tion.jda.api.Permission;
import net.dv8tion.jda.api.entities.Guild;
import net.dv8tion.jda.api.entities.Member;
import net.dv8tion.jda.api.entities.MessageEmbed;
import net.dv8tion.jda.api.entities.Role;
import net.dv8tion.jda.api.entities.User;
import net.dv8tion.jda.api.events.interaction.SlashCommandEvent;
import net.dv8tion.jda.api.interactions.commands.OptionMapping;
import net.dv8tion.jda.api.interactions.commands.OptionType;
import net.dv8tion.jda.api.utils.concurrent.Task;
import org.jetbrains.annotations.NotNull;
@SlashCommand(name = "info")
public class Info {
private final EmbedService embedService;
@Inject
public Info(EmbedService embedService) {
this.embedService = embedService;
}
@SuppressWarnings("ConstantConditions")
@Executable
@SubCommand(
name = "user",
description = "Retrieve your or another member's information",
options = {
@Option(
optionType = OptionType.USER,
name = "member",
description = "A member within the server",
isRequired = false
)
}
)
public void userCommand(@NotNull SlashCommandEvent slashCommandEvent) {
OptionMapping optionalMember = slashCommandEvent.getOption("member");
Member member = optionalMember == null ? slashCommandEvent.getMember() : optionalMember.getAsMember();
MessageEmbed messageEmbed = embedService.getBaseEmbed()
.setTitle(getMemberOnlineStatus(member) + " " + member.getUser().getAsTag() + " " + getMemberBadges(member))
.setColor(getMemberColour(member))
.setThumbnail(member.getUser().getEffectiveAvatarUrl() + "?size=4096")
.addField("Registered", getRegisteredDate(member), false)
.addField("Joined", getJoinedDate(member), false)
.addField("Top Role", getTopRole(member), false)
.addField("Roles (" + member.getRoles().size() + ")", getMemberRoles(member, 20), false)
.setFooter("ID: " + member.getId())
.build();
slashCommandEvent.replyEmbeds(messageEmbed).queue();
}
@SuppressWarnings("ConstantConditions")
@Executable
@SubCommand(
name = "role",
description = "Retrieve your highest role or another role's information",
options = {
@Option(
optionType = OptionType.ROLE,
name = "role",
description = "A role within the server",
isRequired = false
)
}
)
public void roleCommand(@NotNull SlashCommandEvent slashCommandEvent) {
OptionMapping optionalRole = slashCommandEvent.getOption("role");
Guild guild = slashCommandEvent.getGuild();
Role role;
if (optionalRole == null) {
Member member = slashCommandEvent.getMember();
if (member == null) return;
if (member.getRoles().isEmpty()) {
slashCommandEvent.reply("No Role Given").queue();
return;
} else {
role = member.getRoles().get(0);
}
} else {
role = optionalRole.getAsRole();
}
Task<List<Member>> membersWithRoles = guild.findMembersWithRoles(role);
String roleColour = role.getColor() == null ? "N/A" : getColorAsHex(role.getColor());
String isMentionable = role.isMentionable() ? Constants.CHECK : Constants.CROSS;
String isHoisted = role.isHoisted() ? Constants.CHECK : Constants.CROSS;
String isManaged = role.isManaged() ? Constants.CHECK : Constants.CROSS;
String miscString = "Mentionable: " + isMentionable
+ "\n" + "Hoisted: " + isHoisted
+ "\n" + "Managed: " + isManaged;
EnumSet<Permission> permissions = role.getPermissions(slashCommandEvent.getGuildChannel());
System.out.println(permissions);
membersWithRoles.onSuccess(members -> {
long humanCount = members.stream().filter(member -> !member.getUser().isBot()).count();
long botCount = members.stream().filter(member -> member.getUser().isBot()).count();
MessageEmbed messageEmbed = embedService.getBaseEmbed()
.setTitle(role.getName() + " Information")
.setDescription(role.getAsMention() + "\n" + "**Colour:** " + roleColour)
.setColor(role.getColor())
.setThumbnail(guild.getIconUrl() + "?size=4096")
.addField("Creation At", getCreationDate(role), false)
.addField("Members (" + members.size() + ")", "Humans: " + humanCount + "\nBots: " + botCount, true)
.addField("Misc", miscString, true)
.addField("List of Members (" + members.size() + ")", getListOfMembers(members, 20), false)
.setFooter("ID: " + role.getId())
.build();
slashCommandEvent.replyEmbeds(messageEmbed).queue();
});
}
private @NotNull String getColorAsHex (@NotNull Color color) {
return "#" + Integer.toHexString(color.getRGB()).toUpperCase();
}
private Color getMemberColour(@NotNull Member member) {
return member.getColor() == null ? Color.BLACK : member.getColor();
}
private @NotNull String getMemberBadges(@NotNull Member member) {
User user = member.getUser();
String memberBadges = "";
if (user.isBot()) memberBadges += Constants.BADGE_BOT;
if (member.getTimeBoosted() != null) memberBadges += Constants.BADGE_SERVER_BOOST;
memberBadges += user.getFlags().toString()
.substring(1, user.getFlags().toString().length() - 1)
.replace(",", "")
.replace("PARTNER", Constants.BADGE_PARTNER)
.replace("HYPESQUAD_BRAVERY", Constants.BADGE_BRAVERY)
.replace("HYPESQUAD_BRILLIANCE", Constants.BADGE_BRILLIANCE)
.replace("HYPESQUAD_BALANCE", Constants.BADGE_BALANCE)
.replace("VERIFIED_DEVELOPER", Constants.BADGE_EARLY_VERIFIED_BOT_DEVELOPER)
.replace("EARLY_SUPPORTER", Constants.BADGE_EARLY_SUPPORTER)
.replace("SYSTEM", Constants.BADGE_STAFF)
.replace("BUG_HUNTER_LEVEL_1", Constants.BADGE_BUG_HUNTER)
.replace("BUG_HUNTER_LEVEL_2", Constants.BADGE_BUG_HUNTER)
.replace("VERIFIED_BOT", Constants.BADGE_VERIFIED_BOT);
return memberBadges;
}
private @NotNull String getMemberOnlineStatus(@NotNull Member member) {
return member.getOnlineStatus().toString()
.replace("ONLINE", Constants.STATUS_ONLINE)
.replace("IDLE", Constants.STATUS_IDLE)
.replace("DO_NOT_DISTURB", Constants.STATUS_DND)
.replace("OFFLINE", Constants.STATUS_OFFLINE)
.replace("STREAMING", Constants.STATUS_STREAMING);
}
private @NotNull String getTopRole(@NotNull Member member) {
List<Role> memberRoles = member.getRoles();
if (memberRoles.isEmpty()) return "No Roles";
return memberRoles.get(0).getAsMention();
}
private String getListOfMembers(@NotNull List<Member> members, int limit) {
StringJoiner memberJoiner = new StringJoiner(" **|** ");
if (members.isEmpty()) return "No Members In Role";
members.stream()
.limit(limit)
.forEach(member -> memberJoiner.add(member.getAsMention()));
if (members.size() > limit) {
int leftOverMembers = members.size() - limit;
memberJoiner.add(" and ** " + leftOverMembers + " ** more ");
}
return memberJoiner.toString();
}
private String getMemberRoles(@NotNull Member member, int limit) {
StringJoiner memberRolesJoiner = new StringJoiner(" **|** ");
List<Role> memberRoles = member.getRoles();
if (memberRoles.isEmpty()) return "No Roles";
memberRoles.stream()
.limit(limit)
.forEach(role -> memberRolesJoiner.add(role.getAsMention()));
if (memberRoles.size() > limit) {
int leftOverRoles = memberRoles.size() - limit;
memberRolesJoiner.add(" and ** " + leftOverRoles + " ** more ");
}
return memberRolesJoiner.toString();
}
private @NotNull String getCreationDate(@NotNull Role role) {
long timeCreated = role.getTimeCreated().toInstant().getEpochSecond();
return "<t:" + timeCreated + ":F>";
}
private @NotNull String getRegisteredDate(@NotNull Member member) {
long timeCreated = member.getTimeCreated().toInstant().getEpochSecond();
return "<t:" + timeCreated + ":F>";
}
private @NotNull String getJoinedDate(@NotNull Member member) {
long timeJoined = member.getTimeJoined().toInstant().getEpochSecond();
return "<t:" + timeJoined + ":F>";
}
}
Write Preview
Loading…
Cancel
Save