add codeeditor subdomain

Author: aws-asolidu

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

@@ -85,7 +85,8 @@ export async function deeplinkConnect(
     session: string,
     wsUrl: string,
     token: string,
-    domain: string
+    domain: string,
+    appType: string
 ) {
     getLogger().debug(
         `sm:deeplinkConnect: connectionIdentifier: ${connectionIdentifier} session: ${session} wsUrl: ${wsUrl} token: ${token}`
@@ -104,7 +105,8 @@ export async function deeplinkConnect(
             session,
             wsUrl,
             token,
-            domain
+            domain,
+            appType
         )
 
         await startVscodeRemote(

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

@@ -79,15 +79,16 @@ export async function persistSSMConnection(
     domain: string,
     session?: string,
     wsUrl?: string,
-    token?: string
+    token?: string,
+    appType?: string
 ): Promise<void> {
     const { region } = parseArn(appArn)
     const endpoint = DevSettings.instance.get('endpoints', {})['sagemaker'] ?? ''
 
-    // TODO: Hardcoded to 'jupyterlab' due to a bug in Studio that only supports refreshing
-    // the token for both CodeEditor and JupyterLab Apps in the jupyterlab subdomain.
-    // This will be fixed shortly after NYSummit launch to support refresh URL in CodeEditor subdomain.
-    const appSubDomain = 'jupyterlab'
+    let appSubDomain = 'jupyterlab'
+    if (appType && appType.toLowerCase() === 'codeeditor') {
+        appSubDomain = 'code-editor'
+    }
 
     let envSubdomain: string
 

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

@@ -50,7 +50,8 @@ export async function prepareDevEnvConnection(
     session?: string,
     wsUrl?: string,
     token?: string,
-    domain?: string
+    domain?: string,
+    appType?: string
 ) {
     const remoteLogger = configureRemoteConnectionLogger()
     const { ssm, vsc, ssh } = (await ensureDependencies()).unwrap()
@@ -72,7 +73,7 @@ export async function prepareDevEnvConnection(
     if (connectionType === 'sm_lc') {
         await persistLocalCredentials(appArn)
     } else if (connectionType === 'sm_dl') {
-        await persistSSMConnection(appArn, domain ?? '', session, wsUrl, token)
+        await persistSSMConnection(appArn, domain ?? '', session, wsUrl, token, appType)
     }
 
     await startLocalServer(ctx)

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

@@ -18,7 +18,8 @@ export function register(ctx: ExtContext) {
                 params.session,
                 `${params.ws_url}&cell-number=${params['cell-number']}`,
                 params.token,
-                params.domain
+                params.domain,
+                params.app_type
             )
         })
     }
@@ -34,7 +35,8 @@ export function parseConnectParams(query: SearchParams) {
         'session',
         'ws_url',
         'cell-number',
-        'token'
+        'token',
+        'app_type'
     )
     return params
 }