#!/bin/bash

####------------------- Pre-Flight Checks
# Exit on error
set -e

# Check if the script is run as root
if [[ $(id -u) -ne 0 ]]; then
    echo "Please run this script as root."
    exit 1
fi

# Check if the working directory is 'client' and contains package.json
if [[ ! -f "$(pwd)/package.json" ]]; then
    echo "Error: Please make sure the working directory is 'client' and contains package.json."
    exit 1
fi

# Check to make sure the pi user exists
if ! id "pi" &>/dev/null; then
    echo "Error: User pi does not exist."
    exit 1
fi

# Install Node Repo
# Get the CPU architecture
cpu_arch=$(uname -m)

# Print the CPU architecture for verification
echo "Detected CPU Architecture: $cpu_arch"

# Check if the architecture is ARMv6
if [[ "$cpu_arch" == "armv6"* ]]; then
  echo "----- CPU Architecture is ARMv6 or compatible. -----"
  echo "----- CPU Architectre is not compatible with dependencies of this project, please use a newer CPU architecture -----"
  exit
fi

curl -fsSL https://deb.nodesource.com/setup_current.x | sudo -E bash -

# Update the system
apt update
apt upgrade -y

# Install the necessary packages
echo "Installing dependencies..."
apt install -y \
  nodejs \
  libasound-dev \
  portaudio19-dev \
  libportaudio2 \
  libpulse-dev \
  pulseaudio \
  apulse \
  git \
  ffmpeg \
  python3 \
  python3-pip 

echo "Setting up Pulse Audio"

# Ensure pulse audio is running as system so the service can see the audio device
systemctl --global disable pulseaudio.service pulseaudio.socket

# Update the PulseAudio config to disable autospawning
sed -i 's/autospawn = .*$/autospawn = no/' /etc/pulse/client.conf

# Add the system PulseAudio service
echo "[Unit]
Description=PulseAudio system server

[Service]
Type=notify
ExecStart=pulseaudio --daemonize=no --system --realtime --log-target=journal

[Install]
WantedBy=multi-user.target" >> /etc/systemd/system/PulseAudio.service

# Add the root user to the pulse-access group
usermod -aG pulse-access root
usermod -aG pulse-access pi

# Enable the PulseAudio service
systemctl enable PulseAudio.service

####------------------- Install and setup node
# Run npm install to install dependencies listed in package.json
echo "Installing npm dependencies..."
npm install

# Get rid of PEP 668
rm -rf /usr/lib/python3.11/EMTERNALL-MANAGED # Not sure if this was an attrocious fat finger or if this is needed, doesn't throw an error, so...
rm -rf /usr/lib/python3.11/EXTERNALLY-MANAGED

# Getting the Python DAB
echo "Installing PDAB and Dependencies"
git clone -b DRBv3 https://git.vpn.cusano.net/logan/Python-Discord-Audio-Bot.git ./pdab
pip3 install -r ./pdab/requirements.txt

# Create a systemd service file for the DRB Client
echo "Adding DRB Node service..."
service_content="[Unit]
Description=Discord-Radio-Bot_v3
After=syslog.target network.target nss-lookup.target network-online.target
Requires=network-online.target

[Service]
User=1000
Group=1000
WorkingDirectory=$(pwd)
ExecStart=/bin/bash -- serviceStart.sh
RestartSec=5
Restart=on-failure

[Install]
WantedBy=multi-user.target"

# Write the systemd service file
echo "$service_content" > /etc/systemd/system/discord-radio-bot.service

# Reload systemd daemon
systemctl daemon-reload
systemctl enable discord-radio-bot.service

echo "\n\n\t\tDRB Client install completed!\n\n"

####------------------- OP25 Installation
# Clone OP25 from the git repository
echo "Cloning OP25 from the git repository..."
git clone -b gr310 https://github.com/boatbod/op25.git

# Navigate to the OP25 directory
ogPwd=$(pwd)
cd op25

# Edit the startup script to use the active.cfg.json config file generated by the app
echo "Editing startup script..."
sed -i 's/p25_rtl_example.json/active.cfg.json/g' op25-multi_rx.sh

# Move the startup script to the apps dir
mv op25-multi_rx.sh op25/gr-op25_repeater/apps/

# Install the OP25 service
echo "Adding OP25 service..."
service_content="[Unit]
Description=op25-multi_rx
After=syslog.target network.target nss-lookup.target network-online.target
Requires=network-online.target

[Service]
User=1000
Group=1000
WorkingDirectory=$(pwd)/op25/gr-op25_repeater/apps
ExecStart=/bin/bash -- op25-multi_rx.sh
RestartSec=5
Restart=on-failure

[Install]
WantedBy=multi-user.target"

# Write the systemd service file
echo "$service_content" > /etc/systemd/system/op25-multi_rx.service

# Reload systemd daemon
systemctl daemon-reload

# Install OP25 using the provided installation script
echo "Installing OP25..."
./install.sh

echo "\n\n\t\tOP25 installation completed!\n\n"

# Setting permissions on the directories created
cd $ogPwd
chown -R 1000:1000 ./*
echo "Permissions set on the client directory!"

echo "\n\n\t\tNode installation Complete!"

# Prompt the user for reboot confirmation
read -p "This script has installed all required components for the DRB client. Are you okay with rebooting? If not, you will have to reboot later before running the applications to finish the installation. (Reboot?: y/n): " confirm

# Convert user input to lowercase for case-insensitive comparison
confirm="${confirm,,}"

echo "To configure the app, please go to http://localhost:3000 after the reboot to configure this app"
echo "Thank you for joining the network!"

# Prompt user to press any key before rebooting
read -rsp $'System will now reboot, press any key to continue or Ctrl+C to cancel...\n' -n1 key
echo "Rebooting..."
reboot