#!/usr/bin/bash

IGREEN="\033[0;92m"     # Intense Green
IYELLOW="\033[0;93m"    # Intense Yellow
IRED='\033[0;91m'       # Intense Red
NO_COLOR="\033[0m"      # Text Reset

[[ -z $1 ]] && exit 1

if [[ -z $2 ]]; then
    output=$([[ $(hostname) = \s\a\t\a\n ]] && cargo run -- add_url "$1" || ./Hentai add_url "$1")
else
    output=$([[ $(hostname) = \s\a\t\a\n ]] && cargo run -- add_custom_name "$1" "$2" || ./Hentai add_custom_name "$1" "$2")
fi

# output="https://rule34.xxx/index.php?page=post&s=list&tags=donnard # Artists/donnard/Rule34"

echo "$output"
[[ $output =~ "#" ]] || exit 1

url=$(echo "$output" | awk '{print $1}')
path=$(echo "$output" | awk '{print $3}')

if [[ $(hostname) = \s\a\t\a\n ]]; then
    [[ -d ./Artists ]] || mkdir ./Artists
    rm -rf ./gallery-dl/*.sqlite3 ./gallery-dl/log.txt ./Artists/* || exit 1
fi

echo "command: gallery-dl --config ./gallery-dl/config.json --directory $path $url"

for ((n=0;n<5;n++)); do
    gallery-dl --config ./gallery-dl/config.json --directory "$path" "$url"

    [[ $(echo $?) -eq 0 ]] && (echo -e "${IGREEN}Success${NO_COLOR}"; break)
    
    echo -e "${IYELLOW}download failed${NO_COLOR}"

    [[ $(ping -c 3 v4.ip.hopeless-cloud.xyz > /dev/null 2>&1) -ne 0 ]] && (echo -e "${IRED}no internet${NO_COLOR}"; exit 1)
    [[ $(ping -c 3 v6.ip.hopeless-cloud.xyz > /dev/null 2>&1) -ne 0 ]] && (echo -e "${IRED}no internet${NO_COLOR}"; exit 1)

    echo -e "${IGREEN}has internet${NO_COLOR}"
done