From 42cb27f6e0da1e990d26ea058f5d1a762ae59704 Mon Sep 17 00:00:00 2001 From: Jon Shaulis Date: Tue, 6 Sep 2016 01:29:06 -0400 Subject: [PATCH] Change options to be robust --- README.md | 41 ++++++--- plexupdate.sh | 243 ++++++++++++++++++++++++++++++++------------------ 2 files changed, 181 insertions(+), 103 deletions(-) diff --git a/README.md b/README.md index c530449..02afca6 100644 --- a/README.md +++ b/README.md @@ -17,43 +17,39 @@ Download it as a [zip file](https://github.com/mrworf/plexupdate/archive/master. ``` wget https://github.com/mrworf/plexupdate/archive/master.zip && unzip master.zip && mv plexupdate-master plexupdate && rm master.zip ``` -Note that unzip is requered (`sudo apt-get install unzip`). +Note that unzip is required (`sudo apt-get install unzip`). ####Using git to clone (Recommended) Using git is way easier and recommended, if you ask me. ``` git clone https://github.com/mrworf/plexupdate.git ``` -Note that git is requered (`sudo apt-get install git-all`) +Note that git is required (`sudo apt-get install git-all`) The main benefit with git clone is that you can update to latest version very easily. If you want to use the auto update feature, you must be using a git clone. ## 2. Setting it up -plexupdate.sh looks for a file named `.plexupdate` located in your home directory. Please note that I'm referring to the home directory of the user who is running the plexupdate.sh ... If that user is someone else (root for instance) you'll need to make sure that user has the config file set up properly as well. - -The contents of this file are usually: +To quickly setup plexupdate.sh, you should run it the first time like below: ``` -EMAIL="my.email@plex-server.com" -PASS="my-secret-plex-password" -DOWNLOADDIR="/a/folder/to/save/the/files/in" +./plexupdate.sh --email="my.email@plex-server.com" --pass="my-secret-plex-password" --dldir="/a/folder/to/save/the/files/in" --saveconfig ``` -Obviously you need to change these so they match your account information. And if you don't put anything as a `DOWNLOADDIR`, the tool will use the folder you're executing the script from. So take care. +Obviously you need to change these so they match your account information. And if you don't put anything as for the ```--dldir``` option, the tool will use the folder you're executing the script from. So take care. ## 3. Advanced options -You can point out a different file than ```.plexupdate``` by providing it as the last argument to the script. It HAS to be the LAST argument, or it will be ignored. Any options set by the config file can be overriden with command-line options. +You can point out a different file than ```.plexupdate``` by providing it as the argument to the ```--config``` option. Any options set by the config file can be overridden with command-line options. There are also a few additional options for the more enterprising user. Setting any of these to `yes` will enable the function. - PLEXSERVER - If set, and combined with AUTOINSTALL, the script will automatically check if the server is in-use and deferr the update. Great for crontab users. PLEXSERVER should be set to the IP/DNS of your Plex Media Server, which typically is 127.0.0.1 + If set, and combined with AUTOINSTALL, the script will automatically check if the server is in-use and defer the update. Great for crontab users. PLEXSERVER should be set to the IP/DNS of your Plex Media Server, which typically is 127.0.0.1 - AUTOUPDATE Makes plexupdate.sh automatically update itself using git. Note! This will fail if git isn't available on the command line. - AUTOINSTALL - Automatically installs the newly downloaded version. Currently works for debian based systems as well as rpm based distros. Will fail miserably if you're not root. + Automatically installs the newly downloaded version. Currently works for Debian based systems as well as rpm based distros. Will fail miserably if you're not root. - AUTODELETE Once successfully downloaded and installed, it will delete the package (want not, waste not? ;-)) - PUBLIC @@ -76,7 +72,24 @@ Most of these options can be specified on the command-line as well, this is just If you want to use plexupdate as either a cron job or as a [systemd job](https://github.com/mrworf/plexupdate/wiki/Running-plexupdate-daily-as-a-systemd-timer), the -c option should do what you want. All non-error exit codes will be set to 0 and no output will be printed to stdout unless something has actually been done. (a new version was downloaded, installed, etc) -If you don't even want to know when something has been done, you can combine this with the -q option and you will only receive output in the event of an error. Everything else will just silenty finish without producing any output. +If you don't even want to know when something has been done, you can combine this with the -q option and you will only receive output in the event of an error. Everything else will just silently finish without producing any output. + +### Command Line Options + +Several new command line options are available. They can be specified in any order. + +- ```--config ``` + Defines the location the script should look for the config file. +- ```--email ``` + Email to sign in to Plex.tv +- ```--pass ``` + Password to sign in to Plex.tv +- ```--dldir ``` + This is the folder that the files will be downloaded to. +- ```--server ``` + This is the address that Plex Media Server is on. Setting this will enable a check to see if users are on the server prior to the software being updated. +- ```--saveconfig``` + Saves the configuration as it is currently. This will take whatever is in the config file, plus whatever is specified on the command line and will save the config file with that information. Any information in the config file that plexupdate.sh does not understand or use WILL BE LOST. # Running it @@ -86,7 +99,7 @@ Overall it tries to give you hints regarding why it isn't doing what you expecte # Trivia -- "kaka" is swedish for "cookie" +- "kaka" is Swedish for "cookie" # TL;DR Open a terminal or SSH on the server running Plex Media Center diff --git a/plexupdate.sh b/plexupdate.sh index a72900f..ab98436 100755 --- a/plexupdate.sh +++ b/plexupdate.sh @@ -67,64 +67,13 @@ REDHAT_INSTALL="yum -y install" DEBIAN_INSTALL="dpkg -i" DISTRO_INSTALL="" -# Sanity, make sure wget is in our path... -if ! hash wget 2>/dev/null; then - echo "ERROR: This script requires wget in the path. It could also signify that you don't have the tool installed." >&2 - exit 1 -fi - -# Allow manual control of configfile -HASCFG="${@: -1}" -if [ ! -z "${HASCFG}" -a ! "${HASCFG:0:1}" = "-" ]; then - if [ -f "${HASCFG}" ]; then - source "${HASCFG}" - else - echo "ERROR: Cannot load configuration ${HASCFG}" >&2 - exit 1 - fi -else - # Load settings from config file if it exists - # Also, respect SUDO_USER and try that first - if [ ! -z "${SUDO_USER}" ]; then - # Make sure nothing bad comes from this (since we use eval) - ERROR=0 - if [[ $SUDO_USER == *";"* ]]; then ERROR=1 ; # Allows more commands - elif [[ $SUDO_USER == *" "* ]]; then ERROR=1 ; # Space is not a good thing - elif [[ $SUDO_USER == *"&"* ]]; then ERROR=1 ; # Spinning off the command is bad - elif [[ $SUDO_USER == *"<"* ]]; then ERROR=1 ; # No redirection - elif [[ $SUDO_USER == *">"* ]]; then ERROR=1 ; # No redirection - elif [[ $SUDO_USER == *"|"* ]]; then ERROR=1 ; # No pipes - elif [[ $SUDO_USER == *"~"* ]]; then ERROR=1 ; # No tilde - fi - if [ ${ERROR} -gt 0 ]; then - echo "ERROR: SUDO_USER variable is COMPROMISED: \"${SUDO_USER}\"" >&2 - exit 255 - fi - - # Try using original user's config - CONFIGDIR="$( eval cd ~${SUDO_USER} 2>/dev/null && pwd )" - if [ -z "${CONFIGDIR}" ]; then - echo "WARNING: SUDO_USER \"${SUDO_USER}\" does not have a valid home directory, ignoring." >&2 - fi - - if [ ! -z "${CONFIGDIR}" -a -f "${CONFIGDIR}/.plexupdate" ]; then - #echo "INFO: Using \"${SUDO_USER}\" configuration: ${CONFIGDIR}/.plexupdate" - source "${CONFIGDIR}/.plexupdate" - elif [ -f ~/.plexupdate ]; then - # Fallback for compatibility - source ~/.plexupdate - fi - elif [ -f ~/.plexupdate ]; then - # Fallback for compatibility - source ~/.plexupdate - fi -fi - # Current pages we need - Do not change unless Plex.tv changes again URL_LOGIN=https://plex.tv/users/sign_in.json URL_DOWNLOAD=https://plex.tv/api/downloads/1.json?channel=plexpass URL_DOWNLOAD_PUBLIC=https://plex.tv/api/downloads/1.json +# Functions for rest of script + cronexit() { # Don't give anything but true error codes if in CRON mode RAWEXIT=$1 @@ -135,27 +84,33 @@ cronexit() { } usage() { - echo "Usage: $(basename $0) [-acfhopqsSuU] [config file]" + echo "Usage: $(basename $0) [-acdfFhlpqsuU]" echo "" - echo " config file overrides the default ~/.plexupdate" - echo " If used, it must be the LAST option or it will be ignored" echo "" - echo " -a Auto install if download was successful (requires root)" + echo " -a Auto install if download was successful (requires root)" echo " -c Cron mode, only fatal errors return non-zero cronexit code" - echo " -d Auto delete after auto install" - echo " -f Force download even if it's the same version or file" - echo " already exists unless checksum passes" - echo " -F Force download always" - echo " -h This help" - echo " -l List available builds and distros" - echo " -p Public Plex Media Server version" - echo " -q Quiet mode. No stdout, only stderr and cronexit codes" - echo " -r Print download URL and exit" - echo " -s Auto start (needed for some distros)" - echo " -u Auto update plexupdate.sh before running it (experimental)" - echo " -U Do not autoupdate plexupdate.sh (experimental, default)" - echo - cronexit 0 + echo " -d Auto delete after auto install" + echo " -f Force download even if it's the same version or file" + echo " already exists unless checksum passes" + echo " -F Force download always" + echo " -h This help" + echo " -l List available builds and distros" + echo " -p Public Plex Media Server version" + echo " -q Quiet mode. No stdout, only stderr and cronexit codes" + echo " -r Print download URL and exit" + echo " -s Auto start (needed for some distros)" + echo " -u Auto update plexupdate.sh before running it (experimental)" + echo " -U Do not autoupdate plexupdate.sh (experimental, default)" + echo "" + echo " Long Argument Options:" + echo " --config Configuration file to use" + echo " --dldir Download directory to use" + echo " --email Plex.TV email address" + echo " --pass Plex.TV password" + echo " --server Address of Plex Server" + echo " --saveconfig Save the command line arguments to " + echo + cronexit 0 } running() { @@ -185,37 +140,147 @@ running() { # Parse commandline ALLARGS=( "$@" ) -optstring="acCdfFhlpqrSsuU" +optstring="acCdfFhlpqrSsuU -l config:,dldir:,email:,pass:,server:,saveconfig" getopt -T >/dev/null if [ $? -eq 4 ]; then optstring="-o $optstring" fi -set -- $(getopt $optstring -- "$@") +GETOPTRES=$(getopt $optstring -- "$@") +if [ $? -eq 1 ]; then + cronexit 1 +fi + +set -- ${GETOPTRES} while true; do case "$1" in - (-h) usage;; - (-a) AUTOINSTALL=yes;; - (-c) CRON=yes;; + (-h) usage;; + (-a) AUTOINSTALL_CL=yes;; + (-c) CRON_CL=yes;; (-C) echo "ERROR: CRON option has changed, please review README.md" >&2; cronexit 255;; - (-d) AUTODELETE=yes;; - (-f) FORCE=yes;; - (-F) FORCEALL=yes;; - (-l) LISTOPTS=yes;; - (-p) PUBLIC=yes;; - (-q) QUIET=yes;; - (-r) PRINT_URL=yes;; - (-s) AUTOSTART=yes;; + (-d) AUTODELETE_CL=yes;; + (-f) FORCE_CL=yes;; + (-F) FORCEALL_CL=yes;; + (-l) LISTOPTS=yes;; + (-p) PUBLIC_CL=yes;; + (-q) QUIET_CL=yes;; + (-r) PRINT_URL=yes;; + (-s) AUTOSTART_CL=yes;; (-S) echo "ERROR: SILENT option has been removed, please use QUIET (-q) instead" >&2; cronexit 255;; - (-u) AUTOUPDATE=yes;; - (-U) IGNOREAUTOUPDATE=yes;; - (--) ;; - (-*) echo "ERROR: unrecognized option $1" >&2; usage; cronexit 1;; - (*) break;; + (-u) AUTOUPDATE_CL=yes;; + (-U) IGNOREAUTOUPDATE=yes;; + + (--config) shift; CONFIGFILE=$(echo "$1" | tr -d "'");; + (--dldir) shift; DOWNLOADDIR_CL=$(echo "$1" | tr -d "'");; + (--email) shift; EMAIL_CL=$(echo "$1" | tr -d "'");; + (--pass) shift; PASS_CL=$(echo "$1" | tr -d "'");; + (--server) shift; PLEXSERVER_CL=$(echo "$1" | tr -d "'");; + (--saveconfig) SAVECONFIG=yes;; + + (--) ;; + (-*) echo "ERROR: unrecognized option $1" >&2; usage; cronexit 1;; + (*) break;; esac shift done +# Sanity, make sure wget is in our path... +if ! hash wget 2>/dev/null; then + echo "ERROR: This script requires wget in the path. It could also signify that you don't have the tool installed." >&2 + exit 1 +fi + +# Allow manual control of configfile +if [ ! -z "${CONFIGFILE}" ]; then + if [ -f "${CONFIGFILE}" ]; then + source "${CONFIGFILE}" + else + echo "ERROR: Cannot load configuration ${CONFIGFILE}" >&2 + exit 1 + fi +else + # Load settings from config file if it exists + # Also, respect SUDO_USER and try that first + if [ ! -z "${SUDO_USER}" ]; then + # Make sure nothing bad comes from this (since we use eval) + ERROR=0 + if [[ $SUDO_USER == *";"* ]]; then ERROR=1 ; # Allows more commands + elif [[ $SUDO_USER == *" "* ]]; then ERROR=1 ; # Space is not a good thing + elif [[ $SUDO_USER == *"&"* ]]; then ERROR=1 ; # Spinning off the command is bad + elif [[ $SUDO_USER == *"<"* ]]; then ERROR=1 ; # No redirection + elif [[ $SUDO_USER == *">"* ]]; then ERROR=1 ; # No redirection + elif [[ $SUDO_USER == *"|"* ]]; then ERROR=1 ; # No pipes + elif [[ $SUDO_USER == *"~"* ]]; then ERROR=1 ; # No tilde + fi + if [ ${ERROR} -gt 0 ]; then + echo "ERROR: SUDO_USER variable is COMPROMISED: \"${SUDO_USER}\"" >&2 + exit 255 + fi + + # Try using original user's config + CONFIGDIR="$( eval cd ~${SUDO_USER} 2>/dev/null && pwd )" + if [ -z "${CONFIGDIR}" ]; then + echo "WARNING: SUDO_USER \"${SUDO_USER}\" does not have a valid home directory, ignoring." >&2 + fi + + if [ ! -z "${CONFIGDIR}" -a -f "${CONFIGDIR}/.plexupdate" ]; then + #echo "INFO: Using \"${SUDO_USER}\" configuration: ${CONFIGDIR}/.plexupdate" + CONFIGFILE="${CONFIGDIR}/.plexupdate" + source "${CONFIGDIR}/.plexupdate" + elif [ -f ~/.plexupdate ]; then + # Fallback for compatibility + CONFIGFILE="${HOME}/.plexupdate" # tilde expansion won't happen later. + source ~/.plexupdate + fi + elif [ -f ~/.plexupdate ]; then + # Fallback for compatibility + CONFIGFILE="${HOME}/.plexupdate" + source ~/.plexupdate + fi +fi + +# The way I wrote this, it assumes that whatever we put on the command line is what we want and should override +# any values in the configuration file. As a result, we need to check if they've been set on the command line +# and overwrite the values that may have been loaded with the config file + +for VAR in AUTOINSTALL CRON AUTODELETE DOWNLOADDIR EMAIL PASS FORCE FORCEALL PUBLIC QUIET AUTOSTART AUTOUPDATE PLEXSERVER +do + VAR2="$VAR""_CL" + if [ ! -z ${!VAR2} ]; then + eval $VAR=${!VAR2} + fi +done + +# This will destroy and recreate the config file. Any settings that are set in the config file but are no longer +# valid will NOT be saved. +if [ "${SAVECONFIG}" = "yes" ]; then + echo "# Config file for plexupdate" >${CONFIGFILE:="${HOME}/.plexupdate"} + + for VAR in AUTOINSTALL CRON AUTODELETE DOWNLOADDIR EMAIL PASS FORCE FORCEALL PUBLIC QUIET AUTOSTART AUTOUPDATE PLEXSERVER + do + if [ ! -z ${!VAR} ]; then + + # The following keys have defaults set in this file. We don't want to include these values if they are the default. + if [ ${VAR} = "FORCE" \ + -o ${VAR} = "FORCEALL" \ + -o ${VAR} = "PUBLIC" \ + -o ${VAR} = "AUTOINSTALL" \ + -o ${VAR} = "AUTODELETE" \ + -o ${VAR} = "AUTOUPDATE" \ + -o ${VAR} = "AUTOSTART" \ + -o ${VAR} = "CRON" \ + -o ${VAR} = "QUIET" ]; then + + if [ ${!VAR} = "yes" ]; then + echo "${VAR}=${!VAR}" >> ${CONFIGFILE} + fi + else + echo "${VAR}=${!VAR}" >> ${CONFIGFILE} + fi + fi + done +fi + if [ "${IGNOREAUTOUPDATE}" = "yes" ]; then AUTOUPDATE=no fi