#!/bin/bash

# Tente de réparer les conteneurs LXC à partir des snapshots et des sauvegardes.

# Récupère le dossier du script
if [ "${0:0:1}" == "/" ]; then script_dir="$(dirname "$0")"; else script_dir="$(echo $PWD/$(dirname "$0" | cut -d '.' -f2) | sed 's@/$@@')"; fi

source $script_dir/ynh_lxd
source $script_dir/ynh_lxd_demo
source /usr/share/yunohost/helpers

app=${__APP__:-yunohost_demo}
final_path=$(ynh_app_setting_get --app=$app --key=final_path)
lxdbr_demo_network=$(ynh_app_setting_get --app=$app --key=lxdbr_demo_network)
lxc_ip1=$(ynh_app_setting_get --app=$app --key=lxc_ip1)
lxc_ip2=$(ynh_app_setting_get --app=$app --key=lxc_ip2)
lxc_name1=$(ynh_app_setting_get --app=$app --key=lxc_name1)
lxc_name2=$(ynh_app_setting_get --app=$app --key=lxc_name2)

ynh_print_info --message=">> Starting demo restore from crash."

ynh_print_info --message="> Disabling switch cron."
sed -i "s/.*demo_switch.sh/#&/" /etc/cron.d/demo_switch	# Le cron est commenté durant l'opération de maintenance.

ynh_print_info --message="> Deleting locks and stoping LXC containers."
ynh_secure_remove --file="/var/lib/lxd/$lxc_name1.lock_fileS"
ynh_secure_remove --file="/var/lib/lxd/$lxc_name2.lock_fileS"
ynh_secure_remove --file="/var/lib/lxd/$lxc_name1.lock_fileU"
ynh_secure_remove --file="/var/lib/lxd/$lxc_name2.lock_fileU"

ynh_lxc_demo_stop --name="$lxc_name1"
ynh_lxc_demo_stop --name="$lxc_name2"

# Vérifie l'état des conteneurs.
ynh_lxc_check_container_start --name=$lxc_name1
LXC1_STATUS=$?
ynh_lxc_check_container_start --name=$lxc_name2
LXC2_STATUS=$?

if [ $LXC1_STATUS -eq 1 ]; then
	ynh_print_info --message="> LXC container $lxc_name1 is broken."
else
	ynh_print_info --message="> LXC container $lxc_name1 is working."
fi
if [ $LXC2_STATUS -eq 1 ]; then
	ynh_print_info --message="> LXC container $lxc_name2 is broken."
else
	ynh_print_info --message="> LXC container $lxc_name2 is working."
fi

# Restauration des snapshots
if [ $LXC1_STATUS -eq 1 ]; then
	ynh_lxc_demo_restore_from_snapshot  --name=$lxc_name1
	LXC1_STATUS=$?
fi
if [ $LXC2_STATUS -eq 1 ]; then
	ynh_lxc_demo_restore_from_snapshot  --name=$lxc_name2
	LXC2_STATUS=$?
fi

# Restauration des archives des snapshots
if [ $LXC1_STATUS -eq 1 ]; then
	ynh_lxc_demo_restore_from_archive  --name=$lxc_name1
	LXC1_STATUS=$?
fi
if [ $LXC2_STATUS -eq 1 ]; then
	ynh_lxc_demo_restore_from_archive  --name=$lxc_name2
	LXC2_STATUS=$?
fi

# Si des erreurs persistent, tente de cloner depuis un conteneur sain
if [ $LXC1_STATUS -eq 1 ] && [ $LXC2_STATUS -eq 0 ] ; then
	ynh_lxc_clone --source=$lxc_name2 --destination=$lxc_name1
	LXC1_STATUS=$?
fi
if [ $LXC2_STATUS -eq 1 ] && [ $LXC1_STATUS -eq 0 ]; then
	ynh_lxc_clone --source=$lxc_name1 --destination=$lxc_name2
	LXC2_STATUS=$?
fi

# Résultats finaux
if [ $LXC1_STATUS -eq 1 ] || [ $LXC2_STATUS -eq 1 ]; then
	if [ $LXC1_STATUS -eq 1 ]; then
		ynh_print_info --message="> $lxc_name1 LXC container can't be repaired..."
	fi
	if [ $LXC2_STATUS -eq 1 ]; then
		ynh_print_info --message="> $lxc_name2 LXC container can't be repaired..."
	fi
else
	ynh_print_info --message="> The 2 LXC containers are working."
fi

ynh_print_info --message="> Enabling switch cron."
sed -i "s/#*\*/\*/" /etc/cron.d/demo_switch	# Le cron est décommenté
ynh_print_info --message="> Restart the demo."
$final_path/demo_start.sh

ynh_print_info --message=">> Finished demo restore from crash."