Test cookie grabbing

aws · Aug 30, 2024 · a179e11 · a179e11
1 parent feadcc6
commit a179e11
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 1 deletion.
diff --git a/patched-vscode/extensions/package.json b/patched-vscode/extensions/package.json
@@ -16,5 +16,13 @@
   },
   "resolutions": {
     "node-gyp-build": "4.8.1"
-  }
+  },
+  "contributes": {
+    "commands": [
+        {
+            "command": "sagemaker.logCookie",
+            "title": "SageMaker: Log Cookie"
+        }
+    ]
+}
 }
diff --git a/patched-vscode/extensions/sagemaker-extension/src/extension.ts b/patched-vscode/extensions/sagemaker-extension/src/extension.ts
@@ -134,4 +134,14 @@ export function activate(context: vscode.ExtensionContext) {
         initialize(sagemakerCookie);
         updateStatusItemWithMetadata(context);
     });
+
+    // New command
+    const logCookieCommand = vscode.commands.registerCommand('sagemaker.logCookie', () => {
+        vscode.commands.executeCommand(PARSE_SAGEMAKER_COOKIE_COMMAND).then(response => {
+            const sagemakerCookie: SagemakerCookie = response as SagemakerCookie;
+            console.log('SageMaker Cookie:', sagemakerCookie);
+        });
+    });
+
+    context.subscriptions.push(logCookieCommand);
 }