forked from SteamWar/SteamWar
Add bust to RatsChannel threads
This commit is contained in:
@@ -38,6 +38,7 @@ import net.dv8tion.jda.api.entities.Activity;
|
|||||||
import net.dv8tion.jda.api.entities.Guild;
|
import net.dv8tion.jda.api.entities.Guild;
|
||||||
import net.dv8tion.jda.api.entities.Member;
|
import net.dv8tion.jda.api.entities.Member;
|
||||||
import net.dv8tion.jda.api.entities.Role;
|
import net.dv8tion.jda.api.entities.Role;
|
||||||
|
import net.dv8tion.jda.api.entities.channel.concrete.ThreadChannel;
|
||||||
import net.dv8tion.jda.api.entities.emoji.Emoji;
|
import net.dv8tion.jda.api.entities.emoji.Emoji;
|
||||||
import net.dv8tion.jda.api.events.message.MessageReceivedEvent;
|
import net.dv8tion.jda.api.events.message.MessageReceivedEvent;
|
||||||
import net.dv8tion.jda.api.exceptions.ErrorResponseException;
|
import net.dv8tion.jda.api.exceptions.ErrorResponseException;
|
||||||
@@ -190,9 +191,17 @@ public class DiscordBot {
|
|||||||
new DiscordTeamEvent(),
|
new DiscordTeamEvent(),
|
||||||
new ChannelListener(),
|
new ChannelListener(),
|
||||||
new DiscordSchemUpload(),
|
new DiscordSchemUpload(),
|
||||||
new RatsChannel()
|
new RatsChannel.RastChannelListener()
|
||||||
);
|
);
|
||||||
|
|
||||||
|
config.getRatsRoleToThread().forEach((roleId, threadId) -> {
|
||||||
|
Role role = DiscordBot.getGuild().getRoleById(roleId);
|
||||||
|
if (role == null) return;
|
||||||
|
ThreadChannel threadChannel = DiscordBot.getGuild().getThreadChannelById(threadId);
|
||||||
|
if (threadChannel == null) return;
|
||||||
|
new RatsChannel(role, threadChannel);
|
||||||
|
});
|
||||||
|
|
||||||
commandSetup(jda.retrieveCommands().complete(), jda.updateCommands());
|
commandSetup(jda.retrieveCommands().complete(), jda.updateCommands());
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
@@ -19,11 +19,11 @@
|
|||||||
|
|
||||||
package de.steamwar.velocitycore.discord.channels;
|
package de.steamwar.velocitycore.discord.channels;
|
||||||
|
|
||||||
import de.steamwar.velocitycore.discord.DiscordBot;
|
|
||||||
import de.steamwar.velocitycore.discord.listeners.ChannelListener;
|
|
||||||
import de.steamwar.messages.Chatter;
|
import de.steamwar.messages.Chatter;
|
||||||
import de.steamwar.messages.Message;
|
import de.steamwar.messages.Message;
|
||||||
import de.steamwar.sql.SteamwarUser;
|
import de.steamwar.sql.SteamwarUser;
|
||||||
|
import de.steamwar.velocitycore.discord.DiscordBot;
|
||||||
|
import de.steamwar.velocitycore.discord.listeners.ChannelListener;
|
||||||
import lombok.AllArgsConstructor;
|
import lombok.AllArgsConstructor;
|
||||||
import lombok.Getter;
|
import lombok.Getter;
|
||||||
import net.dv8tion.jda.api.entities.User;
|
import net.dv8tion.jda.api.entities.User;
|
||||||
@@ -31,7 +31,6 @@ import net.dv8tion.jda.api.entities.channel.middleman.MessageChannel;
|
|||||||
import net.dv8tion.jda.api.events.interaction.component.GenericComponentInteractionCreateEvent;
|
import net.dv8tion.jda.api.events.interaction.component.GenericComponentInteractionCreateEvent;
|
||||||
import net.dv8tion.jda.api.events.message.MessageReceivedEvent;
|
import net.dv8tion.jda.api.events.message.MessageReceivedEvent;
|
||||||
import net.dv8tion.jda.api.utils.messages.MessageCreateBuilder;
|
import net.dv8tion.jda.api.utils.messages.MessageCreateBuilder;
|
||||||
import net.dv8tion.jda.api.utils.messages.MessageCreateData;
|
|
||||||
import net.kyori.adventure.text.Component;
|
import net.kyori.adventure.text.Component;
|
||||||
import net.kyori.adventure.text.serializer.plain.PlainTextComponentSerializer;
|
import net.kyori.adventure.text.serializer.plain.PlainTextComponentSerializer;
|
||||||
|
|
||||||
@@ -57,6 +56,11 @@ public class DiscordChannel extends Chatter.PlayerlessChatter {
|
|||||||
ChannelListener.getChannels().put(this.channel, this);
|
ChannelListener.getChannels().put(this.channel, this);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
public DiscordChannel(MessageChannel channel) {
|
||||||
|
this(SteamwarUser.get(-1), channel);
|
||||||
|
ChannelListener.getChannels().put(this.channel, this);
|
||||||
|
}
|
||||||
|
|
||||||
public void send(String message) {
|
public void send(String message) {
|
||||||
send(new MessageCreateBuilder()
|
send(new MessageCreateBuilder()
|
||||||
.setContent(message
|
.setContent(message
|
||||||
|
|||||||
@@ -2,37 +2,46 @@ package de.steamwar.velocitycore.discord.channels;
|
|||||||
|
|
||||||
import de.steamwar.sql.SteamwarUser;
|
import de.steamwar.sql.SteamwarUser;
|
||||||
import de.steamwar.velocitycore.discord.DiscordBot;
|
import de.steamwar.velocitycore.discord.DiscordBot;
|
||||||
|
import net.dv8tion.jda.api.EmbedBuilder;
|
||||||
import net.dv8tion.jda.api.entities.Member;
|
import net.dv8tion.jda.api.entities.Member;
|
||||||
import net.dv8tion.jda.api.entities.Message;
|
|
||||||
import net.dv8tion.jda.api.entities.Role;
|
import net.dv8tion.jda.api.entities.Role;
|
||||||
import net.dv8tion.jda.api.entities.channel.concrete.ThreadChannel;
|
import net.dv8tion.jda.api.entities.channel.concrete.ThreadChannel;
|
||||||
import net.dv8tion.jda.api.events.guild.member.GuildMemberRoleAddEvent;
|
import net.dv8tion.jda.api.events.guild.member.GuildMemberRoleAddEvent;
|
||||||
import net.dv8tion.jda.api.events.guild.member.GuildMemberRoleRemoveEvent;
|
import net.dv8tion.jda.api.events.guild.member.GuildMemberRoleRemoveEvent;
|
||||||
import net.dv8tion.jda.api.hooks.ListenerAdapter;
|
import net.dv8tion.jda.api.hooks.ListenerAdapter;
|
||||||
import net.dv8tion.jda.api.utils.AttachedFile;
|
import net.dv8tion.jda.api.utils.messages.MessageCreateBuilder;
|
||||||
import net.dv8tion.jda.api.utils.FileUpload;
|
|
||||||
import org.jetbrains.annotations.NotNull;
|
import org.jetbrains.annotations.NotNull;
|
||||||
|
|
||||||
import java.io.IOException;
|
import java.util.HashMap;
|
||||||
import java.io.InputStream;
|
import java.util.List;
|
||||||
import java.net.URL;
|
import java.util.Map;
|
||||||
import java.util.*;
|
|
||||||
import java.util.stream.Collectors;
|
|
||||||
|
|
||||||
public class RatsChannel extends ListenerAdapter {
|
public class RatsChannel extends StaticMessageChannel {
|
||||||
|
|
||||||
private Map<Role, ThreadChannel> ratsChannel = new HashMap<>();
|
private static Map<Role, RatsChannel> listener = new HashMap<>();
|
||||||
|
|
||||||
public RatsChannel() {
|
public RatsChannel(Role role, ThreadChannel threadChannel) {
|
||||||
DiscordBot.getInstance().getConfig().getRatsRoleToThread().forEach((roleId, threadId) -> {
|
super(threadChannel, () -> {
|
||||||
Role role = DiscordBot.getGuild().getRoleById(roleId);
|
MessageCreateBuilder messageCreateBuilder = new MessageCreateBuilder();
|
||||||
if (role != null) return;
|
messageCreateBuilder.setContent("# Ratsmitglieder");
|
||||||
ThreadChannel threadChannel = DiscordBot.getGuild().getThreadChannelById(threadId);
|
|
||||||
if (threadChannel != null) ratsChannel.put(role, threadChannel);
|
for (Member member : DiscordBot.getGuild().getMembersWithRoles(role)) {
|
||||||
ratsChannel.put(role, threadChannel);
|
SteamwarUser steamwarUser = SteamwarUser.get(member.getIdLong());
|
||||||
});
|
if (steamwarUser == null) continue;
|
||||||
|
|
||||||
|
messageCreateBuilder.addEmbeds(new EmbedBuilder()
|
||||||
|
.setTitle(steamwarUser.getUserName())
|
||||||
|
.setImage("api.steamwar.de/data/skin/" + steamwarUser.getUUID().toString())
|
||||||
|
.build());
|
||||||
}
|
}
|
||||||
|
|
||||||
|
return messageCreateBuilder;
|
||||||
|
}, genericComponentInteractionCreateEvent -> {});
|
||||||
|
|
||||||
|
listener.put(role, this);
|
||||||
|
}
|
||||||
|
|
||||||
|
public static class RastChannelListener extends ListenerAdapter {
|
||||||
@Override
|
@Override
|
||||||
public void onGuildMemberRoleAdd(@NotNull GuildMemberRoleAddEvent event) {
|
public void onGuildMemberRoleAdd(@NotNull GuildMemberRoleAddEvent event) {
|
||||||
updateRatsChannel(event.getRoles());
|
updateRatsChannel(event.getRoles());
|
||||||
@@ -44,65 +53,11 @@ public class RatsChannel extends ListenerAdapter {
|
|||||||
}
|
}
|
||||||
|
|
||||||
private void updateRatsChannel(List<Role> roles) {
|
private void updateRatsChannel(List<Role> roles) {
|
||||||
for (Role role : roles) {
|
roles.forEach(role -> {
|
||||||
if (!ratsChannel.containsKey(role)) continue;
|
if (listener.containsKey(role)) {
|
||||||
ThreadChannel channel = ratsChannel.get(role);
|
listener.get(role).update();
|
||||||
|
|
||||||
List<Map.Entry<SteamwarUser, InputStream>> ratMembers = loadImages(role)
|
|
||||||
.entrySet()
|
|
||||||
.stream()
|
|
||||||
.sorted(Comparator.comparing(value -> value.getKey().getUserName()))
|
|
||||||
.collect(Collectors.toList());
|
|
||||||
|
|
||||||
List<Message> messages = new ArrayList<>(channel.getHistory()
|
|
||||||
.retrievePast(100)
|
|
||||||
.complete());
|
|
||||||
// Need to reverse since they are provided from newest (0) to latest (size() - 1)
|
|
||||||
Collections.reverse(messages);
|
|
||||||
|
|
||||||
Iterator<Message> iterator = messages.iterator();
|
|
||||||
if (!iterator.hasNext()) {
|
|
||||||
channel.sendMessage("# Ratsmitglieder")
|
|
||||||
.queue();
|
|
||||||
} else {
|
|
||||||
iterator.next();
|
|
||||||
}
|
}
|
||||||
|
});
|
||||||
for (Map.Entry<SteamwarUser, InputStream> entry : ratMembers) {
|
|
||||||
if (iterator.hasNext()) {
|
|
||||||
Message message = iterator.next();
|
|
||||||
if (message.getContentRaw().startsWith("# " + entry.getKey().getUserName())) {
|
|
||||||
continue;
|
|
||||||
}
|
|
||||||
|
|
||||||
message.editMessage("## " + entry.getKey().getUserName())
|
|
||||||
.setAttachments(AttachedFile.fromData(entry.getValue(), entry.getKey().getUUID().toString() + ".png"))
|
|
||||||
.queue();
|
|
||||||
} else {
|
|
||||||
channel.sendMessage("## " + entry.getKey().getUserName())
|
|
||||||
.addFiles(FileUpload.fromData(entry.getValue(), entry.getKey().getUUID().toString() + ".png"))
|
|
||||||
.queue();
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
while (iterator.hasNext()) {
|
|
||||||
iterator.next().delete().queue();
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
private Map<SteamwarUser, InputStream> loadImages(Role role) {
|
|
||||||
Map<SteamwarUser, InputStream> images = new HashMap<>();
|
|
||||||
for (Member member : DiscordBot.getGuild().getMembersWithRoles(role)) {
|
|
||||||
SteamwarUser steamwarUser = SteamwarUser.get(member.getIdLong());
|
|
||||||
if (steamwarUser == null) continue;
|
|
||||||
|
|
||||||
try {
|
|
||||||
images.put(steamwarUser, new URL("api.steamwar.de/data/skin/" + steamwarUser.getUUID().toString()).openStream());
|
|
||||||
} catch (IOException e) {
|
|
||||||
images.put(steamwarUser, null);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return images;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|||||||
@@ -21,6 +21,7 @@ package de.steamwar.velocitycore.discord.channels;
|
|||||||
|
|
||||||
import net.dv8tion.jda.api.EmbedBuilder;
|
import net.dv8tion.jda.api.EmbedBuilder;
|
||||||
import net.dv8tion.jda.api.entities.Message;
|
import net.dv8tion.jda.api.entities.Message;
|
||||||
|
import net.dv8tion.jda.api.entities.channel.middleman.MessageChannel;
|
||||||
import net.dv8tion.jda.api.events.interaction.component.GenericComponentInteractionCreateEvent;
|
import net.dv8tion.jda.api.events.interaction.component.GenericComponentInteractionCreateEvent;
|
||||||
import net.dv8tion.jda.api.utils.messages.MessageCreateBuilder;
|
import net.dv8tion.jda.api.utils.messages.MessageCreateBuilder;
|
||||||
import net.dv8tion.jda.api.utils.messages.MessageEditData;
|
import net.dv8tion.jda.api.utils.messages.MessageEditData;
|
||||||
@@ -55,6 +56,17 @@ public class StaticMessageChannel extends DiscordChannel {
|
|||||||
update();
|
update();
|
||||||
}
|
}
|
||||||
|
|
||||||
|
public StaticMessageChannel(MessageChannel channel, Supplier<MessageCreateBuilder> supplier, Consumer<GenericComponentInteractionCreateEvent> interaction) {
|
||||||
|
super(channel);
|
||||||
|
this.supplier = supplier;
|
||||||
|
this.interaction = interaction;
|
||||||
|
|
||||||
|
if(getChannel().getLatestMessageIdLong() != 0)
|
||||||
|
message = getChannel().getIterableHistory().complete().stream().filter(m -> m.getAuthor().isBot()).findFirst().orElse(null);
|
||||||
|
|
||||||
|
update();
|
||||||
|
}
|
||||||
|
|
||||||
public void update() {
|
public void update() {
|
||||||
if (message == null) {
|
if (message == null) {
|
||||||
getChannel().sendMessage(supplier.get().build()).queue(m -> message = m);
|
getChannel().sendMessage(supplier.get().build()).queue(m -> message = m);
|
||||||
|
|||||||
Reference in New Issue
Block a user