triggerdotdev--trigger.dev/apps/webapp/app/presenters/v3/WaitpointPresenter.server.ts

import { isWaitpointOutputTimeout, prettyPrintPacket } from "@trigger.dev/core/v3";
import { type PrismaClientOrTransaction, type PrismaReplicaClient } from "~/db.server";
import { clickhouseFactory } from "~/services/clickhouse/clickhouseFactoryInstance.server";
import { generateHttpCallbackUrl } from "~/services/httpCallback.server";
import { logger } from "~/services/logger.server";
import { controlPlaneResolver } from "~/v3/runOpsMigration/controlPlaneResolver.server";
import { readThroughRun } from "~/v3/runOpsMigration/readThrough.server";
import { BasePresenter } from "./basePresenter.server";
import { NextRunListPresenter, type NextRunListItem } from "./NextRunListPresenter.server";
import { waitpointStatusToApiStatus } from "./WaitpointListPresenter.server";

export type WaitpointDetail = NonNullable<Awaited<ReturnType<WaitpointPresenter["call"]>>>;

export class WaitpointPresenter extends BasePresenter {
  constructor(
    prisma?: PrismaClientOrTransaction,
    replica?: PrismaClientOrTransaction,
    private readonly readThroughDeps?: {
      // The new run-ops client + the legacy run-ops read replica (never the legacy writer).
      // Omitted => single-DB / self-host: both default to `_replica` (passthrough).
      newClient?: PrismaClientOrTransaction;
      legacyReplica?: PrismaClientOrTransaction;
      // Resolved boot constant from isSplitEnabled(). When false/absent:
      // the waitpoint lookup is one plain findFirst and the connected-runs hydrate runs passthrough.
      splitEnabled?: boolean;
    }
  ) {
    super(prisma, replica);
  }

  async #findWaitpoint(friendlyId: string, environmentId: string) {
    const where = { friendlyId, environmentId };
    const select = {
      id: true,
      friendlyId: true,
      type: true,
      status: true,
      idempotencyKey: true,
      userProvidedIdempotencyKey: true,
      idempotencyKeyExpiresAt: true,
      inactiveIdempotencyKey: true,
      output: true,
      outputType: true,
      outputIsError: true,
      completedAfter: true,
      completedAt: true,
      createdAt: true,
      tags: true,
      environmentId: true,
    } as const;

    const hydrate = (client: PrismaReplicaClient) => client.waitpoint.findFirst({ where, select });

    if (!this.readThroughDeps) {
      return this._replica.waitpoint.findFirst({ where, select });
    }

    const result = await readThroughRun({
      runId: friendlyId,
      environmentId,
      readNew: (client) => hydrate(client),
      readLegacy: (replica) => hydrate(replica),
      deps: {
        splitEnabled: this.readThroughDeps.splitEnabled,
        newClient:
          (this.readThroughDeps.newClient as PrismaReplicaClient | undefined) ??
          (this._replica as unknown as PrismaReplicaClient),
        legacyReplica:
          (this.readThroughDeps.legacyReplica as PrismaReplicaClient | undefined) ??
          (this._replica as unknown as PrismaReplicaClient),
      },
    });

