Add AWM sign request

crates/avalanche-types/src/lib.rs

@@ -29,7 +29,7 @@ pub mod txs;
 pub mod units;
 pub mod utils;
 pub mod verify;
-
+pub mod warp;
 #[cfg(feature = "avalanchego")]
 #[cfg_attr(docsrs, doc(cfg(feature = "avalanchego")))]
 pub mod avalanchego;

crates/avalanche-types/src/subnet/rpc/snow/engine/common/vm.rs

@@ -18,7 +18,7 @@ use crate::{
     },
 };
 use tokio::sync::mpsc::Sender;
-
+use crate::warp::WarpSignerClient_;
 /// Vm describes the trait that all consensus VMs must implement.
 ///
 /// ref. <https://pkg.go.dev/github.com/ava-labs/avalanchego/snow/engine/common#Vm>
@@ -29,7 +29,7 @@ pub trait CommonVm: AppHandler + Connector + Checkable {
     type ChainHandler: Handle;
     type StaticHandler: Handle;
     type ValidatorState: validators::State;
-
+    type WarpSigner: WarpSignerClient_;
     async fn initialize(
         &mut self,
         ctx: Option<Context<Self::ValidatorState>>,
@@ -40,6 +40,7 @@ pub trait CommonVm: AppHandler + Connector + Checkable {
         to_engine: Sender<Message>,
         fxs: &[Fx],
         app_sender: Self::AppSender,
+        warp_signer: Self::WarpSigner,
     ) -> Result<()>;
     async fn set_state(&self, state: State) -> Result<()>;
     async fn shutdown(&self) -> Result<()>;

crates/avalanche-types/src/subnet/rpc/vm/server.rs

@@ -44,6 +44,7 @@ use prost::bytes::Bytes;
 use semver::Version;
 use tokio::sync::{broadcast, mpsc, RwLock};
 use tonic::{Request, Response};
+use crate::warp::client::WarpSignerClient;
 
 pub struct Server<V> {
     /// Underlying Vm implementation.
@@ -96,6 +97,7 @@ where
             DatabaseManager = DatabaseManager,
             AppSender = AppSenderClient,
             ValidatorState = ValidatorStateClient,
+            WarpSigner=WarpSignerClient,
         > + Send
         + Sync
         + 'static,
@@ -182,7 +184,7 @@ where
                 return tonic::Status::unknown("engine receiver closed unexpectedly");
             }
         });
-
+        let warp_signer = WarpSignerClient::new(client_conn.clone());
         let mut inner_vm = self.vm.write().await;
         inner_vm
             .initialize(
@@ -194,6 +196,7 @@ where
                 tx_engine,
                 &[()],
                 AppSenderClient::new(client_conn.clone()),
+                warp_signer
             )
             .await
             .map_err(|e| tonic::Status::unknown(e.to_string()))?;

crates/avalanche-types/src/warp/client.rs

@@ -0,0 +1,50 @@
+use std::io::{Error, ErrorKind, Read, Result};
+use std::str::FromStr;
+
+use crate::proto::pb::warp::{
+        signer_client,
+        SignRequest,
+        SignResponse,
+    };
+use prost::bytes::Bytes;
+use tonic::transport::Channel;
+use crate::ids::Id;
+
+#[derive(Clone)]
+pub struct WarpSignerClient {
+    inner: signer_client::SignerClient<Channel>,
+}
+
+impl WarpSignerClient {
+    pub fn new(client_conn: Channel) -> Self {
+        Self {
+            inner: signer_client::SignerClient::new(client_conn)
+                .max_decoding_message_size(usize::MAX)
+                .max_encoding_message_size(usize::MAX),
+        }
+    }
+}
+
+#[tonic::async_trait]
+impl super::WarpSignerClient_ for WarpSignerClient {
+    async fn sign(&self,
+                  network_id: u32,
+                  source_chain_id: &str,
+                  payload: &[u8]) -> Result<SignResponse> {
+        let mut client = self.inner.clone();
+        let res = client
+            .sign(SignRequest {
+                network_id,
+                source_chain_id: Bytes::from(Id::from_str(source_chain_id).unwrap().to_vec()),
+                payload: Bytes::from(payload.to_vec()),
+            })
+            .await
+            .map_err(|e| {
+                Error::new(
+                    ErrorKind::Other,
+                    format!("sign failed: {:?}", e),
+                )
+            })?;
+        Ok(res.into_inner())
+    }
+}

crates/avalanche-types/src/warp/mod.rs

@@ -0,0 +1,33 @@
+pub mod client;
+
+use std::io::Result;
+use crate::proto::warp::SignResponse;
+
+#[tonic::async_trait]
+pub trait WarpSignerClient_: Send + Sync + CloneBox {
+    async fn sign(
+        &self,
+        network_id: u32,
+        source_chain_id: &str,
+        payload: &[u8],
+    ) -> Result<SignResponse>;
+}
+
+pub trait CloneBox {
+    fn clone_box(&self) -> Box<dyn WarpSignerClient_ + Send + Sync>;
+}
+
+impl<T> CloneBox for T
+    where
+        T: 'static + WarpSignerClient_ + Clone + Send + Sync,
+{
+    fn clone_box(&self) -> Box<dyn WarpSignerClient_ + Send + Sync> {
+        Box::new(self.clone())
+    }
+}
+
+impl Clone for Box<dyn WarpSignerClient_ + Send + Sync> {
+    fn clone(&self) -> Box<dyn WarpSignerClient_ + Send + Sync> {
+        self.clone_box()
+    }
+}