mirror of https://github.com/sgoudham/Enso-Bot.git
Add InfoUtil interface
parent
bef7a2050e
commit
b916bd1dcf
@ -0,0 +1,204 @@
|
|||||||
|
package me.goudham.util;
|
||||||
|
|
||||||
|
import jakarta.inject.Singleton;
|
||||||
|
import java.awt.Color;
|
||||||
|
import java.util.List;
|
||||||
|
import java.util.StringJoiner;
|
||||||
|
import me.goudham.domain.Constants;
|
||||||
|
import net.dv8tion.jda.api.OnlineStatus;
|
||||||
|
import net.dv8tion.jda.api.entities.ChannelType;
|
||||||
|
import net.dv8tion.jda.api.entities.Emote;
|
||||||
|
import net.dv8tion.jda.api.entities.Guild;
|
||||||
|
import net.dv8tion.jda.api.entities.Member;
|
||||||
|
import net.dv8tion.jda.api.entities.Role;
|
||||||
|
import net.dv8tion.jda.api.entities.TextChannel;
|
||||||
|
import net.dv8tion.jda.api.entities.User;
|
||||||
|
import net.dv8tion.jda.api.entities.VoiceChannel;
|
||||||
|
import net.dv8tion.jda.api.utils.cache.SnowflakeCacheView;
|
||||||
|
import net.dv8tion.jda.api.utils.cache.SortedSnowflakeCacheView;
|
||||||
|
import org.jetbrains.annotations.NotNull;
|
||||||
|
|
||||||
|
@Singleton
|
||||||
|
public class EnsoInfoUtil implements InfoUtil {
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public @NotNull String getColorAsHex(@NotNull Color color) {
|
||||||
|
return "#" + Integer.toHexString(color.getRGB()).toUpperCase();
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public Color getMemberColour(@NotNull Member member) {
|
||||||
|
return member.getColor() == null ? Color.BLACK : member.getColor();
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public @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;
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public @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);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public @NotNull String getTopRole(@NotNull Object object) {
|
||||||
|
String topRole = "";
|
||||||
|
|
||||||
|
if (object instanceof Member member) {
|
||||||
|
List<Role> memberRoles = member.getRoles();
|
||||||
|
if (memberRoles.isEmpty()) return "No Roles";
|
||||||
|
topRole = memberRoles.get(0).getAsMention();
|
||||||
|
} else if (object instanceof Guild guild) {
|
||||||
|
StringJoiner topRoleJoiner = new StringJoiner("");
|
||||||
|
guild.getRoleCache().stream()
|
||||||
|
.findFirst()
|
||||||
|
.ifPresentOrElse(
|
||||||
|
role -> topRoleJoiner.add(role.getAsMention()),
|
||||||
|
() -> topRoleJoiner.setEmptyValue("No Roles")
|
||||||
|
);
|
||||||
|
topRole = topRoleJoiner.toString();
|
||||||
|
}
|
||||||
|
|
||||||
|
return topRole;
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public String getListOfMembers(@NotNull List<Member> members, int limit) {
|
||||||
|
StringJoiner memberJoiner = new StringJoiner(" **|** ");
|
||||||
|
if (members.isEmpty()) return "No Members";
|
||||||
|
|
||||||
|
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();
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public String getGuildEmotes(@NotNull Guild guild, int limit) {
|
||||||
|
SnowflakeCacheView<Emote> emoteCache = guild.getEmoteCache();
|
||||||
|
|
||||||
|
StringJoiner guildEmoteJoiner = new StringJoiner(" ");
|
||||||
|
if (emoteCache.isEmpty()) {
|
||||||
|
guildEmoteJoiner.setEmptyValue("No Emojis");
|
||||||
|
} else {
|
||||||
|
emoteCache.stream()
|
||||||
|
.limit(limit)
|
||||||
|
.forEach(emote -> guildEmoteJoiner.add(emote.getAsMention()));
|
||||||
|
}
|
||||||
|
if (emoteCache.size() > limit) {
|
||||||
|
long leftOverEmojis = emoteCache.size() - limit;
|
||||||
|
guildEmoteJoiner.add(" and ** " + leftOverEmojis + " ** more");
|
||||||
|
}
|
||||||
|
|
||||||
|
return guildEmoteJoiner.toString();
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public String getGuildRoles(@NotNull Guild guild, int limit) {
|
||||||
|
SortedSnowflakeCacheView<Role> roleCache = guild.getRoleCache();
|
||||||
|
|
||||||
|
StringJoiner guildRolesJoiner = new StringJoiner(" **|** ");
|
||||||
|
if (roleCache.isEmpty()) {
|
||||||
|
guildRolesJoiner.setEmptyValue("No Roles");
|
||||||
|
} else {
|
||||||
|
roleCache.stream()
|
||||||
|
.limit(limit)
|
||||||
|
.forEach(role -> guildRolesJoiner.add(role.getAsMention()));
|
||||||
|
}
|
||||||
|
if (roleCache.size() > limit) {
|
||||||
|
long leftOverRoles = roleCache.size() - limit;
|
||||||
|
guildRolesJoiner.add(" and ** " + leftOverRoles + " ** more ");
|
||||||
|
}
|
||||||
|
|
||||||
|
return guildRolesJoiner.toString();
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public 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();
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public long getMemberStatusCount(@NotNull Guild guild, OnlineStatus onlineStatus) {
|
||||||
|
return guild.getMemberCache().stream()
|
||||||
|
.filter(member -> member.getOnlineStatus() == onlineStatus)
|
||||||
|
.count();
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public long getChannelCount(@NotNull Guild guild, ChannelType channelType) {
|
||||||
|
return guild.getChannels().stream()
|
||||||
|
.filter(guildChannel -> guildChannel.getType() == channelType)
|
||||||
|
.count();
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public @NotNull String getJoinedDate(@NotNull Member member) {
|
||||||
|
long timeJoined = member.getTimeJoined().toInstant().getEpochSecond();
|
||||||
|
return "<t:" + timeJoined + ":F>";
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public @NotNull String getCreationDate(Object object) {
|
||||||
|
long time = 0L;
|
||||||
|
|
||||||
|
if (object instanceof Role role) {
|
||||||
|
time = role.getTimeCreated().toInstant().getEpochSecond();
|
||||||
|
} else if (object instanceof Member member) {
|
||||||
|
time = member.getTimeCreated().toInstant().getEpochSecond();
|
||||||
|
} else if (object instanceof Guild guild) {
|
||||||
|
time = guild.getTimeCreated().toInstant().getEpochSecond();
|
||||||
|
} else if (object instanceof VoiceChannel voiceChannel) {
|
||||||
|
time = voiceChannel.getTimeCreated().toInstant().getEpochSecond();
|
||||||
|
} else if (object instanceof TextChannel textChannel) {
|
||||||
|
time = textChannel.getTimeCreated().toInstant().getEpochSecond();
|
||||||
|
}
|
||||||
|
|
||||||
|
return "<t:" + time + ":F>";
|
||||||
|
}
|
||||||
|
}
|
||||||
@ -0,0 +1,25 @@
|
|||||||
|
package me.goudham.util;
|
||||||
|
|
||||||
|
import java.awt.Color;
|
||||||
|
import java.util.List;
|
||||||
|
import net.dv8tion.jda.api.OnlineStatus;
|
||||||
|
import net.dv8tion.jda.api.entities.ChannelType;
|
||||||
|
import net.dv8tion.jda.api.entities.Guild;
|
||||||
|
import net.dv8tion.jda.api.entities.Member;
|
||||||
|
import org.jetbrains.annotations.NotNull;
|
||||||
|
|
||||||
|
public interface InfoUtil {
|
||||||
|
@NotNull String getColorAsHex(@NotNull Color color);
|
||||||
|
Color getMemberColour(@NotNull Member member);
|
||||||
|
@NotNull String getMemberBadges(@NotNull Member member);
|
||||||
|
@NotNull String getMemberOnlineStatus(@NotNull Member member);
|
||||||
|
@NotNull String getTopRole(@NotNull Object object);
|
||||||
|
String getListOfMembers(@NotNull List<Member> members, int limit);
|
||||||
|
String getGuildEmotes(@NotNull Guild guild, int limit);
|
||||||
|
String getGuildRoles(@NotNull Guild guild, int limit);
|
||||||
|
String getMemberRoles(@NotNull Member member, int limit);
|
||||||
|
long getMemberStatusCount(@NotNull Guild guild, OnlineStatus onlineStatus);
|
||||||
|
long getChannelCount(@NotNull Guild guild, ChannelType channelType);
|
||||||
|
@NotNull String getJoinedDate(@NotNull Member member);
|
||||||
|
@NotNull String getCreationDate(Object object);
|
||||||
|
}
|
||||||
Loading…
Reference in New Issue