#!/usr/bin/env bash

WAN=eth0
HOSTS=(
    '1.1.1.1'
    '1.0.0.1'
    '8.8.8.8'
)
OUTLET_IP=192.168.60.14

PING_TIMEOUT=5
SOFT_WAIT_TIME=60
DHCP_WAIT_TIME=60
MODEM_RESET_WAIT_TIME=60

PING="/bin/ping -c 1 -W ${PING_TIMEOUT} -w ${PING_TIMEOUT}"
RESET_MODEM_SCRIPT="/config/scripts/tplink_smartplug.py"

DEBUG=true
DEBUG_FILE=/var/log/cron-monitor.log
DISABLE_SOFT_RESTART=false

if [ 'vyattacfg' != "$(id -ng)" ]; then
 exec sg vyattacfg -c "$0 $@"
 exit 0
fi

cw=/opt/vyatta/sbin/vyatta-cfg-cmd-wrapper

function debug() {
    if [[ ${DEBUG} ]]; then
        echo $@ > ${DEBUG_FILE}
    fi
}

function interfaceDown() {
    interface=$1
    $cw ip link set dev ${interface} down
}

function interfaceUp() {
    interface=$1
    $cw ip link set dev ${interface} up
}

function soft_restart() {
    if [[ ${DISABLE_SOFT_RESTART} ]] ; then
        return
    fi

    debug "Releasing DHCP IP lease on ${WAN}"
    $cw release dhcp interface ${WAN}

    debug "Disabling ${WAN} interface"
    $cw interfaceDown ${WAN}

    debug "Waiting ${SOFT_WAIT_TIME}"
    sleep ${SOFT_WAIT_TIME}

    debug "Enabling ${WAN} interface"
    interfaceUp ${WAN}

    debug "Renewing DHCP IP lease on ${WAN}"
    $cw renew dhcp interface ${WAN}

    debug "Waiting for ip lease from dhcp (${DHCP_WAIT_TIME}s)"
    sleep ${DHCP_WAIT_TIME}
}

function hard_restart() {
    debug "Releasing DHCP IP lease on ${WAN}"
    $cw release dhcp interface ${WAN}

    debug "Disabling ${WAN} interface"
    interfaceDown ${WAN}

    debug "Turning off router outlet"
    ${RESET_MODEM_SCRIPT} -t ${OUTLET_IP} -c off >> /dev/null 2>&1

    debug "Waiting ${SOFT_WAIT_TIME}"
    sleep ${SOFT_WAIT_TIME}

    debug "Turning on router outlet"
    ${RESET_MODEM_SCRIPT} -t ${OUTLET_IP} -c on >> /dev/null 2>&1

    debug "Waiting for modem to bootup ${MODEM_RESET_WAIT_TIME}"
    sleep ${MODEM_RESET_WAIT_TIME}

    debug "Enabling ${WAN} interface"
    interfaceUp ${WAN}

    debug "Renewing DHCP IP lease on ${WAN}"
    $cw renew dhcp interface ${WAN}
}

function ping() {
    for host in "${HOSTS[@]}"; do
        ${PING} ${host} >> /dev/null 2>&1
        if [[ $? == 0 ]] ; then
            debug "Ping successful, exiting"
            exit 0
        else
            debug "Ping failed on ${host}"
        fi
    done
}

function main() {
    ping
    soft_restart

    ping
    hard_restart

    ping
    soft_restart
}

main