diff options
| -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('"', "\\\"") |