shell-scripting-templates/utilities/services.bash

# Functions for working with network services

_httpStatus_() {
    # DESC:
    #         Report the HTTP status of a specified URL
    # ARGS:
    #         $1 (Required) - URL (will work fine without https:// prefix)
    #         $2 (Optional) - Seconds to wait until timeout (Default is 3)
    #         $3 (Optional) - either '--code'  or '--status' (default)
    #         $4 (optional) - CURL opts separated by spaces (Use -L to follow redirects)
    # OUTS:
    #         stdout: Prints the HTTP status code or status message
    # USAGE:
    #         _httpStatus_ URL [timeout] [--code or --status] [curl opts]
    # NOTE:
    #         https://gist.github.com/rsvp/1171304
    # EXAMPLES
    #         $ _httpStatus_ bit.ly
    #         301 Redirection: Moved Permanently
    #
    #         $ _httpStatus_ www.google.com 100 --code

    [[ $# == 0 ]] && fatal "Missing required argument to ${FUNCNAME[0]}"

    local _saveIFS=${IFS}
    IFS=$' \n\t'

    local _url=${1}
    local _timeout=${2:-3}
    local _flag=${3:---status}
    local _arg4=${4:-}
    local _arg5=${5:-}
    local _arg6=${6:-}
    local _arg7=${7:-}
    local _curlops="${_arg4} ${_arg5} ${_arg6} ${_arg7}"
    local _code
    local _status

    #      __________ get the CODE which is numeric:
    _code=$(curl --write-out %{http_code} --silent --connect-timeout "${_timeout}" \
        --no-keepalive ${_curlops} --output /dev/null ${_url})

    #      __________ get the STATUS (from code) which is human interpretable:
    case $_code in
        000) _status="Not responding within ${_timeout} seconds" ;;
        100) _status="Informational: Continue" ;;
        101) _status="Informational: Switching Protocols" ;;
        200) _status="Successful: OK within ${_timeout} seconds" ;;
        201) _status="Successful: Created" ;;
        202) _status="Successful: Accepted" ;;
        203) _status="Successful: Non-Authoritative Information" ;;
        204) _status="Successful: No Content" ;;
        205) _status="Successful: Reset Content" ;;
        206) _status="Successful: Partial Content" ;;
        300) _status="Redirection: Multiple Choices" ;;
        301) _status="Redirection: Moved Permanently" ;;
        302) _status="Redirection: Found residing temporarily under different URI" ;;
        303) _status="Redirection: See Other" ;;
        304) _status="Redirection: Not Modified" ;;
        305) _status="Redirection: Use Proxy" ;;
        306) _status="Redirection: status not defined" ;;
        307) _status="Redirection: Temporary Redirect" ;;
        400) _status="Client Error: Bad Request" ;;
        401) _status="Client Error: Unauthorized" ;;
        402) _status="Client Error: Payment Required" ;;
        403) _status="Client Error: Forbidden" ;;
        404) _status="Client Error: Not Found" ;;
        405) _status="Client Error: Method Not Allowed" ;;
        406) _status="Client Error: Not Acceptable" ;;
        407) _status="Client Error: Proxy Authentication Required" ;;
        408) _status="Client Error: Request Timeout within ${_timeout} seconds" ;;
        409) _status="Client Error: Conflict" ;;
        410) _status="Client Error: Gone" ;;
        411) _status="Client Error: Length Required" ;;
        412) _status="Client Error: Precondition Failed" ;;
        413) _status="Client Error: Request Entity Too Large" ;;
        414) _status="Client Error: Request-URI Too Long" ;;
        415) _status="Client Error: Unsupported Media Type" ;;
        416) _status="Client Error: Requested Range Not Satisfiable" ;;
        417) _status="Client Error: Expectation Failed" ;;
        500) _status="Server Error: Internal Server Error" ;;
        501) _status="Server Error: Not Implemented" ;;
        502) _status="Server Error: Bad Gateway" ;;
        503) _status="Server Error: Service Unavailable" ;;
        504) _status="Server Error: Gateway Timeout within ${_timeout} seconds" ;;
        505) _status="Server Error: HTTP Version Not Supported" ;;
        *) fatal "httpstatus: status not defined." ;;
    esac

    case ${_flag} in
        --status) echo "${_code} ${_status}" ;;
        -s) echo "${_code} ${_status}" ;;
        --code) echo "${_code}" ;;
        -c) echo "${_code}" ;;
        *) echo " _httpStatus_: bad flag" && _safeExit_ ;;
    esac

    IFS="${_saveIFS}"
}

_pushover_() {
    # DESC:
    #         Sends a notification via Pushover
    # ARGS:
    #         $1 (Required) - Title of notification
    #         $2 (Required) - Body of notification
    #         $3 (Required) - User Token
    #         $4 (Required) - API Key
    #         $5 (Optional) - Device
    # OUTS:
    #         0 if success
    #         1 if failure
    # USAGE:  _pushover_ "Title Goes Here" "Message Goes Here"
    # NOTE:   The variables for the two API Keys must have valid values
    #         Credit: http://ryonsherman.blogspot.com/2012/10/shell-script-to-send-pushover.html

    [[ $# -lt 4 ]] && fatal "Missing required argument to ${FUNCNAME[0]}"

    local _pushoverURL="https://api.pushover.net/1/messages.json"
    local _message="${2}"
    local _apiKey="${3}"
    local _userKey="${4}"
    local _device="${5:-}"

    if curl \
        -F "token=${_apiKey}" \
        -F "user=${_userKey}" \
        -F "device=${_device}" \
        -F "title=${_messageTitle}" \
        -F "message=${_message}" \
        "${_pushoverURL}" >/dev/null 2>&1; then
        return 0
    else
        return 1
    fi
}