Merge branch 'main' into postinstall_3

1 files changed, 59 insertions, 0 deletions

diff --git a/packages/bun-vscode/src/features/tasks/tasks.ts b/packages/bun-vscode/src/features/tasks/tasks.ts
new file mode 100644
index 000000000..aabeb3920
--- /dev/null
+++ b/packages/bun-vscode/src/features/tasks/tasks.ts
@@ -0,0 +1,59 @@
+import * as vscode from "vscode";
+import { providePackageJsonTasks } from "./package.json";
+
+interface BunTaskDefinition extends vscode.TaskDefinition {
+  script: string;
+}
+
+export class BunTask extends vscode.Task {
+  declare definition: BunTaskDefinition;
+
+  constructor({
+    script,
+    name,
+    detail,
+    execution,
+    scope = vscode.TaskScope.Workspace,
+  }: {
+    script: string;
+    name: string;
+    detail?: string;
+    scope?: vscode.WorkspaceFolder | vscode.TaskScope.Global | vscode.TaskScope.Workspace;
+    execution?: vscode.ProcessExecution | vscode.ShellExecution | vscode.CustomExecution;
+  }) {
+    super({ type: "bun", script }, scope, name, "bun", execution);
+    this.detail = detail;
+  }
+}
+
+/**
+ * Registers the task provider for the bun extension.
+ */
+export function registerTaskProvider(context: vscode.ExtensionContext) {
+  const taskProvider: vscode.TaskProvider<BunTask> = {
+    provideTasks: async () => await providePackageJsonTasks(),
+    resolveTask: task => resolveTask(task),
+  };
+  context.subscriptions.push(vscode.tasks.registerTaskProvider("bun", taskProvider));
+}
+
+/**
+ * Parses tasks defined in the vscode tasks.json file.
+ * For more information, see https://code.visualstudio.com/api/extension-guides/task-provider
+ */
+export function resolveTask(task: BunTask): BunTask | undefined {
+  // Make sure the task has a script defined
+  const definition: BunTask["definition"] = task.definition;
+  if (!definition.script) return task;
+  const shellCommand = definition.script.startsWith("bun ") ? definition.script : `bun ${definition.script}`;
+
+  const newTask = new vscode.Task(
+    definition,
+    task.scope ?? vscode.TaskScope.Workspace,
+    task.name,
+    "bun",
+    new vscode.ShellExecution(shellCommand),
+  ) as BunTask;
+  newTask.detail = `${shellCommand} - tasks.json`;
+  return newTask;
+}