From d9e7d4498445d39209f65c49b0737233a3f6b52d Mon Sep 17 00:00:00 2001
From: DevilaN <devilan@o2.pl>
Date: Mon, 24 Oct 2022 10:34:53 +0200
Subject: [PATCH] Detect IPv6 and add proper flags for aria2c

---
 services/download/download.sh | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/services/download/download.sh b/services/download/download.sh
index e5520a2..420a807 100755
--- a/services/download/download.sh
+++ b/services/download/download.sh
@@ -4,9 +4,14 @@ set -Eeuo pipefail
 
 mkdir -p /data/.cache /data/StableDiffusion /data/Codeformer /data/GFPGAN /data/ESRGAN /data/BSRGAN /data/RealESRGAN /data/SwinIR /data/LDSR /data/ScuNET /data/embeddings
 
+ARIA2C_OPTS=""
+if [ "`ip -6 addr`" == "" ]; then
+    ARIA2C_OPTS="${ARIA2C_OPTS} --disable-ipv6"
+fi
+
 echo "Downloading, this might take a while..."
 
-aria2c --input-file /docker/links.txt --dir /data --continue
+aria2c ${ARIA2C_OPTS} --input-file /docker/links.txt --dir /data --continue
 
 echo "Checking SHAs..."