#!/usr/bin/env bash
#### webcamd - A webcam Service for multiple Cams and Stream Services.
####
#### Written by Stephan Wendel aka KwadFan <me@stephanwe.de>
#### Copyright 2021
#### https://github.com/mainsail-crew/crowsnest
####
#### This File is distributed under GPLv3
####

# shellcheck enable=requires-variable-braces

# Global Vars
BASE_USER=$(whoami)
TITLE="crowsnest - A Webcam Daemon for Raspberry Pi OS"


### Non root
if [ ${UID} == '0' ]; then
    echo -e "DO NOT RUN THIS SCRIPT AS ROOT!\nExiting..."
    exit 1
fi

### noninteractive Check
if [ -z "${DEBIAN_FRONTEND}" ]; then
    export DEBIAN_FRONTEND=noninteractive
fi

### Functions

### Messages
# Welcome Message
function welcome_msg {
    echo -e "${TITLE}\n"
    echo -e "\tYou will be prompted for your 'sudo' password, if needed."
    echo -e "\tSome Parts of the Updater requires 'root' privileges."
    # Dirty hack to gain root permissions
    sudo echo -e "\n"
}

# Goodbye Message
function goodbye_msg {
    echo -e "\nInstallation complete.\n"
    echo -e "\tIn case something was updated:\n\tPlease reboot your machine!"
    echo -e "I hope you enjoy crowsnest, GoodBye ..."
}

# No longer needed Message
function remove_message {
    echo -e "Found ${1}, remove because it is not longer needed."
}

### General
## These two functions are reused from custompios common.sh
## Credits to guysoft!
## https://github.com/guysoft/CustomPiOS

function install_cleanup_trap() {
    # kills all child processes of the current process on SIGINT or SIGTERM
    trap 'cleanup' SIGINT SIGTERM
}

function cleanup() {
    # make sure that all child processed die when we die
    echo -e "Killed by user ...\r\nGoodBye ...\r"
    # shellcheck disable=2046
    [ -n "$(jobs -pr)" ] && kill $(jobs -pr) && sleep 5 && kill -9 $(jobs -pr)
}

function err_exit {
    if [ "${1}" != "0" ]; then
        echo -e "ERROR: Error ${1} occured on line ${2}"
        echo -e "ERROR: Stopping $(basename "$0")."
        echo -e "Goodbye..."
    fi
    # shellcheck disable=2046
    [ -n "$(jobs -pr)" ] && kill $(jobs -pr) && sleep 5 && kill -9 $(jobs -pr)
    exit 1
}

### Init ERR Trap
trap 'err_exit $? $LINENO' ERR

# Helper funcs
function stop_webcamd {
    if [ "$(sudo systemctl is-active webcamd.service)" = "active" ]; then
        sudo systemctl stop webcamd &> /dev/null
    fi
}

function start_webcamd {
    if [ "$(sudo systemctl is-active webcamd.service)" = "active" ]; then
        sudo systemctl start webcamd &> /dev/null
    fi
}

function daemon_reload {
    echo -en "Reload systemd to enable new deamon ...\r"
    sudo systemctl daemon-reload &> /dev/null
    echo -e "Reload systemd to enable new daemon ... [OK]"
}

function compare_files {
    local installed template
    installed="$(sha256sum "${1}" | awk '{print $1}')"
    template="$(sha256sum "${2}" | awk '{print $1}')"
    if [ -f "${1}" ] && [ "${installed}" != "${template}" ]; then
        echo "1"
    else
        echo "0"
    fi
}

### Import config from custompios.
function import_config {
    if [ -f "${HOME}/crowsnest/custompios/crowsnest/config" ]; then
        # shellcheck disable=SC1091
        source custompios/crowsnest/config
    else
        echo -e "${TITLE}\n"
        echo -e "OOPS!\nConfiguration File missing! Exiting..."
        echo -e "Try to git clone a second time please ...\n"
        exit 1
    fi
}

