#!/bin/bash

set -eu
### --help Info: debug_manager.sh - утилита для отключения дебага Евы.
### --help Usage: debug_manager.sh [enable|disable] [--restart]
### --help Usage:     enable - включит дебаг в конфигурации аппа
### --help Usage:     disable - выключит дебаг в конфигурации аппа
### --help Usage:     --restart - перезапустит апп, если конфигурация поменялась;
### --help Usage:                 иначе, это произойдёт по cron
### --help Usage:     без параметров - пройдёт по всем аппам, проверит, что дебаг
### --help Usage:                      выключен; запомнит время, если обнаружет, что
### --help Usage:                      включили и выключит через пять дней при очередном
### --help Usage:                      запуске по cron.
### --help Usage: debug_manager.sh [enable|disable] --restart
### --help Usage: debug_manager.sh [enable|disable] --restart
### --help Example: debug_manager.sh
### --help Example: debug_manager.sh disable --restart
### --help Example: debug_manager.sh enable

. /opt/fox_utils/crab_sys.sh

date --rfc-3339=seconds

sys::usage "$@"

declare ARG_1 ARG_LIST
sys::arg_parse "$@"


declare NOW DAYS LAST_UPDATE DEBUG NEED_RESTART APP_STATE_CHANGED
NOW="$(date +%s)"
DAYS=4
DEBUG="True"
NEED_RESTART="FALSE"


_get_debug_value(){
	set -e
	local d common_config custom_config config value
	d="$1"
	common_config="$2"
	custom_config="$3"
	(
		cd "$d"
		if python3 manage.py shell "print(config.DEBUG)" &>/dev/null; then
			value="$(python3 manage.py shell "print(config.DEBUG)" 2> /dev/null)"
			value="$(grep -e ^True$ -e ^False$<<<$value)"
			value="${value:-"$DEBUG"}"
			echo $value
			return 0
		fi
		if grep '^DEBUG' "$custom_config" &>/dev/null; then
			config="$custom_config"
		elif grep '^DEBUG' "$common_config" &>/dev/null; then
			config="$common_config"
		elif grep '^DEBUG' "$d/cmf/config.py" &>/dev/null; then
			config="$d/cmf/config.py"
		else
			echo "$DEBUG"
			return 0
		fi
		value=$(sed 's/DEBUG\s*=\s*//g'<<<"$(grep -w -e "^DEBUG\s*=" $config)")
		echo "${value:-$DEBUG}"
	)
	return 0
}


_set_debug_value(){
	local config="$1"
	local value="$2"
	if [[ ! -f "$config" ]]; then
		echo "Файл $config отсутствует. Создаём."
		echo "DEBUG = $value" > $config
	elif grep '^DEBUG' $config; then
		sed -i "s/^DEBUG\s*=.*/DEBUG = $value/g" $config
	else
		echo "DEBUG = $value" > $config
	fi
	return 0
}


switch_debug_value(){
	local d app common_config custom_config debug_current_value value
	d="$1"
	app="$2"
	common_config="$3"
	custom_config="$4"
	debug_current_value="$5"
	echo "Меняю статус дебага для $app: debug_current_value=$debug_current_value"
	if [[ "$debug_current_value" == "True" ]]; then
		_set_debug_value "$custom_config" False
	else
		_set_debug_value "$custom_config" True
	fi
	# Рестарт должен делать либо cron, либо флаг --restart
	# NEED_RESTART="TRUE"
	APP_STATE_CHANGED=TRUE
	return 0
}


cron_disabler(){
	local d app common_config custom_config
	local debug_marker debug_current_value enabled_date msg
	d="$1"
	app="$2"
	common_config="$3"
	custom_config="$4"
	debug_current_value="$5"
	debug_marker="$d/custom/debug_disabler.marker"
	if [[ "$debug_current_value" == "False" ]]; then
		# Могли выключить вручную, тогда маркер больше не нужен
		[[ -f "$debug_marker" ]] && echo "Дебаг выключен. Удаляю маркер $debug_marker"
		[[ -f "$debug_marker" ]] && rm -f "$debug_marker"
		return 0
	fi
	if [[ ! -f "$debug_marker" ]] || [[ ! -s "$debug_marker" ]]; then
		echo $NOW > "$debug_marker"
		echo "Дебаг приложения $app включен, запоминаю дату"
		return 0
	fi
	enabled_date="$(<"$debug_marker")"
	if ((NOW > enabled_date+3600*24*DAYS)); then
		NEED_RESTART="TRUE"
		msg="Отключаю дебаг приложения $app."
		msg+=" debug_current_value=$debug_current_value"
		msg+=", enabled_date=$(date  --rfc-3339=seconds -d @$enabled_date)"
		echo "$msg"
		_set_debug_value "$custom_config" False
		rm -f "$debug_marker"
	fi
	return 0
}


main(){
	local d action app common_config custom_config debug_current_value value
	action="${1:-}"
	for d in $(find /opt/ -mindepth 1 -maxdepth 1 -type d); do
		app="${d##*/}"
		common_config="$d/common/config.py"
		custom_config="$d/custom/config.py"
		if [[ ! -f "$common_config" ]]; then
			continue
		fi
		echo "$d"
		debug_current_value="$(_get_debug_value "$d" "$common_config" "$custom_config")"
		if [[ "$action" == "enable" ]]; then
			value="True"
		elif [[ "$action" == "disable" ]]; then
			value="False"
		else
			# По-умолчанию выключаем дебаг через N ней
			cron_disabler "$d" "$app" "$common_config" "$custom_config" "$debug_current_value"
			continue
		fi
		if [[ "$debug_current_value" != "$value" ]]; then
			switch_debug_value "$d" "$app" "$common_config" "$custom_config" "$debug_current_value"
		fi
	done
	if [[ "$NEED_RESTART" == "TRUE"
		|| "${ARG_RESTART:-}" == "TRUE"
		&& "${APP_STATE_CHANGED:-}" == "TRUE" ]]; then
		/opt/bin/services restart
	fi
	return 0
}

main "$@"

date --rfc-3339=seconds
exit 0