cli: add --rip-lib and --unsigned options

Author: AzyrRuthless

src/main/kotlin/app/morphe/cli/command/PatchCommand.kt

@@ -15,9 +15,13 @@ import picocli.CommandLine.Help.Visibility.ALWAYS
 import picocli.CommandLine.Model.CommandSpec
 import picocli.CommandLine.Spec
 import java.io.File
+import java.io.FileOutputStream
 import java.io.PrintWriter
 import java.io.StringWriter
 import java.util.logging.Logger
+import java.util.zip.ZipEntry
+import java.util.zip.ZipFile
+import java.util.zip.ZipOutputStream
 
 @CommandLine.Command(
     name = "patch",
@@ -181,6 +185,19 @@ internal object PatchCommand : Runnable {
     )
     private var purge: Boolean = false
 
+    @CommandLine.Option(
+        names = ["--rip-lib"],
+        description = ["Remove native libraries for specific architectures (e.g., x86, x86_64)."],
+    )
+    private var ripLibs: List<String> = emptyList()
+
+    @CommandLine.Option(
+        names = ["--unsigned"],
+        description = ["Don't sign the patched APK file."],
+        showDefaultValue = ALWAYS,
+    )
+    private var unsigned: Boolean = false
+
     @CommandLine.Parameters(
         description = ["APK file to patch."],
         arity = "1",
@@ -332,7 +349,12 @@ internal object PatchCommand : Runnable {
         apk.copyTo(temporaryFilesPath.resolve(apk.name), overwrite = true).apply {
             patcherResult.applyTo(this)
         }.let { patchedApkFile ->
-            if (!mount) {
+
+            if (ripLibs.isNotEmpty()) {
+                ripLibraries(patchedApkFile, ripLibs)
+            }
+
+            if (!mount && !unsigned) {
                 ApkUtils.signApk(
                     patchedApkFile,
                     outputFilePath,
@@ -450,4 +472,42 @@ internal object PatchCommand : Runnable {
             }
         logger.info(result)
     }
+
+    private fun ripLibraries(apkFile: File, archsToRemove: List<String>) {
+        if (archsToRemove.isEmpty()) return
+
+        logger.info("Ripping libraries for architectures: $archsToRemove")
+
+        val tempFile = File.createTempFile("temp-rip", ".apk", apkFile.parentFile)
+
+        ZipFile(apkFile).use { zip ->
+            ZipOutputStream(FileOutputStream(tempFile)).use { out ->
+                val entries = zip.entries()
+                while (entries.hasMoreElements()) {
+                    val entry = entries.nextElement()
+                    var shouldRemove = false
+
+                    for (arch in archsToRemove) {
+                        if (entry.name.startsWith("lib/$arch/")) {
+                            shouldRemove = true
+                            break
+                        }
+                    }
+
+                    if (!shouldRemove) {
+                        out.putNextEntry(ZipEntry(entry.name))
+                        zip.getInputStream(entry).copyTo(out)
+                        out.closeEntry()
+                    }
+                }
+            }
+        }
+
+        if (apkFile.delete()) {
+            tempFile.renameTo(apkFile)
+        } else {
+            tempFile.copyTo(apkFile, overwrite = true)
+            tempFile.delete()
+        }
+    }
 }