#292 add new regex extension property to specify what files should be…

… ignored. Set this value by default to the build directory
cortinico · Aug 5, 2024 · 8e881ca · 8e881ca
1 parent 00e230c
commit 8e881ca
Show file tree

Hide file tree

Showing 11 changed files with 222 additions and 81 deletions.
diff --git a/example/build.gradle.kts b/example/build.gradle.kts
@@ -1,6 +1,4 @@
-import app.cash.sqldelight.gradle.SqlDelightWorkerTask
 import app.cash.sqldelight.gradle.VerifyMigrationTask
-import com.ncorti.ktfmt.gradle.tasks.KtfmtBaseTask
 
 plugins {
     kotlin("jvm")
@@ -25,10 +23,4 @@ tasks.withType<Test> { useJUnitPlatform() }
 
 tasks.withType<VerifyMigrationTask> { enabled = false }
 
-tasks.withType<KtfmtBaseTask>(){
-    mustRunAfter(tasks.withType<SqlDelightWorkerTask>())
-}
-
-
 sqldelight { databases { create("Database") { packageName.set("com.example") } } }
-
diff --git a/plugin-build/plugin/src/main/java/com/ncorti/ktfmt/gradle/KtfmtAndroidUtils.kt b/plugin-build/plugin/src/main/java/com/ncorti/ktfmt/gradle/KtfmtAndroidUtils.kt
@@ -14,6 +14,7 @@ internal object KtfmtAndroidUtils {
         project: Project,
         topLevelFormat: TaskProvider<Task>,
         topLevelCheck: TaskProvider<Task>,
+        ktfmtExtension: KtfmtExtension,
         isKmpProject: Boolean = false
     ) {
         fun applyKtfmtForAndroid() {
@@ -43,6 +44,7 @@ internal object KtfmtAndroidUtils {
                         project,
                         sourceSetName,
                         project.files(Callable { srcDirs }),
+                        ktfmtExtension,
                         topLevelFormat,
                         topLevelCheck,
                     )

diff --git a/plugin-build/plugin/src/main/java/com/ncorti/ktfmt/gradle/KtfmtExtension.kt b/plugin-build/plugin/src/main/java/com/ncorti/ktfmt/gradle/KtfmtExtension.kt
@@ -12,6 +12,7 @@ abstract class KtfmtExtension {
         removeUnusedImports.convention(DEFAULT_REMOVE_UNUSED_IMPORTS)
         debuggingPrintOpsAfterFormatting.convention(DEFAULT_DEBUGGING_PRINT_OPTS)
         manageTrailingCommas.convention(DEFAULT_MANAGE_TRAILING_COMMAS)
+        srcSetPathExclusionPattern.convention(DEFAULT_SRC_SET_PATH_EXCLUSION_PATTERN)
     }
 
     /** ktfmt breaks lines longer than maxWidth. Default 100. */
@@ -53,6 +54,15 @@ abstract class KtfmtExtension {
     /** Whether ktfmt should remove imports that are not used. */
     abstract val removeUnusedImports: Property<Boolean>
 
+    /**
+     * Regex to define what sourceSets paths should not be formatted
+     *
+     * For example the sourceSet "main" in the "build" folder would be excluded with the regex
+     * "^(.*[\\/])?build([\\/].*)?$". This does not affect files inside a sourceSet. To exclude a
+     * file inside a sourceSet use the include & exclude options on the ktfmt task.
+     */
+    abstract val srcSetPathExclusionPattern: Property<Regex>
+
     /**
      * Print the Ops generated by KotlinInputAstVisitor to help reason about formatting (i.e.,
      * newline) decisions
@@ -94,5 +104,7 @@ abstract class KtfmtExtension {
         internal const val DEFAULT_REMOVE_UNUSED_IMPORTS: Boolean = true
         internal const val DEFAULT_DEBUGGING_PRINT_OPTS: Boolean = false
         internal const val DEFAULT_MANAGE_TRAILING_COMMAS: Boolean = false
+        internal val DEFAULT_SRC_SET_PATH_EXCLUSION_PATTERN =
+            Regex("^(.*[\\\\/])?build([\\\\/].*)?\$")
     }
 }
diff --git a/plugin-build/plugin/src/main/java/com/ncorti/ktfmt/gradle/KtfmtPlugin.kt b/plugin-build/plugin/src/main/java/com/ncorti/ktfmt/gradle/KtfmtPlugin.kt
@@ -56,34 +56,35 @@ abstract class KtfmtPlugin : Plugin<Project> {
         topLevelFormat = createTopLevelFormatTask(project)
         topLevelCheck = createTopLevelCheckTask(project)
 
-        project.plugins.withId("kotlin") { applyKtfmt(project) }
+        project.plugins.withId("kotlin") { applyKtfmt(project, ktfmtExtension) }
         project.plugins.withId("kotlin-android") {
             if (project.plugins.hasPlugin("org.jetbrains.kotlin.multiplatform")) {
                 project.logger.i("Skipping Android task creation, as KMP is applied")
             } else {
-                applyKtfmtToAndroidProject(project, topLevelFormat, topLevelCheck)
+                applyKtfmtToAndroidProject(project, topLevelFormat, topLevelCheck, ktfmtExtension)
             }
         }
-        project.plugins.withId("org.jetbrains.kotlin.js") { applyKtfmt(project) }
+        project.plugins.withId("org.jetbrains.kotlin.js") { applyKtfmt(project, ktfmtExtension) }
         project.plugins.withId("org.jetbrains.kotlin.multiplatform") {
-            applyKtfmtToMultiplatformProject(project)
+            applyKtfmtToMultiplatformProject(project, ktfmtExtension)
         }
     }
 
-    private fun applyKtfmt(project: Project) {
+    private fun applyKtfmt(project: Project, ktfmtExtension: KtfmtExtension) {
         val extension = project.extensions.getByType(KotlinProjectExtension::class.java)
         extension.sourceSets.all {
             createTasksForSourceSet(
                 project,
                 it.name,
                 it.kotlin.sourceDirectories,
+                ktfmtExtension,
                 topLevelFormat,
                 topLevelCheck,
             )
         }
     }
 
-    private fun applyKtfmtToMultiplatformProject(project: Project) {
+    private fun applyKtfmtToMultiplatformProject(project: Project, ktfmtExtension: KtfmtExtension) {
         val extension = project.extensions.getByType(KotlinMultiplatformExtension::class.java)
         extension.sourceSets.all {
             val name = "kmp ${it.name}"
@@ -96,6 +97,7 @@ abstract class KtfmtPlugin : Plugin<Project> {
                 project,
                 name,
                 it.kotlin.sourceDirectories,
+                ktfmtExtension,
                 topLevelFormat,
                 topLevelCheck,
             )
@@ -104,7 +106,7 @@ abstract class KtfmtPlugin : Plugin<Project> {
         extension.targets.all { kotlinTarget ->
             if (kotlinTarget.platformType == KotlinPlatformType.androidJvm) {
                 applyKtfmtToAndroidProject(
-                    project, topLevelFormat, topLevelCheck, isKmpProject = true)
+                    project, topLevelFormat, topLevelCheck, ktfmtExtension, isKmpProject = true)
             }
         }
     }
@@ -125,8 +127,6 @@ abstract class KtfmtPlugin : Plugin<Project> {
     }
 
     companion object {
-        internal val defaultExcludes = listOf("**/build/**")
-        internal val defaultExcludesRegex = Regex("^(.*[\\\\/])?build([\\\\/].*)?\$")
         internal val defaultIncludes = listOf("**/*.kt", "**/*.kts")
     }
 }
diff --git a/plugin-build/plugin/src/main/java/com/ncorti/ktfmt/gradle/KtfmtPluginUtils.kt b/plugin-build/plugin/src/main/java/com/ncorti/ktfmt/gradle/KtfmtPluginUtils.kt
@@ -2,9 +2,11 @@ package com.ncorti.ktfmt.gradle
 
 import com.ncorti.ktfmt.gradle.tasks.KtfmtCheckTask
 import com.ncorti.ktfmt.gradle.tasks.KtfmtFormatTask
+import java.io.File
 import org.gradle.api.Project
 import org.gradle.api.Task
 import org.gradle.api.file.FileCollection
+import org.gradle.api.provider.Provider
 import org.gradle.api.tasks.TaskProvider
 import org.gradle.language.base.plugins.LifecycleBasePlugin
 import org.jetbrains.kotlin.gradle.tasks.KotlinCompile
@@ -34,15 +36,16 @@ internal object KtfmtPluginUtils {
         project: Project,
         srcSetName: String,
         srcSetDir: FileCollection,
+        ktfmtExtension: KtfmtExtension,
         topLevelFormat: TaskProvider<Task>,
         topLevelCheck: TaskProvider<Task>
     ) {
         if (shouldCreateTasks(srcSetName).not()) {
             return
         }
 
-        val srcCheckTask = createCheckTask(project, srcSetName, srcSetDir)
-        val srcFormatTask = createFormatTask(project, srcSetName, srcSetDir)
+        val srcCheckTask = createCheckTask(project, srcSetName, srcSetDir, ktfmtExtension)
+        val srcFormatTask = createFormatTask(project, srcSetName, srcSetDir, ktfmtExtension)
 
         // When running together with compileKotlin, ktfmt tasks should have precedence as
         // they're editing the source code
@@ -63,7 +66,8 @@ internal object KtfmtPluginUtils {
     private fun createCheckTask(
         project: Project,
         name: String,
-        srcDir: FileCollection
+        srcDir: FileCollection,
+        ktfmtExtension: KtfmtExtension,
     ): TaskProvider<KtfmtCheckTask> {
         val capitalizedName =
             name.split(" ").joinToString("") {
@@ -77,20 +81,21 @@ internal object KtfmtPluginUtils {
                 charArray.concatToString()
             }
         val taskName = "$TASK_NAME_CHECK$capitalizedName"
-        val inputDirs = project.provider { srcDir.toList() }
+
+        val inputDirs = project.getSelectedSrcSets(srcDir, ktfmtExtension)
         return project.tasks.register(taskName, KtfmtCheckTask::class.java) {
             it.description =
                 "Run Ktfmt formatter for sourceSet '$name' on project '${project.name}'"
             it.setSource(inputDirs)
             it.setIncludes(KtfmtPlugin.defaultIncludes)
-            it.setExcludes(KtfmtPlugin.defaultExcludes)
         }
     }
 
     private fun createFormatTask(
         project: Project,
         name: String,
-        srcDir: FileCollection
+        srcDir: FileCollection,
+        ktfmtExtension: KtfmtExtension,
     ): TaskProvider<KtfmtFormatTask> {
         val srcSetName =
             name.split(" ").joinToString("") {
@@ -104,13 +109,24 @@ internal object KtfmtPluginUtils {
                 charArray.concatToString()
             }
         val taskName = "$TASK_NAME_FORMAT$srcSetName"
-        val inputDirs = project.provider { srcDir.toList() }
+
+        val inputDirs = project.getSelectedSrcSets(srcDir, ktfmtExtension)
         return project.tasks.register(taskName, KtfmtFormatTask::class.java) {
             it.description =
                 "Run Ktfmt formatter validation for sourceSet '$name' on project '${project.name}'"
             it.setSource(inputDirs)
             it.setIncludes(KtfmtPlugin.defaultIncludes)
-            it.setExcludes(KtfmtPlugin.defaultExcludes)
+        }
+    }
+
+    private fun Project.getSelectedSrcSets(
+        srcDir: FileCollection,
+        ktfmtExtension: KtfmtExtension
+    ): Provider<List<File>> {
+        val excludedSourceSets = ktfmtExtension.srcSetPathExclusionPattern
+
+        return provider {
+            srcDir.toList().filterNot { it.absolutePath.matches(excludedSourceSets.get()) }
         }
     }
 }
diff --git a/plugin-build/plugin/src/main/java/com/ncorti/ktfmt/gradle/tasks/KtfmtBaseTask.kt b/plugin-build/plugin/src/main/java/com/ncorti/ktfmt/gradle/tasks/KtfmtBaseTask.kt
@@ -1,24 +1,22 @@
 package com.ncorti.ktfmt.gradle.tasks
 
 import com.ncorti.ktfmt.gradle.FormattingOptionsBean
-import com.ncorti.ktfmt.gradle.KtfmtPlugin
 import com.ncorti.ktfmt.gradle.tasks.worker.KtfmtWorkAction
 import com.ncorti.ktfmt.gradle.tasks.worker.Result
 import com.ncorti.ktfmt.gradle.util.d
-import java.io.File
-import java.util.*
+import java.util.UUID
 import org.gradle.api.file.ConfigurableFileCollection
 import org.gradle.api.file.FileCollection
+import org.gradle.api.file.FileTree
 import org.gradle.api.file.ProjectLayout
 import org.gradle.api.provider.Property
-import org.gradle.api.specs.Spec
 import org.gradle.api.tasks.Classpath
 import org.gradle.api.tasks.IgnoreEmptyDirectories
 import org.gradle.api.tasks.Input
 import org.gradle.api.tasks.InputFiles
-import org.gradle.api.tasks.Internal
 import org.gradle.api.tasks.PathSensitive
 import org.gradle.api.tasks.PathSensitivity
+import org.gradle.api.tasks.SkipWhenEmpty
 import org.gradle.api.tasks.SourceTask
 import org.gradle.api.tasks.TaskAction
 import org.gradle.api.tasks.options.Option
@@ -49,11 +47,11 @@ internal constructor(
     @get:Input
     abstract val includeOnly: Property<String>
 
-    @get:PathSensitive(PathSensitivity.RELATIVE)
-    @get:InputFiles
-    @get:IgnoreEmptyDirectories
-    internal val inputFiles: FileCollection
-        get() = super.getSource().filter(defaultExcludesFilter)
+    @PathSensitive(PathSensitivity.RELATIVE)
+    @InputFiles
+    @IgnoreEmptyDirectories
+    @SkipWhenEmpty
+    override fun getSource(): FileTree = super.getSource()
 
     @TaskAction
     internal fun taskAction() {
@@ -64,16 +62,6 @@ internal constructor(
 
     protected abstract fun execute(workQueue: WorkQueue)
 
-    @get:Internal
-    internal val defaultExcludesFilter: Spec<File> =
-        Spec<File> {
-            if (this.excludes.containsAll(KtfmtPlugin.defaultExcludes) && this.excludes.size == 1) {
-                it.absolutePath.matches(KtfmtPlugin.defaultExcludesRegex).not()
-            } else {
-                true
-            }
-        }
-
     internal fun <T : KtfmtWorkAction> FileCollection.submitToQueue(
         queue: WorkQueue,
         action: Class<T>

diff --git a/plugin-build/plugin/src/main/java/com/ncorti/ktfmt/gradle/tasks/KtfmtCheckTask.kt b/plugin-build/plugin/src/main/java/com/ncorti/ktfmt/gradle/tasks/KtfmtCheckTask.kt
@@ -29,7 +29,7 @@ internal constructor(workerExecutor: WorkerExecutor, layout: ProjectLayout) :
     }
 
     override fun execute(workQueue: WorkQueue) {
-        val results = inputFiles.submitToQueue(workQueue, KtfmtCheckAction::class.java)
+        val results = source.submitToQueue(workQueue, KtfmtCheckAction::class.java)
 
         logger.d("Check results: $results")
         val failures = results.filterIsInstance<Result.Failure>()

diff --git a/plugin-build/plugin/src/main/java/com/ncorti/ktfmt/gradle/tasks/KtfmtFormatTask.kt b/plugin-build/plugin/src/main/java/com/ncorti/ktfmt/gradle/tasks/KtfmtFormatTask.kt
@@ -6,9 +6,7 @@ import com.ncorti.ktfmt.gradle.util.KtfmtUtils
 import com.ncorti.ktfmt.gradle.util.d
 import com.ncorti.ktfmt.gradle.util.i
 import javax.inject.Inject
-import org.gradle.api.file.FileCollection
 import org.gradle.api.file.ProjectLayout
-import org.gradle.api.tasks.OutputFiles
 import org.gradle.workers.WorkQueue
 import org.gradle.workers.WorkerExecutor
 
@@ -18,16 +16,13 @@ abstract class KtfmtFormatTask
 internal constructor(workerExecutor: WorkerExecutor, layout: ProjectLayout) :
     KtfmtBaseTask(workerExecutor, layout) {
 
-    @get:OutputFiles
-    protected val outputFiles: FileCollection
-        get() = inputFiles
-
     init {
         group = KtfmtUtils.GROUP_FORMATTING
+        outputs.upToDateWhen { true }
     }
 
     override fun execute(workQueue: WorkQueue) {
-        val results = inputFiles.submitToQueue(workQueue, KtfmtFormatAction::class.java)
+        val results = source.submitToQueue(workQueue, KtfmtFormatAction::class.java)
 
         logger.d("Format results: $results")
         val failures = results.filterIsInstance<Result.Failure>()

diff --git a/plugin-build/plugin/src/test/java/com/ncorti/ktfmt/gradle/tasks/KtfmtBaseTaskTest.kt b/plugin-build/plugin/src/test/java/com/ncorti/ktfmt/gradle/tasks/KtfmtBaseTaskTest.kt
@@ -39,27 +39,7 @@ internal class KtfmtBaseTaskTest {
         val file = createTempFile(fileName = "file.kt", content = "")
         underTest.source(file)
 
-        assertThat(underTest.inputFiles.files.contains(file)).isTrue()
-    }
-
-    @Test
-    fun `files in build folder are excluded`() {
-        val underTest = project.tasks.getByName("ktfmtFormatMain") as KtfmtBaseTask
-        val buildFile = createTempFile(fileName = "something/build/folder/file.kt", content = "")
-        underTest.source(buildFile)
-
-        assertThat(underTest.inputFiles.files.contains(buildFile)).isFalse()
-    }
-
-    @Test
-    fun `files in build folder are not excluded if users provided a custom exclude`() {
-        val underTest = project.tasks.getByName("ktfmtFormatMain") as KtfmtBaseTask
-        val buildFile = createTempFile(fileName = "something/build/folder/file.kt", content = "")
-        underTest.source(buildFile)
-
-        underTest.exclude("**/generated/**")
-
-        assertThat(underTest.inputFiles.files.contains(buildFile)).isTrue()
+        assertThat(underTest.source.files.contains(file)).isTrue()
     }
 
     private fun createTempFile(