wehub-resource-sync
63 lines
2.2 KiB
TypeScript
63 lines
2.2 KiB
TypeScript
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");
|
|
}
|
|
}
|
|
}
|