fix(java): remove unnecessary imports and update method signature for…

… `prepareContainingClassContext` to be more flexible.
unit-mesh · Apr 16, 2024 · 34b4a3c · 34b4a3c
1 parent 43f421f
commit 34b4a3c
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/java/src/main/kotlin/cc/unitmesh/idea/service/JavaRelatedContext.kt b/java/src/main/kotlin/cc/unitmesh/idea/service/JavaRelatedContext.kt
@@ -4,8 +4,6 @@ import com.intellij.openapi.roots.ProjectFileIndex
 import com.intellij.psi.*
 import com.intellij.psi.util.PsiTreeUtil
 import com.intellij.psi.util.PsiUtil
-import org.jetbrains.annotations.NotNull
-import org.jetbrains.annotations.Nullable
 
 object JavaRelatedContext {
     fun findRelatedClassesAndCleanUp(method: PsiMethod): List<PsiClass> {
@@ -33,7 +31,7 @@ object JavaRelatedContext {
         return classDescription
     }
 
-    fun prepareContainingClassContext(@NotNull method: PsiMethod): PsiClass? {
+    fun prepareContainingClassContext(method: PsiMethod): PsiClass? {
         val file = method.containingFile.copy() as PsiFile
         val containingClass = method.containingClass ?: return null
         val psiElement = PsiTreeUtil.findSameElementInCopy(method, file)