diff --git a/src/main/kotlin/net/moonleay/lilJudd/buttons/component/EditButtonManager.kt b/src/main/kotlin/net/moonleay/lilJudd/buttons/component/EditButtonManager.kt
index 070f25a..f0df57f 100644
--- a/src/main/kotlin/net/moonleay/lilJudd/buttons/component/EditButtonManager.kt
+++ b/src/main/kotlin/net/moonleay/lilJudd/buttons/component/EditButtonManager.kt
@@ -18,8 +18,9 @@
package net.moonleay.lilJudd.buttons.component
+import net.moonleay.lilJudd.buttons.matchplanner.AcceptEditButton
import net.moonleay.lilJudd.buttons.matchplanner.CancelEditButton
-import net.moonleay.lilJudd.buttons.matchplanner.SignMeUpEditButton
+import net.moonleay.lilJudd.buttons.matchplanner.DeclineEditButton
import net.moonleay.lilJudd.buttons.timeplanner.IsAvailableEditButton
import net.moonleay.lilJudd.buttons.timeplanner.MaybeAvailableEditButton
import net.moonleay.lilJudd.buttons.timeplanner.NotAvailableEditButton
@@ -29,7 +30,8 @@ object EditButtonManager {
IsAvailableEditButton(),
MaybeAvailableEditButton(),
NotAvailableEditButton(),
- SignMeUpEditButton(),
- CancelEditButton()
+ AcceptEditButton(),
+ CancelEditButton(),
+ DeclineEditButton(),
)
}
diff --git a/src/main/kotlin/net/moonleay/lilJudd/buttons/matchplanner/DeclineEditButton.kt b/src/main/kotlin/net/moonleay/lilJudd/buttons/matchplanner/DeclineEditButton.kt
new file mode 100644
index 0000000..67b193e
--- /dev/null
+++ b/src/main/kotlin/net/moonleay/lilJudd/buttons/matchplanner/DeclineEditButton.kt
@@ -0,0 +1,119 @@
+/*
+ * lilJudd
+ * Copyright (C) 2023 moonleay
+ *
+ * This program is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation, either version 3 of the License, or
+ * (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program. If not, see .
+ */
+
+package net.moonleay.lilJudd.buttons.matchplanner
+
+import dev.kord.common.entity.Snowflake
+import dev.kord.core.behavior.edit
+import dev.kord.core.behavior.interaction.response.PublicMessageInteractionResponseBehavior
+import dev.kord.core.entity.Guild
+import dev.kord.core.entity.User
+import dev.kord.core.entity.channel.MessageChannel
+import dev.kord.core.entity.interaction.ButtonInteraction
+import dev.kord.rest.builder.message.modify.embed
+import net.moonleay.lilJudd.Bot
+import net.moonleay.lilJudd.buttons.component.IEditButton
+import net.moonleay.lilJudd.data.entry.MatchPlanningDataData
+import net.moonleay.lilJudd.data.tables.MatchPlanningData
+import net.moonleay.lilJudd.util.EmbedUtil
+import net.moonleay.lilJudd.util.Logger
+import net.moonleay.lilJudd.util.MessageUtil
+import org.jetbrains.exposed.sql.and
+import org.jetbrains.exposed.sql.select
+import org.jetbrains.exposed.sql.transactions.transaction
+
+class DeclineEditButton : IEditButton {
+ override val id: String = "public.edit.btn.matchmanagement.decline"
+
+ override suspend fun onInteraction(
+ interaction: ButtonInteraction,
+ response: PublicMessageInteractionResponseBehavior,
+ guild: Guild,
+ user: User
+ ) {
+ val m = interaction.message
+ val eb = MessageUtil.getAClonedEmbed(m.embeds[0])
+ var shouldEditButton = false
+ lateinit var mpdd: MatchPlanningDataData
+ var found = false
+ transaction {
+ for (pnr in MatchPlanningData.select {
+ MatchPlanningData.messageid eq (interaction.message.id.value.toString()) and (
+ MatchPlanningData.serverid eq (guild.id.value.toString())) and (
+ MatchPlanningData.channelid eq (interaction.channelId.value.toString()))
+ }) {
+ mpdd = MatchPlanningDataData(
+ pnr[MatchPlanningData.id].value,
+ pnr[MatchPlanningData.serverid],
+ pnr[MatchPlanningData.channelid],
+ pnr[MatchPlanningData.matchtype],
+ pnr[MatchPlanningData.registererid],
+ pnr[MatchPlanningData.roleid],
+ pnr[MatchPlanningData.opponentName],
+ pnr[MatchPlanningData.messageid],
+ pnr[MatchPlanningData.timestamp],
+ pnr[MatchPlanningData.jobstr]
+ )
+ found = true
+ }
+ }
+ if (!found || mpdd == null) {
+ return
+ }
+ val role = guild.getRoleOrNull(Snowflake(mpdd.roleid)) ?: return
+ val member = interaction.user.asMember(guild.id) ?: return
+ if (m.embeds[0].fields[0].value.contains(user.id.value.toString())) {
+ if (member.roleIds.contains(Snowflake(mpdd.roleid))) {
+ Logger.out("Removing role from ${member.username}")
+ member.removeRole(role.id)
+ }
+ // remove the user from the 1st list in the embed
+ Logger.out("Removing ${user.username} from the 1st list in the embed")
+ eb.fields = EmbedUtil.replaceXWithYinValuesAtTable(user.id.value.toString(), "", eb, 1).fields
+ shouldEditButton = true
+ }
+ if (!m.embeds[0].fields[1].value.contains(user.id.value.toString())) {
+ if (member.roleIds.contains(Snowflake(mpdd.roleid))) {
+ Logger.out("Removing role from ${member.username}")
+ member.removeRole(role.id)
+ }
+ // Add the user to the list in the embed
+ Logger.out("Adding ${user.username} to the 2nd list in the embed")
+ eb.fields = EmbedUtil.addXToValuesAtTable(user.id.value.toString(), eb, 2).fields
+ shouldEditButton = true
+ }
+ if (m.embeds[0].fields[1].value.contains(user.id.value.toString())) {
+ // Remove the user from all tables
+ Logger.out("Removing ${user.username} from the 2nd list in the embed")
+ eb.fields = EmbedUtil.replaceXWithYinValuesAtTable(user.id.value.toString(), "", eb, 2).fields
+ shouldEditButton = true
+ }
+ if (shouldEditButton) {
+ // update the message
+ Bot.bot.kordRef.getChannelOf(interaction.channelId)!!.getMessage(m.id).edit {
+ this.embed {
+ this.color = eb.color
+ this.title = eb.title
+ this.description = eb.description
+ this.fields = eb.fields
+ this.footer = eb.footer
+ }
+ }
+ }
+ }
+}