#!/bin/bash

declare -i time=60

declare -i delay=5

declare -i count=5

declare -i okCount=0

declare -i elapsed=0

declare isUp="false"

# Display usage

usage(){

echo -e "\e[32m╭──────────────────────────────────────────────────────────────╮"

echo -e "│ 🌍 \e[94murl-check.sh \e[96mCheck URL endpoint for HTTP responses 🚀\e[32m │"

echo -e "╰──────────────────────────────────────────────────────────────╯"

echo -e "\n\e[95mParameters:\e[37m"

echo -e " -u, --url \e[33mURL to check (required)\e[37m"

echo -e " [-t, --time] \e[33mMaximum number of seconds to poll for \e[92m(default: 60)\e[37m"

echo -e " [-d, --delay] \e[33mDelay in seconds between requests \e[92m(default: 5)\e[37m"

echo -e " [-c, --count] \e[33mHow many successes to receive before exiting \e[92m(default: 5)\e[37m"

echo -e " [-s, --search] \e[33mOptional content check, grep for this string in HTTP body \e[92m(default: none)\e[37m"

echo -e " [-h, --help] \e[33mShow this help text\e[37m"

}

OPTS=`getopt -o u:t:d:c:s:h --long url:,time:,delay:,count:,search:,help -n 'parse-options' -- "$@"`

if [ $? != 0 ] ; then echo "Failed parsing options." >&2 ; usage; exit 1 ; fi

eval set -- "$OPTS"

while true; do

case "$1" in

-u | --url ) url="$2"; shift; shift;;

-t | --time ) time="$2"; shift; shift;;

-d | --delay ) delay="$2"; shift; shift;;

-c | --count ) count="$2"; shift; shift;;

-s | --search ) search="$2"; shift; shift;;

-h | --help ) HELP=true; shift ;;

-- ) shift; break ;;

* ) break ;;

esac

done

if [[ ${HELP} = true ]] || [ -z ${url} ]; then

usage

exit 0

fi

# Check for impossible parameter combination ie. too many checks and delays in given time limit

if (( $delay * $count > $time)); then

echo -e "\e[31m### Error! The time ($time) provided is too short given the delay ($delay) and count ($count)\e[0m"

exit 1

fi

echo -e "\n\e[36m### Polling \e[33m$url\e[36m for ${time}s, to get $count OK results, with a ${delay}s delay\e[0m\n"

# Generate tmp filename

tmpfile=$(echo $url | md5sum)

# Main loop

while [ "$isUp" != "true" ]

do

# Break out of loop if max time has elapsed

if (( $elapsed >= $time )); then break; fi

timestamp=$(date "+%Y/%m/%d %H:%M:%S")

# Main CURL test, output to file and return http_code

urlstatus=$(curl -o "/tmp/$tmpfile" --silent --write-out '%{http_code}' "$url")

if [ $urlstatus -eq 000 ]; then

# Code 000 means DNS, network error or malformed URL

msg="\e[95mSite not found or other error"

else

if (( $urlstatus >= 200 )) && (( $urlstatus < 300 )); then

# Check returned content with grep if check specified

if [ ! -z "$search" ]; then

grep -q "$search" "/tmp/$tmpfile"

# Only count as a success if string grep passed

if (( $? == 0)); then

((okCount=okCount + 1))

msg="✅ \e[32m$urlstatus 🔍 Content check for '$search' passed"

else

msg="❌ \e[91m$urlstatus 🔍 Content check for '$search' failed"

fi

else

# Good status code

((okCount=okCount + 1))

msg="✅ \e[32m$urlstatus "

fi

if (( $okCount >= $count )); then isUp="true"; fi

else

# Bad status code

msg="❌ \e[91m$urlstatus "

fi

fi

# Output message + timestamp then delay

echo -e "### $timestamp: $msg\e[0m"

sleep $delay

((elapsed=elapsed + delay))

done

rm "/tmp/$tmpfile"

# Final result check

if [ "$isUp" == "true" ]; then

echo -e "\n\e[32m### Result: $url is UP! 🤩\e[0m"

exit 0

else

echo -e "\n\e[91m### Result: $url is DOWN! 😢\e[0m"

exit 1

fi