install_rtop.sh

#!/bin/bash

# Check if the script is run with sudo
if [ "$EUID" -ne 0 ]; then
  echo "Please run this script with sudo."
  exit 1
fi

# Check if libQt5Widgets.so.5 is installed
if ! ldconfig -p | grep -q libQt5Widgets.so.5; then
  echo "libQt5Widgets.so.5 is not found on your system."
  read -p "Do you want to install it now? (y/n): " choice
  case "$choice" in
    y|Y )
      echo "Installing libqt5widgets5..."
      apt-get update
      apt-get install -y libqt5widgets5
      ;;
    * )
      echo "libQt5Widgets.so.5 is required"
      echo "Please install it with $ sudo apt-get install libqt5widgets5"
      ;;
  esac
else
  echo "libQt5Widgets.so.5 is already installed."
fi

# Get the current non-root user
CURRENT_USER=$(logname)

# Determine the directory of the setup script
SCRIPT_DIR=$(dirname "$(realpath "$0")")

# Check if rtop is in the same directory as the script
RTOP_PATH="$SCRIPT_DIR/rtop"
if [ -f "$RTOP_PATH" ]; then
  echo "rtop found at $RTOP_PATH"
else
  echo "Error: rtop not found in the current directory ($SCRIPT_DIR). Please ensure rtop is present."
  exit 1
fi

# Copy rtop to /usr/local/bin if not already there
if [ ! -f /usr/local/bin/rtop ]; then
  cp "$RTOP_PATH" /usr/local/bin/
  chmod +x /usr/local/bin/rtop
  echo "rtop has been copied to /usr/local/bin."
else
  echo "rtop is already present in /usr/local/bin."
fi

# Define the sudoers entry
SUDOERS_ENTRY="$CURRENT_USER ALL=(ALL) NOPASSWD: /usr/local/bin/rtop"

# Create a new file in /etc/sudoers.d/
SUDOERS_FILE="/etc/sudoers.d/rtop_nopasswd"

# Check if the file already exists
if [ -f "$SUDOERS_FILE" ]; then
  echo "The sudoers file for rtop already exists. Updating it..."
  echo "$SUDOERS_ENTRY" | tee "$SUDOERS_FILE" > /dev/null
else
  echo "$SUDOERS_ENTRY" | tee "$SUDOERS_FILE" > /dev/null
fi

# Set correct permissions for the file
chmod 0440 "$SUDOERS_FILE"

# Create the wrapper script
WRAPPER_SCRIPT="/usr/local/bin/rtop-wrapper"
echo "#!/bin/bash" | tee "$WRAPPER_SCRIPT" > /dev/null
echo "sudo /usr/local/bin/rtop \"\$@\"" | tee -a "$WRAPPER_SCRIPT" > /dev/null

# Make the wrapper script executable
chmod +x "$WRAPPER_SCRIPT"

# Add the alias to the user's .bashrc
BASHRC_FILE="/home/$CURRENT_USER/.bashrc"
ALIAS_CMD="alias rtop='rtop-wrapper'"

# Check if the alias already exists in .bashrc
if grep -Fxq "$ALIAS_CMD" "$BASHRC_FILE"; then
  echo "The alias for rtop already exists in $BASHRC_FILE."
else
  echo "$ALIAS_CMD" | tee -a "$BASHRC_FILE" > /dev/null
  echo "The alias for rtop has been added to $BASHRC_FILE."
fi

# Define source and destination paths
ICON_SOURCE="$SCRIPT_DIR/rtop_icon.png"
ICON_DEST_DIR="/usr/local/share/icons"
ICON_DEST="$ICON_DEST_DIR/rtop_icon.png"

# Check if the source icon file exists
if [ -f "$ICON_SOURCE" ]; then
  # Check if the destination directory exists
  if [ ! -d "$ICON_DEST_DIR" ]; then
    # Create the destination directory
    mkdir -p "$ICON_DEST_DIR"
    echo "Created directory $ICON_DEST_DIR."
  fi
  # Copy the icon to the destination directory
  cp "$ICON_SOURCE" "$ICON_DEST"
  echo "Icon has been copied to $ICON_DEST."
else
  echo "Warning: Icon not found at $ICON_SOURCE. The .desktop file will be created without an icon."
fi

# Create .desktop file for rtop
DESKTOP_FILE="/usr/share/applications/rtop.desktop"
tee "$DESKTOP_FILE" > /dev/null << EOF
[Desktop Entry]
Name=Rtop
Comment=Remote system monitoring tool
Exec=sudo /usr/local/bin/rtop
Icon=$ICON_DEST
Terminal=true
Type=Application
Categories=Utility;
EOF

# Ensure the .desktop file is executable
chmod +x "$DESKTOP_FILE"

echo "Setup complete. A .desktop file has been created for rtop."
echo "Please log out and log back in, or start a new shell session for changes to take effect."
echo "You can now find rtop in your applications menu and pin it to the dock if desired."