Function type annotations

Author: nicolas-guichard

semanticdb-kotlinc/src/main/kotlin/com/sourcegraph/semanticdb_kotlinc/AnalyzerCheckers.kt

@@ -248,6 +248,12 @@ open class AnalyzerCheckers(session: FirSession) : FirAdditionalCheckersExtensio
             val ktFile = context.containingFile?.sourceFile ?: return
             val visitor = visitors[ktFile]
             visitor?.visitNamedFunction(declaration, getIdentifier(source), context)
+
+            val klass = declaration.returnTypeRef.toClassLikeSymbol(context.session)
+            val klassSource = declaration.returnTypeRef.source
+            if (klass != null && klassSource != null && klassSource.kind !is KtFakeSourceElementKind) {
+                visitor?.visitClassReference(klass, getIdentifier(klassSource), context)
+            }
         }
     }
 
@@ -297,6 +303,12 @@ open class AnalyzerCheckers(session: FirSession) : FirAdditionalCheckersExtensio
             val ktFile = context.containingFile?.sourceFile ?: return
             val visitor = visitors[ktFile]
             visitor?.visitParameter(declaration, getIdentifier(source), context)
+
+            val klass = declaration.returnTypeRef.toClassLikeSymbol(context.session)
+            val klassSource = declaration.returnTypeRef.source
+            if (klass != null && klassSource != null && klassSource.kind !is KtFakeSourceElementKind) {
+                visitor?.visitClassReference(klass, getIdentifier(klassSource), context)
+            }
         }
     }
 

semanticdb-kotlinc/src/test/kotlin/com/sourcegraph/semanticdb_kotlinc/test/AnalyzerTest.kt

@@ -444,6 +444,65 @@ class AnalyzerTest {
         assertSoftly(document.symbolsList) { withClue(this) { symbols.forEach(::shouldContain) } }
     }
 
+    @Test
+    fun `function return type`(@TempDir path: Path) {
+        val document =
+            compileSemanticdb(
+                path,
+                """
+            package sample
+
+            fun foo(arg: Int): Boolean = true
+            """)
+
+        val occurrences =
+            arrayOf(
+                SymbolOccurrence {
+                    role = Role.DEFINITION
+                    symbol = "sample/foo()."
+                    range {
+                        startLine = 2
+                        startCharacter = 4
+                        endLine = 2
+                        endCharacter = 7
+                    }
+                },
+                SymbolOccurrence {
+                    role = Role.DEFINITION
+                    symbol = "sample/foo().(arg)"
+                    range {
+                        startLine = 2
+                        startCharacter = 8
+                        endLine = 2
+                        endCharacter = 11
+                    }
+                },
+                SymbolOccurrence {
+                    role = Role.REFERENCE
+                    symbol = "kotlin/Int#"
+                    range {
+                        startLine = 2
+                        startCharacter = 13
+                        endLine = 2
+                        endCharacter = 16
+                    }
+                },
+                SymbolOccurrence {
+                    role = Role.REFERENCE
+                    symbol = "kotlin/Boolean#"
+                    range {
+                        startLine = 2
+                        startCharacter = 19
+                        endLine = 2
+                        endCharacter = 26
+                    }
+                },
+            )
+        assertSoftly(document.occurrencesList) {
+            withClue(this) { occurrences.forEach(::shouldContain) }
+        }
+    }
+
     @Test
     fun `exception test`(@TempDir path: Path) {
         val buildPath = File(path.resolve("build").toString()).apply { mkdir() }