mirror of
https://github.com/AuxXxilium/arv-v-dsm.git
synced 2024-12-03 21:27:01 +07:00
141 lines
2.5 KiB
Bash
141 lines
2.5 KiB
Bash
#!/usr/bin/env bash
|
||
set -u
|
||
|
||
VERSION="9"
|
||
HEADER="VirtualDSM Agent"
|
||
|
||
# Functions
|
||
|
||
error () { echo -e "\E[1;31m❯ ERROR: $1\E[0m" ; }
|
||
info () { echo -e "\E[1;34m❯\E[1;36m $1\E[0m" ; }
|
||
|
||
finish() {
|
||
|
||
echo "❯ $HEADER: Shutting down.."
|
||
exit
|
||
|
||
}
|
||
|
||
function checkNMI {
|
||
|
||
local nmi
|
||
nmi=$(cat /proc/interrupts | grep NMI | sed 's/[^1-9]*//g')
|
||
|
||
if [ "$nmi" != "" ]; then
|
||
|
||
info "Received shutdown request through NMI.."
|
||
|
||
/usr/syno/sbin/synoshutdown -s > /dev/null
|
||
finish
|
||
|
||
fi
|
||
}
|
||
|
||
function downloadUpdate {
|
||
|
||
TMP="/tmp/agent.sh"
|
||
rm -f "${TMP}"
|
||
|
||
# Auto update the agent
|
||
|
||
URL="https://raw.githubusercontent.com/vdsm/virtual-dsm/master/agent/agent.sh"
|
||
|
||
remote_size=$(curl -sIk -m 4 "${URL}" | grep -i "content-length:" | tr -d " \t" | cut -d ':' -f 2)
|
||
remote_size=${remote_size//$'\r'}
|
||
|
||
[[ "$remote_size" == "" || "$remote_size" == "0" ]] && return
|
||
|
||
remote_size=$(($remote_size+0))
|
||
((remote_size<100)) && return
|
||
|
||
SCRIPT=$(readlink -f "${BASH_SOURCE[0]}")
|
||
local_size=$(stat -c%s "$SCRIPT")
|
||
local_size=$(($local_size+0))
|
||
|
||
[[ remote_size -eq local_size ]] && return
|
||
|
||
if ! curl -sfk -m 10 -o "${TMP}" "${URL}"; then
|
||
error "$HEADER: curl error ($?)" && return
|
||
fi
|
||
|
||
if [ ! -f "${TMP}" ]; then
|
||
error "$HEADER: update error, file not found.." && return
|
||
fi
|
||
|
||
line=$(head -1 "${TMP}")
|
||
|
||
if [[ "$line" != "#!/usr/bin/env bash" ]]; then
|
||
error "$HEADER: update error, invalid header: $line" && return
|
||
fi
|
||
|
||
if cmp --silent -- "${TMP}" "${SCRIPT}"; then
|
||
error "$HEADER: update file is already equal? (${local_size} / ${remote_size})" && return
|
||
fi
|
||
|
||
mv -f "${TMP}" "${SCRIPT}"
|
||
chmod 755 "${SCRIPT}"
|
||
|
||
info "$HEADER: succesfully installed update..."
|
||
|
||
}
|
||
|
||
function installPackages {
|
||
|
||
for filename in /usr/local/packages/*.spk; do
|
||
if [ -f "$filename" ]; then
|
||
|
||
BASE=$(basename "$filename" .spk)
|
||
BASE="${BASE%%-*}"
|
||
|
||
[[ $BASE == "ActiveInsight" ]] && continue
|
||
|
||
info "Installing package ${BASE}.."
|
||
|
||
/usr/syno/bin/synopkg install "$filename" > /dev/null
|
||
/usr/syno/bin/synopkg start "$BASE" > /dev/null &
|
||
|
||
rm "$filename"
|
||
|
||
fi
|
||
done
|
||
|
||
}
|
||
|
||
trap finish SIGINT SIGTERM
|
||
|
||
ts=$(date +%s%N)
|
||
|
||
echo ""
|
||
echo "❯ Started $HEADER v$VERSION..."
|
||
|
||
checkNMI
|
||
|
||
# Install packages
|
||
|
||
first_run=false
|
||
|
||
for filename in /usr/local/packages/*.spk; do
|
||
if [ -f "$filename" ]; then
|
||
first_run=true
|
||
fi
|
||
done
|
||
|
||
if [ "$first_run" = true ]; then
|
||
|
||
installPackages
|
||
|
||
else
|
||
|
||
downloadUpdate
|
||
|
||
fi
|
||
|
||
# Wait for NMI interrupt as a shutdown signal
|
||
|
||
while true; do
|
||
|
||
checkNMI
|
||
sleep 2 & wait $!
|
||
|
||
done
|