#!/bin/bash
#This bash script installs a modification to the Proxmox Virtual Environment (PVE) web user interface (UI) to display temperature information.

# Filepaths
pvemanagerlib="/usr/share/pve-manager/js/pvemanagerlib.js"
nodespm="/usr/share/perl5/PVE/API2/Nodes.pm"
backuplocation="/root/backup"
timestamp=$(date '+%Y-%m-%d_%H-%M-%S')


################### code below #############
echo ""

# Function to display usage information
function usage {
  echo "Usage: $0 [install | uninstall]"
  echo ""
  exit 1
}

#!/bin/bash

# Define a function to install packages
install_packages () {
  # Check if the 'sensors' command is available on the system
  if ! command -v sensors &> /dev/null; then
    # If the 'sensors' command is not available, prompt the user to install lm-sensors
    read -p "lm-sensors is not installed. Would you like to install it? (y/n) " choice
    case "$choice" in
      y|Y )
        # If the user chooses to install lm-sensors, update the package list and install the package
        apt-get update
        apt-get install lm-sensors
        ;;
      n|N )
        # If the user chooses not to install lm-sensors, exit the script with a zero status code
        echo "Decided to not install lm-sensors. The mod cannot run without"
        exit 0
        ;;
      * )
        # If the user enters an invalid input, print an error message and exit the script with a non-zero status code
        echo "Invalid input. Exiting..."
        exit 1
        ;;
    esac
  fi
}

# Call the 'install_packages' function to check if lm-sensors is installed and install it if necessary
install_packages


# Function to install the modification
function install_mod {
  # Create backup of original files
  mkdir -p $backuplocation

  # Add new line to Nodes.pm file
  if [[ -z $(cat $nodespm | grep -e "$res->{thermalstate}") ]]; then
    # Create backup of original file
    cp "$nodespm" "$backuplocation/Nodes.pm.$timestamp"
    echo "Backup of $nodespm saved to $backuplocation/Nodes.pm.$timestamp"

    sed -i '/my $dinfo = df('\''\/'\'', 1);/i\'$'\t''$res->{thermalstate} = `sensors -j`;\n'$'\t''$res->{thermalstate2} = `sensors -j`;\n' "$nodespm"
    echo "Added thermalstate to $nodespm"
  else
    echo "Thermalstate already added to $nodespm"
  fi

  # Add new item to the items array in PVE.node.StatusView
  if [[ -z $(cat "$pvemanagerlib" | grep -e "itemId: 'thermal'") ]]; then
    # Create backup of original file
    cp "$pvemanagerlib" "$backuplocation/pvemanagerlib.js.$timestamp"
    echo "Backup of $pvemanagerlib saved to $backuplocation/pvemanagerlib.js.$timestamp"
    
    # Expand space in StatusView
    sed -i "/Ext.define('PVE\.node\.StatusView'/,/\},/ {
      s/\(bodyPadding:\) '[^']*'/\1 '20 15 20 15'/
      s/height: [0-9]\+/minHeight: 360,\n\tflex: 1/}" "$pvemanagerlib"

    echo "Expanded space in $pvemanagerlib"

    sed -i "/^Ext.define('PVE.node.StatusView',/ {
      :a;
      /items:/!{N;ba;}
      :b;
      /swap.*},/!{N;bb;}
      a\
      \\
        {\n\
            itemId: 'thermal',\n\
            colspan: 1,\n\
            printBar: false,\n\
            title: gettext('CPU Thermal State'),\n\
            iconCls: 'fa fa-fw fa-thermometer-half',\n\
            textField: 'thermalstate',\n\
            renderer: function(value){\n\
              let objValue = JSON.parse(value);\n\
              const address = \"coretemp-isa-0000\";\n\
              if(objValue.hasOwnProperty(address)) \{\n\
                  const items = objValue[address],\n\
                    itemPrefix = \"Core \",\n\
                    coreKeys = Object.keys(items).filter(item => \{\n\
                      return String(item).startsWith(itemPrefix);\n\
                    \}),\n\
                    tempInputOffset = 2; // see tempN_input for \"Core 0\"\n\
\n\
                  let temps = [];\n\
                  coreKeys.forEach((coreKey, index) => \{\n\
                      try \{\n\
                          let temp = items[itemPrefix + index][\`temp\$\{tempInputOffset + index\}_input\`];\n\
                          temps.push(\`Core \$\{index\}: \$\{temp\}°C\`);\n\
                      \} catch(e) \{ /*_*/ \}\n\
                  });\n\
\n\
                  const cols = 4,\n\
                    result = temps.map((strTemp, index, arr) => { return strTemp + (index + 1 < arr.length ? ((index + 1) % cols === 0 ? '<br>' : ' | ') : '')});\n\
                  return result.join('');\n\
              \}\n\
            }\n\
        },\n\
        {\n\
            itemId: 'thermal2',\n\
            colspan: 1,\n\
            printBar: false,\n\
            title: gettext('NVME Thermal State'),\n\
            iconCls: 'fa fa-fw fa-thermometer-half',\n\
            textField: 'thermalstate2',\n\
            renderer: function(value){\n\
              let objValue = JSON.parse(value);\n\
              const addressPrefix = \"nvme-pci-\",\n\
                nvmeKeys = Object.keys(objValue).filter(item => \{ return String(item).startsWith(addressPrefix); \}),\n\
                sensorName = \"Composite\",\n\
                tempInputNo = 1;\n\
\n\
                let temps = [];\n\
                nvmeKeys.forEach((nvmeKey, index) => \{\n\
                  try \{\n\
                    let temp = objValue[nvmeKey][sensorName][\`temp\$\{tempInputNo\}_input\`]\n\
                    temps.push(\`Drive \$\{index\}: \$\{temp\}&deg;C\`);\n\
                  \} catch(e) \{ /*_*/ \}\n\
                \});\n\
\n\
                const cols = 4,\n\
                  result = temps.map((strTemp, index, arr) => \{ return strTemp + (index + 1 < arr.length ? ((index + 1) % cols === 0 ? '<br>' : ' | ') : '')\});\n\
                return result.join('');\n\
            \}\n\
        },
    }" $pvemanagerlib
    echo "Added new item to the items array in $pvemanagerlib"
  else
    echo "New item already added to items array $pvemanagerlib"
  fi

  # Restart pveproxy
  systemctl restart pveproxy
}

