Sulkta-OSS/rustypipe
Archived
2
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

feat(cli): add vdata argument

Browse source
This commit is contained in:
ThetaDev 2023-08-22 23:35:07 +02:00
parent e2eda901b1
commit 93e5ad22e9
1 changed files with 9 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

14
cli/src/main.rs
View file

@ -22,6 +22,9 @@ struct Cli {
/// Always generate a report (used for debugging)
#[clap(long)]
report: bool,
/// YouTube visitor data cookie
#[clap(long)]
vdata: Option<String>,
}
#[derive(Subcommand)]
@ -391,13 +394,14 @@ async fn main() {
let cli = Cli::parse();
let mut storage_dir = dirs::data_dir().expect("no data dir");
storage_dir.push("rustypipe");
std::fs::create_dir_all(&storage_dir).expect("could not create data dir");
let mut rp = RustyPipe::builder().storage_dir(storage_dir);
let mut rp = RustyPipe::builder().visitor_data_opt(cli.vdata);
if cli.report {
rp = rp.report();
} else {
let mut storage_dir = dirs::data_dir().expect("no data dir");
storage_dir.push("rustypipe");
std::fs::create_dir_all(&storage_dir).expect("could not create data dir");
rp = rp.storage_dir(storage_dir);
}
let rp = rp.build().unwrap();