mirror of
https://github.com/crunchy-labs/crunchy-cli.git
synced 2026-01-21 12:12:00 -06:00
510 lines
20 KiB
Rust
510 lines
20 KiB
Rust
use crate::cli::log::tab_info;
|
|
use crate::cli::utils::{download_segments, find_resolution, FFmpegPreset};
|
|
use crate::utils::context::Context;
|
|
use crate::utils::format::{format_string, Format};
|
|
use crate::utils::log::progress;
|
|
use crate::utils::os::{free_file, has_ffmpeg, is_special_file};
|
|
use crate::utils::parse::{parse_url, UrlFilter};
|
|
use crate::utils::sort::{sort_formats_after_seasons, sort_seasons_after_number};
|
|
use crate::Execute;
|
|
use anyhow::{bail, Result};
|
|
use crunchyroll_rs::media::{Resolution, VariantData};
|
|
use crunchyroll_rs::{
|
|
Episode, Locale, Media, MediaCollection, Movie, MovieListing, Season, Series,
|
|
};
|
|
use log::{debug, error, info, warn};
|
|
use std::fs::File;
|
|
use std::path::{Path, PathBuf};
|
|
use std::process::{Command, Stdio};
|
|
|
|
#[derive(Debug, clap::Parser)]
|
|
#[clap(about = "Download a video")]
|
|
#[command(arg_required_else_help(true))]
|
|
pub struct Download {
|
|
#[arg(help = format!("Audio language. Can only be used if the provided url(s) point to a series. \
|
|
Available languages are: {}", Locale::all().into_iter().map(|l| l.to_string()).collect::<Vec<String>>().join(", ")))]
|
|
#[arg(long_help = format!("Audio language. Can only be used if the provided url(s) point to a series. \
|
|
Available languages are:\n{}", Locale::all().into_iter().map(|l| format!("{:<6} → {}", l.to_string(), l.to_human_readable())).collect::<Vec<String>>().join("\n ")))]
|
|
#[arg(short, long, default_value_t = crate::utils::locale::system_locale())]
|
|
audio: Locale,
|
|
#[arg(help = format!("Subtitle language. Available languages are: {}", Locale::all().into_iter().map(|l| l.to_string()).collect::<Vec<String>>().join(", ")))]
|
|
#[arg(long_help = format!("Subtitle language. If set, the subtitle will be burned into the video and cannot be disabled. \
|
|
Available languages are: {}", Locale::all().into_iter().map(|l| l.to_string()).collect::<Vec<String>>().join(", ")))]
|
|
#[arg(short, long)]
|
|
subtitle: Option<Locale>,
|
|
|
|
#[arg(help = "Name of the output file")]
|
|
#[arg(long_help = "Name of the output file.\
|
|
If you use one of the following pattern they will get replaced:\n \
|
|
{title} → Title of the video\n \
|
|
{series_name} → Name of the series\n \
|
|
{season_name} → Name of the season\n \
|
|
{audio} → Audio language of the video\n \
|
|
{resolution} → Resolution of the video\n \
|
|
{padded_season_number} → Number of the season padded to double digits\n \
|
|
{season_number} → Number of the season\n \
|
|
{padded_episode_number} → Number of the episode padded to double digits\n \
|
|
{episode_number} → Number of the episode\n \
|
|
{series_id} → ID of the series\n \
|
|
{season_id} → ID of the season\n \
|
|
{episode_id} → ID of the episode")]
|
|
#[arg(short, long, default_value = "{title}.ts")]
|
|
output: String,
|
|
|
|
#[arg(help = "Video resolution")]
|
|
#[arg(long_help = "The video resolution.\
|
|
Can either be specified via the pixels (e.g. 1920x1080), the abbreviation for pixels (e.g. 1080p) or 'common-use' words (e.g. best). \
|
|
Specifying the exact pixels is not recommended, use one of the other options instead. \
|
|
Crunchyroll let you choose the quality with pixel abbreviation on their clients, so you might be already familiar with the available options. \
|
|
The available common-use words are 'best' (choose the best resolution available) and 'worst' (worst resolution available)")]
|
|
#[arg(short, long, default_value = "best")]
|
|
#[arg(value_parser = crate::utils::clap::clap_parse_resolution)]
|
|
resolution: Resolution,
|
|
|
|
#[arg(help = format!("Presets for video converting. Can be used multiple times. \
|
|
Available presets: \n {}", FFmpegPreset::all().into_iter().map(|p| format!("{}: {}", p.to_string(), p.description())).collect::<Vec<String>>().join("\n ")))]
|
|
#[arg(long_help = format!("Presets for video converting. Can be used multiple times. \
|
|
Generally used to minify the file size with keeping (nearly) the same quality. \
|
|
It is recommended to only use this if you download videos with high resolutions since low resolution videos tend to result in a larger file with any of the provided presets. \
|
|
Available presets: \n {}", FFmpegPreset::all().into_iter().map(|p| format!("{}: {}", p.to_string(), p.description())).collect::<Vec<String>>().join("\n ")))]
|
|
#[arg(long)]
|
|
#[arg(value_parser = FFmpegPreset::parse)]
|
|
ffmpeg_preset: Vec<FFmpegPreset>,
|
|
|
|
#[arg(help = "Url(s) to Crunchyroll episodes or series")]
|
|
urls: Vec<String>,
|
|
}
|
|
|
|
#[async_trait::async_trait(?Send)]
|
|
impl Execute for Download {
|
|
fn pre_check(&self) -> Result<()> {
|
|
if has_ffmpeg() {
|
|
debug!("FFmpeg detected")
|
|
} else if PathBuf::from(&self.output)
|
|
.extension()
|
|
.unwrap_or_default()
|
|
.to_string_lossy()
|
|
!= "ts"
|
|
{
|
|
bail!("File extension is not '.ts'. If you want to use a custom file format, please install ffmpeg")
|
|
} else if !self.ffmpeg_preset.is_empty() {
|
|
bail!("FFmpeg is required to use (ffmpeg) presets")
|
|
}
|
|
|
|
let _ = FFmpegPreset::ffmpeg_presets(self.ffmpeg_preset.clone())?;
|
|
if self.ffmpeg_preset.len() == 1
|
|
&& self.ffmpeg_preset.get(0).unwrap() == &FFmpegPreset::Nvidia
|
|
{
|
|
warn!("Skipping 'nvidia' hardware acceleration preset since no other codec preset was specified")
|
|
}
|
|
|
|
Ok(())
|
|
}
|
|
|
|
async fn execute(self, ctx: Context) -> Result<()> {
|
|
let mut parsed_urls = vec![];
|
|
|
|
for (i, url) in self.urls.iter().enumerate() {
|
|
let progress_handler = progress!("Parsing url {}", i + 1);
|
|
match parse_url(&ctx.crunchy, url.clone(), true).await {
|
|
Ok((media_collection, url_filter)) => {
|
|
parsed_urls.push((media_collection, url_filter));
|
|
progress_handler.stop(format!("Parsed url {}", i + 1))
|
|
}
|
|
Err(e) => bail!("url {} could not be parsed: {}", url, e),
|
|
}
|
|
}
|
|
|
|
for (i, (media_collection, url_filter)) in parsed_urls.into_iter().enumerate() {
|
|
let progress_handler = progress!("Fetching series details");
|
|
let formats = match media_collection {
|
|
MediaCollection::Series(series) => {
|
|
debug!("Url {} is series ({})", i + 1, series.title);
|
|
formats_from_series(&self, series, &url_filter).await?
|
|
}
|
|
MediaCollection::Season(season) => {
|
|
debug!(
|
|
"Url {} is season {} ({})",
|
|
i + 1,
|
|
season.metadata.season_number,
|
|
season.title
|
|
);
|
|
formats_from_season(&self, season, &url_filter).await?
|
|
}
|
|
MediaCollection::Episode(episode) => {
|
|
debug!(
|
|
"Url {} is episode {} ({}) of season {} ({}) of {}",
|
|
i + 1,
|
|
episode.metadata.episode_number,
|
|
episode.title,
|
|
episode.metadata.season_number,
|
|
episode.metadata.season_title,
|
|
episode.metadata.series_title
|
|
);
|
|
format_from_episode(&self, episode, &url_filter, false)
|
|
.await?
|
|
.map(|fmt| vec![fmt])
|
|
}
|
|
MediaCollection::MovieListing(movie_listing) => {
|
|
debug!("Url {} is movie listing ({})", i + 1, movie_listing.title);
|
|
format_from_movie_listing(&self, movie_listing, &url_filter).await?
|
|
}
|
|
MediaCollection::Movie(movie) => {
|
|
debug!("Url {} is movie ({})", i + 1, movie.title);
|
|
format_from_movie(&self, movie, &url_filter)
|
|
.await?
|
|
.map(|fmt| vec![fmt])
|
|
}
|
|
};
|
|
|
|
let Some(formats) = formats else {
|
|
progress_handler.stop(format!("Skipping url {} (no matching episodes found)", i + 1));
|
|
continue;
|
|
};
|
|
progress_handler.stop(format!("Loaded series information for url {}", i + 1));
|
|
|
|
if log::max_level() == log::Level::Debug {
|
|
let seasons = sort_formats_after_seasons(formats.clone());
|
|
debug!("Series has {} seasons", seasons.len());
|
|
for (i, season) in seasons.into_iter().enumerate() {
|
|
info!("Season {} ({})", i + 1, season.get(0).unwrap().season_title);
|
|
for format in season {
|
|
info!(
|
|
"{}: {}px, {:.02} FPS (S{:02}E{:02})",
|
|
format.title,
|
|
format.stream.resolution,
|
|
format.stream.fps,
|
|
format.season_number,
|
|
format.number,
|
|
)
|
|
}
|
|
}
|
|
} else {
|
|
for season in sort_formats_after_seasons(formats.clone()) {
|
|
let first = season.get(0).unwrap();
|
|
info!(
|
|
"{} Season {} ({})",
|
|
first.series_name, first.season_number, first.season_title
|
|
);
|
|
|
|
for (i, format) in season.into_iter().enumerate() {
|
|
tab_info!(
|
|
"{}. {} » {}px, {:.2} FPS (S{:02}E{:02})",
|
|
i + 1,
|
|
format.title,
|
|
format.stream.resolution,
|
|
format.stream.fps,
|
|
format.season_number,
|
|
format.number
|
|
)
|
|
}
|
|
}
|
|
}
|
|
|
|
for format in formats {
|
|
let mut path = PathBuf::from(&self.output);
|
|
path = free_file(
|
|
path.with_file_name(format_string(
|
|
if let Some(fname) = path.file_name() {
|
|
fname.to_str().unwrap()
|
|
} else {
|
|
"{title}.ts"
|
|
}
|
|
.to_string(),
|
|
&format,
|
|
true,
|
|
)),
|
|
);
|
|
|
|
info!(
|
|
"Downloading {} to '{}'",
|
|
format.title,
|
|
if is_special_file(&path) {
|
|
path.to_str().unwrap()
|
|
} else {
|
|
path.file_name().unwrap().to_str().unwrap()
|
|
}
|
|
);
|
|
tab_info!("Episode: S{:02}E{:02}", format.season_number, format.number);
|
|
tab_info!("Audio: {}", format.audio);
|
|
tab_info!(
|
|
"Subtitles: {}",
|
|
self.subtitle
|
|
.clone()
|
|
.map_or("None".to_string(), |l| l.to_string())
|
|
);
|
|
tab_info!("Resolution: {}", format.stream.resolution);
|
|
tab_info!("FPS: {:.2}", format.stream.fps);
|
|
|
|
let extension = path.extension().unwrap_or_default().to_string_lossy();
|
|
|
|
if (!extension.is_empty() && extension != "ts") || !self.ffmpeg_preset.is_empty() {
|
|
download_ffmpeg(&ctx, &self, format.stream, path.as_path()).await?;
|
|
} else if path.to_str().unwrap() == "-" {
|
|
let mut stdout = std::io::stdout().lock();
|
|
download_segments(&ctx, &mut stdout, None, format.stream).await?;
|
|
} else {
|
|
// create parent directory if it does not exist
|
|
if let Some(parent) = path.parent() {
|
|
if !parent.exists() {
|
|
std::fs::create_dir_all(parent)?
|
|
}
|
|
}
|
|
let mut file = File::options().create(true).write(true).open(&path)?;
|
|
download_segments(&ctx, &mut file, None, format.stream).await?
|
|
}
|
|
}
|
|
}
|
|
|
|
Ok(())
|
|
}
|
|
}
|
|
|
|
async fn download_ffmpeg(
|
|
ctx: &Context,
|
|
download: &Download,
|
|
variant_data: VariantData,
|
|
target: &Path,
|
|
) -> Result<()> {
|
|
let (input_presets, output_presets) =
|
|
FFmpegPreset::ffmpeg_presets(download.ffmpeg_preset.clone())?;
|
|
|
|
// create parent directory if it does not exist
|
|
if let Some(parent) = target.parent() {
|
|
if !parent.exists() {
|
|
std::fs::create_dir_all(parent)?
|
|
}
|
|
}
|
|
|
|
let mut ffmpeg = Command::new("ffmpeg")
|
|
.stdin(Stdio::piped())
|
|
.stdout(Stdio::null())
|
|
.stderr(Stdio::piped())
|
|
.arg("-y")
|
|
.args(input_presets)
|
|
.args(["-f", "mpegts", "-i", "pipe:"])
|
|
.args(
|
|
if target
|
|
.extension()
|
|
.unwrap_or_default()
|
|
.to_string_lossy()
|
|
.is_empty()
|
|
{
|
|
vec!["-f", "mpegts"]
|
|
} else {
|
|
vec![]
|
|
}
|
|
.as_slice(),
|
|
)
|
|
.args(output_presets)
|
|
.arg(target.to_str().unwrap())
|
|
.spawn()?;
|
|
|
|
download_segments(ctx, &mut ffmpeg.stdin.take().unwrap(), None, variant_data).await?;
|
|
|
|
let _progress_handler = progress!("Generating output file");
|
|
ffmpeg.wait()?;
|
|
info!("Output file generated");
|
|
|
|
Ok(())
|
|
}
|
|
|
|
async fn formats_from_series(
|
|
download: &Download,
|
|
series: Media<Series>,
|
|
url_filter: &UrlFilter,
|
|
) -> Result<Option<Vec<Format>>> {
|
|
if !series.metadata.audio_locales.is_empty()
|
|
&& !series.metadata.audio_locales.contains(&download.audio)
|
|
{
|
|
error!(
|
|
"Series {} is not available with {} audio",
|
|
series.title, download.audio
|
|
);
|
|
return Ok(None);
|
|
}
|
|
|
|
let mut seasons = series.seasons().await?;
|
|
|
|
// filter any season out which does not contain the specified audio language
|
|
for season in sort_seasons_after_number(seasons.clone()) {
|
|
// check if the current iterated season has the specified audio language
|
|
if !season
|
|
.iter()
|
|
.any(|s| s.metadata.audio_locales.contains(&download.audio))
|
|
{
|
|
error!(
|
|
"Season {} of series {} is not available with {} audio",
|
|
season.first().unwrap().metadata.season_number,
|
|
series.title,
|
|
download.audio
|
|
);
|
|
}
|
|
|
|
// remove all seasons with the wrong audio for the current iterated season number
|
|
seasons.retain(|s| {
|
|
s.metadata.season_number != season.first().unwrap().metadata.season_number
|
|
|| s.metadata.audio_locales.contains(&download.audio)
|
|
})
|
|
}
|
|
|
|
let mut formats = vec![];
|
|
for season in seasons {
|
|
if let Some(fmts) = formats_from_season(download, season, url_filter).await? {
|
|
formats.extend(fmts)
|
|
}
|
|
}
|
|
|
|
Ok(some_vec_or_none(formats))
|
|
}
|
|
|
|
async fn formats_from_season(
|
|
download: &Download,
|
|
season: Media<Season>,
|
|
url_filter: &UrlFilter,
|
|
) -> Result<Option<Vec<Format>>> {
|
|
if !season.metadata.audio_locales.contains(&download.audio) {
|
|
error!(
|
|
"Season {} ({}) is not available with {} audio",
|
|
season.metadata.season_number, season.title, download.audio
|
|
);
|
|
return Ok(None);
|
|
} else if !url_filter.is_season_valid(season.metadata.season_number) {
|
|
return Ok(None);
|
|
}
|
|
|
|
let mut formats = vec![];
|
|
|
|
for episode in season.episodes().await? {
|
|
if let Some(fmt) = format_from_episode(download, episode, url_filter, true).await? {
|
|
formats.push(fmt)
|
|
}
|
|
}
|
|
|
|
Ok(some_vec_or_none(formats))
|
|
}
|
|
|
|
async fn format_from_episode(
|
|
download: &Download,
|
|
episode: Media<Episode>,
|
|
url_filter: &UrlFilter,
|
|
filter_audio: bool,
|
|
) -> Result<Option<Format>> {
|
|
if filter_audio && episode.metadata.audio_locale != download.audio {
|
|
error!(
|
|
"Episode {} ({}) of season {} ({}) of {} has no {} audio",
|
|
episode.metadata.episode_number,
|
|
episode.title,
|
|
episode.metadata.season_number,
|
|
episode.metadata.season_title,
|
|
episode.metadata.series_title,
|
|
download.audio
|
|
);
|
|
return Ok(None);
|
|
} else if !url_filter.is_episode_valid(
|
|
episode.metadata.episode_number,
|
|
episode.metadata.season_number,
|
|
) {
|
|
return Ok(None);
|
|
}
|
|
|
|
let streams = episode.streams().await?;
|
|
let streaming_data = if let Some(subtitle) = &download.subtitle {
|
|
if !streams.subtitles.keys().cloned().any(|x| &x == subtitle) {
|
|
error!(
|
|
"Episode {} ({}) of season {} ({}) of {} has no {} subtitles",
|
|
episode.metadata.episode_number,
|
|
episode.title,
|
|
episode.metadata.season_number,
|
|
episode.metadata.season_title,
|
|
episode.metadata.series_title,
|
|
subtitle
|
|
);
|
|
return Ok(None);
|
|
}
|
|
streams.hls_streaming_data(Some(subtitle.clone())).await?
|
|
} else {
|
|
streams.hls_streaming_data(None).await?
|
|
};
|
|
|
|
let Some(stream) = find_resolution(streaming_data, &download.resolution) else {
|
|
bail!(
|
|
"Resolution ({}x{}) is not available for episode {} ({}) of season {} ({}) of {}",
|
|
download.resolution.width,
|
|
download.resolution.height,
|
|
episode.metadata.episode_number,
|
|
episode.title,
|
|
episode.metadata.season_number,
|
|
episode.metadata.season_title,
|
|
episode.metadata.series_title
|
|
)
|
|
};
|
|
|
|
Ok(Some(Format::new_from_episode(episode, stream)))
|
|
}
|
|
|
|
async fn format_from_movie_listing(
|
|
download: &Download,
|
|
movie_listing: Media<MovieListing>,
|
|
url_filter: &UrlFilter,
|
|
) -> Result<Option<Vec<Format>>> {
|
|
let mut formats = vec![];
|
|
|
|
for movie in movie_listing.movies().await? {
|
|
if let Some(fmt) = format_from_movie(download, movie, url_filter).await? {
|
|
formats.push(fmt)
|
|
}
|
|
}
|
|
|
|
Ok(some_vec_or_none(formats))
|
|
}
|
|
|
|
async fn format_from_movie(
|
|
download: &Download,
|
|
movie: Media<Movie>,
|
|
_: &UrlFilter,
|
|
) -> Result<Option<Format>> {
|
|
let streams = movie.streams().await?;
|
|
let mut streaming_data = if let Some(subtitle) = &download.subtitle {
|
|
if !streams.subtitles.keys().cloned().any(|x| &x == subtitle) {
|
|
error!("Movie {} has no {} subtitles", movie.title, subtitle);
|
|
return Ok(None);
|
|
}
|
|
streams.hls_streaming_data(Some(subtitle.clone())).await?
|
|
} else {
|
|
streams.hls_streaming_data(None).await?
|
|
};
|
|
|
|
streaming_data.sort_by(|a, b| a.resolution.width.cmp(&b.resolution.width).reverse());
|
|
let stream = {
|
|
match download.resolution.height {
|
|
u64::MAX => streaming_data.into_iter().next().unwrap(),
|
|
u64::MIN => streaming_data.into_iter().last().unwrap(),
|
|
_ => {
|
|
if let Some(streaming_data) = streaming_data.into_iter().find(|v| {
|
|
download.resolution.height == u64::MAX
|
|
|| v.resolution.height == download.resolution.height
|
|
}) {
|
|
streaming_data
|
|
} else {
|
|
bail!(
|
|
"Resolution ({}x{}) is not available for movie {}",
|
|
download.resolution.width,
|
|
download.resolution.height,
|
|
movie.title
|
|
)
|
|
}
|
|
}
|
|
}
|
|
};
|
|
|
|
Ok(Some(Format::new_from_movie(movie, stream)))
|
|
}
|
|
|
|
fn some_vec_or_none<T>(v: Vec<T>) -> Option<Vec<T>> {
|
|
if v.is_empty() {
|
|
None
|
|
} else {
|
|
Some(v)
|
|
}
|
|
}
|