diff --git a/application/src/main/java/org/togetherjava/tjbot/commands/code/CodeMessageAutoDetection.java b/application/src/main/java/org/togetherjava/tjbot/commands/code/CodeMessageAutoDetection.java index d8adecc0d1..64572420b2 100644 --- a/application/src/main/java/org/togetherjava/tjbot/commands/code/CodeMessageAutoDetection.java +++ b/application/src/main/java/org/togetherjava/tjbot/commands/code/CodeMessageAutoDetection.java @@ -63,7 +63,7 @@ public void onMessageReceived(MessageReceivedEvent event) { return; } - codeMessageHandler.addAndHandleCodeMessage(originalMessage); + codeMessageHandler.addAndHandleCodeMessage(originalMessage, true); } private boolean isHelpThread(MessageReceivedEvent event) { diff --git a/application/src/main/java/org/togetherjava/tjbot/commands/code/CodeMessageHandler.java b/application/src/main/java/org/togetherjava/tjbot/commands/code/CodeMessageHandler.java index a45a5c0f80..341c8ad351 100644 --- a/application/src/main/java/org/togetherjava/tjbot/commands/code/CodeMessageHandler.java +++ b/application/src/main/java/org/togetherjava/tjbot/commands/code/CodeMessageHandler.java @@ -27,20 +27,16 @@ import javax.annotation.Nullable; import java.awt.Color; -import java.util.LinkedHashMap; -import java.util.List; -import java.util.Map; -import java.util.Optional; +import java.util.*; import java.util.function.Function; import java.util.regex.Pattern; import java.util.stream.Collectors; -import java.util.stream.Stream; /** * Handles code in registered messages and offers code actions to the user, such as formatting their * code. *

- * Messages can be registered by using {@link #addAndHandleCodeMessage(Message)}. + * Messages can be registered by using {@link #addAndHandleCodeMessage(Message, boolean)}. *

* Code actions are automatically updated whenever the code in the original message is edited or * deleted. @@ -111,10 +107,12 @@ public void acceptComponentIdGenerator(ComponentIdGenerator generator) { * corresponding code-reply to the author. * * @param originalMessage the code message to add to this handler + * @param showDeleteButton whether the code-actions should initially have a delete button or not */ - public void addAndHandleCodeMessage(Message originalMessage) { + public void addAndHandleCodeMessage(Message originalMessage, boolean showDeleteButton) { // Suggest code actions and remember the message <-> reply - MessageCreateData codeReply = createCodeReplyMessage(originalMessage.getIdLong()); + MessageCreateData codeReply = + createCodeReplyMessage(originalMessage.getIdLong(), showDeleteButton); originalMessage.reply(codeReply) .onSuccess(replyMessage -> originalMessageToCodeReply.put(originalMessage.getIdLong(), @@ -122,22 +120,24 @@ public void addAndHandleCodeMessage(Message originalMessage) { .queue(); } - private MessageCreateData createCodeReplyMessage(long originalMessageId) { + private MessageCreateData createCodeReplyMessage(long originalMessageId, + boolean showDeleteButton) { + List