Add InputMedia encoder (bot4s#86)

mmvpm · Jun 2, 2024 · eba5108 · eba5108
1 parent f209aa3
commit eba5108
Show file tree

Hide file tree

Showing 2 changed files with 23 additions and 4 deletions.
diff --git a/core/src/com/bot4s/telegram/marshalling/CirceEncoders.scala b/core/src/com/bot4s/telegram/marshalling/CirceEncoders.scala
@@ -343,10 +343,16 @@ trait CirceEncoders {
  deriveConfiguredEncoder[DeclineChatJoinRequest]
 
  implicit val inputFileEncoder: Encoder[InputFile] = Encoder.instance {
- _ match {
- case InputFile.FileId(fileId) => fileId.asJson
- case _ => io.circe.Json.Null
- }
+ case InputFile.FileId(fileId) => fileId.asJson
+ case _ => io.circe.Json.Null
+ }
+
+ implicit val inputMediaEncoder: Encoder[InputMedia] = Encoder.instance {
+ case q: InputMediaPhoto => q.asJson
+ case q: InputMediaVideo => q.asJson
+ case q: InputMediaAnimation => q.asJson
+ case q: InputMediaAudio => q.asJson
+ case q: InputMediaDocument => q.asJson
  }
 
  implicit val sendLocationEncoder: Encoder[SendLocation] = deriveConfiguredEncoder[SendLocation]

diff --git a/core/test/src/com/bot4s/telegram/marshalling/MarshallingSuite.scala b/core/test/src/com/bot4s/telegram/marshalling/MarshallingSuite.scala
@@ -1,6 +1,7 @@
 package com.bot4s.telegram.marshalling
 
 import com.bot4s.telegram.api.TestUtils
+import com.bot4s.telegram.methods.SendMediaGroup
 import com.bot4s.telegram.models.CountryCode.CountryCode
 import com.bot4s.telegram.models.Currency.Currency
 import com.bot4s.telegram.models.MaskPositionType.MaskPositionType
@@ -183,4 +184,16 @@ class MarshallingSuite extends AnyFlatSpec with MockFactory with Matchers with T
  toJson[StickerFormat](StickerFormat.Video) shouldBe """"video""""
  }
 
+ it should "encode InputMediaPhoto" in {
+ val inputMedia = InputMediaPhoto("media", None)
+ toJson[InputMedia](inputMedia) shouldBe """{"media":"media","type":"photo"}"""
+ }
+
+ it should "encode SendMediaGroup" in {
+ val inputMedia = InputMediaPhoto("media", None)
+ val sendMediaGroup = SendMediaGroup(ChatId(123), Array(inputMedia))
+ toJson[SendMediaGroup](sendMediaGroup) shouldBe """{"chat_id":123,"media":[{"media":"media","type":"photo"}]}"""
+ }
+
+
 }