hr567/word-freq-analyzer
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Use typed arguments for default values

Browse Source 
Signed-off-by: hr567 <hr567@hr567.me>
This commit is contained in:
hr567 2024-06-04 14:45:46 +08:00
parent 72574323ba
commit 7e8f4f8a00
Signed by: hr567
SSH Key Fingerprint: SHA256:AUrHz/3TKmc6tf4wqaaEPV/BGQ4aULSp7g76FpqujYw
2 changed files with 8 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
Cargo.toml
View File

@ -6,7 +6,7 @@ edition = "2021"
# See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
[dependencies]
clap = { version = "4.5.4", features = ["derive", "string"] }
clap = { version = "4.5.4", features = ["derive"] }
dashmap = { version = "5.5.3", features = ["serde"] }
futures = "0.3.30"
indicatif = "0.17.8"

16
src/main.rs
View File

@ -2,6 +2,7 @@ use std::cell::RefCell;
use std::collections::{BTreeMap, HashMap, VecDeque};
use std::env::current_dir;
use std::fmt::Write;
use std::fs::read_to_string;
use std::io;
use std::path::PathBuf;
use std::rc::{Rc, Weak};
@ -21,13 +22,11 @@ use tracing::Instrument;
#[derive(Parser, Debug)]
#[command(version, about, long_about = None)]
struct Args {
#[arg(short, long, default_value =
current_dir().unwrap().into_os_string())]
#[arg(short, long, default_value_os_t = current_dir().unwrap())]
directory: PathBuf,
#[arg(short, long, default_value = "16")]
#[arg(short, long, default_value_t = 16)]
jobs: usize,
#[arg(short = 'f', long, default_value =
current_dir().unwrap().join("keywords.txt").into_os_string())]
#[arg(short = 'f', long, default_value_os_t = current_dir().unwrap().join("keywords.txt"))]
keywords_file: PathBuf,
#[arg(last = true)]
keywords: Vec<String>,
@ -35,10 +34,9 @@ struct Args {
static ARGS: Lazy<Args> = Lazy::new(|| {
let mut args = Args::parse();
if args.keywords_file.exists() {
let content =
std::fs::read_to_string(&args.keywords_file).expect("failed to read keywords file");
args.keywords.extend(content.lines().map(|s| s.to_string()));
if args.keywords.is_empty() && args.keywords_file.exists() {
let content = read_to_string(&args.keywords_file).expect("failed to read keywords file");
args.keywords = content.split_whitespace().map(|s| s.to_string()).collect();
}
args
});