Merge branch 'master' into yair/expose_capabilities_to_chat

.chatversion

@@ -1 +1 @@
-v2.22.0
+v2.24.0

Common/src/main/java/com/tabnineCommon/binary/requests/fileLifecycle/Workspace.kt

@@ -0,0 +1,17 @@
+package com.tabnineCommon.binary.requests.fileLifecycle
+
+import com.google.gson.annotations.SerializedName
+import com.tabnineCommon.binary.BinaryRequest
+import com.tabnineCommon.binary.requests.EmptyResponse
+
+data class Workspace(
+    @SerializedName("root_paths") private val rootPaths: List<String>
+) : BinaryRequest<EmptyResponse> {
+    override fun response(): Class<EmptyResponse> {
+        return EmptyResponse::class.java
+    }
+
+    override fun serialize(): Any {
+        return mapOf("Workspace" to this)
+    }
+}

Common/src/main/java/com/tabnineCommon/chat/ChatMessagesRouter.kt

@@ -2,6 +2,7 @@ package com.tabnineCommon.chat
 
 import InitHandler
 import InsertAtCursorHandler
+import WorkspaceFoldersHandler
 import com.google.gson.JsonElement
 import com.intellij.openapi.diagnostic.Logger
 import com.intellij.openapi.project.Project
@@ -41,7 +42,8 @@ object ChatMessagesRouter {
         "get_settings" to GetChatSettingsHandler(gson),
         "update_settings" to UpdateChatSettingsHandler(gson),
         "get_server_url" to GetServerUrlHandler(gson),
-        "navigate_to_location" to NavigateToLocationHandler(gson)
+        "navigate_to_location" to NavigateToLocationHandler(gson),
+        "workspace_folders" to WorkspaceFoldersHandler(gson)
     )
 
     fun handleRawMessage(rawRequest: String, project: Project): String {

Common/src/main/java/com/tabnineCommon/chat/commandHandlers/WorkspaceFoldersHandler.kt

@@ -0,0 +1,20 @@
+import com.google.gson.Gson
+import com.google.gson.JsonElement
+import com.intellij.openapi.components.ServiceManager
+import com.intellij.openapi.project.Project
+import com.tabnineCommon.chat.commandHandlers.ChatMessageHandler
+import com.tabnineCommon.lifecycle.WorkspaceListenerService
+
+data class WorkspaceFoldersPayload(private val rootPaths: List<String>)
+
+class WorkspaceFoldersHandler(gson: Gson) : ChatMessageHandler<Unit, WorkspaceFoldersPayload>(gson) {
+    private val workspaceService = ServiceManager.getService(WorkspaceListenerService::class.java)
+
+    override fun handle(payload: Unit?, project: Project): WorkspaceFoldersPayload? {
+        val rootPaths = workspaceService.getWorkspaceRootPaths(project) ?: return null
+        return WorkspaceFoldersPayload(rootPaths)
+    }
+
+    override fun deserializeRequest(data: JsonElement?) {
+    }
+}

Common/src/main/java/com/tabnineCommon/lifecycle/LifecycleInitializer.kt

@@ -4,5 +4,6 @@ import com.intellij.openapi.components.ServiceManager
 
 fun initializeLifecycleEndpoints() {
     ServiceManager.getService(BinaryStateService::class.java).startUpdateLoop()
+    ServiceManager.getService(WorkspaceListenerService::class.java).start()
     TabnineFileEditorListener.registerListener()
 }

Common/src/main/java/com/tabnineCommon/lifecycle/WorkspaceListenerService.kt

@@ -0,0 +1,66 @@
+package com.tabnineCommon.lifecycle
+
+import com.intellij.openapi.components.Service
+import com.intellij.openapi.diagnostic.Logger
+import com.intellij.openapi.project.Project
+import com.intellij.openapi.project.ProjectManager
+import com.intellij.openapi.roots.ProjectRootManager
+import com.intellij.util.concurrency.AppExecutorUtil
+import com.tabnineCommon.binary.requests.fileLifecycle.Workspace
+import com.tabnineCommon.general.DependencyContainer
+import java.net.URL
+import java.util.concurrent.TimeUnit
+import java.util.concurrent.atomic.AtomicBoolean
+
+const val UPDATE_WORKSPACE_INITIAL_DELAY = 5L
+const val UPDATE_WORKSPACE_INTERVAL = 30L
+
+@Service
+class WorkspaceListenerService {
+    private val binaryRequestFacade = DependencyContainer.instanceOfBinaryRequestFacade()
+    private val scheduler = AppExecutorUtil.getAppScheduledExecutorService()
+    private val started = AtomicBoolean(false)
+
+    fun start() {
+        if (started.getAndSet(true)) return
+
+        scheduler.scheduleWithFixedDelay(
+            { updateWorkspaceRootPaths() },
+            UPDATE_WORKSPACE_INITIAL_DELAY,
+            UPDATE_WORKSPACE_INTERVAL,
+            TimeUnit.SECONDS
+        )
+    }
+
+    private fun updateWorkspaceRootPaths() {
+        val rootPaths = ProjectManager.getInstance()
+            .openProjects
+            .map { getWorkspaceRootPaths(it) ?: emptyList() }
+            .reduceOrNull { acc, cur -> acc.plus(cur) }
+
+        if (rootPaths.isNullOrEmpty()) return
+        Logger.getInstance(javaClass).info("All root paths collected: $rootPaths")
+
+        binaryRequestFacade.executeRequest(Workspace(rootPaths))
+    }
+
+    fun getWorkspaceRootPaths(project: Project): List<String>? {
+        if (project.isDisposed) {
+            Logger.getInstance(javaClass).warn("Project ${project.name} is disposed, skipping root paths resolution")
+            return null
+        }
+
+        val rootPaths = mutableListOf<String>()
+        for (contentRootUrl in ProjectRootManager.getInstance(project).contentRootUrls) {
+            val url = URL(contentRootUrl)
+            if (url.protocol != "file") {
+                Logger.getInstance(javaClass).debug("$url in project ${project.name} has unsupported protocol (${url.protocol})")
+                continue
+            }
+            rootPaths.add(url.path)
+        }
+
+        Logger.getInstance(javaClass).debug("Root paths for project ${project.name} found: $rootPaths")
+        return rootPaths
+    }
+}