Fix 3: Enable Both slot Flash options

Author: fatalcoder524

app/src/main/assets/flash_ak3.sh

@@ -13,9 +13,11 @@ TMP=$F/tmp;
 
 $F/busybox rm -rf $TMP 2>/dev/null;
 $F/busybox mkdir -p $TMP;
+$F/busybox sed -i "/export ZIPFILE=\"\$3\";/a export SLOT=\"\$4\";" $F/update-binary;
+$F/busybox sed -i 's/\[ -e \/dev\/block\/$byname\/system \] || slot=\$(find_slot);/[ -e \/dev\/block\/$byname\/system ] || slot=$SLOT;/' $F/update-binary;
 
 # update-binary <RECOVERY_API_VERSION> <OUTFD> <ZIPFILE>
-AKHOME=$TMP/anykernel $F/busybox ash $F/update-binary 3 1 "$Z";
+AKHOME=$TMP/anykernel $F/busybox ash $F/update-binary 3 1 "$Z" "$S";
 RC=$?;
 
 $F/busybox rm -rf $TMP;

app/src/main/java/com/github/capntrips/kernelflasher/ui/screens/slot/SlotViewModel.kt

@@ -507,7 +507,7 @@ class SlotViewModel(
                 _wasFlashSuccess.value = false
                 val files = File(context.filesDir.canonicalPath)
                 val flashScript = File(files, "flash_ak3.sh")
-                val result = Shell.Builder.create().setFlags(Shell.FLAG_MOUNT_MASTER or Shell.FLAG_REDIRECT_STDERR).build().newJob().add("F=$files Z=\"$zip\" /system/bin/sh $flashScript").to(flashOutput).exec()
+                val result = Shell.Builder.create().setFlags(Shell.FLAG_MOUNT_MASTER or Shell.FLAG_REDIRECT_STDERR).build().newJob().add("F=$files Z=\"$zip\" S=\"slotSuffix\" /system/bin/sh $flashScript").to(flashOutput).exec()
                 if (result.isSuccess) {
                     log(context, "Kernel flashed successfully")
                     _wasFlashSuccess.value = true