#! /usr/bin/env bash
# Standard Installation Script for BirdNET-stream for Debian Based Linux distros
# set -x
set -e

DEBUG=${DEBUG:-0}

REQUIREMENTS="git ffmpeg python3 python3-pip python3-dev"
REPOSITORY=${REPOSITORY:-https://github.com/UncleSamulus/BirdNET-stream.git}
WORKDIR="$(pwd)/BirdNET-stream"
PYTHON_VENV="./.venv/birdnet-stream"

debug() {
    [[ $DEBUG -eq 1 ]] && echo "$@"
}

install_requirements() {
    requirements=$1
    # Install requirements
    missing_requirements=""
    for requirement in $requirements; do
        if ! dpkg -s "$requirement" >/dev/null 2>&1; then
            missing_requirements="$missing_requirements $requirement"
        fi
    done
    if [[ -n "$missing_requirements" ]]; then
        debug "Installing missing requirements: $missing_requirements"
        sudo apt-get install -y "$missing_requirements"
    fi
}

# Install BirdNET-stream
install_birdnetstream() {
    # Check if repo is not already installed
    if [[ -d "$DIR" ]]; then
        debug "BirdNET-stream is already installed, use update script (not implemented yet)"
        exit 1 
    else
        debug "Installing BirdNET-stream"
        debug "Creating BirdNET-stream directory"
        mkdir -p "$WORKDIR"
        # Clone BirdNET-stream
        cd "$WORKDIR"
        debug "Cloning BirdNET-stream from $REPOSITORY"
        git clone --recurse-submodules "$REPOSITORY" .
        debug "Creating python3 virtual environment $PYTHON_VENV"
        python3 -m venv $PYTHON_VENV
        debug "Activating $PYTHON_VENV"
        source "$PYTHON_VENV/bin/activate"
        debug "Installing python packages"
        pip3 install -U pip
        pip3 install -r requirements.txt
        debug "Creating ./var directory"
        mkdir -p ./var/{charts,chunks/{in,out}}
    fi
}

# Install systemd services
install_birdnetstream_services() {
    GROUP=birdnet
    debug "Setting up BirdNET stream systemd services"
    services="birdnet_recording.service birdnet_analyzis.service birdnet_miner.timer birdnet_miner.service birdnet_plotter.service birdnet_plotter.timer"
    read -r -a services_array <<< "$services"
    for service in "${services_array[@]}"; do
        sudo cp "daemon/systemd/templates/$service" "/etc/systemd/system/"
        variables="DIR USER GROUP"
        for variable in $variables; do
            sudo sed -i "s|<$variable>|${!variable}|g" "/etc/systemd/system/$service"
        done
    done
    sudo systemctl daemon-reload
    sudo systemctl enable --now birdnet_recording.service birdnet_analyzis.service birdnet_miner.timer birdnet_plotter.timer
}

install_php8() {
    # Remove previously installed php version
    sudo apt-get remove --purge php*
    # Install required packages for php
    sudo apt-get install -y lsb-release ca-certificates apt-transport-https software-properties-common gnupg2
    # Get php package from sury repo
    echo "deb https://packages.sury.org/php/ $(lsb_release -sc) main" | sudo tee /etc/apt/sources.list.d/sury-php.list
    sudo wget -qO - https://packages.sury.org/php/apt.gpg | sudo gpg --no-default-keyring --keyring gnupg-ring:/etc/apt/trusted.gpg.d/debian-php-8.gpg --import
    sudo chmod 644 /etc/apt/trusted.gpg.d/debian-php-8.gpg
    update
    sudo apt-get install php8.1
    # Install and enable php-fpm
    sudo apt-get install php8.1-fpm
    sudo systemctl enable php8.1-fpm
    # Install php packages
    sudo apt-get install php8.1-{sqlite3,curl,intl}
}

install_composer() {
    cd /tmp
    php -r "copy('https://getcomposer.org/installer', 'composer-setup.php');"\nphp -r "if (hash_file('sha384', 'composer-setup.php') === '55ce33d7678c5a611085589f1f3ddf8b3c52d662cd01d4ba75c0ee0459970c2200a51f492d557530c71c15d8dba01eae') { echo 'Installer verified'; } else { echo 'Installer corrupt'; unlink('composer-setup.php'); } echo PHP_EOL;"\nphp composer-setup.php\nphp -r "unlink('composer-setup.php');"
    sudo mv ./composer.phar /usr/local/bin/composer
    cd -
}

install_nodejs() {
    # Install nodejs
    curl -o- https://raw.githubusercontent.com/nvm-sh/nvm/v0.39.1/install.sh | bash
    export NVM_DIR="$([[ -z "${XDG_CONFIG_HOME-}" ]] && printf %s "${HOME}/.nvm" || printf %s "${XDG_CONFIG_HOME}/nvm")"
    [[ -s "$NVM_DIR/nvm.sh" ]] && \. "$NVM_DIR/nvm.sh" # This loads nvm
    nvm install 16
    nvm use 16
    install_requirements "npm"
    # Install yarn
    sudo npm install -g yarn
}

install_web_interface() {
    debug "Setting up web interface"
    install_requirements "nginx"
    # Install php 8.1
    install_php8
    # Install composer
    install_composer
    # Install nodejs 16
    install_nodejs
    # Install Symfony web app
    cd "$WORKDIR"
    cd www
    debug "Creating nginx configuration"
    sudo cp nginx.conf /etc/nginx/sites-available/birdnet-stream.conf
    sudo mkdir /var/log/nginx/birdnet/
    sudo ln -s /etc/nginx/sites-available/birdnet-stream.conf /etc/nginx/sites-available/birdnet-stream.conf
    debug "Info: Please edit /etc/nginx/sites-available/birdnet-stream.conf to set the correct server name and paths"
    sudo systemctl enable --now nginx
    sudo systemctl restart nginx
    debug "Retrieving composer dependencies"
    composer install
    debug "PHP dependencies installed"
    debug "Installing nodejs dependencies"
    yarn install
    debug "npm dependencies installed"
    debug "Building assets"
    yarn build
    debug "Webpack assets built"
    debug "Web interface is available"
    debug "Please restart nginx after double check of /etc/nginx/sites-available/birdnet-stream.conf"
}

change_value() {
    local variable_name
    variable_name="$1"
    local variable_new_value
    variable_new_value="$2"
    local variable_filepath="$3"
    sed -i "s|$variable_name=.*|$variable_name=\"$variable_new_value\"|g" "$variable_filepath"
}

install_config() {
    debug "Updating config"
    cd "$WORKDIR"
    cp ./config/birdnet.conf.example ./config/birdnet.conf
    config_filepath="$WORKDIR/config/birdnet.conf"
    change_value "DIR" "$WORKDIR" "$config_filepath"
    change_value "PYTHON_VENV" "$PYTHON_VENV" "$config_filepath"
    change_value "AUDIO_RECORDING" "true" "$config_filepath"
    source "$config_filepath"
    cd www
    debug "Setup webapp .env"
    cp .env.local.example .env.local
    change_value "RECORDS_DIR" "$CHUNKS_FOLDER" ".env.local"
}

main() {
    install_requirements "$REQUIREMENTS"
    install_birdnetstream
    install_birdnetstream_services
    install_web_interface
    install_config
    update_permissions "$WORKDIR"
    debug "Installation done"
}

main