apibara-dna-sdk 2.0.1

The SDK to connect and stream data from Apibara DNA services.
Documentation 
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

use tonic::{
    metadata::{AsciiMetadataValue, KeyAndValueRef},
    service::Interceptor,
};

pub use tonic::metadata::{errors::InvalidMetadataValue, MetadataMap};

/// A metadata key.
pub type MetadataKey = tonic::metadata::MetadataKey<tonic::metadata::Ascii>;
/// A metadata value.
pub type MetadataValue = tonic::metadata::MetadataValue<tonic::metadata::Ascii>;

pub const AUTHORIZATION_HEADER: &str = "authorization";

#[derive(Default, Clone)]
pub struct MetadataInterceptor {
    metadata: MetadataMap,
}

impl MetadataInterceptor {
    pub fn with_metadata(metadata: MetadataMap) -> Self {
        Self { metadata }
    }

    pub fn insert_bearer_token(
        &mut self,
        token: String,
    ) -> Result<Option<MetadataValue>, InvalidMetadataValue> {
        let token = AsciiMetadataValue::try_from(format!("Bearer {token}"))?;
        Ok(self.metadata.insert(AUTHORIZATION_HEADER, token))
    }
}

impl Interceptor for MetadataInterceptor {
    fn call(
        &mut self,
        mut request: tonic::Request<()>,
    ) -> Result<tonic::Request<()>, tonic::Status> {
        let req_meta = request.metadata_mut();

        for kv in self.metadata.iter() {
            match kv {
                KeyAndValueRef::Ascii(key, value) => {
                    req_meta.insert(key, value.clone());
                }
                KeyAndValueRef::Binary(key, value) => {
                    req_meta.insert_bin(key, value.clone());
                }
            }
        }

        Ok(request)
    }
}