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
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
use crate::errors::Result;
use crate::CERTIFICATES;
use gouth::Builder;
use std::sync::Arc;
use tonic::{
metadata::{Ascii, MetadataValue},
service::Interceptor,
transport::{Certificate, Channel, ClientTlsConfig},
Status,
};
#[derive(Clone)]
pub struct TokenInterceptor(Arc<String>);
impl TokenInterceptor {
fn new(token_header_val: Arc<String>) -> TokenInterceptor {
TokenInterceptor(token_header_val)
}
}
pub fn new_interceptor(token_header_val: Arc<String>) -> TokenInterceptor {
TokenInterceptor::new(token_header_val)
}
impl Interceptor for TokenInterceptor {
fn call(
&mut self,
request: tonic::Request<()>,
) -> core::result::Result<tonic::Request<()>, Status> {
let mut req = request;
#[allow(deprecated)]
let meta_result = MetadataValue::<Ascii>::from_str(&self.0);
return match meta_result {
Ok(meta) => {
req.metadata_mut().insert("authorization", meta);
Ok(req)
}
Err(some_error) => Err(tonic::Status::internal(format!(
"new_interceptor: Error when getting MetadataValue from token {:?}",
some_error
))),
};
}
}
pub(crate) async fn new_grpc_channel(
domain_name: &'static str,
channel_url: &'static str,
timeout_secs: Option<u64>,
) -> Result<Channel> {
let tls_config = ClientTlsConfig::new()
.ca_certificate(Certificate::from_pem(CERTIFICATES))
.domain_name(domain_name);
if let Some(timeout) = timeout_secs {
Ok(Channel::from_static(channel_url)
.tls_config(tls_config.clone())?
.timeout(std::time::Duration::from_secs(timeout))
.connect()
.await?)
} else {
Ok(Channel::from_static(channel_url)
.tls_config(tls_config.clone())?
.connect()
.await?)
}
}
#[allow(clippy::rc_buffer)]
pub(crate) fn get_token(google_credentials: impl AsRef<str>) -> Result<Arc<String>> {
let token = Builder::new().json(google_credentials).build()?;
let token_header_val: Arc<String> = token.header_value()?;
Ok(token_header_val)
}