revanced-cli/src/main/kotlin/app/revanced/cli/command/ListPatchesCommand.kt

package app.revanced.cli.command

import app.revanced.patcher.PatchBundleLoader
import app.revanced.patcher.annotation.Package
import app.revanced.patcher.extensions.PatchExtensions.compatiblePackages
import app.revanced.patcher.extensions.PatchExtensions.description
import app.revanced.patcher.extensions.PatchExtensions.options
import app.revanced.patcher.extensions.PatchExtensions.patchName
import app.revanced.patcher.patch.PatchClass
import app.revanced.patcher.patch.PatchOption
import picocli.CommandLine.*
import picocli.CommandLine.Help.Visibility.ALWAYS
import java.io.File
import java.util.logging.Logger


@Command(name = "list-patches", description = ["List patches from supplied patch bundles"])
internal object ListPatchesCommand : Runnable {
    private val logger = Logger.getLogger(ListPatchesCommand::class.java.name)

    @Parameters(
        description = ["Paths to patch bundles"], arity = "1..*"
    )
    private lateinit var patchBundles: Array<File>

    @Option(
        names = ["-d", "--with-descriptions"], description = ["List their descriptions"], showDefaultValue = ALWAYS
    )
    private var withDescriptions: Boolean = true

    @Option(
        names = ["-p", "--with-packages"],
        description = ["List the packages the patches are compatible with"],
        showDefaultValue = ALWAYS
    )
    private var withPackages: Boolean = false

    @Option(
        names = ["-v", "--with-versions"],
        description = ["List the versions of the apps the patches are compatible with"],
        showDefaultValue = ALWAYS
    )
    private var withVersions: Boolean = false

    @Option(
        names = ["-o", "--with-options"], description = ["List the options of the patches"], showDefaultValue = ALWAYS
    )
    private var withOptions: Boolean = false

    override fun run() {
        fun Package.buildString() = buildString {
            if (withVersions && versions.isNotEmpty()) {
                appendLine("Package name: $name")
                appendLine("Compatible versions:")
                append(versions.joinToString("\n") { version -> version }.prependIndent("\t"))
            } else append("Package name: $name")
        }

        fun PatchOption<*>.buildString() = buildString {
            appendLine("Title: $title")
            appendLine("Description: $description")

            value?.let {
                appendLine("Key: $key")
                append("Value: $it")
            } ?: append("Key: $key")
        }

        fun PatchClass.buildString() = buildString {
            append("Name: $patchName")

            if (withDescriptions) append("\nDescription: $description")

            if (withOptions && options != null) {
                appendLine("\nOptions:")
                append(
                    options!!.joinToString("\n\n") { option -> option.buildString() }.prependIndent("\t")
                )
            }

            if (withPackages && compatiblePackages != null) {
                appendLine("\nCompatible packages:")
                append(
                    compatiblePackages!!.joinToString("\n") { it.buildString() }.prependIndent("\t")
                )
            }
        }

        logger.info(PatchBundleLoader.Jar(*patchBundles).joinToString("\n\n") { it.buildString() })
    }
}