feat(amazonq): Implement aws/syncModuleDependencies call (#5414)

* syncModuleDependencies call barebones

* implement syncModuleDependencies logic

* detekt

* return completableFuture

* ExtensionPoint impl

* detekt

* fix EachExtensionSafe

* update ExtensionPoint namespace

* private functions not needed

* detekt

* cs.launch for the ModuleDependencyServiceHandler

* Tests

* update xml

* move xml impl to src root

* feedback

* change message to notification instead of Request

Author: samgst-amazon

plugins/amazonq/shared/jetbrains-community/resources/META-INF/module-amazonq.xml

@@ -9,4 +9,5 @@
     <extensions defaultExtensionNs="com.intellij">
         <applicationService serviceImplementation="migration.software.aws.toolkits.jetbrains.services.codewhisperer.explorer.CodeWhispererExplorerActionManager"/>
     </extensions>
+
 </idea-plugin>

plugins/amazonq/shared/jetbrains-community/src/software/aws/toolkits/jetbrains/services/amazonq/lsp/AmazonQLanguageServer.kt

@@ -8,13 +8,17 @@ import org.eclipse.lsp4j.jsonrpc.services.JsonNotification
 import org.eclipse.lsp4j.jsonrpc.services.JsonRequest
 import org.eclipse.lsp4j.services.LanguageServer
 import software.aws.toolkits.jetbrains.services.amazonq.lsp.model.aws.credentials.UpdateCredentialsPayload
+import software.aws.toolkits.jetbrains.services.amazonq.lsp.model.aws.dependencies.SyncModuleDependenciesParams
 import java.util.concurrent.CompletableFuture
 
 /**
  * Remote interface exposed by the Amazon Q language server
  */
 @Suppress("unused")
 interface AmazonQLanguageServer : LanguageServer {
+    @JsonNotification("aws/syncModuleDependencies")
+    fun syncModuleDependencies(params: SyncModuleDependenciesParams): CompletableFuture<Unit>
+
     @JsonRequest("aws/credentials/token/update")
     fun updateTokenCredentials(payload: UpdateCredentialsPayload): CompletableFuture<ResponseMessage>
 

plugins/amazonq/shared/jetbrains-community/src/software/aws/toolkits/jetbrains/services/amazonq/lsp/AmazonQLspService.kt

@@ -23,6 +23,7 @@ import kotlinx.coroutines.CoroutineScope
 import kotlinx.coroutines.Deferred
 import kotlinx.coroutines.TimeoutCancellationException
 import kotlinx.coroutines.async
+import kotlinx.coroutines.launch
 import kotlinx.coroutines.runBlocking
 import kotlinx.coroutines.sync.Mutex
 import kotlinx.coroutines.sync.withLock
@@ -45,6 +46,7 @@ import software.aws.toolkits.core.utils.info
 import software.aws.toolkits.core.utils.warn
 import software.aws.toolkits.jetbrains.isDeveloperMode
 import software.aws.toolkits.jetbrains.services.amazonq.lsp.auth.DefaultAuthCredentialsService
+import software.aws.toolkits.jetbrains.services.amazonq.lsp.dependencies.DefaultModuleDependenciesService
 import software.aws.toolkits.jetbrains.services.amazonq.lsp.encryption.JwtEncryptionManager
 import software.aws.toolkits.jetbrains.services.amazonq.lsp.model.createExtendedClientMetadata
 import software.aws.toolkits.jetbrains.services.amazonq.lsp.textdocument.TextDocumentServiceHandler
@@ -310,6 +312,9 @@ private class AmazonQServerInstance(private val project: Project, private val cs
         DefaultAuthCredentialsService(project, encryptionManager, this)
         TextDocumentServiceHandler(project, this)
         WorkspaceServiceHandler(project, this)
+        cs.launch {
+            DefaultModuleDependenciesService(project, this@AmazonQServerInstance)
+        }
     }
 
     override fun dispose() {

plugins/amazonq/shared/jetbrains-community/src/software/aws/toolkits/jetbrains/services/amazonq/lsp/dependencies/DefaultModuleDependenciesService.kt

@@ -0,0 +1,52 @@
+// Copyright 2025 Amazon.com, Inc. or its affiliates. All Rights Reserved.
+// SPDX-License-Identifier: Apache-2.0
+
+package software.aws.toolkits.jetbrains.services.amazonq.lsp.dependencies
+
+import com.intellij.openapi.Disposable
+import com.intellij.openapi.module.ModuleManager
+import com.intellij.openapi.project.Project
+import com.intellij.openapi.roots.ModuleRootEvent
+import com.intellij.openapi.roots.ModuleRootListener
+import software.aws.toolkits.jetbrains.services.amazonq.lsp.AmazonQLspService
+import software.aws.toolkits.jetbrains.services.amazonq.lsp.dependencies.ModuleDependencyProvider.Companion.EP_NAME
+import software.aws.toolkits.jetbrains.services.amazonq.lsp.model.aws.dependencies.SyncModuleDependenciesParams
+import java.util.concurrent.CompletableFuture
+
+class DefaultModuleDependenciesService(
+    private val project: Project,
+    serverInstance: Disposable,
+) : ModuleDependenciesService,
+    ModuleRootListener {
+
+    init {
+        project.messageBus.connect(serverInstance).subscribe(
+            ModuleRootListener.TOPIC,
+            this
+        )
+        // project initiation with initial list of dependencies
+        syncAllModules()
+    }
+
+    override fun rootsChanged(event: ModuleRootEvent) {
+        if (event.isCausedByFileTypesChange) return
+        // call on change with updated dependencies
+        syncAllModules()
+    }
+
+    override fun syncModuleDependencies(params: SyncModuleDependenciesParams): CompletableFuture<Unit> =
+        AmazonQLspService.executeIfRunning(project) { languageServer ->
+            languageServer.syncModuleDependencies(params)
+        }?.toCompletableFuture() ?: CompletableFuture.failedFuture(IllegalStateException("LSP Server not running"))
+
+    private fun syncAllModules() {
+        ModuleManager.getInstance(project).modules.forEach { module ->
+            EP_NAME.forEachExtensionSafe {
+                if (it.isApplicable(module)) {
+                    syncModuleDependencies(it.createParams(module))
+                    return@forEachExtensionSafe
+                }
+            }
+        }
+    }
+}

plugins/amazonq/shared/jetbrains-community/src/software/aws/toolkits/jetbrains/services/amazonq/lsp/dependencies/ModuleDependenciesService.kt

@@ -0,0 +1,11 @@
+// Copyright 2025 Amazon.com, Inc. or its affiliates. All Rights Reserved.
+// SPDX-License-Identifier: Apache-2.0
+
+package software.aws.toolkits.jetbrains.services.amazonq.lsp.dependencies
+
+import software.aws.toolkits.jetbrains.services.amazonq.lsp.model.aws.dependencies.SyncModuleDependenciesParams
+import java.util.concurrent.CompletableFuture
+
+interface ModuleDependenciesService {
+    fun syncModuleDependencies(params: SyncModuleDependenciesParams): CompletableFuture<Unit>
+}

plugins/amazonq/shared/jetbrains-community/src/software/aws/toolkits/jetbrains/services/amazonq/lsp/dependencies/ModuleDependencyProvider.kt

@@ -0,0 +1,17 @@
+// Copyright 2025 Amazon.com, Inc. or its affiliates. All Rights Reserved.
+// SPDX-License-Identifier: Apache-2.0
+
+package software.aws.toolkits.jetbrains.services.amazonq.lsp.dependencies
+
+import com.intellij.openapi.extensions.ExtensionPointName
+import com.intellij.openapi.module.Module
+import software.aws.toolkits.jetbrains.services.amazonq.lsp.model.aws.dependencies.SyncModuleDependenciesParams
+
+interface ModuleDependencyProvider {
+    companion object {
+        val EP_NAME = ExtensionPointName<ModuleDependencyProvider>("software.aws.toolkits.jetbrains.moduleDependencyProvider")
+    }
+
+    fun isApplicable(module: Module): Boolean
+    fun createParams(module: Module): SyncModuleDependenciesParams
+}

plugins/amazonq/shared/jetbrains-community/src/software/aws/toolkits/jetbrains/services/amazonq/lsp/dependencies/providers/JavaModuleDependencyProvider.kt

@@ -0,0 +1,46 @@
+// Copyright 2025 Amazon.com, Inc. or its affiliates. All Rights Reserved.
+// SPDX-License-Identifier: Apache-2.0
+
+package software.aws.toolkits.jetbrains.services.amazonq.lsp.dependencies.providers
+
+import com.intellij.openapi.module.Module
+import com.intellij.openapi.projectRoots.JavaSdkType
+import com.intellij.openapi.roots.ModuleRootManager
+import com.intellij.openapi.roots.OrderRootType
+import com.intellij.openapi.vfs.VfsUtil
+import software.aws.toolkits.jetbrains.services.amazonq.lsp.dependencies.ModuleDependencyProvider
+import software.aws.toolkits.jetbrains.services.amazonq.lsp.model.aws.dependencies.SyncModuleDependenciesParams
+
+internal class JavaModuleDependencyProvider : ModuleDependencyProvider {
+    override fun isApplicable(module: Module): Boolean =
+        ModuleRootManager.getInstance(module).sdk?.sdkType is JavaSdkType
+
+    override fun createParams(module: Module): SyncModuleDependenciesParams {
+        val sourceRoots = getSourceRoots(module)
+        val dependencies = mutableListOf<String>()
+
+        ModuleRootManager.getInstance(module).orderEntries().forEachLibrary { library ->
+            library.getUrls(OrderRootType.CLASSES).forEach { url ->
+                dependencies.add(VfsUtil.urlToPath(url))
+            }
+            true
+        }
+
+        return SyncModuleDependenciesParams(
+            moduleName = module.name,
+            programmingLanguage = "Java",
+            files = sourceRoots,
+            dirs = dependencies,
+            includePatterns = emptyList(),
+            excludePatterns = emptyList()
+        )
+    }
+
+    private fun getSourceRoots(module: Module): List<String> =
+        ModuleRootManager.getInstance(module).contentEntries
+            .flatMap { contentEntry ->
+                contentEntry.sourceFolders
+                    .filter { !it.isTestSource }
+                    .mapNotNull { it.file?.path }
+            }
+}

plugins/amazonq/shared/jetbrains-community/src/software/aws/toolkits/jetbrains/services/amazonq/lsp/dependencies/providers/PythonModuleDependencyProvider.kt

@@ -0,0 +1,45 @@
+// Copyright 2025 Amazon.com, Inc. or its affiliates. All Rights Reserved.
+// SPDX-License-Identifier: Apache-2.0
+
+package software.aws.toolkits.jetbrains.services.amazonq.lsp.dependencies.providers
+
+import com.intellij.openapi.module.Module
+import com.intellij.openapi.roots.ModuleRootManager
+import com.jetbrains.python.packaging.management.PythonPackageManager
+import com.jetbrains.python.sdk.PythonSdkUtil
+import software.aws.toolkits.jetbrains.services.amazonq.lsp.dependencies.ModuleDependencyProvider
+import software.aws.toolkits.jetbrains.services.amazonq.lsp.model.aws.dependencies.SyncModuleDependenciesParams
+
+internal class PythonModuleDependencyProvider : ModuleDependencyProvider {
+    override fun isApplicable(module: Module): Boolean =
+        PythonSdkUtil.findPythonSdk(module) != null
+
+    override fun createParams(module: Module): SyncModuleDependenciesParams {
+        val sourceRoots = getSourceRoots(module)
+        val dependencies = mutableListOf<String>()
+
+        PythonSdkUtil.findPythonSdk(module)?.let { sdk ->
+            val packageManager = PythonPackageManager.forSdk(module.project, sdk)
+            packageManager.installedPackages.forEach { pkg ->
+                dependencies.add(pkg.name)
+            }
+        }
+
+        return SyncModuleDependenciesParams(
+            moduleName = module.name,
+            programmingLanguage = "Python",
+            files = sourceRoots,
+            dirs = dependencies,
+            includePatterns = emptyList(),
+            excludePatterns = emptyList()
+        )
+    }
+
+    private fun getSourceRoots(module: Module): List<String> =
+        ModuleRootManager.getInstance(module).contentEntries
+            .flatMap { contentEntry ->
+                contentEntry.sourceFolders
+                    .filter { !it.isTestSource }
+                    .mapNotNull { it.file?.path }
+            }
+}

plugins/amazonq/shared/jetbrains-community/src/software/aws/toolkits/jetbrains/services/amazonq/lsp/model/aws/dependencies/SyncModuleDependenciesParams.kt

@@ -0,0 +1,13 @@
+// Copyright 2025 Amazon.com, Inc. or its affiliates. All Rights Reserved.
+// SPDX-License-Identifier: Apache-2.0
+
+package software.aws.toolkits.jetbrains.services.amazonq.lsp.model.aws.dependencies
+
+class SyncModuleDependenciesParams(
+    val moduleName: String,
+    val programmingLanguage: String,
+    val files: List<String>,
+    val dirs: List<String>,
+    val includePatterns: List<String>,
+    val excludePatterns: List<String>,
+)