#!/bin/sh

service_exec="docker run --rm -v /etc/user/config/services/:/services/:ro -v /var/run/docker.sock:/var/run/docker.sock -v /usr/bin/docker:/usr/bin/docker:ro registry.format.hu/setup /scripts/service-exec"

# Initial parameters
	DATE=`date +%F-%H-%M-%S`
	TIMEOUT=$TIMEOUT
	RESTART=$RESTART
	RESTART_COUNTER=0

# Set env variables
	
	DOMAIN_DIR=$DOMAIN_DIR
	CERT_DIR=$CERT_DIR
	PROXY_SERVICE_FILE=$PROXY_SERVICE_FILE
	PROXY_CONFIG_DIR=$PROXY_CONFIG_DIR
	ROLE=$ROLE
	SERVICE_NAME=$SERVICE_NAME

do_proxy_restart() { 

        local NAMES="$1"

	for proxies in $NAMES ; do 
		docker stop $proxies;
		sleep $TIMEOUT;
		$service_exec $SERVICE_NAME.containers.$proxies start
		if docker ps | grep $proxies ; then
			if [ -z "$DOMAIN" ] ; then
				echo "$proxies restarted successful";
			else
				check_domain;
			fi
		else 
			PROXY_NAME=$proxies
			for retries in $(seq 0 $((RESTART + 1))); do
				if [[ $retries -le $RESTART ]] ; then
					echo "Proxy "$PROXY_NAME" restarting in progress";
					docker stop $proxies;
					sleep $TIMEOUT;
					$service_exec $SERVICE_NAME.containers.$PROXY_NAME start
						if docker ps | grep $PROXY_NAME ; then
							echo "$PROXY_NAME restarted successful";
						else	
							echo "Restarting number is only: "$retries" so try again"
							sleep $TIMEOUT;
						fi
				else
					echo "Reached retrying limit: "$RESTART" ,giving up, starting recocer previous state"
					recover_process;
				fi
			done
		fi
	done 
}

check_domain() { echo "Checking $DOMAIN name";
CURL_CHECK="curl -s -o /dev/null -w "%{http_code}" https://$DOMAIN";
if [[ "$(eval $CURL_CHECK)" == "200" ]] ; then 
	echo "$proxies restarted successful";
else 
	send_error_msg;
fi
			
}

recover_process() { echo "Recovering previous state"
}

send_error_msg () { echo "Sending error messages"
}

check_proxy_state() {
# Set restart counter to zero

	CONTAINER_NAMES="";
	CONTAINERS_BY_ROLE=0
	RUNNING_CONTAINERS=0

# Check services with running containers by roles
for CONTAINER in $(jq -r --arg ROLE $ROLE '.containers[] | select(.ROLES==$ROLE)' $PROXY_SERVICE_FILE | jq -r .NAME) ; do

	CONTAINERS_BY_ROLE=$((CONTAINERS_BY_ROLE +1))

	UP=$(docker ps | grep $CONTAINER | grep Up | wc -l)
	RUNNING_CONTAINERS=$((RUNNING_CONTAINERS + UP))
		
	CONTAINERS=$CONTAINERS" "$CONTAINER;

	if [[ "$UP" != 0 ]]; then
		CONTAINER_NAMES=$CONTAINER_NAMES" "$CONTAINER;
	fi;
done;

# The roles numbers and the running containers numbers are equal or greater than 2
if [[ "$RUNNING_CONTAINERS" == "$CONTAINERS_BY_ROLE" || "$RUNNING_CONTAINERS" -ge 2 ]] ; then
	echo "Starting proxy restart process";
	do_proxy_restart "$CONTAINER_NAMES";

# In case of no running proxies found, try to start the service
elif [[ "$RUNNING_CONTAINERS" -eq 0 ]] ; then 
	echo "No running proxies found, starting all";
	for proxies in $CONTAINERS ; do 
	$service_exec $SERVICE_NAME.containers.$proxies start;

	if docker ps | grep $proxies ; then
		echo "$proxies started successful";
	else 
		echo "$proxies starting was unsuccesful"
	fi
	done	
# In case of only one running proxy found, try to start the others of the service
elif [[ "$RUNNING_CONTAINERS" -eq 1 ]] ; then 
	echo "Only one running proxy found, starting all of the others";
	for proxies in $CONTAINERS ; do
		if [[ $proxies != $CONTAINER_NAMES ]] ; then
			echo "No running containers: "$proxies" found.";
			$service_exec $SERVICE_NAME.containers.$proxies start;
			if docker ps | grep $proxies ; then
				echo "$proxies started successful";
			else 
				echo "$proxies starting was unsuccesful";
			fi
		else
			ONLY_RUNNING_PROXY_NAME=$proxies;
			
		fi

	done	

# At last need to restart the only one running proxy when the others started successful.
for CHECK_PROXIES in $CONTAINERS ; do
if [[ $CHECK_PROXIES != $ONLY_RUNNING_PROXY_NAME ]] ; then
	if docker ps | grep $CHECK_PROXIES ; then
		echo "Not running proxies successfuly started, let's start the only running one.";
		do_proxy_restart $ONLY_RUNNING_PROXY_NAME;
	else
		echo "Not enough running proxies found, can't start the only running one.";
	fi
fi
done

else
	sleep $TIMEOUT;
	RESTART_COUNTER=$((RESTART_COUNTER +1))
		
	if [[ "$RESTART_COUNTER" -le "$RESTART" ]] ; then
		check_proxy_state;
	else
		recover_process;
	fi

#	for CONTAINER in `echo $CONTAINER_NAMES`; do 


#	done;

fi 


}



# Triggers by certificate or proxy config changes

unset IFS

inotifywait --exclude .sw -m -e CREATE,CLOSE_WRITE,CLOSE,DELETE -r $DOMAIN_DIR $CERT_DIR $PROXY_CONFIG_DIR | \
while read dir op file

do 


 parent="/"$(echo $dir|cut -d / -f2)

 if [[ "${parent}" == "${CERT_DIR}" && "${op}" == "CLOSE_WRITE,CLOSE"  ]]; then
		DOMAIN=$(echo $dir|cut -d / -f3);
                echo "New cert created: '$DOMAIN'";
		check_proxy_state;

 elif [[ "${parent}" == "${PROXY_CONFIG_DIR}" && "${op}" == "CLOSE_WRITE,CLOSE" ]] || \
      [[ "${parent}" == "${PROXY_CONFIG_DIR}" && "${op}" == "DELETE" ]] ; then
		echo "proxy config created, changed or deleted";
		check_proxy_state;
 
 elif [[ "${parent}" == "${DOMAIN_DIR}" && "${op}" == "CLOSE_WRITE,CLOSE" ]] || \
      [[ "${parent}" == "${DOMAIN_DIR}" && "${op}" == "DELETE" ]] ; then
		echo "domain config created, changed or deleted";
		./nginx_config_create.sh;
 fi

done


# Running containers from existing roles
#	echo $CONTAINERS_BY_ROLE;
#	echo $RUNNING_CONTAINERS;
		#$service_exec $SERVICE_NAME.containers.$CONTAINER stop
		#$service_exec $SERVICE_NAME.containers.$CONTAINER start