feat: properly make use of logging facade

This commit is contained in:
oSumAtrIX 2023-08-24 04:41:44 +02:00
parent 45dd15f679
commit 41898d7547
No known key found for this signature in database
GPG Key ID: A9B3094ACDB604B4
12 changed files with 79 additions and 104 deletions

View File

@ -11,10 +11,13 @@ import app.revanced.patcher.patch.PatchOption
import picocli.CommandLine.* import picocli.CommandLine.*
import picocli.CommandLine.Help.Visibility.ALWAYS import picocli.CommandLine.Help.Visibility.ALWAYS
import java.io.File import java.io.File
import java.util.logging.Logger
@Command(name = "list-patches", description = ["List patches from supplied patch bundles"]) @Command(name = "list-patches", description = ["List patches from supplied patch bundles"])
internal object ListPatchesCommand : Runnable { internal object ListPatchesCommand : Runnable {
private val logger = Logger.getLogger(ListPatchesCommand::class.java.name)
@Parameters( @Parameters(
description = ["Paths to patch bundles"], description = ["Paths to patch bundles"],
arity = "1..*" arity = "1..*"

View File

@ -1,20 +1,47 @@
package app.revanced.cli.command package app.revanced.cli.command
import app.revanced.cli.logging.impl.DefaultCliLogger
import app.revanced.patcher.patch.PatchClass import app.revanced.patcher.patch.PatchClass
import picocli.CommandLine import picocli.CommandLine
import picocli.CommandLine.Command import picocli.CommandLine.Command
import picocli.CommandLine.IVersionProvider import picocli.CommandLine.IVersionProvider
import java.util.* import java.util.*
import java.util.logging.*
fun main(args: Array<String>) { fun main(args: Array<String>) {
System.setProperty("java.util.logging.SimpleFormatter.format", "%4\$s: %5\$s %n")
Logger.getLogger("").apply {
handlers.forEach {
it.close()
removeHandler(it)
}
object : Handler() {
override fun publish(record: LogRecord) = formatter.format(record).let {
if (record.level.intValue() > Level.INFO.intValue()) {
System.err.write(it.toByteArray())
} else {
System.out.write(it.toByteArray())
}
}
override fun flush() {
System.out.flush()
System.err.flush()
}
override fun close() = flush()
}.also {
it.level = Level.ALL
it.formatter = SimpleFormatter()
}.let(::addHandler)
}
CommandLine(MainCommand).execute(*args) CommandLine(MainCommand).execute(*args)
} }
internal typealias PatchList = List<PatchClass> internal typealias PatchList = List<PatchClass>
internal val logger = DefaultCliLogger()
object CLIVersionProvider : IVersionProvider { object CLIVersionProvider : IVersionProvider {
override fun getVersion(): Array<String> { override fun getVersion(): Array<String> {
Properties().apply { Properties().apply {

View File

@ -6,12 +6,15 @@ import app.revanced.utils.Options.setOptions
import picocli.CommandLine import picocli.CommandLine
import picocli.CommandLine.Help.Visibility.ALWAYS import picocli.CommandLine.Help.Visibility.ALWAYS
import java.io.File import java.io.File
import java.util.logging.Logger
@CommandLine.Command( @CommandLine.Command(
name = "options", name = "options",
description = ["Generate options file from patches"], description = ["Generate options file from patches"],
) )
internal object OptionsCommand : Runnable { internal object OptionsCommand : Runnable {
private val logger = Logger.getLogger(OptionsCommand::class.java.name)
@CommandLine.Parameters( @CommandLine.Parameters(
description = ["Paths to patch bundles"], description = ["Paths to patch bundles"],
arity = "1..*" arity = "1..*"
@ -41,10 +44,10 @@ internal object OptionsCommand : Runnable {
override fun run() = if (!path.exists() || overwrite) override fun run() = if (!path.exists() || overwrite)
with(PatchBundleLoader.Jar(*patchBundles)) { with(PatchBundleLoader.Jar(*patchBundles)) {
if (update) setOptions(path, logger) if (update) setOptions(path)
Options.serialize(this, prettyPrint = true) Options.serialize(this, prettyPrint = true)
.let(path::writeText) .let(path::writeText)
} }
else logger.error("Options file already exists, use --override to override it") else logger.severe("Options file already exists, use --override to override it")
} }

View File

@ -1,6 +1,5 @@
package app.revanced.cli.command package app.revanced.cli.command
import app.revanced.cli.patcher.logging.impl.PatcherLogger
import app.revanced.patcher.PatchBundleLoader import app.revanced.patcher.PatchBundleLoader
import app.revanced.patcher.Patcher import app.revanced.patcher.Patcher
import app.revanced.patcher.PatcherOptions import app.revanced.patcher.PatcherOptions
@ -20,6 +19,7 @@ import kotlinx.coroutines.runBlocking
import picocli.CommandLine import picocli.CommandLine
import picocli.CommandLine.Help.Visibility.ALWAYS import picocli.CommandLine.Help.Visibility.ALWAYS
import java.io.File import java.io.File
import java.util.logging.Logger
@CommandLine.Command( @CommandLine.Command(
@ -27,6 +27,8 @@ import java.io.File
description = ["Patch the supplied APK file with the supplied patches and integrations"] description = ["Patch the supplied APK file with the supplied patches and integrations"]
) )
internal object PatchCommand: Runnable { internal object PatchCommand: Runnable {
private val logger = Logger.getLogger(PatchCommand::class.java.name)
@CommandLine.Parameters( @CommandLine.Parameters(
description = ["APK file to be patched"], description = ["APK file to be patched"],
arity = "1..1" arity = "1..1"
@ -144,15 +146,13 @@ internal object PatchCommand: Runnable {
// region Prepare // region Prepare
if (!apk.exists()) { if (!apk.exists()) {
logger.error("Input file ${apk.name} does not exist") logger.severe("Input file ${apk.name} does not exist")
return return
} }
val adbManager = deviceSerial?.let { serial -> val adbManager = deviceSerial?.let { serial ->
if (mount) AdbManager.RootAdbManager(serial, logger) else AdbManager.UserAdbManager( if (mount) AdbManager.RootAdbManager(serial)
serial, else AdbManager.UserAdbManager(serial)
logger
)
} }
// endregion // endregion
@ -167,7 +167,7 @@ internal object PatchCommand: Runnable {
logger.info("Setting patch options") logger.info("Setting patch options")
optionsFile.let { optionsFile.let {
if (it.exists()) patches.setOptions(it, logger) if (it.exists()) patches.setOptions(it)
else Options.serialize(patches, prettyPrint = true).let(it::writeText) else Options.serialize(patches, prettyPrint = true).let(it::writeText)
} }
@ -181,7 +181,6 @@ internal object PatchCommand: Runnable {
resourceCachePath, resourceCachePath,
aaptBinaryPath.path, aaptBinaryPath.path,
resourceCachePath.absolutePath, resourceCachePath.absolutePath,
PatcherLogger
) )
) )
@ -193,7 +192,7 @@ internal object PatchCommand: Runnable {
runBlocking { runBlocking {
apply(false).collect { patchResult -> apply(false).collect { patchResult ->
patchResult.exception?.let { patchResult.exception?.let {
logger.error("${patchResult.patchName} failed:\n${patchResult.exception}") logger.severe("${patchResult.patchName} failed:\n${patchResult.exception}")
} ?: logger.info("${patchResult.patchName} succeeded") } ?: logger.info("${patchResult.patchName} succeeded")
} }
} }
@ -288,7 +287,7 @@ internal object PatchCommand: Runnable {
it.isEmpty() || it.any { version -> version == packageVersion } it.isEmpty() || it.any { version -> version == packageVersion }
} }
if (!matchesVersion) return@patch logger.warn( if (!matchesVersion) return@patch logger.warning(
"${patch.patchName} is incompatible with version $packageVersion. " + "${patch.patchName} is incompatible with version $packageVersion. " +
"This patch is only compatible with version " + "This patch is only compatible with version " +
packages.joinToString(";") { pkg -> packages.joinToString(";") { pkg ->
@ -296,14 +295,14 @@ internal object PatchCommand: Runnable {
} }
) )
} ?: return@patch logger.trace( } ?: return@patch logger.fine(
"${patch.patchName} is incompatible with $packageName. " + "${patch.patchName} is incompatible with $packageName. " +
"This patch is only compatible with " + "This patch is only compatible with " +
packages.joinToString(", ") { `package` -> `package`.name } packages.joinToString(", ") { `package` -> `package`.name }
) )
return@let return@let
} ?: logger.trace("$formattedPatchName: No constraint on packages.") } ?: logger.fine("$formattedPatchName: No constraint on packages.")
/** /**
* Check if the patch is explicitly included. * Check if the patch is explicitly included.
@ -329,7 +328,7 @@ internal object PatchCommand: Runnable {
val included = implicitlyIncluded || exclusivelyIncluded val included = implicitlyIncluded || exclusivelyIncluded
if (!included) return@patch logger.info("${patch.patchName} excluded by default") // Case 1. if (!included) return@patch logger.info("${patch.patchName} excluded by default") // Case 1.
logger.trace("Adding $formattedPatchName") logger.fine("Adding $formattedPatchName")
add(patch) add(patch)
} }

View File

@ -3,6 +3,7 @@ package app.revanced.cli.command
import app.revanced.utils.adb.AdbManager import app.revanced.utils.adb.AdbManager
import picocli.CommandLine.* import picocli.CommandLine.*
import picocli.CommandLine.Help.Visibility.ALWAYS import picocli.CommandLine.Help.Visibility.ALWAYS
import java.util.logging.Logger
@Command( @Command(
@ -10,6 +11,8 @@ import picocli.CommandLine.Help.Visibility.ALWAYS
description = ["Uninstall a patched APK file from the devices with the supplied ADB device serials"] description = ["Uninstall a patched APK file from the devices with the supplied ADB device serials"]
) )
internal object UninstallCommand : Runnable { internal object UninstallCommand : Runnable {
private val logger = Logger.getLogger(UninstallCommand::class.java.name)
@Parameters( @Parameters(
description = ["ADB device serials"], description = ["ADB device serials"],
arity = "1..*" arity = "1..*"
@ -33,12 +36,12 @@ internal object UninstallCommand : Runnable {
override fun run() = try { override fun run() = try {
deviceSerials.forEach {deviceSerial -> deviceSerials.forEach {deviceSerial ->
if (unmount) { if (unmount) {
AdbManager.RootAdbManager(deviceSerial, logger) AdbManager.RootAdbManager(deviceSerial)
} else { } else {
AdbManager.UserAdbManager(deviceSerial, logger) AdbManager.UserAdbManager(deviceSerial)
}.uninstall(packageName) }.uninstall(packageName)
} }
} catch (e: AdbManager.DeviceNotFoundException) { } catch (e: AdbManager.DeviceNotFoundException) {
logger.error(e.toString()) logger.severe(e.toString())
} }
} }

View File

@ -1,8 +0,0 @@
package app.revanced.cli.logging
internal interface CliLogger {
fun error(msg: String)
fun info(msg: String)
fun trace(msg: String)
fun warn(msg: String)
}

View File

@ -1,30 +0,0 @@
package app.revanced.cli.logging.impl
import app.revanced.cli.command.MainCommand
import app.revanced.cli.logging.CliLogger
import java.util.logging.Logger
import java.util.logging.SimpleFormatter
internal class DefaultCliLogger(
private val logger: Logger = Logger.getLogger(MainCommand::class.java.name),
private val errorLogger: Logger = Logger.getLogger(logger.name + "Err")
) : CliLogger {
init {
logger.useParentHandlers = false
if (logger.handlers.isEmpty()) {
logger.addHandler(FlushingStreamHandler(System.out, SimpleFormatter()))
}
}
companion object {
init {
System.setProperty("java.util.logging.SimpleFormatter.format", "%4\$s: %5\$s %n")
}
}
override fun error(msg: String) = errorLogger.severe(msg)
override fun info(msg: String) = logger.info(msg)
override fun trace(msg: String) = logger.finest(msg)
override fun warn(msg: String) = errorLogger.warning(msg)
}

View File

@ -1,13 +0,0 @@
package app.revanced.cli.logging.impl
import java.io.OutputStream
import java.util.logging.Formatter
import java.util.logging.LogRecord
import java.util.logging.StreamHandler
internal class FlushingStreamHandler(out: OutputStream, format: Formatter) : StreamHandler(out, format) {
override fun publish(record: LogRecord) {
super.publish(record)
flush()
}
}

View File

@ -1,13 +0,0 @@
package app.revanced.cli.patcher.logging.impl
import app.revanced.cli.logging.impl.DefaultCliLogger
import java.util.logging.Logger
internal object PatcherLogger : app.revanced.patcher.logging.Logger{
private val logger = DefaultCliLogger(Logger.getLogger(app.revanced.patcher.Patcher::class.java.name))
override fun error(msg: String) = logger.error(msg)
override fun info(msg: String) = logger.info(msg)
override fun warn(msg: String)= logger.warn(msg)
override fun trace(msg: String)= logger.trace(msg)
}

View File

@ -1,16 +1,18 @@
package app.revanced.utils package app.revanced.utils
import app.revanced.cli.command.PatchList import app.revanced.cli.command.PatchList
import app.revanced.cli.logging.CliLogger
import app.revanced.patcher.extensions.PatchExtensions.options import app.revanced.patcher.extensions.PatchExtensions.options
import app.revanced.patcher.extensions.PatchExtensions.patchName import app.revanced.patcher.extensions.PatchExtensions.patchName
import app.revanced.patcher.patch.NoSuchOptionException import app.revanced.patcher.patch.NoSuchOptionException
import app.revanced.utils.Options.PatchOption.Option import app.revanced.utils.Options.PatchOption.Option
import com.fasterxml.jackson.module.kotlin.jacksonObjectMapper import com.fasterxml.jackson.module.kotlin.jacksonObjectMapper
import java.io.File import java.io.File
import java.util.logging.Logger
internal object Options { internal object Options {
private val logger = Logger.getLogger(Options::class.java.name)
private var mapper = jacksonObjectMapper() private var mapper = jacksonObjectMapper()
/** /**
@ -53,9 +55,8 @@ internal object Options {
* Sets the options for the patches in the list. * Sets the options for the patches in the list.
* *
* @param json The JSON string containing the options. * @param json The JSON string containing the options.
* @param logger The logger to use for logging.
*/ */
fun PatchList.setOptions(json: String, logger: CliLogger? = null) { fun PatchList.setOptions(json: String) {
filter { it.options?.any() == true }.let { patches -> filter { it.options?.any() == true }.let { patches ->
if (patches.isEmpty()) return if (patches.isEmpty()) return
@ -66,9 +67,9 @@ internal object Options {
it.options.forEach { option -> it.options.forEach { option ->
try { try {
patch.options?.set(option.key, option.value) patch.options?.set(option.key, option.value)
?: logger?.warn("${patch.patchName} has no options") ?: logger.warning("${patch.patchName} has no options")
} catch (e: NoSuchOptionException) { } catch (e: NoSuchOptionException) {
logger?.error(e.message ?: "Unknown error") logger.info(e.message ?: "Unknown error")
} }
} }
} }
@ -80,10 +81,9 @@ internal object Options {
* Sets the options for the patches in the list. * Sets the options for the patches in the list.
* *
* @param file The file containing the JSON string containing the options. * @param file The file containing the JSON string containing the options.
* @param logger The logger to use for logging.
* @see setOptions * @see setOptions
*/ */
fun PatchList.setOptions(file: File, logger: CliLogger? = null) = setOptions(file.readText(), logger) fun PatchList.setOptions(file: File) = setOptions(file.readText())
/** /**
* Data class for a patch and its [Option]s. * Data class for a patch and its [Option]s.

View File

@ -1,6 +1,5 @@
package app.revanced.utils.adb package app.revanced.utils.adb
import app.revanced.cli.logging.CliLogger
import app.revanced.utils.adb.AdbManager.Apk import app.revanced.utils.adb.AdbManager.Apk
import app.revanced.utils.adb.Constants.COMMAND_CREATE_DIR import app.revanced.utils.adb.Constants.COMMAND_CREATE_DIR
import app.revanced.utils.adb.Constants.COMMAND_DELETE import app.revanced.utils.adb.Constants.COMMAND_DELETE
@ -19,18 +18,21 @@ import se.vidstige.jadb.managers.Package
import se.vidstige.jadb.managers.PackageManager import se.vidstige.jadb.managers.PackageManager
import java.io.Closeable import java.io.Closeable
import java.io.File import java.io.File
import java.util.logging.Logger
/** /**
* Adb manager. Used to install and uninstall [Apk] files. * Adb manager. Used to install and uninstall [Apk] files.
* *
* @param deviceSerial The serial of the device. * @param deviceSerial The serial of the device.
*/ */
internal sealed class AdbManager(deviceSerial: String? = null, protected val logger: CliLogger? = null) : Closeable { internal sealed class AdbManager(deviceSerial: String? = null) : Closeable {
protected val logger: Logger = Logger.getLogger(AdbManager::class.java.name)
protected val device = JadbConnection().devices.find { device -> device.serial == deviceSerial } protected val device = JadbConnection().devices.find { device -> device.serial == deviceSerial }
?: throw DeviceNotFoundException(deviceSerial) ?: throw DeviceNotFoundException(deviceSerial)
init { init {
logger?.trace("Established connection to $deviceSerial") logger.fine("Established connection to $deviceSerial")
} }
/** /**
@ -39,7 +41,7 @@ internal sealed class AdbManager(deviceSerial: String? = null, protected val log
* @param apk The [Apk] file. * @param apk The [Apk] file.
*/ */
open fun install(apk: Apk) { open fun install(apk: Apk) {
logger?.info("Finished installing ${apk.file.name}") logger.info("Finished installing ${apk.file.name}")
} }
/** /**
@ -48,23 +50,23 @@ internal sealed class AdbManager(deviceSerial: String? = null, protected val log
* @param packageName The package name. * @param packageName The package name.
*/ */
open fun uninstall(packageName: String) { open fun uninstall(packageName: String) {
logger?.info("Finished uninstalling $packageName") logger.info("Finished uninstalling $packageName")
} }
/** /**
* Closes the [AdbManager] instance. * Closes the [AdbManager] instance.
*/ */
override fun close() { override fun close() {
logger?.trace("Closed") logger.fine("Closed")
} }
class RootAdbManager(deviceSerial: String, logger: CliLogger? = null) : AdbManager(deviceSerial, logger) { class RootAdbManager(deviceSerial: String) : AdbManager(deviceSerial) {
init { init {
if (!device.hasSu()) throw IllegalArgumentException("Root required on $deviceSerial. Task failed") if (!device.hasSu()) throw IllegalArgumentException("Root required on $deviceSerial. Task failed")
} }
override fun install(apk: Apk) { override fun install(apk: Apk) {
logger?.info("Installing by mounting") logger.info("Installing by mounting")
val applyReplacement = getPlaceholderReplacement( val applyReplacement = getPlaceholderReplacement(
apk.packageName ?: throw IllegalArgumentException("Package name is required") apk.packageName ?: throw IllegalArgumentException("Package name is required")
@ -86,7 +88,7 @@ internal sealed class AdbManager(deviceSerial: String? = null, protected val log
} }
override fun uninstall(packageName: String) { override fun uninstall(packageName: String) {
logger?.info("Uninstalling $packageName by unmounting and deleting the package") logger.info("Uninstalling $packageName by unmounting and deleting the package")
val applyReplacement = getPlaceholderReplacement(packageName) val applyReplacement = getPlaceholderReplacement(packageName)
@ -103,7 +105,7 @@ internal sealed class AdbManager(deviceSerial: String? = null, protected val log
} }
} }
class UserAdbManager(deviceSerial: String, logger: CliLogger? = null) : AdbManager(deviceSerial, logger) { class UserAdbManager(deviceSerial: String) : AdbManager(deviceSerial) {
private val packageManager = PackageManager(device) private val packageManager = PackageManager(device)
override fun install(apk: Apk) { override fun install(apk: Apk) {
@ -113,7 +115,7 @@ internal sealed class AdbManager(deviceSerial: String? = null, protected val log
} }
override fun uninstall(packageName: String) { override fun uninstall(packageName: String) {
logger?.info("Uninstalling $packageName") logger.info("Uninstalling $packageName")
packageManager.uninstall(Package(packageName)) packageManager.uninstall(Package(packageName))

View File

@ -1,6 +1,5 @@
package app.revanced.utils.signing package app.revanced.utils.signing
import app.revanced.cli.command.logger
import com.android.apksig.ApkSigner import com.android.apksig.ApkSigner
import org.bouncycastle.asn1.x500.X500Name import org.bouncycastle.asn1.x500.X500Name
import org.bouncycastle.asn1.x509.SubjectPublicKeyInfo import org.bouncycastle.asn1.x509.SubjectPublicKeyInfo
@ -16,10 +15,13 @@ import java.math.BigInteger
import java.security.* import java.security.*
import java.security.cert.X509Certificate import java.security.cert.X509Certificate
import java.util.* import java.util.*
import java.util.logging.Logger
internal class ApkSigner( internal class ApkSigner(
private val signingOptions: SigningOptions private val signingOptions: SigningOptions
) { ) {
private val logger = Logger.getLogger(ApkSigner::class.java.name)
private val signer: ApkSigner.Builder private val signer: ApkSigner.Builder
private val passwordCharArray = signingOptions.password.toCharArray() private val passwordCharArray = signingOptions.password.toCharArray()