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

Go to download

Show more of this group Show more artifacts with this name
Show all versions of traceplugin Show documentation

gradle plugin allowing to add automatic logs (or other process) at the start of each traced method

There is a newer version: 0.6.2

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.")
                }
            }
        }
    }
}