feat(amazonq): Implement aws/syncModuleDependencies call

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

@@ -0,0 +1,9 @@
+<!-- Copyright 2025 Amazon.com, Inc. or its affiliates. All Rights Reserved. -->
+<!-- SPDX-License-Identifier: Apache-2.0 -->
+
+<idea-plugin>
+    <extensions defaultExtensionNs="com.intellij">
+        <moduleDependencyProvider
+            implementation="software.aws.toolkits.jetbrains.services.amazonq.lsp.dependencies.providers.JavaModuleDependencyProvider"/>
+    </extensions>
+</idea-plugin>

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

@@ -6,7 +6,17 @@
     <xi:include href="/META-INF/plugin-codetransform.xml" />
     <xi:include href="/META-INF/plugin-codewhisperer.xml" />
 
+    <extensionPoints>
+        <extensionPoint name="moduleDependencyProvider"
+                        interface="software.aws.toolkits.jetbrains.services.amazonq.lsp.dependencies.ModuleDependencyProvider"
+                        dynamic="true"/>
+    </extensionPoints>
+
     <extensions defaultExtensionNs="com.intellij">
         <applicationService serviceImplementation="migration.software.aws.toolkits.jetbrains.services.codewhisperer.explorer.CodeWhispererExplorerActionManager"/>
     </extensions>
+
+    <depends optional="true" config-file="java-support.xml">com.intellij.modules.java</depends>
+    <depends optional="true" config-file="python-support.xml">com.intellij.modules.python</depends>
+
 </idea-plugin>

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

@@ -0,0 +1,9 @@
+<!-- Copyright 2025 Amazon.com, Inc. or its affiliates. All Rights Reserved. -->
+<!-- SPDX-License-Identifier: Apache-2.0 -->
+
+<idea-plugin>
+    <extensions defaultExtensionNs="com.intellij">
+        <moduleDependencyProvider
+            implementation="software.aws.toolkits.jetbrains.services.amazonq.lsp.dependencies.providers.PythonModuleDependencyProvider"/>
+    </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 {
+    @JsonRequest("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

@@ -44,6 +44,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
@@ -301,6 +302,7 @@ private class AmazonQServerInstance(private val project: Project, private val cs
         }
 
         DefaultAuthCredentialsService(project, encryptionManager, this)
+        DefaultModuleDependenciesService(project, this)
         TextDocumentServiceHandler(project, this)
         WorkspaceServiceHandler(project, this)
     }

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

@@ -0,0 +1,120 @@
+// 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.Module
+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 com.intellij.openapi.roots.ModuleRootManager
+import com.intellij.openapi.roots.OrderRootType
+import com.intellij.openapi.vfs.VfsUtil
+import com.jetbrains.python.packaging.management.PythonPackageManager
+import com.jetbrains.python.sdk.PythonSdkUtil
+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> =
+        CompletableFuture<Unit>().also { completableFuture ->
+            AmazonQLspService.executeIfRunning(project) { languageServer ->
+                languageServer.syncModuleDependencies(params)
+                completableFuture.complete(null)
+            } ?: completableFuture.completeExceptionally(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
+                }
+            }
+        }
+    }
+
+    private fun getSourceRoots(module: Module): List<String> {
+        val sourceRoots = mutableListOf<String>()
+
+        // Get all source roots from content entries
+        ModuleRootManager.getInstance(module).contentEntries
+            .flatMap { contentEntry ->
+                contentEntry.sourceFolders
+                    .filter { !it.isTestSource }
+                    .mapNotNull { it.file?.path }
+            }
+            .forEach { path -> sourceRoots.add(path) }
+
+        return sourceRoots
+    }
+
+    private fun createJavaParams(module: Module): SyncModuleDependenciesParams {
+        val sourceRoots = getSourceRoots(module)
+        val dependencies = mutableListOf<String>()
+
+        // Get library dependencies
+        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 createPythonParams(module: Module): SyncModuleDependenciesParams {
+        val sourceRoots = getSourceRoots(module)
+        val dependencies = mutableListOf<String>()
+
+        // Get Python packages
+        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()
+        )
+    }
+}

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>("com.intellij.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,47 @@
+// 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
+
+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> {
+        return 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,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.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
+
+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> {
+        return 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
+
+public class SyncModuleDependenciesParams(
+    val moduleName: String,
+    val programmingLanguage: String,
+    val files: List<String>,
+    val dirs: List<String>,
+    val includePatterns: List<String>,
+    val excludePatterns: List<String>,
+)

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

@@ -0,0 +1,22 @@
+// 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.util
+
+import com.intellij.diagnostic.PluginException
+import com.intellij.openapi.diagnostic.Logger
+import com.intellij.openapi.extensions.ExtensionPointName
+
+val LOG = Logger.getInstance("software.aws.toolkits.jetbrains.services.amazonq.lsp.util.ExtensionPointUtils")
+
+inline fun <T : Any> ExtensionPointName<T>.forEachExtensionSafe(action: (T) -> Unit) {
+    for (extension in extensionList) {
+        try {
+            action(extension)
+        } catch (e: PluginException) {
+            LOG.warn("Failed to process extension ${extension::class.java.name}", e)
+        } catch (e: Exception) {
+            LOG.warn("Error processing extension ${extension::class.java.name}", e)
+        }
+    }
+}