+# This project has been sunset as Crunchyroll moved to a DRM-only system. See [#362](https://github.com/crunchy-labs/crunchy-cli/issues/362).
-# crunchyroll-go
+# crunchy-cli
-A [Go](https://golang.org) library & cli for the undocumented [crunchyroll](https://www.crunchyroll.com) api. To use it, you need a crunchyroll premium account to for full (api) access.
+👇 A Command-line downloader for [Crunchyroll](https://www.crunchyroll.com).
-# 🖥️ CLI
+> We are in no way affiliated with, maintained, authorized, sponsored, or officially associated with Crunchyroll LLC or any of its subsidiaries or affiliates.
+> The official Crunchyroll website can be found at [www.crunchyroll.com](https://www.crunchyroll.com/).
## ✨ Features
-- Download single videos and entire series from [crunchyroll](https://www.crunchyroll.com).
-- Archive episode or seasons in an `.mkv` file with multiple subtitles and audios and compress them to gzip or zip files.
-- Specify a range which episodes to download from an anime.
+- Download single videos and entire series from [Crunchyroll](https://www.crunchyroll.com).
+- Archive episodes or seasons in an `.mkv` file with multiple subtitles and audios.
+- Specify a range of episodes to download from an anime.
+- Search through the Crunchyroll collection and return metadata (title, duration, direct stream link, ...) of all media types.
## 💾 Get the executable
-- 📥 Download the latest binaries [here](https://github.com/ByteDream/crunchyroll-go/releases/latest) or get it from below:
- - [Linux (x64)](https://smartrelease.bytedream.org/github/ByteDream/crunchyroll-go/crunchy-{tag}_linux)
- - [Windows (x64)](https://smartrelease.bytedream.org/github/ByteDream/crunchyroll-go/crunchy-{tag}_windows.exe)
- - [MacOS (x64)](https://smartrelease.bytedream.org/github/ByteDream/crunchyroll-go/crunchy-{tag}_darwin)
-- If you use Arch btw. or any other Linux distro which is based on Arch Linux, you can download the package via the [AUR](https://aur.archlinux.org/packages/crunchyroll-go/):
- ```
- $ yay -S crunchyroll-go
- ```
-- 🛠 Build it yourself
- - use `make` (requires `go` to be installed):
- ```
- $ git clone https://github.com/ByteDream/crunchyroll-go
- $ cd crunchyroll-go
- $ make && sudo make install
- ```
- - use `go`:
- ```
- $ git clone https://github.com/ByteDream/crunchyroll-go
- $ cd crunchyroll-go/cmd/crunchyroll-go
- $ go build -o crunchy
+### 📥 Download the latest binaries
+
+Check out the [releases](https://github.com/crunchy-labs/crunchy-cli/releases) tab and get the binary from the latest (pre-)release.
+
+### 📦 Get it via a package manager
+
+- [AUR](https://aur.archlinux.org/)
+
+ If you're using Arch or an Arch based Linux distribution you are able to install our [AUR](https://aur.archlinux.org/) package.
+ You need an [AUR helper](https://wiki.archlinux.org/title/AUR_helpers) like [yay](https://github.com/Jguer/yay) to install it.
+
+ ```shell
+ # this package builds crunchy-cli manually (recommended)
+ $ yay -S crunchy-cli
+ # this package installs the latest pre-compiled release binary
+ $ yay -S crunchy-cli-bin
```
-## 📝 Examples
+- [Scoop](https://scoop.sh/)
-_Before reading_: Because of the huge functionality not all cases can be covered in the README.
-Make sure to check the [wiki](https://github.com/ByteDream/crunchyroll-go/wiki/Cli), further usages and options are described there.
+ For Windows users, we support the [scoop](https://scoop.sh/#/) command-line installer.
+
+ ```shell
+ $ scoop bucket add extras
+ $ scoop install extras/crunchy-cli
+ ```
+
+- [Homebrew](https://brew.sh/)
+
+ For macOS/linux users, we support the [brew](https://brew.sh/#/) command-line installer. Packages are compiled by the [homebrew project](https://formulae.brew.sh/formula/crunchy-cli), and will also install the `openssl@3` and `ffmpeg` dependencies.
+
+ ```shell
+ $ brew install crunchy-cli
+ ```
+
+ Supported archs: `x86_64_linux`, `arm64_monterey`, `sonoma`, `ventura`
+
+- [Nix](https://nixos.org/)
+
+ This requires [nix](https://nixos.org) and you'll probably need `--extra-experimental-features "nix-command flakes"`, depending on your configurations.
+
+ ```shell
+ $ nix github:crunchy-labs/crunchy-cli
+ ```
+
+### 🛠 Build it yourself
+
+Since we do not support every platform and architecture you may have to build the project yourself.
+This requires [git](https://git-scm.com/) and [Cargo](https://doc.rust-lang.org/cargo).
+
+```shell
+$ git clone https://github.com/crunchy-labs/crunchy-cli
+$ cd crunchy-cli
+# either just build it (will be available in ./target/release/crunchy-cli)...
+$ cargo build --release
+# ... or install it globally
+$ cargo install --force --path .
+```
+
+## 🖥️ Usage
+
+> All shown commands are examples 🧑🏼🍳
+
+### Global Flags
+
+crunchy-cli requires you to log in.
+Though you can use a non-premium account, you will not have access to premium content without a subscription.
+You can authenticate with your credentials (email:password) or by using a refresh token.
+
+- Credentials
+
+ ```shell
+ $ crunchy-cli --credentials "email:password"
+ ```
+
+- Stay Anonymous
+
+ Login without an account (you won't be able to access premium content):
+
+ ```shell
+ $ crunchy-cli --anonymous
+ ```
+
+### Global settings
+
+You can set specific settings which will be
+
+- Verbose output
+
+ If you want to include debug information in the output, use the `-v` / `--verbose` flag to show it.
+
+ ```shell
+ $ crunchy-cli -v
+ ```
+
+ This flag can't be used in combination with `-q` / `--quiet`.
+
+- Quiet output
+
+ If you want to hide all output, use the `-q` / `--quiet` flag to do so.
+ This is especially useful if you want to pipe the output video to an external program (like a video player).
+
+ ```shell
+ $ crunchy-cli -q
+ ```
+
+ This flag can't be used in combination with `-v` / `--verbose`.
+
+- Language
+
+ By default, the resulting metadata like title or description are shown in your system language (if Crunchyroll supports it, else in English).
+ If you want to show the results in another language, use the `--lang` flag to set it.
+
+ ```shell
+ $ crunchy-cli --lang de-DE
+ ```
+
+- Experimental fixes
+
+ Crunchyroll constantly changes and breaks its services or just delivers incorrect answers.
+ The `--experimental-fixes` flag tries to fix some of those issues.
+ As the *experimental* in `--experimental-fixes` states, these fixes may or may not break other functionality.
+
+ ```shell
+ $ crunchy-cli --experimental-fixes
+ ```
+
+ For an overview which parts this flag affects, see the [documentation](https://docs.rs/crunchyroll-rs/latest/crunchyroll_rs/crunchyroll/struct.CrunchyrollBuilder.html) of the underlying Crunchyroll library, all functions beginning with `stabilization_` are applied.
+
+- Proxy
+
+ The `--proxy` flag supports https and socks5 proxies to route all your traffic through.
+ This may be helpful to bypass the geo-restrictions Crunchyroll has on certain series.
+ You are also able to set in which part of the cli a proxy should be used.
+ Instead of a normal url you can also use: `:` (only proxies api requests), `:` (only proxies download traffic), `:` (proxies api requests through the first url and download traffic through the second url).
+
+ ```shell
+ $ crunchy-cli --proxy socks5://127.0.0.1:8080
+ ```
+
+ Make sure that proxy can either forward TLS requests, which is needed to bypass the (cloudflare) bot protection, or that it is configured so that the proxy can bypass the protection itself.
+
+- User Agent
+
+ There might be cases where a custom user agent is necessary, e.g. to bypass the cloudflare bot protection (#104).
+ In such cases, the `--user-agent` flag can be used to set a custom user agent.
+
+ ```shell
+ $ crunchy-cli --user-agent "Mozilla/4.0 (compatible; MSIE 8.0; Windows NT 5.1; Trident/4.0)"
+ ```
+
+ Default is the user agent, defined in the underlying [library](https://github.com/crunchy-labs/crunchyroll-rs).
+
+- Speed limit
+
+ If you want to limit how fast requests/downloads should be, you can use the `--speed-limit` flag. Allowed units are `B` (bytes), `KB` (kilobytes) and `MB` (megabytes).
+
+ ```shell
+ $ crunchy-cli --speed-limit 10MB
+ ```
### Login
-Before you can do something, you have to login first.
+The `login` command can store your session, so you don't have to authenticate every time you execute a command.
-This can be performed via crunchyroll account email and password.
-```
-$ crunchy login user@example.com password
+```shell
+# save the refresh token which gets generated when login with credentials.
+# your email and password won't be stored at any time on disk
+$ crunchy-cli login --credentials "email:password"
```
-or via session id
-```
-$ crunchy login --session-id 8e9gs135defhga790dvrf2i0eris8gts
-```
+With the session stored, you do not need to pass `--credentials` / `--anonymous` anymore when you want to execute a command.
### Download
-By default, the cli tries to download the episode with your system language as audio.
-If no streams with your system language are available, the video will be downloaded with japanese audio and hardsubbed subtitles in your system language.
-**If your system language is not supported, an error message will be displayed and en-US (american english) will be chosen as language.**
+The `download` command lets you download episodes with a specific audio language and optional subtitles.
-```
-$ crunchy download https://www.crunchyroll.com/darling-in-the-franxx/episode-1-alone-and-lonesome-759575
-```
+**Supported urls**
-With `-r best` the video(s) will have the best available resolution (mostly 1920x1080 / Full HD).
-```
-$ crunchy download -r best https://www.crunchyroll.com/darling-in-the-franxx/episode-1-alone-and-lonesome-759575
-```
+- Single episode (with [episode filtering](#episode-filtering))
+ ```shell
+ $ crunchy-cli download https://www.crunchyroll.com/watch/GRDQPM1ZY/alone-and-lonesome
+ ```
+- Series (with [episode filtering](#episode-filtering))
+ ```shell
+ $ crunchy-cli download https://www.crunchyroll.com/series/GY8VEQ95Y/darling-in-the-franxx
+ ```
-The file is by default saved as a `.ts` (mpeg transport stream) file.
-`.ts` files may can't be played or are looking very weird (it depends on the video player you are using).
-With the `-o` flag, you can change the name (and file ending) of the output file.
-So if you want to save it as, for example, `mp4` file, just name it `whatever.mp4`.
-**You need [ffmpeg](https://ffmpeg.org) to store the video in other file formats.**
-```
-$ crunchy download -o "daaaaaaaaaaaaaaaarling.ts" https://www.crunchyroll.com/darling-in-the-franxx/episode-1-alone-and-lonesome-759575
-```
+**Options**
-With the `--audio` flag you can specify which audio the video should have and with `--subtitle` which subtitle it should have.
-Type `crunchy help download` to see all available locales.
-```
-$ crunchy download --audio ja-JP --subtitle de-DE https://www.crunchyroll.com/darling-in-the-franxx
-```
+- Audio language
-##### Flags
+ Set the audio language with the `-a` / `--audio` flag.
+ This only works if the url points to a series since episode urls are language specific.
-The following flags can be (optional) passed to modify the [download](#download) process.
+ ```shell
+ $ crunchy-cli download -a de-DE https://www.crunchyroll.com/series/GY8VEQ95Y/darling-in-the-franxx
+ ```
-| Short | Extended | Description |
-|-------|----------------|--------------------------------------------------------------------------------|
-| `-a` | `--audio` | Forces audio of the video(s). |
-| `-s` | `--subtitle` | Forces subtitle of the video(s). |
-| `-d` | `--directory` | Directory to download the video(s) to. |
-| `-o` | `--output` | Name of the output file. |
-| `-r` | `--resolution` | The resolution of the video(s). `best` for best resolution, `worst` for worst. |
-| `-g` | `--goroutines` | Sets how many parallel segment downloads should be used. |
+ Default is your system locale. If not supported by Crunchyroll, `en-US` (American English) is the default.
+
+- Subtitle language
+
+ Besides the audio, you can specify the subtitle language by using the `-s` / `--subtitle` flag.
+ In formats that support it (.mp4, .mov and .mkv ), subtitles are stored as soft-subs. All other formats are hardsubbed: the subtitles will be burned into the video track (cf. [hardsub](https://www.urbandictionary.com/define.php?term=hardsub)) and thus can not be turned off.
+
+ ```shell
+ $ crunchy-cli download -s de-DE https://www.crunchyroll.com/series/GY8VEQ95Y/darling-in-the-franxx
+ ```
+
+ Default is none.
+
+- Output template
+
+ Define an output template by using the `-o` / `--output` flag.
+
+ ```shell
+ $ crunchy-cli download -o "ditf.mp4" https://www.crunchyroll.com/watch/GRDQPM1ZY/alone-and-lonesome
+ ```
+
+ Default is `{title}.mp4`. See the [Template Options section](#output-template-options) below for more options.
+
+- Output template for special episodes
+
+ Define an output template which only gets used when the episode is a special (episode number is 0 or has non-zero decimal places) by using the `--output-special` flag.
+
+ ```shell
+ $ crunchy-cli download --output-specials "Special EP - {title}" https://www.crunchyroll.com/watch/GY8D975JY/veldoras-journal
+ ```
+
+ Default is the template, set by the `-o` / `--output` flag. See the [Template Options section](#output-template-options) below for more options.
+
+- Universal output
+
+ The output template options can be forced to get sanitized via the `--universal-output` flag to be valid across all supported operating systems (Windows has a lot of characters which aren't allowed in filenames...).
+
+ ```shell
+ $ crunchy-cli download --universal-output -o https://www.crunchyroll.com/watch/G7PU4XD48/tales-veldoras-journal-2
+ ```
+
+- Resolution
+
+ The resolution for videos can be set via the `-r` / `--resolution` flag.
+
+ ```shell
+ $ crunchy-cli download -r worst https://www.crunchyroll.com/watch/GRDQPM1ZY/alone-and-lonesome
+ ```
+
+ Default is `best`.
+
+- Language tagging
+
+ You can force the usage of a specific language tagging in the output file with the `--language-tagging` flag.
+ This might be useful as some video players doesn't recognize the language tagging Crunchyroll uses internally.
+
+ ```shell
+ $ crunchy-cli download --language-tagging ietf https://www.crunchyroll.com/watch/GRDQPM1ZY/alone-and-lonesome
+ ```
+
+- FFmpeg Preset
+
+ You can specify specific built-in presets with the `--ffmpeg-preset` flag to convert videos to a specific coding while downloading.
+ Multiple predefined presets how videos should be encoded (h264, h265, av1, ...) are available, you can see them with `crunchy-cli download --help`.
+ If you need more specific ffmpeg customizations you could either convert the output file manually or use ffmpeg output arguments as value for this flag.
+
+ ```shell
+ $ crunchy-cli download --ffmpeg-preset av1-lossless https://www.crunchyroll.com/watch/GRDQPM1ZY/alone-and-lonesome
+ ```
+
+- FFmpeg threads
+
+ If you want to manually set how many threads FFmpeg should use, you can use the `--ffmpeg-threads` flag. This does not work with every codec/preset and is skipped entirely when specifying custom ffmpeg output arguments instead of a preset for `--ffmpeg-preset`.
+
+ ```shell
+ $ crunchy-cli download --ffmpeg-threads 4 https://www.crunchyroll.com/watch/GRDQPM1ZY/alone-and-lonesome
+ ```
+
+- Skip existing
+
+ If you re-download a series but want to skip episodes you've already downloaded, the `--skip-existing` flag skips the already existing/downloaded files.
+
+ ```shell
+ $ crunchy-cli download --skip-existing https://www.crunchyroll.com/series/GY8VEQ95Y/darling-in-the-franxx
+ ```
+
+- Skip specials
+
+ If you doesn't want to download special episodes, use the `--skip-specials` flag to skip the download of them.
+
+ ```shell
+ $ crunchy-cli download --skip-specials https://www.crunchyroll.com/series/GYZJ43JMR/that-time-i-got-reincarnated-as-a-slime[S2]
+ ```
+
+- Include chapters
+
+ Crunchyroll sometimes provide information about skippable events like the intro or credits.
+ These information can be stored as chapters in the resulting video file via the `--include-chapters` flag.
+
+ ```shell
+ $ crunchy-cli download --include-chapters https://www.crunchyroll.com/watch/G0DUND0K2/the-journeys-end
+ ```
+
+- Yes
+
+ Sometimes different seasons have the same season number (e.g. Sword Art Online Alicization and Alicization War of Underworld are both marked as season 3), in such cases an interactive prompt is shown which needs user further user input to decide which season to download.
+ The `--yes` flag suppresses this interactive prompt and just downloads all seasons.
+
+ ```shell
+ $ crunchy-cli download --yes https://www.crunchyroll.com/series/GR49G9VP6/sword-art-online
+ ```
+
+ If you've passed the `-q` / `--quiet` [global flag](#global-settings), this flag is automatically set.
+
+- Force hardsub
+
+ If you want to burn-in the subtitles, even if the output format/container supports soft-subs (e.g. `.mp4`), use the `--force-hardsub` flag to do so.
+
+ ```shell
+ $ crunchy-cli download --force-hardsub -s en-US https://www.crunchyroll.com/watch/GRDQPM1ZY/alone-and-lonesome
+ ```
+
+- Threads
+
+ To increase the download speed, video segments are downloaded simultaneously by creating multiple threads.
+ If you want to manually specify how many threads to use when downloading, do this with the `-t` / `--threads` flag.
+
+ ```shell
+ $ crunchy-cli download -t 1 https://www.crunchyroll.com/series/GY8VEQ95Y/darling-in-the-franxx
+ ```
+
+ The default thread count is the count of cpu threads your pc has.
### Archive
-Archive works just like [download](#download). It downloads the given videos as `.mkv` files and stores all (soft) subtitles in it.
-Default audio locales are japanese and your system language (if available) but you can set more or less with the `--language` flag.
+The `archive` command lets you download episodes with multiple audios and subtitles and merges it into a `.mkv` file.
+
+**Supported urls**
+
+- Single episode (with [episode filtering](#episode-filtering))
+ ```shell
+ $ crunchy-cli archive https://www.crunchyroll.com/watch/GRDQPM1ZY/alone-and-lonesome
+ ```
+- Series (with [episode filtering](#episode-filtering))
+ ```shell
+ $ crunchy-cli archive https://www.crunchyroll.com/series/GY8VEQ95Y/darling-in-the-franxx
+ ```
+
+**Options**
+
+- Audio languages
+
+ Set the audio language with the `-a` / `--audio` flag. Can be used multiple times.
+
+ ```shell
+ $ crunchy-cli archive -a ja-JP -a de-DE https://www.crunchyroll.com/series/GY8VEQ95Y/darling-in-the-franxx
+ ```
+
+ Default is your system locale (if not supported by Crunchyroll, `en-US` (American English) and `ja-JP` (Japanese) are used).
+
+- Subtitle languages
+
+ Besides the audio, you can specify the subtitle language by using the `-s` / `--subtitle` flag.
+
+ ```shell
+ $ crunchy-cli archive -s de-DE https://www.crunchyroll.com/series/GY8VEQ95Y/darling-in-the-franxx
+ ```
+
+ Default is `all` subtitles.
+
+- Output template
+
+ Define an output template by using the `-o` / `--output` flag.
+ _crunchy-cli_ exclusively uses the [`.mkv`](https://en.wikipedia.org/wiki/Matroska) container format, because of its ability to store multiple audio, video and subtitle tracks at once.
+
+ ```shell
+ $ crunchy-cli archive -o "{title}.mkv" https://www.crunchyroll.com/series/GY8VEQ95Y/darling-in-the-franxx
+ ```
+
+ Default is `{title}.mkv`. See the [Template Options section](#output-template-options) below for more options.
+
+- Output template for special episodes
+
+ Define an output template which only gets used when the episode is a special (episode number is 0 or has non-zero decimal places) by using the `--output-special` flag.
+ _crunchy-cli_ exclusively uses the [`.mkv`](https://en.wikipedia.org/wiki/Matroska) container format, because of its ability to store multiple audio, video and subtitle tracks at once.
+
+ ```shell
+ $ crunchy-cli archive --output-specials "Special EP - {title}" https://www.crunchyroll.com/watch/GY8D975JY/veldoras-journal
+ ```
+
+ Default is the template, set by the `-o` / `--output` flag. See the [Template Options section](#output-template-options) below for more options.
+
+- Universal output
+
+ The output template options can be forced to get sanitized via the `--universal-output` flag to be valid across all supported operating systems (Windows has a lot of characters which aren't allowed in filenames...).
+
+ ```shell
+ $ crunchy-cli archive --universal-output -o https://www.crunchyroll.com/watch/G7PU4XD48/tales-veldoras-journal-2
+ ```
+
+- Resolution
+
+ The resolution for videos can be set via the `-r` / `--resolution` flag.
+
+ ```shell
+ $ crunchy-cli archive -r worst https://www.crunchyroll.com/series/GY8VEQ95Y/darling-in-the-franxx
+ ```
+
+ Default is `best`.
+
+- Merge behavior
+
+ Due to censorship or additional intros, some episodes have multiple lengths for different languages.
+ In the best case, when multiple audio & subtitle tracks are used, there is only one *video* track and all other languages can be stored as audio-only.
+ But, as said, this is not always the case.
+ With the `-m` / `--merge` flag you can define the behaviour when an episodes' video tracks differ in length.
+ Valid options are `audio` - store one video and all other languages as audio only; `video` - store the video + audio for every language; `auto` - detect if videos differ in length: if so, behave like `video` - otherwise like `audio`; `sync` - detect if videos differ in length: if so, it tries to find the offset of matching audio parts and removes the offset from the beginning, otherwise it behaves like `audio`.
+ Subtitles will always match the primary audio and video.
+
+ ```shell
+ $ crunchy-cli archive -m audio https://www.crunchyroll.com/series/GY8VEQ95Y/darling-in-the-franxx
+ ```
+
+ Default is `auto`.
+
+- Merge time tolerance
+
+ Sometimes two video tracks are downloaded with `--merge` set to `auto` even if they only differ some milliseconds in length which shouldn't be noticeable to the viewer.
+ To prevent this, you can specify a range in milliseconds with the `--merge-time-tolerance` flag that only downloads one video if the length difference is in the given range.
+
+ ```shell
+ $ crunchy-cli archive -m auto --merge-time-tolerance 100 https://www.crunchyroll.com/series/GY8VEQ95Y/darling-in-the-franxx
+ ```
+
+ Default are `200` milliseconds.
+
+- Merge sync tolerance
+
+ Sometimes two video tracks are downloaded with `--merge` set to `sync` because the audio fingerprinting fails to identify matching audio parts (e.g. opening).
+ To prevent this, you can use the `--merge-sync-tolerance` flag to specify the difference by which two fingerprints are considered equal.
+
+ ```shell
+ $ crunchy-cli archive -m sync --merge-sync-tolerance 3 https://www.crunchyroll.com/watch/GRDQPM1ZY/alone-and-lonesome
+ ```
+
+ Default is `6`.
+
+- Merge sync precision
+
+ If you use `--merge` set to `sync` and the syncing seems to be not accurate enough or takes to long, you can use the `--sync-precision` flag to specify the amount of offset determination runs from which the final offset is calculated.
+
+ ```shell
+ $ crunchy-cli archive -m sync --merge-sync-precision 3 https://www.crunchyroll.com/watch/GRDQPM1ZY/alone-and-lonesome
+ ```
+
+ Default is `4`.
+
+- Language tagging
+
+ You can force the usage of a specific language tagging in the output file with the `--language-tagging` flag.
+ This might be useful as some video players doesn't recognize the language tagging Crunchyroll uses internally.
+
+ ```shell
+ $ crunchy-cli archive --language-tagging ietf https://www.crunchyroll.com/series/GY8VEQ95Y/darling-in-the-franxx
+ ```
+
+- FFmpeg Preset
+
+ You can specify specific built-in presets with the `--ffmpeg-preset` flag to convert videos to a specific coding while downloading.
+ Multiple predefined presets how videos should be encoded (h264, h265, av1, ...) are available, you can see them with `crunchy-cli archive --help`.
+ If you need more specific ffmpeg customizations you could either convert the output file manually or use ffmpeg output arguments as value for this flag.
+
+ ```shell
+ $ crunchy-cli archive --ffmpeg-preset av1-lossless https://www.crunchyroll.com/watch/GRDQPM1ZY/alone-and-lonesome
+ ```
+
+- FFmpeg threads
+
+ If you want to manually set how many threads FFmpeg should use, you can use the `--ffmpeg-threads` flag. This does not work with every codec/preset and is skipped entirely when specifying custom ffmpeg output arguments instead of a preset for `--ffmpeg-preset`.
+
+ ```shell
+ $ crunchy-cli archive --ffmpeg-threads 4 https://www.crunchyroll.com/watch/GRDQPM1ZY/alone-and-lonesome
+ ```
+
+- Default subtitle
+
+ `--default-subtitle` Set which subtitle language is to be flagged as **default** and **forced**.
+
+ ```shell
+ $ crunchy-cli archive --default-subtitle en-US https://www.crunchyroll.com/series/GY8VEQ95Y/darling-in-the-franxx
+ ```
+
+ Default is none.
+
+- Include fonts
+
+ You can include the fonts required by subtitles directly into the output file with the `--include-fonts` flag. This will use the embedded font for subtitles instead of the system font when playing the video in a video player which supports it.
+
+ ```shell
+ $ crunchy-cli archive --include-fonts https://www.crunchyroll.com/series/GY8VEQ95Y/darling-in-the-franxx
+ ```
+
+- Include chapters
+
+ Crunchyroll sometimes provide information about skippable events like the intro or credits.
+ These information can be stored as chapters in the resulting video file via the `--include-chapters` flag.
+ This flag only works if `--merge` is set to `audio` because chapters cannot be mapped to a specific video steam.
+
+ ```shell
+ $ crunchy-cli archive --include-chapters https://www.crunchyroll.com/watch/G0DUND0K2/the-journeys-end
+ ```
+
+- Skip existing
+
+ If you re-download a series but want to skip episodes you've already downloaded, the `--skip-existing` flag skips the already existing/downloaded files.
+
+ ```shell
+ $ crunchy-cli archive --skip-existing https://www.crunchyroll.com/series/GY8VEQ95Y/darling-in-the-franxx
+ ```
+
+- Skip existing method
+
+ By default, already existing files are determined by their name and the download of the corresponding episode is skipped.
+ But sometimes Crunchyroll adds dubs or subs to an already existing episode and these changes aren't recognized and `--skip-existing` just skips it.
+ This behavior can be changed by the `--skip-existing-method` flag. Valid options are `audio` and `subtitle` (if the file already exists but the audio/subtitle are less from what should be downloaded, the episode gets downloaded and the file overwritten).
+
+ ```shell
+ $ crunchy-cli archive --skip-existing-method audio --skip-existing-method video https://www.crunchyroll.com/series/GY8VEQ95Y/darling-in-the-franxx
+ ```
+
+- Skip specials
+
+ If you doesn't want to download special episodes, use the `--skip-specials` flag to skip the download of them.
+
+ ```shell
+ $ crunchy-cli archive --skip-specials https://www.crunchyroll.com/series/GYZJ43JMR/that-time-i-got-reincarnated-as-a-slime[S2]
+ ```
+
+- Yes
+
+ Sometimes different seasons have the same season number (e.g. Sword Art Online Alicization and Alicization War of Underworld are both marked as season 3), in such cases an interactive prompt is shown which needs user further user input to decide which season to download.
+ The `--yes` flag suppresses this interactive prompt and just downloads all seasons.
+
+ ```shell
+ $ crunchy-cli archive --yes https://www.crunchyroll.com/series/GR49G9VP6/sword-art-online
+ ```
+
+ If you've passed the `-q` / `--quiet` [global flag](#global-settings), this flag is automatically set.
+
+- Threads
+
+ To increase the download speed, video segments are downloaded simultaneously by creating multiple threads.
+ If you want to manually specify how many threads to use when downloading, do this with the `-t` / `--threads` flag.
+
+ ```shell
+ $ crunchy-cli archive -t 1 https://www.crunchyroll.com/series/GY8VEQ95Y/darling-in-the-franxx
+ ```
+
+ The default thread count is the count of cpu threads your pc has.
+
+### Search
+
+The `search` command is a powerful tool to query the Crunchyroll library.
+It behaves like the regular search on the website but is able to further process the results and return everything it can find, from the series title down to the raw stream url.
+_Using this command with the `--anonymous` flag or a non-premium account may return incomplete results._
+
+**Supported urls/input**
+
+- Single episode (with [episode filtering](#episode-filtering))
+ ```shell
+ $ crunchy-cli search https://www.crunchyroll.com/watch/GRDQPM1ZY/alone-and-lonesome
+ ```
+- Series (with [episode filtering](#episode-filtering))
+ ```shell
+ $ crunchy-cli search https://www.crunchyroll.com/series/GY8VEQ95Y/darling-in-the-franxx
+ ```
+- Search input
+ ```shell
+ $ crunchy-cli search "darling in the franxx"
+ ```
+
+**Options**
+
+- Audio
+
+ Set the audio language to search via the `--audio` flag. Can be used multiple times.
+
+ ```shell
+ $ crunchy-cli search --audio en-US https://www.crunchyroll.com/series/GY8VEQ95Y/darling-in-the-franxx
+ ```
+
+ Default is your system locale.
+
+- Result limit
+
+ If your input is a search term instead of an url, you have multiple options to control which results to process.
+ The `--search-top-results-limit` flag sets the limit of top search results to process.
+ `--search-series-limit` sets the limit of only series, `--search-movie-listing-limit` of only movie listings, `--search-episode-limit` of only episodes and `--search-music-limit` of only concerts and music videos.
+
+ ```shell
+ $ crunchy-cli search --search-top-results-limit 10 "darling in the franxx"
+ # only return series which have 'darling' in it. do not return top results which might also be non-series items
+ $ crunchy-cli search --search-top-results-limit 0 --search-series-limit 10 "darling"
+ # this returns 2 top results, 3 movie listings, 5 episodes and 1 music item as result
+ $ crunchy-cli search --search-top-results-limit 2 --search-movie-listing-limit 3 --search-episode-limit 5 --search-music-limit 1 "test"
+ ```
+
+ Default is `5` for `--search-top-results-limit`, `0` for all others.
+
+- Output template
+
+ The search command is designed to show only the specific information you want.
+ This is done with the `-o`/`--output` flag.
+ You can specify keywords in a specific pattern, and they will get replaced in the output text.
+ The required pattern for this begins with `{{`, then the keyword, and closes with `}}` (e.g. `{{episode.title}}`).
+ For example, if you want to get the title of an episode, you can use `Title: {{episode.title}}` and `{{episode.title}}` will be replaced with the episode title.
+ You can see all supported keywords with `crunchy-cli search --help`.
+
+ ```shell
+ $ crunchy-cli search -o "{{series.title}}" https://www.crunchyroll.com/series/GY8VEQ95Y/darling-in-the-franxx
+ ```
+
+ Default is `S{{season.number}}E{{episode.number}} - {{episode.title}}`.
+
+---
+
+#### Output Template Options
+
+You can use various template options to change how the filename is processed. The following tags are available:
+
+- `{title}` → Title of the video
+- `{series_name}` → Name of the series
+- `{season_name}` → Name of the season
+- `{audio}` → Audio language of the video
+- `{width}` → Width of the video
+- `{height}` → Height of the video
+- `{season_number}` → Number of the season
+- `{episode_number}` → Number of the episode
+- `{relative_episode_number}` → Number of the episode relative to its season
+- `{sequence_number}` → Like `{episode_number}` but without possible non-number characters
+- `{relative_sequence_number}` → Like `{relative_episode_number}` but with support for episode 0's and .5's
+- `{release_year}` → Release year of the video
+- `{release_month}` → Release month of the video
+- `{release_day} ` → Release day of the video
+- `{series_id}` → ID of the series
+- `{season_id}` → ID of the season
+- `{episode_id}` → ID of the episode
+
+Example:
-Archive a file
```shell
-$ crunchy archive https://www.crunchyroll.com/darling-in-the-franxx/darling-in-the-franxx/episode-1-alone-and-lonesome-759575
+$ crunchy-cli archive -o "[S{season_number}E{episode_number}] {title}.mkv" https://www.crunchyroll.com/series/G8DHV7W21/dragon-ball
+# Output file: '[S01E01] Secret of the Dragon Ball.mkv'
```
-Downloads the first two episode of Darling in the FranXX and stores it compressed in a file.
-```shell
-$ crunchy archive -c "ditf.tar.gz" https://www.crunchyroll.com/darling-in-the-franxx/darling-in-the-franxx
+#### Episode filtering
+
+Filters patterns can be used to download a specific range of episodes from a single series.
+
+A filter pattern may consist of either a season, an episode, or a combination of the two.
+When used in combination, seasons `S` must be defined before episodes `E`.
+
+There are many possible patterns, for example:
+
+- `...[E5]` - Download the fifth episode.
+- `...[S1]` - Download the whole first season.
+- `...[-S2]` - Download the first two seasons.
+- `...[S3E4-]` - Download everything from season three, episode four, onwards.
+- `...[S1E4-S3]` - Download season one, starting at episode four, then download season two and three.
+- `...[S3,S5]` - Download season three and five.
+- `...[S1-S3,S4E2-S4E6]` - Download season one to three, then episodes two to six from season four.
+
+In practice, it would look like this:
+
+```
+https://www.crunchyroll.com/series/GY8VEQ95Y/darling-in-the-franxx[E1-E5]
```
-##### Flags
+# 📜 Disclaimer
-The following flags can be (optional) passed to modify the [archive](#archive) process.
+This tool is meant for private use only.
+You need a [Crunchyroll Premium](https://www.crunchyroll.com/welcome#plans) subscription to access premium content.
-| Short | Extended | Description |
-|-------|----------------|---------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------|
-| `-l` | `--language` | Audio locale which should be downloaded. Can be used multiple times. |
-| `-d` | `--directory` | Directory to download the video(s) to. |
-| `-o` | `--output` | Name of the output file. |
-| `-m` | `--merge` | Sets the behavior of the stream merging. Valid behaviors are 'auto', 'audio', 'video'. See the [wiki](https://github.com/ByteDream/crunchyroll-go/wiki/Cli#archive) for more information. |
-| `-c` | `--compress` | If is set, all output will be compresses into an archive. This flag sets the name of the compressed output file and the file ending specifies the compression algorithm (gzip, tar, zip are supported). |
-| `-r` | `--resolution` | The resolution of the video(s). `best` for best resolution, `worst` for worst. |
-| `-g` | `--goroutines` | Sets how many parallel segment downloads should be used. |
-
-### Help
-
-- General help
- ```shell
- $ crunchy help
- ```
-- Login help
- ```shell
- $ crunchy help login
- ```
-- Download help
- ```shell
- $ crunchy help download
- ```
-
-- Archive help
- ```shell
- $ crunchy help archive
- ```
-
-### Global flags
-
-These flags you can use across every sub-command
-
-| Flag | Description |
-|------|------------------------------------------------------|
-| `-q` | Disables all output. |
-| `-v` | Shows additional debug output. |
-| `-p` | Use a proxy to hide your ip / redirect your traffic. |
-
-# 📚 Library
-
-Download the library via `go get`
-```shell
-$ go get github.com/ByteDream/crunchyroll-go
-```
-
-The documentation is available on [pkg.go.dev](https://pkg.go.dev/github.com/ByteDream/crunchyroll-go).
-
-Examples how to use the library and some features of it are described in the [wiki](https://github.com/ByteDream/crunchyroll-go/wiki/Library).
-
-# ☝️ Disclaimer
-
-This tool is **ONLY** meant to be used for private purposes.
-To use this tool you need crunchyroll premium anyway, so there is no reason why rip and share the episodes.
-
-**The responsibility for what happens to the downloaded videos lies entirely with the user who downloaded them.**
+**You are entirely responsible for what happens when you use crunchy-cli.**
# ⚖ License
-This project is licensed under the GNU Lesser General Public License v3.0 (LGPL-3.0) - see the [LICENSE](LICENSE) file for more details.
+This project is licensed under the MIT License - see the [LICENSE](LICENSE) file for more details.
diff --git a/build.rs b/build.rs
new file mode 100644
index 0000000..313cb6d
--- /dev/null
+++ b/build.rs
@@ -0,0 +1,122 @@
+use clap::{Command, CommandFactory};
+use clap_complete::shells;
+use std::path::{Path, PathBuf};
+
+fn main() -> std::io::Result<()> {
+ let rustls_tls = cfg!(feature = "rustls-tls");
+ let native_tls = cfg!(feature = "native-tls");
+ let openssl_tls = cfg!(any(feature = "openssl-tls", feature = "openssl-tls-static"));
+
+ if rustls_tls as u8 + native_tls as u8 + openssl_tls as u8 > 1 {
+ let active_tls_backend = if openssl_tls {
+ "openssl"
+ } else if native_tls {
+ "native tls"
+ } else {
+ "rustls"
+ };
+
+ println!("cargo:warning=Multiple tls backends are activated (through the '*-tls' features). Consider to activate only one as it is not possible to change the backend during runtime. The active backend for this build will be '{}'.", active_tls_backend)
+ }
+
+ // note that we're using an anti-pattern here / violate the rust conventions. build script are
+ // not supposed to write outside of 'OUT_DIR'. to have the generated files in the build "root"
+ // (the same directory where the output binary lives) is much simpler than in 'OUT_DIR' since
+ // its nested in sub directories and is difficult to find (at least more difficult than in the
+ // build root)
+ let unconventional_out_dir =
+ std::path::PathBuf::from(std::env::var_os("OUT_DIR").ok_or(std::io::ErrorKind::NotFound)?)
+ .parent()
+ .unwrap()
+ .parent()
+ .unwrap()
+ .parent()
+ .unwrap()
+ .to_path_buf();
+
+ let completions_dir = exist_or_create_dir(unconventional_out_dir.join("completions"))?;
+ let manpage_dir = exist_or_create_dir(unconventional_out_dir.join("manpages"))?;
+
+ generate_completions(completions_dir)?;
+ generate_manpages(manpage_dir)?;
+
+ Ok(())
+}
+
+fn exist_or_create_dir(path: PathBuf) -> std::io::Result {
+ if !path.exists() {
+ std::fs::create_dir(path.clone())?
+ }
+ Ok(path)
+}
+
+fn generate_completions(out_dir: PathBuf) -> std::io::Result<()> {
+ let mut command: Command = crunchy_cli_core::Cli::command();
+
+ clap_complete::generate_to(
+ shells::Bash,
+ &mut command.clone(),
+ "crunchy-cli",
+ out_dir.clone(),
+ )?;
+ clap_complete::generate_to(
+ shells::Elvish,
+ &mut command.clone(),
+ "crunchy-cli",
+ out_dir.clone(),
+ )?;
+ println!(
+ "{}",
+ clap_complete::generate_to(
+ shells::Fish,
+ &mut command.clone(),
+ "crunchy-cli",
+ out_dir.clone(),
+ )?
+ .to_string_lossy()
+ );
+ clap_complete::generate_to(
+ shells::PowerShell,
+ &mut command.clone(),
+ "crunchy-cli",
+ out_dir.clone(),
+ )?;
+ clap_complete::generate_to(shells::Zsh, &mut command, "crunchy-cli", out_dir)?;
+
+ Ok(())
+}
+
+fn generate_manpages(out_dir: PathBuf) -> std::io::Result<()> {
+ fn generate_command_manpage(
+ mut command: Command,
+ base_path: &Path,
+ sub_name: &str,
+ ) -> std::io::Result<()> {
+ let (file_name, title) = if sub_name.is_empty() {
+ command = command.name("crunchy-cli");
+ ("crunchy-cli.1".to_string(), "crunchy-cli".to_string())
+ } else {
+ command = command.name(format!("crunchy-cli {}", sub_name));
+ (
+ format!("crunchy-cli-{}.1", sub_name),
+ format!("crunchy-cli-{}", sub_name),
+ )
+ };
+
+ let mut command_buf = vec![];
+ let man = clap_mangen::Man::new(command)
+ .title(title)
+ .date(chrono::Utc::now().format("%b %d, %Y").to_string());
+ man.render(&mut command_buf)?;
+
+ std::fs::write(base_path.join(file_name), command_buf)
+ }
+
+ generate_command_manpage(crunchy_cli_core::Cli::command(), &out_dir, "")?;
+ generate_command_manpage(crunchy_cli_core::Archive::command(), &out_dir, "archive")?;
+ generate_command_manpage(crunchy_cli_core::Download::command(), &out_dir, "download")?;
+ generate_command_manpage(crunchy_cli_core::Login::command(), &out_dir, "login")?;
+ generate_command_manpage(crunchy_cli_core::Search::command(), &out_dir, "search")?;
+
+ Ok(())
+}
diff --git a/cmd/crunchyroll-go/cmd/archive.go b/cmd/crunchyroll-go/cmd/archive.go
deleted file mode 100644
index a36f806..0000000
--- a/cmd/crunchyroll-go/cmd/archive.go
+++ /dev/null
@@ -1,805 +0,0 @@
-package cmd
-
-import (
- "archive/tar"
- "archive/zip"
- "bufio"
- "bytes"
- "compress/gzip"
- "context"
- "fmt"
- "github.com/ByteDream/crunchyroll-go"
- "github.com/ByteDream/crunchyroll-go/utils"
- "github.com/grafov/m3u8"
- "github.com/spf13/cobra"
- "io"
- "os"
- "os/exec"
- "os/signal"
- "path/filepath"
- "regexp"
- "runtime"
- "sort"
- "strconv"
- "strings"
- "sync"
- "time"
-)
-
-var (
- archiveLanguagesFlag []string
-
- archiveDirectoryFlag string
- archiveOutputFlag string
-
- archiveMergeFlag string
-
- archiveCompressFlag string
-
- archiveResolutionFlag string
-
- archiveGoroutinesFlag int
-)
-
-var archiveCmd = &cobra.Command{
- Use: "archive",
- Short: "Stores the given videos with all subtitles and multiple audios in a .mkv file",
- Args: cobra.MinimumNArgs(1),
-
- PreRunE: func(cmd *cobra.Command, args []string) error {
- out.Debug("Validating arguments")
-
- if !hasFFmpeg() {
- return fmt.Errorf("ffmpeg is needed to run this command correctly")
- }
- out.Debug("FFmpeg detected")
-
- if filepath.Ext(archiveOutputFlag) != ".mkv" {
- return fmt.Errorf("currently only matroska / .mkv files are supported")
- }
-
- for _, locale := range archiveLanguagesFlag {
- if !utils.ValidateLocale(crunchyroll.LOCALE(locale)) {
- // if locale is 'all', match all known locales
- if locale == "all" {
- archiveLanguagesFlag = allLocalesAsStrings()
- break
- }
- return fmt.Errorf("%s is not a valid locale. Choose from: %s", locale, strings.Join(allLocalesAsStrings(), ", "))
- }
- }
- out.Debug("Using following audio locales: %s", strings.Join(archiveLanguagesFlag, ", "))
-
- var found bool
- for _, mode := range []string{"auto", "audio", "video"} {
- if archiveMergeFlag == mode {
- out.Debug("Using %s merge behavior", archiveMergeFlag)
- found = true
- break
- }
- }
- if !found {
- return fmt.Errorf("'%s' is no valid merge flag. Use 'auto', 'audio' or 'video'", archiveMergeFlag)
- }
-
- if archiveCompressFlag != "" {
- found = false
- for _, algo := range []string{".tar", ".tar.gz", ".tgz", ".zip"} {
- if strings.HasSuffix(archiveCompressFlag, algo) {
- out.Debug("Using %s compression", algo)
- found = true
- break
- }
- }
- if !found {
- return fmt.Errorf("'%s' is no valid compress algorithm. Valid algorithms / file endings are '.tar', '.tar.gz', '.zip'",
- archiveCompressFlag)
- }
- }
-
- switch archiveResolutionFlag {
- case "1080p", "720p", "480p", "360p", "240p":
- intRes, _ := strconv.ParseFloat(strings.TrimSuffix(archiveResolutionFlag, "p"), 84)
- archiveResolutionFlag = fmt.Sprintf("%dx%s", int(intRes*(16/9)), strings.TrimSuffix(archiveResolutionFlag, "p"))
- case "1920x1080", "1280x720", "640x480", "480x360", "428x240", "best", "worst":
- default:
- return fmt.Errorf("'%s' is not a valid resolution", archiveResolutionFlag)
- }
- out.Debug("Using resolution '%s'", archiveResolutionFlag)
-
- return nil
- },
- RunE: func(cmd *cobra.Command, args []string) error {
- loadCrunchy()
-
- return archive(args)
- },
-}
-
-func init() {
- archiveCmd.Flags().StringSliceVarP(&archiveLanguagesFlag,
- "language",
- "l",
- []string{string(systemLocale(false)), string(crunchyroll.JP)},
- "Audio locale which should be downloaded. Can be used multiple times")
-
- cwd, _ := os.Getwd()
- archiveCmd.Flags().StringVarP(&archiveDirectoryFlag,
- "directory",
- "d",
- cwd,
- "The directory to store the files into")
- archiveCmd.Flags().StringVarP(&archiveOutputFlag,
- "output",
- "o",
- "{title}.mkv",
- "Name of the output file. If you use the following things in the name, the will get replaced:\n"+
- "\t{title} » Title of the video\n"+
- "\t{series_name} » Name of the series\n"+
- "\t{season_name} » Name of the season\n"+
- "\t{season_number} » Number of the season\n"+
- "\t{episode_number} » Number of the episode\n"+
- "\t{resolution} » Resolution of the video\n"+
- "\t{fps} » Frame Rate of the video\n"+
- "\t{audio} » Audio locale of the video\n"+
- "\t{subtitle} » Subtitle locale of the video")
-
- archiveCmd.Flags().StringVarP(&archiveMergeFlag,
- "merge",
- "m",
- "auto",
- "Sets the behavior of the stream merging. Valid behaviors are 'auto', 'audio', 'video'")
-
- archiveCmd.Flags().StringVarP(&archiveCompressFlag,
- "compress",
- "c",
- "",
- "If is set, all output will be compresses into an archive (every url generates a new one). "+
- "This flag sets the name of the compressed output file. The file ending specifies the compression algorithm. "+
- "The following algorithms are supported: gzip, tar, zip")
-
- archiveCmd.Flags().StringVarP(&archiveResolutionFlag,
- "resolution",
- "r",
- "best",
- "The video resolution. Can either be specified via the pixels, the abbreviation for pixels, or 'common-use' words\n"+
- "\tAvailable pixels: 1920x1080, 1280x720, 640x480, 480x360, 428x240\n"+
- "\tAvailable abbreviations: 1080p, 720p, 480p, 360p, 240p\n"+
- "\tAvailable common-use words: best (best available resolution), worst (worst available resolution)")
-
- archiveCmd.Flags().IntVarP(&archiveGoroutinesFlag,
- "goroutines",
- "g",
- runtime.NumCPU(),
- "Number of parallel segment downloads")
-
- rootCmd.AddCommand(archiveCmd)
-}
-
-func archive(urls []string) error {
- for i, url := range urls {
- out.SetProgress("Parsing url %d", i+1)
- episodes, err := archiveExtractEpisodes(url)
- if err != nil {
- out.StopProgress("Failed to parse url %d", i+1)
- return err
- }
- out.StopProgress("Parsed url %d", i+1)
-
- var compressFile *os.File
- var c compress
-
- if archiveCompressFlag != "" {
- compressFile, err = os.Create(generateFilename(archiveCompressFlag, ""))
- if err != nil {
- return fmt.Errorf("failed to create archive file: %v", err)
- }
- if strings.HasSuffix(archiveCompressFlag, ".tar") {
- c = newTarCompress(compressFile)
- } else if strings.HasSuffix(archiveCompressFlag, ".tar.gz") || strings.HasSuffix(archiveCompressFlag, ".tgz") {
- c = newGzipCompress(compressFile)
- } else if strings.HasSuffix(archiveCompressFlag, ".zip") {
- c = newZipCompress(compressFile)
- }
- }
-
- for _, season := range episodes {
- out.Info("%s Season %d", season[0].SeriesName, season[0].SeasonNumber)
-
- for j, info := range season {
- out.Info("\t%d. %s » %spx, %.2f FPS (S%02dE%02d)",
- j+1,
- info.Title,
- info.Resolution,
- info.FPS,
- info.SeasonNumber,
- info.EpisodeNumber)
- }
- }
- out.Empty()
-
- for _, season := range episodes {
- for _, info := range season {
- var filename string
- var writeCloser io.WriteCloser
- if c != nil {
- filename = info.Format(archiveOutputFlag)
- writeCloser, err = c.NewFile(info)
- if err != nil {
- return fmt.Errorf("failed to pre generate new archive file: %v", err)
- }
- } else {
- dir := info.Format(downloadDirectoryFlag)
- if _, err = os.Stat(dir); os.IsNotExist(err) {
- if err = os.MkdirAll(dir, 0777); err != nil {
- return fmt.Errorf("error while creating directory: %v", err)
- }
- }
- filename = generateFilename(info.Format(archiveOutputFlag), dir)
- writeCloser, err = os.Create(filename)
- if err != nil {
- return fmt.Errorf("failed to create new file: %v", err)
- }
- }
-
- if err = archiveInfo(info, writeCloser, filename); err != nil {
- writeCloser.Close()
- if f, ok := writeCloser.(*os.File); ok {
- os.Remove(f.Name())
- } else {
- c.Close()
- compressFile.Close()
- os.RemoveAll(compressFile.Name())
- }
- return err
- }
-
- writeCloser.Close()
- }
- }
- if c != nil {
- c.Close()
- }
- if compressFile != nil {
- compressFile.Close()
- }
- }
- return nil
-}
-
-func archiveInfo(info formatInformation, writeCloser io.WriteCloser, filename string) error {
- out.Debug("Entering season %d, episode %d with %d additional formats", info.SeasonNumber, info.EpisodeNumber, len(info.additionalFormats))
-
- downloadProgress, err := createArchiveProgress(info)
- if err != nil {
- return fmt.Errorf("error while setting up downloader: %v", err)
- }
- defer func() {
- if downloadProgress.Total != downloadProgress.Current {
- fmt.Println()
- }
- }()
-
- rootFile, err := os.CreateTemp("", fmt.Sprintf("%s_*.ts", strings.TrimSuffix(filepath.Base(filename), filepath.Ext(filename))))
- if err != nil {
- return fmt.Errorf("failed to create temp file: %v", err)
- }
- defer os.Remove(rootFile.Name())
- defer rootFile.Close()
-
- ctx, cancel := context.WithCancel(context.Background())
- defer cancel()
- downloader := crunchyroll.NewDownloader(ctx, rootFile, downloadGoroutinesFlag, func(segment *m3u8.MediaSegment, current, total int, file *os.File) error {
- // check if the context was cancelled.
- // must be done in to not print any progress messages if ctrl+c was pressed
- if ctx.Err() != nil {
- return nil
- }
-
- if out.IsDev() {
- downloadProgress.UpdateMessage(fmt.Sprintf("Downloading %d/%d (%.2f%%) » %s", current, total, float32(current)/float32(total)*100, segment.URI), false)
- } else {
- downloadProgress.Update()
- }
-
- if current == total {
- downloadProgress.UpdateMessage("Merging segments", false)
- }
- return nil
- })
-
- sig := make(chan os.Signal, 1)
- signal.Notify(sig, os.Interrupt)
- go func() {
- select {
- case <-sig:
- signal.Stop(sig)
- out.Exit("Exiting... (may take a few seconds)")
- out.Exit("To force exit press ctrl+c (again)")
- cancel()
- // os.Exit(1) is not called since an immediate exit after the cancel function does not let
- // the download process enough time to stop gratefully. A result of this is that the temporary
- // directory where the segments are downloaded to will not be deleted
- case <-ctx.Done():
- // this is just here to end the goroutine and prevent it from running forever without a reason
- }
- }()
- out.Debug("Set up signal catcher")
-
- var additionalDownloaderOpts []string
- var mergeMessage string
- switch archiveMergeFlag {
- case "auto":
- additionalDownloaderOpts = []string{"-vn"}
- for _, format := range info.additionalFormats {
- if format.Video.Bandwidth != info.format.Video.Bandwidth {
- // revoke the changed FFmpegOpts above
- additionalDownloaderOpts = []string{}
- break
- }
- }
- if len(additionalDownloaderOpts) > 0 {
- mergeMessage = "merging audio for additional formats"
- } else {
- mergeMessage = "merging video for additional formats"
- }
- case "audio":
- additionalDownloaderOpts = []string{"-vn"}
- mergeMessage = "merging audio for additional formats"
- case "video":
- mergeMessage = "merging video for additional formats"
- }
-
- out.Info("Downloading episode `%s` to `%s` (%s)", info.Title, filepath.Base(filename), mergeMessage)
- out.Info("\tEpisode: S%02dE%02d", info.SeasonNumber, info.EpisodeNumber)
- out.Info("\tAudio: %s", info.Audio)
- out.Info("\tSubtitle: %s", info.Subtitle)
- out.Info("\tResolution: %spx", info.Resolution)
- out.Info("\tFPS: %.2f", info.FPS)
-
- var videoFiles, audioFiles, subtitleFiles []string
- defer func() {
- for _, f := range append(append(videoFiles, audioFiles...), subtitleFiles...) {
- os.RemoveAll(f)
- }
- }()
-
- var f []string
- if f, err = archiveDownloadVideos(downloader, filepath.Base(filename), true, info.format); err != nil {
- if err != ctx.Err() {
- return fmt.Errorf("error while downloading: %v", err)
- }
- return err
- }
- videoFiles = append(videoFiles, f[0])
-
- if len(additionalDownloaderOpts) == 0 {
- var videos []string
- downloader.FFmpegOpts = additionalDownloaderOpts
- if videos, err = archiveDownloadVideos(downloader, filepath.Base(filename), true, info.additionalFormats...); err != nil {
- return fmt.Errorf("error while downloading additional videos: %v", err)
- }
- downloader.FFmpegOpts = []string{}
- videoFiles = append(videoFiles, videos...)
- } else {
- var audios []string
- if audios, err = archiveDownloadVideos(downloader, filepath.Base(filename), false, info.additionalFormats...); err != nil {
- return fmt.Errorf("error while downloading additional videos: %v", err)
- }
- audioFiles = append(audioFiles, audios...)
- }
-
- sort.Sort(utils.SubtitlesByLocale(info.format.Subtitles))
-
- sortSubtitles, _ := strconv.ParseBool(os.Getenv("SORT_SUBTITLES"))
- if sortSubtitles && len(archiveLanguagesFlag) > 0 {
- // this sort the subtitle locales after the languages which were specified
- // with the `archiveLanguagesFlag` flag
- for _, language := range archiveLanguagesFlag {
- for i, subtitle := range info.format.Subtitles {
- if subtitle.Locale == crunchyroll.LOCALE(language) {
- info.format.Subtitles = append([]*crunchyroll.Subtitle{subtitle}, append(info.format.Subtitles[:i], info.format.Subtitles[i+1:]...)...)
- break
- }
- }
- }
- }
-
- var subtitles []string
- if subtitles, err = archiveDownloadSubtitles(filepath.Base(filename), info.format.Subtitles...); err != nil {
- return fmt.Errorf("error while downloading subtitles: %v", err)
- }
- subtitleFiles = append(subtitleFiles, subtitles...)
-
- if err = archiveFFmpeg(ctx, writeCloser, videoFiles, audioFiles, subtitleFiles); err != nil {
- return fmt.Errorf("failed to merge files: %v", err)
- }
-
- downloadProgress.UpdateMessage("Download finished", false)
-
- signal.Stop(sig)
- out.Debug("Stopped signal catcher")
-
- out.Empty()
- out.Empty()
-
- return nil
-}
-
-func createArchiveProgress(info formatInformation) (*downloadProgress, error) {
- var progressCount int
- if err := info.format.InitVideo(); err != nil {
- return nil, fmt.Errorf("error while initializing a video: %v", err)
- }
- // + number of segments a video has +1 is for merging
- progressCount += int(info.format.Video.Chunklist.Count()) + 1
- for _, f := range info.additionalFormats {
- if f == info.format {
- continue
- }
-
- if err := f.InitVideo(); err != nil {
- return nil, err
- }
- // + number of segments a video has +1 is for merging
- progressCount += int(f.Video.Chunklist.Count()) + 1
- }
-
- downloadProgress := &downloadProgress{
- Prefix: out.InfoLog.Prefix(),
- Message: "Downloading video",
- // number of segments a video +1 is for the success message
- Total: progressCount + 1,
- Dev: out.IsDev(),
- Quiet: out.IsQuiet(),
- }
- if out.IsDev() {
- downloadProgress.Prefix = out.DebugLog.Prefix()
- }
-
- return downloadProgress, nil
-}
-
-func archiveDownloadVideos(downloader crunchyroll.Downloader, filename string, video bool, formats ...*crunchyroll.Format) ([]string, error) {
- var files []string
-
- for _, format := range formats {
- var name string
- if video {
- name = fmt.Sprintf("%s_%s_video_*.ts", filename, format.AudioLocale)
- } else {
- name = fmt.Sprintf("%s_%s_audio_*.aac", filename, format.AudioLocale)
- }
-
- f, err := os.CreateTemp("", name)
- if err != nil {
- return nil, err
- }
- files = append(files, f.Name())
-
- downloader.Writer = f
- if err = format.Download(downloader); err != nil {
- f.Close()
- for _, file := range files {
- os.Remove(file)
- }
- return nil, err
- }
- f.Close()
-
- out.Debug("Downloaded '%s' video", format.AudioLocale)
- }
-
- return files, nil
-}
-
-func archiveDownloadSubtitles(filename string, subtitles ...*crunchyroll.Subtitle) ([]string, error) {
- var files []string
-
- for _, subtitle := range subtitles {
- f, err := os.CreateTemp("", fmt.Sprintf("%s_%s_subtitle_*.ass", filename, subtitle.Locale))
- if err != nil {
- return nil, err
- }
- files = append(files, f.Name())
-
- if err := subtitle.Save(f); err != nil {
- f.Close()
- for _, file := range files {
- os.Remove(file)
- }
- return nil, err
- }
- f.Close()
-
- out.Debug("Downloaded '%s' subtitles", subtitle.Locale)
- }
-
- return files, nil
-}
-
-func archiveFFmpeg(ctx context.Context, dst io.Writer, videoFiles, audioFiles, subtitleFiles []string) error {
- var input, maps, metadata []string
- re := regexp.MustCompile(`(?m)_([a-z]{2}-([A-Z]{2}|[0-9]{3}))_(video|audio|subtitle)`)
-
- for i, video := range videoFiles {
- input = append(input, "-i", video)
- maps = append(maps, "-map", strconv.Itoa(i))
- locale := crunchyroll.LOCALE(re.FindStringSubmatch(video)[1])
- metadata = append(metadata, fmt.Sprintf("-metadata:s:v:%d", i), fmt.Sprintf("language=%s", utils.LocaleLanguage(locale)))
- metadata = append(metadata, fmt.Sprintf("-metadata:s:a:%d", i), fmt.Sprintf("language=%s", utils.LocaleLanguage(locale)))
- }
-
- for i, audio := range audioFiles {
- input = append(input, "-i", audio)
- maps = append(maps, "-map", strconv.Itoa(i+len(videoFiles)))
- locale := crunchyroll.LOCALE(re.FindStringSubmatch(audio)[1])
- metadata = append(metadata, fmt.Sprintf("-metadata:s:a:%d", i), fmt.Sprintf("language=%s", utils.LocaleLanguage(locale)))
- }
-
- for i, subtitle := range subtitleFiles {
- input = append(input, "-i", subtitle)
- maps = append(maps, "-map", strconv.Itoa(i+len(videoFiles)+len(audioFiles)))
- locale := crunchyroll.LOCALE(re.FindStringSubmatch(subtitle)[1])
- metadata = append(metadata, fmt.Sprintf("-metadata:s:s:%d", i), fmt.Sprintf("title=%s", utils.LocaleLanguage(locale)))
- }
-
- commandOptions := []string{"-y"}
- commandOptions = append(commandOptions, input...)
- commandOptions = append(commandOptions, maps...)
- commandOptions = append(commandOptions, metadata...)
- // we have to create a temporary file here because it must be seekable
- // for ffmpeg.
- // ffmpeg could write to dst too, but this would require to re-encode
- // the audio which results in much higher time and resource consumption
- // (0-1 second with the temp file, ~20 seconds with re-encoding on my system)
- file, err := os.CreateTemp("", "")
- if err != nil {
- return err
- }
- file.Close()
- defer os.Remove(file.Name())
-
- commandOptions = append(commandOptions, "-c", "copy", "-f", "matroska", file.Name())
-
- // just a little nicer debug output to copy and paste the ffmpeg for debug reasons
- if out.IsDev() {
- var debugOptions []string
-
- for _, option := range commandOptions {
- if strings.HasPrefix(option, "title=") {
- debugOptions = append(debugOptions, "title=\""+strings.TrimPrefix(option, "title=")+"\"")
- } else if strings.HasPrefix(option, "language=") {
- debugOptions = append(debugOptions, "language=\""+strings.TrimPrefix(option, "language=")+"\"")
- } else if strings.Contains(option, " ") {
- debugOptions = append(debugOptions, "\""+option+"\"")
- } else {
- debugOptions = append(debugOptions, option)
- }
- }
- out.Debug("FFmpeg merge command: ffmpeg %s", strings.Join(debugOptions, " "))
- }
-
- var errBuf bytes.Buffer
- cmd := exec.CommandContext(ctx, "ffmpeg", commandOptions...)
- cmd.Stderr = &errBuf
- if err = cmd.Run(); err != nil {
- return fmt.Errorf(errBuf.String())
- }
-
- file, err = os.Open(file.Name())
- if err != nil {
- return err
- }
- defer file.Close()
-
- _, err = bufio.NewWriter(dst).ReadFrom(file)
- return err
-}
-
-func archiveExtractEpisodes(url string) ([][]formatInformation, error) {
- var hasJapanese bool
- languagesAsLocale := []crunchyroll.LOCALE{crunchyroll.JP}
- for _, language := range archiveLanguagesFlag {
- locale := crunchyroll.LOCALE(language)
- if locale == crunchyroll.JP {
- hasJapanese = true
- } else {
- languagesAsLocale = append(languagesAsLocale, locale)
- }
- }
-
- episodes, err := extractEpisodes(url, languagesAsLocale...)
- if err != nil {
- return nil, err
- }
-
- if !hasJapanese && len(episodes[1:]) == 0 {
- return nil, fmt.Errorf("no episodes found")
- }
-
- for i, eps := range episodes {
- if len(eps) == 0 {
- out.SetProgress("%s has no matching episodes", languagesAsLocale[i])
- } else if len(episodes[0]) > len(eps) {
- out.SetProgress("%s has %d less episodes than existing in japanese (%d)", languagesAsLocale[i], len(episodes[0])-len(eps), len(episodes[0]))
- }
- }
-
- if !hasJapanese {
- episodes = episodes[1:]
- }
-
- eps := make(map[int]map[int]*formatInformation)
- for _, lang := range episodes {
- for _, season := range utils.SortEpisodesBySeason(lang) {
- if _, ok := eps[season[0].SeasonNumber]; !ok {
- eps[season[0].SeasonNumber] = map[int]*formatInformation{}
- }
- for _, episode := range season {
- format, err := episode.GetFormat(archiveResolutionFlag, "", false)
- if err != nil {
- return nil, fmt.Errorf("error while receiving format for %s: %v", episode.Title, err)
- }
-
- if _, ok := eps[episode.SeasonNumber][episode.EpisodeNumber]; !ok {
- eps[episode.SeasonNumber][episode.EpisodeNumber] = &formatInformation{
- format: format,
- additionalFormats: make([]*crunchyroll.Format, 0),
-
- Title: episode.Title,
- SeriesName: episode.SeriesTitle,
- SeasonName: episode.SeasonTitle,
- SeasonNumber: episode.SeasonNumber,
- EpisodeNumber: episode.EpisodeNumber,
- Resolution: format.Video.Resolution,
- FPS: format.Video.FrameRate,
- Audio: format.AudioLocale,
- }
- } else {
- eps[episode.SeasonNumber][episode.EpisodeNumber].additionalFormats = append(eps[episode.SeasonNumber][episode.EpisodeNumber].additionalFormats, format)
- }
- }
- }
- }
-
- var infoFormat [][]formatInformation
- for _, e := range eps {
- var tmpFormatInfo []formatInformation
-
- var keys []int
- for episodeNumber := range e {
- keys = append(keys, episodeNumber)
- }
- sort.Ints(keys)
-
- for _, key := range keys {
- tmpFormatInfo = append(tmpFormatInfo, *e[key])
- }
-
- infoFormat = append(infoFormat, tmpFormatInfo)
- }
-
- return infoFormat, nil
-}
-
-type compress interface {
- io.Closer
-
- NewFile(information formatInformation) (io.WriteCloser, error)
-}
-
-func newGzipCompress(file *os.File) *tarCompress {
- gw := gzip.NewWriter(file)
- return &tarCompress{
- parent: gw,
- dst: tar.NewWriter(gw),
- }
-}
-
-func newTarCompress(file *os.File) *tarCompress {
- return &tarCompress{
- dst: tar.NewWriter(file),
- }
-}
-
-type tarCompress struct {
- compress
-
- wg sync.WaitGroup
-
- parent *gzip.Writer
- dst *tar.Writer
-}
-
-func (tc *tarCompress) Close() error {
- // we have to wait here in case the actual content isn't copied completely into the
- // writer yet
- tc.wg.Wait()
-
- var err, err2 error
- if tc.parent != nil {
- err2 = tc.parent.Close()
- }
- err = tc.dst.Close()
-
- if err != nil && err2 != nil {
- // best way to show double errors at once that i've found
- return fmt.Errorf("%v\n%v", err, err2)
- } else if err == nil && err2 != nil {
- err = err2
- }
-
- return err
-}
-
-func (tc *tarCompress) NewFile(information formatInformation) (io.WriteCloser, error) {
- rp, wp := io.Pipe()
- go func() {
- tc.wg.Add(1)
- defer tc.wg.Done()
- var buf bytes.Buffer
- io.Copy(&buf, rp)
-
- header := &tar.Header{
- Name: filepath.Join(fmt.Sprintf("S%2d", information.SeasonNumber), information.Title),
- ModTime: time.Now(),
- Mode: 0644,
- Typeflag: tar.TypeReg,
- // fun fact: i did not set the size for quiet some time because i thought that it isn't
- // required. well because of this i debugged this part for multiple hours because without
- // proper size information only a tiny amount gets copied into the tar (or zip) writer.
- // this is also the reason why the file content is completely copied into a buffer before
- // writing it to the writer. i could bypass this and save some memory but this requires
- // some rewriting and im nearly at the (planned) finish for version 2 so nah in the future
- // maybe
- Size: int64(buf.Len()),
- }
- tc.dst.WriteHeader(header)
- io.Copy(tc.dst, &buf)
- }()
- return wp, nil
-}
-
-func newZipCompress(file *os.File) *zipCompress {
- return &zipCompress{
- dst: zip.NewWriter(file),
- }
-}
-
-type zipCompress struct {
- compress
-
- wg sync.WaitGroup
-
- dst *zip.Writer
-}
-
-func (zc *zipCompress) Close() error {
- zc.wg.Wait()
- return zc.dst.Close()
-}
-
-func (zc *zipCompress) NewFile(information formatInformation) (io.WriteCloser, error) {
- rp, wp := io.Pipe()
- go func() {
- zc.wg.Add(1)
- defer zc.wg.Done()
-
- var buf bytes.Buffer
- io.Copy(&buf, rp)
-
- header := &zip.FileHeader{
- Name: filepath.Join(fmt.Sprintf("S%2d", information.SeasonNumber), information.Title),
- Modified: time.Now(),
- Method: zip.Deflate,
- UncompressedSize64: uint64(buf.Len()),
- }
- header.SetMode(0644)
-
- hw, _ := zc.dst.CreateHeader(header)
- io.Copy(hw, &buf)
- }()
-
- return wp, nil
-}
diff --git a/cmd/crunchyroll-go/cmd/download.go b/cmd/crunchyroll-go/cmd/download.go
deleted file mode 100644
index bbf9a3e..0000000
--- a/cmd/crunchyroll-go/cmd/download.go
+++ /dev/null
@@ -1,325 +0,0 @@
-package cmd
-
-import (
- "context"
- "fmt"
- "github.com/ByteDream/crunchyroll-go"
- "github.com/ByteDream/crunchyroll-go/utils"
- "github.com/grafov/m3u8"
- "github.com/spf13/cobra"
- "os"
- "os/signal"
- "path/filepath"
- "runtime"
- "sort"
- "strconv"
- "strings"
-)
-
-var (
- downloadAudioFlag string
- downloadSubtitleFlag string
-
- downloadDirectoryFlag string
- downloadOutputFlag string
-
- downloadResolutionFlag string
-
- downloadGoroutinesFlag int
-)
-
-var getCmd = &cobra.Command{
- Use: "download",
- Short: "Download a video",
- Args: cobra.MinimumNArgs(1),
-
- PreRunE: func(cmd *cobra.Command, args []string) error {
- out.Debug("Validating arguments")
-
- if filepath.Ext(downloadOutputFlag) != ".ts" {
- if !hasFFmpeg() {
- return fmt.Errorf("the file ending for the output file (%s) is not `.ts`. "+
- "Install ffmpeg (https://ffmpeg.org/download.html) to use other media file endings (e.g. `.mp4`)", downloadOutputFlag)
- } else {
- out.Debug("Custom file ending '%s' (ffmpeg is installed)", filepath.Ext(downloadOutputFlag))
- }
- }
-
- if !utils.ValidateLocale(crunchyroll.LOCALE(downloadAudioFlag)) {
- return fmt.Errorf("%s is not a valid audio locale. Choose from: %s", downloadAudioFlag, strings.Join(allLocalesAsStrings(), ", "))
- } else if downloadSubtitleFlag != "" && !utils.ValidateLocale(crunchyroll.LOCALE(downloadSubtitleFlag)) {
- return fmt.Errorf("%s is not a valid subtitle locale. Choose from: %s", downloadSubtitleFlag, strings.Join(allLocalesAsStrings(), ", "))
- }
- out.Debug("Locales: audio: %s / subtitle: %s", downloadAudioFlag, downloadSubtitleFlag)
-
- switch downloadResolutionFlag {
- case "1080p", "720p", "480p", "360p", "240p":
- intRes, _ := strconv.ParseFloat(strings.TrimSuffix(downloadResolutionFlag, "p"), 84)
- downloadResolutionFlag = fmt.Sprintf("%dx%s", int(intRes*(16/9)), strings.TrimSuffix(downloadResolutionFlag, "p"))
- case "1920x1080", "1280x720", "640x480", "480x360", "428x240", "best", "worst":
- default:
- return fmt.Errorf("'%s' is not a valid resolution", downloadResolutionFlag)
- }
- out.Debug("Using resolution '%s'", downloadResolutionFlag)
-
- return nil
- },
- RunE: func(cmd *cobra.Command, args []string) error {
- loadCrunchy()
-
- return download(args)
- },
-}
-
-func init() {
- getCmd.Flags().StringVarP(&downloadAudioFlag, "audio",
- "a",
- string(systemLocale(false)),
- "The locale of the audio. Available locales: "+strings.Join(allLocalesAsStrings(), ", "))
- getCmd.Flags().StringVarP(&downloadSubtitleFlag,
- "subtitle",
- "s",
- "",
- "The locale of the subtitle. Available locales: "+strings.Join(allLocalesAsStrings(), ", "))
-
- cwd, _ := os.Getwd()
- getCmd.Flags().StringVarP(&downloadDirectoryFlag,
- "directory",
- "d",
- cwd,
- "The directory to download the file(s) into")
- getCmd.Flags().StringVarP(&downloadOutputFlag,
- "output",
- "o",
- "{title}.ts",
- "Name of the output file. "+
- "If you use the following things in the name, the will get replaced:\n"+
- "\t{title} » Title of the video\n"+
- "\t{series_name} » Name of the series\n"+
- "\t{season_name} » Name of the season\n"+
- "\t{season_number} » Number of the season\n"+
- "\t{episode_number} » Number of the episode\n"+
- "\t{resolution} » Resolution of the video\n"+
- "\t{fps} » Frame Rate of the video\n"+
- "\t{audio} » Audio locale of the video\n"+
- "\t{subtitle} » Subtitle locale of the video")
-
- getCmd.Flags().StringVarP(&downloadResolutionFlag,
- "resolution",
- "r",
- "best",
- "The video resolution. Can either be specified via the pixels, the abbreviation for pixels, or 'common-use' words\n"+
- "\tAvailable pixels: 1920x1080, 1280x720, 640x480, 480x360, 428x240\n"+
- "\tAvailable abbreviations: 1080p, 720p, 480p, 360p, 240p\n"+
- "\tAvailable common-use words: best (best available resolution), worst (worst available resolution)")
-
- getCmd.Flags().IntVarP(&downloadGoroutinesFlag,
- "goroutines",
- "g",
- runtime.NumCPU(),
- "Sets how many parallel segment downloads should be used")
-
- rootCmd.AddCommand(getCmd)
-}
-
-func download(urls []string) error {
- for i, url := range urls {
- out.SetProgress("Parsing url %d", i+1)
- episodes, err := downloadExtractEpisodes(url)
- if err != nil {
- out.StopProgress("Failed to parse url %d", i+1)
- return err
- }
- out.StopProgress("Parsed url %d", i+1)
-
- for _, season := range episodes {
- out.Info("%s Season %d", season[0].SeriesName, season[0].SeasonNumber)
-
- for j, info := range season {
- out.Info("\t%d. %s » %spx, %.2f FPS (S%02dE%02d)",
- j+1,
- info.Title,
- info.Resolution,
- info.FPS,
- info.SeasonNumber,
- info.EpisodeNumber)
- }
- }
- out.Empty()
-
- for _, season := range episodes {
- for _, info := range season {
- dir := info.Format(downloadDirectoryFlag)
- if _, err = os.Stat(dir); os.IsNotExist(err) {
- if err = os.MkdirAll(dir, 0777); err != nil {
- return fmt.Errorf("error while creating directory: %v", err)
- }
- }
- file, err := os.Create(generateFilename(info.Format(downloadOutputFlag), dir))
- if err != nil {
- return fmt.Errorf("failed to create output file: %v", err)
- }
-
- if err = downloadInfo(info, file); err != nil {
- file.Close()
- os.Remove(file.Name())
- return err
- }
- file.Close()
- }
- }
- }
- return nil
-}
-
-func downloadInfo(info formatInformation, file *os.File) error {
- out.Debug("Entering season %d, episode %d", info.SeasonNumber, info.EpisodeNumber)
-
- if err := info.format.InitVideo(); err != nil {
- return fmt.Errorf("error while initializing the video: %v", err)
- }
-
- downloadProgress := &downloadProgress{
- Prefix: out.InfoLog.Prefix(),
- Message: "Downloading video",
- // number of segments a video has +2 is for merging and the success message
- Total: int(info.format.Video.Chunklist.Count()) + 2,
- Dev: out.IsDev(),
- Quiet: out.IsQuiet(),
- }
- if out.IsDev() {
- downloadProgress.Prefix = out.DebugLog.Prefix()
- }
- defer func() {
- if downloadProgress.Total != downloadProgress.Current {
- fmt.Println()
- }
- }()
-
- ctx, cancel := context.WithCancel(context.Background())
- defer cancel()
- downloader := crunchyroll.NewDownloader(ctx, file, downloadGoroutinesFlag, func(segment *m3u8.MediaSegment, current, total int, file *os.File) error {
- // check if the context was cancelled.
- // must be done in to not print any progress messages if ctrl+c was pressed
- if ctx.Err() != nil {
- return nil
- }
-
- if out.IsDev() {
- downloadProgress.UpdateMessage(fmt.Sprintf("Downloading %d/%d (%.2f%%) » %s", current, total, float32(current)/float32(total)*100, segment.URI), false)
- } else {
- downloadProgress.Update()
- }
-
- if current == total {
- downloadProgress.UpdateMessage("Merging segments", false)
- }
- return nil
- })
-
- sig := make(chan os.Signal, 1)
- signal.Notify(sig, os.Interrupt)
- go func() {
- select {
- case <-sig:
- signal.Stop(sig)
- out.Exit("Exiting... (may take a few seconds)")
- out.Exit("To force exit press ctrl+c (again)")
- cancel()
- // os.Exit(1) is not called because an immediate exit after the cancel function does not let
- // the download process enough time to stop gratefully. A result of this is that the temporary
- // directory where the segments are downloaded to will not be deleted
- case <-ctx.Done():
- // this is just here to end the goroutine and prevent it from running forever without a reason
- }
- }()
- out.Debug("Set up signal catcher")
-
- out.Info("Downloading episode `%s` to `%s`", info.Title, filepath.Base(file.Name()))
- out.Info("\tEpisode: S%02dE%02d", info.SeasonNumber, info.EpisodeNumber)
- out.Info("\tAudio: %s", info.Audio)
- out.Info("\tSubtitle: %s", info.Subtitle)
- out.Info("\tResolution: %spx", info.Resolution)
- out.Info("\tFPS: %.2f", info.FPS)
- if err := info.format.Download(downloader); err != nil {
- return fmt.Errorf("error while downloading: %v", err)
- }
-
- downloadProgress.UpdateMessage("Download finished", false)
-
- signal.Stop(sig)
- out.Debug("Stopped signal catcher")
-
- out.Empty()
- out.Empty()
-
- return nil
-}
-
-func downloadExtractEpisodes(url string) ([][]formatInformation, error) {
- episodes, err := extractEpisodes(url, crunchyroll.JP, crunchyroll.LOCALE(downloadAudioFlag))
- if err != nil {
- return nil, err
- }
- japanese := episodes[0]
- custom := episodes[1]
-
- sort.Sort(utils.EpisodesByNumber(japanese))
- sort.Sort(utils.EpisodesByNumber(custom))
-
- var errMessages []string
-
- var final []*crunchyroll.Episode
- if len(japanese) == 0 || len(japanese) == len(custom) {
- final = custom
- } else {
- for _, jp := range japanese {
- before := len(final)
- for _, episode := range custom {
- if jp.SeasonNumber == episode.SeasonNumber && jp.EpisodeNumber == episode.EpisodeNumber {
- final = append(final, episode)
- }
- }
- if before == len(final) {
- errMessages = append(errMessages, fmt.Sprintf("%s has no %s audio, using %s as fallback", jp.Title, crunchyroll.LOCALE(downloadAudioFlag), crunchyroll.JP))
- final = append(final, jp)
- }
- }
- }
-
- if len(errMessages) > 10 {
- for _, msg := range errMessages[:10] {
- out.SetProgress(msg)
- }
- out.SetProgress("... and %d more", len(errMessages)-10)
- } else {
- for _, msg := range errMessages {
- out.SetProgress(msg)
- }
- }
-
- var infoFormat [][]formatInformation
- for _, season := range utils.SortEpisodesBySeason(final) {
- tmpFormatInformation := make([]formatInformation, 0)
- for _, episode := range season {
- format, err := episode.GetFormat(downloadResolutionFlag, crunchyroll.LOCALE(downloadSubtitleFlag), true)
- if err != nil {
- return nil, fmt.Errorf("error while receiving format for %s: %v", episode.Title, err)
- }
- tmpFormatInformation = append(tmpFormatInformation, formatInformation{
- format: format,
-
- Title: episode.Title,
- SeriesName: episode.SeriesTitle,
- SeasonName: episode.SeasonTitle,
- SeasonNumber: episode.SeasonNumber,
- EpisodeNumber: episode.EpisodeNumber,
- Resolution: format.Video.Resolution,
- FPS: format.Video.FrameRate,
- Audio: format.AudioLocale,
- })
- }
- infoFormat = append(infoFormat, tmpFormatInformation)
- }
- return infoFormat, nil
-}
diff --git a/cmd/crunchyroll-go/cmd/logger.go b/cmd/crunchyroll-go/cmd/logger.go
deleted file mode 100644
index 83bc214..0000000
--- a/cmd/crunchyroll-go/cmd/logger.go
+++ /dev/null
@@ -1,193 +0,0 @@
-package cmd
-
-import (
- "fmt"
- "io"
- "log"
- "os"
- "runtime"
- "strings"
- "sync"
- "time"
-)
-
-var prefix, progressDown, progressDownFinish string
-
-func initPrefixBecauseWindowsSucksBallsHard() {
- // dear windows user, please change to a good OS, linux in the best case.
- // MICROSHIT DOES NOT GET IT DONE TO SHOW THE SYMBOLS IN THE ELSE CLAUSE
- // CORRECTLY. NOT IN THE CMD NOR POWERSHELL. WHY TF, IT IS ONE OF THE MOST
- // PROFITABLE COMPANIES ON THIS PLANET AND CANNOT SHOW A PROPER UTF-8 SYMBOL
- // IN THEIR OWN PRODUCT WHICH GETS USED MILLION TIMES A DAY
- if runtime.GOOS == "windows" {
- prefix = "=>"
- progressDown = "|"
- progressDownFinish = "->"
- } else {
- prefix = "➞"
- progressDown = "↓"
- progressDownFinish = "↳"
- }
-}
-
-type progress struct {
- message string
- stop bool
-}
-
-type logger struct {
- DebugLog *log.Logger
- InfoLog *log.Logger
- ErrLog *log.Logger
-
- devView bool
-
- progress chan progress
- done chan interface{}
- lock sync.Mutex
-}
-
-func newLogger(debug, info, err bool) *logger {
- initPrefixBecauseWindowsSucksBallsHard()
-
- debugLog, infoLog, errLog := log.New(io.Discard, prefix+" ", 0), log.New(io.Discard, prefix+" ", 0), log.New(io.Discard, prefix+" ", 0)
-
- if debug {
- debugLog.SetOutput(os.Stdout)
- }
- if info {
- infoLog.SetOutput(os.Stdout)
- }
- if err {
- errLog.SetOutput(os.Stderr)
- }
-
- if debug {
- debugLog = log.New(debugLog.Writer(), "[debug] ", 0)
- infoLog = log.New(infoLog.Writer(), "[info] ", 0)
- errLog = log.New(errLog.Writer(), "[err] ", 0)
- }
-
- return &logger{
- DebugLog: debugLog,
- InfoLog: infoLog,
- ErrLog: errLog,
-
- devView: debug,
- }
-}
-
-func (l *logger) IsDev() bool {
- return l.devView
-}
-
-func (l *logger) IsQuiet() bool {
- return l.DebugLog.Writer() == io.Discard && l.InfoLog.Writer() == io.Discard && l.ErrLog.Writer() == io.Discard
-}
-
-func (l *logger) Debug(format string, v ...interface{}) {
- l.DebugLog.Printf(format, v...)
-}
-
-func (l *logger) Info(format string, v ...interface{}) {
- l.InfoLog.Printf(format, v...)
-}
-
-func (l *logger) Err(format string, v ...interface{}) {
- l.ErrLog.Printf(format, v...)
-}
-
-func (l *logger) Exit(format string, v ...interface{}) {
- fmt.Fprintln(l.ErrLog.Writer(), fmt.Sprintf(format, v...))
-}
-
-func (l *logger) Empty() {
- if !l.devView && l.InfoLog.Writer() != io.Discard {
- fmt.Println("")
- }
-}
-
-func (l *logger) SetProgress(format string, v ...interface{}) {
- if out.InfoLog.Writer() == io.Discard {
- return
- } else if l.devView {
- l.Debug(format, v...)
- return
- }
-
- initialMessage := fmt.Sprintf(format, v...)
-
- p := progress{
- message: initialMessage,
- }
-
- l.lock.Lock()
- if l.done != nil {
- l.progress <- p
- return
- } else {
- l.progress = make(chan progress, 1)
- l.progress <- p
- l.done = make(chan interface{})
- }
-
- go func() {
- states := []string{"-", "\\", "|", "/"}
-
- var count int
-
- for i := 0; ; i++ {
- select {
- case p := <-l.progress:
- if p.stop {
- fmt.Printf("\r" + strings.Repeat(" ", len(prefix)+len(initialMessage)))
- if count > 1 {
- fmt.Printf("\r%s %s\n", progressDownFinish, p.message)
- } else {
- fmt.Printf("\r%s %s\n", prefix, p.message)
- }
-
- if l.done != nil {
- l.done <- nil
- }
- l.progress = nil
-
- l.lock.Unlock()
- return
- } else {
- if count > 0 {
- fmt.Printf("\r%s %s\n", progressDown, p.message)
- }
- l.progress = make(chan progress, 1)
-
- count++
-
- fmt.Printf("\r%s %s", states[i/10%4], initialMessage)
- l.lock.Unlock()
- }
- default:
- if i%10 == 0 {
- fmt.Printf("\r%s %s", states[i/10%4], initialMessage)
- }
- time.Sleep(35 * time.Millisecond)
- }
- }
- }()
-}
-
-func (l *logger) StopProgress(format string, v ...interface{}) {
- if out.InfoLog.Writer() == io.Discard {
- return
- } else if l.devView {
- l.Debug(format, v...)
- return
- }
-
- l.lock.Lock()
- l.progress <- progress{
- message: fmt.Sprintf(format, v...),
- stop: true,
- }
- <-l.done
- l.done = nil
-}
diff --git a/cmd/crunchyroll-go/cmd/login.go b/cmd/crunchyroll-go/cmd/login.go
deleted file mode 100644
index 1303516..0000000
--- a/cmd/crunchyroll-go/cmd/login.go
+++ /dev/null
@@ -1,81 +0,0 @@
-package cmd
-
-import (
- "fmt"
- "github.com/ByteDream/crunchyroll-go"
- "github.com/spf13/cobra"
- "os"
- "os/user"
- "path/filepath"
- "runtime"
-)
-
-var (
- loginPersistentFlag bool
-
- loginSessionIDFlag bool
-)
-
-var loginCmd = &cobra.Command{
- Use: "login",
- Short: "Login to crunchyroll",
- Args: cobra.RangeArgs(1, 2),
-
- Run: func(cmd *cobra.Command, args []string) {
- if loginSessionIDFlag {
- loginSessionID(args[0])
- } else {
- loginCredentials(args[0], args[1])
- }
- },
-}
-
-func init() {
- loginCmd.Flags().BoolVar(&loginPersistentFlag,
- "persistent",
- false,
- "If the given credential should be stored persistent")
-
- loginCmd.Flags().BoolVar(&loginSessionIDFlag,
- "session-id",
- false,
- "Use a session id to login instead of username and password")
-
- rootCmd.AddCommand(loginCmd)
-}
-
-func loginCredentials(user, password string) error {
- out.Debug("Logging in via credentials")
- if _, err := crunchyroll.LoginWithCredentials(user, password, systemLocale(false), client); err != nil {
- out.Err(err.Error())
- os.Exit(1)
- }
-
- return os.WriteFile(loginStorePath(), []byte(fmt.Sprintf("%s\n%s", user, password)), 0600)
-}
-
-func loginSessionID(sessionID string) error {
- out.Debug("Logging in via session id")
- if _, err := crunchyroll.LoginWithSessionID(sessionID, systemLocale(false), client); err != nil {
- out.Err(err.Error())
- os.Exit(1)
- }
-
- return os.WriteFile(loginStorePath(), []byte(sessionID), 0600)
-}
-
-func loginStorePath() string {
- path := filepath.Join(os.TempDir(), ".crunchy")
- if loginPersistentFlag {
- if runtime.GOOS != "windows" {
- usr, _ := user.Current()
- path = filepath.Join(usr.HomeDir, ".config/crunchy")
- }
-
- out.Info("The login information will be stored permanently UNENCRYPTED on your drive (%s)", path)
- } else if runtime.GOOS != "windows" {
- out.Info("Due to security reasons, you have to login again on the next reboot")
- }
-
- return path
-}
diff --git a/cmd/crunchyroll-go/cmd/root.go b/cmd/crunchyroll-go/cmd/root.go
deleted file mode 100644
index 25c8270..0000000
--- a/cmd/crunchyroll-go/cmd/root.go
+++ /dev/null
@@ -1,68 +0,0 @@
-package cmd
-
-import (
- "context"
- "github.com/ByteDream/crunchyroll-go"
- "github.com/spf13/cobra"
- "net/http"
- "os"
- "runtime/debug"
- "strings"
-)
-
-var (
- client *http.Client
- crunchy *crunchyroll.Crunchyroll
- out = newLogger(false, true, true)
-
- quietFlag bool
- verboseFlag bool
- proxyFlag string
-)
-
-var rootCmd = &cobra.Command{
- Use: "crunchyroll",
- Short: "Download crunchyroll videos with ease. See the wiki for details about the cli and library: https://github.com/ByteDream/crunchyroll-go/wiki",
-
- SilenceErrors: true,
- SilenceUsage: true,
-
- PersistentPreRunE: func(cmd *cobra.Command, args []string) (err error) {
- if verboseFlag {
- out = newLogger(true, true, true)
- } else if quietFlag {
- out = newLogger(false, false, false)
- }
-
- out.DebugLog.Printf("Executing `%s` command with %d arg(s)\n", cmd.Name(), len(args))
-
- client, err = createOrDefaultClient(proxyFlag)
- return
- },
-}
-
-func init() {
- rootCmd.PersistentFlags().BoolVarP(&quietFlag, "quiet", "q", false, "Disable all output")
- rootCmd.PersistentFlags().BoolVarP(&verboseFlag, "verbose", "v", false, "Adds debug messages to the normal output")
- rootCmd.PersistentFlags().StringVarP(&proxyFlag, "proxy", "p", "", "Proxy to use")
-}
-
-func Execute() {
- rootCmd.CompletionOptions.DisableDefaultCmd = true
- defer func() {
- if r := recover(); r != nil {
- if out.IsDev() {
- out.Err("%v: %s", r, debug.Stack())
- } else {
- out.Err("Unexpected error: %v", r)
- }
- os.Exit(1)
- }
- }()
- if err := rootCmd.Execute(); err != nil {
- if !strings.HasSuffix(err.Error(), context.Canceled.Error()) {
- out.Exit("An error occurred: %v", err)
- }
- os.Exit(1)
- }
-}
diff --git a/cmd/crunchyroll-go/cmd/utils.go b/cmd/crunchyroll-go/cmd/utils.go
deleted file mode 100644
index b4afb73..0000000
--- a/cmd/crunchyroll-go/cmd/utils.go
+++ /dev/null
@@ -1,407 +0,0 @@
-package cmd
-
-import (
- "fmt"
- "github.com/ByteDream/crunchyroll-go"
- "github.com/ByteDream/crunchyroll-go/utils"
- "net/http"
- "net/url"
- "os"
- "os/exec"
- "os/user"
- "path/filepath"
- "reflect"
- "regexp"
- "runtime"
- "sort"
- "strconv"
- "strings"
- "sync"
- "time"
-)
-
-var (
- // ahh i love windows :)))
- invalidWindowsChars = []string{"\\", "<", ">", ":", "\"", "/", "|", "?", "*"}
- invalidNotWindowsChars = []string{"/"}
-)
-
-var urlFilter = regexp.MustCompile(`(S(\d+))?(E(\d+))?((-)(S(\d+))?(E(\d+))?)?(,|$)`)
-
-// systemLocale receives the system locale
-// https://stackoverflow.com/questions/51829386/golang-get-system-language/51831590#51831590
-func systemLocale(verbose bool) crunchyroll.LOCALE {
- if runtime.GOOS != "windows" {
- if lang, ok := os.LookupEnv("LANG"); ok {
- prefix := strings.Split(lang, "_")[0]
- suffix := strings.Split(strings.Split(lang, ".")[0], "_")[1]
- l := crunchyroll.LOCALE(fmt.Sprintf("%s-%s", prefix, suffix))
- if !utils.ValidateLocale(l) {
- if verbose {
- out.Err("%s is not a supported locale, using %s as fallback", l, crunchyroll.US)
- }
- l = crunchyroll.US
- }
- return l
- }
- } else {
- cmd := exec.Command("powershell", "Get-Culture | select -exp Name")
- if output, err := cmd.Output(); err == nil {
- l := crunchyroll.LOCALE(strings.Trim(string(output), "\r\n"))
- if !utils.ValidateLocale(l) {
- if verbose {
- out.Err("%s is not a supported locale, using %s as fallback", l, crunchyroll.US)
- }
- l = crunchyroll.US
- }
- return l
- }
- }
- if verbose {
- out.Err("Failed to get locale, using %s", crunchyroll.US)
- }
- return crunchyroll.US
-}
-
-func allLocalesAsStrings() (locales []string) {
- for _, locale := range utils.AllLocales {
- locales = append(locales, string(locale))
- }
- sort.Strings(locales)
- return
-}
-
-func createOrDefaultClient(proxy string) (*http.Client, error) {
- if proxy == "" {
- return http.DefaultClient, nil
- } else {
- out.Info("Using custom proxy %s", proxy)
- proxyURL, err := url.Parse(proxy)
- if err != nil {
- return nil, err
- }
- client := &http.Client{
- Transport: &http.Transport{
- DisableCompression: true,
- Proxy: http.ProxyURL(proxyURL),
- },
- Timeout: 30 * time.Second,
- }
- return client, nil
- }
-}
-
-func freeFileName(filename string) (string, bool) {
- ext := filepath.Ext(filename)
- base := strings.TrimSuffix(filename, ext)
- // checks if a .tar stands before the "actual" file ending
- if extraExt := filepath.Ext(base); extraExt == ".tar" {
- ext = extraExt + ext
- base = strings.TrimSuffix(base, extraExt)
- }
- j := 0
- for ; ; j++ {
- if _, stat := os.Stat(filename); stat != nil && !os.IsExist(stat) {
- break
- }
- filename = fmt.Sprintf("%s (%d)%s", base, j+1, ext)
- }
- return filename, j != 0
-}
-
-func loadCrunchy() {
- out.SetProgress("Logging in")
-
- files := []string{filepath.Join(os.TempDir(), ".crunchy")}
-
- if runtime.GOOS != "windows" {
- usr, _ := user.Current()
- files = append(files, filepath.Join(usr.HomeDir, ".config/crunchy"))
- }
-
- var body []byte
- var err error
- for _, file := range files {
- if _, err = os.Stat(file); os.IsNotExist(err) {
- continue
- }
- body, err = os.ReadFile(file)
- break
- }
- if body == nil {
- out.StopProgress("To use this command, login first. Type `%s login -h` to get help", os.Args[0])
- os.Exit(1)
- } else if err != nil {
- out.StopProgress("Failed to read login information: %v", err)
- os.Exit(1)
- }
-
- split := strings.SplitN(string(body), "\n", 2)
- if len(split) == 1 || split[1] == "" {
- if crunchy, err = crunchyroll.LoginWithSessionID(split[0], systemLocale(true), client); err != nil {
- out.StopProgress(err.Error())
- os.Exit(1)
- }
- out.Debug("Logged in with session id %s. BLANK THIS LINE OUT IF YOU'RE ASKED TO POST THE DEBUG OUTPUT SOMEWHERE", split[0])
- } else {
- if crunchy, err = crunchyroll.LoginWithCredentials(split[0], split[1], systemLocale(true), client); err != nil {
- out.StopProgress(err.Error())
- os.Exit(1)
- }
- out.Debug("Logged in with username '%s' and password '%s'. BLANK THIS LINE OUT IF YOU'RE ASKED TO POST THE DEBUG OUTPUT SOMEWHERE", split[0], split[1])
- }
-
- out.StopProgress("Logged in")
-}
-
-func hasFFmpeg() bool {
- return exec.Command("ffmpeg", "-h").Run() == nil
-}
-
-func terminalWidth() int {
- if runtime.GOOS != "windows" {
- cmd := exec.Command("stty", "size")
- cmd.Stdin = os.Stdin
- res, err := cmd.Output()
- if err != nil {
- return 60
- }
- // on alpine linux the command `stty size` does not respond the terminal size
- // but something like "stty: standard input". this may also apply to other systems
- splitOutput := strings.SplitN(strings.ReplaceAll(string(res), "\n", ""), " ", 2)
- if len(splitOutput) == 1 {
- return 60
- }
- width, err := strconv.Atoi(splitOutput[1])
- if err != nil {
- return 60
- }
- return width
- }
- return 60
-}
-
-func generateFilename(name, directory string) string {
- if runtime.GOOS != "windows" {
- for _, char := range invalidNotWindowsChars {
- name = strings.ReplaceAll(name, char, "")
- }
- out.Debug("Replaced invalid characters (not windows)")
- } else {
- for _, char := range invalidWindowsChars {
- name = strings.ReplaceAll(name, char, "")
- }
- out.Debug("Replaced invalid characters (windows)")
- }
-
- filename, changed := freeFileName(filepath.Join(directory, name))
- if changed {
- out.Debug("File `%s` already exists, changing name to `%s`", filepath.Base(name), filepath.Base(filename))
- }
-
- return filename
-}
-
-func extractEpisodes(url string, locales ...crunchyroll.LOCALE) ([][]*crunchyroll.Episode, error) {
- var matches [][]string
-
- lastOpen := strings.LastIndex(url, "[")
- if strings.HasSuffix(url, "]") && lastOpen != -1 && lastOpen < len(url) {
- matches = urlFilter.FindAllStringSubmatch(url[lastOpen+1:len(url)-1], -1)
-
- var all string
- for _, match := range matches {
- all += match[0]
- }
- if all != url[lastOpen+1:len(url)-1] {
- return nil, fmt.Errorf("invalid episode filter")
- }
- url = url[:lastOpen]
- }
-
- final := make([][]*crunchyroll.Episode, len(locales))
- episodes, err := crunchy.ExtractEpisodesFromUrl(url, locales...)
- if err != nil {
- return nil, fmt.Errorf("failed to get episodes: %v", err)
- }
-
- if len(episodes) == 0 {
- return nil, fmt.Errorf("no episodes found")
- }
-
- if matches != nil {
- for _, match := range matches {
- fromSeason, fromEpisode, toSeason, toEpisode := -1, -1, -1, -1
- if match[2] != "" {
- fromSeason, _ = strconv.Atoi(match[2])
- }
- if match[4] != "" {
- fromEpisode, _ = strconv.Atoi(match[4])
- }
- if match[8] != "" {
- toSeason, _ = strconv.Atoi(match[8])
- }
- if match[10] != "" {
- toEpisode, _ = strconv.Atoi(match[10])
- }
-
- if match[6] != "-" {
- toSeason = fromSeason
- toEpisode = fromEpisode
- }
-
- tmpEps := make([]*crunchyroll.Episode, 0)
- for _, episode := range episodes {
- if fromSeason != -1 && (episode.SeasonNumber < fromSeason || (fromEpisode != -1 && episode.EpisodeNumber < fromEpisode)) {
- continue
- } else if fromSeason == -1 && fromEpisode != -1 && episode.EpisodeNumber < fromEpisode {
- continue
- } else if toSeason != -1 && (episode.SeasonNumber > toSeason || (toEpisode != -1 && episode.EpisodeNumber > toEpisode)) {
- continue
- } else if toSeason == -1 && toEpisode != -1 && episode.EpisodeNumber > toEpisode {
- continue
- } else {
- tmpEps = append(tmpEps, episode)
- }
- }
-
- if len(tmpEps) == 0 {
- return nil, fmt.Errorf("no episodes are matching the given filter")
- }
-
- episodes = tmpEps
- }
- }
-
- localeSorted, err := utils.SortEpisodesByAudio(episodes)
- if err != nil {
- return nil, fmt.Errorf("failed to get audio locale: %v", err)
- }
- for i, locale := range locales {
- final[i] = append(final[i], localeSorted[locale]...)
- }
-
- return final, nil
-}
-
-type formatInformation struct {
- // the format to download
- format *crunchyroll.Format
-
- // additional formats which are only used by archive.go
- additionalFormats []*crunchyroll.Format
-
- Title string `json:"title"`
- SeriesName string `json:"series_name"`
- SeasonName string `json:"season_name"`
- SeasonNumber int `json:"season_number"`
- EpisodeNumber int `json:"episode_number"`
- Resolution string `json:"resolution"`
- FPS float64 `json:"fps"`
- Audio crunchyroll.LOCALE `json:"audio"`
- Subtitle crunchyroll.LOCALE `json:"subtitle"`
-}
-
-func (fi formatInformation) Format(source string) string {
- fields := reflect.TypeOf(fi)
- values := reflect.ValueOf(fi)
-
- for i := 0; i < fields.NumField(); i++ {
- var valueAsString string
- switch value := values.Field(i); value.Kind() {
- case reflect.String:
- valueAsString = value.String()
- case reflect.Int:
- valueAsString = fmt.Sprintf("%02d", value.Int())
- case reflect.Float64:
- valueAsString = fmt.Sprintf("%.2f", value.Float())
- case reflect.Bool:
- valueAsString = fields.Field(i).Tag.Get("json")
- if !value.Bool() {
- valueAsString = "no " + valueAsString
- }
- }
-
- if runtime.GOOS != "windows" {
- for _, char := range invalidNotWindowsChars {
- valueAsString = strings.ReplaceAll(valueAsString, char, "")
- }
- out.Debug("Replaced invalid characters (not windows)")
- } else {
- for _, char := range invalidWindowsChars {
- valueAsString = strings.ReplaceAll(valueAsString, char, "")
- }
- out.Debug("Replaced invalid characters (windows)")
- }
-
- source = strings.ReplaceAll(source, "{"+fields.Field(i).Tag.Get("json")+"}", valueAsString)
- }
-
- return source
-}
-
-type downloadProgress struct {
- Prefix string
- Message string
-
- Total int
- Current int
-
- Dev bool
- Quiet bool
-
- lock sync.Mutex
-}
-
-func (dp *downloadProgress) Update() {
- dp.update("", false)
-}
-
-func (dp *downloadProgress) UpdateMessage(msg string, permanent bool) {
- dp.update(msg, permanent)
-}
-
-func (dp *downloadProgress) update(msg string, permanent bool) {
- if dp.Quiet {
- return
- }
-
- if dp.Current >= dp.Total {
- return
- }
-
- dp.lock.Lock()
- defer dp.lock.Unlock()
- dp.Current++
-
- if msg == "" {
- msg = dp.Message
- }
- if permanent {
- dp.Message = msg
- }
-
- if dp.Dev {
- fmt.Printf("%s%s\n", dp.Prefix, msg)
- return
- }
-
- percentage := float32(dp.Current) / float32(dp.Total) * 100
-
- pre := fmt.Sprintf("%s%s [", dp.Prefix, msg)
- post := fmt.Sprintf("]%4d%% %8d/%d", int(percentage), dp.Current, dp.Total)
-
- // i don't really know why +2 is needed here but without it the Printf below would not print to the line end
- progressWidth := terminalWidth() - len(pre) - len(post) + 2
- repeatCount := int(percentage / float32(100) * float32(progressWidth))
- // it can be lower than zero when the terminal is very tiny
- if repeatCount < 0 {
- repeatCount = 0
- }
- progressPercentage := strings.Repeat("=", repeatCount)
- if dp.Current != dp.Total {
- progressPercentage += ">"
- }
-
- fmt.Printf("\r%s%-"+fmt.Sprint(progressWidth)+"s%s", pre, progressPercentage, post)
-}
diff --git a/cmd/crunchyroll-go/main.go b/cmd/crunchyroll-go/main.go
deleted file mode 100644
index efc6a1e..0000000
--- a/cmd/crunchyroll-go/main.go
+++ /dev/null
@@ -1,9 +0,0 @@
-package main
-
-import (
- "github.com/ByteDream/crunchyroll-go/cmd/crunchyroll-go/cmd"
-)
-
-func main() {
- cmd.Execute()
-}
diff --git a/crunchy-cli-core/Cargo.toml b/crunchy-cli-core/Cargo.toml
new file mode 100644
index 0000000..399053f
--- /dev/null
+++ b/crunchy-cli-core/Cargo.toml
@@ -0,0 +1,51 @@
+[package]
+name = "crunchy-cli-core"
+authors = ["Crunchy Labs Maintainers"]
+version = "3.6.7"
+edition = "2021"
+license = "MIT"
+
+[features]
+rustls-tls = ["reqwest/rustls-tls"]
+native-tls = ["reqwest/native-tls", "reqwest/native-tls-alpn"]
+openssl-tls = ["reqwest/native-tls", "reqwest/native-tls-alpn", "dep:rustls-native-certs"]
+openssl-tls-static = ["reqwest/native-tls", "reqwest/native-tls-alpn", "reqwest/native-tls-vendored", "dep:rustls-native-certs"]
+
+[dependencies]
+anyhow = "1.0"
+async-speed-limit = "0.4"
+clap = { version = "4.5", features = ["derive", "string"] }
+chrono = "0.4"
+crunchyroll-rs = { version = "0.11.4", features = ["experimental-stabilizations", "tower"] }
+ctrlc = "3.4"
+dialoguer = { version = "0.11", default-features = false }
+dirs = "5.0"
+derive_setters = "0.1"
+futures-util = { version = "0.3", features = ["io"] }
+fs2 = "0.4"
+http = "1.1"
+indicatif = "0.17"
+lazy_static = "1.4"
+log = { version = "0.4", features = ["std"] }
+num_cpus = "1.16"
+regex = "1.10"
+reqwest = { version = "0.12", features = ["socks", "stream"] }
+rsubs-lib = "~0.3.2"
+rusty-chromaprint = "0.2"
+serde = "1.0"
+serde_json = "1.0"
+serde_plain = "1.0"
+shlex = "1.3"
+sys-locale = "0.3"
+tempfile = "3.10"
+time = "0.3"
+tokio = { version = "1.38", features = ["io-util", "macros", "net", "rt-multi-thread", "time"] }
+tokio-util = "0.7"
+tower-service = "0.3"
+rustls-native-certs = { version = "0.7", optional = true }
+
+[target.'cfg(not(target_os = "windows"))'.dependencies]
+nix = { version = "0.28", features = ["fs"] }
+
+[build-dependencies]
+chrono = "0.4"
diff --git a/crunchy-cli-core/build.rs b/crunchy-cli-core/build.rs
new file mode 100644
index 0000000..b36ec8e
--- /dev/null
+++ b/crunchy-cli-core/build.rs
@@ -0,0 +1,34 @@
+fn main() -> std::io::Result<()> {
+ println!(
+ "cargo:rustc-env=GIT_HASH={}",
+ std::env::var("CRUNCHY_CLI_GIT_HASH")
+ .or::(Ok(get_short_commit_hash()?.unwrap_or_default()))?
+ );
+ println!(
+ "cargo:rustc-env=BUILD_DATE={}",
+ chrono::Utc::now().format("%F")
+ );
+
+ Ok(())
+}
+
+fn get_short_commit_hash() -> std::io::Result> {
+ let git = std::process::Command::new("git")
+ .arg("rev-parse")
+ .arg("--short")
+ .arg("HEAD")
+ .output();
+
+ match git {
+ Ok(cmd) => Ok(Some(
+ String::from_utf8_lossy(cmd.stdout.as_slice()).to_string(),
+ )),
+ Err(e) => {
+ if e.kind() != std::io::ErrorKind::NotFound {
+ Err(e)
+ } else {
+ Ok(None)
+ }
+ }
+ }
+}
diff --git a/crunchy-cli-core/src/archive/command.rs b/crunchy-cli-core/src/archive/command.rs
new file mode 100644
index 0000000..0d1b3a4
--- /dev/null
+++ b/crunchy-cli-core/src/archive/command.rs
@@ -0,0 +1,692 @@
+use crate::utils::context::Context;
+use crate::utils::download::{
+ DownloadBuilder, DownloadFormat, DownloadFormatMetadata, MergeBehavior,
+};
+use crate::utils::ffmpeg::FFmpegPreset;
+use crate::utils::filter::{Filter, FilterMediaScope};
+use crate::utils::format::{Format, SingleFormat};
+use crate::utils::locale::{all_locale_in_locales, resolve_locales, LanguageTagging};
+use crate::utils::log::progress;
+use crate::utils::os::{free_file, has_ffmpeg, is_special_file};
+use crate::utils::parse::parse_url;
+use crate::utils::video::stream_data_from_stream;
+use crate::Execute;
+use anyhow::bail;
+use anyhow::Result;
+use chrono::Duration;
+use crunchyroll_rs::media::{Resolution, Subtitle};
+use crunchyroll_rs::Locale;
+use log::{debug, warn};
+use regex::Regex;
+use std::fmt::{Display, Formatter};
+use std::iter::zip;
+use std::ops::Sub;
+use std::path::{Path, PathBuf};
+use std::process::{Command, Stdio};
+
+#[derive(Clone, Debug, clap::Parser)]
+#[clap(about = "Archive a video")]
+#[command(arg_required_else_help(true))]
+pub struct Archive {
+ #[arg(help = format!("Audio languages. Can be used multiple times. \
+ Available languages are: {}", Locale::all().into_iter().map(|l| l.to_string()).collect::>().join(", ")))]
+ #[arg(long_help = format!("Audio languages. Can be used multiple times. \
+ Available languages are:\n {}\nIETF tagged language codes for the shown available locales can be used too", Locale::all().into_iter().map(|l| format!("{:<6} → {}", l.to_string(), l.to_human_readable())).collect::>().join("\n ")))]
+ #[arg(short, long, default_values_t = vec![Locale::ja_JP, crate::utils::locale::system_locale()])]
+ pub(crate) audio: Vec,
+ #[arg(skip)]
+ output_audio_locales: Vec,
+ #[arg(help = format!("Subtitle languages. Can be used multiple times. \
+ Available languages are: {}", Locale::all().into_iter().map(|l| l.to_string()).collect::>().join(", ")))]
+ #[arg(long_help = format!("Subtitle languages. Can be used multiple times. \
+ Available languages are: {}\nIETF tagged language codes for the shown available locales can be used too", Locale::all().into_iter().map(|l| l.to_string()).collect::>().join(", ")))]
+ #[arg(short, long, default_values_t = Locale::all())]
+ pub(crate) subtitle: Vec,
+ #[arg(skip)]
+ output_subtitle_locales: Vec,
+
+ #[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 \
+ {width} → Width of the video\n \
+ {height} → Height of the video\n \
+ {season_number} → Number of the season\n \
+ {episode_number} → Number of the episode\n \
+ {relative_episode_number} → Number of the episode relative to its season\n \
+ {sequence_number} → Like '{episode_number}' but without possible non-number characters\n \
+ {relative_sequence_number} → Like '{relative_episode_number}' but with support for episode 0's and .5's\n \
+ {release_year} → Release year of the video\n \
+ {release_month} → Release month of the video\n \
+ {release_day} → Release day of the video\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}.mkv")]
+ pub(crate) output: String,
+ #[arg(help = "Name of the output file if the episode is a special")]
+ #[arg(long_help = "Name of the output file if the episode is a special. \
+ If not set, the '-o'/'--output' flag will be used as name template")]
+ #[arg(long)]
+ pub(crate) output_specials: Option,
+
+ #[arg(help = "Sanitize the output file for use with all operating systems. \
+ This option only affects template options and not static characters.")]
+ #[arg(long, default_value_t = false)]
+ pub(crate) universal_output: bool,
+
+ #[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)]
+ pub(crate) resolution: Resolution,
+
+ #[arg(
+ help = "Sets the behavior of the stream merging. Valid behaviors are 'auto', 'sync', 'audio' and 'video'"
+ )]
+ #[arg(
+ long_help = "Because of local restrictions (or other reasons) some episodes with different languages does not have the same length (e.g. when some scenes were cut out). \
+ With this flag you can set the behavior when handling multiple language.
+ Valid options are 'audio' (stores one video and all other languages as audio only), 'video' (stores the video + audio for every language), 'auto' (detects if videos differ in length: if so, behave like 'video' else like 'audio') and 'sync' (detects if videos differ in length: if so, tries to find the offset of matching audio parts and removes it from the beginning, otherwise it behaves like 'audio')"
+ )]
+ #[arg(short, long, default_value = "auto")]
+ #[arg(value_parser = MergeBehavior::parse)]
+ pub(crate) merge: MergeBehavior,
+ #[arg(
+ help = "If the merge behavior is 'auto' or 'sync', consider videos to be of equal lengths if the difference in length is smaller than the specified milliseconds"
+ )]
+ #[arg(long, default_value_t = 200)]
+ pub(crate) merge_time_tolerance: u32,
+ #[arg(
+ help = "If the merge behavior is 'sync', specify the difference by which two fingerprints are considered equal, higher values can help when the algorithm fails"
+ )]
+ #[arg(long, default_value_t = 6)]
+ pub(crate) merge_sync_tolerance: u32,
+ #[arg(
+ help = "If the merge behavior is 'sync', specify the amount of offset determination runs from which the final offset is calculated, higher values will increase the time required but lead to more precise offsets"
+ )]
+ #[arg(long, default_value_t = 4)]
+ pub(crate) merge_sync_precision: u32,
+
+ #[arg(
+ help = "Specified which language tagging the audio and subtitle tracks and language specific format options should have. \
+ Valid options are: 'default' (how Crunchyroll uses it internally), 'ietf' (according to the IETF standard)"
+ )]
+ #[arg(
+ long_help = "Specified which language tagging the audio and subtitle tracks and language specific format options should have. \
+ Valid options are: 'default' (how Crunchyroll uses it internally), 'ietf' (according to the IETF standard; you might run in issues as there are multiple locales which resolve to the same IETF language code, e.g. 'es-LA' and 'es-ES' are both resolving to 'es')"
+ )]
+ #[arg(long)]
+ #[arg(value_parser = LanguageTagging::parse)]
+ pub(crate) language_tagging: Option,
+
+ #[arg(help = format!("Presets for converting the video to a specific coding format. \
+ Available presets: \n {}", FFmpegPreset::available_matches_human_readable().join("\n ")))]
+ #[arg(long_help = format!("Presets for converting the video to a specific coding format. \
+ If you need more specific ffmpeg customizations you can pass ffmpeg output arguments instead of a preset as value. \
+ Available presets: \n {}", FFmpegPreset::available_matches_human_readable().join("\n ")))]
+ #[arg(long)]
+ #[arg(value_parser = FFmpegPreset::parse)]
+ pub(crate) ffmpeg_preset: Option,
+ #[arg(
+ help = "The number of threads used by ffmpeg to generate the output file. Does not work with every codec/preset"
+ )]
+ #[arg(
+ long_help = "The number of threads used by ffmpeg to generate the output file. \
+ Does not work with every codec/preset and is skipped entirely when specifying custom ffmpeg output arguments instead of a preset for `--ffmpeg-preset`. \
+ By default, ffmpeg chooses the thread count which works best for the output codec"
+ )]
+ #[arg(long)]
+ pub(crate) ffmpeg_threads: Option,
+
+ #[arg(
+ help = "Set which subtitle language should be set as default / auto shown when starting a video"
+ )]
+ #[arg(long)]
+ pub(crate) default_subtitle: Option,
+ #[arg(help = "Include fonts in the downloaded file")]
+ #[arg(long)]
+ pub(crate) include_fonts: bool,
+ #[arg(
+ help = "Includes chapters (e.g. intro, credits, ...). Only works if `--merge` is set to 'audio'"
+ )]
+ #[arg(
+ long_help = "Includes chapters (e.g. intro, credits, ...). . Only works if `--merge` is set to 'audio'. \
+ Because chapters are essentially only special timeframes in episodes like the intro, most of the video timeline isn't covered by a chapter.
+ These \"gaps\" are filled with an 'Episode' chapter because many video players are ignore those gaps and just assume that a chapter ends when the next chapter start is reached, even if a specific end-time is set.
+ Also chapters aren't always available, so in this case, just a big 'Episode' chapter from start to end will be created"
+ )]
+ #[arg(long, default_value_t = false)]
+ pub(crate) include_chapters: bool,
+
+ #[arg(help = "Omit closed caption subtitles in the downloaded file")]
+ #[arg(long, default_value_t = false)]
+ pub(crate) no_closed_caption: bool,
+
+ #[arg(help = "Skip files which are already existing by their name")]
+ #[arg(long, default_value_t = false)]
+ pub(crate) skip_existing: bool,
+ #[arg(
+ help = "Only works in combination with `--skip-existing`. Sets the method how already existing files should be skipped. Valid methods are 'audio' and 'subtitle'"
+ )]
+ #[arg(long_help = "Only works in combination with `--skip-existing`. \
+ By default, already existing files are determined by their name and the download of the corresponding episode is skipped. \
+ With this flag you can modify this behavior. \
+ Valid options are 'audio' and 'subtitle' (if the file already exists but the audio/subtitle are less from what should be downloaded, the episode gets downloaded and the file overwritten).")]
+ #[arg(long, default_values_t = SkipExistingMethod::default())]
+ #[arg(value_parser = SkipExistingMethod::parse)]
+ pub(crate) skip_existing_method: Vec,
+ #[arg(help = "Skip special episodes")]
+ #[arg(long, default_value_t = false)]
+ pub(crate) skip_specials: bool,
+
+ #[arg(help = "Skip any interactive input")]
+ #[arg(short, long, default_value_t = false)]
+ pub(crate) yes: bool,
+
+ #[arg(help = "The number of threads used to download")]
+ #[arg(short, long, default_value_t = num_cpus::get())]
+ pub(crate) threads: usize,
+
+ #[arg(help = "Crunchyroll series url(s)")]
+ #[arg(required = true)]
+ pub(crate) urls: Vec,
+}
+
+impl Execute for Archive {
+ fn pre_check(&mut self) -> Result<()> {
+ if !has_ffmpeg() {
+ bail!("FFmpeg is needed to run this command")
+ } else if PathBuf::from(&self.output)
+ .extension()
+ .unwrap_or_default()
+ .to_string_lossy()
+ != "mkv"
+ && !is_special_file(&self.output)
+ && self.output != "-"
+ {
+ bail!("File extension is not '.mkv'. Currently only matroska / '.mkv' files are supported")
+ } else if let Some(special_output) = &self.output_specials {
+ if PathBuf::from(special_output)
+ .extension()
+ .unwrap_or_default()
+ .to_string_lossy()
+ != "mkv"
+ && !is_special_file(special_output)
+ && special_output != "-"
+ {
+ bail!("File extension for special episodes is not '.mkv'. Currently only matroska / '.mkv' files are supported")
+ }
+ }
+
+ if self.include_chapters
+ && !matches!(self.merge, MergeBehavior::Sync)
+ && !matches!(self.merge, MergeBehavior::Audio)
+ {
+ bail!("`--include-chapters` can only be used if `--merge` is set to 'audio' or 'sync'")
+ }
+
+ if !self.skip_existing_method.is_empty() && !self.skip_existing {
+ warn!("`--skip-existing-method` has no effect if `--skip-existing` is not set")
+ }
+
+ self.audio = all_locale_in_locales(self.audio.clone());
+ self.subtitle = all_locale_in_locales(self.subtitle.clone());
+
+ if let Some(language_tagging) = &self.language_tagging {
+ self.audio = resolve_locales(&self.audio);
+ self.subtitle = resolve_locales(&self.subtitle);
+ self.output_audio_locales = language_tagging.convert_locales(&self.audio);
+ self.output_subtitle_locales = language_tagging.convert_locales(&self.subtitle);
+ } else {
+ self.output_audio_locales = self
+ .audio
+ .clone()
+ .into_iter()
+ .map(|l| l.to_string())
+ .collect();
+ self.output_subtitle_locales = self
+ .subtitle
+ .clone()
+ .into_iter()
+ .map(|l| l.to_string())
+ .collect();
+ }
+
+ Ok(())
+ }
+
+ async fn execute(self, ctx: Context) -> Result<()> {
+ if !ctx.crunchy.premium().await {
+ warn!("You may not be able to download all requested videos when logging in anonymously or using a non-premium account")
+ }
+
+ let mut parsed_urls = vec![];
+
+ for (i, url) in self.urls.clone().into_iter().enumerate() {
+ let progress_handler = progress!("Parsing url {}", i + 1);
+ match parse_url(&ctx.crunchy, url.clone(), true).await {
+ Ok((media_collection, url_filter)) => {
+ progress_handler.stop(format!("Parsed url {}", i + 1));
+ parsed_urls.push((media_collection, url_filter))
+ }
+ 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 single_format_collection = Filter::new(
+ url_filter,
+ self.audio.clone(),
+ self.subtitle.clone(),
+ |scope, locales| {
+ let audios = locales.into_iter().map(|l| l.to_string()).collect::>().join(", ");
+ match scope {
+ FilterMediaScope::Series(series) => warn!("Series {} is not available with {} audio", series.title, audios),
+ FilterMediaScope::Season(season) => warn!("Season {} is not available with {} audio", season.season_number, audios),
+ FilterMediaScope::Episode(episodes) => {
+ if episodes.len() == 1 {
+ warn!("Episode {} is not available with {} audio", episodes[0].sequence_number, audios)
+ } else if episodes.len() == 2 {
+ warn!("Season {} is only available with {} audio from episode {} to {}", episodes[0].season_number, audios, episodes[0].sequence_number, episodes[1].sequence_number)
+ } else {
+ unimplemented!()
+ }
+ }
+ }
+ Ok(true)
+ },
+ |scope, locales| {
+ let subtitles = locales.into_iter().map(|l| l.to_string()).collect::>().join(", ");
+ match scope {
+ FilterMediaScope::Series(series) => warn!("Series {} is not available with {} subtitles", series.title, subtitles),
+ FilterMediaScope::Season(season) => warn!("Season {} is not available with {} subtitles", season.season_number, subtitles),
+ FilterMediaScope::Episode(episodes) => {
+ if episodes.len() == 1 {
+ warn!("Episode {} of season {} is not available with {} subtitles", episodes[0].sequence_number, episodes[0].season_title, subtitles)
+ } else if episodes.len() == 2 {
+ warn!("Season {} of season {} is only available with {} subtitles from episode {} to {}", episodes[0].season_number, episodes[0].season_title, subtitles, episodes[0].sequence_number, episodes[1].sequence_number)
+ } else {
+ unimplemented!()
+ }
+ }
+ }
+ Ok(true)
+ },
+ |season| {
+ warn!("Skipping premium episodes in season {season}");
+ Ok(())
+ },
+ Format::has_relative_fmt(&self.output),
+ !self.yes,
+ self.skip_specials,
+ ctx.crunchy.premium().await,
+ )
+ .visit(media_collection)
+ .await?;
+
+ if single_format_collection.is_empty() {
+ progress_handler.stop(format!("Skipping url {} (no matching videos found)", i + 1));
+ continue;
+ }
+ progress_handler.stop(format!("Loaded series information for url {}", i + 1));
+
+ single_format_collection.full_visual_output();
+
+ let download_builder =
+ DownloadBuilder::new(ctx.client.clone(), ctx.rate_limiter.clone())
+ .default_subtitle(self.default_subtitle.clone())
+ .download_fonts(self.include_fonts)
+ .ffmpeg_preset(self.ffmpeg_preset.clone().unwrap_or_default())
+ .ffmpeg_threads(self.ffmpeg_threads)
+ .output_format(Some("matroska".to_string()))
+ .audio_sort(Some(self.audio.clone()))
+ .subtitle_sort(Some(self.subtitle.clone()))
+ .no_closed_caption(self.no_closed_caption)
+ .merge_sync_tolerance(match self.merge {
+ MergeBehavior::Sync => Some(self.merge_sync_tolerance),
+ _ => None,
+ })
+ .merge_sync_precision(match self.merge {
+ MergeBehavior::Sync => Some(self.merge_sync_precision),
+ _ => None,
+ })
+ .threads(self.threads)
+ .audio_locale_output_map(
+ zip(self.audio.clone(), self.output_audio_locales.clone()).collect(),
+ )
+ .subtitle_locale_output_map(
+ zip(self.subtitle.clone(), self.output_subtitle_locales.clone()).collect(),
+ );
+
+ for single_formats in single_format_collection.into_iter() {
+ let (download_formats, mut format) = get_format(&self, &single_formats).await?;
+
+ let mut downloader = download_builder.clone().build();
+ for download_format in download_formats {
+ downloader.add_format(download_format)
+ }
+
+ let formatted_path = if format.is_special() {
+ format.format_path(
+ self.output_specials
+ .as_ref()
+ .map_or((&self.output).into(), |so| so.into()),
+ self.universal_output,
+ self.language_tagging.as_ref(),
+ )
+ } else {
+ format.format_path(
+ (&self.output).into(),
+ self.universal_output,
+ self.language_tagging.as_ref(),
+ )
+ };
+ let (mut path, changed) = free_file(formatted_path.clone());
+
+ if changed && self.skip_existing {
+ let mut skip = true;
+
+ if !self.skip_existing_method.is_empty() {
+ if let Some((audio_locales, subtitle_locales)) =
+ get_video_streams(&formatted_path)?
+ {
+ let method_audio = self
+ .skip_existing_method
+ .contains(&SkipExistingMethod::Audio);
+ let method_subtitle = self
+ .skip_existing_method
+ .contains(&SkipExistingMethod::Subtitle);
+
+ let audio_differ = if method_audio {
+ format
+ .locales
+ .iter()
+ .any(|(a, _)| !audio_locales.contains(a))
+ } else {
+ false
+ };
+ let subtitle_differ = if method_subtitle {
+ format
+ .locales
+ .clone()
+ .into_iter()
+ .flat_map(|(a, mut s)| {
+ // remove the closed caption if the flag is given to omit
+ // closed captions
+ if self.no_closed_caption && a != Locale::ja_JP {
+ s.retain(|l| l != &a)
+ }
+ s
+ })
+ .any(|l| !subtitle_locales.contains(&l))
+ } else {
+ false
+ };
+
+ if (method_audio && audio_differ)
+ || (method_subtitle && subtitle_differ)
+ {
+ skip = false;
+ path.clone_from(&formatted_path)
+ }
+ }
+ }
+
+ if skip {
+ debug!(
+ "Skipping already existing file '{}'",
+ formatted_path.to_string_lossy()
+ );
+ continue;
+ }
+ }
+
+ format.locales.sort_by(|(a, _), (b, _)| {
+ self.audio
+ .iter()
+ .position(|l| l == a)
+ .cmp(&self.audio.iter().position(|l| l == b))
+ });
+ for (_, subtitles) in format.locales.iter_mut() {
+ subtitles.sort_by(|a, b| {
+ self.subtitle
+ .iter()
+ .position(|l| l == a)
+ .cmp(&self.subtitle.iter().position(|l| l == b))
+ })
+ }
+
+ format.visual_output(&path);
+
+ downloader.download(&path).await?
+ }
+ }
+
+ Ok(())
+ }
+}
+
+#[derive(Clone, Debug, Eq, PartialEq)]
+pub(crate) enum SkipExistingMethod {
+ Audio,
+ Subtitle,
+}
+
+impl Display for SkipExistingMethod {
+ fn fmt(&self, f: &mut Formatter<'_>) -> std::fmt::Result {
+ let value = match self {
+ SkipExistingMethod::Audio => "audio",
+ SkipExistingMethod::Subtitle => "subtitle",
+ };
+ write!(f, "{}", value)
+ }
+}
+
+impl SkipExistingMethod {
+ fn parse(s: &str) -> Result {
+ match s.to_lowercase().as_str() {
+ "audio" => Ok(Self::Audio),
+ "subtitle" => Ok(Self::Subtitle),
+ _ => Err(format!("invalid skip existing method '{}'", s)),
+ }
+ }
+
+ fn default<'a>() -> &'a [Self] {
+ &[]
+ }
+}
+
+async fn get_format(
+ archive: &Archive,
+ single_formats: &Vec,
+) -> Result<(Vec, Format)> {
+ let mut format_pairs = vec![];
+ let mut single_format_to_format_pairs = vec![];
+
+ for single_format in single_formats {
+ let stream = single_format.stream().await?;
+ let Some((video, audio, _)) =
+ stream_data_from_stream(&stream, &archive.resolution, None).await?
+ else {
+ if single_format.is_episode() {
+ bail!(
+ "Resolution ({}) is not available for episode {} ({}) of {} season {}",
+ archive.resolution,
+ single_format.episode_number,
+ single_format.title,
+ single_format.series_name,
+ single_format.season_number,
+ )
+ } else {
+ bail!(
+ "Resolution ({}) is not available for {} ({})",
+ archive.resolution,
+ single_format.source_type(),
+ single_format.title
+ )
+ }
+ };
+
+ let subtitles: Vec<(Subtitle, bool)> = archive
+ .subtitle
+ .iter()
+ .flat_map(|s| {
+ let mut subtitles = vec![];
+ if let Some(caption) = stream.captions.get(s) {
+ subtitles.push((caption.clone(), true))
+ }
+ if let Some(subtitle) = stream.subtitles.get(s) {
+ // the subtitle is probably cc if the audio is not japanese or only one subtitle
+ // exists for this stream
+ let cc = single_format.audio != Locale::ja_JP && stream.subtitles.len() == 1;
+ // only include the subtitles if no cc subtitle is already present or if it's
+ // not cc
+ if subtitles.is_empty() || !cc {
+ subtitles.push((subtitle.clone(), cc))
+ }
+ }
+ subtitles
+ })
+ .collect();
+
+ format_pairs.push((single_format, video.clone(), audio, subtitles.clone()));
+ single_format_to_format_pairs.push((single_format.clone(), video, subtitles));
+
+ stream.invalidate().await?
+ }
+
+ let mut download_formats = vec![];
+
+ match archive.merge {
+ MergeBehavior::Video => {
+ for (single_format, video, audio, subtitles) in format_pairs {
+ download_formats.push(DownloadFormat {
+ video: (video, single_format.audio.clone()),
+ audios: vec![(audio, single_format.audio.clone())],
+ subtitles,
+ metadata: DownloadFormatMetadata { skip_events: None },
+ })
+ }
+ }
+ MergeBehavior::Audio => download_formats.push(DownloadFormat {
+ video: (
+ format_pairs.first().unwrap().1.clone(),
+ format_pairs.first().unwrap().0.audio.clone(),
+ ),
+ audios: format_pairs
+ .iter()
+ .map(|(single_format, _, audio, _)| (audio.clone(), single_format.audio.clone()))
+ .collect(),
+ // mix all subtitles together and then reduce them via a map so that only one subtitle
+ // per language exists
+ subtitles: format_pairs
+ .iter()
+ .flat_map(|(_, _, _, subtitles)| subtitles.clone())
+ .collect(),
+ metadata: DownloadFormatMetadata {
+ skip_events: if archive.include_chapters {
+ format_pairs.first().unwrap().0.skip_events().await?
+ } else {
+ None
+ },
+ },
+ }),
+ MergeBehavior::Auto | MergeBehavior::Sync => {
+ let mut d_formats: Vec<(Duration, DownloadFormat)> = vec![];
+
+ for (single_format, video, audio, subtitles) in format_pairs {
+ let closest_format = d_formats.iter_mut().min_by(|(x, _), (y, _)| {
+ x.sub(single_format.duration)
+ .abs()
+ .cmp(&y.sub(single_format.duration).abs())
+ });
+
+ match closest_format {
+ Some(closest_format)
+ if closest_format
+ .0
+ .sub(single_format.duration)
+ .abs()
+ .num_milliseconds()
+ < archive.merge_time_tolerance.into() =>
+ {
+ // If less than `audio_error` apart, use same audio.
+ closest_format
+ .1
+ .audios
+ .push((audio, single_format.audio.clone()));
+ closest_format.1.subtitles.extend(subtitles);
+ }
+ _ => {
+ d_formats.push((
+ single_format.duration,
+ DownloadFormat {
+ video: (video, single_format.audio.clone()),
+ audios: vec![(audio, single_format.audio.clone())],
+ subtitles,
+ metadata: DownloadFormatMetadata {
+ skip_events: if archive.include_chapters {
+ single_format.skip_events().await?
+ } else {
+ None
+ },
+ },
+ },
+ ));
+ }
+ };
+ }
+
+ for (_, d_format) in d_formats.into_iter() {
+ download_formats.push(d_format);
+ }
+ }
+ }
+
+ Ok((
+ download_formats,
+ Format::from_single_formats(single_format_to_format_pairs),
+ ))
+}
+
+fn get_video_streams(path: &Path) -> Result, Vec)>> {
+ let video_streams =
+ Regex::new(r"(?m)Stream\s#\d+:\d+\((?P.+)\):\s(?P(Audio|Subtitle))")
+ .unwrap();
+
+ let ffmpeg = Command::new("ffmpeg")
+ .stdout(Stdio::null())
+ .stderr(Stdio::piped())
+ .arg("-hide_banner")
+ .args(["-i", &path.to_string_lossy()])
+ .output()?;
+ let ffmpeg_output = String::from_utf8(ffmpeg.stderr)?;
+
+ let mut audio = vec![];
+ let mut subtitle = vec![];
+ for cap in video_streams.captures_iter(&ffmpeg_output) {
+ let locale = cap.name("language").unwrap().as_str();
+ let type_ = cap.name("type").unwrap().as_str();
+
+ match type_ {
+ "Audio" => audio.push(Locale::from(locale.to_string())),
+ "Subtitle" => subtitle.push(Locale::from(locale.to_string())),
+ _ => unreachable!(),
+ }
+ }
+
+ if audio.is_empty() && subtitle.is_empty() {
+ Ok(None)
+ } else {
+ Ok(Some((audio, subtitle)))
+ }
+}
diff --git a/crunchy-cli-core/src/archive/mod.rs b/crunchy-cli-core/src/archive/mod.rs
new file mode 100644
index 0000000..670d0c2
--- /dev/null
+++ b/crunchy-cli-core/src/archive/mod.rs
@@ -0,0 +1,3 @@
+mod command;
+
+pub use command::Archive;
diff --git a/crunchy-cli-core/src/download/command.rs b/crunchy-cli-core/src/download/command.rs
new file mode 100644
index 0000000..8e3794f
--- /dev/null
+++ b/crunchy-cli-core/src/download/command.rs
@@ -0,0 +1,483 @@
+use crate::utils::context::Context;
+use crate::utils::download::{DownloadBuilder, DownloadFormat, DownloadFormatMetadata};
+use crate::utils::ffmpeg::{FFmpegPreset, SOFTSUB_CONTAINERS};
+use crate::utils::filter::{Filter, FilterMediaScope};
+use crate::utils::format::{Format, SingleFormat};
+use crate::utils::locale::{resolve_locales, LanguageTagging};
+use crate::utils::log::progress;
+use crate::utils::os::{free_file, has_ffmpeg, is_special_file};
+use crate::utils::parse::parse_url;
+use crate::utils::video::stream_data_from_stream;
+use crate::Execute;
+use anyhow::bail;
+use anyhow::Result;
+use crunchyroll_rs::media::Resolution;
+use crunchyroll_rs::Locale;
+use log::{debug, error, warn};
+use std::collections::HashMap;
+use std::path::Path;
+
+#[derive(Clone, 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::>().join(", ")))]
+ #[arg(long_help = format!("Audio language. Can only be used if the provided url(s) point to a series. \
+ Available languages are:\n {}\nIETF tagged language codes for the shown available locales can be used too", Locale::all().into_iter().map(|l| format!("{:<6} → {}", l.to_string(), l.to_human_readable())).collect::>().join("\n ")))]
+ #[arg(short, long, default_value_t = crate::utils::locale::system_locale())]
+ pub(crate) audio: Locale,
+ #[arg(skip)]
+ output_audio_locale: String,
+ #[arg(help = format!("Subtitle language. Available languages are: {}", Locale::all().into_iter().map(|l| l.to_string()).collect::>().join(", ")))]
+ #[arg(long_help = format!("Subtitle language. If set, the subtitle will be burned into the video and cannot be disabled. \
+ Available languages are: {}\nIETF tagged language codes for the shown available locales can be used too", Locale::all().into_iter().map(|l| l.to_string()).collect::>().join(", ")))]
+ #[arg(short, long)]
+ pub(crate) subtitle: Option,
+ #[arg(skip)]
+ output_subtitle_locale: String,
+
+ #[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 \
+ {width} → Width of the video\n \
+ {height} → Height of the video\n \
+ {season_number} → Number of the season\n \
+ {episode_number} → Number of the episode\n \
+ {relative_episode_number} → Number of the episode relative to its season\n \
+ {sequence_number} → Like '{episode_number}' but without possible non-number characters\n \
+ {relative_sequence_number} → Like '{relative_episode_number}' but with support for episode 0's and .5's\n \
+ {release_year} → Release year of the video\n \
+ {release_month} → Release month of the video\n \
+ {release_day} → Release day of the video\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}.mp4")]
+ pub(crate) output: String,
+ #[arg(help = "Name of the output file if the episode is a special")]
+ #[arg(long_help = "Name of the output file if the episode is a special. \
+ If not set, the '-o'/'--output' flag will be used as name template")]
+ #[arg(long)]
+ pub(crate) output_specials: Option,
+
+ #[arg(help = "Sanitize the output file for use with all operating systems. \
+ This option only affects template options and not static characters.")]
+ #[arg(long, default_value_t = false)]
+ pub(crate) universal_output: bool,
+
+ #[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)]
+ pub(crate) resolution: Resolution,
+
+ #[arg(
+ long,
+ help = "Specified which language tagging the audio and subtitle tracks and language specific format options should have. \
+ Valid options are: 'default' (how Crunchyroll uses it internally), 'ietf' (according to the IETF standard)"
+ )]
+ #[arg(
+ long_help = "Specified which language tagging the audio and subtitle tracks and language specific format options should have. \
+ Valid options are: 'default' (how Crunchyroll uses it internally), 'ietf' (according to the IETF standard; you might run in issues as there are multiple locales which resolve to the same IETF language code, e.g. 'es-LA' and 'es-ES' are both resolving to 'es')"
+ )]
+ #[arg(value_parser = LanguageTagging::parse)]
+ pub(crate) language_tagging: Option,
+
+ #[arg(help = format!("Presets for converting the video to a specific coding format. \
+ Available presets: \n {}", FFmpegPreset::available_matches_human_readable().join("\n ")))]
+ #[arg(long_help = format!("Presets for converting the video to a specific coding format. \
+ If you need more specific ffmpeg customizations you can pass ffmpeg output arguments instead of a preset as value. \
+ Available presets: \n {}", FFmpegPreset::available_matches_human_readable().join("\n ")))]
+ #[arg(long)]
+ #[arg(value_parser = FFmpegPreset::parse)]
+ pub(crate) ffmpeg_preset: Option,
+ #[arg(
+ help = "The number of threads used by ffmpeg to generate the output file. Does not work with every codec/preset"
+ )]
+ #[arg(
+ long_help = "The number of threads used by ffmpeg to generate the output file. \
+ Does not work with every codec/preset and is skipped entirely when specifying custom ffmpeg output arguments instead of a preset for `--ffmpeg-preset`. \
+ By default, ffmpeg chooses the thread count which works best for the output codec"
+ )]
+ #[arg(long)]
+ pub(crate) ffmpeg_threads: Option,
+
+ #[arg(help = "Skip files which are already existing by their name")]
+ #[arg(long, default_value_t = false)]
+ pub(crate) skip_existing: bool,
+ #[arg(help = "Skip special episodes")]
+ #[arg(long, default_value_t = false)]
+ pub(crate) skip_specials: bool,
+
+ #[arg(help = "Includes chapters (e.g. intro, credits, ...)")]
+ #[arg(long_help = "Includes chapters (e.g. intro, credits, ...). \
+ Because chapters are essentially only special timeframes in episodes like the intro, most of the video timeline isn't covered by a chapter.
+ These \"gaps\" are filled with an 'Episode' chapter because many video players are ignore those gaps and just assume that a chapter ends when the next chapter start is reached, even if a specific end-time is set.
+ Also chapters aren't always available, so in this case, just a big 'Episode' chapter from start to end will be created")]
+ #[arg(long, default_value_t = false)]
+ pub(crate) include_chapters: bool,
+
+ #[arg(help = "Skip any interactive input")]
+ #[arg(short, long, default_value_t = false)]
+ pub(crate) yes: bool,
+
+ #[arg(help = "Force subtitles to be always burnt-in")]
+ #[arg(long, default_value_t = false)]
+ pub(crate) force_hardsub: bool,
+
+ #[arg(help = "The number of threads used to download")]
+ #[arg(short, long, default_value_t = num_cpus::get())]
+ pub(crate) threads: usize,
+
+ #[arg(help = "Url(s) to Crunchyroll episodes or series")]
+ #[arg(required = true)]
+ pub(crate) urls: Vec,
+}
+
+impl Execute for Download {
+ fn pre_check(&mut self) -> Result<()> {
+ if !has_ffmpeg() {
+ bail!("FFmpeg is needed to run this command")
+ } else if Path::new(&self.output)
+ .extension()
+ .unwrap_or_default()
+ .is_empty()
+ && !is_special_file(&self.output)
+ && self.output != "-"
+ {
+ bail!("No file extension found. Please specify a file extension (via `-o`) for the output file")
+ }
+
+ if self.subtitle.is_some() {
+ if let Some(ext) = Path::new(&self.output).extension() {
+ if self.force_hardsub {
+ warn!("Hardsubs are forced. Adding subtitles may take a while")
+ } else if !["mkv", "mov", "mp4"].contains(&ext.to_string_lossy().as_ref()) {
+ warn!("Detected a container which does not support softsubs. Adding subtitles may take a while")
+ }
+ }
+ }
+
+ if let Some(special_output) = &self.output_specials {
+ if Path::new(special_output)
+ .extension()
+ .unwrap_or_default()
+ .is_empty()
+ && !is_special_file(special_output)
+ && special_output != "-"
+ {
+ bail!("No file extension found. Please specify a file extension (via `--output-specials`) for the output file")
+ }
+ if let Some(ext) = Path::new(special_output).extension() {
+ if self.force_hardsub {
+ warn!("Hardsubs are forced for special episodes. Adding subtitles may take a while")
+ } else if !["mkv", "mov", "mp4"].contains(&ext.to_string_lossy().as_ref()) {
+ warn!("Detected a container which does not support softsubs. Adding subtitles for special episodes may take a while")
+ }
+ }
+ }
+
+ if let Some(language_tagging) = &self.language_tagging {
+ self.audio = resolve_locales(&[self.audio.clone()]).remove(0);
+ self.subtitle = self
+ .subtitle
+ .as_ref()
+ .map(|s| resolve_locales(&[s.clone()]).remove(0));
+ self.output_audio_locale = language_tagging.for_locale(&self.audio);
+ self.output_subtitle_locale = self
+ .subtitle
+ .as_ref()
+ .map(|s| language_tagging.for_locale(s))
+ .unwrap_or_default()
+ } else {
+ self.output_audio_locale = self.audio.to_string();
+ self.output_subtitle_locale = self
+ .subtitle
+ .as_ref()
+ .map(|s| s.to_string())
+ .unwrap_or_default();
+ }
+
+ Ok(())
+ }
+
+ async fn execute(self, ctx: Context) -> Result<()> {
+ if !ctx.crunchy.premium().await {
+ warn!("You may not be able to download all requested videos when logging in anonymously or using a non-premium account")
+ }
+
+ let mut parsed_urls = vec![];
+
+ let output_supports_softsubs = SOFTSUB_CONTAINERS.contains(
+ &Path::new(&self.output)
+ .extension()
+ .unwrap_or_default()
+ .to_string_lossy()
+ .as_ref(),
+ );
+ let special_output_supports_softsubs = if let Some(so) = &self.output_specials {
+ SOFTSUB_CONTAINERS.contains(
+ &Path::new(so)
+ .extension()
+ .unwrap_or_default()
+ .to_string_lossy()
+ .as_ref(),
+ )
+ } else {
+ output_supports_softsubs
+ };
+
+ for (i, url) in self.urls.clone().into_iter().enumerate() {
+ let progress_handler = progress!("Parsing url {}", i + 1);
+ match parse_url(&ctx.crunchy, url.clone(), true).await {
+ Ok((media_collection, url_filter)) => {
+ progress_handler.stop(format!("Parsed url {}", i + 1));
+ parsed_urls.push((media_collection, url_filter))
+ }
+ 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 single_format_collection = Filter::new(
+ url_filter,
+ vec![self.audio.clone()],
+ self.subtitle.as_ref().map_or(vec![], |s| vec![s.clone()]),
+ |scope, locales| {
+ match scope {
+ FilterMediaScope::Series(series) => bail!("Series {} is not available with {} audio", series.title, locales[0]),
+ FilterMediaScope::Season(season) => {
+ error!("Season {} is not available with {} audio", season.season_number, locales[0]);
+ Ok(false)
+ }
+ FilterMediaScope::Episode(episodes) => {
+ if episodes.len() == 1 {
+ warn!("Episode {} of season {} is not available with {} audio", episodes[0].sequence_number, episodes[0].season_title, locales[0])
+ } else if episodes.len() == 2 {
+ warn!("Season {} is only available with {} audio from episode {} to {}", episodes[0].season_number, locales[0], episodes[0].sequence_number, episodes[1].sequence_number)
+ } else {
+ unimplemented!()
+ }
+ Ok(false)
+ }
+ }
+ },
+ |scope, locales| {
+ match scope {
+ FilterMediaScope::Series(series) => bail!("Series {} is not available with {} subtitles", series.title, locales[0]),
+ FilterMediaScope::Season(season) => {
+ warn!("Season {} is not available with {} subtitles", season.season_number, locales[0]);
+ Ok(false)
+ },
+ FilterMediaScope::Episode(episodes) => {
+ if episodes.len() == 1 {
+ warn!("Episode {} of season {} is not available with {} subtitles", episodes[0].sequence_number, episodes[0].season_title, locales[0])
+ } else if episodes.len() == 2 {
+ warn!("Season {} is only available with {} subtitles from episode {} to {}", episodes[0].season_number, locales[0], episodes[0].sequence_number, episodes[1].sequence_number)
+ } else {
+ unimplemented!()
+ }
+ Ok(false)
+ }
+ }
+ },
+ |season| {
+ warn!("Skipping premium episodes in season {season}");
+ Ok(())
+ },
+ Format::has_relative_fmt(&self.output),
+ !self.yes,
+ self.skip_specials,
+ ctx.crunchy.premium().await,
+ )
+ .visit(media_collection)
+ .await?;
+
+ if single_format_collection.is_empty() {
+ progress_handler.stop(format!("Skipping url {} (no matching videos found)", i + 1));
+ continue;
+ }
+ progress_handler.stop(format!("Loaded series information for url {}", i + 1));
+
+ single_format_collection.full_visual_output();
+
+ let download_builder =
+ DownloadBuilder::new(ctx.client.clone(), ctx.rate_limiter.clone())
+ .default_subtitle(self.subtitle.clone())
+ .force_hardsub(self.force_hardsub)
+ .output_format(if is_special_file(&self.output) || self.output == "-" {
+ Some("mpegts".to_string())
+ } else {
+ None
+ })
+ .ffmpeg_preset(self.ffmpeg_preset.clone().unwrap_or_default())
+ .ffmpeg_threads(self.ffmpeg_threads)
+ .threads(self.threads)
+ .audio_locale_output_map(HashMap::from([(
+ self.audio.clone(),
+ self.output_audio_locale.clone(),
+ )]))
+ .subtitle_locale_output_map(
+ self.subtitle.as_ref().map_or(HashMap::new(), |s| {
+ HashMap::from([(s.clone(), self.output_subtitle_locale.clone())])
+ }),
+ );
+
+ for mut single_formats in single_format_collection.into_iter() {
+ // the vec contains always only one item
+ let single_format = single_formats.remove(0);
+
+ let (download_format, format) = get_format(
+ &self,
+ &single_format,
+ if self.force_hardsub {
+ true
+ } else if single_format.is_special() {
+ !special_output_supports_softsubs
+ } else {
+ !output_supports_softsubs
+ },
+ )
+ .await?;
+
+ let mut downloader = download_builder.clone().build();
+ downloader.add_format(download_format);
+
+ let formatted_path = if format.is_special() {
+ format.format_path(
+ self.output_specials
+ .as_ref()
+ .map_or((&self.output).into(), |so| so.into()),
+ self.universal_output,
+ self.language_tagging.as_ref(),
+ )
+ } else {
+ format.format_path(
+ (&self.output).into(),
+ self.universal_output,
+ self.language_tagging.as_ref(),
+ )
+ };
+ let (path, changed) = free_file(formatted_path.clone());
+
+ if changed && self.skip_existing {
+ debug!(
+ "Skipping already existing file '{}'",
+ formatted_path.to_string_lossy()
+ );
+ continue;
+ }
+
+ format.visual_output(&path);
+
+ downloader.download(&path).await?
+ }
+ }
+
+ Ok(())
+ }
+}
+
+async fn get_format(
+ download: &Download,
+ single_format: &SingleFormat,
+ try_peer_hardsubs: bool,
+) -> Result<(DownloadFormat, Format)> {
+ let stream = single_format.stream().await?;
+ let Some((video, audio, contains_hardsub)) = stream_data_from_stream(
+ &stream,
+ &download.resolution,
+ if try_peer_hardsubs {
+ download.subtitle.clone()
+ } else {
+ None
+ },
+ )
+ .await?
+ else {
+ if single_format.is_episode() {
+ bail!(
+ "Resolution ({}) is not available for episode {} ({}) of {} season {}",
+ download.resolution,
+ single_format.episode_number,
+ single_format.title,
+ single_format.series_name,
+ single_format.season_number,
+ )
+ } else {
+ bail!(
+ "Resolution ({}) is not available for {} ({})",
+ download.resolution,
+ single_format.source_type(),
+ single_format.title
+ )
+ }
+ };
+
+ let subtitle = if contains_hardsub {
+ None
+ } else if let Some(subtitle_locale) = &download.subtitle {
+ if download.audio == Locale::ja_JP {
+ stream
+ .subtitles
+ .get(subtitle_locale)
+ // use closed captions as fallback if no actual subtitles are found
+ .or_else(|| stream.captions.get(subtitle_locale))
+ .cloned()
+ } else {
+ stream
+ .captions
+ .get(subtitle_locale)
+ .or_else(|| stream.subtitles.get(subtitle_locale))
+ .cloned()
+ }
+ } else {
+ None
+ };
+
+ let download_format = DownloadFormat {
+ video: (video.clone(), single_format.audio.clone()),
+ audios: vec![(audio, single_format.audio.clone())],
+ subtitles: subtitle.clone().map_or(vec![], |s| {
+ vec![(
+ s,
+ single_format.audio != Locale::ja_JP && stream.subtitles.len() == 1,
+ )]
+ }),
+ metadata: DownloadFormatMetadata {
+ skip_events: if download.include_chapters {
+ single_format.skip_events().await?
+ } else {
+ None
+ },
+ },
+ };
+ let mut format = Format::from_single_formats(vec![(
+ single_format.clone(),
+ video,
+ subtitle.map_or(vec![], |s| {
+ vec![(
+ s,
+ single_format.audio != Locale::ja_JP && stream.subtitles.len() == 1,
+ )]
+ }),
+ )]);
+ if contains_hardsub {
+ let (_, subs) = format.locales.get_mut(0).unwrap();
+ subs.push(download.subtitle.clone().unwrap())
+ }
+
+ stream.invalidate().await?;
+
+ Ok((download_format, format))
+}
diff --git a/crunchy-cli-core/src/download/mod.rs b/crunchy-cli-core/src/download/mod.rs
new file mode 100644
index 0000000..47ca304
--- /dev/null
+++ b/crunchy-cli-core/src/download/mod.rs
@@ -0,0 +1,3 @@
+mod command;
+
+pub use command::Download;
diff --git a/crunchy-cli-core/src/lib.rs b/crunchy-cli-core/src/lib.rs
new file mode 100644
index 0000000..1c180e8
--- /dev/null
+++ b/crunchy-cli-core/src/lib.rs
@@ -0,0 +1,405 @@
+use crate::utils::context::Context;
+use crate::utils::locale::system_locale;
+use crate::utils::log::{progress, CliLogger};
+use anyhow::bail;
+use anyhow::Result;
+use clap::{Parser, Subcommand};
+use crunchyroll_rs::crunchyroll::CrunchyrollBuilder;
+use crunchyroll_rs::error::Error;
+use crunchyroll_rs::{Crunchyroll, Locale};
+use log::{debug, error, warn, LevelFilter};
+use reqwest::{Client, Proxy};
+use std::{env, fs};
+
+mod archive;
+mod download;
+mod login;
+mod search;
+mod utils;
+
+use crate::utils::rate_limit::RateLimiterService;
+pub use archive::Archive;
+use dialoguer::console::Term;
+pub use download::Download;
+pub use login::Login;
+pub use search::Search;
+
+trait Execute {
+ fn pre_check(&mut self) -> Result<()> {
+ Ok(())
+ }
+ async fn execute(self, ctx: Context) -> Result<()>;
+}
+
+#[derive(Debug, Parser)]
+#[clap(author, version = version(), about)]
+#[clap(name = "crunchy-cli")]
+pub struct Cli {
+ #[clap(flatten)]
+ verbosity: Verbosity,
+
+ #[arg(
+ help = "Overwrite the language in which results are returned. Default is your system language"
+ )]
+ #[arg(global = true, long)]
+ lang: Option,
+
+ #[arg(
+ help = "Enable experimental fixes which may resolve some unexpected errors. Generally not recommended as this flag may crash the program completely"
+ )]
+ #[arg(
+ long_help = "Enable experimental fixes which may resolve some unexpected errors. \
+ It is not recommended to use this this flag regularly, it might cause unexpected errors which may crash the program completely. \
+ If everything works as intended this option isn't needed, but sometimes Crunchyroll mislabels \
+ the audio of a series/season or episode or returns a wrong season number. This is when using this option might help to solve the issue"
+ )]
+ #[arg(global = true, long, default_value_t = false)]
+ experimental_fixes: bool,
+
+ #[clap(flatten)]
+ login_method: login::LoginMethod,
+
+ #[arg(help = "Use a proxy to route all traffic through")]
+ #[arg(long_help = "Use a proxy to route all traffic through. \
+ Make sure that the proxy can either forward TLS requests, which is needed to bypass the (cloudflare) bot protection, or that it is configured so that the proxy can bypass the protection itself. \
+ Besides specifying a simple url, you also can partially control where a proxy should be used: ':' only proxies api requests, ':' only proxies download traffic, ':' proxies api requests through the first url and download traffic through the second url")]
+ #[arg(global = true, long, value_parser = crate::utils::clap::clap_parse_proxies)]
+ proxy: Option<(Option, Option)>,
+
+ #[arg(help = "Use custom user agent")]
+ #[arg(global = true, long)]
+ user_agent: Option,
+
+ #[arg(
+ help = "Maximal speed to download/request (may be a bit off here and there). Must be in format of [B|KB|MB]"
+ )]
+ #[arg(
+ long_help = "Maximal speed to download/request (may be a bit off here and there). Must be in format of [B|KB|MB] (e.g. 500KB or 10MB)"
+ )]
+ #[arg(global = true, long, value_parser = crate::utils::clap::clap_parse_speed_limit)]
+ speed_limit: Option,
+
+ #[clap(subcommand)]
+ command: Command,
+}
+
+fn version() -> String {
+ let package_version = env!("CARGO_PKG_VERSION");
+ let git_commit_hash = env!("GIT_HASH");
+ let build_date = env!("BUILD_DATE");
+
+ if git_commit_hash.is_empty() {
+ package_version.to_string()
+ } else {
+ format!("{} ({} {})", package_version, git_commit_hash, build_date)
+ }
+}
+
+#[derive(Debug, Subcommand)]
+enum Command {
+ Archive(Archive),
+ Download(Download),
+ Login(Login),
+ Search(Search),
+}
+
+#[derive(Debug, Parser)]
+struct Verbosity {
+ #[arg(help = "Verbose output")]
+ #[arg(global = true, short, long)]
+ verbose: bool,
+
+ #[arg(help = "Quiet output. Does not print anything unless it's a error")]
+ #[arg(
+ long_help = "Quiet output. Does not print anything unless it's a error. Can be helpful if you pipe the output to stdout"
+ )]
+ #[arg(global = true, short, long)]
+ quiet: bool,
+}
+
+pub async fn main(args: &[String]) {
+ let mut cli: Cli = Cli::parse_from(args);
+
+ if cli.verbosity.verbose || cli.verbosity.quiet {
+ if cli.verbosity.verbose && cli.verbosity.quiet {
+ eprintln!("Output cannot be verbose ('-v') and quiet ('-q') at the same time");
+ std::process::exit(1)
+ } else if cli.verbosity.verbose {
+ CliLogger::init(LevelFilter::Debug).unwrap()
+ } else if cli.verbosity.quiet {
+ CliLogger::init(LevelFilter::Error).unwrap()
+ }
+ } else {
+ CliLogger::init(LevelFilter::Info).unwrap()
+ }
+
+ debug!("cli input: {:?}", cli);
+
+ match &mut cli.command {
+ Command::Archive(archive) => {
+ // prevent interactive select to be shown when output should be quiet
+ if cli.verbosity.quiet {
+ archive.yes = true;
+ }
+ pre_check_executor(archive).await
+ }
+ Command::Download(download) => {
+ // prevent interactive select to be shown when output should be quiet
+ if cli.verbosity.quiet {
+ download.yes = true;
+ }
+ pre_check_executor(download).await
+ }
+ Command::Login(login) => {
+ if login.remove {
+ if let Some(session_file) = login::session_file_path() {
+ let _ = fs::remove_file(session_file);
+ }
+ return;
+ } else {
+ pre_check_executor(login).await
+ }
+ }
+ Command::Search(search) => pre_check_executor(search).await,
+ };
+
+ let ctx = match create_ctx(&mut cli).await {
+ Ok(ctx) => ctx,
+ Err(e) => {
+ error!("{}", e);
+ std::process::exit(1)
+ }
+ };
+ debug!("Created context");
+
+ ctrlc::set_handler(move || {
+ debug!("Ctrl-c detected");
+ if let Ok(dir) = fs::read_dir(env::temp_dir()) {
+ for file in dir.flatten() {
+ if file
+ .path()
+ .file_name()
+ .unwrap_or_default()
+ .to_str()
+ .unwrap_or_default()
+ .starts_with(".crunchy-cli_")
+ {
+ if file.file_type().map_or(true, |ft| ft.is_file()) {
+ let result = fs::remove_file(file.path());
+ debug!(
+ "Ctrl-c removed temporary file {} {}",
+ file.path().to_string_lossy(),
+ if result.is_ok() {
+ "successfully"
+ } else {
+ "not successfully"
+ }
+ )
+ } else {
+ let result = fs::remove_dir_all(file.path());
+ debug!(
+ "Ctrl-c removed temporary directory {} {}",
+ file.path().to_string_lossy(),
+ if result.is_ok() {
+ "successfully"
+ } else {
+ "not successfully"
+ }
+ )
+ }
+ }
+ }
+ }
+ // when pressing ctrl-c while interactively choosing seasons the cursor stays hidden, this
+ // line shows it again
+ let _ = Term::stdout().show_cursor();
+ std::process::exit(1)
+ })
+ .unwrap();
+ debug!("Created ctrl-c handler");
+
+ match cli.command {
+ Command::Archive(archive) => execute_executor(archive, ctx).await,
+ Command::Download(download) => execute_executor(download, ctx).await,
+ Command::Login(login) => execute_executor(login, ctx).await,
+ Command::Search(search) => execute_executor(search, ctx).await,
+ };
+}
+
+async fn pre_check_executor(executor: &mut impl Execute) {
+ if let Err(err) = executor.pre_check() {
+ error!("Misconfigurations detected: {}", err);
+ std::process::exit(1)
+ }
+}
+
+async fn execute_executor(executor: impl Execute, ctx: Context) {
+ if let Err(mut err) = executor.execute(ctx).await {
+ if let Some(crunchy_error) = err.downcast_mut::() {
+ if let Error::Block { message, .. } = crunchy_error {
+ *message = "Triggered Cloudflare bot protection. Try again later or use a VPN or proxy to spoof your location".to_string()
+ }
+
+ error!("An error occurred: {}", crunchy_error)
+ } else {
+ error!("An error occurred: {}", err)
+ }
+
+ std::process::exit(1)
+ }
+}
+
+async fn create_ctx(cli: &mut Cli) -> Result {
+ let crunchy_client = reqwest_client(
+ cli.proxy.as_ref().and_then(|p| p.0.clone()),
+ cli.user_agent.clone(),
+ );
+ let internal_client = reqwest_client(
+ cli.proxy.as_ref().and_then(|p| p.1.clone()),
+ cli.user_agent.clone(),
+ );
+
+ let crunchy = crunchyroll_session(
+ cli,
+ crunchy_client.clone(),
+ cli.speed_limit
+ .map(|l| RateLimiterService::new(l, crunchy_client)),
+ )
+ .await?;
+
+ Ok(Context {
+ crunchy,
+ client: internal_client.clone(),
+ rate_limiter: cli
+ .speed_limit
+ .map(|l| RateLimiterService::new(l, internal_client)),
+ })
+}
+
+async fn crunchyroll_session(
+ cli: &mut Cli,
+ client: Client,
+ rate_limiter: Option,
+) -> Result {
+ let supported_langs = vec![
+ Locale::ar_ME,
+ Locale::de_DE,
+ Locale::en_US,
+ Locale::es_ES,
+ Locale::es_419,
+ Locale::fr_FR,
+ Locale::it_IT,
+ Locale::pt_BR,
+ Locale::pt_PT,
+ Locale::ru_RU,
+ ];
+ let locale = if let Some(lang) = &cli.lang {
+ if !supported_langs.contains(lang) {
+ bail!(
+ "Via `--lang` specified language is not supported. Supported languages: {}",
+ supported_langs
+ .iter()
+ .map(|l| format!("`{}` ({})", l, l.to_human_readable()))
+ .collect::>()
+ .join(", ")
+ )
+ }
+ lang.clone()
+ } else {
+ let mut lang = system_locale();
+ if !supported_langs.contains(&lang) {
+ warn!("Recognized system locale is not supported. Using en-US as default. Use `--lang` to overwrite the used language");
+ lang = Locale::en_US
+ }
+ lang
+ };
+
+ let mut builder = Crunchyroll::builder()
+ .locale(locale)
+ .client(client.clone())
+ .stabilization_locales(cli.experimental_fixes)
+ .stabilization_season_number(cli.experimental_fixes);
+ if let Command::Download(download) = &cli.command {
+ builder = builder.preferred_audio_locale(download.audio.clone())
+ }
+ if let Some(rate_limiter) = rate_limiter {
+ builder = builder.middleware(rate_limiter)
+ }
+
+ let root_login_methods_count =
+ cli.login_method.credentials.is_some() as u8 + cli.login_method.anonymous as u8;
+
+ let progress_handler = progress!("Logging in");
+ if root_login_methods_count == 0 {
+ if let Some(login_file_path) = login::session_file_path() {
+ if login_file_path.exists() {
+ let session = fs::read_to_string(login_file_path)?;
+ if let Some((token_type, token)) = session.split_once(':') {
+ match token_type {
+ "refresh_token" => {
+ return match builder.login_with_refresh_token(token).await {
+ Ok(crunchy) => Ok(crunchy),
+ Err(e) => {
+ if let Error::Request { message, .. } = &e {
+ if message.starts_with("invalid_grant") {
+ bail!("The stored login is expired, please login again")
+ }
+ }
+ Err(e.into())
+ }
+ }
+ }
+ "etp_rt" => bail!("The stored login method (etp-rt) isn't supported anymore. Please login again using your credentials"),
+ _ => (),
+ }
+ }
+ bail!("Could not read stored session ('{}')", session)
+ }
+ }
+ bail!("Please use a login method ('--credentials' or '--anonymous')")
+ } else if root_login_methods_count > 1 {
+ bail!("Please use only one login method ('--credentials' or '--anonymous')")
+ }
+
+ let crunchy = if let Some(credentials) = &cli.login_method.credentials {
+ if let Some((email, password)) = credentials.split_once(':') {
+ builder.login_with_credentials(email, password).await?
+ } else {
+ bail!("Invalid credentials format. Please provide your credentials as email:password")
+ }
+ } else if cli.login_method.anonymous {
+ builder.login_anonymously().await?
+ } else {
+ bail!("should never happen")
+ };
+
+ progress_handler.stop("Logged in");
+
+ Ok(crunchy)
+}
+
+fn reqwest_client(proxy: Option, user_agent: Option) -> Client {
+ let mut builder = CrunchyrollBuilder::predefined_client_builder();
+ if let Some(p) = proxy {
+ builder = builder.proxy(p)
+ }
+ if let Some(ua) = user_agent {
+ builder = builder.user_agent(ua)
+ }
+
+ #[cfg(any(feature = "openssl-tls", feature = "openssl-tls-static"))]
+ let client = {
+ let mut builder = builder.use_native_tls().tls_built_in_root_certs(false);
+
+ for certificate in rustls_native_certs::load_native_certs().unwrap() {
+ builder =
+ builder.add_root_certificate(reqwest::Certificate::from_der(&certificate).unwrap())
+ }
+
+ builder.build().unwrap()
+ };
+ #[cfg(not(any(feature = "openssl-tls", feature = "openssl-tls-static")))]
+ let client = builder.build().unwrap();
+
+ client
+}
diff --git a/crunchy-cli-core/src/login/command.rs b/crunchy-cli-core/src/login/command.rs
new file mode 100644
index 0000000..4da1898
--- /dev/null
+++ b/crunchy-cli-core/src/login/command.rs
@@ -0,0 +1,55 @@
+use crate::utils::context::Context;
+use crate::Execute;
+use anyhow::bail;
+use anyhow::Result;
+use clap::Parser;
+use crunchyroll_rs::crunchyroll::SessionToken;
+use log::info;
+use std::fs;
+use std::path::PathBuf;
+
+#[derive(Debug, clap::Parser)]
+#[clap(about = "Save your login credentials persistent on disk")]
+pub struct Login {
+ #[arg(help = "Remove your stored credentials (instead of saving them)")]
+ #[arg(long)]
+ pub remove: bool,
+}
+
+impl Execute for Login {
+ async fn execute(self, ctx: Context) -> Result<()> {
+ if let Some(login_file_path) = session_file_path() {
+ fs::create_dir_all(login_file_path.parent().unwrap())?;
+
+ match ctx.crunchy.session_token().await {
+ SessionToken::RefreshToken(refresh_token) => {
+ fs::write(login_file_path, format!("refresh_token:{}", refresh_token))?
+ }
+ SessionToken::EtpRt(_) => bail!("Login with etp_rt isn't supported anymore. Please use your credentials to login"),
+ SessionToken::Anonymous => bail!("Anonymous login cannot be saved"),
+ }
+
+ info!("Saved login");
+
+ Ok(())
+ } else {
+ bail!("Cannot find config path")
+ }
+ }
+}
+
+#[derive(Clone, Debug, Parser)]
+pub struct LoginMethod {
+ #[arg(
+ help = "Login with credentials (email and password). Must be provided as email:password"
+ )]
+ #[arg(global = true, long)]
+ pub credentials: Option,
+ #[arg(help = "Login anonymously / without an account")]
+ #[arg(global = true, long, default_value_t = false)]
+ pub anonymous: bool,
+}
+
+pub fn session_file_path() -> Option {
+ dirs::config_dir().map(|config_dir| config_dir.join("crunchy-cli").join("session"))
+}
diff --git a/crunchy-cli-core/src/login/mod.rs b/crunchy-cli-core/src/login/mod.rs
new file mode 100644
index 0000000..8c1220a
--- /dev/null
+++ b/crunchy-cli-core/src/login/mod.rs
@@ -0,0 +1,3 @@
+mod command;
+
+pub use command::{session_file_path, Login, LoginMethod};
diff --git a/crunchy-cli-core/src/search/command.rs b/crunchy-cli-core/src/search/command.rs
new file mode 100644
index 0000000..8032bed
--- /dev/null
+++ b/crunchy-cli-core/src/search/command.rs
@@ -0,0 +1,222 @@
+use crate::search::filter::FilterOptions;
+use crate::search::format::Format;
+use crate::utils::context::Context;
+use crate::utils::parse::{parse_url, UrlFilter};
+use crate::Execute;
+use anyhow::{bail, Result};
+use crunchyroll_rs::common::StreamExt;
+use crunchyroll_rs::search::QueryResults;
+use crunchyroll_rs::{Episode, Locale, MediaCollection, MovieListing, MusicVideo, Series};
+use log::warn;
+use std::sync::Arc;
+
+#[derive(Debug, clap::Parser)]
+#[clap(about = "Search in videos")]
+#[command(arg_required_else_help(true))]
+pub struct Search {
+ #[arg(help = format!("Audio languages to include. \
+ Available languages are: {}", Locale::all().into_iter().map(|l| l.to_string()).collect::>().join(", ")))]
+ #[arg(long_help = format!("Audio languages to include. \
+ Available languages are:\n {}", Locale::all().into_iter().map(|l| format!("{:<6} → {}", l.to_string(), l.to_human_readable())).collect::>().join("\n ")))]
+ #[arg(long, default_values_t = vec![crate::utils::locale::system_locale()])]
+ audio: Vec,
+
+ #[arg(help = "Limit of search top search results")]
+ #[arg(long, default_value_t = 5)]
+ search_top_results_limit: u32,
+ #[arg(help = "Limit of search series results")]
+ #[arg(long, default_value_t = 0)]
+ search_series_limit: u32,
+ #[arg(help = "Limit of search movie listing results")]
+ #[arg(long, default_value_t = 0)]
+ search_movie_listing_limit: u32,
+ #[arg(help = "Limit of search episode results")]
+ #[arg(long, default_value_t = 0)]
+ search_episode_limit: u32,
+ #[arg(help = "Limit of search music results")]
+ #[arg(long, default_value_t = 0)]
+ search_music_limit: u32,
+
+ /// Format of the output text.
+ ///
+ /// You can specify keywords in a specific pattern and they will get replaced in the output text.
+ /// The required pattern for this begins with `{{`, then the keyword, and closes with `}}` (e.g. `{{episode.title}}`).
+ /// For example, if you want to get the title of an episode, you can use `Title {{episode.title}}` and `{{episode.title}}` will be replaced with the episode title
+ ///
+ /// See the following list for all keywords and their meaning:
+ /// series.id → Series id
+ /// series.title → Series title
+ /// series.description → Series description
+ /// series.release_year → Series release year
+ ///
+ /// season.id → Season id
+ /// season.title → Season title
+ /// season.description → Season description
+ /// season.number → Season number
+ /// season.episodes → Number of episodes the season has
+ ///
+ /// episode.id → Episode id
+ /// episode.title → Episode title
+ /// episode.description → Episode description
+ /// episode.locale → Episode locale/language
+ /// episode.number → Episode number
+ /// episode.sequence_number → Episode number. This number is unique unlike `episode.number` which sometimes can be duplicated
+ /// episode.duration → Episode duration in milliseconds
+ /// episode.air_date → Episode air date as unix timestamp
+ /// episode.premium_only → If the episode is only available with Crunchyroll premium
+ ///
+ /// movie_listing.id → Movie listing id
+ /// movie_listing.title → Movie listing title
+ /// movie_listing.description → Movie listing description
+ ///
+ /// movie.id → Movie id
+ /// movie.title → Movie title
+ /// movie.description → Movie description
+ /// movie.duration → Movie duration in milliseconds
+ /// movie.premium_only → If the movie is only available with Crunchyroll premium
+ ///
+ /// music_video.id → Music video id
+ /// music_video.title → Music video title
+ /// music_video.description → Music video description
+ /// music_video.duration → Music video duration in milliseconds
+ /// music_video.premium_only → If the music video is only available with Crunchyroll premium
+ ///
+ /// concert.id → Concert id
+ /// concert.title → Concert title
+ /// concert.description → Concert description
+ /// concert.duration → Concert duration in milliseconds
+ /// concert.premium_only → If the concert is only available with Crunchyroll premium
+ ///
+ /// stream.locale → Stream locale/language
+ /// stream.dash_url → Stream url in DASH format. You need to set the `Authorization` header to `Bearer ` when requesting this url
+ /// stream.is_drm → If `stream.dash_url` is DRM encrypted
+ ///
+ /// subtitle.locale → Subtitle locale/language
+ /// subtitle.url → Url to the subtitle
+ ///
+ /// account.token → Access token to make request to restricted endpoints. This token is only valid for a max. of 5 minutes
+ /// account.id → Internal ID of the user account
+ /// account.profile_name → Profile name of the account
+ /// account.email → Email address of the account
+ #[arg(short, long, verbatim_doc_comment)]
+ #[arg(default_value = "S{{season.number}}E{{episode.number}} - {{episode.title}}")]
+ output: String,
+
+ input: String,
+}
+
+impl Execute for Search {
+ async fn execute(self, ctx: Context) -> Result<()> {
+ if !ctx.crunchy.premium().await {
+ warn!("Using `search` anonymously or with a non-premium account may return incomplete results")
+ }
+
+ if self.output.contains("{{stream.is_drm}}") {
+ warn!("The `{{{{stream.is_drm}}}}` option is deprecated as it isn't reliable anymore and will be removed soon")
+ }
+
+ let input = if crunchyroll_rs::parse::parse_url(&self.input).is_some() {
+ match parse_url(&ctx.crunchy, self.input.clone(), true).await {
+ Ok(ok) => vec![ok],
+ Err(e) => bail!("url {} could not be parsed: {}", self.input, e),
+ }
+ } else {
+ let mut output = vec![];
+
+ let query = resolve_query(&self, ctx.crunchy.query(&self.input)).await?;
+ output.extend(query.0.into_iter().map(|m| (m, UrlFilter::default())));
+ output.extend(
+ query
+ .1
+ .into_iter()
+ .map(|s| (s.into(), UrlFilter::default())),
+ );
+ output.extend(
+ query
+ .2
+ .into_iter()
+ .map(|m| (m.into(), UrlFilter::default())),
+ );
+ output.extend(
+ query
+ .3
+ .into_iter()
+ .map(|e| (e.into(), UrlFilter::default())),
+ );
+ output.extend(
+ query
+ .4
+ .into_iter()
+ .map(|m| (m.into(), UrlFilter::default())),
+ );
+
+ output
+ };
+
+ let crunchy_arc = Arc::new(ctx.crunchy);
+ for (media_collection, url_filter) in input {
+ let filter_options = FilterOptions {
+ audio: self.audio.clone(),
+ url_filter,
+ };
+
+ let format = Format::new(self.output.clone(), filter_options, crunchy_arc.clone())?;
+ println!("{}", format.parse(media_collection).await?);
+ }
+
+ Ok(())
+ }
+}
+
+macro_rules! resolve_query {
+ ($limit:expr, $vec:expr, $item:expr) => {
+ if $limit > 0 {
+ let mut item_results = $item;
+ while let Some(item) = item_results.next().await {
+ $vec.push(item?);
+ if $vec.len() >= $limit as usize {
+ break;
+ }
+ }
+ }
+ };
+}
+
+async fn resolve_query(
+ search: &Search,
+ query_results: QueryResults,
+) -> Result<(
+ Vec,
+ Vec,
+ Vec,
+ Vec,
+ Vec,
+)> {
+ let mut media_collection = vec![];
+ let mut series = vec![];
+ let mut movie_listing = vec![];
+ let mut episode = vec![];
+ let mut music_video = vec![];
+
+ resolve_query!(
+ search.search_top_results_limit,
+ media_collection,
+ query_results.top_results
+ );
+ resolve_query!(search.search_series_limit, series, query_results.series);
+ resolve_query!(
+ search.search_movie_listing_limit,
+ movie_listing,
+ query_results.movie_listing
+ );
+ resolve_query!(search.search_episode_limit, episode, query_results.episode);
+ resolve_query!(search.search_music_limit, music_video, query_results.music);
+
+ Ok((
+ media_collection,
+ series,
+ movie_listing,
+ episode,
+ music_video,
+ ))
+}
diff --git a/crunchy-cli-core/src/search/filter.rs b/crunchy-cli-core/src/search/filter.rs
new file mode 100644
index 0000000..3bb6d9f
--- /dev/null
+++ b/crunchy-cli-core/src/search/filter.rs
@@ -0,0 +1,47 @@
+use crate::utils::parse::UrlFilter;
+use crunchyroll_rs::{Episode, Locale, MovieListing, Season, Series};
+
+pub struct FilterOptions {
+ pub audio: Vec,
+ pub url_filter: UrlFilter,
+}
+
+impl FilterOptions {
+ pub fn check_series(&self, series: &Series) -> bool {
+ self.check_audio_language(&series.audio_locales)
+ }
+
+ pub fn filter_seasons(&self, mut seasons: Vec) -> Vec {
+ seasons.retain(|s| {
+ self.check_audio_language(&s.audio_locales)
+ && self.url_filter.is_season_valid(s.season_number)
+ });
+ seasons
+ }
+
+ pub fn filter_episodes(&self, mut episodes: Vec) -> Vec {
+ episodes.retain(|e| {
+ self.check_audio_language(&[e.audio_locale.clone()])
+ && self
+ .url_filter
+ .is_episode_valid(e.sequence_number, e.season_number)
+ });
+ episodes
+ }
+
+ pub fn check_movie_listing(&self, movie_listing: &MovieListing) -> bool {
+ self.check_audio_language(
+ &movie_listing
+ .audio_locale
+ .clone()
+ .map_or(vec![], |a| vec![a.clone()]),
+ )
+ }
+
+ fn check_audio_language(&self, audio: &[Locale]) -> bool {
+ if !self.audio.is_empty() {
+ return self.audio.iter().any(|a| audio.contains(a));
+ }
+ true
+ }
+}
diff --git a/crunchy-cli-core/src/search/format.rs b/crunchy-cli-core/src/search/format.rs
new file mode 100644
index 0000000..cf3c5bc
--- /dev/null
+++ b/crunchy-cli-core/src/search/format.rs
@@ -0,0 +1,687 @@
+use crate::search::filter::FilterOptions;
+use anyhow::{bail, Result};
+use crunchyroll_rs::media::{Stream, Subtitle};
+use crunchyroll_rs::{
+ Concert, Crunchyroll, Episode, Locale, MediaCollection, Movie, MovieListing, MusicVideo,
+ Season, Series,
+};
+use regex::Regex;
+use serde::Serialize;
+use serde_json::{Map, Value};
+use std::collections::HashMap;
+use std::ops::Range;
+use std::sync::Arc;
+
+#[derive(Default, Serialize)]
+struct FormatSeries {
+ pub id: String,
+ pub title: String,
+ pub description: String,
+ pub release_year: u32,
+}
+
+impl From<&Series> for FormatSeries {
+ fn from(value: &Series) -> Self {
+ Self {
+ id: value.id.clone(),
+ title: value.title.clone(),
+ description: value.description.clone(),
+ release_year: value.series_launch_year.unwrap_or_default(),
+ }
+ }
+}
+
+#[derive(Default, Serialize)]
+struct FormatSeason {
+ pub id: String,
+ pub title: String,
+ pub description: String,
+ pub number: u32,
+ pub episodes: u32,
+}
+
+impl From<&Season> for FormatSeason {
+ fn from(value: &Season) -> Self {
+ Self {
+ id: value.id.clone(),
+ title: value.title.clone(),
+ description: value.description.clone(),
+ number: value.season_number,
+ episodes: value.number_of_episodes,
+ }
+ }
+}
+
+#[derive(Default, Serialize)]
+struct FormatEpisode {
+ pub id: String,
+ pub title: String,
+ pub description: String,
+ pub locale: Locale,
+ pub number: u32,
+ pub sequence_number: f32,
+ pub duration: i64,
+ pub air_date: i64,
+ pub premium_only: bool,
+}
+
+impl From<&Episode> for FormatEpisode {
+ fn from(value: &Episode) -> Self {
+ Self {
+ id: value.id.clone(),
+ title: value.title.clone(),
+ description: value.description.clone(),
+ locale: value.audio_locale.clone(),
+ number: value.episode_number.unwrap_or_default(),
+ sequence_number: value.sequence_number,
+ duration: value.duration.num_milliseconds(),
+ air_date: value.episode_air_date.timestamp(),
+ premium_only: value.is_premium_only,
+ }
+ }
+}
+
+#[derive(Default, Serialize)]
+struct FormatMovieListing {
+ pub id: String,
+ pub title: String,
+ pub description: String,
+}
+
+impl From<&MovieListing> for FormatMovieListing {
+ fn from(value: &MovieListing) -> Self {
+ Self {
+ id: value.id.clone(),
+ title: value.title.clone(),
+ description: value.description.clone(),
+ }
+ }
+}
+
+#[derive(Default, Serialize)]
+struct FormatMovie {
+ pub id: String,
+ pub title: String,
+ pub description: String,
+ pub duration: i64,
+ pub premium_only: bool,
+}
+
+impl From<&Movie> for FormatMovie {
+ fn from(value: &Movie) -> Self {
+ Self {
+ id: value.id.clone(),
+ title: value.title.clone(),
+ description: value.description.clone(),
+ duration: value.duration.num_milliseconds(),
+ premium_only: value.is_premium_only,
+ }
+ }
+}
+
+#[derive(Default, Serialize)]
+struct FormatMusicVideo {
+ pub id: String,
+ pub title: String,
+ pub description: String,
+ pub duration: i64,
+ pub premium_only: bool,
+}
+
+impl From<&MusicVideo> for FormatMusicVideo {
+ fn from(value: &MusicVideo) -> Self {
+ Self {
+ id: value.id.clone(),
+ title: value.title.clone(),
+ description: value.description.clone(),
+ duration: value.duration.num_milliseconds(),
+ premium_only: value.is_premium_only,
+ }
+ }
+}
+
+#[derive(Default, Serialize)]
+struct FormatConcert {
+ pub id: String,
+ pub title: String,
+ pub description: String,
+ pub duration: i64,
+ pub premium_only: bool,
+}
+
+impl From<&Concert> for FormatConcert {
+ fn from(value: &Concert) -> Self {
+ Self {
+ id: value.id.clone(),
+ title: value.title.clone(),
+ description: value.description.clone(),
+ duration: value.duration.num_milliseconds(),
+ premium_only: value.is_premium_only,
+ }
+ }
+}
+
+#[derive(Default, Serialize)]
+struct FormatStream {
+ pub locale: Locale,
+ pub dash_url: String,
+ pub is_drm: bool,
+}
+
+impl From<&Stream> for FormatStream {
+ fn from(value: &Stream) -> Self {
+ Self {
+ locale: value.audio_locale.clone(),
+ dash_url: value.url.clone(),
+ is_drm: false,
+ }
+ }
+}
+
+#[derive(Default, Serialize)]
+struct FormatSubtitle {
+ pub locale: Locale,
+ pub url: String,
+}
+
+impl From<&Subtitle> for FormatSubtitle {
+ fn from(value: &Subtitle) -> Self {
+ Self {
+ locale: value.locale.clone(),
+ url: value.url.clone(),
+ }
+ }
+}
+
+#[derive(Default, Serialize)]
+struct FormatAccount {
+ pub token: String,
+ pub id: String,
+ pub profile_name: String,
+ pub email: String,
+}
+
+impl FormatAccount {
+ pub async fn async_from(value: &Crunchyroll) -> Result {
+ let account = value.account().await?;
+
+ Ok(Self {
+ token: value.access_token().await,
+ id: account.account_id,
+ profile_name: account.profile_name,
+ email: account.email,
+ })
+ }
+}
+
+#[derive(Clone, Debug, Eq, PartialEq, Hash)]
+enum Scope {
+ Series,
+ Season,
+ Episode,
+ MovieListing,
+ Movie,
+ MusicVideo,
+ Concert,
+ Stream,
+ Subtitle,
+ Account,
+}
+
+macro_rules! must_match_if_true {
+ ($condition:expr => $media_collection:ident | $field:pat => $expr:expr) => {
+ if $condition {
+ match &$media_collection {
+ $field => Some($expr),
+ _ => panic!(),
+ }
+ } else {
+ None
+ }
+ };
+}
+
+pub struct Format {
+ pattern: Vec<(Range, Scope, String)>,
+ pattern_count: HashMap,
+ input: String,
+ filter_options: FilterOptions,
+ crunchyroll: Arc,
+}
+
+impl Format {
+ pub fn new(
+ input: String,
+ filter_options: FilterOptions,
+ crunchyroll: Arc,
+ ) -> Result {
+ let scope_regex = Regex::new(r"(?m)\{\{\s*(?P\w+)\.(?P\w+)\s*}}").unwrap();
+ let mut pattern = vec![];
+ let mut pattern_count = HashMap::new();
+
+ macro_rules! generate_field_check {
+ ($($scope:expr => $struct_:ident)+) => {
+ HashMap::from([
+ $(
+ (
+ $scope,
+ serde_json::from_value::>(serde_json::to_value($struct_::default()).unwrap()).unwrap()
+ )
+ ),+
+ ])
+ };
+ }
+ let field_check = generate_field_check!(
+ Scope::Series => FormatSeries
+ Scope::Season => FormatSeason
+ Scope::Episode => FormatEpisode
+ Scope::MovieListing => FormatMovieListing
+ Scope::Movie => FormatMovie
+ Scope::MusicVideo => FormatMusicVideo
+ Scope::Concert => FormatConcert
+ Scope::Stream => FormatStream
+ Scope::Subtitle => FormatSubtitle
+ Scope::Account => FormatAccount
+ );
+
+ for capture in scope_regex.captures_iter(&input) {
+ let full = capture.get(0).unwrap();
+ let scope = capture.name("scope").unwrap().as_str();
+ let field = capture.name("field").unwrap().as_str();
+
+ let format_pattern_scope = match scope {
+ "series" => Scope::Series,
+ "season" => Scope::Season,
+ "episode" => Scope::Episode,
+ "movie_listing" => Scope::MovieListing,
+ "movie" => Scope::Movie,
+ "music_video" => Scope::MusicVideo,
+ "concert" => Scope::Concert,
+ "stream" => Scope::Stream,
+ "subtitle" => Scope::Subtitle,
+ "account" => Scope::Account,
+ _ => bail!("'{}.{}' is not a valid keyword", scope, field),
+ };
+
+ if field_check
+ .get(&format_pattern_scope)
+ .unwrap()
+ .get(field)
+ .is_none()
+ {
+ bail!("'{}.{}' is not a valid keyword", scope, field)
+ }
+
+ pattern.push((
+ full.start()..full.end(),
+ format_pattern_scope.clone(),
+ field.to_string(),
+ ));
+ *pattern_count.entry(format_pattern_scope).or_default() += 1
+ }
+
+ Ok(Self {
+ pattern,
+ pattern_count,
+ input,
+ filter_options,
+ crunchyroll,
+ })
+ }
+
+ pub async fn parse(&self, media_collection: MediaCollection) -> Result {
+ match &media_collection {
+ MediaCollection::Series(_)
+ | MediaCollection::Season(_)
+ | MediaCollection::Episode(_) => {
+ self.check_scopes(vec![
+ Scope::Series,
+ Scope::Season,
+ Scope::Episode,
+ Scope::Stream,
+ Scope::Subtitle,
+ Scope::Account,
+ ])?;
+
+ self.parse_series(media_collection).await
+ }
+ MediaCollection::MovieListing(_) | MediaCollection::Movie(_) => {
+ self.check_scopes(vec![
+ Scope::MovieListing,
+ Scope::Movie,
+ Scope::Stream,
+ Scope::Subtitle,
+ Scope::Account,
+ ])?;
+
+ self.parse_movie_listing(media_collection).await
+ }
+ MediaCollection::MusicVideo(_) => {
+ self.check_scopes(vec![
+ Scope::MusicVideo,
+ Scope::Stream,
+ Scope::Subtitle,
+ Scope::Account,
+ ])?;
+
+ self.parse_music_video(media_collection).await
+ }
+ MediaCollection::Concert(_) => {
+ self.check_scopes(vec![
+ Scope::Concert,
+ Scope::Stream,
+ Scope::Subtitle,
+ Scope::Account,
+ ])?;
+
+ self.parse_concert(media_collection).await
+ }
+ }
+ }
+
+ async fn parse_series(&self, media_collection: MediaCollection) -> Result {
+ let series_empty = self.check_pattern_count_empty(Scope::Series);
+ let season_empty = self.check_pattern_count_empty(Scope::Season);
+ let episode_empty = self.check_pattern_count_empty(Scope::Episode);
+ let stream_empty = self.check_pattern_count_empty(Scope::Stream)
+ && self.check_pattern_count_empty(Scope::Subtitle);
+ let account_empty = self.check_pattern_count_empty(Scope::Account);
+
+ #[allow(clippy::type_complexity)]
+ let mut tree: Vec<(Season, Vec<(Episode, Vec)>)> = vec![];
+
+ let series = if !series_empty {
+ let series = match &media_collection {
+ MediaCollection::Series(series) => series.clone(),
+ MediaCollection::Season(season) => season.series().await?,
+ MediaCollection::Episode(episode) => episode.series().await?,
+ _ => panic!(),
+ };
+ if !self.filter_options.check_series(&series) {
+ return Ok("".to_string());
+ }
+ series
+ } else {
+ Series::default()
+ };
+ if !season_empty || !episode_empty || !stream_empty {
+ let tmp_seasons = match &media_collection {
+ MediaCollection::Series(series) => series.seasons().await?,
+ MediaCollection::Season(season) => vec![season.clone()],
+ MediaCollection::Episode(_) => vec![],
+ _ => panic!(),
+ };
+ let mut seasons = vec![];
+ for season in tmp_seasons {
+ seasons.push(season.clone());
+ for version in season.versions {
+ if season.id == version.id {
+ continue;
+ }
+ if self.filter_options.audio.contains(&version.audio_locale) {
+ seasons.push(version.season().await?)
+ }
+ }
+ }
+ tree.extend(
+ self.filter_options
+ .filter_seasons(seasons)
+ .into_iter()
+ .map(|s| (s, vec![])),
+ )
+ } else {
+ tree.push((Season::default(), vec![]))
+ }
+ if !episode_empty || !stream_empty {
+ match &media_collection {
+ MediaCollection::Episode(episode) => {
+ let mut episodes = vec![episode.clone()];
+ for version in &episode.versions {
+ if episode.id == version.id {
+ continue;
+ }
+ if self.filter_options.audio.contains(&version.audio_locale) {
+ episodes.push(version.episode().await?)
+ }
+ }
+ tree.push((
+ Season::default(),
+ episodes
+ .into_iter()
+ .filter(|e| self.filter_options.audio.contains(&e.audio_locale))
+ .map(|e| (e, vec![]))
+ .collect(),
+ ))
+ }
+ _ => {
+ for (season, episodes) in tree.iter_mut() {
+ episodes.extend(
+ self.filter_options
+ .filter_episodes(season.episodes().await?)
+ .into_iter()
+ .map(|e| (e, vec![])),
+ )
+ }
+ }
+ };
+ } else {
+ for (_, episodes) in tree.iter_mut() {
+ episodes.push((Episode::default(), vec![]))
+ }
+ }
+ if !stream_empty {
+ for (_, episodes) in tree.iter_mut() {
+ for (episode, streams) in episodes {
+ let stream = episode.stream_maybe_without_drm().await?;
+ stream.clone().invalidate().await?;
+ streams.push(stream)
+ }
+ }
+ } else {
+ for (_, episodes) in tree.iter_mut() {
+ for (_, streams) in episodes {
+ streams.push(Stream::default())
+ }
+ }
+ }
+
+ let mut output = vec![];
+ let account_map = if !account_empty {
+ self.serializable_to_json_map(FormatAccount::async_from(&self.crunchyroll).await?)
+ } else {
+ Map::default()
+ };
+ let series_map = self.serializable_to_json_map(FormatSeries::from(&series));
+ for (season, episodes) in tree {
+ let season_map = self.serializable_to_json_map(FormatSeason::from(&season));
+ for (episode, streams) in episodes {
+ let episode_map = self.serializable_to_json_map(FormatEpisode::from(&episode));
+ for stream in streams {
+ let stream_map = self.serializable_to_json_map(FormatStream::from(&stream));
+
+ output.push(
+ self.replace_all(
+ HashMap::from([
+ (Scope::Account, &account_map),
+ (Scope::Series, &series_map),
+ (Scope::Season, &season_map),
+ (Scope::Episode, &episode_map),
+ (Scope::Stream, &stream_map),
+ ]),
+ stream,
+ )
+ .unwrap_or_default(),
+ )
+ }
+ }
+ }
+
+ Ok(output.join("\n"))
+ }
+
+ async fn parse_movie_listing(&self, media_collection: MediaCollection) -> Result {
+ let movie_listing_empty = self.check_pattern_count_empty(Scope::MovieListing);
+ let movie_empty = self.check_pattern_count_empty(Scope::Movie);
+ let stream_empty = self.check_pattern_count_empty(Scope::Stream);
+
+ let mut tree: Vec<(Movie, Vec)> = vec![];
+
+ let movie_listing = if !movie_listing_empty {
+ let movie_listing = match &media_collection {
+ MediaCollection::MovieListing(movie_listing) => movie_listing.clone(),
+ MediaCollection::Movie(movie) => movie.movie_listing().await?,
+ _ => panic!(),
+ };
+ if !self.filter_options.check_movie_listing(&movie_listing) {
+ return Ok("".to_string());
+ }
+ movie_listing
+ } else {
+ MovieListing::default()
+ };
+ if !movie_empty || !stream_empty {
+ let movies = match &media_collection {
+ MediaCollection::MovieListing(movie_listing) => movie_listing.movies().await?,
+ MediaCollection::Movie(movie) => vec![movie.clone()],
+ _ => panic!(),
+ };
+ tree.extend(movies.into_iter().map(|m| (m, vec![])))
+ }
+ if !stream_empty {
+ for (movie, streams) in tree.iter_mut() {
+ streams.push(movie.stream_maybe_without_drm().await?)
+ }
+ } else {
+ for (_, streams) in tree.iter_mut() {
+ streams.push(Stream::default())
+ }
+ }
+
+ let mut output = vec![];
+ let movie_listing_map =
+ self.serializable_to_json_map(FormatMovieListing::from(&movie_listing));
+ for (movie, streams) in tree {
+ let movie_map = self.serializable_to_json_map(FormatMovie::from(&movie));
+ for stream in streams {
+ let stream_map = self.serializable_to_json_map(FormatStream::from(&stream));
+
+ output.push(
+ self.replace_all(
+ HashMap::from([
+ (Scope::MovieListing, &movie_listing_map),
+ (Scope::Movie, &movie_map),
+ (Scope::Stream, &stream_map),
+ ]),
+ stream,
+ )
+ .unwrap_or_default(),
+ )
+ }
+ }
+
+ Ok(output.join("\n"))
+ }
+
+ async fn parse_music_video(&self, media_collection: MediaCollection) -> Result {
+ let music_video_empty = self.check_pattern_count_empty(Scope::MusicVideo);
+ let stream_empty = self.check_pattern_count_empty(Scope::Stream);
+
+ let music_video = must_match_if_true!(!music_video_empty => media_collection|MediaCollection::MusicVideo(music_video) => music_video.clone()).unwrap_or_default();
+ let stream = must_match_if_true!(!stream_empty => media_collection|MediaCollection::MusicVideo(music_video) => music_video.stream_maybe_without_drm().await?).unwrap_or_default();
+
+ let music_video_map = self.serializable_to_json_map(FormatMusicVideo::from(&music_video));
+ let stream_map = self.serializable_to_json_map(FormatStream::from(&stream));
+
+ let output = self
+ .replace_all(
+ HashMap::from([
+ (Scope::MusicVideo, &music_video_map),
+ (Scope::Stream, &stream_map),
+ ]),
+ stream,
+ )
+ .unwrap_or_default();
+ Ok(output)
+ }
+
+ async fn parse_concert(&self, media_collection: MediaCollection) -> Result {
+ let concert_empty = self.check_pattern_count_empty(Scope::Concert);
+ let stream_empty = self.check_pattern_count_empty(Scope::Stream);
+
+ let concert = must_match_if_true!(!concert_empty => media_collection|MediaCollection::Concert(concert) => concert.clone()).unwrap_or_default();
+ let stream = must_match_if_true!(!stream_empty => media_collection|MediaCollection::Concert(concert) => concert.stream_maybe_without_drm().await?).unwrap_or_default();
+
+ let concert_map = self.serializable_to_json_map(FormatConcert::from(&concert));
+ let stream_map = self.serializable_to_json_map(FormatStream::from(&stream));
+
+ let output = self
+ .replace_all(
+ HashMap::from([(Scope::Concert, &concert_map), (Scope::Stream, &stream_map)]),
+ stream,
+ )
+ .unwrap_or_default();
+ Ok(output)
+ }
+
+ fn serializable_to_json_map(&self, s: S) -> Map {
+ serde_json::from_value(serde_json::to_value(s).unwrap()).unwrap()
+ }
+
+ fn check_pattern_count_empty(&self, scope: Scope) -> bool {
+ self.pattern_count.get(&scope).cloned().unwrap_or_default() == 0
+ }
+
+ fn check_scopes(&self, available_scopes: Vec) -> Result<()> {
+ for (_, scope, field) in self.pattern.iter() {
+ if !available_scopes.contains(scope) {
+ bail!(
+ "'{}.{}' is not a valid keyword",
+ format!("{:?}", scope).to_lowercase(),
+ field
+ )
+ }
+ }
+ Ok(())
+ }
+
+ fn replace_all(
+ &self,
+ values: HashMap>,
+ mut stream: Stream,
+ ) -> Option {
+ if stream.subtitles.is_empty() {
+ if !self.check_pattern_count_empty(Scope::Subtitle) {
+ return None;
+ }
+ stream
+ .subtitles
+ .insert(Locale::Custom("".to_string()), Subtitle::default());
+ }
+
+ let mut output = vec![];
+ for (_, subtitle) in stream.subtitles {
+ let subtitle_map = self.serializable_to_json_map(FormatSubtitle::from(&subtitle));
+ let mut tmp_values = values.clone();
+ tmp_values.insert(Scope::Subtitle, &subtitle_map);
+ output.push(self.replace(tmp_values))
+ }
+
+ Some(output.join("\n"))
+ }
+
+ fn replace(&self, values: HashMap>) -> String {
+ let mut output = self.input.clone();
+ let mut offset = 0;
+ for (range, scope, field) in &self.pattern {
+ let item =
+ serde_plain::to_string(values.get(scope).unwrap().get(field.as_str()).unwrap())
+ .unwrap();
+ let start = (range.start as i32 + offset) as usize;
+ let end = (range.end as i32 + offset) as usize;
+ output.replace_range(start..end, &item);
+ offset += item.len() as i32 - range.len() as i32;
+ }
+
+ output
+ }
+}
diff --git a/crunchy-cli-core/src/search/mod.rs b/crunchy-cli-core/src/search/mod.rs
new file mode 100644
index 0000000..839c844
--- /dev/null
+++ b/crunchy-cli-core/src/search/mod.rs
@@ -0,0 +1,5 @@
+mod command;
+mod filter;
+mod format;
+
+pub use command::Search;
diff --git a/crunchy-cli-core/src/utils/clap.rs b/crunchy-cli-core/src/utils/clap.rs
new file mode 100644
index 0000000..35de71f
--- /dev/null
+++ b/crunchy-cli-core/src/utils/clap.rs
@@ -0,0 +1,61 @@
+use crate::utils::parse::parse_resolution;
+use crunchyroll_rs::media::Resolution;
+use regex::Regex;
+use reqwest::Proxy;
+
+pub fn clap_parse_resolution(s: &str) -> Result {
+ parse_resolution(s.to_string()).map_err(|e| e.to_string())
+}
+
+pub fn clap_parse_proxies(s: &str) -> Result<(Option, Option), String> {
+ let double_proxy_regex =
+ Regex::new(r"^(?P(https?|socks5h?)://.+):(?P(https?|socks5h?)://.+)$")
+ .unwrap();
+
+ if let Some(capture) = double_proxy_regex.captures(s) {
+ // checks if the input is formatted like 'https://example.com:socks5://examples.com' and
+ // splits the string into 2 separate proxies at the middle colon
+
+ let first = capture.name("first").unwrap().as_str();
+ let second = capture.name("second").unwrap().as_str();
+ Ok((
+ Some(Proxy::all(first).map_err(|e| format!("first proxy: {e}"))?),
+ Some(Proxy::all(second).map_err(|e| format!("second proxy: {e}"))?),
+ ))
+ } else if s.starts_with(':') {
+ // checks if the input is formatted like ':https://example.com' and returns a proxy on the
+ // second tuple position
+ Ok((
+ None,
+ Some(Proxy::all(s.trim_start_matches(':')).map_err(|e| e.to_string())?),
+ ))
+ } else if s.ends_with(':') {
+ // checks if the input is formatted like 'https://example.com:' and returns a proxy on the
+ // first tuple position
+ Ok((
+ Some(Proxy::all(s.trim_end_matches(':')).map_err(|e| e.to_string())?),
+ None,
+ ))
+ } else {
+ // returns the same proxy for both tuple positions
+ let proxy = Proxy::all(s).map_err(|e| e.to_string())?;
+ Ok((Some(proxy.clone()), Some(proxy)))
+ }
+}
+
+pub fn clap_parse_speed_limit(s: &str) -> Result {
+ let quota = s.to_lowercase();
+
+ let bytes = if let Ok(b) = quota.parse() {
+ b
+ } else if let Ok(b) = quota.trim_end_matches('b').parse::() {
+ b
+ } else if let Ok(kb) = quota.trim_end_matches("kb").parse::() {
+ kb * 1024
+ } else if let Ok(mb) = quota.trim_end_matches("mb").parse::() {
+ mb * 1024 * 1024
+ } else {
+ return Err("Invalid speed limit".to_string());
+ };
+ Ok(bytes)
+}
diff --git a/crunchy-cli-core/src/utils/context.rs b/crunchy-cli-core/src/utils/context.rs
new file mode 100644
index 0000000..693174d
--- /dev/null
+++ b/crunchy-cli-core/src/utils/context.rs
@@ -0,0 +1,9 @@
+use crate::utils::rate_limit::RateLimiterService;
+use crunchyroll_rs::Crunchyroll;
+use reqwest::Client;
+
+pub struct Context {
+ pub crunchy: Crunchyroll,
+ pub client: Client,
+ pub rate_limiter: Option,
+}
diff --git a/crunchy-cli-core/src/utils/download.rs b/crunchy-cli-core/src/utils/download.rs
new file mode 100644
index 0000000..2e8f321
--- /dev/null
+++ b/crunchy-cli-core/src/utils/download.rs
@@ -0,0 +1,1453 @@
+use crate::utils::ffmpeg::FFmpegPreset;
+use crate::utils::filter::real_dedup_vec;
+use crate::utils::fmt::format_time_delta;
+use crate::utils::log::progress;
+use crate::utils::os::{cache_dir, is_special_file, temp_directory, temp_named_pipe, tempfile};
+use crate::utils::rate_limit::RateLimiterService;
+use crate::utils::sync::{sync_audios, SyncAudio};
+use anyhow::{bail, Result};
+use chrono::{NaiveTime, TimeDelta};
+use crunchyroll_rs::media::{SkipEvents, SkipEventsEvent, StreamData, StreamSegment, Subtitle};
+use crunchyroll_rs::Locale;
+use indicatif::{ProgressBar, ProgressDrawTarget, ProgressFinish, ProgressStyle};
+use log::{debug, warn, LevelFilter};
+use regex::Regex;
+use reqwest::Client;
+use rsubs_lib::{SSA, VTT};
+use std::borrow::Borrow;
+use std::cmp::Ordering;
+use std::collections::{BTreeMap, HashMap};
+use std::io::Write;
+use std::ops::Add;
+use std::path::{Path, PathBuf};
+use std::process::{Command, Stdio};
+use std::sync::Arc;
+use std::time::Duration;
+use std::{env, fs};
+use tempfile::TempPath;
+use time::Time;
+use tokio::io::{AsyncBufReadExt, AsyncReadExt, BufReader};
+use tokio::select;
+use tokio::sync::mpsc::unbounded_channel;
+use tokio::sync::Mutex;
+use tokio::task::JoinSet;
+use tokio_util::sync::CancellationToken;
+use tower_service::Service;
+
+#[derive(Clone, Debug)]
+pub enum MergeBehavior {
+ Video,
+ Audio,
+ Auto,
+ Sync,
+}
+
+impl MergeBehavior {
+ pub fn parse(s: &str) -> Result {
+ Ok(match s.to_lowercase().as_str() {
+ "video" => MergeBehavior::Video,
+ "audio" => MergeBehavior::Audio,
+ "auto" => MergeBehavior::Auto,
+ "sync" => MergeBehavior::Sync,
+ _ => return Err(format!("'{}' is not a valid merge behavior", s)),
+ })
+ }
+}
+
+#[derive(Clone, derive_setters::Setters)]
+pub struct DownloadBuilder {
+ client: Client,
+ rate_limiter: Option,
+ ffmpeg_preset: FFmpegPreset,
+ default_subtitle: Option,
+ output_format: Option,
+ audio_sort: Option>,
+ subtitle_sort: Option>,
+ force_hardsub: bool,
+ download_fonts: bool,
+ no_closed_caption: bool,
+ merge_sync_tolerance: Option,
+ merge_sync_precision: Option,
+ threads: usize,
+ ffmpeg_threads: Option,
+ audio_locale_output_map: HashMap,
+ subtitle_locale_output_map: HashMap,
+}
+
+impl DownloadBuilder {
+ pub fn new(client: Client, rate_limiter: Option) -> DownloadBuilder {
+ Self {
+ client,
+ rate_limiter,
+ ffmpeg_preset: FFmpegPreset::default(),
+ default_subtitle: None,
+ output_format: None,
+ audio_sort: None,
+ subtitle_sort: None,
+ force_hardsub: false,
+ download_fonts: false,
+ no_closed_caption: false,
+ merge_sync_tolerance: None,
+ merge_sync_precision: None,
+ threads: num_cpus::get(),
+ ffmpeg_threads: None,
+ audio_locale_output_map: HashMap::new(),
+ subtitle_locale_output_map: HashMap::new(),
+ }
+ }
+
+ pub fn build(self) -> Downloader {
+ Downloader {
+ client: self.client,
+ rate_limiter: self.rate_limiter,
+ ffmpeg_preset: self.ffmpeg_preset,
+ default_subtitle: self.default_subtitle,
+ output_format: self.output_format,
+ audio_sort: self.audio_sort,
+ subtitle_sort: self.subtitle_sort,
+
+ force_hardsub: self.force_hardsub,
+ download_fonts: self.download_fonts,
+ no_closed_caption: self.no_closed_caption,
+
+ merge_sync_tolerance: self.merge_sync_tolerance,
+ merge_sync_precision: self.merge_sync_precision,
+
+ download_threads: self.threads,
+ ffmpeg_threads: self.ffmpeg_threads,
+
+ formats: vec![],
+
+ audio_locale_output_map: self.audio_locale_output_map,
+ subtitle_locale_output_map: self.subtitle_locale_output_map,
+ }
+ }
+}
+
+struct FFmpegVideoMeta {
+ path: TempPath,
+ length: TimeDelta,
+ start_time: Option,
+}
+
+struct FFmpegAudioMeta {
+ path: TempPath,
+ locale: Locale,
+ start_time: Option,
+ video_idx: usize,
+}
+
+struct FFmpegSubtitleMeta {
+ path: TempPath,
+ locale: Locale,
+ cc: bool,
+ start_time: Option,
+ video_idx: usize,
+}
+
+pub struct DownloadFormat {
+ pub video: (StreamData, Locale),
+ pub audios: Vec<(StreamData, Locale)>,
+ pub subtitles: Vec<(Subtitle, bool)>,
+ pub metadata: DownloadFormatMetadata,
+}
+
+pub struct DownloadFormatMetadata {
+ pub skip_events: Option,
+}
+
+pub struct Downloader {
+ client: Client,
+ rate_limiter: Option,
+
+ ffmpeg_preset: FFmpegPreset,
+ default_subtitle: Option,
+ output_format: Option,
+ audio_sort: Option>,
+ subtitle_sort: Option>,
+
+ force_hardsub: bool,
+ download_fonts: bool,
+ no_closed_caption: bool,
+
+ merge_sync_tolerance: Option,
+ merge_sync_precision: Option,
+
+ download_threads: usize,
+ ffmpeg_threads: Option,
+
+ formats: Vec,
+
+ audio_locale_output_map: HashMap,
+ subtitle_locale_output_map: HashMap,
+}
+
+impl Downloader {
+ pub fn add_format(&mut self, format: DownloadFormat) {
+ self.formats.push(format);
+ }
+
+ pub async fn download(mut self, dst: &Path) -> Result<()> {
+ // `.unwrap_or_default()` here unless https://doc.rust-lang.org/stable/std/path/fn.absolute.html
+ // gets stabilized as the function might throw error on weird file paths
+ let required = self.check_free_space(dst).await.unwrap_or_default();
+ if let Some((path, tmp_required)) = &required.0 {
+ let kb = (*tmp_required as f64) / 1024.0;
+ let mb = kb / 1024.0;
+ let gb = mb / 1024.0;
+ warn!(
+ "You may have not enough disk space to store temporary files. The temp directory ({}) should have at least {}{} free space",
+ path.to_string_lossy(),
+ if gb < 1.0 { mb.ceil().to_string() } else { format!("{:.2}", gb) },
+ if gb < 1.0 { "MB" } else { "GB" }
+ )
+ }
+ if let Some((path, dst_required)) = &required.1 {
+ let kb = (*dst_required as f64) / 1024.0;
+ let mb = kb / 1024.0;
+ let gb = mb / 1024.0;
+ warn!(
+ "You may have not enough disk space to store the output file. The directory {} should have at least {}{} free space",
+ path.to_string_lossy(),
+ if gb < 1.0 { mb.ceil().to_string() } else { format!("{:.2}", gb) },
+ if gb < 1.0 { "MB" } else { "GB" }
+ )
+ }
+
+ if let Some(audio_sort_locales) = &self.audio_sort {
+ self.formats.sort_by(|a, b| {
+ audio_sort_locales
+ .iter()
+ .position(|l| l == &a.video.1)
+ .cmp(&audio_sort_locales.iter().position(|l| l == &b.video.1))
+ });
+ }
+ for format in self.formats.iter_mut() {
+ if let Some(audio_sort_locales) = &self.audio_sort {
+ format.audios.sort_by(|(_, a), (_, b)| {
+ audio_sort_locales
+ .iter()
+ .position(|l| l == a)
+ .cmp(&audio_sort_locales.iter().position(|l| l == b))
+ })
+ }
+ if let Some(subtitle_sort) = &self.subtitle_sort {
+ format
+ .subtitles
+ .sort_by(|(a_subtitle, a_cc), (b_subtitle, b_cc)| {
+ let ordering = subtitle_sort
+ .iter()
+ .position(|l| l == &a_subtitle.locale)
+ .cmp(&subtitle_sort.iter().position(|l| l == &b_subtitle.locale));
+ if matches!(ordering, Ordering::Equal) {
+ a_cc.cmp(b_cc).reverse()
+ } else {
+ ordering
+ }
+ })
+ }
+ }
+
+ let mut video_offset = None;
+ let mut audio_offsets = HashMap::new();
+ let mut subtitle_offsets = HashMap::new();
+ let mut raw_audios = vec![];
+ let mut videos = vec![];
+ let mut audios = vec![];
+ let mut subtitles = vec![];
+ let mut fonts = vec![];
+ let mut chapters = None;
+ let mut max_len = TimeDelta::min_value();
+ let mut max_frames = 0;
+ let fmt_space = self
+ .formats
+ .iter()
+ .flat_map(|f| {
+ f.audios
+ .iter()
+ .map(|(_, locale)| format!("Downloading {} audio", locale).len())
+ })
+ .max()
+ .unwrap();
+
+ // downloads all audios
+ for (i, format) in self.formats.iter().enumerate() {
+ for (stream_data, locale) in &format.audios {
+ let path = self
+ .download_audio(
+ stream_data,
+ format!("{:<1$}", format!("Downloading {} audio", locale), fmt_space),
+ )
+ .await?;
+ raw_audios.push(SyncAudio {
+ format_id: i,
+ path,
+ locale: locale.clone(),
+ sample_rate: stream_data.sampling_rate().unwrap(),
+ video_idx: i,
+ })
+ }
+ }
+
+ if self.formats.len() > 1 && self.merge_sync_tolerance.is_some() {
+ let _progress_handler =
+ progress!("Syncing video start times (this might take some time)");
+ let mut offsets = sync_audios(
+ &raw_audios,
+ self.merge_sync_tolerance.unwrap(),
+ self.merge_sync_precision.unwrap(),
+ )?;
+ drop(_progress_handler);
+
+ let mut offset_pre_checked = false;
+ if let Some(tmp_offsets) = &offsets {
+ let formats_with_offset: Vec = self
+ .formats
+ .iter()
+ .enumerate()
+ .map(|(i, f)| {
+ len_from_segments(&f.video.0.segments())
+ - tmp_offsets.get(&i).copied().unwrap_or_default()
+ })
+ .collect();
+ let min = formats_with_offset.iter().min().unwrap();
+ let max = formats_with_offset.iter().max().unwrap();
+
+ if max.num_seconds() - min.num_seconds() > 15 {
+ warn!("Found difference of >15 seconds after sync, so the application was skipped");
+ offsets = None;
+ offset_pre_checked = true
+ }
+ }
+
+ if let Some(offsets) = offsets {
+ let mut root_format_idx = 0;
+ let mut root_format_offset = u64::MAX;
+
+ for (i, format) in self.formats.iter().enumerate() {
+ let offset = offsets.get(&i).copied().unwrap_or_default();
+ let format_offset = offset.num_milliseconds() as u64;
+ if format_offset < root_format_offset {
+ root_format_idx = i;
+ root_format_offset = format_offset;
+ }
+
+ for _ in &format.audios {
+ if let Some(offset) = &offsets.get(&i) {
+ audio_offsets.insert(i, **offset);
+ }
+ }
+ for _ in &format.subtitles {
+ if let Some(offset) = &offsets.get(&i) {
+ subtitle_offsets.insert(i, **offset);
+ }
+ }
+ }
+
+ let mut root_format = self.formats.remove(root_format_idx);
+
+ let mut audio_prepend = vec![];
+ let mut subtitle_prepend = vec![];
+ let mut audio_append = vec![];
+ let mut subtitle_append = vec![];
+ for (i, format) in self.formats.into_iter().enumerate() {
+ if i < root_format_idx {
+ audio_prepend.extend(format.audios);
+ subtitle_prepend.extend(format.subtitles);
+ } else {
+ audio_append.extend(format.audios);
+ subtitle_append.extend(format.subtitles);
+ }
+ }
+ root_format.audios.splice(0..0, audio_prepend);
+ root_format.subtitles.splice(0..0, subtitle_prepend);
+ root_format.audios.extend(audio_append);
+ root_format.subtitles.extend(subtitle_append);
+
+ self.formats = vec![root_format];
+ video_offset = offsets.get(&root_format_idx).copied();
+ for raw_audio in raw_audios.iter_mut() {
+ raw_audio.video_idx = root_format_idx;
+ }
+ } else {
+ for format in &mut self.formats {
+ format.metadata.skip_events = None
+ }
+ if !offset_pre_checked {
+ warn!("Couldn't find reliable sync positions")
+ }
+ }
+ }
+
+ // add audio metadata
+ for raw_audio in raw_audios {
+ audios.push(FFmpegAudioMeta {
+ path: raw_audio.path,
+ locale: raw_audio.locale,
+ start_time: audio_offsets.get(&raw_audio.format_id).copied(),
+ video_idx: raw_audio.video_idx,
+ })
+ }
+
+ // downloads all videos
+ for (i, format) in self.formats.iter().enumerate() {
+ let path = self
+ .download_video(
+ &format.video.0,
+ format!("{:<1$}", format!("Downloading video #{}", i + 1), fmt_space),
+ None,
+ )
+ .await?;
+
+ let (len, fps) = get_video_stats(&path)?;
+ if max_len < len {
+ max_len = len
+ }
+ let frames = ((len.num_milliseconds() as f64
+ - video_offset.unwrap_or_default().num_milliseconds() as f64)
+ / 1000.0
+ * fps) as u64;
+ if max_frames < frames {
+ max_frames = frames
+ }
+
+ videos.push(FFmpegVideoMeta {
+ path,
+ length: len,
+ start_time: video_offset,
+ })
+ }
+
+ for (i, format) in self.formats.iter().enumerate() {
+ if format.subtitles.is_empty() {
+ continue;
+ }
+
+ let progress_spinner = if log::max_level() == LevelFilter::Info {
+ let progress_spinner = ProgressBar::new_spinner()
+ .with_style(
+ ProgressStyle::with_template(
+ format!(
+ ":: {:<1$} {{msg}} {{spinner}}",
+ "Downloading subtitles", fmt_space
+ )
+ .as_str(),
+ )
+ .unwrap()
+ .tick_strings(&["—", "\\", "|", "/", ""]),
+ )
+ .with_finish(ProgressFinish::Abandon);
+ progress_spinner.enable_steady_tick(Duration::from_millis(100));
+ Some(progress_spinner)
+ } else {
+ None
+ };
+
+ for (j, (subtitle, cc)) in format.subtitles.iter().enumerate() {
+ if *cc && self.no_closed_caption {
+ continue;
+ }
+
+ if let Some(pb) = &progress_spinner {
+ let mut progress_message = pb.message();
+ if !progress_message.is_empty() {
+ progress_message += ", "
+ }
+ progress_message += &subtitle.locale.to_string();
+ if *cc {
+ progress_message += " (CC)";
+ }
+ if i.min(videos.len() - 1) != 0 {
+ progress_message += &format!(" [Video: #{}]", i + 1);
+ }
+ pb.set_message(progress_message)
+ }
+
+ let path = self
+ .download_subtitle(subtitle.clone(), videos[i.min(videos.len() - 1)].length)
+ .await?;
+ debug!(
+ "Downloaded {} subtitles{}",
+ subtitle.locale,
+ cc.then_some(" (cc)").unwrap_or_default(),
+ );
+ subtitles.push(FFmpegSubtitleMeta {
+ path,
+ locale: subtitle.locale.clone(),
+ cc: *cc,
+ start_time: subtitle_offsets.get(&j).cloned(),
+ video_idx: i,
+ })
+ }
+ }
+
+ for format in self.formats.iter() {
+ if let Some(skip_events) = &format.metadata.skip_events {
+ let (file, path) = tempfile(".chapter")?.into_parts();
+ chapters = Some((
+ (file, path),
+ [
+ skip_events.recap.as_ref().map(|e| ("Recap", e)),
+ skip_events.intro.as_ref().map(|e| ("Intro", e)),
+ skip_events.credits.as_ref().map(|e| ("Credits", e)),
+ skip_events.preview.as_ref().map(|e| ("Preview", e)),
+ ]
+ .into_iter()
+ .flatten()
+ .collect::>(),
+ ));
+ }
+ }
+
+ if self.download_fonts
+ && !self.force_hardsub
+ && dst.extension().unwrap_or_default().to_str().unwrap() == "mkv"
+ {
+ let mut font_names = vec![];
+ for subtitle in subtitles.iter() {
+ font_names.extend(get_subtitle_stats(&subtitle.path)?)
+ }
+ real_dedup_vec(&mut font_names);
+
+ let progress_spinner = if log::max_level() == LevelFilter::Info {
+ let progress_spinner = ProgressBar::new_spinner()
+ .with_style(
+ ProgressStyle::with_template(
+ format!(
+ ":: {:<1$} {{msg}} {{spinner}}",
+ "Downloading fonts", fmt_space
+ )
+ .as_str(),
+ )
+ .unwrap()
+ .tick_strings(&["—", "\\", "|", "/", ""]),
+ )
+ .with_finish(ProgressFinish::Abandon);
+ progress_spinner.enable_steady_tick(Duration::from_millis(100));
+ Some(progress_spinner)
+ } else {
+ None
+ };
+ for font_name in font_names {
+ if let Some(pb) = &progress_spinner {
+ let mut progress_message = pb.message();
+ if !progress_message.is_empty() {
+ progress_message += ", "
+ }
+ progress_message += &font_name;
+ pb.set_message(progress_message)
+ }
+ if let Some((font, cached)) = self.download_font(&font_name).await? {
+ if cached {
+ if let Some(pb) = &progress_spinner {
+ let mut progress_message = pb.message();
+ progress_message += " (cached)";
+ pb.set_message(progress_message)
+ }
+ debug!("Downloaded font {} (cached)", font_name);
+ } else {
+ debug!("Downloaded font {}", font_name);
+ }
+
+ fonts.push(font)
+ }
+ }
+ }
+
+ let mut input = vec![];
+ let mut maps = vec![];
+ let mut attachments = vec![];
+ let mut metadata = vec![];
+
+ for (i, meta) in videos.iter().enumerate() {
+ if let Some(start_time) = meta.start_time {
+ input.extend(["-itsoffset".to_string(), format_time_delta(&start_time)])
+ }
+ input.extend(["-i".to_string(), meta.path.to_string_lossy().to_string()]);
+ maps.extend(["-map".to_string(), i.to_string()]);
+ metadata.extend([
+ format!("-metadata:s:v:{}", i),
+ format!(
+ "title={}",
+ if videos.len() == 1 {
+ "Default".to_string()
+ } else {
+ format!("#{}", i + 1)
+ }
+ ),
+ ]);
+ // the empty language metadata is created to avoid that metadata from the original track
+ // is copied
+ metadata.extend([format!("-metadata:s:v:{}", i), "language=".to_string()])
+ }
+ for (i, meta) in audios.iter().enumerate() {
+ if let Some(start_time) = meta.start_time {
+ input.extend(["-itsoffset".to_string(), format_time_delta(&start_time)])
+ }
+ input.extend(["-i".to_string(), meta.path.to_string_lossy().to_string()]);
+ maps.extend(["-map".to_string(), (i + videos.len()).to_string()]);
+ metadata.extend([
+ format!("-metadata:s:a:{}", i),
+ format!(
+ "language={}",
+ self.audio_locale_output_map
+ .get(&meta.locale)
+ .unwrap_or(&meta.locale.to_string())
+ ),
+ ]);
+ metadata.extend([
+ format!("-metadata:s:a:{}", i),
+ format!(
+ "title={}",
+ if videos.len() == 1 {
+ meta.locale.to_human_readable()
+ } else {
+ format!(
+ "{} [Video: #{}]",
+ meta.locale.to_human_readable(),
+ meta.video_idx + 1
+ )
+ }
+ ),
+ ]);
+ }
+
+ for (i, font) in fonts.iter().enumerate() {
+ attachments.extend(["-attach".to_string(), font.to_string_lossy().to_string()]);
+ metadata.extend([
+ format!("-metadata:s:t:{}", i),
+ "mimetype=font/woff2".to_string(),
+ ])
+ }
+
+ // this formats are supporting embedding subtitles into the video container instead of
+ // burning it into the video stream directly
+ let container_supports_softsubs = !self.force_hardsub
+ && ["mkv", "mov", "mp4"]
+ .contains(&dst.extension().unwrap_or_default().to_str().unwrap());
+
+ if container_supports_softsubs {
+ for (i, meta) in subtitles.iter().enumerate() {
+ if let Some(start_time) = meta.start_time {
+ input.extend(["-itsoffset".to_string(), format_time_delta(&start_time)])
+ }
+ input.extend(["-i".to_string(), meta.path.to_string_lossy().to_string()]);
+ maps.extend([
+ "-map".to_string(),
+ (i + videos.len() + audios.len()).to_string(),
+ ]);
+ metadata.extend([
+ format!("-metadata:s:s:{}", i),
+ format!(
+ "language={}",
+ self.subtitle_locale_output_map
+ .get(&meta.locale)
+ .unwrap_or(&meta.locale.to_string())
+ ),
+ ]);
+ metadata.extend([
+ format!("-metadata:s:s:{}", i),
+ format!("title={}", {
+ let mut title = meta.locale.to_human_readable();
+ if meta.cc {
+ title += " (CC)"
+ }
+ if videos.len() > 1 {
+ title += &format!(" [Video: #{}]", meta.video_idx + 1)
+ }
+ title
+ }),
+ ]);
+ }
+ }
+
+ if let Some(((file, path), chapters)) = chapters.as_mut() {
+ write_ffmpeg_chapters(file, max_len, chapters)?;
+ input.extend(["-i".to_string(), path.to_string_lossy().to_string()]);
+ maps.extend([
+ "-map_metadata".to_string(),
+ (videos.len()
+ + audios.len()
+ + container_supports_softsubs
+ .then_some(subtitles.len())
+ .unwrap_or_default())
+ .to_string(),
+ ])
+ }
+
+ let preset_custom = matches!(self.ffmpeg_preset, FFmpegPreset::Custom(_));
+ let (input_presets, mut output_presets) = self.ffmpeg_preset.into_input_output_args();
+ let fifo = temp_named_pipe()?;
+
+ let mut command_args = vec![
+ "-y".to_string(),
+ "-hide_banner".to_string(),
+ "-vstats_file".to_string(),
+ fifo.path().to_string_lossy().to_string(),
+ ];
+ command_args.extend(input_presets);
+ command_args.extend(input);
+ command_args.extend(maps);
+ command_args.extend(attachments);
+ command_args.extend(metadata);
+ if !preset_custom {
+ if let Some(ffmpeg_threads) = self.ffmpeg_threads {
+ command_args.extend(vec!["-threads".to_string(), ffmpeg_threads.to_string()])
+ }
+ }
+
+ // set default subtitle
+ if let Some(default_subtitle) = self.default_subtitle {
+ if let Some(position) = subtitles.iter().position(|m| m.locale == default_subtitle) {
+ if container_supports_softsubs {
+ match dst.extension().unwrap_or_default().to_str().unwrap() {
+ "mov" | "mp4" => output_presets.extend([
+ "-movflags".to_string(),
+ "faststart".to_string(),
+ "-c:s".to_string(),
+ "mov_text".to_string(),
+ ]),
+ _ => (),
+ }
+ } else {
+ // remove '-c:v copy' and '-c:a copy' from output presets as its causes issues with
+ // burning subs into the video
+ let mut last = String::new();
+ let mut remove_count = 0;
+ for (i, s) in output_presets.clone().iter().enumerate() {
+ if (last == "-c:v" || last == "-c:a") && s == "copy" {
+ // remove last
+ output_presets.remove(i - remove_count - 1);
+ remove_count += 1;
+ output_presets.remove(i - remove_count);
+ remove_count += 1;
+ }
+ last.clone_from(s);
+ }
+
+ output_presets.extend([
+ "-vf".to_string(),
+ format!(
+ "ass='{}'",
+ // ffmpeg doesn't removes all ':' and '\' from the filename when using
+ // the ass filter. well, on windows these characters are used in
+ // absolute paths, so they have to be correctly escaped here
+ if cfg!(windows) {
+ subtitles
+ .get(position)
+ .unwrap()
+ .path
+ .to_str()
+ .unwrap()
+ .replace('\\', "\\\\")
+ .replace(':', "\\:")
+ } else {
+ subtitles
+ .get(position)
+ .unwrap()
+ .path
+ .to_string_lossy()
+ .to_string()
+ }
+ ),
+ ])
+ }
+ }
+
+ if container_supports_softsubs {
+ if let Some(position) = subtitles
+ .iter()
+ .position(|meta| meta.locale == default_subtitle)
+ {
+ command_args.extend([
+ format!("-disposition:s:s:{}", position),
+ "default".to_string(),
+ ])
+ }
+ }
+ }
+
+ // set the 'forced' flag to CC subtitles
+ for (i, subtitle) in subtitles.iter().enumerate() {
+ if !subtitle.cc {
+ continue;
+ }
+
+ command_args.extend([format!("-disposition:s:s:{}", i), "forced".to_string()])
+ }
+
+ command_args.extend(output_presets);
+ if let Some(output_format) = self.output_format {
+ command_args.extend(["-f".to_string(), output_format]);
+ }
+
+ // prepend './' to the path on linux since ffmpeg may interpret the path incorrectly if it's just the filename.
+ // see https://github.com/crunchy-labs/crunchy-cli/issues/303 for example
+ if !cfg!(windows)
+ && dst
+ .parent()
+ .map_or(true, |p| p.to_string_lossy().is_empty())
+ {
+ command_args.push(Path::new("./").join(dst).to_string_lossy().to_string());
+ } else {
+ command_args.push(dst.to_string_lossy().to_string())
+ }
+
+ debug!("ffmpeg {}", command_args.join(" "));
+
+ // create parent directory if it does not exist
+ if let Some(parent) = dst.parent() {
+ if !parent.exists() {
+ fs::create_dir_all(parent)?
+ }
+ }
+
+ let ffmpeg = Command::new("ffmpeg")
+ // pass ffmpeg stdout to real stdout only if output file is stdout
+ .stdout(if dst.to_str().unwrap() == "-" {
+ Stdio::inherit()
+ } else {
+ Stdio::null()
+ })
+ .stderr(Stdio::piped())
+ .args(command_args)
+ .spawn()?;
+ let ffmpeg_progress_cancel = CancellationToken::new();
+ let ffmpeg_progress_cancellation_token = ffmpeg_progress_cancel.clone();
+ let ffmpeg_progress = tokio::spawn(async move {
+ ffmpeg_progress(
+ max_frames,
+ fifo,
+ format!("{:<1$}", "Generating output file", fmt_space + 1),
+ ffmpeg_progress_cancellation_token,
+ )
+ .await
+ });
+
+ let result = ffmpeg.wait_with_output()?;
+ if !result.status.success() {
+ ffmpeg_progress.abort();
+ bail!("{}", String::from_utf8_lossy(result.stderr.as_slice()))
+ }
+ ffmpeg_progress_cancel.cancel();
+ ffmpeg_progress.await?
+ }
+
+ async fn check_free_space(
+ &self,
+ dst: &Path,
+ ) -> Result<(Option<(PathBuf, u64)>, Option<(PathBuf, u64)>)> {
+ let mut all_stream_data = vec![];
+ for format in &self.formats {
+ all_stream_data.push(&format.video.0);
+ all_stream_data.extend(format.audios.iter().map(|(a, _)| a))
+ }
+ let mut estimated_required_space: u64 = 0;
+ for stream_data in all_stream_data {
+ let segments = stream_data.segments();
+
+ // sum the length of all streams up
+ estimated_required_space += estimate_stream_data_file_size(stream_data, &segments);
+ }
+
+ let tmp_stat = fs2::statvfs(temp_directory()).unwrap();
+ let mut dst_file = if dst.is_absolute() {
+ dst.to_path_buf()
+ } else {
+ env::current_dir()?.join(dst)
+ };
+ for ancestor in dst_file.ancestors() {
+ if ancestor.exists() {
+ dst_file = ancestor.to_path_buf();
+ break;
+ }
+ }
+ let dst_stat = fs2::statvfs(&dst_file).unwrap();
+
+ let mut tmp_space = tmp_stat.available_space();
+ let mut dst_space = dst_stat.available_space();
+
+ // this checks if the partition the two directories are located on are the same to prevent
+ // that the space fits both file sizes each but not together. this is done by checking the
+ // total space if each partition and the free space of each partition (the free space can
+ // differ by 10MB as some tiny I/O operations could be performed between the two calls which
+ // are checking the disk space)
+ if tmp_stat.total_space() == dst_stat.total_space()
+ && (tmp_stat.available_space() as i64 - dst_stat.available_space() as i64).abs() < 10240
+ {
+ tmp_space *= 2;
+ dst_space *= 2;
+ }
+
+ let mut tmp_required = None;
+ let mut dst_required = None;
+
+ if tmp_space < estimated_required_space {
+ tmp_required = Some((temp_directory(), estimated_required_space))
+ }
+ if (!is_special_file(dst) && dst.to_string_lossy() != "-")
+ && dst_space < estimated_required_space
+ {
+ dst_required = Some((dst_file, estimated_required_space))
+ }
+ Ok((tmp_required, dst_required))
+ }
+
+ async fn download_video(
+ &self,
+ stream_data: &StreamData,
+ message: String,
+ max_segments: Option,
+ ) -> Result {
+ let tempfile = tempfile(".mp4")?;
+ let (mut file, path) = tempfile.into_parts();
+
+ self.download_segments(&mut file, message, stream_data, max_segments)
+ .await?;
+
+ Ok(path)
+ }
+
+ async fn download_audio(&self, stream_data: &StreamData, message: String) -> Result {
+ let tempfile = tempfile(".m4a")?;
+ let (mut file, path) = tempfile.into_parts();
+
+ self.download_segments(&mut file, message, stream_data, None)
+ .await?;
+
+ Ok(path)
+ }
+
+ async fn download_subtitle(
+ &self,
+ subtitle: Subtitle,
+ max_length: TimeDelta,
+ ) -> Result {
+ let buf = subtitle.data().await?;
+ let mut ass = match subtitle.format.as_str() {
+ "ass" => SSA::parse(String::from_utf8_lossy(&buf))?,
+ "vtt" => VTT::parse(String::from_utf8_lossy(&buf))?.to_ssa(),
+ _ => bail!("unknown subtitle format: {}", subtitle.format),
+ };
+ // subtitles aren't always correct sorted and video players may have issues with that. to
+ // prevent issues, the subtitles are sorted
+ // (https://github.com/crunchy-labs/crunchy-cli/issues/208)
+ ass.events.sort_by(|a, b| a.start.cmp(&b.start));
+ // it might be the case that the start and/or end time are greater than the actual video
+ // length. this might also result in issues with video players, thus the times are stripped
+ // to be at most as long as `max_length`
+ // (https://github.com/crunchy-labs/crunchy-cli/issues/32)
+ for i in (0..ass.events.len()).rev() {
+ let max_len = Time::from_hms(0, 0, 0)
+ .unwrap()
+ .add(Duration::from_millis(max_length.num_milliseconds() as u64));
+
+ if ass.events[i].start > max_len {
+ if ass.events[i].end > max_len {
+ ass.events[i].start = max_len
+ }
+ ass.events[i].end = max_len
+ } else {
+ break;
+ }
+ }
+
+ // without this additional info, subtitle look very messy in some video player
+ // (https://github.com/crunchy-labs/crunchy-cli/issues/66)
+ ass.info
+ .additional_fields
+ .insert("ScaledBorderAndShadow".to_string(), "yes".to_string());
+
+ let tempfile = tempfile(".ass")?;
+ let path = tempfile.into_temp_path();
+
+ fs::write(&path, ass.to_string())?;
+
+ Ok(path)
+ }
+
+ async fn download_font(&self, name: &str) -> Result> {
+ let Some((_, font_file)) = FONTS.iter().find(|(f, _)| f == &name) else {
+ return Ok(None);
+ };
+
+ let cache_dir = cache_dir("fonts")?;
+ let file = cache_dir.join(font_file);
+ if file.exists() {
+ return Ok(Some((file, true)));
+ }
+
+ // the speed limiter does not apply to this
+ let font = self
+ .client
+ .get(format!(
+ "https://static.crunchyroll.com/vilos-v2/web/vilos/assets/libass-fonts/{}",
+ font_file
+ ))
+ .send()
+ .await?
+ .bytes()
+ .await?;
+ fs::write(&file, font)?;
+
+ Ok(Some((file, false)))
+ }
+
+ async fn download_segments(
+ &self,
+ writer: &mut impl Write,
+ message: String,
+ stream_data: &StreamData,
+ max_segments: Option,
+ ) -> Result<()> {
+ let mut segments = stream_data.segments();
+ if let Some(max_segments) = max_segments {
+ segments = segments
+ .drain(0..max_segments.min(segments.len() - 1))
+ .collect();
+ }
+ let total_segments = segments.len();
+
+ let count = Arc::new(Mutex::new(0));
+
+ let progress = if log::max_level() == LevelFilter::Info {
+ let estimated_file_size = estimate_stream_data_file_size(stream_data, &segments);
+
+ let progress = ProgressBar::new(estimated_file_size)
+ .with_style(
+ ProgressStyle::with_template(
+ ":: {msg} {bytes:>10} {bytes_per_sec:>12} [{wide_bar}] {percent:>3}%",
+ )
+ .unwrap()
+ .progress_chars("##-"),
+ )
+ .with_message(message)
+ .with_finish(ProgressFinish::Abandon);
+ Some(progress)
+ } else {
+ None
+ };
+
+ let cpus = self.download_threads.min(segments.len());
+ let mut segs: Vec> = Vec::with_capacity(cpus);
+ for _ in 0..cpus {
+ segs.push(vec![])
+ }
+ for (i, segment) in segments.clone().into_iter().enumerate() {
+ segs[i - ((i / cpus) * cpus)].push(segment);
+ }
+
+ let (sender, mut receiver) = unbounded_channel();
+
+ let mut join_set: JoinSet> = JoinSet::new();
+ for num in 0..cpus {
+ let thread_sender = sender.clone();
+ let thread_segments = segs.remove(0);
+ let thread_client = self.client.clone();
+ let mut thread_rate_limiter = self.rate_limiter.clone();
+ let thread_count = count.clone();
+ join_set.spawn(async move {
+ let after_download_sender = thread_sender.clone();
+
+ // the download process is encapsulated in its own function. this is done to easily
+ // catch errors which get returned with `...?` and `bail!(...)` and that the thread
+ // itself can report that an error has occurred
+ let download = || async move {
+ for (i, segment) in thread_segments.into_iter().enumerate() {
+ let mut retry_count = 0;
+ let buf = loop {
+ let request = thread_client
+ .get(&segment.url)
+ .timeout(Duration::from_secs(60));
+ let response = if let Some(rate_limiter) = &mut thread_rate_limiter {
+ rate_limiter.call(request.build()?).await.map_err(anyhow::Error::new)
+ } else {
+ request.send().await.map_err(anyhow::Error::new)
+ };
+
+ let err = match response {
+ Ok(r) => match r.bytes().await {
+ Ok(b) => break b.to_vec(),
+ Err(e) => anyhow::Error::new(e)
+ }
+ Err(e) => e,
+ };
+
+ if retry_count == 5 {
+ bail!("Max retry count reached ({}), multiple errors occurred while receiving segment {}: {}", retry_count, num + (i * cpus), err)
+ }
+ debug!("Failed to download segment {} ({}). Retrying, {} out of 5 retries left", num + (i * cpus), err, 5 - retry_count);
+
+ retry_count += 1;
+ };
+
+ let mut c = thread_count.lock().await;
+ debug!(
+ "Downloaded segment [{}/{} {:.2}%] {}",
+ num + (i * cpus) + 1,
+ total_segments,
+ ((*c + 1) as f64 / total_segments as f64) * 100f64,
+ segment.url
+ );
+
+ thread_sender.send((num as i32 + (i * cpus) as i32, buf))?;
+
+ *c += 1;
+ }
+ Ok(())
+ };
+
+
+ let result = download().await;
+ if result.is_err() {
+ after_download_sender.send((-1, vec![]))?;
+ }
+
+ result
+ });
+ }
+ // drop the sender already here so it does not outlive all download threads which are the only
+ // real consumers of it
+ drop(sender);
+
+ // this is the main loop which writes the data. it uses a BTreeMap as a buffer as the write
+ // happens synchronized. the download consist of multiple segments. the map keys are representing
+ // the segment number and the values the corresponding bytes
+ let mut data_pos = 0;
+ let mut buf: BTreeMap> = BTreeMap::new();
+ while let Some((pos, bytes)) = receiver.recv().await {
+ // if the position is lower than 0, an error occurred in the sending download thread
+ if pos < 0 {
+ break;
+ }
+
+ if let Some(p) = &progress {
+ let progress_len = p.length().unwrap();
+ let estimated_segment_len = (stream_data.bandwidth / 8)
+ * segments.get(pos as usize).unwrap().length.as_secs();
+ let bytes_len = bytes.len() as u64;
+
+ p.set_length(progress_len - estimated_segment_len + bytes_len);
+ p.inc(bytes_len)
+ }
+
+ // check if the currently sent bytes are the next in the buffer. if so, write them directly
+ // to the target without first adding them to the buffer.
+ // if not, add them to the buffer
+ if data_pos == pos {
+ writer.write_all(bytes.borrow())?;
+ data_pos += 1;
+ } else {
+ buf.insert(pos, bytes);
+ }
+ // check if the buffer contains the next segment(s)
+ while let Some(b) = buf.remove(&data_pos) {
+ writer.write_all(b.borrow())?;
+ data_pos += 1;
+ }
+ }
+
+ // if any error has occurred while downloading it gets returned here
+ while let Some(joined) = join_set.join_next().await {
+ joined??
+ }
+
+ // write the remaining buffer, if existent
+ while let Some(b) = buf.remove(&data_pos) {
+ writer.write_all(b.borrow())?;
+ data_pos += 1;
+ }
+
+ if !buf.is_empty() {
+ bail!(
+ "Download buffer is not empty. Remaining segments: {}",
+ buf.into_keys()
+ .map(|k| k.to_string())
+ .collect::>()
+ .join(", ")
+ )
+ }
+
+ Ok(())
+ }
+}
+
+fn estimate_stream_data_file_size(stream_data: &StreamData, segments: &[StreamSegment]) -> u64 {
+ (stream_data.bandwidth / 8) * segments.iter().map(|s| s.length.as_secs()).sum::()
+}
+
+/// Get the length and fps of a video.
+fn get_video_stats(path: &Path) -> Result<(TimeDelta, f64)> {
+ let video_length = Regex::new(r"Duration:\s(?P
+
+ 
-
- 
+
+ 
-
- 
+
+ 
-
- 
+
+ 
-
- 
+
+ 
-
- 
+
+ 