com.ybo.trackingplugin.TrackingPlugin.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

import com.ybo.trackingplugin.extension.TrackingPluginExtension
import com.ybo.trackingplugin.tasks.ProcessTraceTask
import com.ybo.trackingplugin.tasks.TraceProcessingParams
import com.ybo.trackingplugin.tasks.UnprocessTraceTask
import org.gradle.api.Plugin
import org.gradle.api.Project
import org.gradle.configurationcache.extensions.capitalized
import org.gradle.kotlin.dsl.register

class TrackingPlugin : Plugin {
    override fun apply(project: Project) {
        with(project) {
            val trackingConfig =
                project.extensions.create(
                    TraceProcessingParams.TRACING_FEATURE_CONFIG,
                    TrackingPluginExtension::class.java,
                )

            with(pluginManager) {
                apply("com.android.application")
                apply("org.jetbrains.kotlin.android")
            }

            tasks.register(TraceProcessingParams.PROCESSING_TASK_NAME) {
                listOfConfigs = trackingConfig.configurationHandler.configs
                doLast {
                    println("PROCESSTRACE processing trace")
                }
            }

            tasks.register(TraceProcessingParams.REVERSE_PROCESSING_TASK_NAME) {
                listOfConfigs = trackingConfig.configurationHandler.configs
                doLast {
                    println("PROCESSTRACE Put trace annotation back to normal")
                }
            }

            afterEvaluate {
                trackingConfig.trackables.forEach { taskName ->
                    tasks.getByName(taskName).let {
                        tasks.register("${TraceProcessingParams.TRACABLE_TASKS_PREFIX}${it.name.capitalized()}") {
                            group = TraceProcessingParams.TASK_PROCESSING_GROUP
                            // Hook prepareBuild to execute before the assemble task
                            dependsOn(TraceProcessingParams.PROCESSING_TASK_NAME)

                            // Hook trackable to execute after prepareBuild
                            dependsOn(it.name)

                            // Hook postpareBuild to execute after the trackable task
                            finalizedBy(TraceProcessingParams.REVERSE_PROCESSING_TASK_NAME)
                            it.finalizedBy(TraceProcessingParams.REVERSE_PROCESSING_TASK_NAME)
                        }
                    }
                }
            }
        }
    }
}