diff options
| author |   alemi <me@alemi.dev> | 2023-12-02 06:15:06 +0100 |
|---|---|---|
| committer | alemi <me@alemi.dev> | 2023-12-02 06:15:06 +0100 |
| commit | 232d47cdc93d36b61ff16077a546085fa54bb1f6 (patch) | |
| tree | 7d6c953eaa12b62a05c874e1d6c5e29701282336 | |
| parent | 5490a28b75378050d63e95f4a866b8f599341352 (diff) | |
| -rw-r--r-- | src/counter.rs | 9 |
1 files changed, 8 insertions, 1 deletions
diff --git a/src/counter.rs b/src/counter.rs index 6e12cd0..bfa8a3b 100644 --- a/src/counter.rs +++ b/src/counter.rs @@ -16,11 +16,18 @@ fn main() { probabilities.sort_by_key(|x| x.1); + let mut uncensored = Vec::new(); let censored = std::env::args().nth(1).as_deref().unwrap_or("") == "--censored"; + if censored { + for arg in std::env::args().skip(2) { + uncensored.push(arg); + } + } + println!("{{"); for (line, count, chance) in probabilities { - let l = if censored { + let l = if censored && !uncensored.iter().any(|x| line.starts_with(x)) { "■".repeat(line.len()) } else { line.replace('"', "\\\"") |