#!/bin/bash

# 1 2 3 ..
CHECKPOINT=$1

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

GOORM="0"
if [ -d "/workspace/firstContainer" ]; then
  GOORM="1"
fi

RELATIVE_DIR=`dirname "$0"`
cd $RELATIVE_DIR

if [ ! -d "SD" ]; then
  cd SD
else
  cd SD*
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"
		;;
	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"
		;;
	6)
        CK_LINK="https://huggingface.co/zuv0/test/resolve/main/grimoire_p0666.safetensors"
		;;
    a)
        CK_LINK="https://huggingface.co/cagliostrolab/animagine-xl-3.1/resolve/main/animagine-xl-3.1.safetensors"
        ;;
	*)
		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"
    else
      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