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

setglobal currentVersion = '"1.11.1'"
setglobal configuredClient = ''""
setglobal configuredPython = ''""
source ~/.bash_profile ## allows grabbing enviornment variable
setglobal apiKey = $TASTE_API_KEY
setglobal info = '"0'" ## indicates if we want extra info
setglobal search = '"0'" ## indivates that we want results on the item itself

## 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
  }
}

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"
    }
  }

}

## This function gets 3 results similar to the item of interest
proc getSimilar
{
  export PYTHONIOENCODING=utf8 #necessary for python in some cases
  setglobal media = $[ echo @Argv | tr " " +]
  setglobal response = $[httpGet "https://tastedive.com/api/similar?q=$media&k=$apiKey&info=$info]
  ## Extrapolate the information by parsing the JSON
  setglobal nameOne = $[echo $response | python -c "import sys, json; print json.load(sys.stdin)['Similar']['Results'][0]['Name']" !2 >  /dev/null || do { echo "Error: Did you search a valid item?]
  setglobal typeOne = $[echo $response | python -c "import sys, json; print json.load(sys.stdin)['Similar']['Results'][0]['Type']" !2 >  /dev/null]
  setglobal nameTwo = $[echo $response | python -c "import sys, json; print json.load(sys.stdin)['Similar']['Results'][1]['Name']" !2 >  /dev/null]
  setglobal typeTwo = $[echo $response | python -c "import sys, json; print json.load(sys.stdin)['Similar']['Results'][1]['Type']" !2 >  /dev/null]
  setglobal nameThree = $[echo $response | python -c "import sys, json; print json.load(sys.stdin)['Similar']['Results'][2]['Name']" !2 >  /dev/null]
  setglobal typeThree = $[echo $response | python -c "import sys, json; print json.load(sys.stdin)['Similar']['Results'][2]['Type']" !2 >  /dev/null]
  if [[ $info == "1" ]]{ ## if we want more detailed info we have to grab a few more fields
    setglobal wikiOne = $[echo $response | python -c "import sys, json; print json.load(sys.stdin)['Similar']['Results'][0]['wTeaser']" !2 >  /dev/null]
    setglobal wikiTwo = $[echo $response | python -c "import sys, json; print json.load(sys.stdin)['Similar']['Results'][1]['wTeaser']" !2 >  /dev/null]
    setglobal wikiThree = $[echo $response | python -c "import sys, json; print json.load(sys.stdin)['Similar']['Results'][2]['wTeaser']" !2 >  /dev/null]
    setglobal youtube = $[echo $response | python -c "import sys, json; print json.load(sys.stdin)['Similar']['Results'][0]['yUrl']" !2 >  /dev/null]
  }
}

## This function grabs all the information it can on the item of interest itself
proc getInfo
{
  export PYTHONIOENCODING=utf8 #necessary for python in some cases
  setglobal media = $[ echo @Argv | tr " " +]
  setglobal response = $[httpGet "https://tastedive.com/api/similar?q=$media&k=$apiKey&info=$info]
  setglobal name = $[echo $response | python -c "import sys, json; print json.load(sys.stdin)['Similar']['Info'][0]['Name']]
  setglobal type = $[echo $response | python -c "import sys, json; print json.load(sys.stdin)['Similar']['Info'][0]['Type']]
  if [[ $info == "1" ]] {
    setglobal wiki = $[echo $response | python -c "import sys, json; print json.load(sys.stdin)['Similar']['Info'][0]['wTeaser']]
    setglobal youtube = $[echo $response | python -c "import sys, json; print json.load(sys.stdin)['Similar']['Info'][0]['yUrl']]
  } else {
    setglobal wiki = '"None'"
    setglobal youtube = '"None'"
  }
}

proc printResults
{
  if [[ $info == "1" ]]{
    echo "==================================="
    echo
    echo $nameOne: $typeOne
    echo $wikiOne
    echo
    echo
    echo $nameTwo: $typeTwo
    echo $wikiTwo
    echo
    echo
    echo $nameThree: $typeThree
    echo $wikiThree
    echo
    if [[ $youtube != "None" ]]{ echo $youtube;}
    echo
    echo "==================================="
  } else {
    echo "==================================="
    echo $nameOne: $typeOne
    echo $nameTwo: $typeTwo
    echo $nameThree: $typeThree
    echo "==================================="
  }
}

proc printInfo
{
  echo "==================================="
  echo
  echo $name: $type
  echo $wiki
  echo
  if [[ $youtube != "None" ]]{ echo $youtube;}
  echo "==================================="
}


proc usage
{
  echo "Taste"
  echo "Description: A recommendation engine that provides 3 similar items based on some input topic."
  echo "  Taste also has the ability to provide information on the item of interest."
  echo "  Supports: shows, books, music, artists, movies, authors, games"
  echo "Usage: taste [flag] [item]"
  echo "  -i Get more information on similar items"
  echo "  -s Get information on the item itself"
  echo "  -u Update Bash-Snippet Tools"
  echo "  -h Show the help"
  echo "  -v Get the tool version"
  echo "Examples:"
  echo "  taste -i Kendrick Lamar"
  echo "  taste Catcher in the Rye"
  echo "  taste -s Red Hot Chili Peppers"
}

if [[ $apiKey == "" ]]{
  echo "Error: API key not setup properly"
  echo "To get an API key visit https://tastedive.com/account/api_access"
  echo 'After getting the API key run the following command: export TASTE_API_KEY="yourAPIKeyGoesHere"'
  echo "After following all the steps and issues still persist try adding export TASTE_API_KEY manually to your .bash_profile"
  exit 1
}
getConfiguredPython || exit 1
getConfiguredClient || exit 1
checkInternet || exit 1


while getopts "uvhis" 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 i
      if [[ $search == "0" ]]{
        setglobal info = '"1'"
      } else {
        echo "Error: the options -i and -s are mutually exclusive (-s already uses -i)"
        exit 1
      }
    
    with s
      if [[ $info != "1" ]]{
        setglobal search = '"1'"
        setglobal info = '"1'"
      } else {
        echo "Error: the options -i and -s are mutually exclusive (-s already uses -i)"
        exit 1
      }
    
    with :
      echo "Option -$OPTARG requires an argument." > !2
      exit 1
    
  }
}

if [[ $# == 0 ]] {
  usage
} elif [[ $1 == "update" ]]{
  update
} elif [[ $1 == "help" ]]{
  usage
} else {
  if [[ $search == "0" ]]{
    if [[ $info == "0" ]]{
      getSimilar @Argv || exit 1 ## exit if we return 1 (chances are movie was not found)
      printResults
    } else {
      getSimilar $(@:2) || exit 1
      printResults
    }
  } else {
    getInfo $(@:2) || exit 1 ## exit if we return 1 (chances are movie was not found)
    printInfo
  }
}