hatter/cargo-todo
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

feat: add count outputs

Browse Source
This commit is contained in:
Hatter Jiang
2020-08-02 01:37:01 +08:00
parent 6e114512ae
commit 258ded8e78
1 changed files with 13 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
src/main.rs
View File

@@ -82,6 +82,8 @@ fn main() -> std::io::Result<()> {
} }
} }
let total_count = tokens.len();
if let Some(members) = matches.values_of("member") { if let Some(members) = matches.values_of("member") {
let members = members.collect::<HashSet<&str>>(); let members = members.collect::<HashSet<&str>>();
tokens = tokens.into_iter() tokens = tokens.into_iter()
@@ -113,10 +115,18 @@ fn main() -> std::io::Result<()> {
tokens = tokens.into_iter().take(head_count).collect::<Vec<Token>>(); tokens = tokens.into_iter().take(head_count).collect::<Vec<Token>>();
} }
if matches.is_present("inline") { if tokens.is_empty() {
tokens.iter().for_each(|i| i.inline()); println!("{}", "Cool, cannot find any TODO in this project".green().bold())
} else { } else {
tokens.iter().for_each(|i| println!("{}", i)); if matches.is_present("inline") {
tokens.iter().for_each(|i| i.inline());
} else {
tokens.iter().for_each(|i| println!("{}", i));
}
println!("Found total {} TODOs, {} displayed",
format!(" {} ", total_count.to_string()).bold().underline(),
format!(" {} ", tokens.len().to_string()).bold().green().underline(),
);
} }
Ok(()) Ok(())