Remove webhookId from behavior

kordlib · Apr 5, 2021 · a58d340 · a58d340
1 parent 5942a23
commit a58d340
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 12 deletions.
diff --git a/core/src/main/kotlin/behavior/MessageBehavior.kt b/core/src/main/kotlin/behavior/MessageBehavior.kt
@@ -40,8 +40,6 @@ interface MessageBehavior : KordEntity, Strategizable {
      */
     val channel get() = MessageChannelBehavior(channelId, kord)
 
-    val webhookId: Snowflake?
-
     /**
      * Requests to get the channel this message was send in.
      */
@@ -200,21 +198,19 @@ interface MessageBehavior : KordEntity, Strategizable {
      */
     override fun withStrategy(
         strategy: EntitySupplyStrategy<*>,
-    ): MessageBehavior = MessageBehavior(channelId, id, kord, webhookId, strategy)
+    ): MessageBehavior = MessageBehavior(channelId, id, kord)
 
 }
 
 fun MessageBehavior(
     channelId: Snowflake,
     messageId: Snowflake,
     kord: Kord,
-    webhookId: Snowflake? = null,
     strategy: EntitySupplyStrategy<*> = kord.resources.defaultStrategy,
 ) = object : MessageBehavior {
     override val channelId: Snowflake = channelId
     override val id: Snowflake = messageId
     override val kord: Kord = kord
-    override val webhookId: Snowflake? = webhookId
     override val supplier: EntitySupplier = strategy.supply(kord)
 
     override fun hashCode(): Int = Objects.hash(id)
@@ -244,8 +240,6 @@ suspend inline fun MessageBehavior.edit(builder: MessageModifyBuilder.() -> Unit
         callsInPlace(builder, InvocationKind.EXACTLY_ONCE)
     }
 
-    require(webhookId == null) { "Cannot perform edits on webhook messages use editWebhook() instead" }
-
     val response =
         kord.rest.channel.editMessage(channelId = channelId, messageId = id, builder = builder)
     val data = MessageData.from(response)
@@ -264,18 +258,17 @@ suspend inline fun MessageBehavior.edit(builder: MessageModifyBuilder.() -> Unit
  */
 @OptIn(ExperimentalContracts::class)
 suspend inline fun MessageBehavior.editWebhookMessage(
+    webhookId: Snowflake,
     token: String,
     builder: EditWebhookMessageBuilder.() -> Unit
 ): Message {
     contract {
         callsInPlace(builder, InvocationKind.EXACTLY_ONCE)
     }
 
-    require(webhookId != null) { "Cannot perform edits on  non webhook messages use edit() instead" }
-
     val response =
         kord.rest.webhook.editWebhookMessage(
-            webhookId = webhookId!!,
+            webhookId = webhookId,
             messageId = id,
             token = token,
             builder = builder

diff --git a/core/src/main/kotlin/entity/Message.kt b/core/src/main/kotlin/entity/Message.kt
@@ -219,7 +219,7 @@ class Message(
      *
      * Returns null if this message was not send using a webhook.
      */
-    override val webhookId: Snowflake? get() = data.webhookId.value
+    val webhookId: Snowflake? get() = data.webhookId.value
 
     /**
      * Returns itself.

diff --git a/core/src/main/kotlin/entity/channel/MessageChannel.kt b/core/src/main/kotlin/entity/channel/MessageChannel.kt
@@ -8,7 +8,6 @@ import dev.kord.core.behavior.channel.MessageChannelBehavior
 import dev.kord.core.entity.Message
 import dev.kord.core.supplier.EntitySupplyStrategy
 import dev.kord.core.toInstant
-import dev.kord.core.toSnowflakeOrNull
 import java.time.Instant
 
 /**