From c78c648b4e5fe99a2f7ddf81a51ed144c7f6a0dc Mon Sep 17 00:00:00 2001
From: Bartosz Nowak <bartekn337@gmail.com>
Date: Mon, 29 Apr 2024 15:40:03 +0200
Subject: [PATCH] deposit&balance in peer

---
 crates/peer/src/registry.rs | 28 +++++++++++++++++++++++++---
 1 file changed, 25 insertions(+), 3 deletions(-)

diff --git a/crates/peer/src/registry.rs b/crates/peer/src/registry.rs
index 342ec86..de07a93 100644
--- a/crates/peer/src/registry.rs
+++ b/crates/peer/src/registry.rs
@@ -79,7 +79,7 @@ where
         Box::pin(stream)
     }
 
-    pub async fn stake<S>(
+    pub async fn deposit<S>(
         &self,
         amount: FieldElement,
         account: SingleOwnerAccount<P, S>,
@@ -90,14 +90,36 @@ where
         let result = account
             .execute(vec![Call {
                 to: self.registry_address,
-                selector: get_selector_from_name("stake").unwrap(),
+                selector: get_selector_from_name("deposit").unwrap(),
                 calldata: vec![amount],
             }])
             .send()
             .await
             .unwrap();
 
-        trace!("Stake result: {:?}", result);
+        trace!("Deposit result: {:?}", result);
+        Ok(())
+    }
+
+    pub async fn balance<S>(
+        &self,
+        target: FieldElement,
+        account: SingleOwnerAccount<P, S>,
+    ) -> Result<(), Box<dyn Error>>
+    where
+        S: Signer + Sync + Send + 'static,
+    {
+        let result = account
+            .execute(vec![Call {
+                to: self.registry_address,
+                selector: get_selector_from_name("balance").unwrap(),
+                calldata: vec![target],
+            }])
+            .send()
+            .await
+            .unwrap();
+
+        trace!("Balance result: {:?}", result);
         Ok(())
     }
 }