#!/bin/bash
# TODO: If $stop similar to "Bus Station" then reject and suggest user tries supplying a route number to see when their next bus leaves the station
read stop

# function to provide data to a for loop
_jq() {
  echo ${json} | base64 --decode | jq -r ${1}
}

_liveresult() {
  livetimes=$(curl --fail --silent https://admin.libertybus.je/api/v1/soj/stop_updates/${stop})
  exitcode=$?
  # sort livetimes by service_number and eta
  livetimes=$(echo ${livetimes} | jq 'sort_by((.service_number | sub("[^0-9]$"; "") | tonumber), (.service_number | sub("[^A-Z]$"; "") | tostring), .eta)')
  livetimeslen=$(echo ${livetimes} | jq '. | length')
  if [ $exitcode != 0 ]; then
    echo "cURL error - exiting."
    exit 1
  elif [ $livetimeslen = 0 ]; then
    echo "No results found."
    exit 1
  else
    for json in $(echo "${livetimes}" | jq -r '.[] | @base64'); do
      if [[ $(_jq '.service_number') != $outroute ]]; then
        result=$(echo $result | sed 's/,$//')
        result="$result | Dest: $(_jq '.destination')[$(_jq '.service_number')] - ETAs: $(_jq '.eta'), "
      else
        result="$result $(_jq '.eta'), "
      fi
      outroute=$(_jq '.service_number')
    done
    result=$(echo $result | sed 's/,$//' | sed 's/|//' )
    echo -e $result
    exit 0
  fi
}

_stopsreturn() {
  stops=$(curl --fail --silent https://admin.libertybus.je/api/v1/stops/"${stop}")
  exitcode=$?
  stopslen=$(echo ${stops} | jq '. | length')
  # if length of $stops array is 1 then only one bus stop matches and we can return live times for that stop
  if [ $exitcode != 0 ]; then
    echo "cURL error - exiting."
    exit 1
  elif [ $stopslen = 1 ]; then
    stop=$(echo ${stops} | jq '.[].stop_id')
    _liveresult
  elif [ $stopslen = 0 ]; then
    echo "No results found."
    exit 1
  # multiple stops returned in $stops so we will return stop names and codes so user can retry
  else
    for json in $(echo "${stops}" | jq -r '.[] | @base64'); do
      result="$result Stop: $(_jq '.name') - Code: $(_jq '.stop_id') |"
      #result="$result Code: $(_jq '.stop_id') |"
    done
    result=$(echo $result | sed 's/ |$//')
    echo -e $result
    exit 0
  fi
}

if [ "$stop" == "-h" ] || [ "$stop" == "--help" ]; then
  echo "Usage: '!bus <stop>' where <stop> is either the 4 digit stop code, the full unique or partial bus stop name. If a bus stop name is used and there are multiple matches then a list of matching stops will be returned with their codes."
  exit 0
elif [[ "$stop" =~ ^-?[0-9]+$ ]] && [ $(echo ${#stop} == 4) ]; then
  _liveresult
elif [[ -z "${stop// }" ]]; then
  echo "Input empty. Usage: '!bus <stop>' where <stop> is either the 4 digit stop code, the full unique or partial bus stop name. If a bus stop name is used and there are multiple matches then a list of matching stops will be returned with their codes."
  exit 1
else
  _stopsreturn
fi