cal.pub0.org/packages/prisma/seed-insights.ts

import type { Prisma } from "@prisma/client";
import { BookingStatus, PrismaClient } from "@prisma/client";
import { v4 as uuidv4 } from "uuid";

import dayjs from "@calcom/dayjs";
import { hashPassword } from "@calcom/features/auth/lib/hashPassword";

const shuffle = (
  booking: any,
  year: number,
  eventTypesToPick: Prisma.EventTypeGetPayload<{
    select: {
      id: true;
      userId: true;
      title: true;
      length: true;
      description: true;
    };
  }>[],
  usersIdsToPick?: number[]
) => {
  const startTime = dayjs(booking.startTime)
    .add(Math.floor(Math.random() * 365), "day")
    .add(Math.floor(Math.random() * 24), "hour")
    .add(Math.floor(Math.random() * 60), "minute")
    .set("y", year);
  const randomEvent = eventTypesToPick[Math.floor(Math.random() * eventTypesToPick.length)];

  const endTime = dayjs(startTime).add(Math.floor(Math.random() * randomEvent.length), "minute");

  booking.title = randomEvent.title;
  booking.description = randomEvent.description;
  booking.startTime = startTime.toISOString();
  booking.endTime = endTime.toISOString();
  booking.createdAt = startTime.subtract(1, "day").toISOString();

  // Pick a random status
  const randomStatusIndex = Math.floor(Math.random() * Object.keys(BookingStatus).length);
  const statusKey = Object.keys(BookingStatus)[randomStatusIndex];

  booking.status = BookingStatus[statusKey];

  booking.rescheduled = Math.random() > 0.5 && Math.random() > 0.5 && Math.random() > 0.5;

  if (booking.rescheduled) {
    booking.status = "CANCELLED";
  }
  const randomEventTypeId = randomEvent.id;

  booking.eventTypeId = randomEventTypeId;
  booking.uid = uuidv4();

  if (usersIdsToPick && usersIdsToPick.length > 0) {
    booking.userId = Math.random() > 0.5 ? usersIdsToPick[0] : usersIdsToPick[1];
  } else {
    booking.userId = randomEvent.userId;
  }

  if (booking.userId === undefined || booking.userId === null) {
    console.log({ randomEvent, usersIdsToPick });
    console.log("This should not happen");
  }

  return booking;
};

