chore: update branch

Signed-off-by: Adithya Krishna <adi@documenso.com>

packages/prisma/migrations/20240205040421_add_teams/migration.sql

@@ -0,0 +1,187 @@
+/*
+  Warnings:
+
+  - A unique constraint covering the columns `[teamId]` on the table `Subscription` will be added. If there are existing duplicate values, this will fail.
+
+*/
+-- CreateEnum
+CREATE TYPE "TeamMemberRole" AS ENUM ('ADMIN', 'MANAGER', 'MEMBER');
+
+-- CreateEnum
+CREATE TYPE "TeamMemberInviteStatus" AS ENUM ('ACCEPTED', 'PENDING');
+
+-- AlterTable
+ALTER TABLE "Document" ADD COLUMN     "teamId" INTEGER;
+
+-- AlterTable
+ALTER TABLE "Subscription" ADD COLUMN     "teamId" INTEGER,
+ALTER COLUMN "userId" DROP NOT NULL;
+
+-- CreateTable
+CREATE TABLE "Team" (
+    "id" SERIAL NOT NULL,
+    "name" TEXT NOT NULL,
+    "url" TEXT NOT NULL,
+    "createdAt" TIMESTAMP(3) NOT NULL DEFAULT CURRENT_TIMESTAMP,
+    "customerId" TEXT,
+    "ownerUserId" INTEGER NOT NULL,
+
+    CONSTRAINT "Team_pkey" PRIMARY KEY ("id")
+);
+
+-- CreateTable
+CREATE TABLE "TeamPending" (
+    "id" SERIAL NOT NULL,
+    "name" TEXT NOT NULL,
+    "url" TEXT NOT NULL,
+    "createdAt" TIMESTAMP(3) NOT NULL DEFAULT CURRENT_TIMESTAMP,
+    "customerId" TEXT NOT NULL,
+    "ownerUserId" INTEGER NOT NULL,
+
+    CONSTRAINT "TeamPending_pkey" PRIMARY KEY ("id")
+);
+
+-- CreateTable
+CREATE TABLE "TeamMember" (
+    "id" SERIAL NOT NULL,
+    "teamId" INTEGER NOT NULL,
+    "createdAt" TIMESTAMP(3) NOT NULL DEFAULT CURRENT_TIMESTAMP,
+    "role" "TeamMemberRole" NOT NULL,
+    "userId" INTEGER NOT NULL,
+
+    CONSTRAINT "TeamMember_pkey" PRIMARY KEY ("id")
+);
+
+-- CreateTable
+CREATE TABLE "TeamEmail" (
+    "teamId" INTEGER NOT NULL,
+    "createdAt" TIMESTAMP(3) NOT NULL DEFAULT CURRENT_TIMESTAMP,
+    "name" TEXT NOT NULL,
+    "email" TEXT NOT NULL,
+
+    CONSTRAINT "TeamEmail_pkey" PRIMARY KEY ("teamId")
+);
+
+-- CreateTable
+CREATE TABLE "TeamEmailVerification" (
+    "teamId" INTEGER NOT NULL,
+    "name" TEXT NOT NULL,
+    "email" TEXT NOT NULL,
+    "token" TEXT NOT NULL,
+    "expiresAt" TIMESTAMP(3) NOT NULL,
+    "createdAt" TIMESTAMP(3) NOT NULL DEFAULT CURRENT_TIMESTAMP,
+
+    CONSTRAINT "TeamEmailVerification_pkey" PRIMARY KEY ("teamId")
+);
+
+-- CreateTable
+CREATE TABLE "TeamTransferVerification" (
+    "teamId" INTEGER NOT NULL,
+    "userId" INTEGER NOT NULL,
+    "name" TEXT NOT NULL,
+    "email" TEXT NOT NULL,
+    "token" TEXT NOT NULL,
+    "expiresAt" TIMESTAMP(3) NOT NULL,
+    "createdAt" TIMESTAMP(3) NOT NULL DEFAULT CURRENT_TIMESTAMP,
+    "clearPaymentMethods" BOOLEAN NOT NULL DEFAULT false,
+
+    CONSTRAINT "TeamTransferVerification_pkey" PRIMARY KEY ("teamId")
+);
+
+-- CreateTable
+CREATE TABLE "TeamMemberInvite" (
+    "id" SERIAL NOT NULL,
+    "teamId" INTEGER NOT NULL,
+    "createdAt" TIMESTAMP(3) NOT NULL DEFAULT CURRENT_TIMESTAMP,
+    "email" TEXT NOT NULL,
+    "status" "TeamMemberInviteStatus" NOT NULL DEFAULT 'PENDING',
+    "role" "TeamMemberRole" NOT NULL,
+    "token" TEXT NOT NULL,
+
+    CONSTRAINT "TeamMemberInvite_pkey" PRIMARY KEY ("id")
+);
+
+-- CreateIndex
+CREATE UNIQUE INDEX "Team_url_key" ON "Team"("url");
+
+-- CreateIndex
+CREATE UNIQUE INDEX "Team_customerId_key" ON "Team"("customerId");
+
+-- CreateIndex
+CREATE UNIQUE INDEX "TeamPending_url_key" ON "TeamPending"("url");
+
+-- CreateIndex
+CREATE UNIQUE INDEX "TeamPending_customerId_key" ON "TeamPending"("customerId");
+
+-- CreateIndex
+CREATE UNIQUE INDEX "TeamMember_userId_teamId_key" ON "TeamMember"("userId", "teamId");
+
+-- CreateIndex
+CREATE UNIQUE INDEX "TeamEmail_teamId_key" ON "TeamEmail"("teamId");
+
+-- CreateIndex
+CREATE UNIQUE INDEX "TeamEmail_email_key" ON "TeamEmail"("email");
+
+-- CreateIndex
+CREATE UNIQUE INDEX "TeamEmailVerification_teamId_key" ON "TeamEmailVerification"("teamId");
+
+-- CreateIndex
+CREATE UNIQUE INDEX "TeamEmailVerification_token_key" ON "TeamEmailVerification"("token");
+
+-- CreateIndex
+CREATE UNIQUE INDEX "TeamTransferVerification_teamId_key" ON "TeamTransferVerification"("teamId");
+
+-- CreateIndex
+CREATE UNIQUE INDEX "TeamTransferVerification_token_key" ON "TeamTransferVerification"("token");
+
+-- CreateIndex
+CREATE UNIQUE INDEX "TeamMemberInvite_token_key" ON "TeamMemberInvite"("token");
+
+-- CreateIndex
+CREATE UNIQUE INDEX "TeamMemberInvite_teamId_email_key" ON "TeamMemberInvite"("teamId", "email");
+
+-- CreateIndex
+CREATE UNIQUE INDEX "Subscription_teamId_key" ON "Subscription"("teamId");
+
+-- AddForeignKey
+ALTER TABLE "Subscription" ADD CONSTRAINT "Subscription_teamId_fkey" FOREIGN KEY ("teamId") REFERENCES "Team"("id") ON DELETE CASCADE ON UPDATE CASCADE;
+
+-- AddForeignKey
+ALTER TABLE "Document" ADD CONSTRAINT "Document_teamId_fkey" FOREIGN KEY ("teamId") REFERENCES "Team"("id") ON DELETE SET NULL ON UPDATE CASCADE;
+
+-- AddForeignKey
+ALTER TABLE "Team" ADD CONSTRAINT "Team_ownerUserId_fkey" FOREIGN KEY ("ownerUserId") REFERENCES "User"("id") ON DELETE RESTRICT ON UPDATE CASCADE;
+
+-- AddForeignKey
+ALTER TABLE "TeamPending" ADD CONSTRAINT "TeamPending_ownerUserId_fkey" FOREIGN KEY ("ownerUserId") REFERENCES "User"("id") ON DELETE CASCADE ON UPDATE CASCADE;
+
+-- AddForeignKey
+ALTER TABLE "TeamMember" ADD CONSTRAINT "TeamMember_userId_fkey" FOREIGN KEY ("userId") REFERENCES "User"("id") ON DELETE RESTRICT ON UPDATE CASCADE;
+
+-- AddForeignKey
+ALTER TABLE "TeamMember" ADD CONSTRAINT "TeamMember_teamId_fkey" FOREIGN KEY ("teamId") REFERENCES "Team"("id") ON DELETE CASCADE ON UPDATE CASCADE;
+
+-- AddForeignKey
+ALTER TABLE "TeamEmail" ADD CONSTRAINT "TeamEmail_teamId_fkey" FOREIGN KEY ("teamId") REFERENCES "Team"("id") ON DELETE CASCADE ON UPDATE CASCADE;
+
+-- AddForeignKey
+ALTER TABLE "TeamEmailVerification" ADD CONSTRAINT "TeamEmailVerification_teamId_fkey" FOREIGN KEY ("teamId") REFERENCES "Team"("id") ON DELETE CASCADE ON UPDATE CASCADE;
+
+-- AddForeignKey
+ALTER TABLE "TeamTransferVerification" ADD CONSTRAINT "TeamTransferVerification_teamId_fkey" FOREIGN KEY ("teamId") REFERENCES "Team"("id") ON DELETE CASCADE ON UPDATE CASCADE;
+
+-- AddForeignKey
+ALTER TABLE "TeamMemberInvite" ADD CONSTRAINT "TeamMemberInvite_teamId_fkey" FOREIGN KEY ("teamId") REFERENCES "Team"("id") ON DELETE CASCADE ON UPDATE CASCADE;
+
+ALTER TABLE "Subscription"
+ADD CONSTRAINT teamId_or_userId_check
+CHECK (
+  (
+    "teamId" IS NOT NULL
+    AND "userId" IS NULL
+  )
+  OR (
+    "teamId" IS NULL
+    AND "userId" IS NOT NULL
+  )
+);

