Suspending paginator DSLs

Kord-Extensions · Aug 10, 2021 · d0a9171 · d0a9171
1 parent 6e59997
commit d0a9171
Show file tree

Hide file tree

Showing 2 changed files with 23 additions and 2 deletions.
diff --git a/...sions/src/main/kotlin/com/kotlindiscord/kord/extensions/commands/MessageCommandContext.kt b/...sions/src/main/kotlin/com/kotlindiscord/kord/extensions/commands/MessageCommandContext.kt
@@ -18,6 +18,7 @@ import dev.kord.core.entity.Message
 import dev.kord.core.entity.User
 import dev.kord.core.event.message.MessageCreateEvent
 import dev.kord.rest.builder.message.create.MessageCreateBuilder
+import dev.kord.rest.builder.message.modify.MessageModifyBuilder
 
 /**
  * Command context object representing the context given to message commands.
@@ -85,7 +86,7 @@ public open class MessageCommandContext<T : Arguments>(
         targetChannel: MessageChannelBehavior? = null,
         targetMessage: Message? = null,
 
-        body: PaginatorBuilder.() -> Unit
+        body: suspend PaginatorBuilder.() -> Unit
     ): MessageButtonPaginator {
         val builder = PaginatorBuilder(command.extension, getLocale(), defaultGroup = defaultGroup)
 
@@ -137,4 +138,24 @@ public open class MessageCommandContext<T : Arguments>(
 
         return components
     }
+
+    /**
+     * Convenience function for adding components to your message via the [Components] class.
+     *
+     * @see Components
+     */
+    public suspend fun MessageModifyBuilder.components(
+        timeoutSeconds: Long? = null,
+        body: suspend Components.() -> Unit
+    ): Components {
+        val components = Components(command.extension)
+
+        body(components)
+
+        with(components) {
+            setup(timeoutSeconds)
+        }
+
+        return components
+    }
 }
diff --git a/...s/src/main/kotlin/com/kotlindiscord/kord/extensions/commands/slash/SlashCommandContext.kt b/...s/src/main/kotlin/com/kotlindiscord/kord/extensions/commands/slash/SlashCommandContext.kt
@@ -97,7 +97,7 @@ public open class SlashCommandContext<T : Arguments>(
      */
     public suspend fun paginator(
         defaultGroup: String = "",
-        body: PaginatorBuilder.() -> Unit
+        body: suspend PaginatorBuilder.() -> Unit
     ): InteractionButtonPaginator {
         val builder = PaginatorBuilder(command.extension, getLocale(), defaultGroup = defaultGroup)