Add file path for KSFile. Fix fileName documentation

google · Sep 28, 2020 · 4f4c624 · 4f4c624
1 parent f5b6219
commit 4f4c624
Show file tree

Hide file tree

Showing 3 changed files with 14 additions and 1 deletion.
diff --git a/api/src/main/kotlin/com/google/devtools/ksp/symbol/KSFile.kt b/api/src/main/kotlin/com/google/devtools/ksp/symbol/KSFile.kt
@@ -28,7 +28,12 @@ interface KSFile : KSDeclarationContainer, KSAnnotated {
     val packageName: KSName
 
     /**
-     * Absolute path of this source file.
+     * File name of this source file.
      */
     val fileName: String
+
+    /**
+     * Absolute path of this source file.
+     */
+    val filePath: String
 }
diff --git a/compiler-plugin/src/main/kotlin/com/google/devtools/ksp/symbol/impl/java/KSFileJavaImpl.kt b/compiler-plugin/src/main/kotlin/com/google/devtools/ksp/symbol/impl/java/KSFileJavaImpl.kt
@@ -45,6 +45,10 @@ class KSFileJavaImpl private constructor(val psi: PsiJavaFile) : KSFile {
         psi.name
     }
 
+    override val filePath: String by lazy {
+        psi.virtualFile.path
+    }
+
     override val packageName: KSName = KSNameImpl.getCached(if (psi.packageName == "") "<root>" else psi.packageName)
 
     override fun <D, R> accept(visitor: KSVisitor<D, R>, data: D): R {

diff --git a/compiler-plugin/src/main/kotlin/com/google/devtools/ksp/symbol/impl/kotlin/KSFileImpl.kt b/compiler-plugin/src/main/kotlin/com/google/devtools/ksp/symbol/impl/kotlin/KSFileImpl.kt
@@ -51,6 +51,10 @@ class KSFileImpl private constructor(val file: KtFile) : KSFile {
         file.name
     }
 
+    override val filePath: String by lazy {
+        file.virtualFilePath
+    }
+
     override fun <D, R> accept(visitor: KSVisitor<D, R>, data: D): R {
         return visitor.visitFile(this, data)
     }