Re-sorted some things

src/main/kotlin/quest/laxla/supertrouper/App.kt

@@ -3,7 +3,9 @@ package quest.laxla.supertrouper
 import com.kotlindiscord.kord.extensions.ExtensibleBot
 import com.kotlindiscord.kord.extensions.utils.env
 import com.kotlindiscord.kord.extensions.utils.envOrNull
+import dev.kord.gateway.PrivilegedIntent
 import kotlinx.coroutines.runBlocking
+import quest.laxla.supertrouper.messaging.PrivateMassagingExtension
 
 private val token = env("TOKEN")
 private val testingServer = envOrNull("TESTING_SERVER")
@@ -16,6 +18,7 @@ fun main() = runBlocking {
 
 		extensions {
 			add(::MaintenanceExtension)
+			@OptIn(PrivilegedIntent::class)
 			add(::PrivateMassagingExtension)
 		}
 	}.start()

src/main/kotlin/quest/laxla/supertrouper/messaging/PrivateMassagingExtension.kt

@@ -0,0 +1,149 @@
+package quest.laxla.supertrouper.messaging
+
+import com.kotlindiscord.kord.extensions.checks.anyGuild
+import com.kotlindiscord.kord.extensions.checks.isNotBot
+import com.kotlindiscord.kord.extensions.extensions.*
+import com.kotlindiscord.kord.extensions.types.EphemeralInteractionContext
+import dev.kord.common.entity.OverwriteType
+import dev.kord.common.entity.Permission
+import dev.kord.core.behavior.GuildBehavior
+import dev.kord.core.behavior.channel.createTextChannel
+import dev.kord.core.behavior.channel.edit
+import dev.kord.core.behavior.createCategory
+import dev.kord.core.entity.User
+import dev.kord.core.entity.channel.Category
+import dev.kord.core.entity.channel.TextChannel
+import dev.kord.core.event.guild.MemberJoinEvent
+import dev.kord.gateway.Intent
+import dev.kord.gateway.PrivilegedIntent
+import dev.kord.rest.builder.channel.addMemberOverwrite
+import dev.kord.rest.builder.channel.addRoleOverwrite
+import kotlinx.coroutines.flow.count
+import quest.laxla.supertrouper.*
+
+@PrivilegedIntent
+class PrivateMassagingExtension : TrouperExtension() {
+	override suspend fun setup() {
+		intents += Intent.GuildMembers
+
+		slashCommandCheck { anyGuild(); isNotBot() }
+		userCommandCheck { anyGuild(); isNotBot() }
+
+		event<MemberJoinEvent> {
+			action {
+				if (event.member.isEligible && event.guild.members.count() < memberLimit) getOrCreateChannel(
+					getOrCreateCategory(event.guild),
+					event.member
+				)
+			}
+		}
+
+		ephemeralSlashCommand(::TargetedArguments) {
+			name = "pm"
+			description = "Get a link to a user's private messages channel"
+
+			action {
+				executeFindCommand(getOrCreateCategory(guild!!), target.asUser())
+			}
+		}
+
+		ephemeralUserCommand {
+			name = "Private Message"
+
+			action {
+				executeFindCommand(getOrCreateCategory(guild!!), targetUsers.single())
+			}
+		}
+
+		ephemeralSlashCommand(::TargetedArguments) slash@{
+			name = "sync"
+			description = "Syncs a private message channel's permissions with the category."
+
+			requirePermission(Permission.ManageRoles)
+
+			action {
+				executeSyncCommand(getOrCreateCategory(guild!!), target.asUser())
+			}
+		}
+
+		ephemeralUserCommand {
+			name = "Sync PM Channel"
+
+			requirePermission(Permission.ManageRoles)
+
+			action {
+				executeSyncCommand(getOrCreateCategory(guild!!), user.asUser())
+			}
+		}
+	}
+
+	private suspend fun EphemeralInteractionContext.executeSyncCommand(category: Category, user: User) {
+		val channel = getChannel(category, user)
+		if (channel == null) {
+			respond { content = "${user.mention} does not have a private messaging channel in this server." }
+			return
+		}
+
+		val userMention = user.mention
+		val channelMention = channel.mention
+
+		channel.edit {
+			reason = "Syncing $userMention"
+
+			sync(
+				overwrite(kord.selfId, OverwriteType.Member, allowed = privateMessageBotPermissions),
+				overwrite(user.id, OverwriteType.Member, allowed = privateMessageOwnerPermissions),
+				defaults = category.permissionOverwrites
+			)
+		}
+
+		respond {
+			content = "Synced $channelMention for $userMention successfully."
+		}
+	}
+
+	private suspend fun EphemeralInteractionContext.executeFindCommand(category: Category, user: User) {
+		if (user.isEligible) {
+			val channel = getOrCreateChannel(category, user)
+			respond { content = channel.mention }
+		} else respond {
+			content = user.mention + " is not eligible for private messaging."
+		}
+	}
+
+	private suspend fun getOrCreateCategory(guild: GuildBehavior) = getCategory(guild) ?: createCategory(guild)
+
+	private suspend fun createCategory(guild: GuildBehavior) = guild.createCategory(PrivateMessagesCategoryName) {
+		reason = "Private messaging category was missing."
+		nsfw = false
+
+		addMemberOverwrite(kord.selfId) {
+			allowed += privateMessageBotPermissions
+		}
+
+		addRoleOverwrite(guild.id) {
+			denied += Permission.ViewChannel
+		}
+	}
+
+	private suspend fun getOrCreateChannel(category: Category, user: User) =
+		getChannel(category, user) ?: createChannel(category, user)
+
+	private suspend fun createChannel(category: Category, user: User): TextChannel {
+		val mention = user.mention
+
+		val channel = category.createTextChannel(user.username) {
+			reason = "Created a PM with $mention."
+			nsfw = category.data.nsfw.discordBoolean
+			topic = "This channel is $mention's private message."
+
+			sync(
+				overwrite(kord.selfId, OverwriteType.Member, allowed = privateMessageBotPermissions),
+				overwrite(user.id, OverwriteType.Member, allowed = privateMessageOwnerPermissions),
+				defaults = category.permissionOverwrites
+			)
+		}
+
+		return channel
+	}
+}

