miden-node-rpc 0.16.0-rc.5

Miden node's front-end RPC server
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60

use miden_node_proto::generated as proto;
use miden_node_tracing::{debug, miden_instrument};

use super::{Request, RpcBackend, RpcService};
use crate::{COMPONENT, LOG_TARGET};

#[tonic::async_trait]
impl proto::server::rpc_api::GetTransactionEncryptionKey for RpcService {
    type Input = ();
    type Output = proto::transaction::TransactionEncryptionKey;

    fn decode(request: ()) -> tonic::Result<Self::Input> {
        Ok(request)
    }

    fn encode(output: Self::Output) -> tonic::Result<proto::transaction::TransactionEncryptionKey> {
        Ok(output)
    }

    #[miden_instrument(
        target = COMPONENT,
        name = "get_transaction_encryption_key",
        err,
    )]
    async fn handle(
        &self,
        _input: Self::Input,
        metadata: &tonic::metadata::MetadataMap,
        _extensions: &tonic::codegen::http::Extensions,
    ) -> tonic::Result<Self::Output> {
        let original_accept_header = metadata.get(http::header::ACCEPT.as_str()).cloned();

        debug!(target: LOG_TARGET, "Getting transaction encryption key");

        let mut forwarded_request = Request::new(());
        if let Some(accept) = original_accept_header {
            forwarded_request.metadata_mut().insert(http::header::ACCEPT.as_str(), accept);
        }

        // Nodes connected to validators ask one directly, otherwise the request is forwarded. The
        // encryption key is shared by the whole validator set, so any single validator serves.
        let validator = match &self.backend {
            RpcBackend::Sequencer { validators, .. } => validators.random(),
            RpcBackend::FullNode { pre_auth: Some(pre_auth), .. } => pre_auth.validators().random(),
            RpcBackend::FullNode { source_rpc, pre_auth: None, .. } => {
                return source_rpc
                    .as_ref()
                    .clone()
                    .get_transaction_encryption_key(forwarded_request)
                    .await
                    .map(tonic::Response::into_inner);
            },
        };
        validator
            .clone()
            .get_transaction_encryption_key(forwarded_request)
            .await
            .map(tonic::Response::into_inner)
    }
}