Merge pull request #44 from koxudaxi/fix_element_detection

Fix element detection
koxudaxi · Jan 5, 2023 · a1b2ed6 · a1b2ed6
2 parents 637f03d + 9454d16
commit a1b2ed6
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/src/com/koxudaxi/ruff/RuffInspection.kt b/src/com/koxudaxi/ruff/RuffInspection.kt
@@ -6,6 +6,7 @@ import com.intellij.credentialStore.toByteArrayAndClear
 import com.intellij.openapi.editor.Document
 import com.intellij.openapi.module.ModuleUtil
 import com.intellij.psi.PsiDocumentManager
+import com.intellij.psi.PsiElement
 import com.intellij.psi.PsiElementVisitor
 import com.intellij.psi.util.PsiTreeUtil
 import com.jetbrains.python.inspections.PyInspection
@@ -41,9 +42,9 @@ class RuffInspection : PyInspection() {
                 runRuff(sdk, stdin, *args.toTypedArray())
             } ?: return
             parseJsonResponse(response).forEach {
-                val pyElement = getPyElement(it, pyFile, document) ?: return@forEach
+                val psiElement = getPyElement(it, pyFile, document) ?: return@forEach
                 registerProblem(
-                    pyElement,
+                    psiElement,
                     it.message,
                     it.fix?.let { fix ->
                         RuffQuickFix.create(fix, document)
@@ -52,10 +53,10 @@ class RuffInspection : PyInspection() {
 
         }
 
-        private fun getPyElement(result: Result, pyFile: PyFile, document: Document): PyElement? {
+        private fun getPyElement(result: Result, pyFile: PyFile, document: Document): PsiElement? {
             val start = document.getLineStartOffset(result.location.row - 1) + result.location.column - 1
             val end = document.getLineStartOffset(result.endLocation.row - 1) + result.endLocation.column - 1
-            return PsiTreeUtil.findElementOfClassAtRange(pyFile, start, end, PyElement::class.java)
+            return PsiTreeUtil.findElementOfClassAtRange(pyFile, start, end, PsiElement::class.java)
         }
     }