Add declaring type information for Kotlin top level functions

Author: jevanlingen

rewrite-kotlin/src/main/kotlin/org/openrewrite/kotlin/KotlinTypeMapping.kt

@@ -682,10 +682,10 @@ class KotlinTypeMapping(
             return null
         }
         val signature = signatureBuilder.methodCallSignature(fir)
-        val existing = typeCache.get<Method>(signature)
+        /*val existing = typeCache.get<Method>(signature)
         if (existing != null) {
             return existing
-        }
+        }*/
         return methodInvocationType(fir, signature)
     }
 
@@ -777,6 +777,11 @@ class KotlinTypeMapping(
                     is Parameterized -> type.type
                     else -> Unknown.getInstance()
                 }
+            } else if (resolvedSymbol.callableId.classId == null) {
+                methodDeclarationType(resolvedSymbol.fir, null)
+                declaringType = ShallowClass.build(resolvedSymbol.callableId.packageName.toString() + "." + firFile.name)
+                    //.withMethods(mutableListOf(methodDeclarationType(resolvedSymbol.fir, null)))
+                //TODO: <need to add methods information here too
             }
         } else {
             declaringType = TypeUtils.asFullyQualified(type(function.typeRef))

rewrite-kotlin/src/main/kotlin/org/openrewrite/kotlin/KotlinTypeSignatureBuilder.kt

@@ -381,6 +381,8 @@ class KotlinTypeSignatureBuilder(private val firSession: FirSession, private val
                 ) {
                     declaringSig = "kotlin.Library"
                 }
+            } else if (resolvedSymbol.callableId.classId == null) {
+                declaringSig = resolvedSymbol.callableId.packageName.toString() + "." + firFile.name
             }
         } else if (sym is FirFunctionSymbol<*>) {
             declaringSig = signature(function.typeRef)

rewrite-kotlin/src/test/java/org/openrewrite/kotlin/tree/KTSTest.java

@@ -44,9 +44,29 @@ void topLevelAssignmentExpression() {
     void topLevelFunctionCall() {
         rewriteRun(
           kotlinScript(
-                """
-            println("foo")
             """
+              package org.example
+              
+              fun one() = "one"
+              fun two() = "two"
+              """
+          ),
+          kotlinScript(
+            """
+            import org.example.one
+            one("foo")
+            """,
+            spec -> spec.afterRecipe(cu -> {
+                assertThat(cu.getTypesInUse().getUsedMethods())
+                  .singleElement()
+                    .satisfies(m -> {
+                       assertThat(m.getDeclaringType())
+                         .satisfies(it -> {
+                             assertThat(it.getFullyQualifiedName()).isEqualTo("org.example.openRewriteFile1.kts");
+                             assertThat(it.getMethods()).hasSize(2);
+                         });
+                    });
+            })
           )
         );
     }
@@ -70,7 +90,6 @@ void dslSample() {
         rewriteRun(
           spec -> spec.typeValidationOptions(TypeValidation.none()),
           kotlinScript(
-            //language=none
             """
             plugins {
                 id("org.flywaydb.flyway") version "8.0.2"