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

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

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

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

impl From<Vec<WorkspaceTags>> for TagsFile {
    fn from(vec: Vec<WorkspaceTags>) -> Self {
        let mut tags_file = TagsFile { tags: None };
        let mut tags_vec = Vec::new();
        for workspace_tags in vec {
            for tag_vec in workspace_tags.tags {
                for tag in tag_vec.data {
                    let entry = Tag {
                        id: tag.id,
                        name: Some(tag.attributes.name.clone()), // We do this to make importing with a file easier
                        attributes: Some(tag.attributes),
                        workspace: Some(workspace_tags.workspace.clone()),
                    };
                    tags_vec.push(entry);
                }
            }
        }
        tags_file.tags = Some(tags_vec);
        tags_file
    }
}

impl TagsFile {
    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 tags_file: Self =
            serde_json::from_reader(reader).map_err(ToolError::Json)?;
        Ok(tags_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(())
    }
}