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

235 lines
5.8 KiB
TypeScript

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 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("insights"),
name: "Insights Admin",
role: "ADMIN",
username: "insights",
completedOnboarding: true,
},
});
}
let insightsUser = await prisma.user.findFirst({
where: {
email: "insightuser@example.com",
},
});
if (!insightsUser) {
insightsUser = await prisma.user.create({
data: {
email: "insightuser@example.com",
password: await hashPassword("insightsuser"),
name: "Insights User",
role: "USER",
username: "insights-user",
completedOnboarding: true,
},
});
}
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,
},
],
});
await prisma.team.update({
where: {
id: insightsTeam.id,
},
data: {
members: {
connect: {
userId_teamId: {
userId: insightsAdmin.id,
teamId: insightsTeam.id,
},
},
},
},
});
}
await prisma.team.update({
where: {
id: insightsTeam.id,
},
data: {
members: {
connect: {
userId_teamId: {
userId: insightsUser.id,
teamId: insightsTeam.id,
},
},
},
},
});
let teamEvents = await prisma.eventType.findMany({
where: {
teamId: insightsTeam.id,
},
});
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",
},
],
});
} else {
teamEvents = await prisma.eventType.findMany({
where: {
teamId: insightsTeam.id,
},
});
}
const baseBooking: Prisma.BookingCreateManyInput = {
uid: "demoUID",
title: "Team Meeting",
description: "Team Meeting",
startTime: dayjs().toISOString(),
endTime: dayjs().toISOString(),
userId: insightsUser.id,
eventTypeId: teamEvents[0].id,
};
const shuffle = (booking: typeof baseBooking, year) => {
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 = teamEvents[Math.floor(Math.random() * teamEvents.length)];
const endTime = dayjs(startTime).add(Math.floor(Math.random() * randomEvent.length), "minute");
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 = teamEvents[Math.floor(Math.random() * teamEvents.length)].id;
booking.eventTypeId = randomEventTypeId;
booking.uid = uuidv4();
if (insightsUser && insightsAdmin) {
booking.userId = Math.random() > 0.5 ? insightsUser.id : insightsAdmin.id;
}
return booking;
};
// Create past bookings
await prisma.booking.createMany({
data: [...new Array(100).fill(0).map(() => shuffle({ ...baseBooking }, dayjs().get("y") - 2))],
});
await prisma.booking.createMany({
data: [...new Array(100).fill(0).map(() => shuffle({ ...baseBooking }, dayjs().get("y") - 1))],
});
await prisma.booking.createMany({
data: [...new Array(100).fill(0).map(() => shuffle({ ...baseBooking }, dayjs().get("y")))],
});
}
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);
});