diff --git a/pihole b/pihole index d4681752..fff3417d 100755 --- a/pihole +++ b/pihole @@ -23,17 +23,6 @@ source "${colfile}" resolver="pihole-FTL" -# Must be root to use this tool -if [[ ! $EUID -eq 0 ]];then - if [[ -x "$(command -v sudo)" ]]; then - exec sudo bash "$0" "$@" - exit $? - else - echo -e " ${CROSS} sudo is needed to run pihole commands. Please run this script as root or install sudo." - exit 1 - fi -fi - webpageFunc() { source "${PI_HOLE_SCRIPT_DIR}/webpage.sh" main "$@" @@ -430,6 +419,21 @@ if [[ $# = 0 ]]; then helpFunc fi +case "${1}" in + "-h" | "help" | "--help" ) helpFunc;; +esac + +# Must be root to use this tool +if [[ ! $EUID -eq 0 ]];then + if [[ -x "$(command -v sudo)" ]]; then + exec sudo bash "$0" "$@" + exit $? + else + echo -e " ${CROSS} sudo is needed to run pihole commands. Please run this script as root or install sudo." + exit 1 + fi +fi + # Handle redirecting to specific functions based on arguments case "${1}" in "-w" | "whitelist" ) listFunc "$@";;