feat: Adding ATX-FES apis support for transformation

app/aws-lsp-antlr4-runtimes/package.json

@@ -12,7 +12,7 @@
         "webpack": "webpack"
     },
     "dependencies": {
-        "@aws/language-server-runtimes": "0.3.6",
+        "@aws/language-server-runtimes": "0.3.7",
         "@aws/lsp-antlr4": "*",
         "antlr4-c3": "^3.4.2",
         "antlr4ng": "^3.0.14"

app/aws-lsp-buildspec-runtimes/package.json

@@ -7,7 +7,7 @@
         "compile": "tsc --build"
     },
     "dependencies": {
-        "@aws/language-server-runtimes": "0.3.6",
+        "@aws/language-server-runtimes": "0.3.7",
         "@aws/lsp-buildspec": "^0.0.1"
     }
 }

app/aws-lsp-cloudformation-runtimes/package.json

@@ -7,7 +7,7 @@
         "compile": "tsc --build"
     },
     "dependencies": {
-        "@aws/language-server-runtimes": "0.3.6",
+        "@aws/language-server-runtimes": "0.3.7",
         "@aws/lsp-cloudformation": "^0.0.1"
     }
 }

app/aws-lsp-codewhisperer-runtimes/package.json

@@ -23,7 +23,7 @@
         "local-build": "node scripts/local-build.js"
     },
     "dependencies": {
-        "@aws/language-server-runtimes": "0.3.6",
+        "@aws/language-server-runtimes": "0.3.7",
         "@aws/lsp-codewhisperer": "*",
         "copyfiles": "^2.4.1",
         "cross-env": "^7.0.3",

app/aws-lsp-codewhisperer-runtimes/src/agent-standalone.ts

@@ -6,6 +6,8 @@ import {
     CodeWhispererServer,
     QAgenticChatServerProxy,
     QConfigurationServerTokenProxy,
+    TransformConfigurationServerTokenProxy,
+    AtxNetTransformServerTokenProxy,
     QLocalProjectContextServerProxy,
     QNetTransformServerTokenProxy,
     WorkspaceContextServerTokenProxy,
@@ -28,6 +30,8 @@ const props = {
         CodeWhispererServer,
         CodeWhispererSecurityScanServerTokenProxy,
         QConfigurationServerTokenProxy,
+        TransformConfigurationServerTokenProxy,
+        AtxNetTransformServerTokenProxy,
         QNetTransformServerTokenProxy,
         QAgenticChatServerProxy,
         IdentityServer.create,

app/aws-lsp-codewhisperer-runtimes/src/iam-standalone.ts

@@ -2,6 +2,6 @@ import { standalone } from '@aws/language-server-runtimes/runtimes'
 import { CodeWhispererServerIAM, QChatServerIAMProxy } from '@aws/lsp-codewhisperer'
 import { createIAMRuntimeProps } from './standalone-common'
 
-const props = createIAMRuntimeProps('0.1.0', [CodeWhispererServerIAM, QChatServerIAMProxy])
+const props = createIAMRuntimeProps('0.1.0', [CodeWhispererServerIAM as any, QChatServerIAMProxy as any])
 
 standalone(props)

app/aws-lsp-codewhisperer-runtimes/src/iam-webworker.ts

@@ -8,7 +8,7 @@ import { AmazonQServiceServerIAM } from '@aws/lsp-codewhisperer/out/shared/amazo
 // to inject the AmazonQServiceServer does not work for webworker as it triggers missing polyfill errors
 const props: RuntimeProps = {
     version: '1.0.0',
-    servers: [AmazonQServiceServerIAM, CodeWhispererServerIAM, QChatServerIAM],
+    servers: [AmazonQServiceServerIAM as any, CodeWhispererServerIAM as any, QChatServerIAM as any],
     name: 'AWS CodeWhisperer',
 }
 

app/aws-lsp-codewhisperer-runtimes/src/standalone-common.ts

@@ -8,10 +8,10 @@ const createRuntimePropsFactory =
     (version: string, servers: Server[], name = 'AWS CodeWhisperer'): RuntimeProps => {
         return {
             version,
-            servers: [AmazonQServiceServer, ...servers],
+            servers: [AmazonQServiceServer as any, ...servers],
             name,
         }
     }
 
-export const createIAMRuntimeProps = createRuntimePropsFactory(AmazonQServiceServerIAM)
-export const createTokenRuntimeProps = createRuntimePropsFactory(AmazonQServiceServerToken)
+export const createIAMRuntimeProps = createRuntimePropsFactory(AmazonQServiceServerIAM as any)
+export const createTokenRuntimeProps = createRuntimePropsFactory(AmazonQServiceServerToken as any)

app/aws-lsp-codewhisperer-runtimes/src/token-standalone.ts

@@ -4,6 +4,8 @@ import {
     CodeWhispererServerTokenProxy,
     QChatServerTokenProxy,
     QConfigurationServerTokenProxy,
+    TransformConfigurationServerTokenProxy,
+    AtxNetTransformServerTokenProxy,
     QNetTransformServerTokenProxy,
     QLocalProjectContextServerProxy,
     WorkspaceContextServerTokenProxy,
@@ -20,6 +22,8 @@ const props = createTokenRuntimeProps(VERSION, [
     CodeWhispererServerTokenProxy,
     CodeWhispererSecurityScanServerTokenProxy,
     QConfigurationServerTokenProxy,
+    TransformConfigurationServerTokenProxy,
+    AtxNetTransformServerTokenProxy as any,
     QNetTransformServerTokenProxy,
     QChatServerTokenProxy,
     IdentityServer.create,

app/aws-lsp-identity-runtimes/package.json

@@ -7,7 +7,7 @@
         "compile": "tsc --build"
     },
     "dependencies": {
-        "@aws/language-server-runtimes": "0.3.6",
+        "@aws/language-server-runtimes": "0.3.7",
         "@aws/lsp-identity": "^0.0.1"
     }
 }

app/aws-lsp-json-runtimes/package.json

@@ -11,7 +11,7 @@
         "webpack": "webpack"
     },
     "dependencies": {
-        "@aws/language-server-runtimes": "0.3.6",
+        "@aws/language-server-runtimes": "0.3.7",
         "@aws/lsp-json": "*"
     },
     "devDependencies": {

app/aws-lsp-notification-runtimes/package.json

@@ -7,7 +7,7 @@
         "compile": "tsc --build"
     },
     "dependencies": {
-        "@aws/language-server-runtimes": "0.3.6",
+        "@aws/language-server-runtimes": "0.3.7",
         "@aws/lsp-notification": "^0.0.1"
     }
 }

app/aws-lsp-partiql-runtimes/package.json

@@ -11,7 +11,7 @@
         "package": "npm run compile && npm run compile:webpack"
     },
     "dependencies": {
-        "@aws/language-server-runtimes": "0.3.6",
+        "@aws/language-server-runtimes": "0.3.7",
         "@aws/lsp-partiql": "0.0.19"
     },
     "devDependencies": {

app/aws-lsp-s3-runtimes/package.json

@@ -10,7 +10,7 @@
         "compile": "tsc --build"
     },
     "dependencies": {
-        "@aws/language-server-runtimes": "0.3.6",
+        "@aws/language-server-runtimes": "0.3.7",
         "@aws/lsp-s3": "^0.0.1"
     }
 }

