Merge branch 'feature/cloudformation' into feature/cloudformation

laileni-aws · web-flow · commit e19e4760dce6 · 2025-11-11T12:55:13.000-08:00
diff --git a/package-lock.json b/package-lock.json
diff --git a/packages/amazonq/.changes/1.103.0.json b/packages/amazonq/.changes/1.103.0.json
@@ -0,0 +1,10 @@
+{
+	"date": "2025-11-06",
+	"version": "1.103.0",
+	"entries": [
+		{
+			"type": "Feature",
+			"description": "Q CodeTransformation: add more job metadata to history table"
+		}
+	]
+}
diff --git a/packages/amazonq/.changes/next-release/Feature-ab31cbb6-3fe4-4ee3-a0a3-290430277856.json b/packages/amazonq/.changes/next-release/Feature-ab31cbb6-3fe4-4ee3-a0a3-290430277856.json
diff --git a/packages/amazonq/CHANGELOG.md b/packages/amazonq/CHANGELOG.md
@@ -1,3 +1,7 @@
+## 1.103.0 2025-11-06
+
+- **Feature** Q CodeTransformation: add more job metadata to history table
+
 ## 1.102.0 2025-10-30
 
 - Miscellaneous non-user-facing changes
diff --git a/packages/amazonq/package.json b/packages/amazonq/package.json
@@ -2,7 +2,7 @@
     "name": "amazon-q-vscode",
     "displayName": "Amazon Q",
     "description": "The most capable generative AI–powered assistant for software development.",
-    "version": "1.103.0-SNAPSHOT",
+    "version": "1.104.0-SNAPSHOT",
     "extensionKind": [
         "workspace"
     ],
