fixed issues and simplified.

Author: Neil-Tomar

src/main/java/net/discordjug/javabot/systems/user_commands/format_code/FormatCodeInteractionHandler.java

@@ -5,13 +5,16 @@
 import net.discordjug.javabot.data.config.BotConfig;
 import net.discordjug.javabot.util.Checks;
 import net.discordjug.javabot.util.Responses;
+import net.dv8tion.jda.api.components.Component;
 import net.dv8tion.jda.api.components.actionrow.ActionRow;
 import net.dv8tion.jda.api.components.buttons.Button;
 import net.dv8tion.jda.api.components.selections.SelectOption;
 import net.dv8tion.jda.api.components.selections.StringSelectMenu;
 import net.dv8tion.jda.api.entities.Member;
 import net.dv8tion.jda.api.entities.Message;
-import java.util.stream.Stream;
+
+import java.util.ArrayList;
+
 import net.dv8tion.jda.api.events.interaction.component.ButtonInteractionEvent;
 import net.dv8tion.jda.api.events.interaction.component.StringSelectInteractionEvent;
 import org.jspecify.annotations.NonNull;
@@ -40,8 +43,8 @@ public class FormatCodeInteractionHandler implements ButtonHandler, StringSelect
 	 * @param total       the number of messages making up the block
 	 * @return the delete-all button
 	 */
-	public static Button createDeleteAllButton(long requesterID, int total){
-		return Button.secondary(ComponentIdBuilder.build(COMPONENT_ID,requesterID,total),"\uD83D\uDDD1\uFE0F");
+	public static Button createDeleteAllButton(long requesterID, int total) {
+		return Button.secondary(ComponentIdBuilder.build(COMPONENT_ID, requesterID, total), "\uD83D\uDDD1\uFE0F");
 	}
 
 	private static StringSelectMenu languageMenu(String customId) {
@@ -85,7 +88,7 @@ public void handleButton(ButtonInteractionEvent event, @NonNull Button button) {
 
 		LinkedMessages.resolveBefore(event.getMessage(), Integer.parseInt(id[2]),
 				messages -> event.getChannel().purgeMessages(messages),
-				() -> Responses.error(event.getHook(),"Could not delete the code block").queue());
+				() -> Responses.error(event.getHook(), "Could not delete the code block").queue());
 	}
 
 	@Override
@@ -106,32 +109,38 @@ public void handleStringSelectMenu(@NonNull StringSelectInteractionEvent event,
 		Language language = Language.fromString(values.getFirst());
 		event.deferEdit().queue();
 
-		Runnable onError = () -> Responses.error(event.getHook(),
-				"Could not update the code block — it may have changed or no longer be yours.").queue();
 		LinkedMessages.resolveAfter(event.getMessage(), Integer.parseInt(id[2]),
 				messages -> {
 					if (ownsBlock(messages, event.getUser().getIdLong())) {
 						messages.forEach(message ->
 								message.editMessage(withLanguage(message.getContentRaw(), language)).queue());
 					} else {
-						onError.run();
+						Responses.error(event.getHook(),"Could not update the code block — it may have changed or no longer be yours.").queue();
 					}
-				},
-				onError);
+				}, () -> Responses.error(event.getHook(), "Could not update the code block — it may have changed or no longer be yours.").queue());
 	}
 
 	private static boolean ownsBlock(List<Message> messages, long userId) {
-		Message last = messages.get(messages.size() - 1);
-		boolean lastIsUsers = buttonsOf(last).anyMatch(button -> isDeleteButtonOf(button, userId));
-		boolean noOtherHasButton = messages.subList(0, messages.size() - 1).stream()
-				.flatMap(FormatCodeInteractionHandler::buttonsOf)
-				.noneMatch(FormatCodeInteractionHandler::isDeleteButton);
+		Message last = messages.getFirst();
+		boolean lastIsUsers = buttonsOf(last).stream().anyMatch(button -> isDeleteButtonOf(button, userId));
+		boolean noOtherHasButton = messages.size() <= 1 ;
+		for (Message message : messages.subList(0,messages.size()-1)){
+			for(Button button: buttonsOf(message)){
+				if(isDeleteButton(button)){
+					noOtherHasButton = true;
+				}
+			}
+		}
 		return lastIsUsers && noOtherHasButton;
 	}
 
-	private static Stream<Button> buttonsOf(Message message) {
-		return message.getComponents().stream()
-				.flatMap(component -> component instanceof ActionRow row ? row.getButtons().stream() : Stream.of());
+	private static List<Button> buttonsOf(Message message) {
+		for (Component component: message.getComponents()) {
+			if(component instanceof ActionRow row) {
+				return row.getButtons();
+			}
+		}
+		return new ArrayList<>();
 	}
 
 	private static boolean isDeleteButton(Button button) {