documenso/packages/prisma/seed/initial-seed.ts

import fs from 'node:fs';
import path from 'node:path';

import { hashSync } from '@documenso/lib/server-only/auth/hash';

import { prisma } from '..';
import { DocumentDataType, DocumentSource, Role, TeamMemberRole } from '../client';

export const seedDatabase = async () => {
  const examplePdf = fs
    .readFileSync(path.join(__dirname, '../../../assets/example.pdf'))
    .toString('base64');

  const exampleUser = await prisma.user.upsert({
    where: {
      email: 'example@documenso.com',
    },
    create: {
      name: 'Example User',
      email: 'example@documenso.com',
      emailVerified: new Date(),
      password: hashSync('password'),
      roles: [Role.USER],
    },
    update: {},
  });

  const adminUser = await prisma.user.upsert({
    where: {
      email: 'admin@documenso.com',
    },
    create: {
      name: 'Admin User',
      email: 'admin@documenso.com',
      emailVerified: new Date(),
      password: hashSync('password'),
      roles: [Role.USER, Role.ADMIN],
    },
    update: {},
  });

  const examplePdfData = await prisma.documentData.upsert({
    where: {
      id: 'clmn0kv5k0000pe04vcqg5zla',
    },
    create: {
      id: 'clmn0kv5k0000pe04vcqg5zla',
      type: DocumentDataType.BYTES_64,
      data: examplePdf,
      initialData: examplePdf,
    },
    update: {},
  });

  await prisma.document.create({
    data: {
      source: DocumentSource.DOCUMENT,
      title: 'Example Document',
      documentDataId: examplePdfData.id,
      userId: exampleUser.id,
      Recipient: {
        create: {
          name: String(adminUser.name),
          email: adminUser.email,
          token: Math.random().toString(36).slice(2, 9),
        },
      },
    },
  });

  const testUsers = [
    'test@documenso.com',
    'test2@documenso.com',
    'test3@documenso.com',
    'test4@documenso.com',
  ];

  const createdUsers = [];

  for (const email of testUsers) {
    const testUser = await prisma.user.upsert({
      where: {
        email: email,
      },
      create: {
        name: 'Test User',
        email: email,
        emailVerified: new Date(),
        password: hashSync('password'),
        roles: [Role.USER],
      },
      update: {},
    });

    createdUsers.push(testUser);
  }

  const team1 = await prisma.team.create({
    data: {
      name: 'Team 1',
      url: 'team1',
      ownerUserId: createdUsers[0].id,
    },
  });

  const team2 = await prisma.team.create({
    data: {
      name: 'Team 2',
      url: 'team2',
      ownerUserId: createdUsers[1].id,
    },
  });

  for (const team of [team1, team2]) {
    await prisma.teamMember.createMany({
      data: [
        {
          teamId: team.id,
          userId: createdUsers[1].id,
          role: TeamMemberRole.ADMIN,
        },
        {
          teamId: team.id,
          userId: createdUsers[2].id,
          role: TeamMemberRole.MEMBER,
        },
      ],
    });
  }
};