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

import {
  type RuntimeEnvironment,
  type PrismaClient,
  type RuntimeEnvironmentType,
} from "@trigger.dev/database";
import { prisma } from "~/db.server";
import { logger } from "~/services/logger.server";
import { type UserFromSession } from "~/services/session.server";

export type MinimumEnvironment = Pick<RuntimeEnvironment, "id" | "type" | "slug" | "paused"> & {
  orgMember: null | {
    userId: string | undefined;
  };
};

export class SelectBestEnvironmentPresenter {
  #prismaClient: PrismaClient;

  constructor(prismaClient: PrismaClient = prisma) {
    this.#prismaClient = prismaClient;
  }

  public async call({ user }: { user: UserFromSession }) {
    const { project, organization } = await this.getBestProject(user);
    const environment = await this.selectBestEnvironment(project.id, user, project.environments);

    return {
      project,
      organization,
      environment,
    };
  }

  async getBestProject(user: UserFromSession) {
    //try get current project from cookie
    const projectId = user.dashboardPreferences.currentProjectId;

    if (projectId) {
      const project = await this.#prismaClient.project.findFirst({
        where: {
          id: projectId,
          deletedAt: null,
          organization: { members: { some: { userId: user.id } } },
        },
        include: {
          organization: true,
          environments: {
            select: {
              id: true,
              type: true,
              slug: true,
              parentEnvironmentId: true,
              paused: true,
              orgMember: {
                select: {
                  userId: true,
                },
              },
            },
          },
        },
      });
      if (project) {
        return { project, organization: project.organization };
      }
    }

    //failing that, we pick the most recently modified project
    const projects = await this.#prismaClient.project.findMany({
      include: {
        organization: true,
        environments: {
          select: {
            id: true,
            type: true,
            slug: true,
            parentEnvironmentId: true,
            paused: true,
            orgMember: {
              select: {
                userId: true,
              },
            },
          },
        },
      },
      where: {
        deletedAt: null,
        organization: {
          members: { some: { userId: user.id } },
        },
      },
      orderBy: {
        updatedAt: "desc",
      },
      take: 1,
    });

    if (projects.length === 0) {
      throw new Response("Not Found", { status: 404 });
    }

    return { project: projects[0], organization: projects[0].organization };
  }

  async selectBestProjectFromProjects({
    user,
    projectSlug,
    projects,
  }: {
    user: UserFromSession;
    projectSlug: string | undefined;
    projects: {
      id: string;
      slug: string;
      name: string;
      updatedAt: Date;
    }[];
  }) {
    if (projectSlug) {
      const proj = projects.find((p) => p.slug === projectSlug);
      if (proj) {
        return proj;
      }

      if (!proj) {
        logger.info("Not Found: project", {
          projectSlug,
          projects,
        });
      }
    }

    const currentProjectId = user.dashboardPreferences.currentProjectId;
    const project = projects.find((p) => p.id === currentProjectId);
    if (project) {
      return project;
    }

    //most recently updated
    return projects.sort((a, b) => b.updatedAt.getTime() - a.updatedAt.getTime()).at(0);
  }

  async selectBestEnvironment<
    T extends {
      id: string;
      type: RuntimeEnvironmentType;
      slug: string;
      parentEnvironmentId: string | null;
      orgMember: { userId: string } | null;
    },
  >(projectId: string, user: UserFromSession, environments: T[]): Promise<T> {
    //try get current environment from prefs
    const currentEnvironmentId: string | undefined =
      user.dashboardPreferences.projects[projectId]?.currentEnvironment.id;

    const currentEnvironment = environments.find((env) => env.id === currentEnvironmentId);
    if (currentEnvironment) {
      return currentEnvironment;
    }

    //otherwise show their dev environment
    const yourDevEnvironment = environments.find(
      // Return the default dev environment (the root, no parent), not a branch
      (env) =>
        env.type === "DEVELOPMENT" &&
        env.parentEnvironmentId === null &&
        env.orgMember?.userId === user.id
    );
    if (yourDevEnvironment) {
      return yourDevEnvironment;
    }

    //otherwise show their prod environment
    const prodEnvironment = environments.find((env) => env.type === "PRODUCTION");
    if (prodEnvironment) {
      return prodEnvironment;
    }

    throw new Error("No environments found");
  }
}