#!/bin/bash

usage() {
  echo "$0 all <version name>"
  echo -e "\tBuild binaries, zip, and sign Magisk"
  echo -e "\tThis is equlivant to first <build>, then <zip>"
  echo "$0 clean"
  echo -e "\tCleanup compiled / generated files"
  echo "$0 build"
  echo -e "\tBuild the binaries with ndk"
  echo "$0 zip <version name>"
  echo -e "\tZip and sign Magisk"
  echo "$0 uninstaller"
  echo -e "\tZip and sign the uninstaller"
  exit 1
}

cleanup() {
  echo "************************"
  echo "* Cleaning up"
  echo "************************"
  ndk-build clean 2>/dev/null
  ls zip_static/arm/* | grep -v "busybox" | xargs rm -rfv
  ls zip_static/arm64/* | grep -v "busybox" | xargs rm -rfv
  ls zip_static/x86/* | grep -v "busybox" | xargs rm -rfv
  ls zip_static/x64/* | grep -v "busybox" | xargs rm -rfv
  rm -rfv zip_static/META-INF/com/google/android/update-binary
  rm -rfv zip_static/common/custom_ramdisk_patch.sh
  rm -rfv zip_static/common/magisksu.sh
  rm -rfv zip_static/common/init.magisk.rc
  rm -rfv zip_static/common/magic_mask.sh
  rm -rfv uninstaller/common
  rm -rfv uninstaller/arm
  rm -rfv uninstaller/arm64
  rm -rfv uninstaller/x86
  rm -rfv uninstaller/x64
}

mkcp() {
  [ ! -d "$2" ] && mkdir -p "$2"
  cp -afv $1 $2
}

error() {
  echo -e "\n! $1\n"
  exit 1
}

build_bin() {
  echo "************************"
  echo "* Building binaries"
  echo "************************"
  [ -z `which ndk-build` ] && error "Please add ndk-build to PATH!"
  ndk-build -j4 || error "Magisk binary tools build failed...."
  echo "************************"
  echo "* Copying binaries"
  echo "************************"
  mkcp "libs/armeabi-v7a/*" zip_static/arm
  mkcp libs/armeabi-v7a/bootimgtools uninstaller/arm
  mkcp "libs/arm64-v8a/*" zip_static/arm64
  mkcp libs/arm64-v8a/bootimgtools uninstaller/arm64
  mkcp "libs/x86/*" zip_static/x86
  mkcp libs/x86/bootimgtools uninstaller/x86
  mkcp "libs/x86_64/*" zip_static/x64
  mkcp libs/x86_64/bootimgtools uninstaller/x64
}

zip_package() {
  [ ! -f "zip_static/arm/bootimgtools" ] && error "Missing binaries!! Please run '$0 build' before zipping"
  echo "************************"
  echo "* Adding version info"
  echo "************************"
  sed "s/MAGISK_VERSION_STUB/Magisk v$1 Boot Image Patcher/g" scripts/flash_script.sh > zip_static/META-INF/com/google/android/update-binary
  sed "s/MAGISK_VERSION_STUB/setprop magisk.version \"$1\"/g" scripts/magic_mask.sh > zip_static/common/magic_mask.sh
  echo "************************"
  echo "* Zipping Magisk v$1"
  echo "************************"
  cp -afv scripts/custom_ramdisk_patch.sh zip_static/common/custom_ramdisk_patch.sh
  cp -afv scripts/magisksu.sh zip_static/common/magisksu.sh
  cp -afv scripts/init.magisk.rc zip_static/common/init.magisk.rc
  cd zip_static
  find . -type f -exec chmod 644 {} \;
  find . -type d -exec chmod 755 {} \;
  rm -rf "../Magisk-v$1.zip"
  zip "../Magisk-v$1.zip" -r .
  cd ../
  sign_zip "Magisk-v$1.zip"
}

zip_uninstaller() {
  [ ! -f "uninstaller/arm/bootimgtools" ] && error "Missing binaries!! Please run '$0 build' before zipping"
  echo "************************"
  echo "* Zipping uninstaller"
  echo "************************"
  mkcp scripts/magisk_uninstaller.sh uninstaller/common
  cd uninstaller
  find . -type f -exec chmod 644 {} \;
  find . -type d -exec chmod 755 {} \;
  TIMESTAMP=`date "+%Y%m%d"`
  rm -rf "../Magisk-uninstaller-$TIMESTAMP.zip"
  zip "../Magisk-uninstaller-$TIMESTAMP.zip" -r .
  cd ../
  sign_zip "Magisk-uninstaller-$TIMESTAMP.zip"
}

sign_zip() {
  if [ ! -f "ziptools/zipadjust" ]; then
    echo "************************"
    echo "* Compiling ZipAdjust"
    echo "************************"
    gcc -o ziptools/zipadjust ziptools/src/*.c -lz || error "ZipAdjust Build failed...."
    chmod 755 ziptools/zipadjust
  fi
  echo "************************"
  echo "* First sign $1"
  echo "************************"
  java -jar "ziptools/signapk.jar" "ziptools/test.certificate.x509.pem" "ziptools/test.key.pk8" "$1" "${1%.*}-firstsign.zip"
  echo "************************"
  echo "* Adjusting $1"
  echo "************************"
  ziptools/zipadjust "${1%.*}-firstsign.zip" "${1%.*}-adjusted.zip"
  echo "************************"
  echo "* Final sign $1"
  echo "************************"
  java -jar "ziptools/minsignapk.jar" "ziptools/test.certificate.x509.pem" "ziptools/test.key.pk8" "${1%.*}-adjusted.zip" "${1%.*}-signed.zip"

  mv "${1%.*}-signed.zip" "$1"
  rm "${1%.*}-adjusted.zip" "${1%.*}-firstsign.zip"
}

DIR="$(cd "$(dirname "$0")"; pwd)"
cd "$DIR"

case $1 in
  "all" )
    [ -z "$2" ] && echo -e "! Missing version number\n" && usage
    build_bin
    zip_package $2
    ;;
  "clean" )
    cleanup
    ;;
  "build" )
    build_bin
    ;;
  "zip" )
    [ -z "$2" ] && echo -e "! Missing version number\n" && usage
    zip_package $2
    ;;
  "uninstaller" )
    zip_uninstaller
    ;;
  * )
    usage
    ;;
esac