Introduce tooling model for Kapt gradle projects

A Kapt model for Gradle projects including 'kotlin-kapt'plugin is
introduced. Its model builder KaptModelBuilder produces an
implementation of these models and can be queried through the Gradle
Tooling API when needed. Model builder is registered in
Kapt3GradleSubplugin.

libraries/tools/kotlin-gradle-plugin/src/main/kotlin/org/jetbrains/kotlin/gradle/internal/kapt/Kapt3KotlinGradleSubplugin.kt

@@ -18,10 +18,12 @@ import org.gradle.api.tasks.TaskDependency
 import org.gradle.api.tasks.compile.AbstractCompile
 import org.gradle.api.tasks.compile.JavaCompile
 import org.gradle.process.CommandLineArgumentProvider
+import org.gradle.tooling.provider.model.ToolingModelBuilderRegistry
 import org.jetbrains.kotlin.gradle.internal.Kapt3GradleSubplugin.Companion.getKaptGeneratedClassesDir
 import org.jetbrains.kotlin.gradle.internal.Kapt3GradleSubplugin.Companion.getKaptGeneratedKotlinSourcesDir
 import org.jetbrains.kotlin.gradle.internal.Kapt3GradleSubplugin.Companion.getKaptGeneratedSourcesDir
 import org.jetbrains.kotlin.gradle.internal.Kapt3KotlinGradleSubplugin.Companion.KAPT_WORKER_DEPENDENCIES_CONFIGURATION_NAME
+import org.jetbrains.kotlin.gradle.model.builder.KaptModelBuilder
 import org.jetbrains.kotlin.gradle.tasks.isWorkerAPISupported
 import org.jetbrains.kotlin.gradle.plugin.*
 import org.jetbrains.kotlin.gradle.tasks.CompilerPluginOptions
@@ -31,9 +33,10 @@ import java.io.ByteArrayOutputStream
 import java.io.File
 import java.io.ObjectOutputStream
 import java.util.*
+import javax.inject.Inject
 
 // apply plugin: 'kotlin-kapt'
-class Kapt3GradleSubplugin : Plugin<Project> {
+class Kapt3GradleSubplugin @Inject internal constructor(private val registry: ToolingModelBuilderRegistry) : Plugin<Project> {
     companion object {
         fun isEnabled(project: Project) = project.plugins.findPlugin(Kapt3GradleSubplugin::class.java) != null
 
@@ -61,6 +64,7 @@ class Kapt3GradleSubplugin : Plugin<Project> {
                 } ?: project.logger.error("Kotlin plugin should be enabled before 'kotlin-kapt'")
             }
         }
+        registry.register(KaptModelBuilder())
     }
 }
 

libraries/tools/kotlin-gradle-plugin/src/main/kotlin/org/jetbrains/kotlin/gradle/model/builder/KaptModelBuilder.kt

@@ -0,0 +1,51 @@
+/*
+ * Copyright 2010-2018 JetBrains s.r.o. Use of this source code is governed by the Apache 2.0 license
+ * that can be found in the license/LICENSE.txt file.
+ */
+
+package org.jetbrains.kotlin.gradle.model.builder
+
+import org.gradle.api.Project
+import org.gradle.tooling.provider.model.ToolingModelBuilder
+import org.jetbrains.kotlin.gradle.internal.KaptTask
+import org.jetbrains.kotlin.gradle.model.Kapt
+import org.jetbrains.kotlin.gradle.model.KaptSourceSet
+import org.jetbrains.kotlin.gradle.model.impl.KaptImpl
+import org.jetbrains.kotlin.gradle.model.impl.KaptSourceSetImpl
+
+/**
+ * [ToolingModelBuilder] for [Kapt] models.
+ * This model builder is registered for Kapt Gradle sub-plugin.
+ */
+class KaptModelBuilder : ToolingModelBuilder {
+
+    override fun canBuild(modelName: String): Boolean {
+        return modelName == Kapt::class.java.name
+    }
+
+    override fun buildAll(modelName: String, project: Project): Any? {
+        if (modelName == Kapt::class.java.name) {
+            val kaptTasks = project.tasks.withType(KaptTask::class.java)
+            return KaptImpl(project.name, kaptTasks.map { it.createKaptSourceSet() })
+        }
+        return null
+    }
+
+    companion object {
+
+        private fun KaptTask.createKaptSourceSet(): KaptSourceSet {
+            val sourceSetName = kotlinCompileTask.sourceSetName
+            return KaptSourceSetImpl(
+                sourceSetName,
+                if (sourceSetName.contains(
+                        "test",
+                        true
+                    )
+                ) KaptSourceSet.KaptSourceSetType.TEST else KaptSourceSet.KaptSourceSetType.PRODUCTION,
+                destinationDir,
+                kotlinSourcesDestinationDir,
+                classesDir
+            )
+        }
+    }
+}

libraries/tools/kotlin-gradle-plugin/src/main/kotlin/org/jetbrains/kotlin/gradle/model/impl/KaptImpl.kt

