first commit
This commit is contained in:
569
calcom/packages/prisma/seed-insights.ts
Normal file
569
calcom/packages/prisma/seed-insights.ts
Normal file
@@ -0,0 +1,569 @@
|
||||
import type { Prisma } from "@prisma/client";
|
||||
import { PrismaClient } from "@prisma/client";
|
||||
import { v4 as uuidv4 } from "uuid";
|
||||
|
||||
import dayjs from "@calcom/dayjs";
|
||||
import { hashPassword } from "@calcom/features/auth/lib/hashPassword";
|
||||
import { BookingStatus } from "@calcom/prisma/enums";
|
||||
|
||||
function getRandomRatingFeedback() {
|
||||
const feedbacks = [
|
||||
"Great chat!",
|
||||
"Okay-ish",
|
||||
"Quite Poor",
|
||||
"Excellent chat!",
|
||||
"Could be better",
|
||||
"Wonderful!",
|
||||
];
|
||||
return feedbacks[Math.floor(Math.random() * feedbacks.length)];
|
||||
}
|
||||
|
||||
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");
|
||||
}
|
||||
|
||||
booking.rating = Math.floor(Math.random() * 5) + 1; // Generates a random rating from 1 to 5
|
||||
booking.ratingFeedback = getRandomRatingFeedback(); // Random feedback from a predefined list
|
||||
booking.noShowHost = Math.random() < 0.5;
|
||||
|
||||
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: {
|
||||
create: {
|
||||
hash: 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: {
|
||||
create: {
|
||||
hash: 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: {
|
||||
create: {
|
||||
hash: 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);
|
||||
});
|
||||
Reference in New Issue
Block a user