com.spirytusz.spi.weaver.transform.scan.InputScanner.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.scan

import com.android.build.api.transform.TransformInvocation
import com.spirytusz.spi.weaver.log.Logger
import com.spirytusz.spi.weaver.transform.cache.CacheManager
import com.spirytusz.spi.weaver.transform.data.ServiceImplInfo
import com.spirytusz.spi.weaver.transform.data.ServiceInfo
import com.spirytusz.spi.weaver.transform.global.TransformContext
import com.spirytusz.spi.weaver.transform.scan.base.IInputScanner
import com.spirytusz.spi.weaver.transform.scan.base.InputScannerDispatcher
import com.spirytusz.spi.weaver.transform.scan.directory.FullDirectoryInputScanner
import com.spirytusz.spi.weaver.transform.scan.directory.IncrementalDirectoryInputScanner
import com.spirytusz.spi.weaver.transform.scan.jar.FullJarInputScanner
import com.spirytusz.spi.weaver.transform.scan.jar.IncrementalJarInputScanner

class InputScanner(transformContext: TransformContext) : IInputScanner {

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

    private val targetClassCollector =
        TargetClassCollector(ClassFilter(transformContext.configProvider))
    private val cacheHelper = CacheManager(transformContext)

    private val jarInputScanner by lazy {
        InputScannerDispatcher(
            incrementalScanner = IncrementalJarInputScanner(targetClassCollector, cacheHelper),
            fullScanner = FullJarInputScanner(targetClassCollector, cacheHelper)
        )
    }

    private val directoryInputScanner by lazy {
        InputScannerDispatcher(
            incrementalScanner = IncrementalDirectoryInputScanner(
                targetClassCollector,
                cacheHelper
            ),
            fullScanner = FullDirectoryInputScanner(targetClassCollector, cacheHelper)
        )
    }

    override val serviceMapping: Map>
        get() = mergeServiceMapping()

    override fun onReceiveInput(transformInvocation: TransformInvocation) {
        Logger.i(TAG) {
            "onReceiveInput() >>> scan start, isIncremental=${transformInvocation.isIncremental}"
        }
        val scanStart = System.currentTimeMillis()
        cacheHelper.init(transformInvocation)
        val incremental = transformInvocation.isIncremental
        val outputProvider = transformInvocation.outputProvider

        if (!incremental) {
            outputProvider.deleteAll()
        }

        jarInputScanner.onReceiveInput(transformInvocation)
        directoryInputScanner.onReceiveInput(transformInvocation)

        cacheHelper.apply()

        val scanEnd = System.currentTimeMillis()
        Logger.i(TAG) { "onReceiveInput() >>> scan end, time cost: [${scanEnd - scanStart}ms]" }
    }

    private fun mergeServiceMapping(): Map> {
        val result = mutableMapOf>()
        result.putAll(jarInputScanner.serviceMapping)
        result.putAll(directoryInputScanner.serviceMapping)
        return result.toMap()
    }
}