#!/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
IN_FILE="$1_generated.txt"
SUCCESS_FILE="$1_success.txt"
FAIL_FILE="$1_failed.txt"

download() {
    SUCCESS="false"

    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"
                SUCCESS="true"
                break
            fi
        else
            if gallery-dl --config ./gallery-dl/config.json --directory "$2" "$1"; then
                echo "$1" >> "$SUCCESS_FILE"
                SUCCESS="true"
                break
            fi
        fi
        echo -e "${IYELLOW}download failed${NO_COLOR}"

        if ! ping -qc 3 v4.ip.hopeless-cloud.xyz > /dev/null 2>&1; then
            echo -e "${IYELLOW}no internet${NO_COLOR}"
            exit 1
        elif ! ping -qc 3 v6.ip.hopeless-cloud.xyz > /dev/null 2>&1; then
            echo -e "${IYELLOW}no internet${NO_COLOR}"
            exit 1
        else
            echo -e "${IGREEN}has internet${NO_COLOR}"
        fi
    done
    
    [[ $SUCCESS != \t\r\u\e ]] && echo "$1" >> "$FAIL_FILE"
}

[[ $(hostname) = \s\a\t\a\n ]] && rm -rf ./Artists/*

if [[ -f $FAIL_FILE ]]; then
    [[ -f "${FAIL_FILE}.old" ]] && rm -f "${FAIL_FILE}.old"
    mv "$FAIL_FILE" "${FAIL_FILE}.old"
fi

if [[ -f $SUCCESS_FILE ]]; then
    [[ -f "${SUCCESS_FILE}.old" ]] && rm -f "${SUCCESS_FILE}.old"
    mv "$SUCCESS_FILE" "${SUCCESS_FILE}.old"
fi

[[ -f $IN_FILE ]] && rm $IN_FILE
[[ $(hostname) = \s\a\t\a\n ]] && cargo run -- generate "$1" || ./Hentai generate "$1"

if [[ $(echo $?) -ne 0 ]]; then
    echo -e "${IRED}Hentai failed to execute properly${NO_COLOR}"
    exit 1
fi

while IFS= read -r line; do
    url=$(echo "$line" | awk '{print $1}')
    path=$(echo "$line" | awk '{print $3}')
    
    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}"