#! /bin/bash
set -e

# Check arguments
if [[ $# != 2 ]]; then
    >&2 echo "Bad usage!"
    echo "Usage: $0 <build dir> <output file>"
    exit 1
fi

# Get paths
ARCH="$(uname -m)"
SYSTEM_LIBS="/usr/lib"
SYSTEM_LIBS64="/usr/lib64"
YUZU_BIN="${1}/bin"
YUZU_BIN_GUI="${YUZU_BIN}/yuzu"

# Make sure executable exists
if [[ $(file -b --mime-type "$YUZU_BIN_GUI") != application/x-pie-executable ]]; then
    >&2 echo "Invalid yuzu executable!"
fi

# Clean up build dir
rm -rf build
mkdir build

# NOTE: some of these aren't used now, but can be reordered in priority when torzu is converted to QT6
# QT5 - /usr/lib/${ARCH}-linux-gnu/qt5 (debian), /usr/lib64/qt5 (fedora), /usr/lib/qt (steam deck)
# QT5 - /usr/lib/${ARCH}-linux-gnu/qt6 (debian), /usr/lib64/qt6 (fedora), /usr/lib/qt6 (steam deck)
QTFOUND="true"
QTDIR="$SYSTEM_LIBS"/${ARCH}-linux-gnu/qt5/plugins
if [ ! -d "$QTDIR" ]; then
  # default qt5 folder not found, check for 64-bit qt5 folder
  QTDIR="$SYSTEM_LIBS64"/qt5/plugins
  if [ ! -d "$QTDIR" ]; then
    # 64-bit qt5 folder not found, check for Steam Deck qt (qt5) folder
    QTDIR="$SYSTEM_LIBS"/qt/plugins
    if [ ! -d "$QTDIR" ]; then
      # Steam Deck qt (qt5) folder not found, check for regular qt6 folder
      QTDIR="$SYSTEM_LIBS"/${ARCH}-linux-gnu/qt6/plugins
      if [ ! -d "$QTDIR" ]; then
        # regular qt6 folder not found, check for 64-bit qt6 folder
        QTDIR="$SYSTEM_LIBS64"/qt6/plugins
        if [ ! -d "$QTDIR" ]; then
          # 64-bit qt6 folder not found, check for Steam Deck qt6 folder
          QTDIR="$SYSTEM_LIBS"/qt6/plugins
          if [ ! -d "$QTDIR" ]; then
            QTFOUND="false"
          fi
        fi
      fi
    fi
  fi
fi
if [ $QTFOUND == "true" ]; then
  echo "QT plugins from $QTDIR will be used."

  # Copy system dependencies used to build and required by the yuzu binary
  # includes:
  #   - '/lib64/ld-linux-x86-64.so.2' or `/lib/ld-linux-aarch64.so.1` file per architecture
  #   - required files from `/usr/lib/x86_64-linux-gnu` or `/usr/lib/aarch64-linux-gnu`
  #   - different for SteamDeck, but still does it automatically
  function copy_libs {
    for lib in $(ldd "$1"); do
      (cp -vn "$lib" ./build/ 2> /dev/null) || true
    done
  }
  echo "Copying main dependencies..."
  copy_libs "$YUZU_BIN_GUI"

  # Copy QT dependency folders, path determined above
  echo "Copying Qt dependencies..."
  mkdir ./build/qt5
  cp -rv "$QTDIR"/{imageformats,platforms,platformthemes,xcbglintegrations} ./build/qt5/

  # Discover indirect dependencies (mostly from runtime-loaded Qt plugins)
  echo "Copying extra dependencies..."
  while true; do
    LIBS="$(find ./build -name \*.so\*)"
    LIB_COUNT=$(echo "$LIBS" | wc -l)
    echo "$LIB_COUNT dependency libraries discovered so far..."
    if [ $LIB_COUNT == "$PREV_LIB_COUNT" ]; then
      break
    fi
    PREV_LIB_COUNT=$LIB_COUNT

    for plib in $LIBS; do
      if [ -f "$plib" ]; then
        copy_libs "$plib"
      fi
    done
  done

  # Copy executable
  cp -v "$YUZU_BIN_GUI" ./build/

  # Copy assets for the appropriate arch
  cp -v ./assets_"${ARCH}"/* ./build/
  # Copy common assets
  cp -v ./assets/* ./build/

  # Strip all libraries and executables
  for file in $(find ./build -type f); do
      (strip -v "$file" 2> /dev/null) || true
  done

  PASSED_CHECKSUM="false"
  FILE=appimagetool.AppImage
  # total number of times to try downloading if a checksum doesn't match
  DL_TRIES=3
  while [ $PASSED_CHECKSUM == "false" ] && [ "$DL_TRIES" -gt 0 ]; do
    case $ARCH in
      x86_64)
        # Static copy from the 'ext-linux-bin' repo.
        # Checksum will need to be changed when/if this file in the repo is updated.
        if ! test -f "$FILE"; then
          echo "Downloading appimagetool for architecture '$ARCH'"
          wget -O appimagetool.AppImage https://github.com/litucks/ext-linux-bin/raw/refs/heads/main/appimage/appimagetool-x86_64.AppImage
        fi
        if [ $(shasum -a 256 appimagetool.AppImage | cut -d' ' -f1) = "110751478abece165a18460acbd7fd1398701f74a9405ad8ac053427d937bd5d" ] ; then
          PASSED_CHECKSUM="true"
        fi
        # DISABLED TO USE THE ABOVE
        # The current continuous release channel option, until a static copy is put in 'ext-linux-bin'.
        # The checksum will pass until the continuous release is updated, then a new one needs to be
        # generated to update this script.
        #if ! test -f "$FILE"; then
        #  echo "Downloading appimagetool for architecture '$ARCH'"
        #  wget -O appimagetool.AppImage https://github.com/AppImage/appimagetool/releases/download/continuous/appimagetool-x86_64.AppImage
        #fi
        #if [ $(shasum -a 256 appimagetool.AppImage | cut -d' ' -f1) = "46fdd785094c7f6e545b61afcfb0f3d98d8eab243f644b4b17698c01d06083d1" ] ; then
        #  PASSED_CHECKSUM="true"
        #fi
        ;;
      aarch64)
        # Currently set to the continuous release channel until a static copy is put in 'ext-linux-bin'.
        # The checksum will pass until the continuous release is updated, then a new one needs to be
        # generated to update this script.
        if ! test -f "$FILE"; then
          echo "Downloading appimagetool for architecture '$ARCH'"
          wget -O appimagetool.AppImage https://github.com/AppImage/appimagetool/releases/download/continuous/appimagetool-aarch64.AppImage
        fi
        if [ $(shasum -a 256 appimagetool.AppImage | cut -d' ' -f1) = "04f45ea45b5aa07bb2b071aed9dbf7a5185d3953b11b47358c1311f11ea94a96" ] ; then
          PASSED_CHECKSUM="true"
        fi
        ;;
      *)
        PASSED_CHECKSUM="invalid_arch"
        ;;
    esac
    # delete the appimagetool downloaded if the checksum doesn't match.
    if [ ! $PASSED_CHECKSUM == "true" ]; then
      rm -f appimagetool.AppImage
    fi
    ((DL_TRIES-=1))
  done
  if [ $PASSED_CHECKSUM == "true" ]; then
    echo "Checksum passed. Proceeding to build image."
    # Build AppImage
    chmod a+x appimagetool.AppImage
    ./appimagetool.AppImage ./build "$2"
  elif [ $PASSED_CHECKSUM == "invalid_arch" ]; then
    echo "No download found for architecture '$ARCH'. Building halted."
  else
    echo "Checksum for appimagetool does not match. Building halted."
    echo "If the file to be downloaded has been changed, a new checksum will need to be generated for this script."
  fi
else
  echo "QT not found, aborting AppImage build."
fi
