#!/bin/bash

# 1 2 3 ..
CHECKPOINT=$1

# 1 2 3 4 5 .. OR 12 235 ..
RC=$2

GOORM="0"
RUNPOD="0"
SAGEMAKER="0"

if [ -d "/workspace/firstContainer" ]; then
  GOORM="1"
elif [ -d "/workspace" ]; then
  RUNPOD="1"
elif [ -d "/home/studio-lab-user" ]; then
  SAGEMAKER="1"
fi

RELATIVE_DIR=""

if [ "$SAGEMAKER" == "1" ]; then
  RELATIVE_DIR='/home/studio-lab-user/sagemaker-studiolab-notebooks'
  else
  RELATIVE_DIR=`dirname "$0"`
fi

echo "$RELATIVE_DIR"
cd $RELATIVE_DIR

if [ -d "SD" ]; then
  cd SD
elif [ -d "CU" ]; then
  cd CU
else
  cd SD*
  cd CU*
fi

cd models
cd Stable-diffusion

CK_LINK=""

case $CHECKPOINT in
	0)
		echo "CHECKPOINT pass."
		;;
	1)
		CK_LINK="https://huggingface.co/Magamanny/Pony-Diffusion-V6-XL/resolve/main/ponyDiffusionV6XL_v6StartWithThisOne.safetensors"
		;;
	2)
		CK_LINK="https://huggingface.co/JosefJilek/moeFussion/resolve/main/moeFussionV1.1.0_PDXL_Lx7-CLIP_VAE_FP16.safetensors"
		;;
	2v2)
		CK_LINK="https://huggingface.co/JosefJilek/moeFussion/resolve/main/moeFussionV1.2.0_NIASR_Lx2_v2.safetensors"
		;;
	3)
		CK_LINK="https://huggingface.co/zuv0/test/resolve/main/MINTSDXL_LollipopMIX_A1-fp32.safetensors"
		;;
	4)
		CK_LINK="https://huggingface.co/Bulkbogan20/autismmix/resolve/main/autismmixSDXL_autismmixPony.safetensors"
		;;
	5)
        CK_LINK="https://huggingface.co/zuv0/test/resolve/main/lemontart_v20.safetensors"
		;;
	5v2)
        CK_LINK="https://huggingface.co/zuv0/test/resolve/main/lemontart_v30.safetensors"
		;;
	6)
        CK_LINK="https://huggingface.co/zuv0/test/resolve/main/grimoire_p0666.safetensors"
		;;
	7)
        CK_LINK="https://huggingface.co/zuv0/test/resolve/main/mizuiropony_v12.safetensors"
		;;
	8)
        CK_LINK="https://huggingface.co/zuv0/test/resolve/main/nekodaxl_.safetensors"
		;;
	11)
        CK_LINK="https://huggingface.co/zuv0/test/resolve/main/perillanium_wpe.safetensors"
		;;
	12)
        CK_LINK="https://huggingface.co/zuv0/test/resolve/main/hassakuXLHentai_v13BetterEyesVersion.safetensors"
		;;
	13)
        CK_LINK="https://huggingface.co/zuv0/test/resolve/main/boleromixPony_v150VAE.safetensors"
		;;
	14)
        CK_LINK="https://huggingface.co/zuv0/test/resolve/main/waiANINSFWPONYXL_v90.safetensors"
		;;
	14v2)
        CK_LINK="https://huggingface.co/zuv0/test/resolve/main/waiANINSFWPONYXL_v12.safetensors"
		;;
	15)
        CK_LINK="https://huggingface.co/zuv0/test/resolve/main/rdxlPixelArt_pony2.safetensors"
		;;
	16)
        CK_LINK="https://huggingface.co/zuv0/test/resolve/main/hosekiLustrousmixPony_v20.safetensors"
		;;
	17)
        CK_LINK="https://huggingface.co/zuv0/test/resolve/main/omdOhmaydaughter_v10.safetensors"
		;;
    a)
        CK_LINK="https://huggingface.co/cagliostrolab/animagine-xl-3.1/resolve/main/animagine-xl-3.1.safetensors"
        ;;
    d)
        CK_LINK="https://huggingface.co/zuv0/test/resolve/main/noobaiXLNAIXL_earlyAccessVersion.safetensors"
        ;;
    d2)
        CK_LINK="https://huggingface.co/zuv0/test/resolve/main/noobaiXLNAIXL_epsilonPred05Version.safetensors"
        ;;
    d3)
        CK_LINK="https://huggingface.co/zuv0/test/resolve/main/noobaiXLNAIXL_epsilonPred075.safetensors"
        ;;
    d4)
        CK_LINK="https://huggingface.co/zuv0/test/resolve/main/noobaiXLNAIXL_epsilonPred10Version.safetensors"
        ;;
    d5)
        CK_LINK="https://huggingface.co/zuv0/test/resolve/main/noobaiXLNAIXL_epsilonPred11Version.safetensors"
        ;;
    d6)
        CK_LINK="https://huggingface.co/Laxhar/noobai-XL-Vpred-1.0/resolve/main/NoobAI-XL-Vpred-v1.0.safetensors"
        ;;
    e)
        CK_LINK="https://huggingface.co/hanzogak/comradeshipXL/resolve/main/comradeshipXL-v14T13.safetensors"
        ;;
    f)
        CK_LINK="https://huggingface.co/zuv0/test/resolve/main/illustriousXPonyMix_v3.safetensors"
        ;;
    h2)
        CK_LINK="https://huggingface.co/zuv0/test/resolve/main/ntrMIXIllustriousXL_v21.safetensors"
        ;;
    h3)
        CK_LINK="https://huggingface.co/zuv0/test/resolve/main/ntrMIXIllustriousXL_v35.safetensors"
        ;;
	j)
        CK_LINK="https://huggingface.co/zuv0/test/resolve/main/lizmix_version15.safetensors"
		;;
	k)
        CK_LINK="https://huggingface.co/zuv0/test/resolve/main/nonnette_v010.safetensors"
		;;
	k2)
        CK_LINK="https://huggingface.co/zuv0/test/resolve/main/nonnette_v020.safetensors"
		;;
	m)
        CK_LINK="https://huggingface.co/zuv0/test/resolve/main/novaAnimeXL_ilV30HappyNewYear.safetensors"
		;;
	n)
        CK_LINK="https://huggingface.co/zuv0/test/resolve/main/waiNSFWIllustrious_v90.safetensors"
		;;
	p)
        CK_LINK="https://huggingface.co/Panchovix/noobai-XL-Vpred-1.0-perpendicular-cyberfix/resolve/main/NoobAI-XL-Vpred-v1.0-cyberfix-perpendicular.safetensors"
		;;
	s)
        CK_LINK="https://huggingface.co/zuv0/test/resolve/main/uncannyValley_uncannyvallyNoob3dV1.safetensors"
		;;
	s2)
        CK_LINK="https://huggingface.co/zuv0/test/resolve/main/uncannyValley_ilxl10Noob.safetensors"
		;;
	t)
        CK_LINK="https://huggingface.co/zuv0/test/resolve/main/hesperidesIllustrious_v10.safetensors"
		;;
	u)
        CK_LINK="https://huggingface.co/JosefJilek/moeFussion/resolve/main/moeFussionV1.3.0_IIZ.safetensors"
		;;
	u2a)
        CK_LINK="https://huggingface.co/JosefJilek/moeFussion/resolve/main/moeFussionV1.4.0_RIZ.safetensors"
		;;
	u2b)
        CK_LINK="https://huggingface.co/JosefJilek/moeFussion/resolve/main/moeFussionV1.4.0_RIZ_vz.safetensors"
		;;
	u2c)
        CK_LINK="https://huggingface.co/JosefJilek/moeFussion/resolve/main/moeFussionV1.4.0_RICZ_vz.safetensors"
		;;
	v)
        CK_LINK="https://huggingface.co/zuv0/test/resolve/main/zukiCuteILL_v50.safetensors"
		;;
	ra)
        CK_LINK="https://huggingface.co/zuv0/test/resolve/main/noobreal_v21.safetensors"
		;;
	rb)
        CK_LINK="https://huggingface.co/zuv0/test/resolve/main/waiREALCN_v14.safetensors"
		;;
    gc)
		rm -rf *.safetensors
		echo "CHECKPOINT remove."
		exit 0
		;;
	*)
		echo "CHECKPOINT error!"
		;;
