redacta 0.0.1

Command line tool for redacting information from text.
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

use clap::Parser;
use redacta::fqdn::FqdnRedactor;
use redacta::ipv4::IPv4Redactor;
use redacta::ipv6::IPv6Redactor;
use redacta::redact::{Redact, redact_logs};
use redacta::regex::RegexRedactor;
use std::io::BufReader;

#[derive(Parser)]
#[command(
    disable_help_subcommand = true,
    override_usage = "<STDIN> | redacta [OPTIONS]"
)]
struct Args {
    /// Enable IPv4 redaction
    #[arg(long)]
    ipv4: bool,
    /// Enable IPv6 redaction
    #[arg(long)]
    ipv6: bool,
    /// Enable FQDN redaction
    #[arg(long)]
    fqdn: bool,
    /// Regex redaction
    #[arg(short, long)]
    regex: Option<String>,
}

fn main() -> anyhow::Result<()> {
    let stdin = std::io::stdin().lock();
    let mut stdout = std::io::stdout().lock();

    redact_logs(
        BufReader::new(stdin),
        &mut stdout,
        &enabled_redactors(&Args::parse())?,
    )
    .unwrap_or_else(|_| println!("failed to redact"));

    Ok(())
}

fn enabled_redactors(args: &Args) -> anyhow::Result<Vec<Box<dyn Redact>>> {
    let mut redactors: Vec<Box<dyn Redact>> = Vec::new();

    if args.ipv4 {
        redactors.push(Box::new(IPv4Redactor::default()));
    }

    if args.ipv6 {
        redactors.push(Box::new(IPv6Redactor::default()));
    }

    if args.fqdn {
        redactors.push(Box::new(FqdnRedactor::default()));
    }

    if let Some(regex) = &args.regex {
        redactors.push(Box::new(RegexRedactor::try_from(regex)?));
    }

    Ok(redactors)
}