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

pub mod error;
pub mod file;

pub use error::ExtrasError;
pub use file::{
    tag::TagsFile, variable::VariablesFile, workspace::WorkspacesFile,
};
use std::path::PathBuf;

use http_cache_surf::{
    CACacheManager, Cache, CacheMode, CacheOptions, HttpCache, HttpCacheOptions,
};
use regex::Regex;
use surf::Client;
use surf_governor::GovernorMiddleware;
use surf_retry::{ExponentialBackoff, RetryMiddleware};
use tfc_toolset::error::ToolError;

pub fn build_governor() -> Result<GovernorMiddleware, ToolError> {
    match GovernorMiddleware::per_second(30) {
        Ok(g) => Ok(g),
        Err(e) => Err(ToolError::General(e.into_inner())),
    }
}

pub fn build_retry() -> RetryMiddleware<ExponentialBackoff> {
    RetryMiddleware::new(
        99,
        ExponentialBackoff::builder().build_with_max_retries(10),
        1,
    )
}

pub fn build_cache_options() -> HttpCacheOptions {
    HttpCacheOptions {
        cache_options: Some(CacheOptions {
            shared: false,
            cache_heuristic: 0.0,
            immutable_min_time_to_live: Default::default(),
            ignore_cargo_cult: false,
        }),
        cache_key: None,
        cache_mode_fn: None,
        cache_bust: None,
    }
}

pub fn default_client(path: Option<PathBuf>) -> Result<Client, ToolError> {
    // Build the http client with a cache, governor, and retry enabled
    Ok(Client::new().with(build_retry()).with(build_governor()?).with(Cache(
        HttpCache {
            mode: CacheMode::Default,
            manager: CACacheManager {
                path: path.unwrap_or_else(|| {
                    dirs::cache_dir().unwrap_or("./".into()).join("tfc-toolset")
                }),
            },
            options: build_cache_options(),
        },
    )))
}

pub fn parse_workspace_name(
    workspace_name: &str,
) -> Result<String, ExtrasError> {
    let re = Regex::new("^[a-zA-Z0-9_-]*$")?;
    let caps = re.captures(workspace_name);
    match caps {
        Some(_) => Ok(workspace_name.to_string()),
        None => Err(ExtrasError::BadWorkspaceName(workspace_name.to_string())),
    }
}

pub fn parse_tag_name(tag_name: &str) -> Result<String, ExtrasError> {
    let re = Regex::new("^[a-zA-Z0-9_:-]*$")?;
    let caps = re.captures(tag_name);
    match caps {
        Some(_) => Ok(tag_name.to_string()),
        None => Err(ExtrasError::BadTagName(tag_name.to_string())),
    }
}