src/main/kotlin/quest/laxla/supertrouper/messaging/PrivateMessaging.kt

@@ -0,0 +1,32 @@
+package quest.laxla.supertrouper.messaging
+
+import com.kotlindiscord.kord.extensions.utils.any
+import com.kotlindiscord.kord.extensions.utils.envOrNull
+import dev.kord.common.entity.Permission
+import dev.kord.core.behavior.GuildBehavior
+import dev.kord.core.behavior.UserBehavior
+import dev.kord.core.entity.User
+import dev.kord.core.entity.channel.Category
+import dev.kord.core.entity.channel.TextChannel
+import kotlinx.coroutines.flow.filterIsInstance
+import kotlinx.coroutines.flow.firstOrNull
+
+const val PrivateMessagesCategoryName = "Private Messages"
+val memberLimit = envOrNull("AUTOMATIC_CHANNEL_CREATION_MEMBER_LIMIT")?.toInt() ?: 30
+val privateMessageOwnerPermissions = Permission.ViewChannel + Permission.ReadMessageHistory
+val privateMessageBotPermissions =
+	privateMessageOwnerPermissions + Permission.ManageChannels + Permission.SendMessages + Permission.ManageMessages
+
+suspend infix fun TextChannel.isOf(user: UserBehavior) = topic?.contains(user.mention) == true || pinnedMessages.any {
+	it.author?.id == kord.selfId && it.mentionedUserIds.singleOrNull() == user.id
+}
+
+suspend fun getChannel(category: Category, user: User) = category.channels.filterIsInstance<TextChannel>().firstOrNull {
+	it.categoryId == category.id && it isOf user
+}
+
+suspend fun getCategory(guild: GuildBehavior) =
+	guild.channels.filterIsInstance<Category>().firstOrNull { it.isUsableForPrivateMessaging }
+
+val User.isEligible get() = !isBot
+val Category.isUsableForPrivateMessaging get() = name.equals(PrivateMessagesCategoryName, ignoreCase = true)