allow downloading game directly from package install

2 files changed, 114 insertions, 17 deletions

diff --git a/.SRCINFO b/.SRCINFO
index 24f1d57884d3..803e328d1e86 100644
--- a/.SRCINFO
+++ b/.SRCINFO
@@ -1,7 +1,7 @@
 pkgbase = factorio
 	pkgdesc = A 2D game about building and maintaining factories.
 	pkgver = 0.13.19
-	pkgrel = 1
+	pkgrel = 2
 	url = http://www.factorio.com/
 	arch = i686
 	arch = x86_64
diff --git a/PKGBUILD b/PKGBUILD
index f6bb957c6bf9..9dc3bde2fdb3 100644
--- a/PKGBUILD
+++ b/PKGBUILD
@@ -1,8 +1,9 @@
 # Maintainer: Sebastien Duthil <duthils@free.fr>
+# Contributor: mickael9 <mickael9 at gmail dot com>
 
 pkgname=factorio
 pkgver=0.13.19
-pkgrel=1
+pkgrel=2
 pkgdesc="A 2D game about building and maintaining factories."
 arch=('i686' 'x86_64')
 url="http://www.factorio.com/"
@@ -15,8 +16,10 @@ sha256sums=('5f62aa7763f9ad367a051371bc16f3c174022bb3380eb221ba06bac395bf9815'
             '67ec2f88afff5d7e0ca5fd3301b5d98655269c161a394368fa0ec49fbc0c0e21')
 if test "$CARCH" == i686; then
   __factorio_arch=i386
+  _url=https://www.factorio.com/get-download/${pkgver}/alpha/linux32
 elif test "$CARCH" == x86_64; then
   __factorio_arch=x64
+  _url=https://www.factorio.com/get-download/${pkgver}/alpha/linux64
 fi
 _gamepkg=factorio_alpha_${__factorio_arch}_$pkgver.tar.gz
 _pkgpaths_tries=("$startdir"
@@ -26,23 +29,26 @@ _pkgpaths_tries=("$startdir"
 build() {
   msg "You need a full copy of this game in order to install it"
 
-  # look for game tarball
-  for pkgpath_try in "${_pkgpaths_tries[@]}" ; do
-    msg "Searching for ${_gamepkg} in dir: \"${pkgpath_try}\""
-    if [[ -f "${pkgpath_try}/${_gamepkg}" ]]; then
-      pkgpath=${pkgpath_try}
-      break
-    fi
-  done
+  _find_pkgpath_from_dir
 
-  # not found: ask for path to game tarball
+  # could not find the path automatically, ask the user
   if [[ ! -f "${pkgpath}/${_gamepkg}" ]]; then
-    error "Game package not found, please type absolute path to ${_gamepkg} (/home/joe):"
-    read pkgpath
-    if [[ ! -f "${pkgpath}/${_gamepkg}" ]]; then
-        error "Unable to find game package."
-        return 1
-    fi
+      _find_pkgpath_from_input
+  fi
+
+  # if user entered nothing, try to download
+  if [[ ! -f "${pkgpath}/${_gamepkg}" ]]; then
+      _find_pkgpath_from_download
+  fi
+
+  # if download failed, ask the user again
+  if [[ ! -f "${pkgpath}/${_gamepkg}" ]]; then
+      _find_pkgpath_from_input
+  fi
+
+  if [[ ! -f "${pkgpath}/${_gamepkg}" ]]; then
+      error "Unable to find game package."
+      return 1
   fi
 
   # unpack game tarball
@@ -67,3 +73,94 @@ package() {
   install -m644 "${srcdir}/factorio.desktop" "${pkgdir}/usr/share/applications/factorio.desktop"
   install -m644 "${srcdir}/LICENSE" "${pkgdir}/usr/share/licenses/factorio/LICENSE"
 }
+
+_find_pkgpath_from_dir() {
+  for pkgpath_try in "${_pkgpaths_tries[@]}" ; do
+    msg "Searching for ${_gamepkg} in dir: \"${pkgpath_try}\""
+    if [[ -f "${pkgpath_try}/${_gamepkg}" ]]; then
+      pkgpath=${pkgpath_try}
+      break
+    fi
+  done
+  if [[ -z "${pkgpath}" ]] ; then
+    error "Game package not found."
+  fi
+}
+
+_find_pkgpath_from_input() {
+    read -p "Please provide the path to the directory containing ${_gamepkg} (e.g. /home/joe). Leave blank to download from https://factorio.org: " pkgpath
+}
+
+_find_pkgpath_from_download() {
+  local cookie=$(mktemp)
+
+  while [[ -z "${pkgpath}" ]] ; do
+    local login
+    local password
+    local file="${SRCDEST}/${_gamepkg}"
+
+    read -p "Username or email: " login
+    if [[ -z "${login}" ]]; then
+      break
+    fi
+    read -sp "Password: " password ; echo
+    if [[ -z "${password}" ]]; then
+      break
+    fi
+
+    msg "Logging in..."
+    local csrf_token=$(
+      curl --silent --fail \
+           --cookie-jar "$cookie" \
+           https://www.factorio.com/login \
+      | grep -Po '(?<=name="csrf_token" type="hidden" value=")[^"]+'
+    )
+
+    if [[ -z "$csrf_token" ]]; then
+      error "Could not find the CSRF token. This script might be broken."
+      break
+    fi
+
+    local output=$(
+      curl --dump-header - \
+           --silent --fail \
+           --cookie-jar "$cookie" \
+           --cookie "$cookie" \
+           https://www.factorio.com/login \
+           --data-urlencode username_or_email="$login" \
+           --data-urlencode password="$password" \
+           --data-urlencode csrf_token="$csrf_token" \
+    )
+
+    if ! echo "$output" | grep -q '^Location: '; then
+      error "Login failed"
+      read -p "Try again? (Y/n) " try_again
+      if [[ "${try_again}" == "n" ]] ; then
+        break
+      else
+        continue
+      fi
+    fi
+
+    msg2 "Logged in"
+    msg "Downloading ${_gamepkg} from $_url ..."
+
+    curl --fail --location \
+         --cookie "${cookie}" \
+         --continue-at - \
+         --output "${file}.part" \
+         "${_url}" \
+    || rm -f "${file}.part"
+
+    if [[ ! -f "${file}.part" ]]; then
+      error "Download failed"
+      break
+    fi
+
+    msg2 "Download successful"
+    mv "${file}"{.part,}
+    pkgpath="${SRCDEST}"
+  done
+
+  rm $cookie
+}