mirror of
https://github.com/alexanderepstein/Bash-Snippets
synced 2018-11-08 02:59:35 +01:00
186 lines
8.7 KiB
Bash
Executable file
186 lines
8.7 KiB
Bash
Executable file
#!/bin/bash
|
|
# Author: Alexander Epstein https://github.com/alexanderepstein
|
|
|
|
currentVersion="1.3.1"
|
|
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 fetch &>/dev/null ; then
|
|
configuredClient="fetch"
|
|
else
|
|
echo "Error: This tool reqires either curl, wget, or fetch to be installed."
|
|
return 1
|
|
fi
|
|
|
|
}
|
|
|
|
## Allows to call the users configured client without if statements everywhere
|
|
httpGet()
|
|
{
|
|
case "$configuredClient" in
|
|
curl) curl -A curl -s "$@";;
|
|
wget) wget -qO- "$@";;
|
|
fetch) fetch -o "...";;
|
|
esac
|
|
}
|
|
|
|
checkInternet()
|
|
{
|
|
echo -e "GET http://google.com HTTP/1.0\n\n" | nc google.com 80 > /dev/null 2>&1 # query google with a get request
|
|
if [ $? -eq 0 ]; then #check if the output is 0, if so no errors have occured and we have connected to google successfully
|
|
return 0
|
|
else
|
|
echo "Error: no active internet connection" >&2 #sent to stderr
|
|
return 1
|
|
fi
|
|
}
|
|
|
|
## This function grabs information about a stock and using python parses the
|
|
## JSON response to extrapolate the information for storage
|
|
getStockInformation()
|
|
{
|
|
stockInfo=$(httpGet "https://www.alphavantage.co/query?function=GLOBAL_QUOTE&symbol=$1&apikey=KPCCCRJVMOGN9L6T") > /dev/null #grab the JSON response
|
|
export PYTHONIOENCODING=utf8 #necessary for python in some cases
|
|
echo $stockInfo | python -c "import sys, json; print json.load(sys.stdin)['Realtime Global Securities Quote']['02. Exchange Name']" > /dev/null 2>&1 || { echo "Not a valid stock symbol" ; exit 1; } #checking if we get any information back from the server if not chances are it isnt a valid stock symbol
|
|
# The rest of the code is just extrapolating the data with python from the JSON response
|
|
exchangeName=$(echo $stockInfo | python -c "import sys, json; print json.load(sys.stdin)['Realtime Global Securities Quote']['02. Exchange Name']")
|
|
latestPrice=$(echo $stockInfo | python -c "import sys, json; print json.load(sys.stdin)['Realtime Global Securities Quote']['03. Latest Price']")
|
|
open=$(echo $stockInfo | python -c "import sys, json; print json.load(sys.stdin)['Realtime Global Securities Quote']['04. Open (Current Trading Day)']")
|
|
high=$(echo $stockInfo | python -c "import sys, json; print json.load(sys.stdin)['Realtime Global Securities Quote']['05. High (Current Trading Day)']")
|
|
low=$(echo $stockInfo | python -c "import sys, json; print json.load(sys.stdin)['Realtime Global Securities Quote']['06. Low (Current Trading Day)']")
|
|
close=$(echo $stockInfo | python -c "import sys, json; print json.load(sys.stdin)['Realtime Global Securities Quote']['07. Close (Previous Trading Day)']")
|
|
priceChange=$(echo $stockInfo | python -c "import sys, json; print json.load(sys.stdin)['Realtime Global Securities Quote']['08. Price Change']")
|
|
priceChangePercentage=$(echo $stockInfo | python -c "import sys, json; print json.load(sys.stdin)['Realtime Global Securities Quote']['09. Price Change Percentage']")
|
|
volume=$(echo $stockInfo | python -c "import sys, json; print json.load(sys.stdin)['Realtime Global Securities Quote']['10. Volume (Current Trading Day)']")
|
|
lastUpdated=$(echo $stockInfo | python -c "import sys, json; print json.load(sys.stdin)['Realtime Global Securities Quote']['11. Last Updated']")
|
|
unset stockInfo # done with the JSON response not needed anymore
|
|
}
|
|
|
|
## This function uses all the variables that are set by getStockInformation and
|
|
## prints them out to the user in a human readable format
|
|
printStockInformation()
|
|
{
|
|
echo
|
|
echo $symbol stock info
|
|
echo "============================================="
|
|
echo "| Exchange Name: $exchangeName"
|
|
echo "| Latest Price: $latestPrice"
|
|
if [[ $open != "--" ]];then echo "| Open (Current Trading Day): $open"; fi ## sometime this is blank only print if value is present
|
|
if [[ $high != "--" ]];then echo "| High (Current Trading Day): $high"; fi ## sometime this is blank only print if value is present
|
|
if [[ $low != "--" ]];then echo "| Low (Current Trading Day): $low"; fi ## sometime this is blank only print if value is present
|
|
echo "| Close (Previous Trading Day): $close"
|
|
echo "| Price Change: $priceChange"
|
|
if [[ $priceChangePercentage != "%" ]];then echo "| Price Change Percentage: $priceChangePercentage"; fi ## sometime this is blank only print if value is present
|
|
if [[ $volume != "--" ]];then echo "| Volume (Current Trading Day): $volume"; fi ## sometime this is blank only print if value is present
|
|
echo "| Last Updated: $lastUpdated"
|
|
echo "============================================="
|
|
echo
|
|
}
|
|
|
|
## This function queries google to determine the stock ticker for a certain company
|
|
## this allows the usage of stocks to be extended where now you can enter stocks appple
|
|
## and it will determine the stock symbol for apple is AAPL and move on from there
|
|
getTicker()
|
|
{
|
|
response=$(httpGet "http://d.yimg.com/autoc.finance.yahoo.com/autoc?query=$1+$2+$3+$4+$5+$6+$7+$8+$9®ion=1&lang=en%22") > /dev/null
|
|
symbol=$(echo $response | python -c "import sys, json; print json.load(sys.stdin)['ResultSet']['Result'][0]['symbol']") # using python to extrapolate the stock symbol
|
|
unset response #just unsets the entire response after using it since all I need is the stock ticker
|
|
}
|
|
|
|
update()
|
|
{
|
|
# Author: Alexander Epstein https://github.com/alexanderepstein
|
|
# Update utility version 1.2.0
|
|
# To test the tool enter in the defualt 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" == "Y" || "$answer" == "y" ]] ;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
|
|
|
|
}
|
|
|
|
usage()
|
|
{
|
|
echo "Stocks tool"
|
|
echo "Usage: stocks [flag or company]"
|
|
echo " -u Update Bash-Snippet Tools"
|
|
echo " -h Show the help"
|
|
echo " -v Get the tool version"
|
|
}
|
|
|
|
getConfiguredClient || exit 1
|
|
checkInternet || exit 1 # check if we have a valid internet connection if this isnt true the rest of the script will not work so stop here
|
|
|
|
|
|
while getopts "uvh" opt; do
|
|
case $opt in
|
|
\?)
|
|
echo "Invalid option: -$OPTARG" >&2
|
|
exit 1
|
|
;;
|
|
h)
|
|
usage
|
|
exit 0
|
|
;;
|
|
v)
|
|
echo "Version $currentVersion"
|
|
exit 0
|
|
;;
|
|
u)
|
|
update
|
|
exit 0
|
|
;;
|
|
:)
|
|
echo "Option -$OPTARG requires an argument." >&2
|
|
exit 1
|
|
;;
|
|
esac
|
|
done
|
|
|
|
|
|
if [[ $1 == "update" ]];then
|
|
update
|
|
elif [[ $1 == "help" ]];then
|
|
usage
|
|
elif [[ $# == "0" ]];then
|
|
usage
|
|
else
|
|
getTicker $1 $2 $3 $4 $5 $6 $7 $8 $9 # the company name might have spaces so passing in all args allows for this
|
|
getStockInformation $symbol # based on the stock symbol exrapolated by the getTicker function get information on the stock
|
|
printStockInformation # print this information out to the user in a human readable format
|
|
fi
|