#!/bin/sh
QMP_PATH="/etc/qmp"

. $QMP_PATH/qmp_common.sh
. $QMP_PATH/qmp_functions.sh
. $QMP_PATH/qmp_gw.sh
. $QMP_PATH/qmp_wireless.sh
. $QMP_PATH/qmp_network.sh
. $QMP_PATH/qmp_update.sh
. $QMP_PATH/qmp_system.sh

offer_default_gw() {
	qmp_gw_default offer $1
	qmp_gw_apply
}

search_default_gw() {
	qmp_gw_default search $1
	qmp_gw_apply
}

disable_default_gw() {
	qmp_gw_default disable $1
	qmp_gw_apply
}

reset_wifi() {
	qmp_reset_wifi
	configure_wifi
}

configure_wifi() {
	# Configure WiFi on qMp config file
	qmp_configure_wifi_initial
	# Configure WiFi on OpenWrt config file
	qmp_configure_wifi
	# Configure the whole network to apply wifi settings
	configure_network
}

configure_gw() {
	qmp_gw_apply
}

apply_services() {
	qmp_set_services
}

configure_network() {
	qmp_configure
	qmp_bmx6_reload
	/etc/init.d/network reload
	/etc/init.d/network restart
	if /etc/init.d/gwck enabled
	then
		/etc/init.d/gwck restart
	fi
	/etc/init.d/dnsmasq restart
	qmp_restart_firewall
}

configure_system() {
	qmp_configure_system
	apply_services
	qmp_bmx6_reload
	/etc/init.d/uhttpd restart
}

enable_ns_xm_ppt() {
	echo 8 > /sys/class/gpio/export
	echo out > /sys/class/gpio/gpio8/direction
	echo 1 > /sys/class/gpio/gpio8/value
}

enable_ns_xw_ppt() {
	echo 2 > /sys/class/gpio/export
	echo out > /sys/class/gpio/gpio2/direction
	echo 1 > /sys/class/gpio/gpio2/value
}

publish_hna() {
	[ -z "$1" ] && help
	qmp_publish_hna_bmx6 $1 $2
}

unpublish_hna() {
	[ -z "$1" ] && help
	qmp_unpublish_hna_bmx6 $1
}

upgrade() {
	qmp_update_upgrade_system $1 && hard_reboot
}

hard_reboot() {
	echo "System is gonna be rebooted now!"
	echo 1 > /proc/sys/kernel/sysrq
	echo b > /proc/sysrq-trigger
}

configure_all() {
	configure_system
	configure_wifi
	configure_network
}

safe_apply() {
	[ -e "/tmp/qmp.save.tar.gz" ] && echo "Found saved state at /tmp/qmp.save.tar.gz. Make sure you want to use it!" \
	|| {
		echo "Cannot found saved state, saving it..."
		save_state
	}

	touch /tmp/qmp.safe.test
	echo "------------------------------------------------------------------------------------"
	echo "File /tmp/qmp.safe.test has been created, after configuring the system you will have
	180 seconds to remove it or the previous state will be recovered"
	echo "------------------------------------------------------------------------------------"

	read -p'Do you agree?[y,N] ' a
	[ "$a" != "y" ] && return

	configure_all

	( sleep 180
		[ -e "/tmp/qmp.safe.test" ] && {
			cp -f /etc/config/qmp /tmp/qmp.wrong
			recover_state
			cp -f /tmp/qmp.wrong /etc/config/qmp.wrong
			hard_reboot
		} || rm -f "/tmp/qmp.safe.test"
	) &
}

save_state() {
	echo "Saving state at $(date) in file /tmp/qmp.save.tar.gz"
	rm -f /tmp/qmp.save.tar.gz 2>/dev/null
	(cd /overlay/etc && tar czf /tmp/qmp.save.tar.gz .)
	[ ! -e /tmp/qmp.save.tar.gz ] && echo "ERROR: cannot save state, exiting..." && exit 1
}

recover_state() {
	echo "Recovering state at $(date) from /tmp/qmp.save.tar.gz"
	[ -e "/tmp/qmp.save.tar.gz" ] && tar xvzf /tmp/qmp.save.tar.gz -C /overlay/etc/ \
	|| echo "Cannot recover state because it has not been saved before"
}

help() {
	echo "Use: $0 <function> [params]"

	echo ""
	echo "Configuration:"
	echo " configure_all			: Configure and apply all settings"
	echo " configure_network		: Configure and apply network settings"
	echo " configure_system 		: Configure and apply system settings (qmp.node section and so on)"
	echo " configure_wifi			: Configure all WiFi devices"
	echo " reset_wifi			: Reset, rescan and configure all the WiFi devices"
	echo " configure_gw			: Configure and apply gateways settings"

	echo ""
	echo "Safe configuration:"
	echo " save_state			: Saves current state of configuration files"
	echo " recover_state			: Recovers previous saved state"
	echo " safe_apply			: Performs a safe configure_all. If something wrong it comes back to old state"

	echo ""
	echo "Gateways:"
	echo " offer_default_gw [ipv4|ipv6]	: Offers default gw to the network IPv4 or IPv6, both versions if no value"
	echo " search_default_gw [ipv4|ipv6]	: Search for a default gw in the network IPv4 or IPv6, both versions if no value"
	echo " disable_default_gw [ipv4|ipv6]	: Disables the search/offer of default GW IPv4 and/or IPv6"
	echo " publish_hna			: Publish an IP range (v4 or v6): publish_hna <IP/NETMASK> [ID]"
	echo " unpublish_hna			: Unpublish a current HNA: unpublish_hna <ID>"

	echo ""
	echo "Other:"
	echo " apply_services			: Start/stop services depending on qmp configuration"
	echo " enable_ns_xm_ppt		: Enable PoE passtrough from NanoStation XM M2/M5 devices. Be careful with this option!"
	echo " enable_ns_xw_ppt		: Enable PoE passtrough from NanoStation XW M2/M5 devices. Be careful with this option!"
	echo " upgrade [URL]			: Upgrade system. By default to the last version, but image url can be provided to force"
	echo " hard_reboot				: Performs a hard reboot (using kernel sysrq)"

	echo ""
	exit 0
}


[ -z "$1" ] && help

echo "executing function $1..."

$@ # 2>/dev/null || echo " Command not found.\n" && help