#!/bin/bash

shopt -s extglob

SCR_NAME_EXEC=$0
SCR_NAME_EXEC_FP=$(realpath "$0")
SCR_NAME=$(basename "$SCR_NAME_EXEC")
SCR_NAME=${SCR_NAME%.*}
RVB_DIR=$HOME/revanced-builder

COLOR_OFF='\033[0m'
COLOR_RED='\033[1;31m'

help_info() {
  cat <<EOF
Usage: $SCR_NAME [command] [options]

Commands:
  run                          Launches the revanced-builder.
                               Running $SCR_NAME_EXEC without arguments will
                               assume this command (i.e. will run the
                               builder)
    --delete-cache
    --dc                       Deletes revanced/ before running builder
    --delete-cache-no-keystore
    --dcnk                     Deletes revanced/ before running builder, but
                               preserving keystore file.
    --delete-cache-after
    --dca                      Deletes revanced/ after running builder
    --delete-cache-after-no-keystore
    --dcank                    Deletes revanced/ after running builder, but
                               preserving keystore file

  reinstall                    Delete everything and start from scratch.
    --delete-keystore          Also delete the signature file. This will
                               make ReVanced use a different signature,
                               which will not allow you to install an
                               updated build over the previously installed
                               one (you'll need to uninstall that first)

  update                       Update the builder to the latest version

  help                         Display this help info
EOF
}

log() {
  echo -e "[$SCR_NAME] $1"
}

error() {
  log "$1"
  [[ "$2" == y ]] && help_info
  exit "${3:-1}"
}

dload_and_install() {
  log "Downloading revanced-builder..."
  curl -sLo revanced-builder.zip https://github.com/reisxd/revanced-builder/archive/refs/heads/main.zip
  log "Unzipping..."
  unzip -qqo revanced-builder.zip
  rm revanced-builder.zip
  mv revanced-builder-main/{.[!.]*,*} .
  log "Installing packages..."
  npm install --omit=dev
  rmdir revanced-builder-main
  [[ -z "$1" ]] && log "Done. Execute \`$SCR_NAME_EXEC run\` to launch the builder."
}

preflight() {
  setup_storage() {
    [[ ! -d "$HOME"/storage ]] && {
      log "You will now get a permission dialog to allow access to storage."
      log "This is needed in order to move the built APK (+ MicroG) to internal storage."
      sleep 5
      termux-setup-storage
    } || {
      log "Already gotten storage access."
    }
  }

  install_dependencies() {
    local JAVA_NF NODE_NF
    which java >/dev/null || JAVA_NF=1
    which node >/dev/null || NODE_NF=1
    [[ -z "$JAVA_NF" ]] && [[ -z "$NODE_NF" ]] && {
      log "Node.js and JDK already installed."
      return
    }
    log "Updating Termux and installing dependencies..."
    pkg update -y
    pkg install nodejs-lts openjdk-17 -y || {
      error "$COLOR_RED
Failed to install Node.js and OpenJDK 17.
Possible reasons (in the order of commonality):
1. Termux was downloaded from Play Store. Termux in Play Store is deprecated, and has packaging bugs. Please install it from F-Droid.
2. Mirrors are down at the moment. Try running \`termux-change-repo\`.
3. Internet connection is unstable.
4. Lack of free storage.$COLOR_OFF" n 2
    }
  }
  
  setup_storage
  install_dependencies

  [[ ! -d "$RVB_DIR" ]] && {
    log "revanced-builder not installed. Installing..."
    mkdir -p "$RVB_DIR"
    cd "$RVB_DIR"
    dload_and_install n
  } || {
    log "revanced-builder found."
    log "All checks done."
    }
}

run_builder() {
  preflight
  termux-wake-lock
  echo
  [[ "$1" == "--delete-cache" ]] || [[ "$1" == "--dc" ]] && {
    delete_cache
  }
  [[ "$1" == "--delete-cache-no-keystore" ]] || [[ "$1" == "--dcnk" ]] && {
    delete_cache_no_keystore
  }
  cd "$RVB_DIR"
  node .
  [[ "$1" == "--delete-cache-after" ]] || [[ "$1" == "--dca" ]] && {
    delete_cache
  }
  [[ "$1" == "--delete-cache-after-no-keystore" ]] || [[ "$1" == "--dcank" ]] && {
    delete_cache_no_keystore
  }
  termux-wake-unlock
}

delete_cache() {
  # Is this even called a cache?
  log "Deleting builder cache..."
  rm -rf "$RVB_DIR"/revanced
}

delete_cache_no_keystore() {
  log "Deleting builder cache preserving keystore..."
  mv "$RVB_DIR"/revanced/revanced.keystore "$HOME"/revanced.keystore
  rm -rf "$RVB_DIR"/revanced
  mkdir -p "$RVB_DIR"/revanced
  mv "$HOME"/revanced.keystore "$RVB_DIR"/revanced/revanced.keystore
}

reinstall_builder() {
  log "Deleting revanced-builder..."
  [[ "$1" != "--delete-keystore" ]] && {
    [[ -f "$RVB_DIR/revanced/revanced.keystore" ]] && {
      mv "$RVB_DIR"/revanced/revanced.keystore "$HOME"/revanced.keystore
      log "Preserving the keystore. If you do not want this, use the --delete-keystore flag."
      log "Execute \`$SCR_NAME_EXEC help\` for more info."
    }
  }
  rm -r "$RVB_DIR"
  mkdir -p "$RVB_DIR"
  [[ -f "$HOME/revanced.keystore" ]] && {
    log "Restoring the keystore..."
    mkdir -p "$RVB_DIR"/revanced
    mv "$HOME"/revanced.keystore "$RVB_DIR"/revanced/revanced.keystore
  }
  log "Reinstalling..."
  cd "$RVB_DIR"
  dload_and_install
}

update_builder() {
  log "Backing up some stuff..."
  [[ -d "$RVB_DIR/revanced" ]] && {
    mkdir -p "$HOME"/revanced_backup
    mv "$RVB_DIR"/revanced/* "$HOME"/revanced_backup
  }
  [[ -f "$RVB_DIR/settings.json" ]] && {
    mv "$RVB_DIR"/settings.json "$HOME"/settings.json
  }
  log "Deleting revanced-builder..."
  rm -r "$RVB_DIR"
  log "Restoring the backup..."
  mkdir -p "$RVB_DIR"
  [[ -d "$HOME/revanced_backup" ]] && {
    mkdir -p "$RVB_DIR"/revanced
    mv "$HOME"/revanced_backup/* "$RVB_DIR"/revanced
  }
  [[ -f "$HOME/settings.json" ]] && {
    mv "$HOME"/settings.json "$RVB_DIR"/settings.json
  }
  log "Updating revanced-builder..."
  cd "$RVB_DIR"
  dload_and_install n
  run_self_update
}

run_self_update() {
  log "Performing self-update..."

  # Download new version
  log "Downloading latest version..."
  ! curl -sLo "$SCR_NAME_EXEC_FP".tmp https://raw.githubusercontent.com/reisxd/revanced-builder/main/android-interface.sh && {
    log "Failed: Error while trying to download new version!"
    error "File requested: https://raw.githubusercontent.com/reisxd/revanced-builder/main/android-interface.sh" n
  } || log "Done."

  # Copy over modes from old version
  OCTAL_MODE=$(stat -c '%a' "$SCR_NAME_EXEC_FP")
  ! chmod "$OCTAL_MODE" "$SCR_NAME_EXEC_FP.tmp" && error "Failed: Error while trying to set mode on $SCR_NAME_EXEC.tmp." n

  # Spawn update script
  cat > updateScript.sh << EOF
#!/bin/bash

# Overwrite old file with new
mv "$SCR_NAME_EXEC_FP.tmp" "$SCR_NAME_EXEC_FP" && {
  echo -e "[$SCR_NAME] Done. Execute '$SCR_NAME_EXEC run' to launch the builder."
  rm \$0
  } || {
  echo "[$SCR_NAME] Failed!"
  }
EOF

  log "Running update process..."
  exec /bin/bash updateScript.sh
}

main() {
  if [[ -z "$@" ]]; then
    run_builder
  elif [[ $# -gt 2 ]]; then
    error "2 optional arguments acceptable, got $#."
  else
    case $1 in
      run)
        run_builder "$2"
      ;;
      reinstall)
        reinstall_builder "$2"
      ;;
      update)
        update_builder
      ;;
      help)
        help_info
      ;;
      *)
        error "Invalid argument(s): $@."
      ;;
    esac
  fi
}

main $@