Add default value for analysis dir

Signed-off-by: hr567 <hr567@hr567.me>
This commit is contained in:
hr567 2024-06-04 11:17:25 +08:00
parent caafa8f715
commit 72574323ba
Signed by: hr567
SSH Key Fingerprint: SHA256:AUrHz/3TKmc6tf4wqaaEPV/BGQ4aULSp7g76FpqujYw
2 changed files with 10 additions and 6 deletions

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"] }
clap = { version = "4.5.4", features = ["derive", "string"] }
dashmap = { version = "5.5.3", features = ["serde"] }
futures = "0.3.30"
indicatif = "0.17.8"

View File

@ -1,5 +1,6 @@
use std::cell::RefCell;
use std::collections::{BTreeMap, HashMap, VecDeque};
use std::env::current_dir;
use std::fmt::Write;
use std::io;
use std::path::PathBuf;
@ -20,20 +21,23 @@ use tracing::Instrument;
#[derive(Parser, Debug)]
#[command(version, about, long_about = None)]
struct Args {
#[arg(short, long)]
#[arg(short, long, default_value =
current_dir().unwrap().into_os_string())]
directory: PathBuf,
#[arg(short, long, default_value = "16")]
jobs: usize,
#[arg(short = 'f', long)]
keywords_file: Option<PathBuf>,
#[arg(short = 'f', long, default_value =
current_dir().unwrap().join("keywords.txt").into_os_string())]
keywords_file: PathBuf,
#[arg(last = true)]
keywords: Vec<String>,
}
static ARGS: Lazy<Args> = Lazy::new(|| {
let mut args = Args::parse();
if let Some(keywords_file) = args.keywords_file.take() {
let content = std::fs::read_to_string(keywords_file).expect("failed to read keywords file");
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()));
}
args