52 lines
1.6 KiB
Rust
52 lines
1.6 KiB
Rust
mod commands;
|
|
mod structs;
|
|
mod utils;
|
|
use clap::{Arg, Command};
|
|
extern crate tokio;
|
|
|
|
fn main() {
|
|
let matches = Command::new("hentaihavenrs")
|
|
.about("hentaihaven.tv downloader in rust")
|
|
.version(env!("CARGO_PKG_VERSION"))
|
|
.subcommand_required(true)
|
|
.subcommand(
|
|
Command::new("search").arg(
|
|
Arg::new("query")
|
|
.takes_value(true)
|
|
.multiple_values(true)
|
|
.help("Search query"),
|
|
),
|
|
)
|
|
.subcommand(
|
|
Command::new("view").visible_aliases(&["info", "show"]).arg(
|
|
Arg::new("id")
|
|
.takes_value(true)
|
|
.multiple_values(true)
|
|
.required(true),
|
|
),
|
|
)
|
|
.subcommand(
|
|
Command::new("download")
|
|
.visible_alias("dl")
|
|
.arg(
|
|
Arg::new("print")
|
|
.long("print")
|
|
.short('p')
|
|
.help("Print the URL to download only"),
|
|
)
|
|
.arg(Arg::new("id").takes_value(true).required(true)),
|
|
)
|
|
.get_matches();
|
|
|
|
let runtime = tokio::runtime::Builder::new_current_thread()
|
|
.enable_all()
|
|
.build()
|
|
.unwrap();
|
|
match matches.subcommand() {
|
|
Some(("search", sub_m)) => runtime.block_on(commands::search(sub_m)),
|
|
Some(("view", sub_m)) => runtime.block_on(commands::view(sub_m)),
|
|
Some(("download", sub_m)) => runtime.block_on(commands::download(sub_m)),
|
|
_ => unreachable!("subcommand_required do your job please"),
|
|
};
|
|
}
|