Merge pull request #109 from wecode-ai/fix_kilo_error

add MainThreadStatusBarShape

Author: qdaxb

jetbrains_plugin/src/main/kotlin/com/sina/weibo/agent/actors/MainThreadLanguageFeaturesShape.kt

@@ -36,7 +36,7 @@ interface MainThreadLanguageFeaturesShape : Disposable {
      * @param selector Document selector
      * @param eventHandle Event handle
      */
-    fun registerCodeLensSupport(handle: Int, selector: List<Map<String, Any?>>, eventHandle: Int?)
+    fun registerCodeLensSupport(handle: Number, selector: List<Map<String, Any?>>, eventHandle: Number?)
 
     /**
      * Emits code lens event.
@@ -459,7 +459,7 @@ class MainThreadLanguageFeatures : MainThreadLanguageFeaturesShape {
         logger.info("Registering document symbol provider: handle=$handle, selector=$selector, label=$label")
     }
 
-    override fun registerCodeLensSupport(handle: Int, selector: List<Map<String, Any?>>, eventHandle: Int?) {
+    override fun registerCodeLensSupport(handle: Number, selector: List<Map<String, Any?>>, eventHandle: Number?) {
         logger.info("Registering code lens support: handle=$handle, selector=$selector, eventHandle=$eventHandle")
     }
 

jetbrains_plugin/src/main/kotlin/com/sina/weibo/agent/actors/MainThreadStatusBarShape.kt

@@ -0,0 +1,62 @@
+// SPDX-FileCopyrightText: 2025 Weibo, Inc.
+//
+// SPDX-License-Identifier: Apache-2.0
+
+package com.sina.weibo.agent.actors
+
+import com.intellij.openapi.Disposable
+import com.intellij.openapi.diagnostic.Logger
+import java.net.URI
+
+/**
+ * Main thread search service interface.
+ * Corresponds to the MainThreadSearchShape interface in VSCode.
+ */
+interface MainThreadStatusBarShape : Disposable {
+    fun setEntry(
+        entryId: String,
+        id: String,
+        extensionId: String?,
+        name: String,
+        text: String,
+        tooltip: Any?,
+        hasTooltipProvider: Boolean,
+        command: Map<Any, Any>?,
+        color: Any?,
+        backgroundColor: Any?,
+        alignLeft: Boolean,
+        priority: Number?,
+        accessibilityInformation: Any?
+    )
+
+}
+
+/**
+ * Implementation of the main thread search service.
+ * Provides search-related functionality for the IDEA platform.
+ */
+class MainThreadStatusBar : MainThreadStatusBarShape {
+    private val logger = Logger.getInstance(MainThreadStatusBar::class.java)
+    override fun setEntry(
+        entryId: String,
+        id: String,
+        extensionId: String?,
+        name: String,
+        text: String,
+        tooltip: Any?,
+        hasTooltipProvider: Boolean,
+        command: Map<Any, Any>?,
+        color: Any?,
+        backgroundColor: Any?,
+        alignLeft: Boolean,
+        priority: Number?,
+        accessibilityInformation: Any?
+    ) {
+        logger.info("Set entry $entryId")
+    }
+
+    override fun dispose() {
+        logger.info("Disposing main thread")
+    }
+
+} 

jetbrains_plugin/src/main/kotlin/com/sina/weibo/agent/core/RPCManager.kt

@@ -34,9 +34,11 @@ class RPCManager(
         setupExtensionRequiredProtocols()
         setupWeCodeRequiredProtocols()
         setupRooCodeFuncitonProtocols()
+        setupKiloCodeFunctionProtocols()
         setupWebviewProtocols()
     }
 
+
     /**
      * Start initializing plugin environment
      * Send configuration and workspace information to extension process
@@ -141,6 +143,7 @@ class RPCManager(
 
         // MainThreadConfiguration
         rpcProtocol.set(ServiceProxyRegistry.MainContext.MainThreadConfiguration, MainThreadConfiguration())
+
     }
 
     /**
@@ -191,6 +194,7 @@ class RPCManager(
 
         //MainThreadDocuments
         rpcProtocol.set(ServiceProxyRegistry.MainContext.MainThreadDocuments, MainThreadDocuments(project))
+
     }
 
     /**
@@ -242,6 +246,10 @@ class RPCManager(
         // MainThreadSecretState
         rpcProtocol.set(ServiceProxyRegistry.MainContext.MainThreadSecretState, MainThreadSecretState())
     }
+    private fun setupKiloCodeFunctionProtocols() {
+        // MainThreadStatusBar
+        rpcProtocol.set(ServiceProxyRegistry.MainContext.MainThreadStatusBar, MainThreadStatusBar())
+    }
 
     private fun setupWebviewProtocols() {
         logger.info("Setting up protocol handlers required for Webview")

jetbrains_plugin/src/main/kotlin/com/sina/weibo/agent/core/ServiceProxyRegistry.kt

@@ -228,7 +228,7 @@ class ServiceProxyRegistry private constructor() {
         val MainThreadProgress = createProxyIdentifier<Any>("MainThreadProgress")
         val MainThreadQuickDiff = createProxyIdentifier<Any>("MainThreadQuickDiff")
         val MainThreadQuickOpen = createProxyIdentifier<Any>("MainThreadQuickOpen")
-        val MainThreadStatusBar = createProxyIdentifier<Any>("MainThreadStatusBar")
+        val MainThreadStatusBar = createProxyIdentifier<MainThreadStatusBarShape>("MainThreadStatusBar")
         val MainThreadSecretState = createProxyIdentifier<MainThreadSecretStateShape>("MainThreadSecretState")
         val MainThreadStorage = createProxyIdentifier<MainThreadStorageShape>("MainThreadStorage")
         val MainThreadSpeech = createProxyIdentifier<Any>("MainThreadSpeechProvider")

jetbrains_plugin/src/main/kotlin/com/sina/weibo/agent/ipc/proxy/RPCProtocol.kt

@@ -680,7 +680,11 @@ class RPCProtocol(
      * Execute handler invocation
      */
     private suspend fun doInvokeHandler(rpcId: Int, methodName: String, args: List<Any?>): Any? {
-        val actor = locals[rpcId] ?: throw IllegalStateException("Unknown actor ${getStringIdentifierForProxy(rpcId)}")
+        val actor = locals[rpcId]
+        if(actor == null) {
+            LOG.error("Unknown actor ${getStringIdentifierForProxy(rpcId)}")
+            return  null
+        }
        // Use reflection to get method with parameter type matching
        val method = try {
            findBestMatchingMethod(actor, methodName, args)