Add AWM sign request

crates/avalanche-types/src/lib.rs

@@ -12,6 +12,9 @@
 //! ecosystem.
 //!
 #![cfg_attr(docsrs, feature(doc_cfg))]
+#[cfg(feature = "avalanchego")]
+#[cfg_attr(docsrs, doc(cfg(feature = "avalanchego")))]
+pub mod avalanchego;
 pub mod avm;
 pub mod choices;
 pub mod codec;
@@ -29,10 +32,7 @@ pub mod txs;
 pub mod units;
 pub mod utils;
 pub mod verify;
-
-#[cfg(feature = "avalanchego")]
-#[cfg_attr(docsrs, doc(cfg(feature = "avalanchego")))]
-pub mod avalanchego;
+pub mod warp;
 
 #[cfg(feature = "coreth")]
 #[cfg_attr(docsrs, doc(cfg(feature = "coreth")))]

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

@@ -1,5 +1,6 @@
 use std::{collections::HashMap, io::Result};
 
+use crate::warp::WarpSignerClient_;
 use crate::{
     ids,
     subnet::rpc::{
@@ -18,7 +19,6 @@ use crate::{
     },
 };
 use tokio::sync::mpsc::Sender;
-
 /// 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

@@ -4,6 +4,7 @@ use std::{
     time::{Duration, Instant},
 };
 
+use crate::warp::client::WarpSignerClient;
 use crate::{
     ids,
     packer::U32_LEN,
@@ -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,43 @@
+use std::io::{Error, ErrorKind, Read, Result};
+use std::str::FromStr;
+
+use crate::ids::Id;
+use crate::proto::pb::warp::{signer_client, SignRequest, SignResponse};
+use prost::bytes::Bytes;
+use tonic::transport::Channel;
+
+#[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 crate::proto::warp::SignResponse;
+use std::io::Result;
+
+#[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()
+    }
+}