#!/bin/bash

function usage() {
	echo "uasge: $0 {start|restart|kill|toc|build|publish|release_src|release|help|-h}"
}

function kill_teedoc() {
	ps aux | grep teedoc | grep python3 | awk '{print $2}' | xargs kill -9 >/dev/null
}

function teedoc_build() {
	teedoc build
	ret=$?
	retry_count=1
	# 等于1表示，文件夹存在，重新运行
	while ((ret==1))
	do 
		teedoc build
		ret=$?
		let "retry_count=$retry_count+1"
		echo $retry_count
	done
}

function restart() {
	teedoc_build
	kill_teedoc
	teedoc serve &
}

function start() {
	teedoc serve &
}

function creat_toc() {
	# 读取所有的 sidebar.yml 生成目录
	echo "生成目录"
	python3 script/auto_creat_toc.py --out_file Quecpython_toc.yml --input_file docs/Quecpython_intro/zh/config.json --action toc

}

function teedoc_release() {
	# 检查是否存在 out文件
	release_filename=Community-document-$(date "+%Y%m%d-%H%M")
	if [ ! -d "out" ]; then
		teedoc_build
	fi
	tar cf ${release_filename}.tar out
	tar jcf ${release_filename}.tar.bz2 ${release_filename}.tar
	rm -rf ${release_filename}.tar
	mv ${release_filename}.tar.bz2 ..
}

function teedoc_src_release() {
	release_filename=Community-document-src-$(date "+%Y%m%d-%H%M")
	tar -zcvf ${release_filename}.tar.gz --exclude-from .tar_excludefile .
	mv ${release_filename}.tar.gz .. 
}

function teedoc_publish_to_server() {
	OUT_DOC_TOP_DIC=./out/doc
	cp ${OUT_DOC_TOP_DIC}/doc/*.html ${OUT_DOC_TOP_DIC}
	cp ${OUT_DOC_TOP_DIC}/doc/*.ico ${OUT_DOC_TOP_DIC}
	cp ${OUT_DOC_TOP_DIC}/doc/*.json ${OUT_DOC_TOP_DIC}
	rm ${OUT_DOC_TOP_DIC}/doc/*.html
	rm ${OUT_DOC_TOP_DIC}/doc/*.ico
	rm ${OUT_DOC_TOP_DIC}/doc/*.json
	# 首页的英文页面
	mkdir ${OUT_DOC_TOP_DIC}/en/assets/
	cp ${OUT_DOC_TOP_DIC}/doc/en/*.html ${OUT_DOC_TOP_DIC}/en/
	cp ${OUT_DOC_TOP_DIC}/doc/en/*.ico ${OUT_DOC_TOP_DIC}/en/
	cp ${OUT_DOC_TOP_DIC}/doc/en/*.json ${OUT_DOC_TOP_DIC}/en/
	cp -r ${OUT_DOC_TOP_DIC}/assets/* ${OUT_DOC_TOP_DIC}/en/assets/
	rm -rf ${OUT_DOC_TOP_DIC}/doc/en/
	
	

	case $1 in
	"root@192.168.25.215:/home/dist/doc")
		echo "确认发布到服务器: "
		sudo chmod -R 777 ${OUT_DOC_TOP_DIC}
		rsync ${OUT_DOC_TOP_DIC}/ $1 -r -i
		;;
	*)
		echo "发布到 /www/wwwroot/test.com/doc 文件夹"
		sudo chmod -R 777 ${OUT_DOC_TOP_DIC}
		rsync ${OUT_DOC_TOP_DIC}/ /www/wwwroot/test.com/doc -r -i >/dev/null
		;;
	esac
}

function copy_file() {
	python3 script/auto_creat_toc.py --input_file pages/index/zh/config.json --action copy
}

function apply_patch() {
	. script/out_doc_static_css_theme_default_light_patch.sh 
}

case $1 in
"start")
	start
	;;
"restart")
	restart
	;;
"kill")
	kill_teedoc
	;;
"--help")
	usage
	;;
"build")
	teedoc_build
	;;
"-h")
	usage
	;;
"release")
	teedoc_release
	;;
"release_src")
	teedoc_src_release
	;;
"publish")
	apply_patch
	teedoc_publish_to_server $2
	;;

"toc")
	# 生成目录
	creat_toc
	;;
"copy")
	# 覆盖相同的文档
	copy_file
	;;
"patch")
	apply_patch
	;;
*)
	if [ $# = 0 ]; then
		# 没有参数，默认
		teedoc_build
	else
		# 错误的参数
		echo "ERROR: $0 $1  错误的参数"
		usage
		exit -1
	fi
	;;
esac