✨ (apply): 增加样式应用功能

Author: 892768447

src/main/kotlin/irony/pycharm/qsseditor/QSSAction.kt

@@ -41,5 +41,6 @@ class QSSAction : AnAction() {
         if (text.isEmpty()) {
             return
         }
+        QSSClient.applyStyle(listOf(text))
     }
 }

src/main/kotlin/irony/pycharm/qsseditor/QSSClient.kt

@@ -11,9 +11,13 @@
 @file:Suppress("ktlint:standard:no-wildcard-imports")
 
 package irony.pycharm.qsseditor
+
 import com.intellij.openapi.diagnostic.logger
+import com.intellij.util.ui.update.MergingUpdateQueue
+import com.intellij.util.ui.update.Update
 import kotlinx.serialization.json.*
 import okhttp3.*
+import org.jetbrains.annotations.Nullable
 import java.util.concurrent.TimeUnit
 
 private val Log = logger<QSSClient>()
@@ -24,7 +28,7 @@ interface MessageListener {
 
 class KeywordsListener : MessageListener {
     override fun onParams(array: JsonArray) {
-        Log.info("onParams: $array")
+        // TODO: 添加关键词
     }
 }
 
@@ -37,14 +41,15 @@ class QSSClient : WebSocketListener() {
         webSocket: WebSocket,
         response: Response,
     ) {
+        twice = 0
         Log.debug("onOpen")
     }
 
     override fun onMessage(
         webSocket: WebSocket,
         text: String,
     ) {
-        Log.debug("onMessage: $text")
+        // Log.debug("onMessage: $text")
         if (text.isNotEmpty()) {
             val json = Json.parseToJsonElement(text).jsonObject
             if (json.getValue(
@@ -79,7 +84,9 @@ class QSSClient : WebSocketListener() {
         t: Throwable,
         response: Response?,
     ) {
-        Log.warn("onFailure, ${t.message}")
+        if (twice < 50) {
+            Log.warn("onFailure, ${t.message}")
+        }
         if (done) {
             Log.info("client require shutdown")
             return
@@ -91,10 +98,13 @@ class QSSClient : WebSocketListener() {
     companion object {
         private var host: String = ""
         private var port: Int = 0
+        private var twice: Int = 0
         private var done: Boolean = false
         private var self: QSSClient? = null
         private var client: OkHttpClient? = null
         private var socket: WebSocket? = null
+        private var queue: MergingUpdateQueue? = null
+        private var messages: MutableMap<String, List<String>> = mutableMapOf()
         private var listener: HashMap<String, MessageListener> = HashMap()
 
         fun connect(
@@ -114,6 +124,8 @@ class QSSClient : WebSocketListener() {
 
             if (self == null) {
                 self = QSSClient()
+                queue = MergingUpdateQueue("QSSClient", 2000, true, MergingUpdateQueue.ANY_COMPONENT, null)
+                queue!!.setRestartTimerOnAdd(true)
             }
             if (client == null) {
                 client = OkHttpClient.Builder().pingInterval(5, TimeUnit.SECONDS).connectTimeout(30, TimeUnit.SECONDS).build()
@@ -129,7 +141,10 @@ class QSSClient : WebSocketListener() {
             host: String,
             port: Int,
         ) {
-            Log.debug("do reconnect")
+            if (twice < 50) {
+                twice = twice.inc()
+                Log.debug("do reconnect")
+            }
             socket = client!!.newWebSocket(Request.Builder().url("ws://$host:$port").build(), self!!)
         }
 
@@ -161,12 +176,44 @@ class QSSClient : WebSocketListener() {
             }.toString()
         }
 
-        fun applyStyle(message: List<String>) {
-            socket!!.send(buildMessage("setStyleSheet", message))
+        private fun sendMessage(
+            method: String,
+            @Nullable message: List<String>,
+            delay: Boolean = false,
+        ) {
+            if (message.isEmpty()) {
+                return
+            }
+            if (delay) {
+                messages.clear()
+                messages[method] = message
+                queue?.queue(
+                    object : Update(method) {
+                        override fun run() {
+                            for (item in messages) {
+                                sendMessage(item.key, item.value, false)
+                            }
+                        }
+                    },
+                )
+            } else {
+                // Log.info("sendMessage: method=$method, message=" + message.joinToString("\n"))
+                socket!!.send(buildMessage(method, message))
+            }
+        }
+
+        fun applyStyle(
+            @Nullable message: List<String>,
+            delay: Boolean = false,
+        ) {
+            sendMessage("setStyleSheet", message, delay)
         }
 
-        fun selectWidget(message: List<String>) {
-            socket!!.send(buildMessage("selectWidget", message))
+        fun selectWidget(
+            @Nullable message: List<String>,
+            delay: Boolean = false,
+        ) {
+            sendMessage("selectWidget", message, delay)
         }
     }
 }

src/main/kotlin/irony/pycharm/qsseditor/QSSDocumentListener.kt

@@ -0,0 +1,12 @@
+package irony.pycharm.qsseditor
+
+import com.intellij.openapi.editor.event.DocumentEvent
+import com.intellij.openapi.editor.event.DocumentListener
+
+class QSSDocumentListener : DocumentListener {
+    override fun documentChanged(event: DocumentEvent) {
+        if (QSSState.instance::auto.get()) {
+            QSSClient.applyStyle(listOf(event.document.text), true)
+        }
+    }
+}

src/main/resources/META-INF/plugin.xml

@@ -50,11 +50,13 @@ Must Install Proxy Server from first(首先需要安装设计师插件)<br/><br/
         <applicationService
                 serviceImplementation="irony.pycharm.qsseditor.QSSState"/>
         <applicationConfigurable
-            parentId="language"
-            groupWeight="-500"
-            instance="irony.pycharm.qsseditor.QSSConfig"
-            id="irony.pycharm.qsseditor.QSSConfig"
-            displayName="QSS Editor"/>
+                parentId="language"
+                groupWeight="-500"
+                instance="irony.pycharm.qsseditor.QSSConfig"
+                id="irony.pycharm.qsseditor.QSSConfig"
+                displayName="QSS Editor"/>
+        <editorFactoryDocumentListener
+                implementation="irony.pycharm.qsseditor.QSSDocumentListener"/>
     </extensions>
 
     <applicationListeners>