Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Added asn component. #246

Open
wants to merge 1 commit into
base: master
Choose a base branch
from
Open
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
203 changes: 203 additions & 0 deletions asn/asn
Original file line number Diff line number Diff line change
@@ -0,0 +1,203 @@
#!/usr/bin/env bash
# Author: IP2Location https://github.com/ip2location
currentVersion="1.23.0"
configuredClient=""

## This function determines which http get tool the system has installed and returns an error if there isnt one
getConfiguredClient()
{
if command -v curl &>/dev/null; then
configuredClient="curl"
elif command -v wget &>/dev/null; then
configuredClient="wget"
elif command -v http &>/dev/null; then
configuredClient="httpie"
elif command -v fetch &>/dev/null; then
configuredClient="fetch"
else
echo "Error: This tool requires either curl, wget, httpie or fetch to be installed." >&2
return 1
fi
}

# The python and JSON functions can be deleted if the script does not need them

## This function determines which version of python is installed on the machine
getConfiguredPython()
{
if command -v python3 &>/dev/null; then
configuredPython="python3"
elif command -v python2 &>/dev/null; then
configuredPython="python2"
elif command -v python &>/dev/null; then
configuredPython="python"
else
echo "Error: This tool requires python to be installed."
return 1
fi
}

## Don't alter the python call if it's coming from a Mac
if [[ $(uname) != "Darwin" ]]; then
python()
{
case "$configuredPython" in
python3) python3 "$@" ;;
python2) python2 "$@" ;;
python) python "$@" ;;
esac
}
fi

## Grabs an element from a a json string and then echoes it to stdout
## $1 = the JSON string
## $n+1 = the elements to be indexed
AccessJsonElement() {
json="$1"
shift
accessor=""
for element in "$@"; do
accessor="${accessor}['$element']"
done
echo "$json" | python -c "from __future__ import print_function; import sys, json; print(json.load(sys.stdin)${accessor})"
return "$?"
}

## Allows to call the users configured client without if statements everywhere
httpGet()
{
case "$configuredClient" in
curl) curl -A curl -s "$@" ;;
wget) wget -qO- "$@" ;;
httpie) http -b GET "$@" ;;
fetch) fetch -q "$@" ;;
esac
}

update()
{
# Author: Alexander Epstein https://github.com/alexanderepstein
# Update utility version 1.2.0
# To test the tool enter in the default values that are in the examples for each variable
repositoryName="Bash-Snippets" #Name of repostiory to be updated ex. Sandman-Lite
githubUserName="alexanderepstein" #username that hosts the repostiory ex. alexanderepstein
nameOfInstallFile="install.sh" # change this if the installer file has a different name be sure to include file extension if there is one
latestVersion=$(httpGet https://api.github.com/repos/$githubUserName/$repositoryName/tags | grep -Eo '"name":.*?[^\\]",'| head -1 | grep -Eo "[0-9.]+" ) #always grabs the tag without the v option

if [[ $currentVersion == "" || $repositoryName == "" || $githubUserName == "" || $nameOfInstallFile == "" ]]; then
echo "Error: update utility has not been configured correctly." >&2
exit 1
elif [[ $latestVersion == "" ]]; then
echo "Error: no active internet connection" >&2
exit 1
else
if [[ "$latestVersion" != "$currentVersion" ]]; then
echo "Version $latestVersion available"
echo -n "Do you wish to update $repositoryName [Y/n]: "
read -r answer
if [[ "$answer" == [Yy] ]]; then
cd ~ || { echo 'Update Failed'; exit 1; }
if [[ -d ~/$repositoryName ]]; then rm -r -f $repositoryName || { echo "Permissions Error: try running the update as sudo"; exit 1; } ; fi
git clone "https://github.com/$githubUserName/$repositoryName" || { echo "Couldn't download latest version"; exit 1; }
cd $repositoryName || { echo 'Update Failed'; exit 1; }
git checkout "v$latestVersion" 2> /dev/null || git checkout "$latestVersion" 2> /dev/null || echo "Couldn't git checkout to stable release, updating to latest commit."
chmod a+x install.sh #this might be necessary in your case but wasnt in mine.
./$nameOfInstallFile "update" || exit 1
cd ..
rm -r -f $repositoryName || { echo "Permissions Error: update succesfull but cannot delete temp files located at ~/$repositoryName delete this directory with sudo"; exit 1; }
else
exit 1
fi
else
echo "$repositoryName is already the latest version"
fi
fi
}

checkInternet()
{
httpGet github.com > /dev/null 2>&1 || { echo "Error: no active internet connection" >&2; return 1; } # query github with a get request
}

usage()
{
cat <<EOF
asn
Description: A simple tool to return the ASN data of an IP address along with it's corresponding geolocation data. This tool is using IP2Location.io API which provided 500 queries per day for keyless query, or 30000 queries per month if you sign up the free plan.
Usage: asn [flags]
-g Lookup an IP address for it's ASN data plus geolocation information
-i Set the IP address to be lookup for
-k Set the IP2Location.io API key (Optional)
-u Update Bash-Snippet Tools
-h Show the help
-v Get the tool version
Examples:
asn -g
asn -i 8.8.8.8 -g
asn -k <YOUR_IP2LOCATIONIO_APIKEY> -g
asn -i 8.8.8.8 -k <YOUR_IP2LOCATIONIO_APIKEY> -g
EOF
}


getasn() {
if [ -n "${IP}" ]; then
echo "Querying for ${IP} now..."
if [ -n "${APIKEY}" ]; then
httpGet "https://api.ip2location.io/?key=${APIKEY}&ip=${IP}"
else
httpGet "https://api.ip2location.io/?ip=${IP}"
fi
else
echo "Querying for server IP now..."
if [ -n "${APIKEY}" ]; then
httpGet "https://api.ip2location.io/?key=${APIKEY}"
else
httpGet "https://api.ip2location.io/"
fi
fi
}

test1() {
if [ -n "${IP}" ]; then
echo "Querying for ${IP} now..."
else
echo "Querying for server IP now..."
fi
}

while getopts "gti:k:hvu" opt; do
case "$opt" in
h) usage
exit 0
;;
g) getConfiguredClient && checkInternet && getasn || exit 1 ;;
i) IP=$OPTARG ;;
k) APIKEY=$OPTARG ;;
t) test1 || exit 1 ;;
v) echo "Version $currentVersion"
exit 0
;;
u) getConfiguredClient || exit 1
checkInternet || exit 1
update
exit 0
;;
esac
done

# special set of first arguments that have a specific behavior across tools
if [[ $# == "0" ]]; then
usage ## if calling the tool with no flags and args chances are you want to return usage
exit 0
elif [[ $# == "1" ]]; then
if [[ $1 == "update" ]]; then
getConfiguredClient || exit 1
checkInternet || exit 1
update || exit 1
exit 0
elif [[ $1 == "help" ]]; then
usage
exit 0
fi
fi