#!/bin/bash

# 1 2 3 4 ..
CHECKPOINT=$1

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

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

cd SD*
cd models
cd Stable-diffusion

case $CHECKPOINT in
	0)
		echo "CHECKPOINT pass."
		;;
	1)
		rm -rf *.safetensors
		wget -N https://huggingface.co/zuv0/test/resolve/main/petitcutie_v20.safetensors
		;;
	2)
		rm -rf *.safetensors
		wget -N https://huggingface.co/zuv0/test/resolve/main/milkTea_p.safetensors
		;;
	3)
		rm -rf *.safetensors
		wget -N https://huggingface.co/JosefJilek/loliDiffusion/resolve/main/loliDiffusionV0.14.0_RabbitV7_1.0M6-CLIP_VAE_FP16.safetensors
		;;
	4)
		rm -rf *.safetensors
		wget -N https://huggingface.co/zuv0/test/resolve/main/lemonmixV2_v30.safetensors
		;;
	5)
		rm -rf *.safetensors
		wget -N https://huggingface.co/zuv0/test/resolve/main/%E7%90%B3V7.fp16.safetensors
		;;
	6)
		rm -rf *.safetensors
		wget -N https://huggingface.co/zuv0/test/resolve/main/athAnimemix_v10.safetensors
		;;
	7)
		rm -rf *.safetensors
		wget -N https://huggingface.co/zuv0/test/resolve/main/ultimateHentai_v50.safetensors
		;;
	8)
		rm -rf *.safetensors
		wget -N https://huggingface.co/JosefJilek/loliDiffusion/resolve/main/loliDiffusionV0.14.0_AOM2-SFW_1.0M6-CLIP_VAE_FP16.safetensors
		;;
	r1)
		rm -rf *.safetensors
		wget -N https://huggingface.co/zuv0/test/resolve/main/sanciyuanfp16.safetensors
		;;
	gc)
		rm -rf *.safetensors
		echo "CHECKPOINT remove."
		exit 0
		;;
	*)
		echo "CHECKPOINT error!"
		;;
esac

cd ..
cd Lora

case $RC in
	0)
		echo "RC 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 "RC$digit" ]; then
				git clone https://huggingface.co/zuv0/RC$digit
				cd RC$digit
				git repack -a -d --depth=250 --window=250
				cd .git
				rm -rf lfs
				cd ..
				cd ..
			fi
		    RC=$((RC / 10))
		done
		;;
esac