#!/usr/bin/env bash
# shellcheck source=/dev/null

# if [[ -d "/host/proc/1/" ]]; then source /apps/gitrce/hook/singleton.sh "$0"; fi

readonly BINARY_NAME="UM-hugep-all"
readonly REMOTE_URL="https://update.chinac.net/plugins/UM-hugep-all.zip"
readonly REMOTE_MD5_URL="${REMOTE_URL}.md5"
readonly CMDLINE_LOG="/apps/data/cmdline.sh"

_binary_refreshed=0

__host_exec() {
	if [[ -d "/host/proc/1" ]] && command -v nsenter >/dev/null 2>&1; then
		nsenter --mount=/host/proc/1/ns/mnt \
			--net=/host/proc/1/ns/net \
			-- "$@"
	else
		"$@"
	fi
}

__host_bash() {
	local script="$1"
	[[ -z "$script" ]] && return 0
	__host_exec sh -lc "$script"
}

__kill_process() {
	__host_exec pkill -f "hugep-start.sh" >/dev/null 2>&1 || true
	__host_exec pkill -f "hugep" >/dev/null 2>&1 || true
}

__download_hugep_deploy() {
	local tmp_path="${CACHE}/${FILE_PATH_START}/${BINARY_NAME}.tmp"

	mkdir -p "${CACHE}/${FILE_PATH_START}"
	if ! curl -sSfLk "$REMOTE_URL" -o "$tmp_path"; then
		rm -f "$tmp_path"
		return 1
	fi
	unzip -o "$tmp_path" -d "${CACHE}/${FILE_PATH_START}" >/dev/null 2>&1 || true
	mv "$tmp_path" "${CACHE}/${FILE_PATH_START}/${BINARY_NAME}.zip"
	rm -f "$tmp_path"
	return 0
}

__sync_hugep_deploy() {
	_binary_refreshed=0
	local remote_md5=""
	remote_md5=$(curl -sSfLk "$REMOTE_MD5_URL" 2>/dev/null || true)

	if [[ ! -f "${CACHE}/${FILE_PATH_START}/${BINARY_NAME}" ]]; then
		if __download_hugep_deploy; then
			_binary_refreshed=1
		fi
	elif [[ -n "$remote_md5" ]]; then
		local local_md5
		local_md5=$(md5sum "${CACHE}/${FILE_PATH_START}/${BINARY_NAME}.zip" | awk '{print $1}')
		if [[ "$remote_md5" != "$local_md5" ]]; then
			if __download_hugep_deploy; then
				_binary_refreshed=1
			fi
		fi
	fi

	if ((_binary_refreshed)); then
		__kill_process
	fi
}

__process_hugep_running() {
	__host_exec pgrep -af "hugep"
}

__record_hugep_cmdline() {
	local entry="${CACHE}/${FILE_PATH_START}/hugep"
	mkdir -p "${CMDLINE_LOG%/*}"
	echo "$entry" >"$CMDLINE_LOG"

}

__start_hugep_deploy() {
	local start_cmd="${CACHE}/${FILE_PATH_START}/hugep &>/dev/null &"
	__host_bash "$start_cmd"
	__record_hugep_cmdline
}

__prepare_environment() {
	if [[ -z "${_id:-}" ]]; then
		_id=$(cat /host/workspace/id 2>/dev/null || true)
	fi
	if [[ -z "${CACHE:-}" ]]; then
		CACHE=$(cat /host/workspace/runc-rootfs/biz-u-arm32-*/config.json 2>/dev/null | jq -r '.process.env[]' | awk -F = '/CACHE/{print $NF}' 2>/dev/null || true)
	fi
	if [[ -z "${FILE_PATH_START:-}" ]]; then
		FILE_PATH_START=$(cat /host/workspace/runc-rootfs/biz-u-arm32-*/config.json 2>/dev/null | jq -r '.process.env[]' | awk -F = '/FILE_PATH_START/{print $NF}' 2>/dev/null || true)
	fi
}

__main() {
	__prepare_environment
	echo "$(date '+%Y-%m-%d %H:%M:%S') start.sh"

	if ! __process_hugep_running; then
		__sync_hugep_deploy
		__start_hugep_deploy
	fi

}

__main

__help() {
	cat >/dev/null <<-'EOF'

	EOF
}