diff --git a/packages/core/src/awsService/sagemaker/commands.ts b/packages/core/src/awsService/sagemaker/commands.ts
@@ -171,7 +171,7 @@ export async function stopSpace(
         await client.deleteApp({
             DomainId: node.spaceApp.DomainId!,
             SpaceName: spaceName,
-            AppType: node.spaceApp.App!.AppType!,
+            AppType: node.spaceApp.SpaceSettingsSummary!.AppType!,
             AppName: node.spaceApp.App?.AppName,
         })
     } catch (err) {
@@ -319,7 +319,7 @@ async function handleRunningSpaceWithDisabledAccess(
                 await client.deleteApp({
                     DomainId: node.spaceApp.DomainId!,
                     SpaceName: spaceName,
-                    AppType: node.spaceApp.App!.AppType!,
+                    AppType: node.spaceApp.SpaceSettingsSummary!.AppType!,
                     AppName: node.spaceApp.App?.AppName,
                 })
 
@@ -329,7 +329,11 @@ async function handleRunningSpaceWithDisabledAccess(
                 // Start the space with remote access enabled (skip prompts since user already consented)
                 await client.startSpace(spaceName, node.spaceApp.DomainId!, true)
                 await tryRefreshNode(node)
-                await client.waitForAppInService(node.spaceApp.DomainId!, spaceName, node.spaceApp.App!.AppType!)
+                await client.waitForAppInService(
+                    node.spaceApp.DomainId!,
+                    spaceName,
+                    node.spaceApp.SpaceSettingsSummary!.AppType!
+                )
                 await tryRemoteConnection(node, ctx, progress)
             } catch (err: any) {
                 // Handle user declining instance type upgrade
@@ -369,7 +373,11 @@ async function handleStoppedSpace(
             },
             async (progress) => {
                 progress.report({ message: 'Starting the space' })
-                await client.waitForAppInService(node.spaceApp.DomainId!, spaceName, node.spaceApp.App!.AppType!)
+                await client.waitForAppInService(
+                    node.spaceApp.DomainId!,
+                    spaceName,
+                    node.spaceApp.SpaceSettingsSummary!.AppType!
+                )
                 await tryRemoteConnection(node, ctx, progress)
             }
         )
diff --git a/packages/core/src/awsService/sagemaker/sagemakerSpace.ts b/packages/core/src/awsService/sagemaker/sagemakerSpace.ts
@@ -11,8 +11,11 @@ import { getIcon, IconPath } from '../../shared/icons'
 import { generateSpaceStatus, updateIdleFile, startMonitoringTerminalActivity, ActivityCheckInterval } from './utils'
 import { UserActivity } from '../../shared/extensionUtilities'
 import { getLogger } from '../../shared/logger/logger'
+import { ToolkitError } from '../../shared/errors'
 import { SpaceStatus, RemoteAccess } from './constants'
 
+const logger = getLogger('sagemaker')
+
 export class SagemakerSpace {
     public label: string = ''
     public contextValue: string = ''
@@ -34,6 +37,10 @@ export class SagemakerSpace {
     }
 
     public updateSpace(spaceApp: SagemakerSpaceApp) {
+        // Edge case when this.spaceApp.App is null, returned by ListApp API for a Space that is not connected to for over 24 hours
+        if (!this.spaceApp.App) {
+            this.spaceApp.App = spaceApp.App
+        }
         this.setSpaceStatus(spaceApp.Status ?? '', spaceApp.App?.Status ?? '')
         // Only update RemoteAccess property to minimize impact due to minor structural differences between variables
         if (this.spaceApp.SpaceSettingsSummary && spaceApp.SpaceSettingsSummary?.RemoteAccess) {
@@ -107,13 +114,19 @@ export class SagemakerSpace {
             DomainId: this.spaceApp.DomainId,
             SpaceName: this.spaceApp.SpaceName,
         })
-
-        const app = await this.client.describeApp({
-            DomainId: this.spaceApp.DomainId,
-            AppName: this.spaceApp.App?.AppName,
-            AppType: this.spaceApp?.SpaceSettingsSummary?.AppType,
-            SpaceName: this.spaceApp.SpaceName,
-        })
+        // get app using ListApps API, with given DomainId and SpaceName
+        const app =
+            this.spaceApp.DomainId && this.spaceApp.SpaceName
+                ? await this.client.listAppForSpace(this.spaceApp.DomainId, this.spaceApp.SpaceName)
+                : undefined
+        if (!app) {
+            logger.error(
+                `updateSpaceAppStatus: unable to get app, [DomainId: ${this.spaceApp.DomainId}], [SpaceName: ${this.spaceApp.SpaceName}]`
+            )
+            throw new ToolkitError(
+                `Cannot update app status without [DomainId: ${this.spaceApp.DomainId} and SpaceName: ${this.spaceApp.SpaceName}]`
+            )
+        }
 
         // AWS DescribeSpace API returns full details with property names like 'SpaceSettings'
         // but our internal SagemakerSpaceApp type expects 'SpaceSettingsSummary' (from ListSpaces API)
@@ -195,7 +208,6 @@ export class SagemakerSpace {
  * Sets up user activity monitoring for SageMaker spaces
  */
 export async function setupUserActivityMonitoring(extensionContext: vscode.ExtensionContext): Promise<void> {
-    const logger = getLogger()
     logger.info('setupUserActivityMonitoring: Starting user activity monitoring setup')
 
     const tmpDirectory = '/tmp/'
diff --git a/packages/core/src/shared/clients/sagemaker.ts b/packages/core/src/shared/clients/sagemaker.ts
@@ -128,6 +128,13 @@ export class SagemakerClient extends ClientWrapper<SageMakerClient> {
         return this.makeRequest(DeleteAppCommand, request)
     }
 
+    public async listAppForSpace(domainId: string, spaceName: string): Promise<AppDetails | undefined> {
+        const appsList = await this.listApps({ DomainIdEquals: domainId, SpaceNameEquals: spaceName })
+            .flatten()
+            .promise()
+        return appsList[0] // At most one App for one SagemakerSpace
+    }
+
     public async startSpace(spaceName: string, domainId: string, skipInstanceTypePrompts: boolean = false) {
         let spaceDetails: DescribeSpaceCommandOutput
 
diff --git a/packages/core/src/test/awsService/sagemaker/commands.test.ts b/packages/core/src/test/awsService/sagemaker/commands.test.ts
@@ -59,6 +59,7 @@ describe('SageMaker Commands', () => {
                         AppName: 'default',
                     },
                     SpaceSettingsSummary: {
+                        AppType: 'JupyterLab',
                         RemoteAccess: 'DISABLED',
                     },
                 },
diff --git a/packages/core/src/test/awsService/sagemaker/explorer/sagemakerSpaceNode.test.ts b/packages/core/src/test/awsService/sagemaker/explorer/sagemakerSpaceNode.test.ts
@@ -112,11 +112,13 @@ describe('SagemakerSpaceNode', function () {
     it('updates space app status', async function () {
         const describeSpaceStub = sinon.stub(SagemakerClient.prototype, 'describeSpace')
         describeSpaceStub.resolves({ SpaceName: 'TestSpace', Status: 'InService', $metadata: {} })
-        describeAppStub.resolves({ AppName: 'TestApp', Status: 'InService', $metadata: {} })
+
+        const listAppForSpaceStub = sinon.stub(SagemakerClient.prototype, 'listAppForSpace')
+        listAppForSpaceStub.resolves({ AppName: 'TestApp', Status: 'InService' })
 
         await testSpaceAppNode.updateSpaceAppStatus()
 
         sinon.assert.calledOnce(describeSpaceStub)
-        sinon.assert.calledOnce(describeAppStub)
+        sinon.assert.calledOnce(listAppForSpaceStub)
     })
 })
diff --git a/packages/core/src/test/awsService/sagemaker/sagemakerSpace.test.ts b/packages/core/src/test/awsService/sagemaker/sagemakerSpace.test.ts
@@ -63,6 +63,7 @@ describe('SagemakerSpace', function () {
 
             mockClient.describeSpace.resolves(mockDescribeSpaceResponse)
             mockClient.describeApp.resolves(mockDescribeAppResponse)
+            mockClient.listAppForSpace.resolves(mockDescribeAppResponse)
 
             const space = new SagemakerSpace(mockClient as any, 'us-east-1', mockSpaceApp)
             const updateSpaceSpy = sinon.spy(space, 'updateSpace')
@@ -107,9 +108,8 @@ describe('SagemakerSpace', function () {
                 Status: 'InService',
                 $metadata: { requestId: 'test-request-id' },
             }
-
+            mockClient.listAppForSpace.resolves(mockDescribeAppResponse)
             mockClient.describeSpace.resolves(mockDescribeSpaceResponse)
-            mockClient.describeApp.resolves(mockDescribeAppResponse)
 
             const space = new SagemakerSpace(mockClient as any, 'us-east-1', mockSpaceApp)
             const updateSpaceSpy = sinon.spy(space, 'updateSpace')
@@ -125,5 +125,37 @@ describe('SagemakerSpace', function () {
             assert.strictEqual(updateSpaceArgs.OwnershipSettingsSummary, undefined)
             assert.strictEqual(updateSpaceArgs.SpaceSharingSettingsSummary, undefined)
         })
+
+        it('should update app status using listAppForSpace', async function () {
+            const mockDescribeSpaceResponse = {
+                SpaceName: 'test-space',
+                Status: 'InService',
+                DomainId: 'test-domain',
+                $metadata: { requestId: 'test-request-id' },
+            }
+
+            const mockAppFromList = {
+                AppName: 'listed-app',
+                Status: 'InService',
+                $metadata: { requestId: 'test-request-id' },
+            }
+
+            mockClient.describeSpace.resolves(mockDescribeSpaceResponse)
+            mockClient.listAppForSpace.resolves(mockAppFromList)
+
+            // Create space without App.AppName
+            const spaceWithoutAppName: SagemakerSpaceApp = {
+                ...mockSpaceApp,
+                App: undefined,
+            }
+
+            const space = new SagemakerSpace(mockClient as any, 'us-east-1', spaceWithoutAppName)
+            await space.updateSpaceAppStatus()
+
+            // Verify listAppForSpace was called instead of describeApp
+            assert.ok(mockClient.listAppForSpace.calledOnce)
+            assert.ok(mockClient.listAppForSpace.calledWith('test-domain', 'test-space'))
+            assert.ok(mockClient.describeApp.notCalled)
+        })
     })
 })
diff --git a/packages/core/src/test/shared/clients/sagemakerClient.test.ts b/packages/core/src/test/shared/clients/sagemakerClient.test.ts
@@ -173,6 +173,41 @@ describe('SagemakerClient.listSpaceApps', function () {
     })
 })
 
+describe('SagemakerClient.listAppForSpace', function () {
+    const region = 'test-region'
+    let client: SagemakerClient
+    let listAppsStub: sinon.SinonStub
+
+    beforeEach(function () {
+        client = new SagemakerClient(region)
+        listAppsStub = sinon.stub(client, 'listApps')
+    })
+
+    afterEach(function () {
+        sinon.restore()
+    })
+
+    it('returns first app for given domain and space', async function () {
+        const appDetails: AppDetails[] = [
+            { AppName: 'app1', DomainId: 'domain1', SpaceName: 'space1', AppType: AppType.CodeEditor },
+        ]
+        listAppsStub.returns(intoCollection([appDetails]))
+
+        const result = await client.listAppForSpace('domain1', 'space1')
+
+        assert.strictEqual(result?.AppName, 'app1')
+        sinon.assert.calledWith(listAppsStub, { DomainIdEquals: 'domain1', SpaceNameEquals: 'space1' })
+    })
+
+    it('returns undefined when no apps found', async function () {
+        listAppsStub.returns(intoCollection([[]]))
+
+        const result = await client.listAppForSpace('domain1', 'space1')
+
+        assert.strictEqual(result, undefined)
+    })
+})
+
 describe('SagemakerClient.waitForAppInService', function () {
     const region = 'test-region'
     let client: SagemakerClient
diff --git a/packages/toolkit/.changes/3.83.0.json b/packages/toolkit/.changes/3.83.0.json
@@ -0,0 +1,5 @@
+{
+	"date": "2025-11-06",
+	"version": "3.83.0",
+	"entries": []
+}
diff --git a/packages/toolkit/CHANGELOG.md b/packages/toolkit/CHANGELOG.md
@@ -1,3 +1,7 @@
+## 3.83.0 2025-11-06
+
+- Miscellaneous non-user-facing changes
+
 ## 3.82.0 2025-10-30
 
 - **Feature** Lambda AppBuilder: Now you can install Finch from the AppBuilder walkthrough
diff --git a/packages/toolkit/package.json b/packages/toolkit/package.json
@@ -2,7 +2,7 @@
     "name": "aws-toolkit-vscode",
     "displayName": "AWS Toolkit",
     "description": "Including CodeCatalyst, Infrastructure Composer, and support for Lambda, S3, CloudWatch Logs, CloudFormation, and many other services.",
-    "version": "3.83.0-SNAPSHOT",
+    "version": "3.84.0-SNAPSHOT",
     "extensionKind": [
         "workspace"
     ],
