summaryrefslogtreecommitdiff
path: root/hib-dlagent
blob: e575123ffae6f73c649c1cd2d8e7987907b43e5c (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
#!/usr/bin/env bash
set -e

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

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

login() {
  /usr/bin/curl -s --cookie-jar "$COOKIE_JAR" \
    --data "username=$USERNAME" --data "password=$PASSWORD" "$LOGIN_PAGE"
}

discover_url() {
  local LISTING_PAGE="$1"
  /usr/bin/curl -s --cookie "$COOKIE_JAR" "$LISTING_PAGE" | grep "/$FILE?" | \
    sed -e "s/.* data-web='\([^']*\)'.*/\1/" | head -n 1
}

usage() {
  echo "hib-dlagent $VERSION"
  echo "Tool to download Humble Indie Bundle binaries by file name"
  echo
  echo "Usage: $0 [OPTIONS] FILE"
  echo "Options:"
  echo " -h         This help"
  echo " -k <key>   Search key's files. Use multiple times for multiple keys"
  echo " -o <file>  Name to use when saving file"
  echo " -p <pass>  Use pass to login. If specified multiple times, the last is"
  echo "            used"
  echo " -s         Print URL to stdout instead of downloading"
  echo " -u <user>  Use user to login. Search account's files. If specified"
  echo "            multiple times, the last is used"
  echo
  echo "If you specify -u, then all of that account's bundles are searched. If"
  echo "a key is associated with a HIB account then you must use -u/-p, since"
  echo "that key only works when logged into that account. It is not helpful to"
  echo "specify -k for bundles associated with an account."
}

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

  while getopts "hk:o:p:su:" opt; do
    case $opt in
      \?)
        exit 1
      ;;
      h)
        usage
        exit 1
      ;;
      k)
        KEYS+=("$OPTARG")
      ;;
      o)
        DESTINATION="$OPTARG"
      ;;
      p)
        PASSWORD="$OPTARG"
      ;;
      s)
        DOWNLOAD=0
      ;;
      u)
        USERNAME="$OPTARG"
      ;;
    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

  COOKIE_JAR=$(mktemp)
  FILE="${1#hib://}"

  if [ -z "$DESTINATION" ]; then
    DESTINATION="${FILE##*/}"
  fi

  if [ -n "$USERNAME" ]; then
    login;
    LISTING_PAGES+=("$HOME_PAGE")
  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" >&1
    exit 1
  fi

  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
    /usr/bin/curl -C - --retry 3 --retry-delay 3 -o "$DESTINATION" "$URL"
  fi
}

main "$@"