Add download() helper function

This commit is contained in:
2024-11-04 10:50:16 -05:00
parent 407cd61710
commit 1a656322e2

View File

@@ -208,7 +208,7 @@ init() {
declare -g CREATEREPO_USER="$USER" # can be root declare -g CREATEREPO_USER="$USER" # can be root
declare -g ID VERSION_ID ARCH NAME declare -g ID VERSION_ID ARCH NAME
declare -g MC_MVERSION MC_PKG MC_RPM MC_ROOT declare -g MC_MVERSION MC_PKG MC_RPM MC_ROOT
declare -ga PKG_INSTALL PKG_REMOVE PKG_UPDATE PKG_QUERY declare -ga PKG_INSTALL PKG_REMOVE PKG_UPDATE PKG_QUERY DOWNLOAD_CMD
declare -ga SERVICES CONTAINERS declare -ga SERVICES CONTAINERS
# Try to save users from themselves # Try to save users from themselves
@@ -411,8 +411,7 @@ get_latest_mc_version() {
mc_version_source="containerized package manager" mc_version_source="containerized package manager"
execute buildah rm "$cnt" execute buildah rm "$cnt"
# Webscrape # Webscrape
elif install_package --silent wget \ elif MC_VERSION=$(download "$BOARD_URL" | grep -o "[0-9][0-9]\.[0-9]\.[0-9]\+" | head -n 1) \
&& MC_VERSION=$(wget -qO- "$BOARD_URL" | grep -o "[0-9][0-9]\.[0-9]\.[0-9]\+" | head -n 1) \
&& [[ $MC_VERSION =~ ([0-9]+.[0-9]+.[0-9]+) ]]; then && [[ $MC_VERSION =~ ([0-9]+.[0-9]+.[0-9]+) ]]; then
mc_version_source="webscrape" mc_version_source="webscrape"
# Hardcoded # Hardcoded
@@ -608,9 +607,8 @@ install_mc_repo() {
repo_file="/etc/apt/sources.list.d/jriver.list" repo_file="/etc/apt/sources.list.d/jriver.list"
repo_text="deb [signed-by=$keyfile arch=amd64,i386,armhf,arm64] http://dist.jriver.com/latest/mediacenter/ $MC_REPO main" repo_text="deb [signed-by=$keyfile arch=amd64,i386,armhf,arm64] http://dist.jriver.com/latest/mediacenter/ $MC_REPO main"
fi fi
install_package wget
echo "Installing JRiver Media Center RPM key" echo "Installing JRiver Media Center RPM key"
wget --quiet --output-document=- http://dist.jriver.com/mediacenter@jriver.com.gpg.key | download "http://dist.jriver.com/mediacenter@jriver.com.gpg.key" |
gpg --dearmor | sudo tee "$keyfile" &>/dev/null gpg --dearmor | sudo tee "$keyfile" &>/dev/null
;; ;;
*) *)
@@ -665,7 +663,7 @@ acquire_deb() {
# Loop through the repositories and attempt to download # Loop through the repositories and attempt to download
for repo in "${repos[@]}"; do for repo in "${repos[@]}"; do
echo "Checking $repo for DEB package" echo "Checking $repo for DEB package"
if execute wget --quiet --output-document "$MC_DEB" "$repo"; then if execute download "$MC_DEB" "$repo"; then
echo "Found" echo "Found"
break break
fi fi
@@ -1542,7 +1540,7 @@ update() {
# Check if we're in a git directory and if it's the installJRMC repository # Check if we're in a git directory and if it's the installJRMC repository
if git -C "$SCRIPT_DIR" rev-parse --is-inside-work-tree &>/dev/null \ if git -C "$SCRIPT_DIR" rev-parse --is-inside-work-tree &>/dev/null \
&& [[ "$(git -C "$SCRIPT_DIR" config --get remote.origin.url)" == *"installJRMC"* ]]; then && [[ "$(git -C "$SCRIPT_DIR" config --get remote.origin.url)" == *"installJRMC"* ]]; then
debug "installJRMC git repository detected. Running git pull..." debug "installJRMC git repository detected. Running git pull"
# Get the current commit hash # Get the current commit hash
local before_pull_hash after_pull_hash local before_pull_hash after_pull_hash
@@ -1553,23 +1551,13 @@ update() {
return 0 return 0
fi fi
after_pull_hash=$(git -C "$SCRIPT_DIR" rev-parse HEAD) after_pull_hash=$(git -C "$SCRIPT_DIR" rev-parse HEAD)
if [[ "$before_pull_hash" != "$after_pull_hash" ]]; then [[ "$before_pull_hash" == "$after_pull_hash" ]] && return 0
echo "installJRMC repository updated. Restarting script..."
exec "$SCRIPT_PATH" "$@" "--no-update"
fi
else # Download the latest version of the script else # Download the latest version of the script
local script_url="https://git.bryanroessler.com/bryan/installJRMC/raw/master/installJRMC" local script_url="https://git.bryanroessler.com/bryan/installJRMC/raw/master/installJRMC"
local tmp; tmp=$(mktemp) local tmp; tmp=$(mktemp)
# Acquire latest version # Acquire latest version
install_package --silent wget download "$tmp" "$script_url" || return 1
if command -v wget &>/dev/null; then
execute wget -q -O "$tmp" "$script_url"
elif command -v curl &>/dev/null; then
execute curl -s -L -o "$tmp" "$script_url"
else
return 1
fi
# Get latest version number # Get latest version number
local remote_version local remote_version
@@ -1578,13 +1566,16 @@ update() {
# Compare versions and update if necessary # Compare versions and update if necessary
if version_greater "$remote_version" "$SCRIPT_VERSION"; then if version_greater "$remote_version" "$SCRIPT_VERSION"; then
echo "Updating installJRMC $SCRIPT_VERSION to $remote_version"
execute mv "$tmp" "$SCRIPT_PATH" execute mv "$tmp" "$SCRIPT_PATH"
execute chmod +x "$SCRIPT_PATH" execute chmod +x "$SCRIPT_PATH"
rm -f "$tmp" execute rm -f "$tmp"
else
return 0
fi
fi
echo "installJRMC updated, restarting"
exec "$SCRIPT_PATH" "$@" "--no-update" exec "$SCRIPT_PATH" "$@" "--no-update"
fi
fi
} }
# @description installJRMC main function # @description installJRMC main function
@@ -1627,7 +1618,6 @@ main() {
fi fi
if (( BUILD_SWITCH )) && [[ $ID != "arch" ]]; then if (( BUILD_SWITCH )) && [[ $ID != "arch" ]]; then
install_package "wget"
[[ -d $OUTPUT_DIR/SOURCES ]] || execute mkdir -p "$OUTPUT_DIR/SOURCES" [[ -d $OUTPUT_DIR/SOURCES ]] || execute mkdir -p "$OUTPUT_DIR/SOURCES"
acquire_deb || { err "Could not download Media Center DEB package"; return 1; } acquire_deb || { err "Could not download Media Center DEB package"; return 1; }
@@ -1702,6 +1692,39 @@ execute() {
"$@" &>/dev/null "$@" &>/dev/null
fi fi
} }
download() {
local url="$1"
local output="${2:-}"
local download_cmd
if command -v wget &>/dev/null; then
download_cmd=(wget --quiet)
elif command -v curl &>/dev/null; then
download_cmd=(curl --silent --location)
else
if install_package --quiet wget; then
download_cmd=(wget --quiet)
elif install_package --quiet curl; then
download_cmd=(curl --silent --location)
else
err "Unable to install wget or curl"
return 1
fi
fi
if [[ ${download_cmd[0]} == "wget" ]]; then
"${download_cmd[@]}" --output-document="${output:--}" "$url"
elif [[ ${download_cmd[0]} == "curl" ]]; then
if [[ -n "$output" ]]; then
"${download_cmd[@]}" --output "$output" "$url"
else
"${download_cmd[@]}" "$url"
fi
else
err "Unsupported download command: ${download_cmd[*]}"
return 1
fi
}
# Roughly turn debugging on, reparse in parse_input() with getopt # Roughly turn debugging on, reparse in parse_input() with getopt
[[ " $* " =~ ( --debug | -d ) ]] && declare -g DEBUG=1 [[ " $* " =~ ( --debug | -d ) ]] && declare -g DEBUG=1