-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
1 parent
0afa3a1
commit 6a4917f
Showing
4 changed files
with
353 additions
and
5 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
349 changes: 349 additions & 0 deletions
349
src/main/kotlin/io/ileukocyte/hibernum/commands/fun/SessionsCommand.kt
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,349 @@ | ||
| package io.ileukocyte.hibernum.commands.`fun` | ||
|
|
||
| import com.google.common.collect.Lists | ||
|
|
||
| import io.ileukocyte.hibernum.Immutable | ||
| import io.ileukocyte.hibernum.builders.buildEmbed | ||
| import io.ileukocyte.hibernum.commands.CommandCategory | ||
| import io.ileukocyte.hibernum.commands.CommandException | ||
| import io.ileukocyte.hibernum.commands.GenericCommand.StaleInteractionHandling | ||
| import io.ileukocyte.hibernum.commands.SlashOnlyCommand | ||
| import io.ileukocyte.hibernum.extensions.* | ||
| import io.ileukocyte.hibernum.extensions.EmbedType.SUCCESS | ||
| import io.ileukocyte.hibernum.extensions.EmbedType.WARNING | ||
| import io.ileukocyte.hibernum.utils.* | ||
|
|
||
| import java.util.concurrent.TimeUnit | ||
|
|
||
| import kotlin.math.ceil | ||
| import kotlin.math.max | ||
| import kotlin.math.min | ||
|
|
||
| import net.dv8tion.jda.api.JDA | ||
| import net.dv8tion.jda.api.entities.Guild | ||
| import net.dv8tion.jda.api.entities.GuildMessageChannel | ||
| import net.dv8tion.jda.api.events.interaction.ModalInteractionEvent | ||
| import net.dv8tion.jda.api.events.interaction.command.SlashCommandInteractionEvent | ||
| import net.dv8tion.jda.api.events.interaction.component.ButtonInteractionEvent | ||
| import net.dv8tion.jda.api.interactions.commands.OptionType.STRING | ||
| import net.dv8tion.jda.api.interactions.commands.build.OptionData | ||
| import net.dv8tion.jda.api.interactions.components.Modal | ||
| import net.dv8tion.jda.api.interactions.components.buttons.Button | ||
| import net.dv8tion.jda.api.interactions.components.text.TextInput | ||
| import net.dv8tion.jda.api.interactions.components.text.TextInputStyle | ||
|
|
||
| import org.jetbrains.kotlin.utils.addToStdlib.applyIf | ||
|
|
||
| class SessionsCommand : SlashOnlyCommand { | ||
| override val name = "sessions" | ||
| override val description = "Sends a list of running game sessions of yours or aborts the one provided by its ID" | ||
| override val options = setOf(OptionData(STRING, "id", "The ID of the session to abort")) | ||
| override val staleInteractionHandling = StaleInteractionHandling.REMOVE_COMPONENTS | ||
| override val neglectProcessBlock = true | ||
|
|
||
| override suspend fun invoke(event: SlashCommandInteractionEvent) { | ||
| val sessions = event.jda.getUserProcesses(event.user) | ||
| .filter { it.command?.category == CommandCategory.FUN } | ||
| .takeUnless { it.isEmpty() } | ||
| ?: throw CommandException("No sessions of yours are currently running!") | ||
|
|
||
| val input = event.getOption("id")?.asString | ||
|
|
||
| if (input === null) { | ||
| val pages = ceil(sessions.size / 5.0).toInt() | ||
|
|
||
| event.replyEmbeds(sessionsListEmbed(sessions, 0, event.jda, event.guild ?: return)) | ||
| .addActionRow( | ||
| pageButtons(event.user.id, 0, pages).takeIf { sessions.size > 5 } | ||
| ?: setOf( | ||
| Button.primary("$name-${event.user.idLong}-abort", "Abort"), | ||
| Button.danger("$name-${event.user.idLong}-exit", "Close"), | ||
| ) | ||
| ).queue() | ||
| } else { | ||
| val session = event.jda.getProcessById(input) | ||
| ?.takeIf { event.user.idLong in it.users && it.command?.category == CommandCategory.FUN } | ||
| ?: throw CommandException("No session of yours has been found by the provided ID!") | ||
|
|
||
| event.replyConfirmation("Are you sure you want to abort the session?") | ||
| .addActionRow( | ||
| Button.danger("$name-${event.user.idLong}-${session.id}-abortc", "Yes"), | ||
| Button.secondary("$name-${event.user.idLong}-exit", "No"), | ||
| ).queue() | ||
| } | ||
| } | ||
|
|
||
| override suspend fun invoke(event: ButtonInteractionEvent) { | ||
| val guild = event.guild ?: return | ||
| val id = event.componentId.removePrefix("$name-").split("-") | ||
|
|
||
| if (event.user.id == id.first()) { | ||
| val type = id.last() | ||
| val sessions = event.jda.getUserProcesses(event.user) | ||
| .filter { it.command?.category == CommandCategory.FUN } | ||
| .takeUnless { it.isEmpty() } | ||
| ?: return | ||
|
|
||
| when (type) { | ||
| "exit" -> event.message.delete().queue() | ||
| "abort" -> { | ||
| val input = TextInput | ||
| .create("$name-id", "Enter the Session ID:", TextInputStyle.SHORT) | ||
| .setMaxLength(4) | ||
| .build() | ||
| val modal = Modal | ||
| .create("$name-modal", "Session Abortion") | ||
| .addActionRow(input) | ||
| .build() | ||
|
|
||
| event.replyModal(modal).queue() | ||
| } | ||
| "abortc" -> { | ||
| val session = event.jda.getProcessById(id[1]) | ||
| ?.takeIf { event.user.idLong in it.users && it.command?.category == CommandCategory.FUN } | ||
| ?: return | ||
|
|
||
| session.kill(event.jda) | ||
|
|
||
| if (session.command is AkinatorCommand) { | ||
| for (userId in session.users) { | ||
| AkinatorCommand.AKIWRAPPERS -= userId | ||
| AkinatorCommand.DECLINED_GUESSES -= userId | ||
| AkinatorCommand.GUESS_TYPES -= userId | ||
| } | ||
| } | ||
|
|
||
| event.editMessageEmbeds(defaultEmbed("The session has been aborted!", SUCCESS)) | ||
| .setComponents(emptyList()) | ||
| .queue(null) { | ||
| event.channel.sendSuccess("The session has been aborted!").queue() | ||
| } | ||
|
|
||
| val description = | ||
| "The ${session.command?.let { it::class.simpleName } ?: event.jda.selfUser.name} session " + | ||
| "running in this channel has been aborted!" | ||
|
|
||
| event.jda.getChannelById(GuildMessageChannel::class.java, session.channel)?.let { channel -> | ||
| session.invoker?.let { | ||
| channel.retrieveMessageById(it).await().delete().queue(null) {} | ||
| } | ||
|
|
||
| channel.sendMessage { | ||
| embeds += defaultEmbed(description, WARNING) { | ||
| text = "This message will self-delete in 5 seconds" | ||
| } | ||
|
|
||
| session.users.mapNotNull { event.jda.getUserById(it)?.asMention }.joinToString() | ||
| .takeUnless { it.isEmpty() } | ||
| ?.let { content += it } | ||
| }.queue({ it.delete().queueAfter(5, TimeUnit.SECONDS, null) {} }) {} | ||
| } | ||
| } | ||
| else -> { | ||
| val page = id[1].toInt() | ||
|
|
||
| when (type) { | ||
| "first" -> { | ||
| val pages = ceil(sessions.size / 5.0).toInt() | ||
|
|
||
| event.editMessageEmbeds(sessionsListEmbed(sessions, 0, event.jda, guild)) | ||
| .setActionRow( | ||
| pageButtons(id.first(), 0, pages).takeIf { sessions.size > 5 } | ||
| ?: setOf( | ||
| Button.primary("$name-${id.first()}-kill", "Kill"), | ||
| Button.danger("$name-${id.first()}-exit", "Close"), | ||
| ) | ||
| ).queue(null) { | ||
| event.message | ||
| .editMessageEmbeds(sessionsListEmbed(sessions, 0, event.jda, guild)) | ||
| .setActionRow( | ||
| pageButtons(id.first(), 0, pages).takeIf { sessions.size > 5 } | ||
| ?: setOf( | ||
| Button.primary("$name-${id.first()}-kill", "Kill"), | ||
| Button.danger("$name-${id.first()}-exit", "Close"), | ||
| ) | ||
| ).queue() | ||
| } | ||
| } | ||
| "last" -> { | ||
| val partition = Lists.partition(sessions.toList(), 5) | ||
| val lastPage = partition.lastIndex | ||
|
|
||
| event.editMessageEmbeds(sessionsListEmbed(sessions, lastPage, event.jda, guild)) | ||
| .setActionRow( | ||
| pageButtons(id.first(), lastPage, partition.size).takeIf { sessions.size > 5 } | ||
| ?: setOf(Button.danger("$name-${id.first()}-exit", "Close")) | ||
| ).queue(null) { | ||
| event.message | ||
| .editMessageEmbeds(sessionsListEmbed(sessions, lastPage, event.jda, guild)) | ||
| .setActionRow( | ||
| pageButtons(id.first(), lastPage, partition.size).takeIf { sessions.size > 5 } | ||
| ?: setOf( | ||
| Button.primary("$name-${id.first()}-kill", "Kill"), | ||
| Button.danger("$name-${id.first()}-exit", "Close"), | ||
| ) | ||
| ).queue() | ||
| } | ||
| } | ||
| "back" -> { | ||
| val newPage = max(0, page.dec()) | ||
| val pages = ceil(sessions.size / 5.0).toInt() | ||
|
|
||
| event.editMessageEmbeds(sessionsListEmbed(sessions, newPage, event.jda, guild)) | ||
| .setActionRow( | ||
| pageButtons(id.first(), newPage, pages).takeIf { sessions.size > 5 } | ||
| ?: setOf( | ||
| Button.primary("$name-${id.first()}-kill", "Kill"), | ||
| Button.danger("$name-${id.first()}-exit", "Close"), | ||
| ) | ||
| ).queue(null) { | ||
| event.message | ||
| .editMessageEmbeds(sessionsListEmbed(sessions, newPage, event.jda, guild)) | ||
| .setActionRow( | ||
| pageButtons(id.first(), newPage, pages).takeIf { sessions.size > 5 } | ||
| ?: setOf( | ||
| Button.primary("$name-${id.first()}-kill", "Kill"), | ||
| Button.danger("$name-${id.first()}-exit", "Close"), | ||
| ) | ||
| ).queue() | ||
| } | ||
| } | ||
| "next" -> { | ||
| val partition = Lists.partition(sessions.toList(), 5) | ||
| val lastPage = partition.lastIndex | ||
| val newPage = min(page.inc(), lastPage) | ||
|
|
||
| event.editMessageEmbeds(sessionsListEmbed(sessions, newPage, event.jda, guild)) | ||
| .setActionRow( | ||
| pageButtons(id.first(), newPage, partition.size).takeIf { sessions.size > 5 } | ||
| ?: setOf( | ||
| Button.primary("$name-${id.first()}-kill", "Kill"), | ||
| Button.danger("$name-${id.first()}-exit", "Close"), | ||
| ) | ||
| ).queue(null) { | ||
| event.message | ||
| .editMessageEmbeds(sessionsListEmbed(sessions, newPage, event.jda, guild)) | ||
| .setActionRow( | ||
| pageButtons(id.first(), newPage, partition.size).takeIf { sessions.size > 5 } | ||
| ?: setOf( | ||
| Button.primary("$name-${id.first()}-kill", "Kill"), | ||
| Button.danger("$name-${id.first()}-exit", "Close"), | ||
| ) | ||
| ).queue() | ||
| } | ||
| } | ||
| } | ||
| } | ||
| } | ||
| } else { | ||
| throw CommandException("You did not invoke the initial command!") | ||
| } | ||
| } | ||
|
|
||
| override suspend fun invoke(event: ModalInteractionEvent) { | ||
| val id = event.getValue("$name-id")?.asString ?: return | ||
| val sessions = event.jda.getProcessById(id) | ||
| ?.takeIf { event.user.idLong in it.users && it.command?.category == CommandCategory.FUN } | ||
| ?: throw CommandException("No session of yours has been found by the provided ID!") | ||
|
|
||
| sessions.kill(event.jda) | ||
|
|
||
| if (sessions.command is AkinatorCommand) { | ||
| for (userId in sessions.users) { | ||
| AkinatorCommand.AKIWRAPPERS -= userId | ||
| AkinatorCommand.DECLINED_GUESSES -= userId | ||
| AkinatorCommand.GUESS_TYPES -= userId | ||
| } | ||
| } | ||
|
|
||
| event.editMessageEmbeds(defaultEmbed("The session has been aborted!", SUCCESS)) | ||
| .setComponents(emptyList()) | ||
| .queue(null) { | ||
| event.messageChannel.sendSuccess("The session has been aborted!").queue() | ||
| } | ||
|
|
||
| val description = | ||
| "The ${sessions.command?.let { it::class.simpleName } ?: event.jda.selfUser.name} session " + | ||
| "running in this channel has been aborted!" | ||
|
|
||
| event.jda.getChannelById(GuildMessageChannel::class.java, sessions.channel)?.let { channel -> | ||
| sessions.invoker?.let { | ||
| channel.retrieveMessageById(it).await().delete().queue(null) {} | ||
| } | ||
|
|
||
| channel.sendMessage { | ||
| embeds += defaultEmbed(description, WARNING) { | ||
| text = "This message will self-delete in 5 seconds" | ||
| } | ||
|
|
||
| sessions.users.mapNotNull { event.jda.getUserById(it)?.asMention }.joinToString() | ||
| .takeUnless { it.isEmpty() } | ||
| ?.let { content += it } | ||
| }.queue({ it.delete().queueAfter(5, TimeUnit.SECONDS, null) {} }) {} | ||
| } | ||
| } | ||
|
|
||
| private fun sessionsListEmbed( | ||
| originalSet: Collection<WaiterProcess>, | ||
| page: Int, | ||
| jda: JDA, | ||
| guild: Guild, | ||
| ) = buildEmbed { | ||
| val partition = Lists.partition(originalSet.toList(), 5) | ||
| val sessions = partition[page] | ||
|
|
||
| color = Immutable.SUCCESS | ||
|
|
||
| for ((index, session) in sessions.withIndex()) { | ||
| val users = session.users.joinToString { id -> | ||
| jda.getUserById(id)?.let { | ||
| if (guild.isMember(it)) { | ||
| it.asMention | ||
| } else { | ||
| it.asTag | ||
| } | ||
| } ?: id.toString() | ||
| } | ||
|
|
||
| val channel = session.channel.let { id -> | ||
| jda.getChannelById(GuildMessageChannel::class.java, id)?.let { | ||
| if (it.guild.idLong == guild.idLong) { | ||
| it.asMention | ||
| } else { | ||
| "${it.name} (${it.guild.name.escapeMarkdown()}, ${it.guild.id})" | ||
| } | ||
| } ?: id.toString() | ||
| } | ||
|
|
||
| field { | ||
| title = "Session #${index.inc() + page * 5}" | ||
| description = """**Command**: ${session.command?.getEffectiveContextName() ?: "Unknown"} | ||
| **Session ID**: ${session.id} | ||
| **Users**: $users | ||
| **Channel**: $channel | ||
| """.trimIndent() | ||
| } | ||
| } | ||
|
|
||
| author { | ||
| name = "Session Manager" | ||
| iconUrl = jda.selfUser.effectiveAvatarUrl | ||
| } | ||
|
|
||
| if (partition.size > 1) { | ||
| footer { text = "Total sessions: ${originalSet.size}" } | ||
| } | ||
| } | ||
|
|
||
| private fun pageButtons(userId: String, page: Int, size: Int) = setOf( | ||
| Button.primary("$name-$userId-abort", "Abort"), | ||
| Button.secondary("$name-$userId-$page-first", "First Page") | ||
| .applyIf(page == 0) { asDisabled() }, | ||
| Button.secondary("$name-$userId-$page-back", "Back") | ||
| .applyIf(page == 0) { asDisabled() }, | ||
| Button.secondary("$name-$userId-$page-next", "Next") | ||
| .applyIf(page == size.dec()) { asDisabled() }, | ||
| Button.secondary("$name-$userId-$page-last", "Last Page") | ||
| .applyIf(page == size.dec()) { asDisabled() }, | ||
| ) | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters