Fix slow operations errors in local templates

Author: Earthcomputer

src/main/kotlin/creator/custom/providers/BuiltinTemplateProvider.kt

@@ -33,6 +33,8 @@ import com.intellij.openapi.progress.ProgressIndicator
 import com.intellij.ui.dsl.builder.bindSelected
 import com.intellij.ui.dsl.builder.panel
 import javax.swing.JComponent
+import kotlinx.coroutines.Dispatchers
+import kotlinx.coroutines.withContext
 
 class BuiltinTemplateProvider : RemoteTemplateProvider() {
 
@@ -59,16 +61,16 @@ class BuiltinTemplateProvider : RemoteTemplateProvider() {
     override suspend fun loadTemplates(
         context: WizardContext,
         repo: MinecraftSettings.TemplateRepo
-    ): Collection<LoadedTemplate> {
+    ): Collection<LoadedTemplate> = withContext(Dispatchers.IO) {
         val remoteTemplates = doLoadTemplates(context, repo, builtinTemplatesInnerPath)
         if (remoteTemplates.isNotEmpty()) {
-            return remoteTemplates
+            return@withContext remoteTemplates
         }
 
         val repoRoot = builtinTemplatesPath.virtualFile
-            ?: return emptyList()
+            ?: return@withContext emptyList()
         repoRoot.refreshSync(context.modalityState)
-        return TemplateProvider.findTemplates(context.modalityState, repoRoot)
+        TemplateProvider.findTemplates(context.modalityState, repoRoot)
     }
 
     override fun setupConfigUi(

src/main/kotlin/creator/custom/providers/LocalTemplateProvider.kt

@@ -39,6 +39,8 @@ import com.intellij.ui.dsl.builder.textValidation
 import java.nio.file.Path
 import javax.swing.JComponent
 import kotlin.io.path.absolute
+import kotlinx.coroutines.Dispatchers
+import kotlinx.coroutines.withContext
 
 class LocalTemplateProvider : TemplateProvider {
 
@@ -49,13 +51,13 @@ class LocalTemplateProvider : TemplateProvider {
     override suspend fun loadTemplates(
         context: WizardContext,
         repo: MinecraftSettings.TemplateRepo
-    ): Collection<LoadedTemplate> {
+    ): Collection<LoadedTemplate> = withContext(Dispatchers.IO) {
         val rootPath = Path.of(repo.data.trim()).absolute()
         val repoRoot = rootPath.virtualFile
-            ?: return emptyList()
+            ?: return@withContext emptyList()
         val modalityState = context.modalityState
         repoRoot.refreshSync(modalityState)
-        return TemplateProvider.findTemplates(modalityState, repoRoot)
+        TemplateProvider.findTemplates(modalityState, repoRoot)
     }
 
     override fun setupConfigUi(

src/main/kotlin/creator/custom/providers/TemplateProvider.kt

@@ -47,6 +47,8 @@ import com.intellij.util.xmlb.annotations.Attribute
 import java.util.ResourceBundle
 import java.util.concurrent.TimeUnit
 import javax.swing.JComponent
+import kotlinx.coroutines.Dispatchers
+import kotlinx.coroutines.withContext
 import org.jetbrains.concurrency.runAsync
 
 /**
@@ -75,12 +77,12 @@ interface TemplateProvider {
 
         fun getAllKeys() = EP_NAME.extensionList.mapNotNull { it.key }
 
-        fun findTemplates(
+        suspend fun findTemplates(
             modalityState: ModalityState,
             repoRoot: VirtualFile,
             templates: MutableList<VfsLoadedTemplate> = mutableListOf(),
             bundle: ResourceBundle? = loadMessagesBundle(modalityState, repoRoot)
-        ): List<VfsLoadedTemplate> {
+        ): List<VfsLoadedTemplate> = withContext(Dispatchers.IO) {
             val templatesToLoad = mutableListOf<VirtualFile>()
             val visitor = object : VirtualFileVisitor<Unit>() {
                 override fun visitFile(file: VirtualFile): Boolean {
@@ -113,7 +115,7 @@ interface TemplateProvider {
                 }
             }
 
-            return templates
+            templates
         }
 
         fun loadMessagesBundle(modalityState: ModalityState, repoRoot: VirtualFile): ResourceBundle? = try {

src/main/kotlin/creator/custom/providers/ZipTemplateProvider.kt

@@ -38,6 +38,8 @@ import com.intellij.ui.dsl.builder.textValidation
 import java.nio.file.Path
 import javax.swing.JComponent
 import kotlin.io.path.isRegularFile
+import kotlinx.coroutines.Dispatchers
+import kotlinx.coroutines.withContext
 
 class ZipTemplateProvider : TemplateProvider {
 
@@ -48,14 +50,14 @@ class ZipTemplateProvider : TemplateProvider {
     override suspend fun loadTemplates(
         context: WizardContext,
         repo: MinecraftSettings.TemplateRepo
-    ): Collection<LoadedTemplate> {
+    ): Collection<LoadedTemplate> = withContext(Dispatchers.IO) {
         val archiveRoot = repo.data + JarFileSystem.JAR_SEPARATOR
         val fs = JarFileSystem.getInstance()
         val rootFile = fs.refreshAndFindFileByPath(archiveRoot)
-            ?: return emptyList()
+            ?: return@withContext emptyList()
         val modalityState = context.modalityState
         rootFile.refreshSync(modalityState)
-        return TemplateProvider.findTemplates(modalityState, rootFile)
+        TemplateProvider.findTemplates(modalityState, rootFile)
     }
 
     override fun setupConfigUi(