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

/* eslint-disable turbo/no-undeclared-env-vars */

import { integrationCatalog } from "../app/services/externalApis/integrationCatalog.server";
import { seedCloud } from "./seedCloud";
import { prisma } from "../app/db.server";

async function seedIntegrationAuthMethods() {
  for (const [_, integration] of Object.entries(integrationCatalog.getIntegrations())) {
    await prisma.integrationDefinition.upsert({
      where: {
        id: integration.identifier,
      },
      create: {
        id: integration.identifier,
        name: integration.name,
        icon: integration.icon ?? integration.identifier,
        instructions: "Instructions go here",
        description: integration.description,
        packageName: integration.packageName,
      },
      update: {
        name: integration.name,
        description: integration.description,
        packageName: integration.packageName,
        icon: integration.icon ?? integration.identifier,
      },
    });

    for (const [key, authMethod] of Object.entries(integration.authenticationMethods)) {
      if (authMethod.type === "oauth2") {
        console.log(`Upserting auth method ${integration.identifier}.${key}`);

        await prisma.integrationAuthMethod.upsert({
          where: {
            definitionId_key: {
              definitionId: integration.identifier,
              key,
            },
          },
          create: {
            key,
            name: authMethod.name,
            description: authMethod.description ?? "",
            type: authMethod.type,
            client: authMethod.client,
            config: authMethod.config,
            scopes: authMethod.scopes,
            definition: {
              connect: {
                id: integration.identifier,
              },
            },
            help: authMethod.help,
          },
          update: {
            name: authMethod.name,
            description: authMethod.description ?? "",
            type: authMethod.type,
            client: authMethod.client,
            config: authMethod.config,
            scopes: authMethod.scopes,
            help: authMethod.help,
          },
        });
      }
    }
  }
}

async function seed() {
  await seedIntegrationAuthMethods();

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

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