feat: add toggle watch mode to command palette

vitest-dev · May 16, 2022 · 7b17988 · 7b17988
1 parent 3c3a386
commit 7b17988
Show file tree

Hide file tree

Showing 3 changed files with 30 additions and 1 deletion.
diff --git a/package.json b/package.json
@@ -44,13 +44,31 @@
         "title": "Start Watch Mode",
         "command": "vitest.startWatching",
         "category": "Vitest"
+      },
+      {
+        "title": "Stop Watch Mode",
+        "command": "vitest.stopWatching",
+        "category": "Vitest"
+      },
+      {
+        "title": "Toggle Watch Mode",
+        "command": "vitest.toggleWatching",
+        "category": "Vitest"
       }
     ],
     "menus": {
       "commandPalette": [
         {
           "command": "vitest.updateSnapshot",
           "when": "false"
+        },
+        {
+          "command": "vitest.startWatching",
+          "when": "false"
+        },
+        {
+          "command": "vitest.stopWatching",
+          "when": "false"
         }
       ],
       "testing/item/context": [
@@ -107,7 +125,7 @@
   },
   "scripts": {
     "vscode:prepublish": "yarn run compile",
-    "compile": "tsup ./src/extension.ts --external vscode",
+    "compile": "tsup ./src/extension.ts --external vscode --minify",
     "watch": "tsup ./src/extension.ts --external vscode --watch --sourcemap",
     "pretest": "yarn run compile && yarn run lint",
     "lint": "eslint src --ext ts",

diff --git a/src/command.ts b/src/command.ts
@@ -2,4 +2,5 @@ export enum Command {
   StartWatching = "vitest.startWatching",
   StopWatching = "vitest.stopWatching",
   UpdateSnapshot = "vitest.updateSnapshot",
+  ToggleWatching = "vitest.toggleWatching",
 }
diff --git a/src/extension.ts b/src/extension.ts
@@ -144,6 +144,16 @@ function registerWatchHandler(
         testWatcher!.dispose();
       },
     ),
+    vscode.commands.registerCommand(
+      Command.ToggleWatching,
+      () => {
+        if (testWatcher.isWatching.value) {
+          testWatcher.dispose();
+        } else {
+          testWatcher.watch();
+        }
+      },
+    ),
   );
 
   ctrl.createRunProfile(