Add locks task to run the current open file in vs code

vsc/extension.ts

@@ -5,9 +5,6 @@ import {
   ExtensionContext,
   commands,
   tasks,
-  ShellExecution,
-  Task,
-  TaskScope,
   TaskDefinition,
   window,
 } from "vscode";
@@ -16,6 +13,7 @@ import {
   LanguageClientOptions,
   ServerOptions,
 } from "vscode-languageclient/node";
+import { LocksTaskProvider } from "./locksTaskProvider";
 
 let lc: LanguageClient;
 
@@ -81,6 +79,8 @@ export function activate(context: ExtensionContext) {
     commands.registerCommand("locks.runCurrentFile", runFileHandler)
   );
 
+  tasks.registerTaskProvider("locks", new LocksTaskProvider());
+
   lc.start();
 }
 

vsc/locksTaskProvider.ts

@@ -0,0 +1,43 @@
+import * as vscode from "vscode";
+
+export class LocksTaskProvider implements vscode.TaskProvider {
+  static TaskType = "locks";
+  private task: vscode.Task | undefined;
+
+  constructor() {
+    this.task = new vscode.Task(
+      {
+        type: LocksTaskProvider.TaskType,
+      },
+      vscode.TaskScope.Workspace,
+      "run",
+      "locks"
+    );
+  }
+
+  async provideTasks(token: vscode.CancellationToken): Promise<vscode.Task[]> {
+    const tasks = [];
+    tasks.push(this.task);
+    return tasks;
+  }
+
+  resolveTask(
+    task: vscode.Task,
+    _token: vscode.CancellationToken
+  ): vscode.Task | undefined {
+    return new vscode.Task(
+      task.definition,
+      vscode.TaskScope.Workspace,
+      task.name,
+      task.source,
+      new vscode.ShellExecution(
+        `locks run ${this.getCurrentOpenFilePath()}`,
+        {}
+      )
+    );
+  }
+
+  private getCurrentOpenFilePath(): string {
+    return vscode.window.activeTextEditor.document.uri.path;
+  }
+}

vsc/package.json

@@ -72,6 +72,13 @@
         "command": "locks.runCurrentFile",
         "title": "Locks: Run Current File"
       }
+    ],
+    "taskDefinitions": [
+      {
+        "type": "locks",
+        "required": [],
+        "properties": {}
+      }
     ]
   },
   "activationEvents": [],