feat: add cleanup for kotlin code

Author: phodal

java/src/main/kotlin/cc/unitmesh/idea/service/JavaRelatedContext.kt

@@ -43,7 +43,7 @@ object JavaRelatedContext {
         return containingClass
     }
 
-    private fun cleanUp(psiClass: PsiClass): PsiClass {
+    fun cleanUp(psiClass: PsiClass): PsiClass {
         val psiElement = psiClass.copy() as PsiClass
         psiElement.containingFile.setName(psiClass.containingFile.name)
         val members = PsiTreeUtil.findChildrenOfType(psiElement, PsiMember::class.java)

java/src/main/kotlin/cc/unitmesh/idea/service/JavaTypeUtil.kt

@@ -46,7 +46,8 @@ object JavaTypeUtil {
             element.parameterList.parameters.filter {
                 it.type is PsiClassReferenceType
             }.map {
-                resolvedClasses[it.name] = (it.type as PsiClassReferenceType).resolve()
+                val resolve = (it.type as PsiClassReferenceType).resolve() ?: return@map null
+                resolvedClasses[it.name] = JavaRelatedContext.cleanUp(resolve)
             }
 
             val outputType = element.returnTypeElement?.type