com.spirytusz.spi.weaver.transform.weave.CodeGenerator.kt Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of spi-compiler Show documentation
spi weaver gradle plugin
There is a newer version: 0.0.2
package com.spirytusz.spi.weaver.transform.weave

import com.android.build.api.transform.Format
import com.android.build.api.transform.QualifiedContent
import com.android.build.api.transform.TransformInvocation
import com.spirytusz.spi.weaver.config.FileConst
import com.spirytusz.spi.weaver.log.Logger
import com.spirytusz.spi.weaver.transform.data.ServiceImplInfo
import com.spirytusz.spi.weaver.transform.data.ServiceInfo
import com.spirytusz.spi.weaver.transform.extensions.safelyWriteBytes
import org.apache.commons.codec.digest.DigestUtils
import java.io.File

class CodeGenerator(
    private val transformInvocation: TransformInvocation,
    private val serviceMapping: Map>
) {

    companion object {
        private const val TAG = "CodeGenerator"
    }

    fun generate() {
        val generateStart = System.currentTimeMillis()
        generateServiceImplCreators()
        generateServiceRegistry()
        val generateEnd = System.currentTimeMillis()
        Logger.i(TAG) { "generate() >>> generate end, time cost: [${generateEnd - generateStart}ms]" }
    }

    private fun generateServiceImplCreators() {
        serviceMapping.values.forEach { impls ->
            impls.forEach { impl ->
                val (creatorName, byteArray) = CallableClassByteCodeGenerator.generate(impl.className)
                val dir = transformInvocation.outputProvider.getContentLocation(
                    DigestUtils.md5Hex(impl.toString()) + FileConst.CLASS_FILE_SUFFIX,
                    mutableSetOf(QualifiedContent.DefaultContentType.CLASSES),
                    mutableSetOf(QualifiedContent.Scope.PROJECT),
                    Format.DIRECTORY
                )
                val file = File(dir, namingClassFile(creatorName))
                file.safelyWriteBytes(byteArray)
                Logger.i(TAG) { "generateServiceImplCreators() >>> write creator $creatorName to $file" }
            }
        }
    }

    private fun generateServiceRegistry() {
        val (className, byteArray) = ServiceRegistryByteCodeGenerator.generate(serviceMapping)
        val dir = transformInvocation.outputProvider.getContentLocation(
            DigestUtils.md5Hex(className) + FileConst.CLASS_FILE_SUFFIX,
            mutableSetOf(QualifiedContent.DefaultContentType.CLASSES),
            mutableSetOf(QualifiedContent.Scope.PROJECT),
            Format.DIRECTORY
        )
        val file = File(dir, namingClassFile(className))
        file.safelyWriteBytes(byteArray)
        Logger.i(TAG) { "generateServiceRegistry() >>> write $className to $file" }
    }

    private fun namingClassFile(className: String): String {
        val name = DigestUtils.md5Hex(className.replace("/", "_"))
        return name + FileConst.CLASS_FILE_SUFFIX
    }
}