triggerdotdev--trigger.dev/apps/webapp/app/models/runtimeEnvironment.server.ts

import type { Prisma, RuntimeEnvironment } from "@trigger.dev/database";
import { prisma } from "~/db.server";
import { getUsername } from "~/utils/username";

export type { RuntimeEnvironment };

export async function findEnvironmentByApiKey(apiKey: string) {
  const environment = await prisma.runtimeEnvironment.findUnique({
    where: {
      apiKey,
    },
    include: {
      project: true,
      organization: true,
      orgMember: true,
    },
  });

  //don't return deleted projects
  if (environment?.project.deletedAt !== null) {
    return null;
  }

  return environment;
}

export async function findEnvironmentByPublicApiKey(apiKey: string) {
  const environment = await prisma.runtimeEnvironment.findUnique({
    where: {
      pkApiKey: apiKey,
    },
    include: {
      project: true,
      organization: true,
      orgMember: true,
    },
  });

  //don't return deleted projects
  if (environment?.project.deletedAt !== null) {
    return null;
  }

  return environment;
}

export async function findEnvironmentById(id: string) {
  const environment = await prisma.runtimeEnvironment.findUnique({
    where: {
      id,
    },
    include: {
      project: true,
      organization: true,
      orgMember: true,
    },
  });

  //don't return deleted projects
  if (environment?.project.deletedAt !== null) {
    return null;
  }

  return environment;
}

export async function createNewSession(environment: RuntimeEnvironment, ipAddress: string) {
  return prisma.$transaction(async (tx) => {
    const session = await tx.runtimeEnvironmentSession.create({
      data: {
        environmentId: environment.id,
        ipAddress,
      },
    });

    await tx.runtimeEnvironment.update({
      where: {
        id: environment.id,
      },
      data: {
        currentSessionId: session.id,
      },
    });

    return session;
  });
}

export async function disconnectSession(environmentId: string) {
  return prisma.$transaction(async (tx) => {
    const environment = await tx.runtimeEnvironment.findUnique({
      where: {
        id: environmentId,
      },
    });

    if (!environment || !environment.currentSessionId) {
      return null;
    }

    const session = await tx.runtimeEnvironmentSession.update({
      where: {
        id: environment.currentSessionId,
      },
      data: {
        disconnectedAt: new Date(),
      },
    });

    await tx.runtimeEnvironment.update({
      where: {
        id: environment.id,
      },
      data: {
        currentSessionId: null,
      },
    });

    return session;
  });
}

type DisplayableInputEnvironment = Prisma.RuntimeEnvironmentGetPayload<{
  select: {
    id: true;
    type: true;
    orgMember: {
      select: {
        user: {
          select: {
            id: true;
            name: true;
            displayName: true;
          };
        };
      };
    };
  };
}>;

export function displayableEnvironments(
  environment: DisplayableInputEnvironment,
  userId: string | undefined
) {
  return {
    id: environment.id,
    type: environment.type,
    userName: environment.orgMember
      ? environment.orgMember.user.id === userId
        ? undefined
        : getUsername(environment.orgMember.user)
      : undefined,
  };
}