app/aws-lsp-yaml-json-webworker/package.json

@@ -11,7 +11,7 @@
         "serve:webpack": "NODE_ENV=development webpack serve"
     },
     "dependencies": {
-        "@aws/language-server-runtimes": "0.3.6",
+        "@aws/language-server-runtimes": "0.3.7",
         "@aws/lsp-json": "*",
         "@aws/lsp-yaml": "*"
     },

app/aws-lsp-yaml-runtimes/package.json

@@ -11,7 +11,7 @@
         "webpack": "webpack"
     },
     "dependencies": {
-        "@aws/language-server-runtimes": "0.3.6",
+        "@aws/language-server-runtimes": "0.3.7",
         "@aws/lsp-yaml": "*"
     },
     "devDependencies": {

app/hello-world-lsp-runtimes/package.json

@@ -15,7 +15,7 @@
     },
     "dependencies": {
         "@aws/hello-world-lsp": "^0.0.1",
-        "@aws/language-server-runtimes": "0.3.6"
+        "@aws/language-server-runtimes": "0.3.7"
     },
     "devDependencies": {
         "@types/chai": "^4.3.5",

chat-client/package.json

@@ -25,7 +25,7 @@
     },
     "dependencies": {
         "@aws/chat-client-ui-types": "0.1.68",
-        "@aws/language-server-runtimes": "0.3.6",
+        "@aws/language-server-runtimes": "0.3.7",
         "@aws/language-server-runtimes-types": "0.1.62",
         "@aws/mynah-ui": "^4.38.0"
     },

client/vscode/package.json

