souschef 0.2.3

CLI utility to interact with Chef Server.
Documentation 
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
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123

use core::error::Error;
use regex::Regex;
use std::fs;
use std::path::PathBuf;
use thiserror::Error;

#[derive(Error, Debug)]
enum KnifeConfigError {
    #[error("Error reading configuration file: {config_file}: {reason}")]
    ReadingConfigurationFile {
        config_file: PathBuf,
        reason: String,
    },

    #[error("Extracting organization from Chef server URL failed: {server_url}")]
    OrganizationExtractFailed { server_url: String },
}

#[derive(Debug, Clone)]
pub struct KnifeConfig {
    pub node_name: String,
    pub client_key: String,
    pub chef_server_url: String,
    pub organization: String,
    pub secret_file: Option<String>,
}

impl KnifeConfig {
    /// from_file - reads the configuration file from file based on the path or profile CLI
    /// argument passed.
    ///
    /// If profile is defined, it would take priority over path.
    pub fn from_file(path: &str, profile: &str) -> Result<Self, Box<dyn Error>> {
        let homedir = match dirs::home_dir() {
            Some(p) => p,
            None => return Err("can't determing $HOME.".into()),
        };
        let off_path = match profile {
            "chef" => homedir.join(".chef/knife.rb"),
            "cinc" => homedir.join(".cinc/knife.rb"),
            _ => {
                if path.starts_with("~/") {
                    let clean_path = path.strip_prefix("~/").unwrap_or("");
                    homedir.join(clean_path)
                } else {
                    PathBuf::from(path)
                }
            }
        };

        let content = match fs::read_to_string(off_path.clone()) {
            Ok(c) => c,
            Err(e) => {
                return Err(KnifeConfigError::ReadingConfigurationFile {
                    config_file: off_path,
                    reason: e.to_string(),
                }
                .to_string()
                .into())
            }
        };

        // Extract Chef server configuration files from knife.rb
        let node_name_re = Regex::new(r#"(?m)^\s*node_name\s*['"]([^'"]+)['"]"#)?;
        let client_key_re = Regex::new(r#"(?m)^\s*client_key\s*['"]([^'"]+)['"]"#)?;
        let server_url_re = Regex::new(r#"(?m)^\s*chef_server_url\s*['"]([^'"]+)['"]"#)?;
        let secret_file_re = Regex::new(r"knife\[:secret_file\]\s*=\s*'([^']*)'")?;

        let node_name = node_name_re
            .captures(&content)
            .ok_or("node_name not found")?[1]
            .to_string();

        let client_key = client_key_re
            .captures(&content)
            .ok_or("client_key not found")?[1]
            .to_string();

        let chef_server_url = server_url_re
            .captures(&content)
            .ok_or("chef_server_url not found")?[1]
            .to_string();

        let secret_file = match secret_file_re.captures(&content) {
            Some(captures) => {
                let cap = captures
                    .get(1)
                    .map_or("".to_string(), |m| m.as_str().to_string());
                Some(cap)
            }
            None => None,
        };

        // Extract the organization from the chef_server_url
        let organization = match chef_server_url.split("/").last() {
            Some(org) => org.to_string(),
            None => {
                return Err(KnifeConfigError::OrganizationExtractFailed {
                    server_url: chef_server_url,
                }
                .into())
            }
        };

        // Expand ~ to home directory if present
        let client_key = if client_key.starts_with('~') {
            homedir
                .join(&client_key[2..])
                .to_string_lossy()
                .into_owned()
        } else {
            client_key
        };

        Ok(KnifeConfig {
            node_name,
            client_key,
            chef_server_url,
            organization,
            secret_file,
        })
    }
}