# Function to uninstall the modification
function uninstall_mod {
  # Find the latest Nodes.pm file using the find command
  latest_Nodes_pm=$(find "$backuplocation" -name "Nodes.pm.*" -type f -printf '%T+ %p\n' 2>/dev/null | sort -r | head -n 1 | awk '{print $2}')

  if [ -z "$latest_Nodes_pm" ]; then
    echo "No Nodes.pm files found"
    exit 1
  fi

  # Remove the latest Nodes.pm file
  echo cp "$latest_Nodes_pm" "$nodespm"
  echo "Copied latest backup to $nodespm"

  # Find the latest pvemanagerlib file using the find command
  latest_pvemanagerlib_js=$(find "$backuplocation" -name "pvemanagerlib.js.*" -type f -printf '%T+ %p\n' 2>/dev/null | sort -r | head -n 1 | awk '{print $2}')

  if [ -z "$latest_pvemanagerlib_js" ]; then
          echo "No latest_pvemanagerlib_js files found"
          exit 1
  fi

  # Remove the latest Nodes.pm file
  echo cp "$latest_pvemanagerlib_js" "$pvemanagerlib"
  echo "Copied latest backup to $pvemanagerlib"

  # Restart pveproxy
  systemctl restart pveproxy
}

# If no arguments were provided or all arguments have been processed, print the usage message
if [[ $# -eq 0 ]]; then
	usage
fi

# Process the arguments using a while loop and a case statement
while [[ $# -gt 0 ]]; do
  case "$1" in
    install)
      echo "Installing the mod..."
      install_packages
      install_mod
      ;;
    uninstall)
      echo "Uninstalling the mod..."
      uninstall_mod
      ;;
    *)
      usage
      ;;
  esac
  shift
done

echo ""
exit 0