Skip to content

Commit

Permalink
updated seed files
Browse files Browse the repository at this point in the history
  • Loading branch information
@teetangh
teetangh committed Dec 12, 2024
1 parent bac34ff commit f7d5d96
Showing 1 changed file with 122 additions and 62 deletions.
184 changes: 122 additions & 62 deletions prisma/seedFiles/createSlotsOfAvailability.ts
View file
Original file line number Diff line number Diff line change
Expand Up @@ -3,89 +3,149 @@ import { DayOfWeek, ScheduleType } from "@prisma/client";
import prisma from "../../lib/prisma";
import { UserWithProfiles } from "./createUsers";

const NUM_SLOTS_PER_CONSULTANT = 20;
const MAX_SLOT_DURATION = 6; // 6 hours
const MIN_SLOT_DURATION = 0.5; // 30 minutes
const MIN_BREAK_DURATION = 0.5; // 30 minutes
const MAX_SLOTS_PER_DAY = 4;

function generateSlotTime(existingSlots: Array<{ start: number; end: number }>) {
// Keep trying until we find a valid slot
let attempts = 0;
while (attempts < 50) { // Prevent infinite loops
// Generate random start hour (0-23)
const startHour = faker.number.int({ min: 0, max: 23 });
// Randomly decide if we want to start at half hour
const startMinute = faker.helpers.arrayElement([0, 0.5]);
const start = startHour + startMinute;

// Generate random duration between 30 mins and 6 hours
const possibleDurations = Array.from(
{ length: MAX_SLOT_DURATION * 2 }, // *2 because we're counting in half hours
(_, i) => (i + 1) * 0.5 // Generate durations from 0.5 to 6 in 0.5 increments
);
const duration = faker.helpers.arrayElement(possibleDurations);
const end = start + duration;

// Verify this slot doesn't overlap with existing slots
const hasOverlap = existingSlots.some(slot => {
// Add MIN_BREAK_DURATION to ensure minimum break between slots
return !(end + MIN_BREAK_DURATION <= slot.start || start >= slot.end + MIN_BREAK_DURATION);
});

if (!hasOverlap) {
return { start, end };
}

attempts++;
}
return null; // Couldn't find a valid slot
}

function generateDaySlots() {
const slots: Array<{ start: number; end: number }> = [];
const numSlots = faker.number.int({ min: 1, max: MAX_SLOTS_PER_DAY });

for (let i = 0; i < numSlots; i++) {
const slot = generateSlotTime(slots);
if (slot) {
slots.push(slot);
}
}

// Sort slots by start time
return slots.sort((a, b) => a.start - b.start);
}

