From a179e111384655048a1ca8d111e73de7e12b9b46 Mon Sep 17 00:00:00 2001
From: Nathan Park <parknate@amazon.com>
Date: Fri, 30 Aug 2024 21:48:49 +0900
Subject: [PATCH] Test cookie grabbing

---
 patched-vscode/extensions/package.json                 | 10 +++++++++-
 .../extensions/sagemaker-extension/src/extension.ts    | 10 ++++++++++
 2 files changed, 19 insertions(+), 1 deletion(-)

diff --git a/patched-vscode/extensions/package.json b/patched-vscode/extensions/package.json
index 2c83af40..df3976d8 100644
--- 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
index b85cec6a..3f1934e5 100644
--- 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);
 }