#!/bin/bash


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


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


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_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"

mtcdt3_id="MTCDT3"
lora_mtcdt3_id="MTCDT3-LORA"
lora_mtcdt3_id868="MTCDT3-LORA-868"
lora_mtcdt3_id915="MTCDT3-LORA-915"

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

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() {
    hw_id=$(mts-io-sysfs show product-id 2> /dev/null)

    # product-id of first lora card
    lora_id=$(mts-io-sysfs show lora/product-id 2> /dev/null)
    lora_eui=$(mts-io-sysfs show lora/eui 2> /dev/null)
    # remove all colons
    lora_eui_raw=${lora_eui//:}
    lora_hw=$(mts-io-sysfs show lora/hw-version 2> /dev/null)

    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
    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

}

setup_mtcdt3() {
    dual_cards_installed=true
    ln -sf /opt/lora/lora_pkt_fwd $pkt_fwd
    ln -sf /opt/lora/lora_pkt_fwd $pkt_fwd_2

    lora_eui=$(mts-io-sysfs show lora/eui)
    lora_2_eui=$(mts-io-sysfs show lora-2/eui)
    
    lora_2_hw=$lora_hw
    lora_2_id=$lora_id
}

hardware_found() {
    if [[ "$hw_id" =~ "$mtcdt3_id" ]]; then
        setup_mtcdt3
    elif [[ "$lora_id" =~ "$lora_mtac_g_id" ]]; then
        setup_mtcdt_2_1
    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
}

setup_mtcdt_2_1() {
    ln -sf /dev/spidev32766.2 /dev/spidev0.0

    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() {
    if [ -d $port1 ] && [[ $(cat $port1/hw-version) = $lora_hw ]]; then
        ln -sf /dev/spidev32766.2 /dev/spidev0.0
    elif [ -d $port2 ] && [[ $(cat $port2/hw-version) = $lora_hw ]]; then
        ln -sf /dev/spidev32765.2 /dev/spidev0.0
    fi
    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
            if [ "$hw" = "MTCAP-0.3" ]; then
                cp /opt/lora/temp_lut.json.MTCAP2.EU868 /var/run/lora/1/temp_lut.json
            fi
        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
            if [ "$hw" = "MTCAP-0.3" ]; then
                cp /opt/lora/temp_lut.json.MTCAP2.US915 /var/run/lora/1/temp_lut.json
            fi
        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
}

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

    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 [[ "$hw_id" =~ "$mtcdt3_id" ]]; then
            mts-io-sysfs store lora/reset 0
            mts-io-sysfs store lora-2/reset 0
            sleep 0.25
            mts-io-sysfs store lora/reset 1
            mts-io-sysfs store lora-2/reset 1
            sleep 0.25
        elif [[ "$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
    start-stop-daemon --start --background --make-pidfile \
        --pidfile $net_server_pidfile --startas /bin/bash -- -c "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"

    if [ "$no_lora_hw" != "true" ]; then
        # start packet forwarder
        if [[ "$hw_id" =~ "$mtcdt3_id" ]]; then
            sleep 0.5
        else
            sleep 4
        fi
        /usr/sbin/start-stop-daemon --chdir $run_dir/1 --start --background --make-pidfile \
            --pidfile $pkt_fwd_pidfile --startas /bin/bash \
            -- -c "exec $pkt_fwd $pkt_fwd_options 2>&1"

        if [ "$dual_cards_installed" == "true" ]; then
            /usr/sbin/start-stop-daemon --chdir $run_dir/2 --start --background --make-pidfile \
                --pidfile $pkt_fwd_2_pidfile --startas /bin/bash \
                -- -c "exec $pkt_fwd_2 $pkt_fwd_options -l $pkt_fwd_2_log"
        fi
    fi

    if [[ ! "$hw_id" =~ "$mtcdt3_id" ]]; then
        renice -n -20 -p $(pgrep lora-network-se)
        renice -n -20 -p $(pgrep $(basename $pkt_fwd))
    fi

    echo "OK"
}


do_stop() {
    echo -n "Stopping $NAME: "
    start-stop-daemon --stop --quiet --oknodo --pidfile $net_server_pidfile --retry TERM/60/KILL/5
    start-stop-daemon --stop --quiet --oknodo --pidfile $pkt_fwd_pidfile --retry 5
    start-stop-daemon --stop --quiet --oknodo --pidfile $pkt_fwd_2_pidfile --retry 5
    rm -f $net_server_pidfile $pkt_fwd_pidfile
    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