export async function createSlotsOfAvailability(
consultants: UserWithProfiles[],
) {
console.log(
`Creating slots of availability for ${consultants.length} consultants...`,
);

for (let i = 0; i < consultants.length; i++) {
const consultant = consultants[i];
if (!consultant.consultantProfile) {
console.warn(`Skipping consultant ${consultant.id} - no profile found`);
continue;
}

try {
const slotType = consultant.consultantProfile.scheduleType;

if (slotType === ScheduleType.WEEKLY) {
// Create weekly slots
for (let j = 0; j < NUM_SLOTS_PER_CONSULTANT; j++) {
const dayOfWeek = faker.helpers.arrayElement(
Object.values(DayOfWeek),
);
const startHour = faker.helpers.arrayElement([
8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23,
]);
const durationHours = faker.helpers.arrayElement([
0.5, 1, 1.5, 2, 2.5, 3,
]);

const startTime = new Date();
startTime.setUTCHours(startHour, startHour % 1 === 0 ? 0 : 30, 0, 0);

const endTime = new Date(startTime);
endTime.setTime(startTime.getTime() + durationHours * 60 * 60 * 1000);

let endDayOfWeek = dayOfWeek;
if (endTime.getUTCHours() < startTime.getUTCHours()) {
// If end time is on the next day, adjust the day of week
const daysOfWeek = Object.values(DayOfWeek);
const currentIndex = daysOfWeek.indexOf(dayOfWeek);
endDayOfWeek = daysOfWeek[(currentIndex + 1) % 7];
}
// Create weekly slots for each day
const daysOfWeek = Object.values(DayOfWeek);

for (const dayOfWeek of daysOfWeek) {
const daySlots = generateDaySlots();

for (const slot of daySlots) {
const startHour = Math.floor(slot.start);
const startMinute = (slot.start % 1) * 60;

const endHour = Math.floor(slot.end);
const endMinute = (slot.end % 1) * 60;

await prisma.slotOfAvailabilityWeekly.create({
data: {
consultantProfileId: consultant.consultantProfile.id,
dayOfWeekforStartTimeInUTC: dayOfWeek,
slotStartTimeInUTC: startTime,
dayOfWeekforEndTimeInUTC: endDayOfWeek,
slotEndTimeInUTC: endTime,
},
});
const startTime = new Date();
startTime.setUTCHours(startHour, startMinute, 0, 0);

const endTime = new Date();
endTime.setUTCHours(endHour, endMinute, 0, 0);

// If end time is before start time, it means the slot crosses midnight
if (endTime <= startTime) {
endTime.setDate(endTime.getDate() + 1);
}

await prisma.slotOfAvailabilityWeekly.create({
data: {
consultantProfileId: consultant.consultantProfile.id,
dayOfWeekforStartTimeInUTC: dayOfWeek,
slotStartTimeInUTC: startTime,
// If slot crosses midnight, end day is next day
dayOfWeekforEndTimeInUTC: endTime <= startTime ?
daysOfWeek[(daysOfWeek.indexOf(dayOfWeek) + 1) % 7] : dayOfWeek,
slotEndTimeInUTC: endTime,
},
});
}
}
} else {
// Create custom slots
for (let j = 0; j < NUM_SLOTS_PER_CONSULTANT; j++) {
const startDate = faker.date.future();
const startHour = faker.helpers.arrayElement([
8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23,
]);
const durationHours = faker.helpers.arrayElement([
0.5, 1, 1.5, 2, 2.5, 3,
]);

const startTime = new Date(startDate);
startTime.setUTCHours(startHour, startHour % 1 === 0 ? 0 : 30, 0, 0);

const endTime = new Date(startTime);
endTime.setTime(startTime.getTime() + durationHours * 60 * 60 * 1000);

if (endTime < startTime) {
// If end time is on the next day, add one day to the end time
endTime.setDate(endTime.getDate() + 1);
}
// Create custom slots for next 7 days
const startDate = new Date();
for (let day = 0; day < 7; day++) {
const date = new Date(startDate);
date.setDate(date.getDate() + day);

const daySlots = generateDaySlots();

for (const slot of daySlots) {
const startHour = Math.floor(slot.start);
const startMinute = (slot.start % 1) * 60;

const endHour = Math.floor(slot.end);
const endMinute = (slot.end % 1) * 60;

await prisma.slotOfAvailabilityCustom.create({
data: {
consultantProfileId: consultant.consultantProfile.id,
slotStartTimeInUTC: startTime,
slotEndTimeInUTC: endTime,
},
});
const startTime = new Date(date);
startTime.setUTCHours(startHour, startMinute, 0, 0);

const endTime = new Date(date);
endTime.setUTCHours(endHour, endMinute, 0, 0);

// If end time is before start time, it means the slot crosses midnight
if (endTime <= startTime) {
endTime.setDate(endTime.getDate() + 1);
}

await prisma.slotOfAvailabilityCustom.create({
data: {
consultantProfileId: consultant.consultantProfile.id,
slotStartTimeInUTC: startTime,
slotEndTimeInUTC: endTime,
},
});
}
}
}
} catch (error) {
Expand Down

0 comments on commit f7d5d96

Please sign in to comment.