triggerdotdev--trigger.dev/apps/webapp/prisma/seed.ts

import { seedCloud } from "./seedCloud";
import { prisma } from "../app/db.server";
import { createEnvironment } from "~/models/organization.server";

async function runDataMigrations() {
  await runStagingEnvironmentMigration();
}

async function runStagingEnvironmentMigration() {
  try {
    await prisma.$transaction(async (tx) => {
      const existingDataMigration = await tx.dataMigration.findUnique({
        where: {
          name: "2023-09-27-AddStagingEnvironments",
        },
      });

      if (existingDataMigration) {
        return;
      }

      await tx.dataMigration.create({
        data: {
          name: "2023-09-27-AddStagingEnvironments",
        },
      });

      console.log("Running data migration 2023-09-27-AddStagingEnvironments");

      const projectsWithoutStagingEnvironments = await tx.project.findMany({
        where: {
          environments: {
            none: {
              type: "STAGING",
            },
          },
        },
        include: {
          organization: true,
        },
      });

      for (const project of projectsWithoutStagingEnvironments) {
        try {
          console.log(
            `Creating staging environment for project ${project.slug} on org ${project.organization.slug}`
          );

          await createEnvironment({
            organization: project.organization,
            project,
            type: "STAGING",
            isBranchableEnvironment: false,
            member: undefined,
            prismaClient: tx,
          });
        } catch (error) {
          console.error(error);
        }
      }

      await tx.dataMigration.update({
        where: {
          name: "2023-09-27-AddStagingEnvironments",
        },
        data: {
          completedAt: new Date(),
        },
      });
    });
  } catch (error) {
    console.error(error);
  }
}

async function seed() {
  if (process.env.NODE_ENV === "development" && process.env.SEED_CLOUD === "enabled") {
    await seedCloud(prisma);
  }

  await runDataMigrations();
}

seed()
  .catch((e) => {
    console.error(e);
    process.exit(1);
  })
  .finally(async () => {
    await prisma.$disconnect();
  });