# Copy Files funcs
function copy_service {
    local servicefile origin
    origin="/etc/systemd/system/webcamd.service"
    if [ ! "${BASE_USER}" == "pi" ]; then
        cp -rf "${origin}" /tmp/webcamd.service
        sudo sed -i 's|pi|'"${BASE_USER}"'|g' /tmp/webcamd.service
        servicefile="/tmp/webcamd.service"
    else
        servicefile="${HOME}/crowsnest/file_templates/webcamd.service"
    fi
    if [ "$(compare_files "${origin}" "${servicefile}")" -eq 1 ]; then
        echo -en "Copying webcamd.service file ...\r"
        sudo cp -rf "${servicefile}" "${origin}" > /dev/null
        echo -e "Copying webcamd.service file ... [OK]\r"
        daemon_reload
    else
        echo -e "No update of '${origin}' required."
    fi
    # Clean Temp File
    if [ -f /tmp/webcamd.service ]; then
        sudo rm -f /tmp/webcamd.service
    fi
}

function copy_logrotate {
    local logrotatefile origin
    origin="/etc/logrotate.d/webcamd"
    if [ ! "${BASE_USER}" == "pi" ]; then
        cp -rf "${origin}" /tmp/logrotate_webcamd
        sudo sed -i 's|pi|'"${BASE_USER}"'|g' /tmp/webcamd.service
        logrotatefile="/tmp/logrotate_webcamd"
    else
        logrotatefile="${HOME}/crowsnest/file_templates/logrotate_webcamd"
    fi
    if [ "$(compare_files "${origin}" "${logrotatefile}")" -eq 1 ]; then
        echo -en "Copying logrotate file ...\r"
        sudo cp -rf "${logrotatefile}" "${origin}" > /dev/null
        echo -e "Copying logrotate file ... [OK]\r"
    else
        echo -e "No update of '${origin}' required."
    fi
}

function copy_raspicam_fix {
    local moduleconf origin
    origin="/etc/modprobe.d/bcm2835-v4l2.conf"
    moduleconf="${HOME}/crowsnest/file_templates/bcm2835-v4l2.conf"
    if [ ! -f "${origin}" ]; then
        echo -en "Copying bcm2835-v4l2.conf file ...\r"
        sudo cp -rf "${moduleconf}" "${origin}" > /dev/null
        echo -e "Copying bcm2835-v4l2.conf file ... [OK]\r"
    else
        if [ "$(compare_files "${origin}" "${moduleconf}")" -eq 1 ]; then
            echo -en "Copying bcm2835-v4l2.conf file ...\r"
            sudo cp -rf "${moduleconf}" "${origin}" > /dev/null
            echo -e "Copying bcm2835-v4l2.conf file ... [OK]\r"
        else
            echo -e "No update of '${origin}' required."
        fi
    fi
}

# Uninstall funcs
function uninstall_v4l2rtsp {
    local bin_path v4l2rtsp_dir
    bin_path="/usr/local/bin/v4l2rtspserver"
    v4l2rtsp_dir="${HOME}/v4l2rtspserver"
    if [ -d "${v4l2rtsp_dir}" ]; then
        remove_message "v4l2rtspserver"
        echo -en "Uninstalling 'v4l2rtspserver' ...\r"
        if [ -x "${bin_path}" ]; then
            sudo rm -f "${bin_path}"
        fi
        sudo rm -rf "${v4l2rtsp_dir}"
        echo -e "Uninstalling 'v4l2rtspserver' ... [OK]\r"
    fi
}

# This removes ustreamer if not inside of crowsnest! ( $HOME/ustreamer )
function uninstall_ustreamer {
    local bin_path ustreamer_dir
    bin_path="/usr/local/bin/ustreamer"
    ustreamer_dir="${HOME}/ustreamer"
    if [ -d "${ustreamer_dir}" ]; then
        remove_message "ustreamer"
        echo -en "Uninstalling '${HOME}/ustreamer' ...\r"
        if [ -x "${bin_path}" ]; then
            sudo rm -f "${bin_path}"
        fi
        sudo rm -rf "${ustreamer_dir}"
        echo -e "Uninstalling '${HOME}/ustreamer' ... [OK]\r"
    fi
}