    return result.source === "new" || result.source === "legacy-replica" ? result.value : null;
  }

  // Connected-run friendlyIds gathered across BOTH stores. The run<->waitpoint join co-locates with
  // the RUN (written on the run's DB), so the waitpoint's own store misses a cross-DB connection; we
  // read the join on each client and resolve the run's friendlyId on that same client, then union.
  // We never relation-select `connectedRuns`: it is not a field on the dedicated subset `Waitpoint`.
  async #connectedRunFriendlyIds(waitpointId: string): Promise<string[]> {
    const replica = this._replica as unknown as PrismaReplicaClient;
    const rawClients: PrismaReplicaClient[] =
      this.readThroughDeps?.splitEnabled === true
        ? [
            (this.readThroughDeps.newClient as PrismaReplicaClient | undefined) ?? replica,
            (this.readThroughDeps.legacyReplica as PrismaReplicaClient | undefined) ?? replica,
          ]
        : [replica];
    const clients = [...new Set(rawClients)];

    const friendlyIds = new Set<string>();
    for (const client of clients) {
      const runIds = await this.#connectedRunIdsOn(client, waitpointId);
      if (runIds.length === 0) {
        continue;
      }
      const runs = await client.taskRun.findMany({
        where: { id: { in: runIds } },
        select: { friendlyId: true },
        take: 5,
      });
      for (const run of runs) {
        friendlyIds.add(run.friendlyId);
      }
      if (friendlyIds.size >= 5) {
        break;
      }
    }
    return Array.from(friendlyIds).slice(0, 5);
  }

  // Schema-aware read of the run ids linked to a waitpoint: the dedicated subset uses the explicit
  // `WaitpointRunConnection` model, the control-plane full schema the implicit `_WaitpointRunConnections`
  // M2M (A = TaskRun.id, B = Waitpoint.id). The dedicated join delegate is absent on the full client.
  async #connectedRunIdsOn(client: PrismaReplicaClient, waitpointId: string): Promise<string[]> {
    const joinDelegate = (
      client as unknown as {
        waitpointRunConnection?: {
          findMany: (args: unknown) => Promise<{ taskRunId: string }[]>;
        };
      }
    ).waitpointRunConnection;
    if (joinDelegate && typeof joinDelegate.findMany === "function") {
      const links = await joinDelegate.findMany({
        where: { waitpointId },
        select: { taskRunId: true },
      });
      return links.map((link) => link.taskRunId);
    }
    const rows = await client.$queryRaw<{ A: string }[]>`
      SELECT "A" FROM "_WaitpointRunConnections" WHERE "B" = ${waitpointId}
    `;
    return rows.map((row) => row.A);
  }

  public async call({
    friendlyId,
    environmentId,
    projectId,
  }: {
    friendlyId: string;
    environmentId: string;
    projectId: string;
  }) {
    const waitpoint = await this.#findWaitpoint(friendlyId, environmentId);

    if (!waitpoint) {
      logger.error(`WaitpointPresenter: Waitpoint not found`, {
        friendlyId,
      });
      return null;
    }

    const environment = await controlPlaneResolver.resolveAuthenticatedEnv(waitpoint.environmentId);

    if (!environment) {
      logger.error(`WaitpointPresenter: environment not found`, { friendlyId });
      return null;
    }

    const output =
      waitpoint.outputType === "application/store"
        ? `/resources/packets/${environmentId}/${waitpoint.output}`
        : typeof waitpoint.output !== "undefined" && waitpoint.output !== null
          ? await prettyPrintPacket(waitpoint.output, waitpoint.outputType ?? undefined)
          : undefined;

    let _isTimeout = false;
    if (waitpoint.outputIsError && output) {
      if (isWaitpointOutputTimeout(output)) {
        _isTimeout = true;
      }
    }

    const connectedRunIds = await this.#connectedRunFriendlyIds(waitpoint.id);
    const connectedRuns: NextRunListItem[] = [];

    if (connectedRunIds.length > 0) {
      const clickhouse = await clickhouseFactory.getClickhouseForOrganization(
        environment.organizationId,
        "standard"
      );
      const runPresenter = new NextRunListPresenter(
        this._prisma,
        clickhouse,
        this.readThroughDeps
          ? {
              newClient: this.readThroughDeps.newClient ?? this._replica,
              legacyReplica: this.readThroughDeps.legacyReplica ?? this._replica,
              splitEnabled: this.readThroughDeps.splitEnabled ?? false,
            }
          : undefined
      );
      const { runs } = await runPresenter.call(environment.organizationId, environmentId, {
        projectId: projectId,
        runId: connectedRunIds,
        pageSize: 5,
        period: "31d",
      });

      connectedRuns.push(...runs);
    }

    return {
      id: waitpoint.friendlyId,
      type: waitpoint.type,
      url: generateHttpCallbackUrl(waitpoint.id, environment.apiKey),
      status: waitpointStatusToApiStatus(waitpoint.status, waitpoint.outputIsError),
      idempotencyKey: waitpoint.idempotencyKey,
      userProvidedIdempotencyKey: waitpoint.userProvidedIdempotencyKey,
      idempotencyKeyExpiresAt: waitpoint.idempotencyKeyExpiresAt,
      inactiveIdempotencyKey: waitpoint.inactiveIdempotencyKey,
      output: output,
      outputType: waitpoint.outputType,
      outputIsError: waitpoint.outputIsError,
      timeoutAt: waitpoint.completedAfter,
      completedAfter: waitpoint.completedAfter,
      completedAt: waitpoint.completedAt,
      createdAt: waitpoint.createdAt,
      tags: waitpoint.tags,
      connectedRuns,
    };
  }
}