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,62 @@
+import { type RuntimeEnvironmentType, type TaskTriggerSource } from "@trigger.dev/database";
+import { sqlDatabaseSchema } from "~/db.server";
+import { findCurrentWorkerDeployment } from "~/v3/models/workerDeployment.server";
+import { BasePresenter } from "./basePresenter.server";
+
+type TaskListOptions = {
+  userId: string;
+  projectId: string;
+  environmentId: string;
+  environmentType: RuntimeEnvironmentType;
+};
+
+export type TaskList = Awaited<ReturnType<TestPresenter["call"]>>;
+export type TaskListItem = NonNullable<TaskList["tasks"]>[0];
+
+export class TestPresenter extends BasePresenter {
+  public async call({ userId, projectId, environmentId, environmentType }: TaskListOptions) {
+    const isDev = environmentType === "DEVELOPMENT";
+    const tasks = await this.#getTasks(environmentId, isDev);
+
+    return {
+      tasks: tasks.map((task) => ({
+        id: task.id,
+        taskIdentifier: task.slug,
+        filePath: task.filePath,
+        friendlyId: task.friendlyId,
+        triggerSource: task.triggerSource,
+      })),
+    };
+  }
+
+  async #getTasks(envId: string, isDev: boolean) {
+    if (isDev) {
+      return await this._replica.$queryRaw<
+        {
+          id: string;
+          version: string;
+          slug: string;
+          filePath: string;
+          friendlyId: string;
+          triggerSource: TaskTriggerSource;
+        }[]
+      >`WITH workers AS (
+          SELECT
+                bw.*,
+                ROW_NUMBER() OVER(ORDER BY string_to_array(bw.version, '.')::int[] DESC) AS rn
+          FROM
+                ${sqlDatabaseSchema}."BackgroundWorker" bw
+          WHERE "runtimeEnvironmentId" = ${envId}
+        ),
+        latest_workers AS (SELECT * FROM workers WHERE rn = 1)
+        SELECT bwt.id, version, slug, "filePath", bwt."friendlyId", bwt."triggerSource"
+        FROM latest_workers
+        JOIN ${sqlDatabaseSchema}."BackgroundWorkerTask" bwt ON bwt."workerId" = latest_workers.id
+        WHERE bwt."triggerSource" != 'AGENT'
+        ORDER BY slug ASC;`;
+    } else {
+      const currentDeployment = await findCurrentWorkerDeployment({ environmentId: envId });
+      return (currentDeployment?.worker?.tasks ?? []).filter((t) => t.triggerSource !== "AGENT");
+    }
+  }
+}