Add run task command

Shopify · Jun 27, 2024 · df86afa · df86afa
1 parent 7b23190
commit df86afa
Show file tree

Hide file tree

Showing 3 changed files with 24 additions and 0 deletions.
diff --git a/vscode/src/common.ts b/vscode/src/common.ts
@@ -19,6 +19,7 @@ export enum Command {
   DebugTest = "rubyLsp.debugTest",
   ShowSyntaxTree = "rubyLsp.showSyntaxTree",
   DisplayAddons = "rubyLsp.displayAddons",
+  RunTask = "rubyLsp.runTask",
 }
 
 export interface RubyInterface {

diff --git a/vscode/src/rubyLsp.ts b/vscode/src/rubyLsp.ts
@@ -370,6 +370,22 @@ export class RubyLsp {
         Command.DebugTest,
         this.testController.debugTest.bind(this.testController),
       ),
+      vscode.commands.registerCommand(
+        Command.RunTask,
+        async (command: string) => {
+          let workspace = this.currentActiveWorkspace();
+
+          if (!workspace) {
+            workspace = await this.showWorkspacePick();
+          }
+
+          if (!workspace) {
+            return;
+          }
+
+          await workspace.execute(command);
+        },
+      ),
     );
   }
 

diff --git a/vscode/src/workspace.ts b/vscode/src/workspace.ts
@@ -263,6 +263,13 @@ export class Workspace implements WorkspaceInterface {
     return this.#rebaseInProgress;
   }
 
+  async execute(command: string) {
+    return asyncExec(command, {
+      env: this.ruby.env,
+      cwd: this.workspaceFolder.uri.fsPath,
+    });
+  }
+
   private registerRestarts(context: vscode.ExtensionContext) {
     this.createRestartWatcher(context, "Gemfile.lock");
     this.createRestartWatcher(context, "gems.locked");