@@ -352,7 +352,7 @@
         "@aws-sdk/credential-providers": "^3.731.1",
         "@aws-sdk/types": "^3.734.0",
         "@aws/chat-client-ui-types": "0.1.68",
-        "@aws/language-server-runtimes": "0.3.6",
+        "@aws/language-server-runtimes": "0.3.7",
         "@types/uuid": "^9.0.8",
         "@types/vscode": "^1.98.0",
         "jose": "^5.2.4",

core/aws-lsp-core/package.json

@@ -28,7 +28,7 @@
         "prepack": "shx cp ../../LICENSE ../../NOTICE ../../SECURITY.md ."
     },
     "dependencies": {
-        "@aws/language-server-runtimes": "0.3.6",
+        "@aws/language-server-runtimes": "0.3.7",
         "@gerhobbelt/gitignore-parser": "^0.2.0-9",
         "cross-spawn": "7.0.6",
         "jose": "^5.2.4",

core/aws-lsp-core/src/credentials/ideCredentialsProvider.test.ts

@@ -21,6 +21,7 @@ describe('IdeCredentialsProvider', function () {
                 warn: sinon.stub(),
                 error: sinon.stub(),
             },
+            onRequest: sinon.stub(),
         } as any
         provider = new IdeCredentialsProvider(mockConnection as any)
     })
@@ -45,4 +46,46 @@ describe('IdeCredentialsProvider', function () {
             )
         })
     })
+
+    describe('initialize', function () {
+        it('registers credential push handlers when IAM credentials are provided', function () {
+            const props = {
+                credentials: {
+                    providesIam: true,
+                },
+            }
+
+            provider.initialize(props)
+
+            assert(mockConnection.onRequest.called)
+        })
+
+        it('does not register handlers when no credentials config provided', function () {
+            const props = {}
+
+            provider.initialize(props)
+
+            assert(!mockConnection.onRequest.called)
+        })
+    })
+
+    describe('hasCredentials', function () {
+        it('returns false when no credentials are set', function () {
+            // IdeCredentialsProvider doesn't expose hasCredentials directly
+            // Test through resolveIamCredentials instead
+            assert.strictEqual(provider['pushedCredentials'], undefined)
+            assert.strictEqual(provider['pushedToken'], undefined)
+        })
+    })
+
+    describe('getCredentials', function () {
+        it('throws NoCredentialsError when no credentials available', async function () {
+            try {
+                await provider.resolveIamCredentials({} as any)
+                throw new Error('Expected error was not thrown')
+            } catch (error) {
+                assert.strictEqual((error as Error).constructor.name, 'NoCredentialsError')
+            }
+        })
+    })
 })

core/aws-lsp-core/src/credentials/ideCredentialsProvider.ts

@@ -17,6 +17,9 @@ export class IdeCredentialsProvider implements CredentialsProvider {
     private pushedCredentials: IamCredentials | undefined
     private pushedToken: BearerToken | undefined
 
+    private pushedAtxCredentials: IamCredentials | undefined
+    private pushedAtxToken: BearerToken | undefined
+
     constructor(
         private readonly connection: Connection,
         key?: string,
@@ -165,6 +168,28 @@ export class IdeCredentialsProvider implements CredentialsProvider {
         return this.pushedToken
     }
 
+    /**
+     * Provides IAM based Atx credentials. Throws NoCredentialsError if no credentials are available
+     */
+    public async resolveIamAtxCredentials(token: CancellationToken): Promise<IamCredentials> {
+        if (!this.pushedAtxCredentials) {
+            throw new NoCredentialsError()
+        }
+
+        return this.pushedAtxCredentials
+    }
+
+    /**
+     * Provides a atx bearer token. Throws NoCredentialsError if bearer token is not available
+     */
+    public async resolveAtxBearerToken(token: CancellationToken): Promise<BearerToken> {
+        if (!this.pushedAtxToken) {
+            throw new NoCredentialsError()
+        }
+
+        return this.pushedAtxToken
+    }
+
     private async decodeCredentialsRequestToken<T>(request: UpdateCredentialsRequest): Promise<T> {
         if (!this.key) {
             throw new Error('no encryption key')

integration-tests/q-agentic-chat-server/package.json

@@ -9,7 +9,7 @@
         "test-integ": "npm run compile && mocha --timeout 30000 \"./out/**/*.test.js\" --retries 2"
     },
     "dependencies": {
-        "@aws/language-server-runtimes": "0.3.6",
+        "@aws/language-server-runtimes": "0.3.7",
         "@aws/lsp-core": "*"
     },
     "devDependencies": {