Sending pubkey

src/ws.rs

@@ -5,7 +5,7 @@ use futures_util::{
     SinkExt, StreamExt,
     stream::{SplitSink, SplitStream},
 };
-use tokio::sync::Mutex;
+use tokio::sync::{Mutex, MutexGuard};
 
 use crate::protocol::{ClientMethod, ServerMethod};
 
@@ -62,4 +62,8 @@ impl EnclaveWebSocket {
 
         Ok(())
     }
+
+    pub async fn tx(&self) -> MutexGuard<'_, SplitSink<WebSocket, Message>> {
+        self.tx.lock().await
+    }
 }