From 471b12d591a4303ae5f9ebef22c9313d3701a878 Mon Sep 17 00:00:00 2001 From: Gareth Coles Date: Sat, 6 May 2023 14:35:15 +0100 Subject: [PATCH] Forums: Tag management commands --- .idea/modules.xml | 2 + .../modes/quilt/extensions/ForumExtension.kt | 116 +++++++++++++++++- 2 files changed, 112 insertions(+), 6 deletions(-) diff --git a/.idea/modules.xml b/.idea/modules.xml index de6965c4..9f8a802d 100644 --- a/.idea/modules.xml +++ b/.idea/modules.xml @@ -2,6 +2,8 @@ + + diff --git a/src/main/kotlin/org/quiltmc/community/modes/quilt/extensions/ForumExtension.kt b/src/main/kotlin/org/quiltmc/community/modes/quilt/extensions/ForumExtension.kt index d9911e67..c9eecdf0 100644 --- a/src/main/kotlin/org/quiltmc/community/modes/quilt/extensions/ForumExtension.kt +++ b/src/main/kotlin/org/quiltmc/community/modes/quilt/extensions/ForumExtension.kt @@ -12,8 +12,11 @@ package org.quiltmc.community.modes.quilt.extensions import com.kotlindiscord.kord.extensions.checks.hasRole import com.kotlindiscord.kord.extensions.checks.or import com.kotlindiscord.kord.extensions.commands.Arguments +import com.kotlindiscord.kord.extensions.commands.application.slash.ephemeralSubCommand import com.kotlindiscord.kord.extensions.commands.converters.impl.channel import com.kotlindiscord.kord.extensions.commands.converters.impl.optionalChannel +import com.kotlindiscord.kord.extensions.commands.converters.impl.optionalTag +import com.kotlindiscord.kord.extensions.commands.converters.impl.tag import com.kotlindiscord.kord.extensions.components.forms.ModalForm import com.kotlindiscord.kord.extensions.extensions.Extension import com.kotlindiscord.kord.extensions.extensions.ephemeralSlashCommand @@ -21,6 +24,7 @@ import com.kotlindiscord.kord.extensions.modules.unsafe.annotations.UnsafeAPI import com.kotlindiscord.kord.extensions.modules.unsafe.extensions.unsafeSubCommand import com.kotlindiscord.kord.extensions.modules.unsafe.types.InitialSlashCommandResponse import com.kotlindiscord.kord.extensions.modules.unsafe.types.ackEphemeral +import com.kotlindiscord.kord.extensions.types.respond import com.kotlindiscord.kord.extensions.utils.extraData import dev.kord.common.annotation.KordUnsafe import dev.kord.common.entity.ChannelType @@ -31,6 +35,7 @@ import dev.kord.core.behavior.edit import dev.kord.core.behavior.interaction.modal import dev.kord.core.behavior.interaction.response.createEphemeralFollowup import dev.kord.core.entity.channel.ForumChannel +import dev.kord.core.entity.channel.thread.TextChannelThread import dev.kord.core.entity.channel.thread.ThreadChannel import kotlinx.coroutines.delay import org.quiltmc.community.* @@ -58,7 +63,7 @@ class ForumExtension : Extension() { } } - unsafeSubCommand(::ForumChannelArgs) { + unsafeSubCommand(::CreatePostArgs) { name = "create-post" description = "Create a Cozy-managed forum post" @@ -112,9 +117,11 @@ class ForumExtension : Extension() { } val thread = forum.startPublicThread(title) { - // TODO: Tags? - message(text) + + if (arguments.tag != null) { + appliedTags = mutableListOf(arguments.tag!!.id) + } } interactionResponse.createEphemeralFollowup { @@ -147,7 +154,7 @@ class ForumExtension : Extension() { } } - unsafeSubCommand(::ForumPostArgs) { + unsafeSubCommand(::EditPostArgs) { name = "edit-post" description = "Edit an existing Cozy-managed forum post" @@ -230,19 +237,116 @@ class ForumExtension : Extension() { } } } + + ephemeralSubCommand(::PostTagArgs) { + name = "add-tag" + description = "Add a tag to the given post" + + action { + val post = arguments.post.asChannelOfOrNull() + val parent = post?.parent?.asChannelOfOrNull() + + if (parent == null) { + respond { + content = "Please provide a forum thread to edit the tags for." + } + + return@action + } + + if (post.appliedTags.contains(arguments.tag.id)) { + respond { + content = "This thread already has that tag applied." + } + + return@action + } + + post.edit { + appliedTags = post.appliedTags.toMutableList() + appliedTags?.add(arguments.tag.id) + } + + respond { + content = "Thread tags updated." + } + } + } + + ephemeralSubCommand(::PostTagArgs) { + name = "remove-tag" + description = "Remove a tag from the given post" + + action { + val post = arguments.post.asChannelOfOrNull() + val parent = post?.parent?.asChannelOfOrNull() + + if (parent == null) { + respond { + content = "Please provide a forum thread to edit the tags for." + } + + return@action + } + + if (!post.appliedTags.contains(arguments.tag.id)) { + respond { + content = "This thread doesn't have that tag applied." + } + + return@action + } + + post.edit { + appliedTags = post.appliedTags.toMutableList() + appliedTags?.remove(arguments.tag.id) + } + + respond { + content = "Thread tags updated." + } + } + } } } - inner class ForumChannelArgs : Arguments() { + inner class PostTagArgs : Arguments() { + override val parseForAutocomplete: Boolean = true + + val post by channel { + name = "post" + description = "Thread to edit the tags for" + + requireChannelType(ChannelType.PublicGuildThread) + } + + val tag by tag { + name = "tag" + description = "Tag to create the post with" + + channelGetter = { post.asChannelOfOrNull()?.parent?.asChannelOfOrNull() } + } + } + + inner class CreatePostArgs : Arguments() { + override val parseForAutocomplete: Boolean = true + val channel by channel { name = "channel" description = "Forum channel to post in" requireChannelType(ChannelType.GuildForum) } + + val tag by optionalTag { + name = "tag" + description = "Tag to create the post with" + + channelGetter = { channel.asChannelOfOrNull() } + } } - inner class ForumPostArgs : Arguments() { + inner class EditPostArgs : Arguments() { val post by optionalChannel { name = "post" description = "Thread to edit the first post for"