devops-armory 0.7.3

Library to make DevOps work easier, based on russh and actix.
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

use std::collections::HashMap;
use std::fs::File;
use std::io::{self, Write};
use std::path::PathBuf;
use std::path::Path;

use super::models::{
    LogrotateConfig,
    CreateSpec
};

pub type LogrotateFileMap = HashMap<PathBuf, LogrotateConfig>;

/// Custom serializer for logrotate file 
fn format_entry(path: &PathBuf, cfg: &LogrotateConfig) -> String {
    let mut s = String::new();
    s.push_str(&format!("{} {{\n", path.display()));

    if let Some(f) = &cfg.frequency { s.push_str(&format!("    {}\n", f)); }
    if let Some(r) = cfg.rotate { s.push_str(&format!("    rotate {}\n", r)); }
    if cfg.compress { s.push_str("    compress\n"); }
    if cfg.delaycompress { s.push_str("    delaycompress\n"); }
    if cfg.missingok { s.push_str("    missingok\n"); }
    if cfg.notifempty { s.push_str("    notifempty\n"); }
    if let Some(create) = &cfg.create {
        // format mode as octal with leading zeros (e.g. 0640)
        s.push_str(&format!(
            "    create {:04o} {} {}\n",
            create.mode, create.owner, create.group
        ));
    }
    if cfg.sharedscripts { s.push_str("    sharedscripts\n"); }
    if let Some(post) = &cfg.postrotate {
        s.push_str("    postrotate\n");
        for line in post.lines() {
            s.push_str("        ");
            s.push_str(line);
            s.push('\n');
        }
        s.push_str("    endscript\n");
    }

    s.push_str("}\n\n");
    s
}

/// Function to write parsed data into file
fn write_logrotate_file(path: &std::path::Path, map: &LogrotateFileMap) -> io::Result<()> {
    let mut file = File::create(path)?;
    // If ordering matters, iterate in a deterministic order:
    let mut entries: Vec<_> = map.iter().collect();
    entries.sort_by_key(|(p, _)| p.clone());
    for (p, cfg) in entries {
        let block = format_entry(p, cfg);
        file.write_all(block.as_bytes())?;
    }
    Ok(())
}

/// Logrotate file creator
/// Create logrotate conf file based on provided function parameters.
pub fn logrotate_parser(
    log_file: &str,
    freq: Option<String>,
    rotation: Option<u32>,
    comp: bool,
    delaycomp: bool,
    missok: bool,
    not_if_empty: bool,
    create_mode: u32,
    create_owner: String,
    create_group: String,
    shared_scripts: bool,
    post_rotate: Option<String>, 
    logrotate_file: &Path
) -> io::Result<()> {
    let mut m = LogrotateFileMap::new();
    m.insert(
        PathBuf::from(log_file),
        LogrotateConfig {
            frequency: freq,
            rotate: rotation,
            compress: comp,
            delaycompress: delaycomp,
            missingok: missok,
            notifempty: not_if_empty,
            create: Some(CreateSpec {
                mode: create_mode,
                owner: create_owner,
                group: create_group,
            }),
            sharedscripts: shared_scripts,
            postrotate: post_rotate,
        },
    );

    write_logrotate_file(std::path::Path::new(logrotate_file), &m)
}