#!/bin/bash

# Usage: sudo ./nexify_rmm_install.sh <ENROLLMENT_KEY>

set -e

ENROLLMENT_KEY="$1"

if [ -z "$ENROLLMENT_KEY" ]; then
  echo "❌ Enrollment key not provided."
  exit 1
fi

INSTALL_DIR="/opt/nexify-rmm"
TEMP_DIR="$INSTALL_DIR/temp"
NODE_MIN_VERSION="20"

echo "📁 Creating directories..."
sudo mkdir -p "$TEMP_DIR"
sudo mkdir -p "$INSTALL_DIR/temp"
cd "$TEMP_DIR"

# Detect OS and install dependencies
install_dependencies() {
  echo "📦 Installing required packages..."

  if [ -f /etc/debian_version ]; then
    sudo apt-get update
    sudo apt-get install -y curl jq unzip

    if ! command -v node &> /dev/null || [ "$(node -v | cut -d. -f1 | tr -d v)" -lt "$NODE_MIN_VERSION" ]; then
      curl -fsSL https://deb.nodesource.com/setup_20.x | sudo -E bash -
      sudo apt-get install -y nodejs
    fi

  elif [ -f /etc/redhat-release ]; then
    sudo dnf install -y curl jq unzip

    if ! command -v node &> /dev/null || [ "$(node -v | cut -d. -f1 | tr -d v)" -lt "$NODE_MIN_VERSION" ]; then
      curl -fsSL https://rpm.nodesource.com/setup_20.x | sudo bash -
      sudo dnf install -y nodejs
    fi

  else
    echo "❌ Unsupported OS. Please install dependencies manually."
    exit 1
  fi
}

install_dependencies

# Ensure Node is now installed
if ! command -v node &> /dev/null || ! command -v npm &> /dev/null; then
  echo "❌ Node.js or npm installation failed. Exiting."
  exit 1
fi

# Download latest agent
echo "⬇️ Downloading Nexify RMM agent..."
AGENT_VERSION=$(curl -s https://api.cloud.brambleit.com/v1/app/rmm/version/agent/linux | jq -r .latest)
AGENT_ZIP="Base_Client_v$AGENT_VERSION.zip"
curl -o "$TEMP_DIR/$AGENT_ZIP" "https://download.cloud.brambleit.com/rmm/agent/linux/$AGENT_ZIP"

echo "🔄 Replacing agent files..."
# Remove all contents of the install dir except config.json (if it exists)
#sudo find "$INSTALL_DIR" -mindepth 1 ! -name 'config.json' -exec rm -rf {} +

echo "📦 Extracting new version of Nexify RMM..."
# Extract new version directly into the install dir
sudo unzip -o "$TEMP_DIR/$AGENT_ZIP" -d "$INSTALL_DIR"
echo "🔄 Moving new version of Nexify RMM..."
sudo cp -r "$INSTALL_DIR/build/"* "$INSTALL_DIR/"
echo "🔄 Cleaning up build folder..."
sudo rm -rf "$INSTALL_DIR/build"

echo "🔧 Installing packages..."
cd "$INSTALL_DIR"
npm install

# Make all .sh files executable
echo "🔧 Making .sh files executable..."
find "$INSTALL_DIR" -type f -name "*.sh" -exec sudo chmod +x {} \;

echo "🔐 Setting folder permissions..."
sudo chown -R root:root "$INSTALL_DIR"
sudo chmod -R 755 "$INSTALL_DIR"

# Ensure config.json is writable if it exists
CONFIG_FILE="$INSTALL_DIR/config.json"

if [ -f "$CONFIG_FILE" ]; then
  echo "🔐 Setting permissions on config.json..."
  sudo chmod 644 "$CONFIG_FILE"
else
  echo "📝 Creating config.json..."
  sudo tee "$CONFIG_FILE" > /dev/null <<EOF
{
  "RMM_Enrolled": false,
  "RMM_ServerURL": "https://api.cloud.brambleit.com",
  "RMM_EnrollmentKey": "$ENROLLMENT_KEY"
}
EOF
  # Set proper permissions
  sudo chmod 644 "$CONFIG_FILE"
fi


# Preserve the config file
echo "⚙️ Config file already exists. Skipping config.json overwrite..."

# Create systemd service file if not exists
SERVICE_FILE="/etc/systemd/system/nexify-rmm-agent.service"
if [ ! -f "$SERVICE_FILE" ]; then
  echo "🛠️ Creating systemd service..."

  sudo tee "$SERVICE_FILE" > /dev/null <<EOF
[Unit]
Description=Nexify RMM Agent
After=network.target

[Service]
ExecStart=$(which node) $INSTALL_DIR/index.js
WorkingDirectory=$INSTALL_DIR
Restart=always
Environment=NODE_ENV=production
StandardOutput=syslog
StandardError=syslog
SyslogIdentifier=nexify-rmm-agent
User=root
Group=root

[Install]
WantedBy=multi-user.target
EOF
fi

# Enable and start service
echo "🔁 Enabling and starting the service..."
sudo systemctl daemon-reexec
sudo systemctl daemon-reload
sudo systemctl enable nexify-rmm-agent
sudo systemctl start nexify-rmm-agent

echo "✅ Nexify RMM agent installed and running."