From a5505ca21b897cac00c8738cc22d299b26a133af Mon Sep 17 00:00:00 2001
From: sschr15 <steven.sschr15@hotmail.com>
Date: Wed, 26 Jul 2023 22:03:01 -0500
Subject: [PATCH] PluralKit AMA support, more configuration possibilities, and
 extra response info

---
 .idea/inspectionProfiles/Project_Default.xml  |   2 +-
 .idea/jarRepositories.xml                     |   5 +
 .idea/kotlinScripting.xml                     |  10 -
 .idea/modules.xml                             |   2 +
 .../cozy/modules/ama/AmaExtension.kt          | 111 +++++++++--
 .../community/cozy/modules/ama/_PkUtils.kt    | 183 ++++++++++++++++++
 .../community/cozy/modules/ama/_Utils.kt      |  91 +++++++--
 .../cozy/modules/ama/data/AmaData.kt          |  12 ++
 .../cozy/modules/ama/data/MemoryAmaData.kt    |  46 ++---
 .../collections/AmaConfigCollection.kt        |   9 +
 .../community/database/entities/UserFlags.kt  |   1 +
 .../community/database/migrations/v23.kt      |   8 +
 12 files changed, 415 insertions(+), 65 deletions(-)
 delete mode 100644 .idea/kotlinScripting.xml
 create mode 100644 module-ama/src/main/kotlin/org/quiltmc/community/cozy/modules/ama/_PkUtils.kt

diff --git a/.idea/inspectionProfiles/Project_Default.xml b/.idea/inspectionProfiles/Project_Default.xml
index 4b2e0d40..aaf184ce 100644
--- a/.idea/inspectionProfiles/Project_Default.xml
+++ b/.idea/inspectionProfiles/Project_Default.xml
@@ -17,7 +17,7 @@
       </option>
       <option name="myCustomValuesEnabled" value="true" />
     </inspection_tool>
-    <inspection_tool class="ProblematicWhitespace" enabled="true" level="TEXT ATTRIBUTES" enabled_by_default="true" editorAttributes="TEXT_STYLE_SUGGESTION" />
+    <inspection_tool class="ProblematicWhitespace" enabled="false" level="TEXT ATTRIBUTES" enabled_by_default="false" editorAttributes="TEXT_STYLE_SUGGESTION" />
     <inspection_tool class="PyPackageRequirementsInspection" enabled="true" level="WARNING" enabled_by_default="true">
       <option name="ignoredPackages">
         <value>
diff --git a/.idea/jarRepositories.xml b/.idea/jarRepositories.xml
index 8f4b8709..1d601402 100644
--- a/.idea/jarRepositories.xml
+++ b/.idea/jarRepositories.xml
@@ -96,5 +96,10 @@
       <option name="name" value="Sleeping Town" />
       <option name="url" value="https://repo.sleeping.town" />
     </remote-repository>
+    <remote-repository>
+      <option name="id" value="MavenLocal" />
+      <option name="name" value="MavenLocal" />
+      <option name="url" value="file:/$USER_HOME$/.m2/repository/" />
+    </remote-repository>
   </component>
 </project>
\ No newline at end of file
diff --git a/.idea/kotlinScripting.xml b/.idea/kotlinScripting.xml
deleted file mode 100644
index 3f791448..00000000
--- a/.idea/kotlinScripting.xml
+++ /dev/null
@@ -1,10 +0,0 @@
-<?xml version="1.0" encoding="UTF-8"?>
-<project version="4">
-  <component name="KotlinScriptingSettings">
-    <option name="suppressDefinitionsCheck" value="true" />
-    <scriptDefinition className="org.jetbrains.kotlin.idea.core.script.StandardIdeScriptDefinition" definitionName="Kotlin Script">
-      <order>2147483647</order>
-      <autoReloadConfigurations>true</autoReloadConfigurations>
-    </scriptDefinition>
-  </component>
-</project>
\ No newline at end of file
diff --git a/.idea/modules.xml b/.idea/modules.xml
index 9f8a802d..128cf299 100644
--- a/.idea/modules.xml
+++ b/.idea/modules.xml
@@ -5,6 +5,8 @@
       <module fileurl="file://$PROJECT_DIR$/.idea/modules/CozyDiscord.main.iml" filepath="$PROJECT_DIR$/.idea/modules/CozyDiscord.main.iml" />
       <module fileurl="file://$PROJECT_DIR$/.idea/modules/CozyDiscord.test.iml" filepath="$PROJECT_DIR$/.idea/modules/CozyDiscord.test.iml" />
       <module fileurl="file://$PROJECT_DIR$/.idea/cozy-discord.iml" filepath="$PROJECT_DIR$/.idea/cozy-discord.iml" />
+      <module fileurl="file://$PROJECT_DIR$/.idea/modules/org.quiltmc.community.CozyDiscord.main.iml" filepath="$PROJECT_DIR$/.idea/modules/org.quiltmc.community.CozyDiscord.main.iml" />
+      <module fileurl="file://$PROJECT_DIR$/.idea/modules/org.quiltmc.community.CozyDiscord.test.iml" filepath="$PROJECT_DIR$/.idea/modules/org.quiltmc.community.CozyDiscord.test.iml" />
     </modules>
   </component>
 </project>
