diff --git a/Cargo.lock b/Cargo.lock index 76631e4..b77319b 100644 --- a/Cargo.lock +++ b/Cargo.lock @@ -418,7 +418,7 @@ dependencies = [ [[package]] name = "detee-shared" version = "0.1.0" -source = "git+ssh://git@gitea.detee.cloud/noormohammedb/detee-shared?branch=stable_01#fce57884937a4ec02acbf2f5b370ab879b1af657" +source = "git+ssh://git@gitea.detee.cloud/noormohammedb/detee-shared?branch=stable_01#65bfa113a44251dd13e5d895014f4a773e367d2e" dependencies = [ "base64", "prost", diff --git a/src/grpc.rs b/src/grpc.rs index 6e69c69..1f6c355 100644 --- a/src/grpc.rs +++ b/src/grpc.rs @@ -426,14 +426,14 @@ impl BrainAppCli for BrainAppCliMock { #[tonic::async_trait] impl BrainAppDaemon for BrainAppDaemonMock { - type RegisterNodeStream = Pin> + Send>>; + type RegisterAppNodeStream = Pin> + Send>>; type BrainMessagesStream = Pin> + Send>>; - async fn register_node( + async fn register_app_node( &self, req: tonic::Request, - ) -> Result, Status> { - let req_data = req.into_inner(); + ) -> Result, Status> { + let req_data = check_sig_from_req(req)?; log::info!( "registering app node_key : {}, owner_key: {}", &req_data.node_pubkey, @@ -569,6 +569,12 @@ impl PubkeyGetter for AirdropReq { } } +impl PubkeyGetter for RegisterAppNodeReq { + fn get_pubkey(&self) -> Option { + None + } +} + fn check_sig_from_req(req: Request) -> Result { let time = match req.metadata().get("timestamp") { Some(t) => t.clone(),