增加进度条显示, 以及下载sha1和modulelist走proxied http client

EmptyIrony · EmptyIrony · commit 544cf39b448c · 2024-08-27T14:08:27.000+08:00
diff --git a/src/main/kotlin/org/tabooproject/intellij/ProjectBuilder.kt b/src/main/kotlin/org/tabooproject/intellij/ProjectBuilder.kt
@@ -5,8 +5,10 @@ import com.intellij.ide.starters.local.StandardAssetsProvider
 import com.intellij.ide.util.projectWizard.ModuleWizardStep
 import com.intellij.ide.util.projectWizard.WizardContext
 import com.intellij.ide.wizard.*
+import com.intellij.openapi.progress.ProgressManager
 import com.intellij.openapi.project.Project
 import com.intellij.openapi.roots.ui.configuration.ModulesProvider
+import com.intellij.openapi.util.ThrowableComputable
 import org.tabooproject.intellij.step.ConfigurationPropertiesStep
 import org.tabooproject.intellij.step.OptionalPropertiesStep
 import org.tabooproject.intellij.util.Assets
@@ -37,11 +39,16 @@ class ProjectBuilder : AbstractNewProjectWizardBuilder() {
             // 资源填充
             .nextStep {
                 object : AssetsNewProjectWizardStep(it) {
-
                     override fun setupAssets(project: Project) {
                         addAssets(StandardAssetsProvider().getGradlewAssets())
-                        val directory = project.basePath!!
-                        Template.downloadAndUnzipFile(directory)
+
+                        ProgressManager.getInstance().runProcessWithProgressSynchronously(
+                            ThrowableComputable {
+                                val directory = project.basePath!!
+                                Template.downloadAndUnzipFile(directory)
+                            },
+                            "Downloading template files", true, project
+                        )
                     }
                 }
             }
diff --git a/src/main/kotlin/org/tabooproject/intellij/Utils.kt b/src/main/kotlin/org/tabooproject/intellij/Utils.kt
@@ -1,14 +1,19 @@
 package org.tabooproject.intellij
 
-import com.intellij.psi.*
+import com.intellij.psi.PsiDirectory
+import com.intellij.psi.PsiElement
+import com.intellij.psi.PsiFile
+import com.intellij.psi.PsiReference
 import okhttp3.OkHttpClient
 import okhttp3.Request
 import org.jetbrains.kotlin.psi.KtAnnotated
+import java.io.IOException
 import java.net.InetSocketAddress
 import java.net.Proxy
 import java.nio.file.Files
 import java.nio.file.Path
 import java.nio.file.Paths
+import java.util.concurrent.TimeUnit
 
 fun createFileWithDirectories(baseDir: String, relativePath: String): Path? {
     val fullPath = Paths.get(baseDir, relativePath)
@@ -28,6 +33,8 @@ fun createOkHttpClientWithSystemProxy(block: OkHttpClient.Builder.() -> Unit = {
     val proxyHost = System.getProperty("http.proxyHost")
     val proxyPort = System.getProperty("http.proxyPort")
 
+    println("proxy host: $proxyHost, proxyPort: $proxyPort")
+
     val clientBuilder = OkHttpClient.Builder().apply {
         block(this)
     }
@@ -70,4 +77,18 @@ private inline fun <reified T : PsiElement> PsiElement.findParent(
 
         el = el.parent ?: return null
     }
+}
+
+fun readFromUrl(url: String): String? {
+    val client = createOkHttpClientWithSystemProxy {
+        connectTimeout(10, TimeUnit.SECONDS)
+        readTimeout(10, TimeUnit.SECONDS)
+    }
+
+    val response = client
+        .newCall(getRequest(url))
+        .execute()
+        .takeIf { it.isSuccessful } ?: throw IOException("Failed to get $url")
+
+    return response.body?.string()
 }
diff --git a/src/main/kotlin/org/tabooproject/intellij/step/ConfigurationPropertiesStep.kt b/src/main/kotlin/org/tabooproject/intellij/step/ConfigurationPropertiesStep.kt
@@ -3,6 +3,8 @@ package org.tabooproject.intellij.step
 import ai.grazie.utils.capitalize
 import com.intellij.ide.util.projectWizard.ModuleWizardStep
 import com.intellij.ide.util.projectWizard.WizardContext
+import com.intellij.openapi.progress.ProgressManager
+import com.intellij.openapi.util.ThrowableComputable
 import com.intellij.ui.dsl.builder.columns
 import com.intellij.ui.dsl.builder.panel
 import kotlinx.coroutines.DelicateCoroutinesApi
@@ -11,6 +13,7 @@ import kotlinx.coroutines.coroutineScope
 import kotlinx.coroutines.launch
 import org.tabooproject.intellij.component.CheckModulePanel
 import org.tabooproject.intellij.util.ResourceLoader
+import org.tabooproject.intellij.util.ResourceLoader.loadModules
 import javax.swing.JComponent
 import javax.swing.JTextField
 
@@ -88,7 +91,7 @@ class ConfigurationPropertiesStep(val context: WizardContext) : ModuleWizardStep
                     row("Plugin name:") {
                         textField()
                             .apply {
-                                property.mainClass = "org.example.${property.name?.lowercase()}.${property.name?.capitalize()}Plugin"
+                                property.mainClass = "org.example.${property.name?.lowercase()}.${property.name?.capitalize()}"
                                 component.text = property.name
                                 component.columns = 30
                             }.onChanged {
@@ -132,6 +135,14 @@ class ConfigurationPropertiesStep(val context: WizardContext) : ModuleWizardStep
     @OptIn(DelicateCoroutinesApi::class)
     override fun _init() {
         if (inited) return
+
+        ProgressManager.getInstance().runProcessWithProgressSynchronously(
+            ThrowableComputable {
+                loadModules()
+            },
+            "Downloading modules list", true, context.project
+        )
+
         GlobalScope.launch {
             coroutineScope {
                 checkModulePanel.setModules(ResourceLoader.getModules())
diff --git a/src/main/kotlin/org/tabooproject/intellij/util/ResourceLoader.kt b/src/main/kotlin/org/tabooproject/intellij/util/ResourceLoader.kt
@@ -6,8 +6,9 @@ import com.jetbrains.rd.util.info
 import groovy.util.logging.Slf4j
 import kotlinx.serialization.json.*
 import org.jetbrains.annotations.NonNls
+import org.tabooproject.intellij.readFromUrl
 import org.tabooproject.intellij.step.Module
-import java.net.URL
+import java.io.IOException
 import java.security.MessageDigest
 
 /**
@@ -26,22 +27,19 @@ object ResourceLoader {
     }
 
     fun loadModules() {
-        val moduleSha1URL = URL(url.plus("Modules.json.sha1"))
-
         val loadLocalModulesToJson = loadLocalModulesToJson()
 
+        val remoteSha1 = readFromUrl(url.plus("Modules.json.sha1")) ?: throw IOException("Failed to get remote sha1")
+
         if (cacheJson != null) {
             val cacheJsonSha1 = sha1(cacheJson.toString())
-            val remoteSha1 = moduleSha1URL.openStream().use { it.bufferedReader().use { it.readText() } }
             if (remoteSha1.trim() == cacheJsonSha1.trim()) {
                 logger.info { "缓存数据正确,不进行更新" }
                 return
             }
         }
 
-
         sha1(loadLocalModulesToJson.toString()).also {
-            val remoteSha1 = moduleSha1URL.openStream().use { it.bufferedReader().use { it.readText() } }
             if (remoteSha1.trim() == it.trim()) {
                 logger.info { "校验通过,本地文件与远程文件一致,无需更新" }
                 if (cacheJson == null) {
@@ -71,14 +69,11 @@ object ResourceLoader {
     }
 
     fun downloadRemoteJson(): JsonObject {
-        val moduleURL = URL(url.plus("Modules.json"))
-        return moduleURL.openStream().use {
-            it.bufferedReader().use { it.readText() }.let { Json.parseToJsonElement(it).jsonObject }
-        }
+        val jsonString = readFromUrl(url.plus("Modules.json")) ?: throw IOException("Failed to get modules")
+        return Json.parseToJsonElement(jsonString).jsonObject
     }
 
     fun getModules(): Map<String, List<Module>> {
-        loadModules()
         return parseModules(cacheJson ?: error("加载失败"))
     }
 
