chore: import upstream snapshot with attribution

2026-07-13 13:32:57 +08:00
commit cd420f9332
4811 changed files with 884702 additions and 0 deletions
@@ -0,0 +1,127 @@
+import assertNever from "assert-never";
+import type { PrismaClientOrTransaction } from "~/db.server";
+import { workerQueue } from "~/services/worker.server";
+import { BaseService } from "../baseService.server";
+import { CancelTaskRunService } from "../cancelTaskRun.server";
+import { ReplayTaskRunService } from "../replayTaskRun.server";
+
+export class PerformBulkActionService extends BaseService {
+  public async performBulkActionItem(bulkActionItemId: string) {
+    const item = await this._prisma.bulkActionItem.findFirst({
+      where: { id: bulkActionItemId },
+      include: {
+        sourceRun: true,
+        destinationRun: true,
+      },
+    });
+
+    if (!item) {
+      return;
+    }
+
+    if (item.status !== "PENDING") {
+      return;
+    }
+
+    switch (item.type) {
+      case "REPLAY": {
+        const service = new ReplayTaskRunService(this._prisma);
+        const result = await service.call(item.sourceRun, { triggerSource: "dashboard" });
+
+        await this._prisma.bulkActionItem.update({
+          where: { id: item.id },
+          data: {
+            destinationRunId: result?.id,
+            status: result ? "COMPLETED" : "FAILED",
+            error: result ? undefined : "Failed to replay task run",
+          },
+        });
+
+        break;
+      }
+      case "CANCEL": {
+        const service = new CancelTaskRunService(this._prisma);
+
+        const result = await service.call(item.sourceRun);
+
+        await this._prisma.bulkActionItem.update({
+          where: { id: item.id },
+          data: {
+            destinationRunId: item.sourceRun.id,
+            status: result ? "COMPLETED" : "FAILED",
+            error: result ? undefined : "Task wasn't cancelable",
+          },
+        });
+
+        break;
+      }
+      default: {
+        assertNever(item.type);
+      }
+    }
+
+    const groupItems = await this._prisma.bulkActionItem.findMany({
+      where: { groupId: item.groupId },
+      select: {
+        status: true,
+      },
+    });
+
+    const isGroupCompleted = groupItems.every((item) => item.status !== "PENDING");
+
+    if (isGroupCompleted) {
+      await this._prisma.bulkActionItem.update({
+        where: { id: item.id },
+        data: {
+          status: "COMPLETED",
+        },
+      });
+    }
+  }
+
+  public async enqueueBulkActionItem(bulkActionItemId: string, groupId: string) {
+    await workerQueue.enqueue(
+      "v3.performBulkActionItem",
+      {
+        bulkActionItemId,
+      },
+      {
+        jobKey: `performBulkActionItem:${bulkActionItemId}`,
+      }
+    );
+  }
+
+  public async call(bulkActionGroupId: string) {
+    const actionGroup = await this._prisma.bulkActionGroup.findFirst({
+      where: { id: bulkActionGroupId },
+      select: { id: true },
+    });
+
+    if (!actionGroup) {
+      return;
+    }
+
+    const items = await this._prisma.bulkActionItem.findMany({
+      where: { groupId: bulkActionGroupId },
+      select: { id: true },
+    });
+
+    for (const item of items) {
+      await this.enqueueBulkActionItem(item.id, bulkActionGroupId);
+    }
+  }
+
+  static async enqueue(bulkActionGroupId: string, tx: PrismaClientOrTransaction, runAt?: Date) {
+    return await workerQueue.enqueue(
+      "v3.performBulkAction",
+      {
+        bulkActionGroupId,
+      },
+      {
+        tx,
+        runAt,
+        jobKey: `performBulkAction:${bulkActionGroupId}`,
+      }
+    );
+  }
+}