From 39926d8767b4986956384bb30a1cc22239dd8349 Mon Sep 17 00:00:00 2001
From: Chaoscaot <chaoscaot444@gmail.com>
Date: Mon, 4 Apr 2022 22:59:30 +0200
Subject: [PATCH] :art: Yeet

---
 .../bot/util/DiscordTicketMessage.java         | 18 +++++++++++++++---
 1 file changed, 15 insertions(+), 3 deletions(-)

diff --git a/src/de/steamwar/bungeecore/bot/util/DiscordTicketMessage.java b/src/de/steamwar/bungeecore/bot/util/DiscordTicketMessage.java
index ea6cea39..21ebded9 100644
--- a/src/de/steamwar/bungeecore/bot/util/DiscordTicketMessage.java
+++ b/src/de/steamwar/bungeecore/bot/util/DiscordTicketMessage.java
@@ -20,6 +20,7 @@
 package de.steamwar.bungeecore.bot.util;
 
 import de.steamwar.bungeecore.bot.SteamwarDiscordBot;
+import de.steamwar.bungeecore.bot.config.DiscordTicketType;
 import de.steamwar.bungeecore.bot.config.SteamwarDiscordBotConfig;
 import lombok.experimental.UtilityClass;
 import net.dv8tion.jda.api.EmbedBuilder;
@@ -32,6 +33,7 @@ import net.dv8tion.jda.api.interactions.components.Button;
 import java.awt.*;
 import java.util.ArrayList;
 import java.util.List;
+import java.util.stream.Collectors;
 
 @UtilityClass
 public class DiscordTicketMessage {
@@ -49,16 +51,26 @@ public class DiscordTicketMessage {
         builder.setTitle("Steamwar-Tickets");
         builder.setColor(Color.RED);
 
-        List<Button> buttons = new ArrayList<>();
-        SteamwarDiscordBotConfig.TICKET_TYPES.forEach((s, discordTicketType) -> buttons.add(discordTicketType.toButton()));
+        List<List<Button>> buttons = new ArrayList<>();
+        chunked(new ArrayList<>(SteamwarDiscordBotConfig.TICKET_TYPES.values()), 5).forEach(discordTicketTypes -> {
+            buttons.add(discordTicketTypes.stream().map(DiscordTicketType::toButton).collect(Collectors.toList()));
+        });
 
         MessageBuilder messageBuilder = new MessageBuilder();
         messageBuilder.setEmbeds(builder.build());
-        messageBuilder.setActionRows(ActionRow.of(buttons));
+        messageBuilder.setActionRows(buttons.stream().map(ActionRow::of).collect(Collectors.toList()));
         if (message != null) {
             message.editMessage(messageBuilder.build()).queue();
         } else {
             channel.sendMessage(messageBuilder.build()).queue();
         }
     }
+
+    private static <T> List<List<T>> chunked(List<T> list, int chunkSize) {
+        List<List<T>> chunks = new ArrayList<>();
+        for (int i = 0; i < list.size(); i += chunkSize) {
+            chunks.add(list.subList(i, Math.min(i + chunkSize, list.size())));
+        }
+        return chunks;
+    }
 }