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

use crate::ExtrasError;
use serde::{Deserialize, Serialize};
use std::{fs::File, io::BufReader, path::Path};
use tfc_toolset::{
    error::ToolError,
    variable::Attributes,
    workspace::{Workspace, WorkspaceVariables},
};

#[derive(Clone, Debug, Deserialize, Serialize)]
pub struct Variable {
    #[serde(skip_serializing_if = "Option::is_none")]
    pub id: Option<String>,
    #[serde(skip_serializing_if = "Option::is_none")]
    pub attributes: Option<Attributes>,
    #[serde(skip_serializing_if = "Option::is_none")]
    pub var: Option<String>,
    pub workspace: Option<Workspace>,
}

#[derive(Clone, Debug, Deserialize, Serialize)]
pub struct VariablesFile {
    #[serde(skip_serializing_if = "Option::is_none")]
    pub variables: Option<Vec<Variable>>,
}

impl From<Vec<WorkspaceVariables>> for VariablesFile {
    fn from(vec: Vec<WorkspaceVariables>) -> Self {
        let mut variables_file = VariablesFile { variables: None };
        let mut variables_vec = Vec::new();
        for workspace_variables in vec {
            for variable in workspace_variables.variables {
                let entry = Variable {
                    id: variable.id,
                    attributes: Some(variable.attributes),
                    var: None,
                    workspace: Some(workspace_variables.workspace.clone()),
                };
                variables_vec.push(entry);
            }
        }
        variables_file.variables = Some(variables_vec);
        variables_file
    }
}

impl VariablesFile {
    pub fn load<P: AsRef<Path>>(path: P) -> Result<Self, ExtrasError> {
        let file = File::open(path).map_err(ToolError::Io)?;
        let reader = BufReader::new(file);
        let variables_file: Self =
            serde_json::from_reader(reader).map_err(ToolError::Json)?;
        Ok(variables_file)
    }

    pub fn save<P: AsRef<Path>>(
        &self,
        path: P,
        pretty: bool,
    ) -> Result<(), ToolError> {
        if pretty {
            serde_json::to_writer_pretty(&File::create(path)?, self)?;
        } else {
            serde_json::to_writer(&File::create(path)?, self)?;
        }
        Ok(())
    }
}