Customizing APK file name with new AGP variant APIs

build.gradle.kts

android {
    onVariantProperties {
        val mainOutput = outputs.single { it.outputType == VariantOutputConfiguration.OutputType.SINGLE }
        tasks.register<CreateRenamedApk>("createRenamedApkFor${name}") {
            this.originalApkFolder.set(artifacts.get(ArtifactType.APK))
            this.builtArtifactsLoader.set(artifacts.getBuiltArtifactsLoader())
            this.newApkFolder.set(layout.buildDirectory.dir("outputs/renamed_apk/${[email protected]}"))
            this.versionCode.set(mainOutput.versionCode)
            this.versionName.set(mainOutput.versionName)
        }
    }
}

abstract class CreateRenamedApk : DefaultTask() {

    @get:InputFiles
    abstract val originalApkFolder: DirectoryProperty

    @get:OutputDirectory
    abstract val newApkFolder: DirectoryProperty

    @get:Internal
    abstract val builtArtifactsLoader: Property<BuiltArtifactsLoader>

    @get:Input
    abstract val versionCode: Property<Int>

    @get:Input
    abstract val versionName: Property<String>

    @TaskAction
    fun taskAction() {
        val builtArtifacts = builtArtifactsLoader.get().load(originalApkFolder.get()) ?: throw RuntimeException("Cannot load APKs")
        File(builtArtifacts.elements.single().outputFile).copyTo(
            File(newApkFolder.asFile.get(), "MyApp-${versionName.get()}-${versionCode.get()}.apk")
        )
    }
}

Any updated solutions? thanks