feat: added dev override for registry URL

Author: ashishrp-aws

packages/amazonq/src/lsp/activation.ts

@@ -7,19 +7,29 @@ import vscode from 'vscode'
 import { startLanguageServer } from './client'
 import { AmazonQLspInstaller, getBundledResourcePaths } from './lspInstaller'
 import { lspSetupStage, ToolkitError, messages, getLogger } from 'aws-core-vscode/shared'
+import { getAmazonQLspConfig } from './config'
 
 export async function activate(ctx: vscode.ExtensionContext): Promise<void> {
+    const config = getAmazonQLspConfig()
+    const registryUrl = config.registryUrl
+
     try {
         await lspSetupStage('all', async () => {
             const installResult = await new AmazonQLspInstaller().resolve()
-            await lspSetupStage('launch', async () => await startLanguageServer(ctx, installResult.resourcePaths))
+            await lspSetupStage(
+                'launch',
+                async () => await startLanguageServer(ctx, installResult.resourcePaths, registryUrl)
+            )
         })
     } catch (err) {
         const e = err as ToolkitError
         getLogger('amazonqLsp').warn(`Failed to start downloaded LSP, falling back to bundled LSP: ${e.message}`)
         try {
             await lspSetupStage('all', async () => {
-                await lspSetupStage('launch', async () => await startLanguageServer(ctx, getBundledResourcePaths(ctx)))
+                await lspSetupStage(
+                    'launch',
+                    async () => await startLanguageServer(ctx, getBundledResourcePaths(ctx), registryUrl)
+                )
             })
         } catch (error) {
             void messages.showViewLogsMessage(

packages/amazonq/src/lsp/client.ts

@@ -83,7 +83,8 @@ export function hasGlibcPatch(): boolean {
 
 export async function startLanguageServer(
     extensionContext: vscode.ExtensionContext,
-    resourcePaths: AmazonQResourcePaths
+    resourcePaths: AmazonQResourcePaths,
+    registryUrl?: string
 ) {
     const toDispose = extensionContext.subscriptions
 
@@ -188,6 +189,7 @@ export async function startLanguageServer(
                         codeReviewInChat: codeReviewInChat,
                         // feature flag for displaying findings found not through CodeReview in the Code Issues Panel
                         displayFindings: true,
+                        registryUrl: registryUrl,
                     },
                     window: {
                         notifications: true,

packages/amazonq/src/lsp/config.ts

@@ -12,6 +12,7 @@ import {
 
 export interface ExtendedAmazonQLSPConfig extends BaseLspInstaller.LspConfig {
     ui?: string
+    registryUrl?: string
 }
 
 // Taken from language server runtimes since they are not exported:
@@ -44,10 +45,12 @@ export const defaultAmazonQLspConfig: ExtendedAmazonQLSPConfig = {
 }
 
 export function getAmazonQLspConfig(): ExtendedAmazonQLSPConfig {
+    const registryConfig = DevSettings.instance.get('amazonqRegistry', {})
     return {
         ...defaultAmazonQLspConfig,
         ...(DevSettings.instance.getServiceConfig('amazonqLsp', {}) as ExtendedAmazonQLSPConfig),
         ...getServiceEnvVarConfig('amazonqLsp', Object.keys(defaultAmazonQLspConfig)),
+        registryUrl: (registryConfig as any)?.registryUrl,
     }
 }
 /**

packages/core/src/shared/settings.ts

@@ -778,6 +778,7 @@ const devSettings = {
     codewhispererService: Record(String, String),
     amazonqLsp: Record(String, String),
     amazonqWorkspaceLsp: Record(String, String),
+    amazonqRegistry: Record(String, String),
     ssoCacheDirectory: String,
     autofillStartUrl: String,
     webAuth: Boolean,