#!/bin/bash


NAME="lora-network-server"
LOCK="/var/lock/$NAME"
ENABLED="yes"


[ -f /etc/default/$NAME ] && source /etc/default/$NAME

daemon_start="/usr/sbin/start-stop-daemon --start --background"
daemon_stop="/usr/sbin/start-stop-daemon --stop --quiet --oknodo"
run_dir=/var/run/lora
conf_dir=/var/config/lora
conf_file=$conf_dir/lora-network-server.conf

net_server=/opt/lora/lora-network-server
net_server_log=/var/log/lora-network-server.log
net_server_pidfile=$run_dir/$NAME.pid

pkt_fwd=$run_dir/1/lora_pkt_fwd
pkt_fwd_log=/var/log/lora-pkt-fwd-1.log
pkt_fwd_pidfile=$run_dir/lora-pkt-fwd-1.pid

pkt_fwd_2=$run_dir/2/lora_pkt_fwd
pkt_fwd_2_log=/var/log/lora-pkt-fwd-2.log
pkt_fwd_2_pidfile=$run_dir/lora-pkt-fwd-2.pid

port1=/sys/devices/platform/mts-io/ap1
port2=/sys/devices/platform/mts-io/ap2

lora_1_0_hw="MTAC-LORA-1.0"
lora_1_5_h_hw="MTAC-LORA-1.5"
lora_2_1_hw="MTAC-LORA-2.1"
lora_2g4_hw="MTAC-LORA-2G4-0.0"

lora_mtac_id="MTAC-LORA"
lora_mtac_003_id="MTAC-003"
lora_mtac_003_id868="MTAC-003E00"
lora_mtac_003_id915="MTAC-003U00"

lora_mtcap_id="MTCAP-LORA"
lora_mtcap_id868="MTCAP-LORA-868"
lora_mtcap_id915="MTCAP-LORA-915"

lora_mtcap3_id="MTCAP3"
lora_mtcap3_id868="MTCAP3-003E00"
lora_mtcap3_id915="MTCAP3-003U00"

lora_mtac_2g4_id="MTAC-LORA-2G4"

lora_mtac_g_id="MTAC-LORA-G"
lora_mtac_g16_id868="MTAC-LORA-G16-868"
lora_mtac_g16_id915="MTAC-LORA-G16-915"
lora_mtac_g64_id868="MTAC-LORA-G64-868"
lora_mtac_g64_id915="MTAC-LORA-G64-915"

pkt_fwd_options=""
dual_cards_installed=false
gps_path="/dev/gps0"

