Add logic to fetch dev endpoint from local package (#176)

* Add logic to fetch dev endpoint from local package

* Use initialization options for endpoint config

---------

Co-authored-by: Kevin DeJong <kddejong@amazon.com>

Author: rongyamazon

src/server/CfnExternal.ts

@@ -36,7 +36,7 @@ export class CfnExternal implements Configurables, Closeable {
     readonly featureFlags: FeatureFlagProvider;
 
     constructor(lsp: LspComponents, core: CfnInfraCore, overrides: Partial<CfnExternal> = {}) {
-        this.awsClient = overrides.awsClient ?? new AwsClient(core.awsCredentials);
+        this.awsClient = overrides.awsClient ?? new AwsClient(core.awsCredentials, core.cloudformationEndpoint);
 
         this.cfnService = overrides.cfnService ?? new CfnService(this.awsClient);
         this.ccapiService = overrides.ccapiService ?? new CcapiService(this.awsClient);

src/server/CfnInfraCore.ts

@@ -16,6 +16,11 @@ import { Configurable, Configurables } from '../utils/Configurable';
 
 interface ExtendedInitializeParams extends InitializeParams {
     initializationOptions?: {
+        aws?: {
+            cloudformation?: {
+                endpoint?: string;
+            };
+        };
         encryption?: {
             key?: string;
         };
@@ -40,6 +45,7 @@ export class CfnInfraCore implements Configurables, Closeable {
 
     readonly awsCredentials: AwsCredentials;
     readonly diagnosticCoordinator: DiagnosticCoordinator;
+    readonly cloudformationEndpoint?: string;
 
     constructor(
         lspComponents: LspComponents,
@@ -59,6 +65,7 @@ export class CfnInfraCore implements Configurables, Closeable {
         this.contextManager = overrides.contextManager ?? new ContextManager(this.syntaxTreeManager);
         this.fileContextManager = overrides.fileContextManager ?? new FileContextManager(this.documentManager);
 
+        this.cloudformationEndpoint = initializeParams.initializationOptions?.aws?.cloudformation?.endpoint;
         const encryptionKey = initializeParams.initializationOptions?.encryption?.key;
 
         this.awsCredentials =

src/services/AwsClient.ts

@@ -9,10 +9,14 @@ type IamClientConfig = {
     region: string;
     credentials: IamCredentials;
     customUserAgent: string;
+    endpoint?: string;
 };
 
 export class AwsClient {
-    constructor(private readonly credentialsProvider: AwsCredentials) {}
+    constructor(
+        private readonly credentialsProvider: AwsCredentials,
+        private readonly cloudformationEndpoint?: string,
+    ) {}
 
     // By default, clients will retry on throttling exceptions 3 times
     public getCloudFormationClient() {
@@ -34,6 +38,7 @@ export class AwsClient {
                 region: credential.region,
                 credentials: credential,
                 customUserAgent: `${ExtensionId}/${ExtensionVersion}`,
+                endpoint: this.cloudformationEndpoint,
             };
         } catch {
             throw new Error('AWS credentials not configured. Authentication required for online features.');