uaws

#!/bin/sh

set -eu

arch=
case $(uname -m) in
  x86_64)
    arch=amd64
    ;;
  arm64)
    arch=arm64
    ;;
  aarch64)
    arch=arm64
    ;;
  *)
    echo "error: unsupported arch $(uname -m)" >&2
    exit 1
    ;;
esac

os=
ext=
case $(uname -s) in
  Linux)
    os=linux
    ;;
  Darwin)
    os=darwin
    ;;
  *)
    echo "error: unsupported OS $(uname -s)" >&2
    exit 1
    ;;
esac

bin_dir="${HOME}/.local/bin"
lib_dir="${HOME}/.local/lib/uawscli"
if [ $(basename $0) = "uaws" ] && [ $0 != "./uaws" ]
then
  # Resolve install path if installed.
  bin_dir=$(cd $(dirname $0) && pwd)
  lib_dir=$(cd ${bin_dir}/.. && pwd)/lib/uawscli
fi

get_download_urls() {
  version=$1

  if [ -z "${GITHUB_TOKEN:-${TRAVIS_BOT_GITHUB_TOKEN:-}}" ]
  then
    # Fallback to use download URL list on gh-pages if token is not available.
    # API access from CI services usually fails due to rate limit.
    curl -sSL --retry 4 \
      https://seqsense.github.io/uawscli/releases/${version}
    exit 0
  fi

  api_auth="-H \"Authorization: token ${GITHUB_TOKEN:-${TRAVIS_BOT_GITHUB_TOKEN}}\""

  gh_api_base=${GITHUB_API_URL_BASE:-https://api.github.com}
  slug=seqsense/uawscli

  ep=latest
  if [ ${version} != "latest" ]
  then
    ep=tags/${version}
  fi
  eval curl \
      ${api_auth} \
      -sSL --retry 4 \
      ${gh_api_base}/repos/${slug}/releases/${ep} \
    | sed -n 's/.*"browser_download_url":[ \t\r\n]*"\([^"]*\)"/\1/p'
}

uaws_install() {
  mkdir -p ${lib_dir}
  mkdir -p ${bin_dir}

  urls=$(get_download_urls $1 | grep -e "_${os}_${arch}${ext}$" || true)
  if [ -z "${urls}" ]
  then
    echo "error: failed to fetch released binary list" >&2
    exit 1
  fi

  shift
  for query in $@
  do
    found=false
    for url in ${urls}
    do
      cmd=$(echo ${url} | sed -n "s|.*/\([^/]*\)_[0-9\.]\{1,\}_${os}_${arch}${ext}$|\1|p")
      if echo ${query} | grep -q -e "^${cmd}"
      then
        file=$(echo ${url} | sed -n 's|.*/\([^/]\{1,\}\)$|\1|p')
        if [ ! -f ${lib_dir}/${file} ]
        then
          echo "info: installing ${url} ${cmd} ${file}" >&2
          curl -sSL ${url} -f -o ${lib_dir}/${file}
          chmod +x ${lib_dir}/${file}
          rm -f ${bin_dir}/${cmd}
          ln -s ${lib_dir}/${file} ${bin_dir}/${cmd}
        else
          echo "info: ${url} is up-to-date" >&2
        fi
        found=true
        break
      fi
    done
    if ! ${found}
    then
      echo "error: ${query} not found" >&2
      exit 1
    fi
  done
}

collect_installed_cmds() {
  ls -1 ${lib_dir} 2> /dev/null \
    | sed "s/\(..*\)_[0-9\.]\{1,\}_${os}_${arch}${ext}/\1/" \
    | sort \
    | uniq || true
}

installed_cmds=$(collect_installed_cmds)

case ${1:-} in
  upgrade)
    uaws_install latest ${installed_cmds}
    exit 0
    ;;
  get_download_urls)
    shift
    get_download_urls ${1:-latest}
    exit 0
    ;;
  install)
    shift

    install_targets=
    install_prefix="${HOME}/.local"
    version=latest
    while [ $# -gt 0 ]
    do
      case $1 in
        -v)
          shift
          if [ $# -lt 1 ]
          then
            echo "error: -v must be followed by a version" >&2
            exit 1
          fi
          version=$1
          ;;
        -i)
          shift
          if [ $# -lt 1 ]
          then
            echo "error: -i must be followed by a path" >&2
            exit 1
          fi
          install_prefix=$1
          ;;
        -*)
          echo "error: unknown option $1" >&2
          exit 1
          ;;
        *)
          install_targets="${install_targets} uaws-$1"
          ;;
      esac
      shift
    done

    if [ -z "${install_targets}" ]
    then
      echo "usage: $(basename $0) install [option...] install_targets" >&2
      echo "option:" >&2
      echo "  -v VERSION    install specific VERSION of uawscli" >&2
      echo "  -i DIR        install prefix DIR (default: ${install_prefix})" >&2
      exit 1
    fi

    bin_dir=${install_prefix}/bin
    lib_dir=${install_prefix}/lib/uawscli
    
    uaws_install ${version} ${install_targets}
    exit 0
    ;;
  "")
    echo "usage: $(basename $0) sub-command ..." >&2
    echo "installed sub-commands:" >&2
    echo ${installed_cmds} | xargs -n1 echo "  "
    exit 0
    ;;
esac

for i in 1 2
do
  selected_cmd=
  for cmd in ${installed_cmds}
  do
    nquery=1
    query="uaws-$@"

    while echo ${query} | grep -q -e "^\S\+ "
    do
      query=$(echo ${query} | sed "s/ /-/")
      nquery=$(expr ${nquery} + 1)
      if echo ${query} | grep -q -e "^${cmd}"
      then
        selected_cmd=${cmd}
        shift ${nquery}
        break
      fi
    done
  done

  if [ -z "${selected_cmd}" ]
  then
    echo "warn: requested command is not installed" >&2
    # not found, try to install
    uaws_install latest uaws-$(echo $@ | sed "s/ /-/g" | head -n1)
    installed_cmds=$(collect_installed_cmds)
    continue
  fi

  exec $(ls -1 ${lib_dir}/${selected_cmd}* | sort -rV | head -n1) "$@"
done