NyxErinys/crunchy-cli
1
0
Fork 0
mirror of https://github.com/crunchy-labs/crunchy-cli.git synced 2026-01-21 04:02:00 -06:00
Code Issues Projects Releases Packages Wiki Activity Actions

Add root flag to set custom user agent

Browse source
This commit is contained in:
bytedream 2023-09-06 03:02:40 +02:00
parent b477ca982c
commit 0f7e6c9120
1 changed files with 12 additions and 14 deletions
Download patch file Download diff file Expand all files Collapse all files

26
crunchy-cli-core/src/lib.rs
View file

@ -63,6 +63,10 @@ pub struct Cli {
#[arg(value_parser = crate::utils::clap::clap_parse_proxy)]
proxy: Option<Proxy>,
#[arg(help = "Use custom user agent")]
#[clap(long)]
user_agent: Option<String>,
#[clap(subcommand)]
command: Command,
}
@ -261,15 +265,17 @@ async fn crunchyroll_session(cli: &mut Cli) -> Result<Crunchyroll> {
lang
};
let proxy = cli.proxy.clone();
let mut builder = Crunchyroll::builder()
.locale(locale)
.client({
let builder = if let Some(p) = &proxy {
CrunchyrollBuilder::predefined_client_builder().proxy(p.clone())
} else {
CrunchyrollBuilder::predefined_client_builder()
};
let mut builder = CrunchyrollBuilder::predefined_client_builder();
if let Some(p) = &cli.proxy {
builder = builder.proxy(p.clone())
}
if let Some(ua) = &cli.user_agent {
builder = builder.user_agent(ua)
}
#[cfg(any(feature = "openssl", feature = "openssl-static"))]
let client = {
let mut builder = builder.use_native_tls().tls_built_in_root_certs(false);
@ -292,14 +298,6 @@ async fn crunchyroll_session(cli: &mut Cli) -> Result<Crunchyroll> {
if let Command::Download(download) = &cli.command {
builder = builder.preferred_audio_locale(download.audio.clone())
}
if let Some(p) = &cli.proxy {
builder = builder.client(
CrunchyrollBuilder::predefined_client_builder()
.proxy(p.clone())
.build()
.unwrap(),
)
}
let root_login_methods_count = cli.login_method.credentials.is_some() as u8
+ cli.login_method.etp_rt.is_some() as u8