esac

#CK_LINK="https://huggingface.co/zuv0/test/resolve/main/waiCUTE_v20.safetensors"
#CK_LINK="https://huggingface.co/GianPehn/PD_for_Anime/resolve/main/pdForAnime_v20.safetensors"
#CK_LINK="https://huggingface.co/Walkearth4/Collection/resolve/main/chenkinAnimeHotbaby_v20.safetensors"
#CK_LINK="https://huggingface.co/zuv0/test/resolve/main/copycatASC_v10.safetensors"
#CK_LINK="https://huggingface.co/zuv0/test/resolve/main/catdash_v10.safetensors"

if [ -n "$CK_LINK" ]; then
  CK_FILE=${CK_LINK##*/}
  if [ -f "$CK_FILE" ]; then
    \mv -f "$CK_FILE" "../$CK_FILE"
    rm -rf *.safetensors
    rm -rf *.ckpt
    \mv -f "../$CK_FILE" "$CK_FILE"
  else
    rm -rf *.safetensors
    rm -rf *.ckpt
    if [ "$GOORM" == "1" ]; then
      #wget –limit-rate=49999k -N "$CK_LINK"
      wget -N "$CK_LINK"
    else
      #wget –limit-rate=249999k -N "$CK_LINK"
      #aria2c -x 8 -s 8 -c --auto-file-renaming=false "$CK_LINK" -d /workspace/SD/models/Stable-diffusion
      #unzip "/workspace/SD/models/Stable-diffusion/*.zip" -d "/workspace/SD/models/Stable-diffusion/"
      wget -N "$CK_LINK"
    fi
  fi
fi

cd ..
cd Lora

case $RC in
	0)
		echo "RCXL pass."
		;;
	*)
		RCLIST="1 2 3 4 5 6 7 8 9"
 
		for var in $RCLIST
		do
		  rm -rf RC$var
		done

		while [ $RC -gt 0 ]; do
		    digit=$((RC % 10))
		    if [ ! -d "RCXL$digit" ]; then
				git clone https://huggingface.co/zuv0/RCXL$digit
				cd RCXL$digit
				git repack -a -d --depth=250 --window=250
				cd .git
				rm -rf lfs
				cd ..
				cd ..
			fi
		    RC=$((RC / 10))
		done
		;;
esac