Add ClaimableBalanceKey (#338)

Synesso · Sep 20, 2020 · 3588573 · 3588573
1 parent 6c4e513
commit 3588573
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 3 deletions.
diff --git a/src/main/scala/stellar/sdk/model/ledger/LedgerKey.scala b/src/main/scala/stellar/sdk/model/ledger/LedgerKey.scala
@@ -1,6 +1,7 @@
 package stellar.sdk.model.ledger
 
 import cats.data.State
+import okio.ByteString
 import stellar.sdk.model.xdr.{Decode, Encodable, Encode}
 import stellar.sdk.model.{Asset, NonNativeAsset}
 import stellar.sdk.{KeyPair, PublicKeyOps}
@@ -13,7 +14,8 @@ object LedgerKey extends Decode {
     widen(AccountKey.decode),
     widen(TrustLineKey.decode),
     widen(OfferKey.decode),
-    widen(DataKey.decode)
+    widen(DataKey.decode),
+    widen(ClaimableBalanceKey.decode)
   )
 }
 
@@ -57,3 +59,13 @@ object DataKey extends Decode {
     name <- string
   } yield DataKey(account, name)
 }
+
+case class ClaimableBalanceKey(id: ByteString) extends LedgerKey {
+  override def encode: LazyList[Byte] = Encode.int(4) ++ Encode.bytes(32, id.toByteArray)
+}
+
+object ClaimableBalanceKey extends Decode {
+  val decode: State[Seq[Byte], ClaimableBalanceKey] = for {
+    id <- bytes(32).map(_.toArray).map(new ByteString(_))
+  } yield ClaimableBalanceKey(id)
+}
diff --git a/src/test/scala/stellar/sdk/model/ledger/LedgerEntryGenerators.scala b/src/test/scala/stellar/sdk/model/ledger/LedgerEntryGenerators.scala
@@ -38,9 +38,13 @@ trait LedgerEntryGenerators extends ArbitraryInput {
     name <- Gen.identifier
   } yield DataKey(account, name)
 
-  val genLedgerKey: Gen[LedgerKey] = Gen.oneOf(genAccountKey, genTrustLineKey, genOfferKey, genDataKey)
+  val genClaimableBalanceKey: Gen[ClaimableBalanceKey] = for {
+    id <- Gen.containerOfN[Array, Byte](32, Gen.posNum[Byte]).map(new ByteString(_))
+  } yield ClaimableBalanceKey(id)
+
+  val genLedgerKey: Gen[LedgerKey] = Gen.oneOf(genAccountKey, genTrustLineKey, genOfferKey, genDataKey, genClaimableBalanceKey)
 
-  implicit val arbLedgerKey = Arbitrary(genLedgerKey)
+  implicit val arbLedgerKey: Arbitrary[LedgerKey] = Arbitrary(genLedgerKey)
 
 
   // LedgerEntries