\ No newline at end of file
diff --git a/module-ama/src/main/kotlin/org/quiltmc/community/cozy/modules/ama/AmaExtension.kt b/module-ama/src/main/kotlin/org/quiltmc/community/cozy/modules/ama/AmaExtension.kt
index 3092636f..253a1c3f 100644
--- a/module-ama/src/main/kotlin/org/quiltmc/community/cozy/modules/ama/AmaExtension.kt
+++ b/module-ama/src/main/kotlin/org/quiltmc/community/cozy/modules/ama/AmaExtension.kt
@@ -8,7 +8,6 @@ package org.quiltmc.community.cozy.modules.ama
 
 import com.kotlindiscord.kord.extensions.checks.anyGuild
 import com.kotlindiscord.kord.extensions.checks.guildFor
-import com.kotlindiscord.kord.extensions.checks.hasPermission
 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
@@ -40,8 +39,10 @@ import dev.kord.core.behavior.edit
 import dev.kord.core.behavior.getChannelOf
 import dev.kord.core.behavior.interaction.modal
 import dev.kord.core.behavior.interaction.response.EphemeralMessageInteractionResponseBehavior
+import dev.kord.core.behavior.interaction.response.MessageInteractionResponseBehavior
 import dev.kord.core.behavior.interaction.response.createEphemeralFollowup
 import dev.kord.core.entity.Message
+import dev.kord.core.entity.User
 import dev.kord.core.entity.channel.Channel
 import dev.kord.core.entity.channel.GuildMessageChannel
 import dev.kord.core.entity.channel.TopGuildChannel