@@ -0,0 +1,35 @@
+/*
+ * Copyright 2010-2018 JetBrains s.r.o. Use of this source code is governed by the Apache 2.0 license
+ * that can be found in the license/LICENSE.txt file.
+ */
+
+package org.jetbrains.kotlin.gradle.model.impl
+
+import org.jetbrains.kotlin.gradle.model.Kapt
+import org.jetbrains.kotlin.gradle.model.KaptSourceSet
+import java.io.Serializable
+
+/**
+ * Implementation of the [Kapt] interface.
+ */
+data class KaptImpl(
+    private val myName: String,
+    private val kaptSourceSets: Collection<KaptSourceSet>
+) : Kapt, Serializable {
+
+    override fun getModelVersion(): Long {
+        return serialVersionUID
+    }
+
+    override fun getName(): String {
+        return myName
+    }
+
+    override fun getKaptSourceSets(): Collection<KaptSourceSet> {
+        return kaptSourceSets
+    }
+
+    companion object {
+        private const val serialVersionUID = 1L
+    }
+}

libraries/tools/kotlin-gradle-plugin/src/main/kotlin/org/jetbrains/kotlin/gradle/model/impl/KaptSourceSetImpl.kt

@@ -0,0 +1,42 @@
+/*
+ * Copyright 2010-2018 JetBrains s.r.o. Use of this source code is governed by the Apache 2.0 license
+ * that can be found in the license/LICENSE.txt file.
+ */
+
+package org.jetbrains.kotlin.gradle.model.impl
+
+import org.jetbrains.kotlin.gradle.model.KaptSourceSet
+import java.io.File
+import java.io.Serializable
+
+/**
+ * Implementation of the [KaptSourceSet] interface.
+ */
+data class KaptSourceSetImpl(
+    private val myName: String,
+    private val myType: KaptSourceSet.KaptSourceSetType,
+    private val myGeneratedSourcesDirectory: File,
+    private val myGeneratedKotlinSourcesDirectory: File,
+    private val myGeneratedClassesDirectory: File
+) : KaptSourceSet, Serializable {
+
+    override fun getName(): String {
+        return myName
+    }
+
+    override fun getType(): KaptSourceSet.KaptSourceSetType {
+        return myType
+    }
+
+    override fun getGeneratedSourcesDirectory(): File {
+        return myGeneratedSourcesDirectory
+    }
+
+    override fun getGeneratedKotlinSourcesDirectory(): File {
+        return myGeneratedKotlinSourcesDirectory
+    }
+
+    override fun getGeneratedClassesDirectory(): File {
+        return myGeneratedClassesDirectory
+    }
+}

libraries/tools/kotlin-gradle-plugin/src/test/kotlin/org/jetbrains/kotlin/gradle/model/builder/KaptModelBuilderTest.kt

@@ -0,0 +1,20 @@
+/*
+ * Copyright 2010-2018 JetBrains s.r.o. Use of this source code is governed by the Apache 2.0 license
+ * that can be found in the license/LICENSE.txt file.
+ */
+
+package org.jetbrains.kotlin.gradle.model.builder
+
+import org.jetbrains.kotlin.gradle.model.Kapt
+import org.junit.Test
+import kotlin.test.assertFalse
+import kotlin.test.assertTrue
+
+class KaptModelBuilderTest {
+    @Test
+    fun testCanBuild() {
+        val modelBuilder = KaptModelBuilder()
+        assertTrue(modelBuilder.canBuild(Kapt::class.java.name))
+        assertFalse(modelBuilder.canBuild("wrongModel"))
+    }
+}

libraries/tools/kotlin-gradle-plugin/src/test/kotlin/org/jetbrains/kotlin/gradle/model/impl/KaptImplTest.kt

@@ -0,0 +1,18 @@
+/*
+ * Copyright 2010-2018 JetBrains s.r.o. Use of this source code is governed by the Apache 2.0 license
+ * that can be found in the license/LICENSE.txt file.
+ */
+
+package org.jetbrains.kotlin.gradle.model.impl
+
+import nl.jqno.equalsverifier.EqualsVerifier
+import nl.jqno.equalsverifier.Warning
+import org.junit.Test
+
+class KaptImplTest {
+    @Test
+    @Throws(Exception::class)
+    fun equals() {
+        EqualsVerifier.forClass(KaptImpl::class.java).suppress(Warning.NULL_FIELDS).verify()
+    }
+}

libraries/tools/kotlin-gradle-plugin/src/test/kotlin/org/jetbrains/kotlin/gradle/model/impl/KaptSourceSetImplTest.kt

@@ -0,0 +1,18 @@
+/*
+ * Copyright 2010-2018 JetBrains s.r.o. Use of this source code is governed by the Apache 2.0 license
+ * that can be found in the license/LICENSE.txt file.
+ */
+
+package org.jetbrains.kotlin.gradle.model.impl
+
+import nl.jqno.equalsverifier.EqualsVerifier
+import nl.jqno.equalsverifier.Warning
+import org.junit.Test
+
+class KaptSourceSetImplTest {
+    @Test
+    @Throws(Exception::class)
+    fun equals() {
+        EqualsVerifier.forClass(KaptSourceSetImpl::class.java).suppress(Warning.NULL_FIELDS).verify()
+    }
+}