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
use crate::{
error::RuarangoErr::Unreachable,
model::{AddHeaders, BuildUrl},
Connection,
};
use anyhow::{Context, Result};
use derive_builder::Builder;
use getset::Getters;
use reqwest::{
header::{HeaderMap, HeaderName, HeaderValue},
Url,
};
use serde::{Deserialize, Serialize};
#[derive(Builder, Clone, Debug, Default, Deserialize, Getters, Serialize)]
#[getset(get = "pub(crate)")]
pub struct Config {
#[builder(setter(into))]
name: String,
#[builder(setter(into))]
collection: String,
#[builder(setter(into))]
key: String,
#[builder(setter(strip_option, into), default)]
if_match: Option<String>,
#[builder(setter(strip_option, into), default)]
if_none_match: Option<String>,
}
impl Config {
fn build_suffix(&self, base: &str) -> String {
format!(
"{}/{}/vertex/{}/{}",
base, self.name, self.collection, self.key
)
}
}
impl BuildUrl for Config {
fn build_url(&self, base: &str, conn: &Connection) -> Result<Url> {
let suffix = self.build_suffix(base);
conn.db_url()
.join(&suffix)
.with_context(|| format!("Unable to build '{}' url", suffix))
}
}
impl AddHeaders for Config {
fn has_header(&self) -> bool {
self.if_match.is_some() || self.if_none_match.is_some()
}
fn add_headers(&self) -> Result<Option<HeaderMap>> {
let mut headers = None;
if self.has_header() {
let mut headers_map = HeaderMap::new();
if let Some(rev) = self.if_match() {
let _ = headers_map.append(
HeaderName::from_static("if-match"),
HeaderValue::from_str(rev)?,
);
headers = Some(headers_map);
} else if let Some(rev) = self.if_none_match() {
let _ = headers_map.append(
HeaderName::from_static("if-none-match"),
HeaderValue::from_str(rev)?,
);
headers = Some(headers_map);
} else {
return Err(Unreachable {
msg: "'if_match' or 'if_none_match' should be true!".to_string(),
}
.into());
}
}
Ok(headers)
}
}