colorizer 2.1.0

Colorizer is a tool for highlighting words in your terminal
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

use std::io::{self, BufRead};
use std::str::FromStr;

use anyhow::anyhow;
use anyhow::Result;
use clap::Parser;
use yaml_rust::YamlLoader;

pub mod color;
pub mod processor;

const APP_NAME: &str = "colorizer";

/// Simple program to highlight words in terminal
#[derive(Parser, Debug)]
#[command(author, version, about, long_about = None)]
struct Args {
    /// color to use for highlighting
    #[arg(short, long, default_value = "red")]
    color: color::Color,

    /// regex pattern to find and colorize
    #[arg(short, long, num_args(0..))]
    regex: Vec<regex::Regex>,

    /// profile to use from config file
    #[arg(short, long, default_value = "default")]
    profile: String,
}

fn main() {
    let args = Args::parse();

    let patterns = get_patterns_from_cfg_by_profile(&args.profile)
        .unwrap_or(vec![])
        .iter()
        .chain(
            args.regex
                .iter()
                .map(|r| (args.color.clone(), r.clone()))
                .collect::<Vec<(color::Color, regex::Regex)>>()
                .iter(),
        )
        .cloned()
        .collect::<Vec<(color::Color, regex::Regex)>>();

    let patterns = patterns
        .iter()
        .map(|(c, r)| (c.clone().into(), r.clone()))
        .collect();

    let processor = processor::TextProcessor::new(patterns);

    let stdin = io::stdin();

    for line in stdin.lock().lines() {
        let res = processor.process_line(line.unwrap());
        println!("{}", res);
    }
}

fn get_patterns_from_cfg_by_profile(
    profile_name: &str,
) -> Result<Vec<(color::Color, regex::Regex)>, anyhow::Error> {
    let cfg_path = home::home_dir()
        .ok_or(anyhow!("failed to get home path"))?
        .join(".config")
        .join(APP_NAME)
        .join("config.yml");

    let cfg_as_str = std::fs::read_to_string(cfg_path)?;

    let cfg = YamlLoader::load_from_str(&cfg_as_str)?;

    let profiles = cfg
        .get(0)
        .and_then(|cfg_root| cfg_root["profiles"].as_hash())
        .ok_or(anyhow!("no profiles found"))?;

    let res: Vec<(color::Color, regex::Regex)> = profiles
        .iter()
        .filter_map(|(k, v)| {
            let k_str = k.as_str()?;
            if k_str == profile_name {
                Some(v)
            } else {
                None
            }
        })
        .flat_map(|v| {
            v.as_hash().map(|colors| {
                colors.iter().flat_map(|(k, v)| {
                    let k_str = k.as_str().unwrap();
                    let v_vec = v.as_vec().unwrap();
                    v_vec.iter().filter_map(|v| {
                        v.as_str().map(|v_str| {
                            (
                                k_str.to_string().into(),
                                regex::Regex::from_str(v_str).unwrap(),
                            )
                        })
                    })
                })
            })
        })
        .flatten()
        .collect();

    Ok(res)
}