RUM-8582: Add kotlin compiler plugin environment

dd-sdk-android-gradle-plugin/build.gradle.kts

@@ -16,6 +16,7 @@ plugins {
     // Build
     id("java-gradle-plugin")
     kotlin("jvm")
+    kotlin("kapt")
 
     // Publish
     `maven-publish`
@@ -46,6 +47,10 @@ dependencies {
     testImplementation(libs.okHttpMock)
     testImplementation(libs.androidToolsPluginGradle)
     testImplementation(libs.kotlinPluginGradle)
+
+    compileOnly(libs.kotlinCompilerEmbeddable)
+    compileOnly(libs.autoServiceAnnotation)
+    kapt(libs.autoService)
 }
 
 kotlinConfig()

dd-sdk-android-gradle-plugin/src/main/kotlin/com/datadog/gradle/plugin/kcp/ComposeTagTransformer.kt

@@ -0,0 +1,12 @@
+package com.datadog.gradle.plugin.kcp
+
+import org.jetbrains.kotlin.backend.common.IrElementTransformerVoidWithContext
+import org.jetbrains.kotlin.backend.common.extensions.IrPluginContext
+import org.jetbrains.kotlin.cli.common.messages.MessageCollector
+
+@Suppress("UnusedPrivateProperty")
+
+class ComposeTagTransformer(
+    private val messageCollector: MessageCollector,
+    private val pluginContext: IrPluginContext
+) : IrElementTransformerVoidWithContext() 

dd-sdk-android-gradle-plugin/src/main/kotlin/com/datadog/gradle/plugin/kcp/DatadogIrExtension.kt

@@ -0,0 +1,20 @@
+package com.datadog.gradle.plugin.kcp
+
+import org.jetbrains.kotlin.backend.common.extensions.FirIncompatiblePluginAPI
+import org.jetbrains.kotlin.backend.common.extensions.IrGenerationExtension
+import org.jetbrains.kotlin.backend.common.extensions.IrPluginContext
+import org.jetbrains.kotlin.cli.common.messages.MessageCollector
+import org.jetbrains.kotlin.ir.declarations.IrModuleFragment
+
+/**
+ * The extension registers all the visitors that need to explore the code being compiled
+ */
+@FirIncompatiblePluginAPI
+class DatadogIrExtension(private val messageCollector: MessageCollector) : IrGenerationExtension {
+
+    override fun generate(moduleFragment: IrModuleFragment, pluginContext: IrPluginContext) {
+        val composeTagTransformer = ComposeTagTransformer(messageCollector, pluginContext)
+
+        moduleFragment.accept(composeTagTransformer, null)
+    }
+}

dd-sdk-android-gradle-plugin/src/main/kotlin/com/datadog/gradle/plugin/kcp/DatadogPluginRegistrar.kt

@@ -0,0 +1,32 @@
+package com.datadog.gradle.plugin.kcp
+
+import com.google.auto.service.AutoService
+import org.jetbrains.kotlin.backend.common.extensions.FirIncompatiblePluginAPI
+import org.jetbrains.kotlin.backend.common.extensions.IrGenerationExtension
+import org.jetbrains.kotlin.cli.common.CLIConfigurationKeys
+import org.jetbrains.kotlin.cli.common.messages.MessageCollector
+import org.jetbrains.kotlin.compiler.plugin.CompilerPluginRegistrar
+import org.jetbrains.kotlin.compiler.plugin.ExperimentalCompilerApi
+import org.jetbrains.kotlin.config.CommonConfigurationKeys
+import org.jetbrains.kotlin.config.CompilerConfiguration
+
+/**
+ * The Registrar makes the plugin visible to the Kotlin Compiler, and provides the extensions to use.
+ */
+@OptIn(ExperimentalCompilerApi::class)
+@AutoService(CompilerPluginRegistrar::class)
+class DatadogPluginRegistrar : CompilerPluginRegistrar() {
+
+    // Supports Kotlin 2.x compiler
+    override val supportsK2 = true
+
+    @OptIn(FirIncompatiblePluginAPI::class)
+    override fun ExtensionStorage.registerExtensions(configuration: CompilerConfiguration) {
+        val messageCollector = configuration.get(CLIConfigurationKeys.MESSAGE_COLLECTOR_KEY, MessageCollector.NONE)
+        val useFir = configuration.getBoolean(CommonConfigurationKeys.USE_FIR)
+
+        if (!useFir) {
+            IrGenerationExtension.registerExtension(DatadogIrExtension(messageCollector))
+        }
+    }
+}

gradle/libs.versions.toml

@@ -40,6 +40,10 @@ kotlinGrammarParser = "c35b50fa44"
 datadogSdk = "2.18.0"
 datadogPluginGradle = "1.15.0"
 
+# Kotlin Compiler Plugin
+kotlinCompilerEmbeddable = "1.8.20"
+autoService = "1.0.1"
+
 [libraries]
 
 # Classpaths
@@ -93,6 +97,11 @@ fuzzyWuzzy = { module = "me.xdrop:fuzzywuzzy", version.ref = "fuzzyWuzzy" }
 datadogSdkRum = { module = "com.datadoghq:dd-sdk-android-rum", version.ref = "datadogSdk" }
 datadogPluginGradle = { module = "com.datadoghq:dd-sdk-android-gradle-plugin", version.ref = "datadogPluginGradle" }
 
+# Kotlin Compiler Plugin
+kotlinCompilerEmbeddable =  { module = "org.jetbrains.kotlin:kotlin-compiler-embeddable", version.ref = "kotlinCompilerEmbeddable" }
+autoService =  { module = "com.google.auto.service:auto-service", version.ref = "autoService" }
+autoServiceAnnotation =  { module = "com.google.auto.service:auto-service-annotations", version.ref = "autoService" }
+
 [bundles]
 
 jUnit5 = [