222 lines
7.6 KiB
Bash
222 lines
7.6 KiB
Bash
#!/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/
|
|
s/\(tableAttrs:.*$\)/trAttrs: \{ valign: 'top' \},\n\t\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\
|
|
// sensors configuration\n\
|
|
const address = \"coretemp-isa-0000\",\n\
|
|
itemPrefix = \"Core \",\n\
|
|
tempInputOffset = 2; // see tempN_input for \"Core 0\"\n\
|
|
// display configuration\n\
|
|
const coresPerRow = 4;\n\
|
|
\n\
|
|
const objValue = JSON.parse(value);\n\
|
|
if(objValue.hasOwnProperty(address)) \{\n\
|
|
const items = objValue[address],\n\
|
|
coreKeys = Object.keys(items).filter(item => \{\n\
|
|
return String(item).startsWith(itemPrefix);\n\
|
|
\});\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 result = temps.map((strTemp, index, arr) => { return strTemp + (index + 1 < arr.length ? ((index + 1) % coresPerRow === 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\
|
|
// sensors configuration\n\
|
|
const addressPrefix = \"nvme-pci-\",\n\
|
|
sensorName = \"Composite\",\n\
|
|
tempInputNo = 1;\n\
|
|
// display configuration\n\
|
|
const driversPerRow = 4;\n\
|
|
\n\
|
|
const objValue = JSON.parse(value),\n\
|
|
nvmeKeys = Object.keys(objValue).filter(item => \{ return String(item).startsWith(addressPrefix); \});\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\}°C\`);\n\
|
|
\} catch(e) \{ /*_*/ \}\n\
|
|
\});\n\
|
|
\n\
|
|
const result = temps.map((strTemp, index, arr) => \{ return strTemp + (index + 1 < arr.length ? ((index + 1) % driversPerRow === 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
|