feat: add rpc calls to compute vault collateral with and without nomi…

…nation Signed-off-by: Gregory Hill <[email protected]>
interlay · Jul 14, 2021 · 1f4c70f · 1f4c70f
1 parent f77d5f2
commit 1f4c70f
Show file tree

Hide file tree

Showing 4 changed files with 68 additions and 0 deletions.
diff --git a/crates/vault-registry/rpc/runtime-api/src/lib.rs b/crates/vault-registry/rpc/runtime-api/src/lib.rs
@@ -14,6 +14,12 @@ sp_api::decl_runtime_apis! {
         Collateral: Codec,
         UnsignedFixedPoint: Codec
     {
+        /// Get the vault's collateral (excluding nomination)
+        fn get_vault_collateral(vault_id: AccountId) -> Result<BalanceWrapper<Collateral>, DispatchError>;
+
+        /// Get the vault's collateral (including nomination)
+        fn get_vault_total_collateral(vault_id: AccountId) -> Result<BalanceWrapper<Collateral>, DispatchError>;
+
         /// Get the total collateralization of the system
         fn get_total_collateralization() -> Result<UnsignedFixedPoint, DispatchError>;
 

diff --git a/crates/vault-registry/rpc/src/lib.rs b/crates/vault-registry/rpc/src/lib.rs
@@ -22,6 +22,20 @@ where
     Collateral: Codec + MaybeDisplay + MaybeFromStr,
     UnsignedFixedPoint: Codec + MaybeDisplay + MaybeFromStr,
 {
+    #[rpc(name = "vaultRegistry_getVaultCollateral")]
+    fn get_vault_collateral(
+        &self,
+        vault_id: AccountId,
+        at: Option<BlockHash>,
+    ) -> JsonRpcResult<BalanceWrapper<Collateral>>;
+
+    #[rpc(name = "vaultRegistry_getVaultTotalCollateral")]
+    fn get_vault_total_collateral(
+        &self,
+        vault_id: AccountId,
+        at: Option<BlockHash>,
+    ) -> JsonRpcResult<BalanceWrapper<Collateral>>;
+
     #[rpc(name = "vaultRegistry_getTotalCollateralization")]
     fn get_total_collateralization(&self, at: Option<BlockHash>) -> JsonRpcResult<UnsignedFixedPoint>;
 
@@ -158,6 +172,34 @@ where
     Collateral: Codec + MaybeDisplay + MaybeFromStr,
     UnsignedFixedPoint: Codec + MaybeDisplay + MaybeFromStr,
 {
+    fn get_vault_collateral(
+        &self,
+        vault_id: AccountId,
+        at: Option<<Block as BlockT>::Hash>,
+    ) -> JsonRpcResult<BalanceWrapper<Collateral>> {
+        let api = self.client.runtime_api();
+        let at = BlockId::hash(at.unwrap_or_else(|| self.client.info().best_hash));
+
+        handle_response(
+            api.get_vault_collateral(&at, vault_id),
+            "Unable to get the vault's collateral.".into(),
+        )
+    }
+
+    fn get_vault_total_collateral(
+        &self,
+        vault_id: AccountId,
+        at: Option<<Block as BlockT>::Hash>,
+    ) -> JsonRpcResult<BalanceWrapper<Collateral>> {
+        let api = self.client.runtime_api();
+        let at = BlockId::hash(at.unwrap_or_else(|| self.client.info().best_hash));
+
+        handle_response(
+            api.get_vault_total_collateral(&at, vault_id),
+            "Unable to get the vault's collateral.".into(),
+        )
+    }
+
     fn get_total_collateralization(&self, at: Option<<Block as BlockT>::Hash>) -> JsonRpcResult<UnsignedFixedPoint> {
         let api = self.client.runtime_api();
         let at = BlockId::hash(at.unwrap_or_else(|| self.client.info().best_hash));

diff --git a/parachain/runtime/src/lib.rs b/parachain/runtime/src/lib.rs
@@ -1090,6 +1090,16 @@ impl_runtime_apis! {
         Balance,
         UnsignedFixedPoint
     > for Runtime {
+        fn get_vault_collateral(vault_id: AccountId) -> Result<BalanceWrapper<Balance>, DispatchError> {
+            let result = VaultRegistry::compute_collateral(&vault_id)?;
+            Ok(BalanceWrapper{amount:result})
+        }
+
+        fn get_vault_total_collateral(vault_id: AccountId) -> Result<BalanceWrapper<Balance>, DispatchError> {
+            let result = VaultRegistry::get_backing_collateral(&vault_id)?;
+            Ok(BalanceWrapper{amount:result})
+        }
+
         fn get_total_collateralization() -> Result<UnsignedFixedPoint, DispatchError> {
             VaultRegistry::get_total_collateralization()
         }

diff --git a/standalone/runtime/src/lib.rs b/standalone/runtime/src/lib.rs
@@ -664,6 +664,16 @@ impl_runtime_apis! {
         Balance,
         UnsignedFixedPoint
     > for Runtime {
+        fn get_vault_collateral(vault_id: AccountId) -> Result<BalanceWrapper<Balance>, DispatchError> {
+            let result = VaultRegistry::compute_collateral(&vault_id)?;
+            Ok(BalanceWrapper{amount:result})
+        }
+
+        fn get_vault_total_collateral(vault_id: AccountId) -> Result<BalanceWrapper<Balance>, DispatchError> {
+            let result = VaultRegistry::get_backing_collateral(&vault_id)?;
+            Ok(BalanceWrapper{amount:result})
+        }
+
         fn get_total_collateralization() -> Result<UnsignedFixedPoint, DispatchError> {
             VaultRegistry::get_total_collateralization()
         }