User: rednesto
Date: 28 Apr 23 10:10
Revision: b49edfa1fab932d25ce11788387c2f1ad975302b
Summary:
Search for mod id reference under progress


TeamCity URL: http://ci.mcdev.io:80/viewModification.html?tab=vcsModificationFiles&modId=8448&personal=false

Index: src/main/kotlin/toml/platform/forge/reference/ModsTomlReferenceContributor.kt
===================================================================
--- src/main/kotlin/toml/platform/forge/reference/ModsTomlReferenceContributor.kt	(revision 2270a96cfa71903027394fac01e31d5fee1b14ca)
+++ src/main/kotlin/toml/platform/forge/reference/ModsTomlReferenceContributor.kt	(revision b49edfa1fab932d25ce11788387c2f1ad975302b)
@@ -17,7 +17,10 @@
 import com.demonwav.mcdev.util.childrenOfType
 import com.demonwav.mcdev.util.findModule
 import com.demonwav.mcdev.util.mapFirstNotNull
+import com.intellij.openapi.progress.ProgressManager
+import com.intellij.openapi.progress.util.ProgressIndicatorBase
 import com.intellij.openapi.project.rootManager
+import com.intellij.openapi.util.Computable
 import com.intellij.openapi.util.Condition
 import com.intellij.openapi.util.TextRange
 import com.intellij.psi.JavaPsiFacade
@@ -126,12 +129,18 @@
         val modAnnotation = JavaPsiFacade.getInstance(element.project).findClass(ForgeConstants.MOD_ANNOTATION, scope)
             ?: return null
         val refScope = GlobalSearchScope.moduleScope(module)
-        return ReferencesSearch.search(modAnnotation, refScope, true).mapFirstNotNull { ref ->
+        return ProgressManager.getInstance().runProcess(
+            Computable {
+                ReferencesSearch.search(modAnnotation, refScope, true).mapFirstNotNull { ref ->
+                    ProgressManager.checkCanceled()
-            ref.element.toUElement()?.getParentOfType<UAnnotation>()
-                ?.takeIf {
-                    it.qualifiedName == ForgeConstants.MOD_ANNOTATION &&
-                        it.findAttributeValue("value")?.evaluateString() == modId
-                }?.getContainingUClass()?.sourcePsi
-        }
+                    ref.element.toUElement()?.getParentOfType<UAnnotation>()
+                        ?.takeIf {
+                            it.qualifiedName == ForgeConstants.MOD_ANNOTATION &&
+                                it.findAttributeValue("value")?.evaluateString() == modId
+                        }?.getContainingUClass()?.sourcePsi
+                }
+            },
+            ProgressIndicatorBase()
+        )
     }
 }