From 4d51e3211740764764a6423f8ead4944e1790341 Mon Sep 17 00:00:00 2001
From: hozan23 <hozan23@proton.me>
Date: Sun, 19 Nov 2023 22:19:06 +0300
Subject: karyons jsonrpc implementation

---
 jsonrpc/src/client.rs | 77 +++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 77 insertions(+)
 create mode 100644 jsonrpc/src/client.rs

(limited to 'jsonrpc/src/client.rs')

diff --git a/jsonrpc/src/client.rs b/jsonrpc/src/client.rs
new file mode 100644
index 0000000..2863204
--- /dev/null
+++ b/jsonrpc/src/client.rs
@@ -0,0 +1,77 @@
+use log::debug;
+use serde::{de::DeserializeOwned, Serialize};
+
+use karyons_core::{async_utils::timeout, utils::random_32};
+use karyons_net::{dial, Conn, Endpoint};
+
+use crate::{
+    message,
+    utils::{read_until, write_all},
+    Error, Result, JSONRPC_VERSION,
+};
+
+/// Represents an RPC client
+pub struct Client {
+    conn: Conn,
+    timeout: Option<u64>,
+}
+
+impl Client {
+    /// Creates a new RPC client.
+    pub fn new(conn: Conn, timeout: Option<u64>) -> Self {
+        Self { conn, timeout }
+    }
+
+    /// Creates a new RPC client using the provided endpoint.
+    pub async fn new_with_endpoint(endpoint: &Endpoint, timeout: Option<u64>) -> Result<Self> {
+        let conn = dial(endpoint).await?;
+        Ok(Self { conn, timeout })
+    }
+
+    /// Calls the named method, waits for the response, and returns the result.
+    pub async fn call<T: Serialize + DeserializeOwned, V: DeserializeOwned>(
+        &self,
+        method: &str,
+        params: T,
+    ) -> Result<V> {
+        let id = serde_json::json!(random_32());
+
+        let request = message::Request {
+            jsonrpc: JSONRPC_VERSION.to_string(),
+            id,
+            method: method.to_string(),
+            params: serde_json::json!(params),
+        };
+
+        let payload = serde_json::to_vec(&request)?;
+        write_all(&self.conn, &payload).await?;
+        debug!("--> {request}");
+
+        let mut buffer = vec![];
+        if let Some(t) = self.timeout {
+            timeout(
+                std::time::Duration::from_secs(t),
+                read_until(&self.conn, &mut buffer),
+            )
+            .await?
+        } else {
+            read_until(&self.conn, &mut buffer).await
+        }?;
+
+        let response = serde_json::from_slice::<message::Response>(&buffer)?;
+        debug!("<-- {response}");
+
+        if let Some(error) = response.error {
+            return Err(Error::CallError(error.code, error.message));
+        }
+
+        if response.id.is_none() || response.id.unwrap() != request.id {
+            return Err(Error::InvalidMsg("Invalid response id"));
+        }
+
+        match response.result {
+            Some(result) => Ok(serde_json::from_value::<V>(result)?),
+            None => Err(Error::InvalidMsg("Invalid response result")),
+        }
+    }
+}
-- 
cgit v1.2.3