From ced0d86c4c8dba5b98c6e2347e5d3225d82917ed Mon Sep 17 00:00:00 2001
From: ghe0 <gheorghe@gheo.tech>
Date: Tue, 12 Nov 2024 18:34:32 +0200
Subject: [PATCH] adding support for ipv6 to the install script

---
 install.sh | 13 +++++++++++--
 1 file changed, 11 insertions(+), 2 deletions(-)

diff --git a/install.sh b/install.sh
index 7031863..73edc1a 100755
--- a/install.sh
+++ b/install.sh
@@ -11,9 +11,13 @@ if [ -z $1 ]; then
 fi
 
 server="$1"
+scp_server="$1"
+
+echo $1 | grep -F ':' && scp_server="[${1}]"
 
 if ! [ -z $2 ]; then
   server="${2}@${server}"
+  echo $1 | grep -F ':' && scp_server="${2}@${scp_server}"
 fi
 
 if [ -z $3 ]; then
@@ -24,6 +28,11 @@ else
   scp="scp -P $3"
 fi
 
+echo "ssh: $ssh"
+echo "scp: $scp"
+echo "server: $server"
+echo "scp_server: $scp_server"
+
 echo
 echo Starting installation...
 echo
@@ -34,10 +43,10 @@ mkdir -p tmp
 tar cf tmp/dtrfs.tar *.sh
 $ssh $server rm -rf ${dir}
 $ssh $server mkdir -p ${dir}
-$scp tmp/dtrfs.tar ${server}:${dir}
+$scp tmp/dtrfs.tar ${scp_server}:${dir}
 $ssh $server tar -xf ${dir}/dtrfs.tar -C ${dir}
 $ssh $server ${dir}/create.sh
 archive=$($ssh $server cat ${dir}/build/.archive_name)
-$scp ${server}:${dir}/build/${archive} tmp/
+$scp ${scp_server}:${dir}/build/${archive} tmp/
 echo
 echo initrd downloaded to: $(pwd)/tmp/${archive}