feat(sagemaker): add progress indicator during space remote connection process (#8247)

## Problem
Users had no idea what was happening during connecting to remote space,
which can take few seconds to connect.
User clicks "Connect" → Nothing visible happens for few seconds → Either
success or failure

## Solution
Adds progress tracking for SageMaker space connections which shows
connection status with space name in progress dialog
Test cases are not being added because it is just showing the user about
progress during Space operations for good user experience.

## Testing 
Tested for both SM-AI and SMUS Spaces. 

1.
https://github.com/user-attachments/assets/20b02db3-7b6b-44d3-8e83-66d4e468da2c

2.
https://github.com/user-attachments/assets/e2196483-b935-45d3-ae00-86e4163d298e
---





- 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.

---------

Co-authored-by: Laxman Reddy <[email protected]>

Author: bhavya2109sharma

packages/core/src/awsService/sagemaker/commands.ts

@@ -191,32 +191,60 @@ export async function openRemoteConnect(
         void vscode.window.showErrorMessage(ConnectFromRemoteWorkspaceMessage)
         return
     }
+
+    const spaceName = node.spaceApp.SpaceName!
     await tryRefreshNode(node)
+
+    // for Stopped SM spaces - check instance type before showing progress
     if (node.getStatus() === 'Stopped') {
         //  In case of SMUS, we pass in a SM Client and for SM AI, it creates a new SM Client.
         const client = sageMakerClient ? sageMakerClient : new SagemakerClient(node.regionCode)
 
         try {
-            await client.startSpace(node.spaceApp.SpaceName!, node.spaceApp.DomainId!)
+            await client.startSpace(spaceName, node.spaceApp.DomainId!)
             await tryRefreshNode(node)
             const appType = node.spaceApp.SpaceSettingsSummary?.AppType
             if (!appType) {
                 throw new ToolkitError('AppType is undefined for the selected space. Cannot start remote connection.', {
                     code: 'undefinedAppType',
                 })
             }
-            await client.waitForAppInService(node.spaceApp.DomainId!, node.spaceApp.SpaceName!, appType)
-            await tryRemoteConnection(node, ctx)
+
+            // Only start showing progress after instance type validation
+            return await vscode.window.withProgress(
+                {
+                    location: vscode.ProgressLocation.Notification,
+                    cancellable: false,
+                    title: `Connecting to ${spaceName}`,
+                },
+                async (progress) => {
+                    progress.report({ message: 'Starting the space.' })
+                    await client.waitForAppInService(node.spaceApp.DomainId!, spaceName, appType)
+                    await tryRemoteConnection(node, ctx, progress)
+                }
+            )
         } catch (err: any) {
             // Ignore InstanceTypeError since it means the user decided not to use an instanceType with more memory
-            if (err.code !== InstanceTypeError) {
-                throw new ToolkitError(`Remote connection failed: ${(err as Error).message}`, {
-                    cause: err as Error,
-                    code: err.code,
-                })
+            // just return without showing progress
+            if (err.code === InstanceTypeError) {
+                return
             }
+            throw new ToolkitError(`Remote connection failed: ${(err as Error).message}`, {
+                cause: err as Error,
+                code: err.code,
+            })
         }
     } else if (node.getStatus() === 'Running') {
-        await tryRemoteConnection(node, ctx)
+        // For running spaces, show progress
+        return await vscode.window.withProgress(
+            {
+                location: vscode.ProgressLocation.Notification,
+                cancellable: false,
+                title: `Connecting to ${spaceName}`,
+            },
+            async (progress) => {
+                await tryRemoteConnection(node, ctx, progress)
+            }
+        )
     }
 }

packages/core/src/awsService/sagemaker/model.ts

@@ -27,12 +27,14 @@ const logger = getLogger('sagemaker')
 
 export async function tryRemoteConnection(
     node: SagemakerSpaceNode | SagemakerUnifiedStudioSpaceNode,
-    ctx: vscode.ExtensionContext
+    ctx: vscode.ExtensionContext,
+    progress: vscode.Progress<{ message?: string; increment?: number }>
 ) {
     const spaceArn = (await node.getSpaceArn()) as string
     const isSMUS = node instanceof SagemakerUnifiedStudioSpaceNode
     const remoteEnv = await prepareDevEnvConnection(spaceArn, ctx, 'sm_lc', isSMUS, node)
     try {
+        progress.report({ message: 'Opening remote session' })
         await startVscodeRemote(
             remoteEnv.SessionProcess,
             remoteEnv.hostname,