@@ -67,11 +68,13 @@ public class AmaExtension : Extension() {
 			ephemeralSubCommand(::AmaConfigArgs, ::AmaConfigModal) {
 				name = "config"
 				description = "Configure your AMA settings"
+				requirePermission(Permission.SendMessages)
 
 				check {
 					anyGuild()
-					hasPermission(Permission.ManageGuild)
-					requirePermission(Permission.SendMessages)
+					with(amaData) {
+						managementChecks()
+					}
 				}
 
 				var buttonMessage: Message?
@@ -123,11 +126,13 @@ public class AmaExtension : Extension() {
 			ephemeralSubCommand {
 				name = "start"
 				description = "Start the AMA for this server"
+				requirePermission(Permission.SendMessages)
 
 				check {
 					anyGuild()
-					hasPermission(Permission.ManageGuild)
-					requirePermission(Permission.SendMessages)
+					with(amaData) {
+						managementChecks()
+					}
 				}
 
 				action {
@@ -135,7 +140,7 @@ public class AmaExtension : Extension() {
 
 					if (config == null) {
 						respond {
-							content = "There is no AMA config for this guild!"
+							content = "There is no AMA config for this guild! Please run `/ama config` first."
 						}
 						return@action
 					}
@@ -171,11 +176,13 @@ public class AmaExtension : Extension() {
 			ephemeralSubCommand {
 				name = "stop"
 				description = "Stop the AMA for this server"
+				requirePermission(Permission.SendMessages)
 
 				check {
 					anyGuild()
-					hasPermission(Permission.ManageGuild)
-					requirePermission(Permission.SendMessages)
+					with(amaData) {
+						managementChecks()
+					}
 				}
 
 				action {
@@ -222,6 +229,9 @@ public class AmaExtension : Extension() {
 			check {
 				anyGuild()
 				failIfNot { event.interaction.componentId.contains("ama-button.") }
+				with(amaData) {
+					userChecks()
+				}
 			}
 
 			action {
@@ -247,8 +257,13 @@ public class AmaExtension : Extension() {
 				val channel = guildFor(event)?.getChannelOf<GuildMessageChannel>(channelId)
 
 				val originalInteractionUser = event.interaction.user
+				val pkMember = checkPK(originalInteractionUser, modalObj.pkId.value, interactionResponse)
+				if (pkMember !is PKResult.Success) {
+					return@action
+				}
+
 				val embedMessage = channel?.createEmbed {
-					questionEmbed(originalInteractionUser, modalObj.question.value, QuestionStatusFlag.NO_FLAG)
+					questionEmbed(originalInteractionUser, modalObj.question.value, QuestionStatusFlag.NO_FLAG, pkMember.result)
 				}
 
 				val answerQueueChannel = guildFor(event)?.getChannelOf<GuildMessageChannel>(config.answerQueueChannel)
@@ -264,6 +279,7 @@ public class AmaExtension : Extension() {
 						questionComponents(
 							embedMessage,
 							originalInteractionUser,
+							pkMember.result,
 							modalObj.question.value,
 							answerQueueChannel,
 							liveChatChannel,
@@ -283,7 +299,12 @@ public class AmaExtension : Extension() {
 
 			initialResponse = InitialSlashCommandResponse.None
 
-			check { anyGuild() }
+			check {
+				anyGuild()
+				with(amaData) {
+					userChecks()
+				}
+			}
 
 			action {
 				if (amaData.isButtonEnabled(guild!!.id) == false) {
@@ -311,8 +332,19 @@ public class AmaExtension : Extension() {
 				val channel = guild?.getChannelOf<GuildMessageChannel>(channelId)
 
 				val originalInteractionUser = event.interaction.user
+
+				val pkMember = checkPK(originalInteractionUser, modalObj.pkId.value, interactionResponse)
+				if (pkMember !is PKResult.Success) {
+					return@action
+				}
+
 				val embedMessage = channel?.createEmbed {
-					questionEmbed(originalInteractionUser, modalObj.question.value, QuestionStatusFlag.NO_FLAG)
+					questionEmbed(
+						originalInteractionUser,
+						modalObj.question.value,
+						QuestionStatusFlag.NO_FLAG,
+						pkMember.result
+					)
 				}
 
 				val answerQueueChannel = guild?.getChannelOf<GuildMessageChannel>(config.answerQueueChannel)
@@ -328,6 +360,7 @@ public class AmaExtension : Extension() {
 						questionComponents(
 							embedMessage,
 							originalInteractionUser,
+							pkMember.result,
 							modalObj.question.value,
 							answerQueueChannel,
 							liveChatChannel,
@@ -348,6 +381,56 @@ public class AmaExtension : Extension() {
 		return topGuildChannel.getEffectivePermissions(kord.selfId).contains(permissions)
 	}
 
+	private suspend fun checkPK(
+		user: User,
+		pkId: String?,
+		interactionResponse: MessageInteractionResponseBehavior?
+	): PKResult<PKMember?> {
+		if (pkId == null) {
+			return PKResult.Success(null)
+		}
+
+		val result = user.getPluralKitMember(pkId)
+		when (result) {
+			PKResult.SystemNotFound -> {
+				if (pkId.isBlank()) {
+					return PKResult.Success(null)
+				}
+
+				interactionResponse?.createEphemeralFollowup {
+					content = "**Error**: No PK system is associated with your Discord account. Please try " +
+						"again, leaving the field blank."
+				}
+			}
+			PKResult.SystemNotAccessible -> {
+				if (pkId.isBlank()) {
+					return PKResult.Success(null)
+				}
+
+				interactionResponse?.createEphemeralFollowup {
+					content = "**Error**: The PK system associated with your Discord account is not " +
+						"accessible. Please try again, leaving the field blank, or contact a staff member."
+				}
+			}
+			PKResult.MemberNotFound -> interactionResponse?.createEphemeralFollowup {
+				content = "**Error**: No PK member was found with the provided information. If you used " +
+					"a member's name, please try again with their ID or UUID."
+			}
+			PKResult.MemberNotFromUser -> interactionResponse?.createEphemeralFollowup {
+				content = "**Error**: The PK member you provided is not from your system. Please try " +
+					"again with a member from your system, or leave the field blank."
+			}
+			PKResult.NotPermitted -> interactionResponse?.createEphemeralFollowup {
+				content = "**Error**: The PK system or member not accessible. Please try again, leaving the field " +
+					"blank, or contact a staff member."
+			}
+			PKResult.NoFronter -> return PKResult.Success(null)
+			is PKResult.Success -> return result
+		}
+
+		return result
+	}
+
 	public inner class AmaConfigArgs : Arguments() {
 		public val answerQueueChannel: Channel by channel {
 			name = "answer-queue-channel"
@@ -436,5 +519,11 @@ public class AmaExtension : Extension() {
 			placeholder = "What are you doing today?"
 			required = true
 		}
+
+		public val pkId: LineTextWidget = lineText {
+			label = "PluralKit Member (Optional)"
+			placeholder = "Either a member name, ID, or UUID"
+			required = false
+		}
 	}
 }
diff --git a/module-ama/src/main/kotlin/org/quiltmc/community/cozy/modules/ama/_PkUtils.kt b/module-ama/src/main/kotlin/org/quiltmc/community/cozy/modules/ama/_PkUtils.kt
new file mode 100644
index 00000000..9a4e4a96
--- /dev/null
+++ b/module-ama/src/main/kotlin/org/quiltmc/community/cozy/modules/ama/_PkUtils.kt
@@ -0,0 +1,183 @@
+package org.quiltmc.community.cozy.modules.ama
+
+import com.kotlindiscord.kord.extensions.modules.extra.pluralkit.api.PKMemberPrivacy
+import com.kotlindiscord.kord.extensions.modules.extra.pluralkit.api.PKProxyTag
+import com.kotlindiscord.kord.extensions.modules.extra.pluralkit.api.PKSystem
+import com.kotlindiscord.kord.extensions.utils.envOrNull
+import com.kotlindiscord.kord.extensions.utils.getKoin
+import dev.kord.common.Color
+import dev.kord.common.entity.optional.Optional
+import dev.kord.core.behavior.UserBehavior
+import io.ktor.client.*
+import io.ktor.client.call.*
+import io.ktor.client.plugins.contentnegotiation.*
+import io.ktor.client.request.*
+import io.ktor.client.statement.*
+import io.ktor.http.*
+import io.ktor.serialization.kotlinx.json.*
+import kotlinx.coroutines.delay
+import kotlinx.datetime.Clock
+import kotlinx.datetime.Instant
+import kotlinx.datetime.toInstant
+import kotlinx.serialization.SerialName
+import kotlinx.serialization.Serializable
+import kotlinx.serialization.json.Json
+import org.quiltmc.community.cozy.modules.ama.data.AmaData
+import kotlin.time.Duration.Companion.seconds
+
+private const val MEMBER_ID_LENGTH = 5
+private val uuidRegex = Regex("^[a-fA-F0-9]{8}(?:-[a-fA-F0-9]{4}){3}-[a-fA-F0-9]{12}$")
+
+private val baseUrl = envOrNull("PLURALKIT_BASE_URL") ?: "https://api.pluralkit.me/v2"
+
+private val client = HttpClient {
+	install(ContentNegotiation) {
+		json(
+			Json { ignoreUnknownKeys = true },
+			ContentType.Any
+		)
+	}
+}
+
+public sealed class PKResult<out T> {
+	public object SystemNotFound : PKResult<Nothing>()
+	public object SystemNotAccessible : PKResult<Nothing>()
+	public object MemberNotFound : PKResult<Nothing>()
+	public object MemberNotFromUser : PKResult<Nothing>()
+	public object NotPermitted : PKResult<Nothing>()
+	public object NoFronter : PKResult<Nothing>()
+	public data class Success<T>(val result: T) : PKResult<T>()
+}
+
+// Using a different PKMember than provided by kordex because it doesn't have the "last message timestamp"
+// or (not documented) "system" fields
+@Serializable
+public data class PKMember(
+	val id: String,
+	val uuid: String,
+	val system: Optional<String> = Optional.Missing(),
+	val name: String,
+	@SerialName("display_name")
+	val displayName: String?,
+	@Serializable(with = ColorHexCodeSerializer::class)
+	val color: Color?,
+	val pronouns: String?,
+	@SerialName("avatar_url")
+	val avatarUrl: String?,
+	@SerialName("webhook_avatar_url")
+	val webhookAvatarUrl: String?,
+	val banner: String?,
+	val description: String?,
+	val created: Instant?,
+	@SerialName("proxy_tags")
+	val proxyTags: List<PKProxyTag>,
+	@SerialName("keep_proxy")
+	val keepProxy: Boolean,
+	@SerialName("autoproxy_enabled")
+	val autoproxyEnabled: Boolean?,
+	@SerialName("message_count")
+	val messageCount: Int?,
+	@SerialName("last_message_timestamp")
+	val lastMessageTimestamp: Instant?,
+	val privacy: PKMemberPrivacy?,
+)
+
+@Serializable
+public data class PKMemberSwitch(
+	val id: String,
+	val timestamp: Instant,
+	val members: List<PKMember>
+)
+
+private suspend inline fun get(url: String, block: HttpRequestBuilder.() -> Unit = {}): HttpResponse {
+	while (true) {
+		val response = client.get(url, block)
+		if (response.status == HttpStatusCode.TooManyRequests) {
+			val retryAfter = response.headers["X-RateLimit-Reset"]?.toInstant()
+			if (retryAfter != null) {
+				val waitTime = retryAfter.toEpochMilliseconds() - Clock.System.now().toEpochMilliseconds()
+				if (waitTime > 0) {
+					delay(waitTime)
+				}
+			} else {
+				delay(1.seconds)
+			}
+		} else {
+			return response
+		}
+	}
+}
+
+public suspend fun UserBehavior.getPluralKitSystem(): PKResult<PKSystem> {
+	val systemRequest = client.get("$baseUrl/systems/$id")
+
+	if (systemRequest.status == HttpStatusCode.NotFound) {
+		return PKResult.SystemNotFound
+	} else if (systemRequest.status != HttpStatusCode.OK) {
+		return PKResult.SystemNotAccessible
+	}
+
+	return systemRequest.body()
+}
+
+public suspend fun UserBehavior.getPluralKitMember(text: String): PKResult<PKMember> {
+	val system = getPluralKitSystem()
+
+	if (system !is PKResult.Success) {
+		@Suppress("UNCHECKED_CAST")
+		return system as PKResult<PKMember>
+	}
+
+	val membersRequest = get("$baseUrl/systems/${system.result.id}/members")
+	if (membersRequest.status == HttpStatusCode.Forbidden) {
+		return PKResult.NotPermitted
+	} else if (membersRequest.status != HttpStatusCode.OK) {
+		return PKResult.SystemNotAccessible
+	}
+
+	val members: List<PKMember> = membersRequest.body()
+
+	val member = members.firstOrNull { it.name.equals(text, ignoreCase = true) }
+	if (member != null) {
+		return PKResult.Success(member)
+	}
+
+	if (text.matches(uuidRegex) || text.length == MEMBER_ID_LENGTH) {
+		val memberRequest = client.get("$baseUrl/members/$text")
+		if (memberRequest.status == HttpStatusCode.OK) {
+			val result: PKMember = memberRequest.body()
+			if (result.system.value == system.result.id) {
+				return PKResult.Success(result)
+			} else if (result.system !is Optional.Missing) {
+				return PKResult.MemberNotFromUser
+			}
+		}
+	} else if (text.isBlank()) {
+		if (getKoin().get<AmaData>().usePluralKitFronter(id)) {
+			val fronter = getMostRecentFronter()
+			if (fronter != null) {
+				return PKResult.Success(fronter)
+			}
+		}
+		return PKResult.NoFronter
+	}
+
+	return PKResult.MemberNotFound
+}
+
+public suspend fun UserBehavior.getFronters(): PKMemberSwitch? {
+	val switchesRequest = client.get("$baseUrl/systems/$id/fronters")
+	if (switchesRequest.status != HttpStatusCode.OK) {
+		return null
+	}
+
+	return switchesRequest.body()
+}
+
+public suspend fun UserBehavior.getMostRecentFronter(): PKMember? {
+	val fronters = getFronters() ?: return null
+
+	if (fronters.members.size == 1) return fronters.members.single()
+
+	return fronters.members.maxByOrNull { it.lastMessageTimestamp ?: Instant.DISTANT_PAST }
+}
diff --git a/module-ama/src/main/kotlin/org/quiltmc/community/cozy/modules/ama/_Utils.kt b/module-ama/src/main/kotlin/org/quiltmc/community/cozy/modules/ama/_Utils.kt
index 0dfe613c..b037a3e6 100644
--- a/module-ama/src/main/kotlin/org/quiltmc/community/cozy/modules/ama/_Utils.kt
+++ b/module-ama/src/main/kotlin/org/quiltmc/community/cozy/modules/ama/_Utils.kt
@@ -15,6 +15,7 @@ import com.kotlindiscord.kord.extensions.components.ComponentContainer
 import com.kotlindiscord.kord.extensions.components.components
 import com.kotlindiscord.kord.extensions.components.ephemeralButton
 import com.kotlindiscord.kord.extensions.utils.loadModule
+import dev.kord.common.Color
 import dev.kord.common.entity.ButtonStyle
 import dev.kord.common.entity.Snowflake
 import dev.kord.core.behavior.channel.createEmbed
@@ -26,6 +27,12 @@ import dev.kord.core.entity.channel.GuildMessageChannel
 import dev.kord.rest.builder.message.EmbedBuilder
 import dev.kord.rest.builder.message.create.embed
 import dev.kord.rest.builder.message.modify.embed
+import kotlinx.serialization.KSerializer
+import kotlinx.serialization.descriptors.PrimitiveKind
+import kotlinx.serialization.descriptors.PrimitiveSerialDescriptor
+import kotlinx.serialization.descriptors.SerialDescriptor
+import kotlinx.serialization.encoding.Decoder
+import kotlinx.serialization.encoding.Encoder
 import org.koin.dsl.bind
 import org.quiltmc.community.cozy.modules.ama.data.AmaData
 import org.quiltmc.community.cozy.modules.ama.enums.QuestionStatusFlag
@@ -36,18 +43,36 @@ public fun ExtensibleBotBuilder.ExtensionsBuilder.extAma(data: AmaData) {
 	add(::AmaExtension)
 }
 
+public val User.uniqueName: String
+	get() = if (this.discriminator == "0") {
+		// Migrated to new username system
+		"@${this.username}"
+	} else {
+		// Still using name#discrim system
+		this.tag
+	}
+
 public fun EmbedBuilder.questionEmbed(
 	interactionUser: User,
 	question: String?,
 	flag: QuestionStatusFlag,
+	pkMember: PKMember?,
 	flaggedBy: User? = null,
 	claimedOrSkippedBy: User? = null,
 	viaStage: Boolean? = null
 ) {
-	author {
-		name = "${interactionUser.tag} (${interactionUser.id})"
-		icon = interactionUser.avatar?.cdnUrl?.toUrl()
+	if (pkMember != null) {
+		author {
+			name = "${pkMember.displayName ?: pkMember.name} (${interactionUser.id})"
+			icon = pkMember.avatarUrl ?: pkMember.webhookAvatarUrl
+		}
+	} else {
+		author {
+			name = "${interactionUser.uniqueName} (${interactionUser.id})"
+			icon = interactionUser.avatar?.cdnUrl?.toUrl()
+		}
 	}
+
 	description = question
 	color = when (flag) {
 		QuestionStatusFlag.ACCEPTED, QuestionStatusFlag.CLAIMED -> DISCORD_GREEN
@@ -61,7 +86,7 @@ public fun EmbedBuilder.questionEmbed(
 		QuestionStatusFlag.FLAGGED ->
 			if (flaggedBy != null) {
 				footer {
-					text = "Flagged by ${flaggedBy.tag}"
+					text = "Flagged by ${flaggedBy.uniqueName}"
 					icon = flaggedBy.avatar?.cdnUrl?.toUrl()
 				}
 			}
@@ -69,7 +94,7 @@ public fun EmbedBuilder.questionEmbed(
 		QuestionStatusFlag.CLAIMED ->
 			if (claimedOrSkippedBy != null) {
 				footer {
-					text = "Claimed by ${claimedOrSkippedBy.tag}"
+					text = "Claimed by ${claimedOrSkippedBy.uniqueName}"
 					icon = claimedOrSkippedBy.avatar?.cdnUrl?.toUrl()
 				}
 			}
@@ -77,7 +102,7 @@ public fun EmbedBuilder.questionEmbed(
 		QuestionStatusFlag.SKIPPED ->
 			if (claimedOrSkippedBy != null) {
 				footer {
-					text = "Skipped by ${claimedOrSkippedBy.tag}"
+					text = "Skipped by ${claimedOrSkippedBy.uniqueName}"
 					icon = claimedOrSkippedBy.avatar?.cdnUrl?.toUrl()
 				}
 			}
@@ -86,7 +111,7 @@ public fun EmbedBuilder.questionEmbed(
 			if (claimedOrSkippedBy != null) {
 				val answerMethod = if (viaStage == true) "stage" else "text"
 				footer {
-					text = "Question answered via $answerMethod by ${claimedOrSkippedBy.tag}"
+					text = "Question answered via $answerMethod by ${claimedOrSkippedBy.uniqueName}"
 				}
 			}
 
@@ -97,6 +122,7 @@ public fun EmbedBuilder.questionEmbed(
 public suspend inline fun ComponentContainer.questionComponents(
 	embedMessage: Message,
 	interactionUser: User,
+	pkMember: PKMember?,
 	question: String?,
 	answerQueueChannel: GuildMessageChannel?,
 	liveChatChannel: GuildMessageChannel?,
@@ -111,7 +137,7 @@ public suspend inline fun ComponentContainer.questionComponents(
 			var answerQueueMessage: Message? = null
 			answerQueueMessage = answerQueueChannel?.createMessage {
 				embed {
-					questionEmbed(interactionUser, question, QuestionStatusFlag.ACCEPTED)
+					questionEmbed(interactionUser, question, QuestionStatusFlag.ACCEPTED, pkMember)
 				}
 				components {
 					ephemeralButton {
@@ -126,6 +152,7 @@ public suspend inline fun ComponentContainer.questionComponents(
 										interactionUser,
 										question,
 										QuestionStatusFlag.CLAIMED,
+										pkMember,
 										claimedOrSkippedBy = event.interaction.user
 									)
 								}
@@ -135,6 +162,7 @@ public suspend inline fun ComponentContainer.questionComponents(
 										claimer,
 										liveChatChannel,
 										interactionUser,
+										pkMember,
 										question,
 										answerQueueMessage
 									)
@@ -154,6 +182,7 @@ public suspend inline fun ComponentContainer.questionComponents(
 										interactionUser,
 										question,
 										QuestionStatusFlag.SKIPPED,
+										pkMember,
 										claimedOrSkippedBy = event.interaction.user
 									)
 								}
@@ -168,7 +197,8 @@ public suspend inline fun ComponentContainer.questionComponents(
 					questionEmbed(
 						interactionUser,
 						question,
-						QuestionStatusFlag.ACCEPTED
+						QuestionStatusFlag.ACCEPTED,
+						pkMember
 					)
 				}
 				components { removeAll() }
@@ -183,7 +213,7 @@ public suspend inline fun ComponentContainer.questionComponents(
 		action {
 			embedMessage.edit {
 				embed {
-					questionEmbed(interactionUser, question, QuestionStatusFlag.DENIED)
+					questionEmbed(interactionUser, question, QuestionStatusFlag.DENIED, pkMember)
 				}
 				components { removeAll() }
 			}
@@ -199,7 +229,13 @@ public suspend inline fun ComponentContainer.questionComponents(
 				var flaggedMessage: Message? = null
 				flaggedMessage = flaggedQueueChannel?.createMessage {
 					embed {
-						questionEmbed(interactionUser, question, QuestionStatusFlag.FLAGGED, event.interaction.user)
+						questionEmbed(
+							interactionUser,
+							question,
+							QuestionStatusFlag.FLAGGED,
+							pkMember,
+							flaggedBy = event.interaction.user
+						)
 					}
 					components {
 						ephemeralButton {
@@ -210,7 +246,7 @@ public suspend inline fun ComponentContainer.questionComponents(
 								var answerQueueMessage: Message? = null
 								answerQueueMessage = answerQueueChannel?.createMessage {
 									embed {
-										questionEmbed(interactionUser, question, QuestionStatusFlag.ACCEPTED)
+										questionEmbed(interactionUser, question, QuestionStatusFlag.ACCEPTED, pkMember)
 									}
 									components {
 										ephemeralButton {
@@ -225,6 +261,7 @@ public suspend inline fun ComponentContainer.questionComponents(
 															interactionUser,
 															question,
 															QuestionStatusFlag.CLAIMED,
+															pkMember,
 															claimedOrSkippedBy = event.interaction.user
 														)
 													}
@@ -234,6 +271,7 @@ public suspend inline fun ComponentContainer.questionComponents(
 															claimer,
 															liveChatChannel,
 															interactionUser,
+															pkMember,
 															question,
 															answerQueueMessage
 														)
@@ -253,6 +291,7 @@ public suspend inline fun ComponentContainer.questionComponents(
 															interactionUser,
 															question,
 															QuestionStatusFlag.SKIPPED,
+															pkMember,
 															claimedOrSkippedBy = event.interaction.user
 														)
 													}
@@ -267,7 +306,8 @@ public suspend inline fun ComponentContainer.questionComponents(
 										questionEmbed(
 											interactionUser,
 											question,
-											QuestionStatusFlag.ACCEPTED
+											QuestionStatusFlag.ACCEPTED,
+											pkMember
 										)
 									}
 									components { removeAll() }
@@ -277,7 +317,8 @@ public suspend inline fun ComponentContainer.questionComponents(
 										questionEmbed(
 											interactionUser,
 											question,
-											QuestionStatusFlag.ACCEPTED
+											QuestionStatusFlag.ACCEPTED,
+											pkMember
 										)
 									}
 									components { removeAll() }
@@ -292,7 +333,7 @@ public suspend inline fun ComponentContainer.questionComponents(
 							action {
 								flaggedMessage?.edit {
 									embed {
-										questionEmbed(interactionUser, question, QuestionStatusFlag.DENIED)
+										questionEmbed(interactionUser, question, QuestionStatusFlag.DENIED, pkMember)
 									}
 									components { removeAll() }
 								}
@@ -301,7 +342,8 @@ public suspend inline fun ComponentContainer.questionComponents(
 										questionEmbed(
 											interactionUser,
 											question,
-											QuestionStatusFlag.DENIED
+											QuestionStatusFlag.DENIED,
+											pkMember
 										)
 									}
 									components { removeAll() }
@@ -316,6 +358,7 @@ public suspend inline fun ComponentContainer.questionComponents(
 							interactionUser,
 							question,
 							QuestionStatusFlag.FLAGGED,
+							pkMember,
 							flaggedBy = event.interaction.user
 						)
 					}
@@ -330,6 +373,7 @@ public suspend inline fun ComponentContainer.answeringButtons(
 	claimer: User,
 	liveChatChannel: GuildMessageChannel?,
 	interactionUser: User,
+	pkMember: PKMember?,
 	question: String?,
 	answerQueueMessage: Message?
 ) {
@@ -349,6 +393,7 @@ public suspend inline fun ComponentContainer.answeringButtons(
 					interactionUser,
 					question,
 					QuestionStatusFlag.ANSWERED,
+					pkMember,
 					claimedOrSkippedBy = claimer,
 					viaStage = true
 				)
@@ -372,6 +417,7 @@ public suspend inline fun ComponentContainer.answeringButtons(
 					interactionUser,
 					question,
 					QuestionStatusFlag.ANSWERED,
+					pkMember,
 					claimedOrSkippedBy = claimer,
 					viaStage = false
 				)
@@ -380,3 +426,16 @@ public suspend inline fun ComponentContainer.answeringButtons(
 		}
 	}
 }
+
+@Suppress("MagicNumber")
+internal object ColorHexCodeSerializer : KSerializer<Color> {
+	override val descriptor: SerialDescriptor =
+		PrimitiveSerialDescriptor("Color", PrimitiveKind.STRING)
+
+	override fun deserialize(decoder: Decoder): Color =
+		Color(decoder.decodeString().trimStart('#').toInt(16))
+
+	override fun serialize(encoder: Encoder, value: Color) {
+		encoder.encodeString(value.rgb.toString(16))
+	}
+}
diff --git a/module-ama/src/main/kotlin/org/quiltmc/community/cozy/modules/ama/data/AmaData.kt b/module-ama/src/main/kotlin/org/quiltmc/community/cozy/modules/ama/data/AmaData.kt
index 8b560701..4df4c234 100644
--- a/module-ama/src/main/kotlin/org/quiltmc/community/cozy/modules/ama/data/AmaData.kt
+++ b/module-ama/src/main/kotlin/org/quiltmc/community/cozy/modules/ama/data/AmaData.kt
@@ -6,6 +6,9 @@
 
 package org.quiltmc.community.cozy.modules.ama.data
 
+import com.kotlindiscord.kord.extensions.checks.hasPermission
+import com.kotlindiscord.kord.extensions.checks.types.CheckContextWithCache
+import dev.kord.common.entity.Permission
 import dev.kord.common.entity.Snowflake
 
 public interface AmaData {
@@ -16,4 +19,13 @@ public interface AmaData {
 	public suspend fun modifyButton(guildId: Snowflake, enabled: Boolean)
 
 	public suspend fun setConfig(config: AmaConfig)
+
+	public suspend fun usePluralKitFronter(user: Snowflake): Boolean
+
+	public suspend fun CheckContextWithCache<*>.managementChecks() {
+		hasPermission(Permission.ManageGuild)
+	}
+
+	public suspend fun CheckContextWithCache<*>.userChecks() {
+	}
 }
diff --git a/module-ama/src/main/kotlin/org/quiltmc/community/cozy/modules/ama/data/MemoryAmaData.kt b/module-ama/src/main/kotlin/org/quiltmc/community/cozy/modules/ama/data/MemoryAmaData.kt
index ba0f7cff..a64c5b9c 100644
--- a/module-ama/src/main/kotlin/org/quiltmc/community/cozy/modules/ama/data/MemoryAmaData.kt
+++ b/module-ama/src/main/kotlin/org/quiltmc/community/cozy/modules/ama/data/MemoryAmaData.kt
@@ -9,42 +9,34 @@ package org.quiltmc.community.cozy.modules.ama.data
 import dev.kord.common.entity.Snowflake
 
 public class MemoryAmaData : AmaData {
-	private val ama: MutableList<AmaConfig> = mutableListOf()
+	private val ama: MutableMap<Snowflake, AmaConfig> = mutableMapOf()
 
 	override suspend fun getConfig(guildId: Snowflake): AmaConfig? =
-		ama.firstOrNull {
-			it.guildId == guildId
-		}
+		ama[guildId]
 
 	override suspend fun isButtonEnabled(guildId: Snowflake): Boolean? =
-		ama.firstOrNull {
-			it.guildId == guildId
-		}?.enabled
+		ama[guildId]?.enabled
 
 	override suspend fun modifyButton(guildId: Snowflake, enabled: Boolean) {
-		val amaConfig = ama.firstOrNull {
-			it.guildId == guildId
-		}
-
-		ama.remove(amaConfig)
-
-		ama.add(
-			AmaConfig(
-				amaConfig!!.guildId,
-				amaConfig.answerQueueChannel,
-				amaConfig.liveChatChannel,
-				amaConfig.buttonChannel,
-				amaConfig.approvalQueueChannel,
-				amaConfig.flaggedQuestionChannel,
-				amaConfig.embedConfig,
-				amaConfig.buttonMessage,
-				amaConfig.buttonId,
-				enabled
-			)
+		val amaConfig = ama[guildId]
+
+		ama[guildId] = AmaConfig(
+			amaConfig!!.guildId,
+			amaConfig.answerQueueChannel,
+			amaConfig.liveChatChannel,
+			amaConfig.buttonChannel,
+			amaConfig.approvalQueueChannel,
+			amaConfig.flaggedQuestionChannel,
+			amaConfig.embedConfig,
+			amaConfig.buttonMessage,
+			amaConfig.buttonId,
+			enabled
 		)
 	}
 
 	override suspend fun setConfig(config: AmaConfig) {
-		ama.add(config)
+		ama[config.guildId] = config
 	}
+
+	override suspend fun usePluralKitFronter(user: Snowflake): Boolean = false
 }
diff --git a/src/main/kotlin/org/quiltmc/community/database/collections/AmaConfigCollection.kt b/src/main/kotlin/org/quiltmc/community/database/collections/AmaConfigCollection.kt
index 975a0cc3..d715214d 100644
--- a/src/main/kotlin/org/quiltmc/community/database/collections/AmaConfigCollection.kt
+++ b/src/main/kotlin/org/quiltmc/community/database/collections/AmaConfigCollection.kt
@@ -6,6 +6,7 @@
 
 package org.quiltmc.community.database.collections
 
+import com.kotlindiscord.kord.extensions.checks.types.CheckContextWithCache
 import com.kotlindiscord.kord.extensions.koin.KordExKoinComponent
 import dev.kord.common.entity.Snowflake
 import org.koin.core.component.inject
@@ -16,10 +17,12 @@ import org.quiltmc.community.cozy.modules.ama.data.AmaData
 import org.quiltmc.community.database.Collection
 import org.quiltmc.community.database.Database
 import org.quiltmc.community.database.entities.AmaEntity
+import org.quiltmc.community.hasBaseModeratorRole
 
 class AmaConfigCollection : KordExKoinComponent, AmaData {
 	private val database: Database by inject()
 	private val col = database.mongo.getCollection<AmaEntity>(name)
+	private val userFlags: UserFlagsCollection by inject()
 
 	override suspend fun getConfig(guildId: Snowflake): AmaConfig? =
 		col.findOne(AmaConfig::guildId eq guildId)?.toAmaConfig()
@@ -35,5 +38,11 @@ class AmaConfigCollection : KordExKoinComponent, AmaData {
 		col.save(AmaEntity.fromAmaConfig(config))
 	}
 
+	override suspend fun usePluralKitFronter(user: Snowflake): Boolean =
+		userFlags.get(user)?.usePKFronter ?: false
+
+	override suspend fun CheckContextWithCache<*>.managementChecks() =
+		hasBaseModeratorRole(includeCommunityManagers = true)
+
 	companion object : Collection("ama_config")
 }
diff --git a/src/main/kotlin/org/quiltmc/community/database/entities/UserFlags.kt b/src/main/kotlin/org/quiltmc/community/database/entities/UserFlags.kt
index 0d8555a0..036a64e1 100644
--- a/src/main/kotlin/org/quiltmc/community/database/entities/UserFlags.kt
+++ b/src/main/kotlin/org/quiltmc/community/database/entities/UserFlags.kt
@@ -21,6 +21,7 @@ data class UserFlags(
 	var hasUsedPK: Boolean = false,
 	var autoPublish: Boolean = true,
 	var syncNicks: Boolean = true,
+	var usePKFronter: Boolean = false,
 ) : Entity<Snowflake> {
 	suspend fun save() {
 		val collection = getKoin().get<UserFlagsCollection>()
diff --git a/src/main/kotlin/org/quiltmc/community/database/migrations/v23.kt b/src/main/kotlin/org/quiltmc/community/database/migrations/v23.kt
index 42015bb2..136eeb98 100644
--- a/src/main/kotlin/org/quiltmc/community/database/migrations/v23.kt
+++ b/src/main/kotlin/org/quiltmc/community/database/migrations/v23.kt
@@ -7,8 +7,16 @@
 package org.quiltmc.community.database.migrations
 
 import org.litote.kmongo.coroutine.CoroutineDatabase
+import org.litote.kmongo.exists
+import org.litote.kmongo.setTo
 import org.quiltmc.community.database.collections.AmaConfigCollection
+import org.quiltmc.community.database.collections.UserFlagsCollection
+import org.quiltmc.community.database.entities.UserFlags
 
 suspend fun v23(db: CoroutineDatabase) {
 	db.createCollection(AmaConfigCollection.name)
+	db.getCollection<UserFlagsCollection>(UserFlagsCollection.name).updateMany(
+		UserFlags::usePKFronter.exists(false),
+		UserFlags::usePKFronter setTo false
+	)
 }