# Remove no longer needed Go installation
function uninstall_go {
    if [ -n "$(whereis -b go | awk '{print $2}')" ]; then
        echo -e "\nFound $(go version)\n"
    else
        echo -e "No Version of Go Lang found ... [SKIPPED]"
    fi
    if  [ -d "/usr/local/go" ] && [ -f "${HOME}/.gorc" ]; then
        sudo rm -rf "$(whereis -b go | awk '{print $2}')"
        rm -f "${HOME}/.gorc"
        sudo rm -rf "${HOME}/golang"
        sed -i '/# Add Go/d;/.gorc/d' "${HOME}/.profile"
        echo -e "\nUninstall complete!"
    fi
}

# remove obsolete RTSPtoWebRTC
function uninstall_rtsp2webrtc {
    if [ -d "${HOME}/crowsnest/bin/RTSPtoWebRTC" ];then
        echo -e "Removing RTSPtoWebRTC ..."
        sudo rm -rf "${HOME}/crowsnest/bin/RTSPtoWebRTC"
        echo -e "Removing RTSPtoWebRTC ... [OK]"
    fi
}


# Install funcs
# Make sure ustreamer submodule is initialized
function sub_init {
    if [ ! -f "${HOME}/crowsnest/bin/ustreamer/Makefile" ]; then
        echo -e "Submodules are not initialized ..."
        git submodule update --init > /dev/null
        echo -e "Submodules are not initialized ... [OK]"
    fi
}

# Build Apps
function build_apps {
    echo -e "\nDependency Check: Check for compiled Apps ..."
    if [ ! -x "bin/ustreamer/src/ustreamer.bin" ] \
    || [ ! -x "bin/rtsp-simple-server/rtsp-simple-server" ]; then
        echo -e "Build dependend Stream Apps ..."
        echo -e "Installing 'ustreamer' Dependencies ..."
        # shellcheck disable=2086
        sudo apt install --yes --no-install-recommends ${CROWSNEST_USTREAMER_DEPS} > /dev/null
        echo -e "Installing 'ustreamer' Dependencies ... [OK]"
        pushd bin > /dev/null || exit 1
        make all
        popd > /dev/null || exit 1
    else
        echo -e "All Apps are present ... [OK]"
        update_ustreamer
        update_rtsp
    fi
}

function update_ustreamer {
    local sm_version us_version
    sm_version="$(cd bin/ustreamer && git describe --always --tags | sed 's/^v//')"
    us_version="$(bin/ustreamer/ustreamer -v)"
    echo -en "Checking ustreamer Version ${sm_version} ... \r"
    if [ "${sm_version}" == "${us_version}" ]; then
        echo -e "Checking ustreamer Version ${us_version} ... [SKIPPED]"
        echo -e "\t==> Version match, no update needed."
    else
        echo -e "Checking ustreamer Version ${sm_version} ... [UPDATE]"
        echo -e "\t==> Version (${us_version}) mismatch, update needed."
        pushd bin/ &> /dev/null || exit 1
        make ustreamer
        popd &> /dev/null || exit 1
        echo -e "Ustreamer Update ... [OK]"
    fi
}

function update_rtsp {
    local dl_version rtsp_version
    dl_version="$(cat bin/rtsp-simple-server/version)"
    rtsp_version="$(bin/rtsp-simple-server/rtsp-simple-server --version)"
    echo -en "Checking rtsp-simple-server Version ... \r"
    if [ "${dl_version}" == "${rtsp_version}" ]; then
        echo -e "Checking rtsp-simple-server Version ${dl_version} ... [SKIPPED]"
        echo -e "\t==> Version match, no update needed."
    else
        echo -e "Checking rtsp-simple-server Version ... [UPDATE]"
        echo -e "\t==> Version (${rtsp_version}) mismatch, update needed."
        pushd bin/ &> /dev/null || exit 1
        make rtsp
        popd &> /dev/null || exit 1
        echo -e "rtsp-simple-server Update ... [OK]"
    fi
}


#### MAIN
install_cleanup_trap
import_config
welcome_msg
stop_webcamd
uninstall_ustreamer
uninstall_v4l2rtsp
uninstall_rtsp2webrtc
uninstall_go
copy_service
copy_logrotate
copy_raspicam_fix
sub_init
build_apps
start_webcamd
goodbye_msg

exit 0