aruna-cache 0.2.3

A library to cache resources for AOS
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
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
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121

use crate::structs::Resource;
use anyhow::{anyhow, Result};
use aruna_rust_api::api::{
    notification::services::v2::Resource as ApiResource,
    storage::models::v2::{generic_resource, InternalRelation, RelationDirection, ResourceVariant},
};
use diesel_ulid::DieselUlid;
use std::str::FromStr;
use tonic::metadata::AsciiMetadataKey;
use tonic::metadata::AsciiMetadataValue;

pub trait GetRef {
    fn get_ref(&self) -> Option<(DieselUlid, Resource)>;
}

pub trait GetName {
    fn get_name(&self) -> String;
}

impl GetName for generic_resource::Resource {
    fn get_name(&self) -> String {
        match self {
            generic_resource::Resource::Project(proj) => proj.name.to_string(),
            generic_resource::Resource::Collection(col) => col.name.to_string(),
            generic_resource::Resource::Dataset(ds) => ds.name.to_string(),
            generic_resource::Resource::Object(obj) => obj.name.to_string(),
        }
    }
}

impl GetRef for ApiResource {
    fn get_ref(&self) -> Option<(DieselUlid, Resource)> {
        let (res_id, associated_id) = if self.persistent_resource_id {
            (
                DieselUlid::from_str(&self.resource_id).ok()?,
                DieselUlid::from_str(&self.associated_id).ok()?,
            )
        } else {
            (
                DieselUlid::from_str(&self.associated_id).ok()?,
                DieselUlid::from_str(&self.resource_id).ok()?,
            )
        };
        let (associated_id, res) = match self.resource_variant() {
            ResourceVariant::Project => (res_id, Resource::Project(associated_id)),
            ResourceVariant::Collection => (res_id, Resource::Collection(associated_id)),
            ResourceVariant::Dataset => (res_id, Resource::Dataset(associated_id)),
            ResourceVariant::Object => (res_id, Resource::Object(associated_id)),
            _ => return None,
        };

        Some((associated_id, res))
    }
}

pub fn internal_relation_to_rel(
    id: Resource,
    int_rel: InternalRelation,
) -> Result<(Resource, Resource)> {
    match int_rel.direction() {
        RelationDirection::Inbound => match int_rel.resource_variant() {
            ResourceVariant::Project => Ok((
                Resource::Project(DieselUlid::from_str(&int_rel.resource_id)?),
                id,
            )),
            ResourceVariant::Collection => Ok((
                Resource::Collection(DieselUlid::from_str(&int_rel.resource_id)?),
                id,
            )),
            ResourceVariant::Dataset => Ok((
                Resource::Dataset(DieselUlid::from_str(&int_rel.resource_id)?),
                id,
            )),
            ResourceVariant::Object => Ok((
                Resource::Object(DieselUlid::from_str(&int_rel.resource_id)?),
                id,
            )),
            _ => Err(anyhow!("Invalid resource variant")),
        },
        RelationDirection::Outbound => match int_rel.resource_variant() {
            ResourceVariant::Project => Ok((
                id,
                Resource::Project(DieselUlid::from_str(&int_rel.resource_id)?),
            )),
            ResourceVariant::Collection => Ok((
                id,
                Resource::Collection(DieselUlid::from_str(&int_rel.resource_id)?),
            )),
            ResourceVariant::Dataset => Ok((
                id,
                Resource::Dataset(DieselUlid::from_str(&int_rel.resource_id)?),
            )),
            ResourceVariant::Object => Ok((
                id,
                Resource::Object(DieselUlid::from_str(&int_rel.resource_id)?),
            )),
            _ => Err(anyhow!("Invalid resource variant")),
        },
        _ => Err(anyhow!("Invalid resource variant")),
    }
}

// Create a client interceptor which always adds the specified api token to the request header
#[derive(Clone)]
pub struct ClientInterceptor {
    pub api_token: String,
}
// Implement a request interceptor which always adds
//  the authorization header with a specific API token to all requests
impl tonic::service::Interceptor for ClientInterceptor {
    fn call(&mut self, request: tonic::Request<()>) -> Result<tonic::Request<()>, tonic::Status> {
        let mut mut_req: tonic::Request<()> = request;
        let metadata = mut_req.metadata_mut();
        metadata.append(
            AsciiMetadataKey::from_bytes("authorization".as_bytes()).unwrap(),
            AsciiMetadataValue::try_from(format!("Bearer {}", self.api_token.as_str())).unwrap(),
        );

        Ok(mut_req)
    }
}