57 lines
1.6 KiB
Bash
Executable File
57 lines
1.6 KiB
Bash
Executable File
#!/usr/bin/bash
|
|
|
|
source "$HOME"/.config/Hentai.conf
|
|
|
|
IGREEN="\033[0;92m" # Intense Green
|
|
IYELLOW="\033[0;93m" # Intense Yellow
|
|
IRED='\033[0;91m' # Intense Red
|
|
NO_COLOR="\033[0m" # Text Reset
|
|
|
|
IN_FILE="${tmp_dir}/${1}_generated.txt"
|
|
SUCCESS_FILE="${log_dir}/${1}_success.txt"
|
|
FAIL_FILE="${log_dir}/${1}_failed.txt"
|
|
|
|
download() {
|
|
for ((n=0;n<5;n++)); do
|
|
if [[ $1 =~ "iwara.tv" ]]; then
|
|
if yt-dlp --config-locations ./yt-dlp/iwara.conf -o "${2}/%(id)s.%(ext)s" "$1"; then
|
|
echo "$1" >> "$SUCCESS_FILE"
|
|
return 0
|
|
fi
|
|
else
|
|
if gallery-dl --config ./gallery-dl/config.json --directory "$2" "$1"; then
|
|
echo "$1" >> "$SUCCESS_FILE"
|
|
return 0
|
|
fi
|
|
fi
|
|
echo -e "${IYELLOW}download failed${NO_COLOR}"
|
|
|
|
if has_internet; then
|
|
echo -e "${IGREEN}has internet${NO_COLOR}"
|
|
else
|
|
echo -e "${IYELLOW}no internet${NO_COLOR}"
|
|
exit 1
|
|
fi
|
|
done
|
|
|
|
echo "$1" >> "$FAIL_FILE"
|
|
}
|
|
|
|
mkdir -p "$tmp_dir" "$log_dir"
|
|
Hentai generate "$1" "$tmp_dir" || exit 1
|
|
|
|
[[ -f $FAIL_FILE ]] && rm -f "$FAIL_FILE" || touch "$FAIL_FILE"
|
|
[[ -f $SUCCESS_FILE ]] && rm -f "$SUCCESS_FILE" || touch "$SUCCESS_FILE"
|
|
|
|
while IFS= read -r line; do
|
|
url=$(awk '{print $1}' <<< "$line")
|
|
path=$(awk '{print $3}' <<< "$line")
|
|
|
|
echo "url: $url"
|
|
echo "path: $path"
|
|
|
|
download $url $path
|
|
done < "$IN_FILE"
|
|
|
|
[[ -s $FAIL_FILE ]] && echo -e "${IRED}Some links failed!${NO_COLOR}" || echo -e "${IGREEN}Success!${NO_COLOR}"
|