com.ybo.trackingplugin.tasks.BrowsingTask.kt Maven / Gradle / Ivy

Go to download
package com.ybo.trackingplugin.tasks

import com.ybo.trackingplugin.TrackingPlugin
import com.ybo.trackingplugin.extension.TraceConfig
import com.ybo.trackingplugin.tasks.data.TraceAnnotationMark
import com.ybo.trackingplugin.tasks.data.getLanguage
import org.gradle.api.DefaultTask
import org.gradle.api.file.FileTree
import org.gradle.api.tasks.Input
import java.io.File

/** Base task browsing through java and kotlin files.*/
open class BrowsingTask : DefaultTask() {

    init {
        group = TraceProcessingParams.TASK_PROCESSING_GROUP
    }

    @Input
    var listOfConfigs: List = emptyList()

    data class TrackedFile(
        val file: File,
        val toBeProcessedMarkToTrack: TraceAnnotationMark,
        val alreadyProcessedMarkToTrack: TraceAnnotationMark,
    )

    fun browseCode(block: (file: TrackedFile, conf: TraceConfig) -> Unit) {
        listOfConfigs.forEach { config ->
            val fileTree: FileTree =
                project.fileTree(config.srcPath + "")
                    .matching {
                        include("**/*.kt")
                        include("**/*.java")
                        if (config.exclude?.isNotEmpty() == true) {
                            config.exclude?.let { it1 -> exclude(*it1) }
                        }
                    }

            fileTree.forEach {
                if (it.canRead()) {
                    val processedMarkToTrack =
                        TraceAnnotationMark(config.alreadyProcessedAnnotation(), it.getLanguage())
                    val toBeProcessedMarkToTrack =
                        TraceAnnotationMark(config.annotation, it.getLanguage())
                    block(
                        TrackedFile(
                            file = it,
                            toBeProcessedMarkToTrack = toBeProcessedMarkToTrack,
                            alreadyProcessedMarkToTrack = processedMarkToTrack,
                        ),
                        config,
                    )
                } else {
                    if (TrackingPlugin.DEBUG) println("we cannot read ${it.name}. moving on.")
                }
            }
        }
    }
}