feat: Support customize output apk signing levels

ReVanced · Apr 18, 2024 · cd01494 · cd01494
1 parent 6558aaf
commit cd01494
Showing 1 changed file with 19 additions and 0 deletions.
diff --git a/src/main/kotlin/app/revanced/cli/command/PatchCommand.kt b/src/main/kotlin/app/revanced/cli/command/PatchCommand.kt
@@ -12,6 +12,7 @@ import app.revanced.patcher.PatcherConfig
 import kotlinx.coroutines.runBlocking
 import picocli.CommandLine
 import picocli.CommandLine.Help.Visibility.ALWAYS
+import picocli.CommandLine.ITypeConverter
 import picocli.CommandLine.Model.CommandSpec
 import picocli.CommandLine.Spec
 import java.io.File
@@ -228,6 +229,14 @@ internal object PatchCommand : Runnable {
         this.aaptBinaryPath = aaptBinaryPath
     }
 
+    @CommandLine.Option(
+        names = ["--signing-levels"],
+        description = ["Output apk signing levels, eg. \"1,2,3\", empty as default."],
+        converter = [SigningLevelsConverter::class],
+        arity = "0..1",
+    )
+    private var signingLevels = setOf<Int>()
+
     override fun run() {
         // region Setup
 
@@ -318,6 +327,9 @@ internal object PatchCommand : Runnable {
             patcherResult.applyTo(this)
         }.let { patchedApkFile ->
             if (!mount) {
+                val signingLevels = signingLevels.ifEmpty {
+                    ApkUtils.readSigningLevels(apk)
+                }
                 ApkUtils.signApk(
                     patchedApkFile,
                     outputFilePath,
@@ -328,6 +340,7 @@ internal object PatchCommand : Runnable {
                         keyStoreEntryAlias,
                         keyStoreEntryPassword,
                     ),
+                    signingLevels,
                 )
             } else {
                 patchedApkFile.copyTo(outputFilePath, overwrite = true)
@@ -418,4 +431,10 @@ internal object PatchCommand : Runnable {
             }
         logger.info(result)
     }
+
+    private class SigningLevelsConverter : ITypeConverter<Set<Int>> {
+        override fun convert(value: String): Set<Int> {
+            return value.split(',').map { it.trim().toInt() }.toSet()
+        }
+    }
 }