const prisma = new PrismaClient();
async function main() {
  // First find if not then create everything
  let insightsAdmin = await prisma.user.findFirst({
    where: {
      email: "insights@example.com",
    },
  });

  if (!insightsAdmin) {
    insightsAdmin = await prisma.user.create({
      data: {
        email: "insights@example.com",
        password: await hashPassword("insightsinsightsinsights...!"),
        name: "Insights Admin",
        role: "ADMIN",
        username: "insights-admin",
        completedOnboarding: true,
      },
    });

    if (!insightsAdmin) {
      // Throw error
      throw new Error("Insights Admin not created");
    }
  }

  let insightsUser = await prisma.user.findFirst({
    where: {
      email: "insightsuser@example.com",
    },
  });

  if (!insightsUser) {
    insightsUser = await prisma.user.create({
      data: {
        email: "insightsuser@example.com",
        password: await hashPassword("insightsuser"),
        name: "Insights User",
        role: "USER",
        username: "insights-user",
        completedOnboarding: true,
      },
    });
    if (!insightsUser) {
      // Throw error
      throw new Error("Insights Admin not created");
    }
  }

  let insightsTeam = await prisma.team.findFirst({
    where: {
      slug: "insights-team",
    },
  });

  if (!insightsTeam) {
    insightsTeam = await prisma.team.create({
      data: {
        name: "Insights",
        slug: "insights-team",
      },
    });
    await prisma.membership.createMany({
      data: [
        {
          teamId: insightsTeam.id,
          userId: insightsAdmin.id,
          role: "OWNER",
          accepted: true,
        },
        {
          teamId: insightsTeam.id,
          userId: insightsUser.id,
          role: "MEMBER",
          accepted: true,
        },
      ],
    });
    if (!insightsTeam) {
      insightsTeam = await prisma.team.create({
        data: {
          name: "Insights",
          slug: "insights-team",
        },
      });
    }

    if (insightsAdmin && insightsTeam) {
      await prisma.team.update({
        where: {
          id: insightsTeam.id,
        },
        data: {
          members: {
            connect: {
              userId_teamId: {
                userId: insightsAdmin.id,
                teamId: insightsTeam.id,
              },
            },
          },
        },
      });
    }
  }

  let teamEvents = await prisma.eventType.findMany({
    where: {
      teamId: insightsTeam.id,
    },
  });

  const userSingleEventsAdmin = await prisma.eventType.findMany({
    where: {
      userId: insightsUser.id,
    },
  });

  const userSingleEvents = await prisma.eventType.findMany({
    where: {
      userId: insightsUser.id,
    },
  });

  if (userSingleEventsAdmin.length === 0 && userSingleEvents.length === 0) {
    await prisma.eventType.create({
      data: {
        title: "Single Event Admin",
        slug: "single-event-admin",
        description: "Single Event Admin",
        length: 60,
        userId: insightsAdmin.id,
        users: {
          connect: {
            id: insightsAdmin.id,
          },
        },
      },
    });
    await prisma.eventType.create({
      data: {
        title: "Single Event",
        slug: "single-event",
        description: "Single Event",
        length: 60,
        userId: insightsUser.id,
        users: {
          connect: {
            id: insightsUser.id,
          },
        },
      },
    });
    const singleEventTypesAdmin = await prisma.eventType.findFirst({
      select: {
        id: true,
        userId: true,
        title: true,
        length: true,
        description: true,
      },
      where: {
        userId: insightsAdmin.id,
      },
    });

    const singleEventTypes = await prisma.eventType.findFirst({
      select: {
        id: true,
        userId: true,
        length: true,
        title: true,
        description: true,
      },
      where: {
        userId: insightsUser.id,
      },
    });

    if (!singleEventTypesAdmin || !singleEventTypes) {
      throw new Error("Could not create single event types");
    }

    const singleEvents = [singleEventTypesAdmin, singleEventTypes];

    // create bookings random for single event type
    const baseBookingSingle: Prisma.BookingCreateManyInput = {
      uid: "demoUIDSingle",
      title: "Single Event",
      description: "Single Event",
      startTime: dayjs().toISOString(),
      endTime: dayjs().toISOString(),
      // This gets overriden in shuffle
      userId: insightsUser.id,
    };

    // Validate if they are null of undefined throw error
    if (!insightsAdmin || !insightsUser || !insightsAdmin.id || !insightsUser.id) {
      throw new Error("Could not find user");
    }

    await prisma.booking.createMany({
      data: [
        ...new Array(10000)
          .fill(0)
          .map(() => shuffle({ ...baseBookingSingle }, dayjs().get("y") - 1, singleEvents)),
      ],
    });

    await prisma.booking.createMany({
      data: [
        ...new Array(10000)
          .fill(0)
          .map(() => shuffle({ ...baseBookingSingle }, dayjs().get("y") - 0, singleEvents)),
      ],
    });
  }

  if (teamEvents.length === 0) {
    await prisma.eventType.createMany({
      data: [
        {
          title: "Team Meeting",
          slug: "team-meeting",
          description: "Team Meeting",
          length: 60,
          teamId: insightsTeam.id,
          schedulingType: "ROUND_ROBIN",
        },
        {
          title: "Team Lunch",
          slug: "team-lunch",
          description: "Team Lunch",
          length: 30,
          teamId: insightsTeam.id,
          schedulingType: "COLLECTIVE",
        },
        {
          title: "Team Coffee",
          slug: "team-coffee",
          description: "Team Coffee",
          length: 15,
          teamId: insightsTeam.id,
          schedulingType: "COLLECTIVE",
        },
      ],
    });
    teamEvents = await prisma.eventType.findMany({
      where: {
        teamId: insightsTeam.id,
      },
    });
  }

  const baseBooking: Prisma.BookingCreateManyInput = {
    uid: "demoUID",
    title: "Team Meeting should be changed in shuffle",
    description: "Team Meeting Should be changed in shuffle",
    startTime: dayjs().toISOString(),
    endTime: dayjs().toISOString(),
    userId: insightsUser.id,
    eventTypeId: teamEvents[0].id,
  };

  // Create past bookings -2y, -1y, -0y
  await prisma.booking.createMany({
    data: [
      ...new Array(10000)
        .fill(0)
        .map(() =>
          shuffle({ ...baseBooking }, dayjs().get("y") - 2, teamEvents, [
            insightsAdmin?.id ?? 0,
            insightsUser?.id ?? 0,
          ])
        ),
    ],
  });

  await prisma.booking.createMany({
    data: [
      ...new Array(10000)
        .fill(0)
        .map(() =>
          shuffle({ ...baseBooking }, dayjs().get("y") - 1, teamEvents, [
            insightsAdmin?.id ?? 0,
            insightsUser?.id ?? 0,
          ])
        ),
    ],
  });

  await prisma.booking.createMany({
    data: [
      ...new Array(10000)
        .fill(0)
        .map(() =>
          shuffle({ ...baseBooking }, dayjs().get("y"), teamEvents, [
            insightsAdmin?.id ?? 0,
            insightsUser?.id ?? 0,
          ])
        ),
    ],
  });
}
main()
  .then(async () => {
    await prisma.$disconnect();
  })
  .catch(async (e) => {
    console.error(e);
    await prisma.user.deleteMany({
      where: {
        email: {
          in: ["insights@example", "insightsuser@example.com"],
        },
      },
    });

    await prisma.team.deleteMany({
      where: {
        slug: "insights",
      },
    });

    await prisma.$disconnect();
    process.exit(1);
  });

