#!/bin/bash
# Author: Alexander Epstein https://github.com/alexanderepstein

setglobal currentVersion = '"1.11.1'"
setglobal configuredClient = ''""
setglobal configuredPython = ''""

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

}

proc getConfiguredPython
{
  if  command -v python2 &>/dev/null  {
    setglobal configuredPython = '"python2'"
  } elif command -v python &>/dev/null  {
    setglobal configuredPython = '"python'"
  } else {
    echo "Error: This tool requires python 2 to be installed."
    return 1
  }
}


proc python
{
  match $configuredPython {
    with python2 python2 @Argv
    with python python @Argv
  }
}

## Allows to call the users configured client without if statements everywhere
proc httpGet
{
  match $configuredClient {
    with curl curl -A curl -s @Argv
    with wget wget -qO- @Argv
    with fetch fetch -o "..."
  }
}

proc 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 test $Status -eq 0 { #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
  }
}

## This function grabs information about a stock and using python parses the
## JSON response to extrapolate the information for storage
proc getStockInformation
{
  setglobal 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 || do { 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
  setglobal exchangeName = $[echo $stockInfo | python -c "import sys, json; print json.load(sys.stdin)['Realtime Global Securities Quote']['02. Exchange Name']]
  setglobal latestPrice = $[echo $stockInfo | python -c "import sys, json; print json.load(sys.stdin)['Realtime Global Securities Quote']['03. Latest Price']]
  setglobal open = $[echo $stockInfo | python -c "import sys, json; print json.load(sys.stdin)['Realtime Global Securities Quote']['04. Open (Current Trading Day)']]
  setglobal high = $[echo $stockInfo | python -c "import sys, json; print json.load(sys.stdin)['Realtime Global Securities Quote']['05. High (Current Trading Day)']]
  setglobal low = $[echo $stockInfo | python -c "import sys, json; print json.load(sys.stdin)['Realtime Global Securities Quote']['06. Low (Current Trading Day)']]
  setglobal close = $[echo $stockInfo | python -c "import sys, json; print json.load(sys.stdin)['Realtime Global Securities Quote']['07. Close (Previous Trading Day)']]
  setglobal priceChange = $[echo $stockInfo | python -c "import sys, json; print json.load(sys.stdin)['Realtime Global Securities Quote']['08. Price Change']]
  setglobal priceChangePercentage = $[echo $stockInfo | python -c "import sys, json; print json.load(sys.stdin)['Realtime Global Securities Quote']['09. Price Change Percentage']]
  setglobal volume = $[echo $stockInfo | python -c "import sys, json; print json.load(sys.stdin)['Realtime Global Securities Quote']['10. Volume (Current Trading Day)']]
  setglobal 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
proc printStockInformation
{
  echo
  echo $symbol stock info
  echo "============================================="
  echo "| Exchange Name: $exchangeName"
  echo "| Latest Price: $latestPrice"
  if [[ $open != "--" ]]{ echo "| Open (Current Trading Day): $open"; } ## sometime this is blank only print if value is present
  if [[ $high != "--" ]]{ echo "| High (Current Trading Day): $high"; } ## sometime this is blank only print if value is present
  if [[ $low != "--" ]]{ echo "| Low (Current Trading Day): $low"; } ## sometime this is blank only print if value is present
  echo "| Close (Previous Trading Day): $close"
  echo "| Price Change: $priceChange"
  if [[ $priceChangePercentage != "%" ]]{ echo "| Price Change Percentage: $priceChangePercentage"; } ## sometime this is blank only print if value is present
  if [[ $volume != "--" ]]{ echo "| Volume (Current Trading Day): $volume"; } ## 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
proc getTicker
{
  setglobal response = $[httpGet "http://d.yimg.com/autoc.finance.yahoo.com/autoc?query=$1+$2+$3+$4+$5+$6+$7+$8+$9&region=1&lang=en%22] > /dev/null
  setglobal 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
}

proc 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
  setglobal repositoryName = '"Bash-Snippets'" #Name of repostiory to be updated ex. Sandman-Lite
  setglobal githubUserName = '"alexanderepstein'" #username that hosts the repostiory ex. alexanderepstein
  setglobal nameOfInstallFile = '"install.sh'" # change this if the installer file has a different name be sure to include file extension if there is one
  setglobal 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 == "" ]]{
    echo "Error: update utility has not been configured correctly." > !2
    exit 1
  } elif [[ $latestVersion == "" ]]{
    echo "Error: no active internet connection" > !2
    exit 1
  } else {
    if [[ "$latestVersion" != "$currentVersion" ]] {
      echo "Version $latestVersion available"
      echo -n "Do you wish to update $repositoryName [Y/n]: "
      read -r answer
      if [[ "$answer" == "Y" || "$answer" == "y" ]] {
        cd  ~ || do { echo 'Update Failed' ; exit 1 ; }
        if [[ -d  ~/$repositoryName ]] { rm -r -f $repositoryName  ||  do { echo "Permissions Error: try running the update as sudo"; exit 1; } ; }
        git clone "https://github.com/$githubUserName/$repositoryName" || do { echo "Couldn't download latest version" ; exit 1; }
        cd $repositoryName ||  do { 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 ||  do { echo "Permissions Error: update succesfull but cannot delete temp files located at ~/$repositoryName delete this directory with sudo"; exit 1; }
      } else {
        exit 1
      }
    } else {
      echo "$repositoryName is already the latest version"
    }
  }

}

proc usage
{
  echo "Stocks"
  echo "Description: Finds the latest information on a certain stock."
  echo "Usage: stocks [flag] or stocks [company/ticker]"
  echo "  -u Update Bash-Snippet Tools"
  echo "  -h Show the help"
  echo "  -v Get the tool version"
  echo "Examples:"
  echo "  stocks AAPL"
  echo "  stocks Tesla"
}

getConfiguredPython || exit 1
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 {
  match $opt {
    with \?
      echo "Invalid option: -$OPTARG" > !2
      exit 1
    
    with h
      usage
      exit 0
    
    with v
      echo "Version $currentVersion"
      exit 0
    
    with u
      update
      exit 0
    
    with :
      echo "Option -$OPTARG requires an argument." > !2
      exit 1
    
  }
}


if [[ $1 == "update" ]]{
  update
  exit 0
} elif [[ $1 == "help" ]]{
  usage
  exit 0
} elif [[ $# == "0" ]]{
  usage
  exit 0
} else {
  getTicker @Argv # 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
  exit 0
}