package me.niloy.banaddon; import com.google.gson.JsonArray; import com.google.gson.JsonElement; import com.google.gson.JsonObject; import com.google.gson.JsonParser; import me.tini.announcer.BanAnnouncer; import me.tini.announcer.PunishmentInfo; import me.tini.announcer.PunishmentListener; import me.tini.announcer.extension.AbstractExtension; import me.tini.announcer.plugin.bukkit.BukkitPunishmentListener; import net.dv8tion.jda.api.EmbedBuilder; import net.dv8tion.jda.api.entities.MessageEmbed; import net.dv8tion.jda.api.entities.channel.concrete.TextChannel; import net.dv8tion.jda.api.entities.emoji.Emoji; import net.dv8tion.jda.api.events.interaction.component.ButtonInteractionEvent; import net.dv8tion.jda.api.hooks.ListenerAdapter; import net.dv8tion.jda.api.interactions.components.ActionRow; import net.dv8tion.jda.api.interactions.components.buttons.Button; import net.dv8tion.jda.api.interactions.components.buttons.ButtonStyle; import org.bukkit.Bukkit; import org.bukkit.configuration.file.YamlConfiguration; import org.bukkit.plugin.Plugin; import org.spicord.Spicord; import org.spicord.bot.DiscordBot; import java.awt.Color; import java.io.File; import java.io.FileReader; import java.io.IOException; import java.io.InputStream; import java.nio.file.Files; import java.util.ArrayList; import java.util.Collections; import java.util.List; import java.util.Map; import java.util.Set; import java.util.UUID; import java.util.concurrent.ConcurrentHashMap; import java.util.logging.Logger; public class BanButtonExtension extends AbstractExtension { private final BanAnnouncer announcer; private final Plugin plugin; private final Logger logger; private final String botName; private static final java.util.regex.Pattern PLACEHOLDER_PATTERN = java.util.regex.Pattern.compile("%([^%]+)%"); private final Set pendingKeys = Collections.newSetFromMap(new ConcurrentHashMap<>()); private final Map responseMap = new ConcurrentHashMap<>(); private volatile boolean listenerRegistered = false; private volatile String cachedChannelId; private volatile boolean channelIdLoaded = false; public BanButtonExtension(BanAnnouncer announcer, Plugin plugin) { this.announcer = announcer; this.plugin = plugin; this.logger = plugin.getLogger(); this.botName = loadBotName(plugin); logger.info("[BanButtonExtension] Loaded. Using bot: " + botName); } // ------------------------------------------------------------------------- // Config // ------------------------------------------------------------------------- private String loadBotName(Plugin plugin) { File dataFolder = new File("plugins/BanAnnouncerButtonAddon"); File configFile = new File(dataFolder, "config.yml"); if (!configFile.exists()) { dataFolder.mkdirs(); try (InputStream in = getClass().getClassLoader() .getResourceAsStream("config.yml")) { if (in != null) { Files.copy(in, configFile.toPath()); } else { YamlConfiguration def = new YamlConfiguration(); def.set("bot-name", "default"); def.save(configFile); } } catch (IOException e) { logger.warning("[BanButtonExtension] Could not save default config: " + e.getMessage()); } } YamlConfiguration cfg = YamlConfiguration.loadConfiguration(configFile); return cfg.getString("bot-name", "default"); } // ------------------------------------------------------------------------- // AbstractExtension // ------------------------------------------------------------------------- @Override public String processPlaceholder(PunishmentInfo info, String placeholder) { String key = buildKey(info); if (pendingKeys.add(key)) { Bukkit.getScheduler().runTaskLaterAsynchronously(plugin, () -> { handleButtonSend(info); pendingKeys.remove(key); }, 1L); } return null; } @Override public PunishmentListener getPunishmentListener() { return new NoOpPunishmentListener(); } // ------------------------------------------------------------------------- // Discord // ------------------------------------------------------------------------- private DiscordBot getBot() { if (!Spicord.isLoaded()) return null; DiscordBot bot = Spicord.getInstance().getBotByName(botName); if (bot == null || !bot.isReady()) return null; return bot; } private void ensureListenerRegistered(DiscordBot bot) { if (listenerRegistered) return; listenerRegistered = true; bot.addEventListener(new ListenerAdapter() { @Override public void onButtonInteraction(ButtonInteractionEvent event) { String customId = event.getComponentId(); ButtonResponse resp = responseMap.remove(customId); if (resp == null) return; event.deferReply(true).queue(hook -> { var edit = hook.editOriginal(resp.content() != null ? resp.content() : ""); if (resp.embed() != null) edit = edit.setEmbeds(resp.embed()); edit.queue(); }); } }); } private void handleButtonSend(PunishmentInfo info) { DiscordBot bot = getBot(); if (bot == null) { logger.warning("[BanButtonExtension] Bot '" + botName + "' is not ready. Check bot-name in BanAnnouncerButtonAddon/config.yml " + "and Spicord's config."); return; } ensureListenerRegistered(bot); String channelId = readChannelId(); if (channelId == null || channelId.isEmpty()) { logger.warning("[BanButtonExtension] channel-to-announce not set in BanAnnouncer/config.yml"); return; } File jsonFile = getEmbedFile(info.getType()); if (jsonFile == null || !jsonFile.exists()) return; List