/**
 * This will create many users in insights teams with bookings 1y in the past
 * Should be run after the main function is executed once
 */
async function createPerformanceData() {
  const createExtraMembers = false; // Turn ON to be executed
  let extraMembersIds;
  const insightsTeam = await prisma.team.findFirst({
    where: {
      slug: "insights-team",
    },
  });

  if (createExtraMembers) {
    const insightsRandomUsers: Prisma.UserCreateManyArgs["data"] = [];
    const numInsightsUsers = 50; // Change this value to adjust the number of insights users to create
    for (let i = 0; i < numInsightsUsers; i++) {
      const timestamp = Date.now();
      const email = `insightsuser${timestamp}@example.com`;
      const insightsUser = {
        email,
        password: await hashPassword("insightsuser"),
        name: `Insights User ${timestamp}`,
        username: `insights-user-${timestamp}`,
        completedOnboarding: true,
      };
      insightsRandomUsers.push(insightsUser);
    }

    await prisma.user.createMany({
      data: insightsRandomUsers,
    });
    // Lets find the ids of the users we just created
    extraMembersIds = await prisma.user.findMany({
      where: {
        email: {
          in: insightsRandomUsers.map((user) => user.email),
        },
      },
      select: {
        id: true,
      },
    });
  }

  if (createExtraMembers) {
    if (insightsTeam === null) {
      console.log("This should not happen");
      throw new Error("Insights team id is undefined or null");
    }

    await prisma.membership.createMany({
      data: extraMembersIds.map((memberId) => ({
        teamId: insightsTeam?.id ?? 1,
        userId: memberId.id,
        role: "MEMBER",
        accepted: true,
      })),
    });

    const updateMemberPromises = extraMembersIds.map((memberId) =>
      prisma.team.update({
        where: {
          id: insightsTeam?.id,
        },
        data: {
          members: {
            connect: {
              userId_teamId: {
                userId: memberId.id,
                teamId: insightsTeam?.id ?? 1,
              },
            },
          },
        },
      })
    );

    await Promise.all(updateMemberPromises);

    // Create events for every Member id
    const createEvents = extraMembersIds.map((memberId) => ({
      title: `Single Event User - ${memberId.id}`,
      slug: `single-event-user-${memberId.id}`,
      description: `Single Event User - ${memberId.id}`,
      length: 30,
      userId: memberId.id,
      users: {
        connect: {
          id: memberId.id,
        },
      },
    }));
    const createEventPromises = createEvents.map((data) =>
      prisma.eventType.create({
        data,
      })
    );
    await Promise.all(createEventPromises);

    // load the events we just created
    const singleEventsCreated = await prisma.eventType.findMany({
      where: {
        userId: {
          in: extraMembersIds.map((memberId) => memberId.id),
        },
      },
    });

    // Create bookings for every single event
    const baseBooking = {
      uid: "demo performance data  booking",
      title: "Single Event Booking Perf",
      description: "Single Event Booking Perf",
      startTime: dayjs().toISOString(),
      endTime: dayjs().toISOString(),
      eventTypeId: singleEventsCreated[0].id,
    };

    await prisma.booking.createMany({
      data: [
        ...new Array(10000)
          .fill(0)
          .map(() => shuffle({ ...baseBooking }, dayjs().get("y"), singleEventsCreated)),
      ],
    });
  }
}

createPerformanceData()
  .then(async () => {
    await prisma.$disconnect();
  })
  .catch(async (e) => {
    console.error(e);
    await prisma.user.deleteMany({
      where: {
        username: {
          contains: "insights-user-",
        },
      },
    });
    await prisma.$disconnect();
    process.exit(1);
  });