packages/prisma/package.json

@@ -21,7 +21,8 @@
     "@prisma/client": "5.4.2",
     "dotenv": "^16.3.1",
     "dotenv-cli": "^7.3.0",
-    "prisma": "5.4.2"
+    "prisma": "5.4.2",
+    "ts-pattern": "^5.0.6"
   },
   "devDependencies": {
     "ts-node": "^10.9.1",

packages/prisma/schema.prisma

@@ -37,6 +37,9 @@ model User {
   Document             Document[]
   Subscription         Subscription[]
   PasswordResetToken   PasswordResetToken[]
+  ownedTeams           Team[]
+  ownedPendingTeams    TeamPending[]
+  teamMembers          TeamMember[]
   twoFactorSecret      String?
   twoFactorEnabled     Boolean              @default(false)
   twoFactorBackupCodes String?
@@ -103,12 +106,14 @@ model Subscription {
   planId            String             @unique
   priceId           String
   periodEnd         DateTime?
-  userId            Int
+  userId            Int?
+  teamId            Int?               @unique
   createdAt         DateTime           @default(now())
   updatedAt         DateTime           @updatedAt
   cancelAtPeriodEnd Boolean            @default(false)
 
-  User User @relation(fields: [userId], references: [id], onDelete: Cascade)
+  team Team? @relation(fields: [teamId], references: [id], onDelete: Cascade)
+  User User? @relation(fields: [userId], references: [id], onDelete: Cascade)
 
   @@index([userId])
 }
@@ -162,6 +167,8 @@ model Document {
   updatedAt      DateTime            @default(now()) @updatedAt
   completedAt    DateTime?
   deletedAt      DateTime?
+  teamId         Int?
+  team           Team?               @relation(fields: [teamId], references: [id])
 
   @@unique([documentDataId])
   @@index([userId])
@@ -301,6 +308,104 @@ model DocumentShareLink {
   @@unique([documentId, email])
 }
 
+enum TeamMemberRole {
+  ADMIN
+  MANAGER
+  MEMBER
+}
+
+enum TeamMemberInviteStatus {
+  ACCEPTED
+  PENDING
+}
+
+model Team {
+  id                   Int                       @id @default(autoincrement())
+  name                 String
+  url                  String                    @unique
+  createdAt            DateTime                  @default(now())
+  customerId           String?                   @unique
+  ownerUserId          Int
+  members              TeamMember[]
+  invites              TeamMemberInvite[]
+  teamEmail            TeamEmail?
+  emailVerification    TeamEmailVerification?
+  transferVerification TeamTransferVerification?
+
+  owner        User          @relation(fields: [ownerUserId], references: [id])
+  subscription Subscription?
+
+  document Document[]
+}
+
+model TeamPending {
+  id          Int      @id @default(autoincrement())
+  name        String
+  url         String   @unique
+  createdAt   DateTime @default(now())
+  customerId  String   @unique
+  ownerUserId Int
+
+  owner User @relation(fields: [ownerUserId], references: [id], onDelete: Cascade)
+}
+
+model TeamMember {
+  id        Int            @id @default(autoincrement())
+  teamId    Int
+  createdAt DateTime       @default(now())
+  role      TeamMemberRole
+  userId    Int
+  user      User           @relation(fields: [userId], references: [id])
+  team      Team           @relation(fields: [teamId], references: [id], onDelete: Cascade)
+
+  @@unique([userId, teamId])
+}
+
+model TeamEmail {
+  teamId    Int      @id @unique
+  createdAt DateTime @default(now())
+  name      String
+  email     String   @unique
+  team      Team     @relation(fields: [teamId], references: [id], onDelete: Cascade)
+}
+
+model TeamEmailVerification {
+  teamId    Int      @id @unique
+  name      String
+  email     String
+  token     String   @unique
+  expiresAt DateTime
+  createdAt DateTime @default(now())
+
+  team Team @relation(fields: [teamId], references: [id], onDelete: Cascade)
+}
+
+model TeamTransferVerification {
+  teamId              Int      @id @unique
+  userId              Int
+  name                String
+  email               String
+  token               String   @unique
+  expiresAt           DateTime
+  createdAt           DateTime @default(now())
+  clearPaymentMethods Boolean  @default(false)
+
+  team Team @relation(fields: [teamId], references: [id], onDelete: Cascade)
+}
+
+model TeamMemberInvite {
+  id        Int                    @id @default(autoincrement())
+  teamId    Int
+  createdAt DateTime               @default(now())
+  email     String
+  status    TeamMemberInviteStatus @default(PENDING)
+  role      TeamMemberRole
+  token     String                 @unique
+  team      Team                   @relation(fields: [teamId], references: [id], onDelete: Cascade)
+
+  @@unique([teamId, email])
+}
+
 enum TemplateType {
   PUBLIC
   PRIVATE

packages/prisma/seed/documents.ts

@@ -0,0 +1,375 @@
+import type { User } from '@prisma/client';
+import { nanoid } from 'nanoid';
+import fs from 'node:fs';
+import path from 'node:path';
+import { match } from 'ts-pattern';
+
+import { prisma } from '..';
+import {
+  DocumentDataType,
+  DocumentStatus,
+  FieldType,
+  Prisma,
+  ReadStatus,
+  SendStatus,
+  SigningStatus,
+} from '../client';
+import { seedTeam } from './teams';
+import { seedUser } from './users';
+
+const examplePdf = fs
+  .readFileSync(path.join(__dirname, '../../../assets/example.pdf'))
+  .toString('base64');
+
+type DocumentToSeed = {
+  sender: User;
+  recipients: (User | string)[];
+  type: DocumentStatus;
+  documentOptions?: Partial<Prisma.DocumentUncheckedCreateInput>;
+};
+
+export const seedDocuments = async (documents: DocumentToSeed[]) => {
+  await Promise.all(
+    documents.map(async (document, i) =>
+      match(document.type)
+        .with(DocumentStatus.DRAFT, async () =>
+          createDraftDocument(document.sender, document.recipients, {
+            key: i,
+            createDocumentOptions: document.documentOptions,
+          }),
+        )
+        .with(DocumentStatus.PENDING, async () =>
+          createPendingDocument(document.sender, document.recipients, {
+            key: i,
+            createDocumentOptions: document.documentOptions,
+          }),
+        )
+        .with(DocumentStatus.COMPLETED, async () =>
+          createCompletedDocument(document.sender, document.recipients, {
+            key: i,
+            createDocumentOptions: document.documentOptions,
+          }),
+        )
+        .exhaustive(),
+    ),
+  );
+};
+
+const createDraftDocument = async (
+  sender: User,
+  recipients: (User | string)[],
+  options: CreateDocumentOptions = {},
+) => {
+  const { key, createDocumentOptions = {} } = options;
+
+  const documentData = await prisma.documentData.create({
+    data: {
+      type: DocumentDataType.BYTES_64,
+      data: examplePdf,
+      initialData: examplePdf,
+    },
+  });
+
+  const document = await prisma.document.create({
+    data: {
+      title: `[TEST] Document ${key} - Draft`,
+      status: DocumentStatus.DRAFT,
+      documentDataId: documentData.id,
+      userId: sender.id,
+      ...createDocumentOptions,
+    },
+  });
+
+  for (const recipient of recipients) {
+    const email = typeof recipient === 'string' ? recipient : recipient.email;
+    const name = typeof recipient === 'string' ? recipient : recipient.name ?? '';
+
+    await prisma.recipient.create({
+      data: {
+        email,
+        name,
+        token: nanoid(),
+        readStatus: ReadStatus.NOT_OPENED,
+        sendStatus: SendStatus.NOT_SENT,
+        signingStatus: SigningStatus.NOT_SIGNED,
+        signedAt: new Date(),
+        Document: {
+          connect: {
+            id: document.id,
+          },
+        },
+        Field: {
+          create: {
+            page: 1,
+            type: FieldType.NAME,
+            inserted: true,
+            customText: name,
+            positionX: new Prisma.Decimal(1),
+            positionY: new Prisma.Decimal(1),
+            width: new Prisma.Decimal(1),
+            height: new Prisma.Decimal(1),
+            documentId: document.id,
+          },
+        },
+      },
+    });
+  }
+};
+
+type CreateDocumentOptions = {
+  key?: string | number;
+  createDocumentOptions?: Partial<Prisma.DocumentUncheckedCreateInput>;
+};
+
+const createPendingDocument = async (
+  sender: User,
+  recipients: (User | string)[],
+  options: CreateDocumentOptions = {},
+) => {
+  const { key, createDocumentOptions = {} } = options;
+
+  const documentData = await prisma.documentData.create({
+    data: {
+      type: DocumentDataType.BYTES_64,
+      data: examplePdf,
+      initialData: examplePdf,
+    },
+  });
+
+  const document = await prisma.document.create({
+    data: {
+      title: `[TEST] Document ${key} - Pending`,
+      status: DocumentStatus.PENDING,
+      documentDataId: documentData.id,
+      userId: sender.id,
+      ...createDocumentOptions,
+    },
+  });
+
+  for (const recipient of recipients) {
+    const email = typeof recipient === 'string' ? recipient : recipient.email;
+    const name = typeof recipient === 'string' ? recipient : recipient.name ?? '';
+
+    await prisma.recipient.create({
+      data: {
+        email,
+        name,
+        token: nanoid(),
+        readStatus: ReadStatus.OPENED,
+        sendStatus: SendStatus.SENT,
+        signingStatus: SigningStatus.NOT_SIGNED,
+        signedAt: new Date(),
+        Document: {
+          connect: {
+            id: document.id,
+          },
+        },
+        Field: {
+          create: {
+            page: 1,
+            type: FieldType.NAME,
+            inserted: true,
+            customText: name,
+            positionX: new Prisma.Decimal(1),
+            positionY: new Prisma.Decimal(1),
+            width: new Prisma.Decimal(1),
+            height: new Prisma.Decimal(1),
+            documentId: document.id,
+          },
+        },
+      },
+    });
+  }
+};
+
+const createCompletedDocument = async (
+  sender: User,
+  recipients: (User | string)[],
+  options: CreateDocumentOptions = {},
+) => {
+  const { key, createDocumentOptions = {} } = options;
+
+  const documentData = await prisma.documentData.create({
+    data: {
+      type: DocumentDataType.BYTES_64,
+      data: examplePdf,
+      initialData: examplePdf,
+    },
+  });
+
+  const document = await prisma.document.create({
+    data: {
+      title: `[TEST] Document ${key} - Completed`,
+      status: DocumentStatus.COMPLETED,
+      documentDataId: documentData.id,
+      userId: sender.id,
+      ...createDocumentOptions,
+    },
+  });
+
+  for (const recipient of recipients) {
+    const email = typeof recipient === 'string' ? recipient : recipient.email;
+    const name = typeof recipient === 'string' ? recipient : recipient.name ?? '';
+
+    await prisma.recipient.create({
+      data: {
+        email,
+        name,
+        token: nanoid(),
+        readStatus: ReadStatus.OPENED,
+        sendStatus: SendStatus.SENT,
+        signingStatus: SigningStatus.SIGNED,
+        signedAt: new Date(),
+        Document: {
+          connect: {
+            id: document.id,
+          },
+        },
+        Field: {
+          create: {
+            page: 1,
+            type: FieldType.NAME,
+            inserted: true,
+            customText: name,
+            positionX: new Prisma.Decimal(1),
+            positionY: new Prisma.Decimal(1),
+            width: new Prisma.Decimal(1),
+            height: new Prisma.Decimal(1),
+            documentId: document.id,
+          },
+        },
+      },
+    });
+  }
+};
+
+/**
+ * Create 5 team documents:
+ * - Completed document with 2 recipients.
+ * - Pending document with 1 recipient.
+ * - Pending document with 4 recipients.
+ * - Draft document with 3 recipients.
+ * - Draft document with 2 recipients.
+ *
+ * Create 3 non team documents where the user is a team member:
+ * - Completed document with 2 recipients.
+ * - Pending document with 1 recipient.
+ * - Draft document with 2 recipients.
+ *
+ * Create 3 non team documents where the user is not a team member, but the recipient is:
+ * - Completed document with 2 recipients.
+ * - Pending document with 1 recipient.
+ * - Draft document with 2 recipients.
+ *
+ * This should result in the following team document dashboard counts:
+ * - 0 Inbox
+ * - 2 Pending
+ * - 1 Completed
+ * - 2 Draft
+ * - 5 All
+ */
+export const seedTeamDocuments = async () => {
+  const team = await seedTeam({
+    createTeamMembers: 4,
+  });
+
+  const documentOptions = {
+    teamId: team.id,
+  };
+
+  const teamMember1 = team.members[1].user;
+  const teamMember2 = team.members[2].user;
+  const teamMember3 = team.members[3].user;
+  const teamMember4 = team.members[4].user;
+
+  const [testUser1, testUser2, testUser3, testUser4] = await Promise.all([
+    seedUser(),
+    seedUser(),
+    seedUser(),
+    seedUser(),
+  ]);
+
+  await seedDocuments([
+    /**
+     * Team documents.
+     */
+    {
+      sender: teamMember1,
+      recipients: [testUser1, testUser2],
+      type: DocumentStatus.COMPLETED,
+      documentOptions,
+    },
+    {
+      sender: teamMember2,
+      recipients: [testUser1],
+      type: DocumentStatus.PENDING,
+      documentOptions,
+    },
+    {
+      sender: teamMember2,
+      recipients: [testUser1, testUser2, testUser3, testUser4],
+      type: DocumentStatus.PENDING,
+      documentOptions,
+    },
+    {
+      sender: teamMember2,
+      recipients: [testUser1, testUser2, teamMember1],
+      type: DocumentStatus.DRAFT,
+      documentOptions,
+    },
+    {
+      sender: team.owner,
+      recipients: [testUser1, testUser2],
+      type: DocumentStatus.DRAFT,
+      documentOptions,
+    },
+    /**
+     * Non team documents where the sender is a team member and recipient is not.
+     */
+    {
+      sender: teamMember1,
+      recipients: [testUser1, testUser2],
+      type: DocumentStatus.COMPLETED,
+    },
+    {
+      sender: teamMember2,
+      recipients: [testUser1],
+      type: DocumentStatus.PENDING,
+    },
+    {
+      sender: teamMember3,
+      recipients: [testUser1, testUser2],
+      type: DocumentStatus.DRAFT,
+    },
+    /**
+     * Non team documents where the sender is not a team member and recipient is.
+     */
+    {
+      sender: testUser1,
+      recipients: [teamMember1, teamMember2],
+      type: DocumentStatus.COMPLETED,
+    },
+    {
+      sender: testUser2,
+      recipients: [teamMember1],
+      type: DocumentStatus.PENDING,
+    },
+    {
+      sender: testUser3,
+      recipients: [teamMember1, teamMember2],
+      type: DocumentStatus.DRAFT,
+    },
+  ]);
+
+  return {
+    team,
+    teamMember1,
+    teamMember2,
+    teamMember3,
+    teamMember4,
+    testUser1,
+    testUser2,
+    testUser3,
+    testUser4,
+  };
+};

packages/prisma/seed/teams.ts

@@ -0,0 +1,177 @@
+import { prisma } from '..';
+import { TeamMemberInviteStatus, TeamMemberRole } from '../client';
+import { seedUser } from './users';
+
+const EMAIL_DOMAIN = `test.documenso.com`;
+
+type SeedTeamOptions = {
+  createTeamMembers?: number;
+  createTeamEmail?: true | string;
+};
+
+export const seedTeam = async ({
+  createTeamMembers = 0,
+  createTeamEmail,
+}: SeedTeamOptions = {}) => {
+  const teamUrl = `team-${Date.now()}`;
+  const teamEmail = createTeamEmail === true ? `${teamUrl}@${EMAIL_DOMAIN}` : createTeamEmail;
+
+  const teamOwner = await seedUser({
+    name: `${teamUrl}-original-owner`,
+    email: `${teamUrl}-original-owner@${EMAIL_DOMAIN}`,
+  });
+
+  const teamMembers = await Promise.all(
+    Array.from({ length: createTeamMembers }).map(async (_, i) => {
+      return seedUser({
+        name: `${teamUrl}-member-${i + 1}`,
+        email: `${teamUrl}-member-${i + 1}@${EMAIL_DOMAIN}`,
+      });
+    }),
+  );
+
+  const team = await prisma.team.create({
+    data: {
+      name: teamUrl,
+      url: teamUrl,
+      ownerUserId: teamOwner.id,
+      members: {
+        createMany: {
+          data: [teamOwner, ...teamMembers].map((user) => ({
+            userId: user.id,
+            role: TeamMemberRole.ADMIN,
+          })),
+        },
+      },
+      teamEmail: teamEmail
+        ? {
+            create: {
+              email: teamEmail,
+              name: teamEmail,
+            },
+          }
+        : undefined,
+    },
+  });
+
+  return await prisma.team.findFirstOrThrow({
+    where: {
+      id: team.id,
+    },
+    include: {
+      owner: true,
+      members: {
+        include: {
+          user: true,
+        },
+      },
+      teamEmail: true,
+    },
+  });
+};
+
+export const unseedTeam = async (teamUrl: string) => {
+  const team = await prisma.team.findUnique({
+    where: {
+      url: teamUrl,
+    },
+    include: {
+      members: true,
+    },
+  });
+
+  if (!team) {
+    return;
+  }
+
+  await prisma.team.delete({
+    where: {
+      url: teamUrl,
+    },
+  });
+
+  await prisma.user.deleteMany({
+    where: {
+      id: {
+        in: team.members.map((member) => member.userId),
+      },
+    },
+  });
+};
+
+export const seedTeamTransfer = async (options: { newOwnerUserId: number; teamId: number }) => {
+  return await prisma.teamTransferVerification.create({
+    data: {
+      teamId: options.teamId,
+      token: Date.now().toString(),
+      expiresAt: new Date(Date.now() + 1000 * 60 * 60 * 24),
+      userId: options.newOwnerUserId,
+      name: '',
+      email: '',
+    },
+  });
+};
+
+export const seedTeamEmail = async ({ email, teamId }: { email: string; teamId: number }) => {
+  return await prisma.teamEmail.create({
+    data: {
+      name: email,
+      email,
+      teamId,
+    },
+  });
+};
+
+export const unseedTeamEmail = async ({ teamId }: { teamId: number }) => {
+  return await prisma.teamEmail.delete({
+    where: {
+      teamId,
+    },
+  });
+};
+
+export const seedTeamInvite = async ({
+  email,
+  teamId,
+  role = TeamMemberRole.ADMIN,
+}: {
+  email: string;
+  teamId: number;
+  role?: TeamMemberRole;
+}) => {
+  return await prisma.teamMemberInvite.create({
+    data: {
+      email,
+      teamId,
+      role,
+      status: TeamMemberInviteStatus.PENDING,
+      token: Date.now().toString(),
+    },
+  });
+};
+
+export const seedTeamEmailVerification = async ({
+  email,
+  teamId,
+}: {
+  email: string;
+  teamId: number;
+}) => {
+  return await prisma.teamEmailVerification.create({
+    data: {
+      teamId,
+      email,
+      name: email,
+      expiresAt: new Date(Date.now() + 1000 * 60 * 60 * 24),
+      token: Date.now().toString(),
+    },
+  });
+};
+
+export const unseedTeamEmailVerification = async ({ teamId }: { teamId: number }) => {
+  return await prisma.teamEmailVerification.delete({
+    where: {
+      teamId,
+    },
+  });
+};

packages/prisma/seed/users.ts

@@ -0,0 +1,34 @@
+import { hashSync } from '@documenso/lib/server-only/auth/hash';
+
+import { prisma } from '..';
+
+type SeedUserOptions = {
+  name?: string;
+  email?: string;
+  password?: string;
+  verified?: boolean;
+};
+
+export const seedUser = async ({
+  name = `user-${Date.now()}`,
+  email = `user-${Date.now()}@test.documenso.com`,
+  password = 'password',
+  verified = true,
+}: SeedUserOptions = {}) => {
+  return await prisma.user.create({
+    data: {
+      name,
+      email,
+      password: hashSync(password),
+      emailVerified: verified ? new Date() : undefined,
+    },
+  });
+};
+
+export const unseedUser = async (userId: number) => {
+  await prisma.user.delete({
+    where: {
+      id: userId,
+    },
+  });
+};