tfc_toolset/
settings.rs

1use crate::error::ToolError;
2use config::{Config, ConfigError, Environment, File};
3use serde::{Deserialize, Serialize};
4use std::{path::PathBuf, str::FromStr};
5
6pub const DEFAULT_TERRAFORM_VERSION: &str = "1.5.7";
7
8#[derive(Clone, Debug, Deserialize, Serialize)]
9#[non_exhaustive]
10pub enum Operators {
11    Equals,
12    NotEquals,
13    Contains,
14    NotContains,
15}
16
17impl FromStr for Operators {
18    type Err = ToolError;
19
20    fn from_str(s: &str) -> Result<Self, Self::Err> {
21        match s {
22            "==" => Ok(Operators::Equals),
23            "!=" => Ok(Operators::NotEquals),
24            "~=" => Ok(Operators::Contains),
25            "!~=" => Ok(Operators::NotContains),
26            _ => Err(ToolError::InvalidQueryOperator(s.to_string())),
27        }
28    }
29}
30
31#[derive(Clone, Debug, Deserialize, Serialize)]
32pub struct Variable {
33    pub key: String,
34    pub operator: Operators,
35    pub value: String,
36}
37
38impl FromStr for Variable {
39    type Err = ToolError;
40
41    fn from_str(s: &str) -> Result<Self, Self::Err> {
42        let parts: Vec<&str> = s.split(':').collect();
43        if parts.len() != 3 {
44            return Err(ToolError::InvalidVariableQuery(s.to_string()));
45        }
46        let operator = Operators::from_str(parts[1])?;
47        Ok(Variable {
48            key: parts[0].to_string(),
49            operator,
50            value: parts[2].to_string(),
51        })
52    }
53}
54
55#[derive(Clone, Debug, Deserialize, Serialize)]
56pub struct Tag {
57    pub operator: Operators,
58    pub name: String,
59}
60
61impl FromStr for Tag {
62    type Err = ToolError;
63
64    fn from_str(s: &str) -> Result<Self, Self::Err> {
65        let parts: Vec<&str> = s.split(':').collect();
66        if parts.len() != 2 {
67            return Err(ToolError::InvalidTagQuery(s.to_string()));
68        }
69        let operator = Operators::from_str(parts[0])?;
70        Ok(Tag { operator, name: parts[1].to_string() })
71    }
72}
73
74#[derive(Clone, Debug, Deserialize, Serialize)]
75pub struct Pagination {
76    pub start_page: String,
77    pub max_depth: String,
78    pub page_size: String,
79}
80
81#[derive(Clone, Debug, Deserialize, Serialize)]
82pub struct Query {
83    pub name: Option<String>,
84    pub wildcard_name: Option<String>,
85    pub variables: Option<Vec<Variable>>,
86    pub tags: Option<Vec<Tag>>,
87}
88
89#[derive(Clone, Debug, Deserialize, Serialize)]
90pub struct Workspaces {
91    pub query: Option<Query>,
92}
93
94#[derive(Clone, Debug, Deserialize)]
95pub struct Core {
96    pub log: String,
97    pub token: String,
98    pub org: String,
99    pub project: Option<String>,
100    pub output: PathBuf,
101    pub save_output: bool,
102    pub pagination: Pagination,
103    pub workspaces: Workspaces,
104    pub terraform_version: String,
105}
106
107impl Core {
108    pub fn new() -> Result<Self, ConfigError> {
109        let s = Config::builder()
110            // Set defaults
111            .set_default("log", "info".to_string())?
112            .set_default("token", "".to_string())?
113            .set_default("org", "".to_string())?
114            .set_default("output", "report.json".to_string())?
115            .set_default("save_output", false)?
116            .set_default("pagination.start_page", "1".to_string())?
117            .set_default("pagination.max_depth", "1".to_string())?
118            .set_default("pagination.page_size", "20".to_string())?
119            .set_default(
120                "terraform_version",
121                DEFAULT_TERRAFORM_VERSION.to_string(),
122            )?
123            .set_default("workspaces.query", None::<String>)?
124            // Start off by merging in the "default" configuration file
125            .add_source(File::with_name("settings.toml").required(false))
126            // Add in settings from the environment
127            // Eg.. `DEBUG=1 ./target/app` would set the `debug` key
128            .add_source(Environment::default())
129            .build()?;
130        s.try_deserialize()
131    }
132}
tfc_toolset/settings.rs

tfc_toolset/
settings.rs
