feat(smus): List connections under compute node (aws#2193)

__Description:__
Add 2 folder nodes under compute node to list Redshit and Spark
connections.
Add tooltip for the listed connections to show critical information.

## Appearance:
<img width="360" height="304" alt="image"
src="https://github.com/user-attachments/assets/4cb3ca4c-14d2-4373-af1d-35147238895b"
/>

---

- Treat all work as PUBLIC. Private `feature/x` branches will not be
squash-merged at release time.
- Your code changes must meet the guidelines in
[CONTRIBUTING.md](https://github.com/aws/aws-toolkit-vscode/blob/master/CONTRIBUTING.md#guidelines).
- License: I confirm that my contribution is made under the terms of the
Apache 2.0 license.

Author: PotatoWKY

packages/core/src/sagemakerunifiedstudio/explorer/nodes/sageMakerUnifiedStudioComputeNode.ts

@@ -10,6 +10,8 @@ import { SageMakerUnifiedStudioSpacesParentNode } from './sageMakerUnifiedStudio
 import { SageMakerUnifiedStudioProjectNode } from './sageMakerUnifiedStudioProjectNode'
 import { SagemakerClient } from '../../../shared/clients/sagemaker'
 import { SmusAuthenticationProvider } from '../../auth/providers/smusAuthenticationProvider'
+import { SageMakerUnifiedStudioConnectionParentNode } from './sageMakerUnifiedStudioConnectionParentNode'
+import { ConnectionType } from '@aws-sdk/client-datazone'
 
 export class SageMakerUnifiedStudioComputeNode implements TreeNode {
     public readonly id = 'smusComputeNode'
@@ -19,7 +21,7 @@ export class SageMakerUnifiedStudioComputeNode implements TreeNode {
     constructor(
         public readonly parent: SageMakerUnifiedStudioProjectNode,
         private readonly extensionContext: vscode.ExtensionContext,
-        private readonly authprovider: SmusAuthenticationProvider,
+        public readonly authProvider: SmusAuthenticationProvider,
         private readonly sagemakerClient: SagemakerClient
     ) {}
 
@@ -35,11 +37,17 @@ export class SageMakerUnifiedStudioComputeNode implements TreeNode {
         const projectId = this.parent.getProject()?.id
 
         if (projectId) {
+            childrenNodes.push(
+                new SageMakerUnifiedStudioConnectionParentNode(this, ConnectionType.REDSHIFT, 'Data warehouse')
+            )
+            childrenNodes.push(
+                new SageMakerUnifiedStudioConnectionParentNode(this, ConnectionType.SPARK, 'Data processing')
+            )
             this.spacesNode = new SageMakerUnifiedStudioSpacesParentNode(
                 this,
                 projectId,
                 this.extensionContext,
-                this.authprovider,
+                this.authProvider,
                 this.sagemakerClient
             )
             childrenNodes.push(this.spacesNode)

packages/core/src/sagemakerunifiedstudio/explorer/nodes/sageMakerUnifiedStudioConnectionNode.ts

@@ -0,0 +1,63 @@
+/*!
+ * Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+ * SPDX-License-Identifier: Apache-2.0
+ */
+
+import * as vscode from 'vscode'
+import { getLogger } from '../../../shared/logger/logger'
+import { TreeNode } from '../../../shared/treeview/resourceTreeDataProvider'
+import { SageMakerUnifiedStudioConnectionParentNode } from './sageMakerUnifiedStudioConnectionParentNode'
+import { ConnectionSummary, ConnectionType } from '@aws-sdk/client-datazone'
+
+export class SageMakerUnifiedStudioConnectionNode implements TreeNode {
+    public resource: SageMakerUnifiedStudioConnectionNode
+    contextValue: string
+    private readonly logger = getLogger()
+    id: string
+    public constructor(
+        private readonly parent: SageMakerUnifiedStudioConnectionParentNode,
+        private readonly connection: ConnectionSummary
+    ) {
+        this.id = connection.name ?? ''
+        this.resource = this
+        this.contextValue = this.getContext()
+        this.logger.debug(`SageMaker Space Node created: ${this.id}`)
+    }
+
+    public async getTreeItem(): Promise<vscode.TreeItem> {
+        const item = new vscode.TreeItem(this.id, vscode.TreeItemCollapsibleState.None)
+        item.contextValue = this.getContext()
+        item.tooltip = new vscode.MarkdownString(this.buildTooltip())
+        return item
+    }
+    private buildTooltip(): string {
+        if (this.connection.type === ConnectionType.REDSHIFT) {
+            const tooltip = ''.concat(
+                '### Compute Details\n\n',
+                `**Type**  \n${this.connection.type}\n\n`,
+                `**Environment ID**  \n${this.connection.environmentId}\n\n`,
+                `**JDBC URL**  \n${this.connection.props?.redshiftProperties?.jdbcUrl}`
+            )
+            return tooltip
+        } else if (this.connection.type === ConnectionType.SPARK) {
+            const tooltip = ''.concat(
+                '### Compute Details\n\n',
+                `**Type**  \n${this.connection.type}\n\n`,
+                `**Glue version**  \n${this.connection.props?.sparkGlueProperties?.glueVersion}\n\n`,
+                `**Worker type**  \n${this.connection.props?.sparkGlueProperties?.workerType}\n\n`,
+                `**Number of workers**  \n${this.connection.props?.sparkGlueProperties?.numberOfWorkers}\n\n`,
+                `**Idle timeout (minutes)**  \n${this.connection.props?.sparkGlueProperties?.idleTimeout}\n\n`
+            )
+            return tooltip
+        } else {
+            return ''
+        }
+    }
+    private getContext(): string {
+        return 'SageMakerUnifiedStudioConnectionNode'
+    }
+
+    public getParent(): TreeNode | undefined {
+        return this.parent
+    }
+}

packages/core/src/sagemakerunifiedstudio/explorer/nodes/sageMakerUnifiedStudioConnectionParentNode.ts

@@ -0,0 +1,57 @@
+/*!
+ * Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+ * SPDX-License-Identifier: Apache-2.0
+ */
+
+import * as vscode from 'vscode'
+import { SageMakerUnifiedStudioComputeNode } from './sageMakerUnifiedStudioComputeNode'
+import { TreeNode } from '../../../shared/treeview/resourceTreeDataProvider'
+import { ListConnectionsCommandOutput, ConnectionType } from '@aws-sdk/client-datazone'
+import { SageMakerUnifiedStudioConnectionNode } from './sageMakerUnifiedStudioConnectionNode'
+import { DataZoneClient } from '../../shared/client/datazoneClient'
+
+// eslint-disable-next-line id-length
+export class SageMakerUnifiedStudioConnectionParentNode implements TreeNode {
+    public resource: SageMakerUnifiedStudioConnectionParentNode
+    contextValue: string
+    public connections: ListConnectionsCommandOutput | undefined
+    public constructor(
+        private readonly parent: SageMakerUnifiedStudioComputeNode,
+        private readonly connectionType: ConnectionType,
+        public id: string
+    ) {
+        this.resource = this
+        this.contextValue = this.getContext()
+    }
+
+    public async getTreeItem(): Promise<vscode.TreeItem> {
+        const item = new vscode.TreeItem(this.id, vscode.TreeItemCollapsibleState.Collapsed)
+        item.contextValue = this.getContext()
+        return item
+    }
+
+    public async getChildren(): Promise<TreeNode[]> {
+        const client = await DataZoneClient.getInstance(this.parent.authProvider)
+        this.connections = await client.fetchConnections(
+            this.parent.parent.project?.domainId,
+            this.parent.parent.project?.id,
+            this.connectionType
+        )
+        const childrenNodes = []
+        if (!this.connections?.items) {
+            return []
+        }
+        for (const connection of this.connections.items) {
+            childrenNodes.push(new SageMakerUnifiedStudioConnectionNode(this, connection))
+        }
+        return childrenNodes
+    }
+
+    private getContext(): string {
+        return 'SageMakerUnifiedStudioConnectionParentNode'
+    }
+
+    public getParent(): TreeNode | undefined {
+        return this.parent
+    }
+}

packages/core/src/sagemakerunifiedstudio/explorer/nodes/sageMakerUnifiedStudioProjectNode.ts

@@ -24,7 +24,7 @@ export class SageMakerUnifiedStudioProjectNode implements TreeNode {
     private readonly onDidChangeEmitter = new vscode.EventEmitter<void>()
     public readonly onDidChangeTreeItem = this.onDidChangeEmitter.event
     public readonly onDidChangeChildren = this.onDidChangeEmitter.event
-    private project?: DataZoneProject
+    public project?: DataZoneProject
     private logger = getLogger()
     private sagemakerClient?: SagemakerClient
 

packages/core/src/sagemakerunifiedstudio/shared/client/datazoneClient.ts

@@ -13,6 +13,7 @@ import {
     PhysicalEndpoint,
     RedshiftPropertiesOutput,
     S3PropertiesOutput,
+    ConnectionType,
 } from '@aws-sdk/client-datazone'
 import { getLogger } from '../../../shared/logger/logger'
 import type { SmusAuthenticationProvider } from '../../auth/providers/smusAuthenticationProvider'
@@ -476,6 +477,18 @@ export class DataZoneClient {
         }
     }
 
+    public async fetchConnections(
+        domain: string | undefined,
+        project: string | undefined,
+        ConnectionType: ConnectionType
+    ): Promise<ListConnectionsCommandOutput> {
+        const datazoneClient = await this.getDataZoneClient()
+        return datazoneClient.listConnections({
+            domainIdentifier: domain,
+            projectIdentifier: project,
+            type: ConnectionType,
+        })
+    }
     /**
      * Lists connections in a DataZone environment
      * @param domainId The DataZone domain identifier

packages/core/src/test/sagemakerunifiedstudio/explorer/nodes/sageMakerUnifiedStudioComputeNode.test.ts

@@ -71,17 +71,16 @@ describe('SageMakerUnifiedStudioComputeNode', function () {
             assert.deepStrictEqual(children, [])
         })
 
-        it('returns spaces node when project is selected', async function () {
+        it('returns connection nodes and spaces node when project is selected', async function () {
             const mockProject = { id: 'project-123', name: 'Test Project' }
             ;(mockParent.getProject as sinon.SinonStub).returns(mockProject)
 
-            // Stub the SpacesParentNode constructor to prevent actual instantiation
-            sinon.stub(SageMakerUnifiedStudioSpacesParentNode.prototype, 'constructor' as any).returns({})
-
             const children = await computeNode.getChildren()
 
-            assert.strictEqual(children.length, 1)
-            assert.ok(children[0] instanceof SageMakerUnifiedStudioSpacesParentNode)
+            assert.strictEqual(children.length, 3)
+            assert.strictEqual(children[0].id, 'Data warehouse')
+            assert.strictEqual(children[1].id, 'Data processing')
+            assert.ok(children[2] instanceof SageMakerUnifiedStudioSpacesParentNode)
         })
     })
 

packages/core/src/test/sagemakerunifiedstudio/explorer/nodes/sageMakerUnifiedStudioConnectionNode.test.ts

@@ -0,0 +1,144 @@
+/*!
+ * Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+ * SPDX-License-Identifier: Apache-2.0
+ */
+
+import assert from 'assert'
+import sinon from 'sinon'
+import * as vscode from 'vscode'
+import { SageMakerUnifiedStudioConnectionNode } from '../../../../sagemakerunifiedstudio/explorer/nodes/sageMakerUnifiedStudioConnectionNode'
+import { SageMakerUnifiedStudioConnectionParentNode } from '../../../../sagemakerunifiedstudio/explorer/nodes/sageMakerUnifiedStudioConnectionParentNode'
+import { ConnectionType, ConnectionSummary } from '@aws-sdk/client-datazone'
+import { getLogger } from '../../../../shared/logger/logger'
+
+describe('SageMakerUnifiedStudioConnectionNode', function () {
+    let connectionNode: SageMakerUnifiedStudioConnectionNode
+    let mockParent: sinon.SinonStubbedInstance<SageMakerUnifiedStudioConnectionParentNode>
+
+    const mockRedshiftConnection: ConnectionSummary = {
+        connectionId: 'conn-1',
+        name: 'Test Redshift Connection',
+        type: ConnectionType.REDSHIFT,
+        environmentId: 'env-1',
+        domainId: 'domain-1',
+        domainUnitId: 'unit-1',
+        physicalEndpoints: [],
+        props: {
+            redshiftProperties: {
+                jdbcUrl: 'jdbc:redshift://test-cluster:5439/testdb',
+            },
+        },
+    }
+
+    const mockSparkConnection: ConnectionSummary = {
+        connectionId: 'conn-2',
+        name: 'Test Spark Connection',
+        type: ConnectionType.SPARK,
+        environmentId: 'env-2',
+        domainId: 'domain-2',
+        domainUnitId: 'unit-2',
+        physicalEndpoints: [],
+        props: {
+            sparkGlueProperties: {
+                glueVersion: '4.0',
+                workerType: 'G.1X',
+                numberOfWorkers: 2,
+                idleTimeout: 30,
+            },
+        },
+    }
+
+    beforeEach(function () {
+        mockParent = {} as any
+        sinon.stub(getLogger(), 'debug')
+    })
+
+    afterEach(function () {
+        sinon.restore()
+    })
+
+    describe('constructor', function () {
+        it('creates instance with correct properties for Redshift connection', function () {
+            connectionNode = new SageMakerUnifiedStudioConnectionNode(mockParent as any, mockRedshiftConnection)
+
+            assert.strictEqual(connectionNode.id, 'Test Redshift Connection')
+            assert.strictEqual(connectionNode.resource, connectionNode)
+            assert.strictEqual(connectionNode.contextValue, 'SageMakerUnifiedStudioConnectionNode')
+        })
+
+        it('creates instance with empty id when connection name is undefined', function () {
+            const connectionWithoutName = { ...mockRedshiftConnection, name: undefined }
+            connectionNode = new SageMakerUnifiedStudioConnectionNode(mockParent as any, connectionWithoutName)
+
+            assert.strictEqual(connectionNode.id, '')
+        })
+    })
+
+    describe('getTreeItem', function () {
+        it('returns correct tree item for Redshift connection', async function () {
+            connectionNode = new SageMakerUnifiedStudioConnectionNode(mockParent as any, mockRedshiftConnection)
+
+            const treeItem = await connectionNode.getTreeItem()
+
+            assert.strictEqual(treeItem.label, 'Test Redshift Connection')
+            assert.strictEqual(treeItem.collapsibleState, vscode.TreeItemCollapsibleState.None)
+            assert.strictEqual(treeItem.contextValue, 'SageMakerUnifiedStudioConnectionNode')
+            assert.ok(treeItem.tooltip instanceof vscode.MarkdownString)
+        })
+
+        it('returns correct tree item for Spark connection', async function () {
+            connectionNode = new SageMakerUnifiedStudioConnectionNode(mockParent as any, mockSparkConnection)
+
+            const treeItem = await connectionNode.getTreeItem()
+
+            assert.strictEqual(treeItem.label, 'Test Spark Connection')
+            assert.ok(treeItem.tooltip instanceof vscode.MarkdownString)
+        })
+    })
+
+    describe('tooltip generation', function () {
+        it('generates correct tooltip for Redshift connection', async function () {
+            connectionNode = new SageMakerUnifiedStudioConnectionNode(mockParent as any, mockRedshiftConnection)
+
+            const treeItem = await connectionNode.getTreeItem()
+            const tooltip = (treeItem.tooltip as vscode.MarkdownString).value
+
+            assert(tooltip.includes('REDSHIFT'))
+            assert(tooltip.includes('env-1'))
+            assert(tooltip.includes('jdbc:redshift://test-cluster:5439/testdb'))
+        })
+
+        it('generates correct tooltip for Spark connection', async function () {
+            connectionNode = new SageMakerUnifiedStudioConnectionNode(mockParent as any, mockSparkConnection)
+
+            const treeItem = await connectionNode.getTreeItem()
+            const tooltip = (treeItem.tooltip as vscode.MarkdownString).value
+
+            assert(tooltip.includes('SPARK'))
+            assert(tooltip.includes('4.0'))
+            assert(tooltip.includes('G.1X'))
+            assert(tooltip.includes('2'))
+            assert(tooltip.includes('30'))
+        })
+
+        it('generates empty tooltip for unknown connection type', async function () {
+            const unknownConnection = { ...mockRedshiftConnection, type: 'UNKNOWN' as ConnectionType }
+            connectionNode = new SageMakerUnifiedStudioConnectionNode(mockParent as any, unknownConnection)
+
+            const treeItem = await connectionNode.getTreeItem()
+            const tooltip = (treeItem.tooltip as vscode.MarkdownString).value
+
+            assert.strictEqual(tooltip, '')
+        })
+    })
+
+    describe('getParent', function () {
+        it('returns the parent node', function () {
+            connectionNode = new SageMakerUnifiedStudioConnectionNode(mockParent as any, mockRedshiftConnection)
+
+            const parent = connectionNode.getParent()
+
+            assert.strictEqual(parent, mockParent)
+        })
+    })
+})