read_lora_hw_info() {
    set +e

    hw_id=$(mts-io-sysfs show hw-version)

    # product-id of first lora card
    lora_id=$(mts-io-sysfs show lora/product-id 2> /dev/null)
    if [ $? -ne 0 ]; then
        lora_id="NONE"
    fi
    lora_eui=$(mts-io-sysfs show lora/eui 2> /dev/null)
    if [ $? -ne 0 ]; then
        lora_eui=$(mts-io-sysfs show mac-eth)
        lora_eui=${lora_eui:0:8}":FF:FF"${lora_eui:8:16}
    fi
    # remove all colons
    lora_eui_raw=${lora_eui//:}
    lora_hw=$(mts-io-sysfs show lora/hw-version 2> /dev/null)
    if [ $? -ne 0 ]; then
        lora_hw="NONE"
    fi

    if [ -d $port1 ] && [[ $(cat $port1/hw-version) =~ "LORA-1.5" ]] && [ -d $port2 ] && [[ $(cat $port2/hw-version) =~ "LORA-1.5" ]]; then
        dual_cards_installed=true
    elif [ -d $port1 ] && [[ $(cat $port1/hw-version) =~ "MTAC-003" ]] && [ -d $port2 ] && [[ $(cat $port2/hw-version) =~ "MTAC-003" ]]; then
        dual_cards_installed=true
    elif [ -d $port1 ] && [[ $(cat $port1/hw-version) =~ "LORA-2G4" ]] && [ -d $port2 ] && [[ $(cat $port2/hw-version) =~ "LORA-2G4" ]]; then
        dual_cards_installed=true
    fi

    if [[ "$dual_cards_installed" = "true" ]] && [ -d $port2 ] && [[ $(cat $port2/hw-version) =~ "LORA-1.5" || $(cat $port2/hw-version) =~ "LORA-2G4" || $(cat $port2/hw-version) =~ "MTAC-003" ]]; then
        # product-id of first lora card
        lora_2_id=$(mts-io-sysfs show lora-2/product-id 2> /dev/null)
        lora_2_hw=$(mts-io-sysfs show lora-2/hw-version 2> /dev/null)
        lora_2_eui=$(mts-io-sysfs show lora-2/eui 2> /dev/null)
        # remove all colons
        lora_2_eui_raw=${lora_2_eui//:}
    fi

    set -e

    # Check USB id for Semtech Card
    if [[ $(lsusb | grep 05c9:5740) =~ "05c9:5740" ]] && [[ ! "$lora_id" =~ "$lora_mtac_2g4_id" ]]; then
       lora_id=$lora_mtac_2g4_id
       lora_hw=$lora_2g4_hw
       lora_eui=$(mts-io-sysfs show mac-eth)
       lora_eui=${lora_eui:0:8}":FF:FF"${lora_eui:8:16}
       LORA_CAPABLE=true
    # Check USB id for Semtech Card in Bootloader mode
    elif [[ $(lsusb | grep 0483:df11) =~ "0483:df11" ]]; then
        echo "Semtech 2g4 MTAC card detected in Bootloader Mode, remove power from Conduit and restore"
        return 1
    fi
}

setup_mtcdt_2g4() {
    mts-io-sysfs store lora/reset 0
    mts-io-sysfs store lora/boot 0
    sleep 0.25
    mts-io-sysfs store lora/reset 1
    GLOBAL_CONF=/opt/lora/global_conf.json.2g4
    set +e
    diff $GLOBAL_CONF /opt/lora/global_conf.json 1>/dev/null
    if [ $? -ne 0 ]; then
        cp $GLOBAL_CONF /opt/lora/global_conf.json
    fi
    set -e
    ln -sf /opt/lora/lora_pkt_fwd_2g4 $pkt_fwd
    ln -sf /opt/lora/lora_pkt_fwd_2g4 $pkt_fwd_2
    return 0
}

setup_mtcdt_2_1() {
    if [[ "$lora_id" = "$lora_mtac_g16_id868" ]]; then
        GLOBAL_CONF=/opt/lora/global_conf.json.MTAC_LORA_2_1_loc_single_antenna_16ch_EU868
    elif [[ "$lora_id" = "$lora_mtac_g16_id915" ]]; then
        GLOBAL_CONF=/opt/lora/global_conf.json.MTAC_LORA_2_1_loc_single_antenna_16ch_US915
    elif [[ "$lora_id" = "$lora_mtac_g16_id868" ]]; then
        GLOBAL_CONF=/opt/lora/global_conf.json.MTAC_LORA_2_1_test_single_antenna_64ch_8x8_EU868
    elif [[ "$lora_id" = "$lora_mtac_g64_id915" ]]; then
        GLOBAL_CONF=/opt/lora/global_conf.json.MTAC_LORA_2_1_loc_single_antenna_64ch_US915
    else
        return 1
    fi

    set +e
    diff $GLOBAL_CONF /opt/lora/global_conf.json 1>/dev/null
    if [ $? -ne 0 ]; then
        cp $GLOBAL_CONF /opt/lora/global_conf.json
    fi
    set -e

    ln -sf /opt/lora/pkt_forwarder $pkt_fwd

    pkt_fwd_options=" -g"
}

setup_mtcdt003() {
    if [[ "$lora_id" =~ "$lora_mtac_003_id868" ]]; then
        GLOBAL_CONF=/opt/lora/global_conf.json.MTAC_003_0_0.EU868
    elif [[ "$lora_id" =~ "$lora_mtac_003_id915" ]]; then
        GLOBAL_CONF=/opt/lora/global_conf.json.MTAC_003_0_0.US915
    else
        return 1
    fi

    ln -sf /opt/lora/lora_pkt_fwd_sx1303 $pkt_fwd
    ln -sf /opt/lora/lora_pkt_fwd_sx1303 $pkt_fwd_2

    # ignore error of diff command
    set +e
    diff $GLOBAL_CONF /opt/lora/global_conf.json 1>/dev/null
    if [ $? -ne 0 ]; then
        cp $GLOBAL_CONF /opt/lora/global_conf.json
    fi
    set -e

    return 0
}

setup_mtcap3() {
    if [[ "$lora_id" = "$lora_mtcap3_id868" ]]; then
        GLOBAL_CONF=/opt/lora/global_conf.json.MTCAP3.EU868
    elif [[ "$lora_id" = "$lora_mtcap3_id915" ]]; then
        GLOBAL_CONF=/opt/lora/global_conf.json.MTCAP3.US915
    else
        return 1
    fi

    ln -sf /opt/lora/lora_pkt_fwd_sx1303 $pkt_fwd

    # ignore error of diff command
    set +e
    diff $GLOBAL_CONF /opt/lora/global_conf.json 1>/dev/null
    if [ $? -ne 0 ]; then
        cp $GLOBAL_CONF /opt/lora/global_conf.json
    fi
    set -e

    return 0
}

setup_mtcdt() {
    GLOBAL_CONF=/opt/lora/global_conf.json.MTAC_LORA_1_0
    if [ "$lora_hw" = "$lora_1_0_hw" ] && [[ ! "$lora_id" =~ .*-SPI ]]; then
        ln -sf /opt/lora/basic_pkt_fwd-usb $pkt_fwd
    else
        if [ "$lora_hw" = "$lora_1_5_h_hw" ]; then
            GLOBAL_CONF=/opt/lora/global_conf.json.MTAC_LORA_1_5
        fi
        ln -sf /opt/lora/lora_pkt_fwd $pkt_fwd
        ln -sf /opt/lora/lora_pkt_fwd $pkt_fwd_2
    fi

    # ignore error of diff command
    set +e
    diff $GLOBAL_CONF /opt/lora/global_conf.json 1>/dev/null
    if [ $? -ne 0 ]; then
        cp $GLOBAL_CONF /opt/lora/global_conf.json
    fi
    set -e

    return 0
}

setup_mtcap() {
    hw=$(mts-io-sysfs show hw-version 2> /dev/null)

    if [ "$lora_id" = "$lora_mtcap_id868" ]; then
        if [ "$hw" = "MTCAP-0.1" ]; then
            GLOBAL_CONF=/opt/lora/global_conf.json.MTCAP_LORA_1_5.EU868
        else
            GLOBAL_CONF=/opt/lora/global_conf.json.MTCAP2_LORA_1_5.EU868
        fi
    elif [ "$lora_id" = "$lora_mtcap_id915" ]; then
        if [ "$hw" = "MTCAP-0.1" ]; then
            GLOBAL_CONF=/opt/lora/global_conf.json.MTCAP_LORA_1_5.US915
        else
            GLOBAL_CONF=/opt/lora/global_conf.json.MTCAP2_LORA_1_5.US915
        fi
    else
        return 1
    fi

    ln -sf /opt/lora/lora_pkt_fwd $pkt_fwd

    set +e
    diff $GLOBAL_CONF /opt/lora/global_conf.json 1>/dev/null
    if [ $? -ne 0 ]; then
        cp $GLOBAL_CONF /opt/lora/global_conf.json
    fi
    set -e

    return 0
}


hardware_found() {
    if [[ "$lora_id" =~ "$lora_mtac_g_id" ]]; then
        setup_mtcdt_2_1
    elif [[ "$lora_id" =~ "$lora_mtac_2g4_id" ]]; then
        setup_mtcdt_2g4
    elif [[ "$lora_id" =~ "$lora_mtac_003_id" ]]; then
        setup_mtcdt003
    elif [[ "$lora_id" =~ "$lora_mtac_id" ]]; then
        setup_mtcdt
    elif [[ "$lora_id" =~ "$lora_mtcap3_id" ]]; then
        setup_mtcap3
    elif [[ "$lora_id" =~ "$lora_mtcap_id" ]]; then
        setup_mtcap
    else
        return 1
    fi
}

do_start() {
    # create run directory
    mkdir -p $run_dir/1
    mkdir -p $run_dir/2

    # remove any existing json configs from other init scripts
    rm -rf $run_dir/1/*.json
    rm -rf $run_dir/2/*.json

    read_lora_hw_info

    if ! [ -f $conf_file ]; then
        echo "$0: $conf_file missing"
        exit 1
    fi

    if hardware_found; then
        echo "Found $lora_id with $lora_hw hardware"

        if [[ "$lora_id" =~ "$lora_mtac_g_id" ]]; then
            echo Resetting 2.1 LoRa
            /usr/sbin/mts-util-lora2-reset -g -f >/dev/null 2>&1
            sleep 0.25
        elif [[ ! "$lora_id" =~ "$lora_mtcap3_id" ]] && [[ ! "$lora_id" =~ "$lora_mtac_003_id" ]]; then
            mts-io-sysfs store lora/reset 0
            if [ "$dual_cards_installed" == true ]; then
                mts-io-sysfs store lora-2/reset 0
            fi
            sleep 0.25
            mts-io-sysfs store lora/reset 1
            if [ "$dual_cards_installed" == true ]; then
            mts-io-sysfs store lora-2/reset 1
            fi
            sleep 0.25
        fi
    else
        echo "$0: Lora hardware not detected"
        # Use MAC Address for lora eui
        lora_eui=$(mts-io-sysfs show mac-eth)
        lora_eui=${lora_eui:0:8}":FF:FF"${lora_eui:8:16}
        lora_hw=$lora_mtcap_id915
        lora_id=$lora_1_5_h_hw
        no_lora_hw=true
    fi

    if [ "$dual_cards_installed" == "true" ]; then
        lora_2_args="--lora-hw-2 $lora_2_hw --lora-prod-2 $lora_2_id --lora-eui-2 $lora_2_eui"
    fi

    echo -n "Starting $NAME: "
    # start network server

    lns_exec="exec $net_server -c $conf_file --lora-eui $lora_eui --lora-hw-1 $lora_hw --lora-prod-1 $lora_id $lora_2_args \
        --lora-path $run_dir  --noconsole 2>&1"
    $daemon_start --make-pidfile --pidfile $net_server_pidfile --startas /bin/bash -- -c "$lns_exec"

    echo "check process lora-network-server with pidfile $net_server_pidfile
    program start = \"$daemon_start --make-pidfile --pidfile $net_server_pidfile --startas /bin/bash -- -c '$lns_exec'\"
    program stop = \"$daemon_stop --pidfile $net_server_pidfile --retry TERM/60/KILL/5\"" > /etc/monit.d/lora-network-server

    if [ "$no_lora_hw" != "true" ]; then
        # start packet forwarder

        sleep 4
        pkf_exec="exec $pkt_fwd $pkt_fwd_options -l $pkt_fwd_log"

        $daemon_start --chdir $run_dir/1 --make-pidfile --pidfile $pkt_fwd_pidfile --startas /bin/bash -- -c "$pkf_exec"

        echo "check process lora-pkt-fwd-1 with pidfile $pkt_fwd_pidfile
        program start = \"$daemon_start --chdir $run_dir/1 --make-pidfile --pidfile $pkt_fwd_pidfile --startas /bin/bash -- -c '$pkf_exec'\"
        program stop = \"$daemon_stop --pidfile $pkt_fwd_pidfile --retry 5\"" > /etc/monit.d/lora-pkt-fwd-1

        if [ "$dual_cards_installed" == "true" ]; then
            pkf_2_exec="exec $pkt_fwd_2 $pkt_fwd_options -l $pkt_fwd_2_log"
            $daemon_start --chdir $run_dir/2 --make-pidfile --pidfile $pkt_fwd_2_pidfile --startas /bin/bash -- -c "$pkf_2_exec"
            echo "check process lora-pkt-fwd-2 with pidfile $pkt_fwd_2_pidfile
            program start = \"$daemon_start --chdir $run_dir/2 --make-pidfile --pidfile $pkt_fwd_2_pidfile --startas /bin/bash -- -c '$pkf_2_exec'\"
            program stop = \"$daemon_stop --pidfile $pkt_fwd_2_pidfile --retry 5\"" > /etc/monit.d/lora-pkt-fwd-2
        fi
    fi

    renice -n -20 -p $(pgrep lora-network-se)
    renice -n -20 -p $(pgrep $(basename $pkt_fwd))
    monit reload

    echo "OK"
}


do_stop() {
    echo -n "Stopping $NAME: "
    if monit summary lora-network-server > /dev/null 2>&1; then
        monit unmonitor lora-network-server
        start-stop-daemon --stop --quiet --oknodo --pidfile $net_server_pidfile --retry TERM/60/KILL/5
        rm -f $net_server_pidfile
    fi
    if monit summary lora-pkt-fwd-1 > /dev/null 2>&1; then
        monit unmonitor lora-pkt-fwd-1
        start-stop-daemon --stop --quiet --oknodo --pidfile $pkt_fwd_pidfile --retry 5
        rm -f $pkt_fwd_pidfile
    fi
    if monit summary lora-pkt-fwd-2 > /dev/null 2>&1; then
        monit unmonitor lora-pkt-fwd-2
        start-stop-daemon --stop --quiet --oknodo --pidfile $pkt_fwd_2_pidfile --retry 5
        rm -f $pkt_fwd_2_pidfile
    fi
    rm -rf /etc/monit.d/lora*
    monit reload

    echo "OK"
}


if [ "$ENABLED" != "yes" ]; then
    echo "$NAME: disabled in /etc/default"
    exit
fi

force_stop() {
    do_stop
    rm -fr $LOCK
}

function try_lock() {
    if mkdir $LOCK; then
        trap "rm -fr $LOCK" EXIT
    else
        echo "Lora Network Server lock not acquired, resource in use"
        exit 1
    fi
}

case "$1" in
    "start")
    try_lock
    do_start
        ;;
    "stop")
    force_stop
        ;;
    "restart")
    ## Stop the service and regardless of whether it was
        ## running or not, start it again.
        try_lock
        do_stop
        do_start
        ;;
    *)
        ## If no parameters are given, print which are avaiable.
        echo "Usage: $0 {start|stop|restart}"
        exit 1
    ;;
esac