path: root/hib-dlagent

#!/usr/bin/env bash
set -e

VERSION=0.6
LOGIN_PAGE=https://www.humblebundle.com/login
HOME_PAGE=https://www.humblebundle.com/home
COOKIE_JAR=

SCRIPT_PATH='/usr/share/hib-dlagent'
CONFIG_PATH='/etc/hib-dlagent'

FILE=
DESTINATION=
DOWNLOAD=1
USERNAME=
PASSWORD=
KEYS=()
STORAGE=

discover_url() {
  local LISTING_PAGE="$1"
  # Running a new shell in order to split PHANTOM_JS_OPTIONS into multiple arguments
  sh -c "phantomjs --config='$CONFIG_PATH'/phantomjs-config.json --cookies-file='$COOKIE_JAR' \
         $PHANTOM_JS_OPTIONS \
         '$SCRIPT_PATH'/discover-url.coffee '$LISTING_PAGE' '$FILE' '$USERNAME' '$PASSWORD'"
}

usage() {
  cat <<EOF
hib-dlagent $VERSION
Tool to download Humble Indie Bundle binaries by file name

Usage: $0 [OPTIONS] FILE
Options:
 -d <dir>     Directory for searching and saving files. The normal save location
              is still used, but will be a symbolic link
 -h           This help
 -k <key>     Search key's files. Use multiple times for multiple keys
 -o <file>    Name to use when saving file
 -p <pass>    Use pass to login. If specified multiple times, the last is used
 -P <pname>   The name of the password available via gnome-keyring-query
 -s           Print URL to stdout instead of downloading. Incompatible with -d
 -u <user>    Use user to login. Search account's files. If specified multiple
              times, the last is used
 -S <dir>     Directory where PhantomJS scripts are located.
 -c <dir>     Directory where configuration files are located.
 -j <options> Additional options passed to PhantomJS
 -l           Turn on logging

If you specify -u, then all of that account's bundles are searched. If a key is
associated with a HIB account then you must use -u/-p, since that key only works
when logged into that account. It is not helpful to specify -k for bundles
associated with an account.
EOF
}

handle_download() {
  local SAVE_FILE="$1"
  local LISTING_PAGES=()

  COOKIE_JAR=$(mktemp)

  if [ -n "$USERNAME" ]; then
    LISTING_PAGES+=("$HOME_PAGE")
    if [ -z "$PASSWORD" ]; then
      read -rsp 'Enter Humble account password: ' PASSWORD
      echo
    fi
  fi

  for KEY in "${KEYS[@]}"; do
    LISTING_PAGES+=("https://www.humblebundle.com/downloads?key=$KEY")
  done

  if [ -z "$LISTING_PAGES" ]; then
    echo "You must specify at least one of -u and -k" >&2
    exit 1
  fi

  local URL
  for LISTING_PAGE in "${LISTING_PAGES[@]}"; do
    URL=$(discover_url "$LISTING_PAGE")
    if [ -n "$URL" ]; then break; fi
  done

  # Cookie no longer necessary.
  rm "$COOKIE_JAR"

  if [ ! -n "$URL" ]; then
    echo "Could not find URL for file: $FILE" >&2
    exit 2
  fi

  if [ $DOWNLOAD -eq 0 ]; then
    echo "$URL"
  else
    curl -C - --retry 3 --retry-delay 3 -o "$SAVE_FILE" "$URL"
  fi
}

main() {
  if [ $# -eq 0 ]; then
    usage
    exit 1
  fi

  while getopts "hd:k:o:p:P:su:S:c:j:l" opt; do
    case $opt in
      \?)
        exit 1
      ;;
      h)
        usage
        exit 1
      ;;
      d)
        STORAGE="$OPTARG"
      ;;
      k)
        KEYS+=("$OPTARG")
      ;;
      o)
        DESTINATION="$OPTARG"
      ;;
      p)
        PASSWORD="$OPTARG"
      ;;
      P)
        PASSWORD="$(gnome-keyring-query get "$OPTARG")"
      ;;
      s)
        DOWNLOAD=0
      ;;
      u)
        USERNAME="$OPTARG"
      ;;
      S)
        SCRIPT_PATH="$OPTARG"
      ;;
      c)
        CONFIG_PATH="$OPTARG"
      ;;
      j)
        PHANTOM_JS_OPTIONS="$OPTARG"
      ;;
      l)
        export LOG=1
      ;;
    esac
  done

  shift $(($OPTIND - 1))

  if [ $# == 0 ]; then
    echo "Missing argument FILE" >&2
    exit 1
  fi

  if [ $# != 1 ]; then
    echo "Unexpected argument: $2" >&2
    exit 1
  fi

  if [ $DOWNLOAD -eq 0 -a -n "$STORAGE" ]; then
    echo "-s and -d are incompatible" >&2
    exit 1
  fi

  FILE="$1"

  if [ "$FILE" != "$(basename "$FILE")" ]; then
    # We need a simple file name without any slashes. Instead of erroring, we
    # strip the file name out so that hib://filename is a valid argument.
    FILE=$(basename "$FILE")
    # Since we are doing something fancy, give the user a clue that we modified
    # their input.
    echo "Searching for $FILE" >&2
  fi

  if [ -z "$DESTINATION" ]; then
    DESTINATION="."
  fi

  if [ -d "$DESTINATION" ]; then
    DESTINATION="$DESTINATION/$FILE"
  fi

  if [ -z "$STORAGE" ]; then
    handle_download "$DESTINATION"
  else
    local STORAGE_FILE
    STORAGE_FILE=$(find "$STORAGE" -name "$FILE")
    if [ -z "$STORAGE_FILE" ]; then
      STORAGE_FILE="$STORAGE/$FILE"
      handle_download "$STORAGE_FILE"
    fi

    if [ ! "$STORAGE_FILE" -ef "$DESTINATION" ]; then
      ln -s "$STORAGE_FILE" "$DESTINATION"
      if [ ! "$STORAGE_FILE" -ef "$DESTINATION" ]; then
        # If DESTINATION is in a different directory than the CWD and STORAGE
        # is a relative path, then ln -s will produce a broken link. To produce
        # a valid link we resolve the path with readlink. This is unfortunate
        # because having relative links is nicer and we will undo any symlinks
        # in the path. Therefore, we detect when things have broken and only
        # use readlink when we must.
        STORAGE_FILE=$(readlink -f "$STORAGE_FILE")
        rm "$DESTINATION"
        ln -s "$STORAGE_FILE" "$DESTINATION"
      fi
    fi
  fi
}

main "$@"