Improve studio access token commands (#5088)

Author: julieg18

extension/package.json

@@ -124,11 +124,6 @@
         "command": "dvc.addStudioAccessToken",
         "category": "DVC"
       },
-      {
-        "title": "Update Studio Access Token",
-        "command": "dvc.updateStudioAccessToken",
-        "category": "DVC"
-      },
       {
         "title": "Add Self-Hosted Studio URL",
         "command": "dvc.addStudioUrl",
@@ -465,6 +460,11 @@
         "command": "dvc.showSetup",
         "category": "DVC"
       },
+      {
+        "title": "Update Studio Access Token",
+        "command": "dvc.updateStudioAccessToken",
+        "category": "DVC"
+      },
       {
         "title": "Update Self-Hosted Studio URL",
         "command": "dvc.updateStudioUrl",
@@ -727,10 +727,6 @@
           "command": "dvc.addStudioUrl",
           "when": "dvc.commands.available && dvc.project.available && !dvc.studio.selfhosted"
         },
-        {
-          "command": "dvc.updateStudioAccessToken",
-          "when": "dvc.commands.available && dvc.project.available && dvc.studio.connected"
-        },
         {
           "command": "dvc.addTarget",
           "when": "false"
@@ -955,6 +951,10 @@
           "command": "dvc.stopAllRunningExperiments",
           "when": "dvc.commands.available && dvc.project.available && dvc.experiment.running"
         },
+        {
+          "command": "dvc.updateStudioAccessToken",
+          "when": "dvc.commands.available && dvc.project.available && dvc.studio.connected"
+        },
         {
           "command": "dvc.updateStudioUrl",
           "when": "dvc.commands.available && dvc.project.available && dvc.studio.selfhosted"

extension/src/setup/index.ts

@@ -306,7 +306,7 @@ export class Setup
   }
 
   public async saveStudioAccessToken() {
-    const cwd = this.dvcRoots[0] || getFirstWorkspaceFolder()
+    const cwd = this.getCwd()
 
     if (!cwd) {
       return