From 81cfc3e39f704f01e344f2e79dcfacd73579852c Mon Sep 17 00:00:00 2001
From: Yan Zhulanow <yan.zhulanow@jetbrains.com>
Date: Wed, 19 Sep 2018 21:25:52 +0300
Subject: [PATCH] Kapt: Support symlinked Java source files (KT-26817)

---
 .../jetbrains/kotlin/kapt3/base/KaptPaths.kt  | 10 +--
 .../kapt3/kapt3-base/test/KaptPathsTest.kt    | 67 +++++++++++++++++++
 2 files changed, 72 insertions(+), 5 deletions(-)
 create mode 100644 plugins/kapt3/kapt3-base/test/KaptPathsTest.kt

diff --git a/plugins/kapt3/kapt3-base/src/org/jetbrains/kotlin/kapt3/base/KaptPaths.kt b/plugins/kapt3/kapt3-base/src/org/jetbrains/kotlin/kapt3/base/KaptPaths.kt
index ecccf99775d..209bf146bb3 100644
--- a/plugins/kapt3/kapt3-base/src/org/jetbrains/kotlin/kapt3/base/KaptPaths.kt
+++ b/plugins/kapt3/kapt3-base/src/org/jetbrains/kotlin/kapt3/base/KaptPaths.kt
@@ -7,6 +7,7 @@ package org.jetbrains.kotlin.kapt3.base
 
 import org.jetbrains.kotlin.kapt3.base.util.KaptLogger
 import java.io.File
+import java.nio.file.Files
 
 class KaptPaths(
     val projectBaseDir: File?,
@@ -23,11 +24,10 @@ class KaptPaths(
 
     fun collectJavaSourceFiles(): List<File> {
         return (javaSourceRoots + stubsOutputDir)
-            .map { it.canonicalFile }
-            .distinct()
-            .flatMap { root ->
-                root.walk().filter { it.isFile && it.extension == "java" }.toList()
-            }
+            .sortedBy { Files.isSymbolicLink(it.toPath()) } // Get non-symbolic paths first
+            .flatMap { root -> root.walk().filter { it.isFile && it.extension == "java" }.toList() }
+            .sortedBy { Files.isSymbolicLink(it.toPath()) } // This time is for .java files
+            .distinctBy { it.canonicalPath }
     }
 }
 
diff --git a/plugins/kapt3/kapt3-base/test/KaptPathsTest.kt b/plugins/kapt3/kapt3-base/test/KaptPathsTest.kt
new file mode 100644
index 00000000000..3676ffdae15
--- /dev/null
+++ b/plugins/kapt3/kapt3-base/test/KaptPathsTest.kt
@@ -0,0 +1,67 @@
+/*
+ * 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.kapt.base.test
+
+import junit.framework.TestCase
+import org.jetbrains.kotlin.kapt3.base.KaptPaths
+import org.junit.Test
+import java.io.File
+import java.nio.file.Files
+
+class KaptPathsTest : TestCase() {
+    @Test
+    fun testSymbolicLinks() {
+        val tempDir = Files.createTempDirectory("kapt-test").toFile()
+        try {
+            fun File.writeJavaClass() = apply {
+                parentFile.mkdirs()
+                writeText("public class $nameWithoutExtension {}")
+            }
+
+            val outputDir = File(tempDir, "stubs").apply { mkdir() }
+            val otherDir = File(tempDir, "other").apply { mkdir() }
+            val javaRootDir = File(tempDir, "java").apply { mkdir() }
+
+            val simpleJava = File(tempDir, "Simple.java").writeJavaClass()
+            val otherJava = File(otherDir, "Other.java").writeJavaClass()
+            val notJava = File(otherDir, "NotJava.not").writeJavaClass()
+            File(javaRootDir, "JavaRoot.java").writeJavaClass()
+
+            val symlinkToOtherJava = Files.createSymbolicLink(File(tempDir, "Other.java").toPath(), otherJava.toPath()).toFile()
+            val symlinkToNotJava = Files.createSymbolicLink(File(tempDir, "NotJava.java").toPath(), notJava.toPath()).toFile()
+            val symlinkToJavaRootDir = Files.createSymbolicLink(File(tempDir, "java2").toPath(), javaRootDir.toPath()).toFile()
+
+            val javaRoots = listOf(simpleJava, symlinkToOtherJava, symlinkToNotJava, symlinkToJavaRootDir, javaRootDir)
+
+            val paths = KaptPaths(
+                projectBaseDir = null,
+                compileClasspath = emptyList(),
+                annotationProcessingClasspath = emptyList(),
+                javaSourceRoots = javaRoots,
+                sourcesOutputDir = outputDir,
+                classFilesOutputDir = outputDir,
+                stubsOutputDir = outputDir,
+                incrementalDataOutputDir = null
+            )
+
+            val javaSourceFiles = paths.collectJavaSourceFiles()
+
+            fun assertContains(path: String) {
+                val available by lazy { javaSourceFiles.joinToString { it.toRelativeString(tempDir) } }
+                assertTrue("Can't find path $path\nAvailable: $available",
+                           javaSourceFiles.any { it.toRelativeString(tempDir) == path })
+            }
+
+            assertEquals(4, javaSourceFiles.size)
+            assertContains("Simple.java")
+            assertContains("Other.java")
+            assertContains("NotJava.java")
+            assertContains("java/JavaRoot.java")
+        } finally {
+            tempDir.deleteRecursively()
+        }
+    }
+}
\ No newline at end of file