rallly/packages/database/prisma/seed/scheduled-events.ts

import { ScheduledEventInviteStatus } from "@prisma/client";
import { Prisma, ScheduledEventStatus } from "@prisma/client"; // Ensure Prisma is imported
import dayjs from "dayjs";
import { faker } from "@faker-js/faker";

import { prisma } from "@rallly/database";
import { randInt } from "./utils";

// Realistic event titles and descriptions
function generateEventDetails() {
  const titles = [
    "Team Sync Meeting",
    "Product Strategy Session",
    "Design Review",
    "Engineering Standup",
    "Client Check-in Call",
    "Marketing Campaign Kickoff",
    "Sales Pipeline Review",
    "HR Training Workshop",
    "Finance Budget Planning",
    "All Hands Company Update",
    "Sprint Retrospective",
    "User Research Debrief",
    "Technical Deep Dive",
    "Content Calendar Planning",
    "Partnership Discussion",
  ];
  const descriptions = [
    "Discussing project updates and blockers.",
    "Aligning on the product roadmap for the next quarter.",
    "Gathering feedback on the latest UI mockups.",
    "Quick daily updates from the engineering team.",
    "Reviewing progress and addressing client concerns.",
    "Launching the new social media campaign.",
    "Analyzing the current sales funnel and opportunities.",
    "Mandatory compliance training session.",
    "Meeting to finalize budget decisions across different departments.",
    "Sharing company performance and upcoming goals.",
    "Reflecting on the past sprint, celebrating successes, and identifying areas for improvement.",
    "Sharing key insights gathered from recent user research sessions.",
    "Exploring the architecture of the new microservice.",
    "Planning blog posts and social media content for the month.",
    "Exploring potential collaboration opportunities with external partners.",
    "Team building activity to foster collaboration.",
    "Workshop on improving presentation skills.",
    "Onboarding session for new hires.",
    "Reviewing customer feedback and planning improvements.",
    "Brainstorming session for new feature ideas.",
  ];

  return {
    title: faker.helpers.arrayElement(titles),
    description: faker.helpers.arrayElement(descriptions),
  };
}

async function createScheduledEventForUser(userId: string) {
  const { title, description } = generateEventDetails();
  const isAllDay = Math.random() < 0.3; // ~30% chance of being all-day

  let startTime: Date;
  let endTime: Date | null;

  if (isAllDay) {
    const startDate = dayjs(
      faker.datatype.boolean() ? faker.date.past(1) : faker.date.soon(30),
    )
      .startOf("day")
      .toDate();
    startTime = startDate;

    // Decide if it's a multi-day event
    const isMultiDay = Math.random() < 0.2; // ~20% chance of multi-day
    if (isMultiDay) {
      const durationDays = faker.datatype.number({ min: 1, max: 3 });
      // End date is the start of the day *after* the last full day
      endTime = dayjs(startDate)
        .add(durationDays + 1, "day")
        .toDate();
    } else {
      // Single all-day event ends at the start of the next day
      endTime = dayjs(startDate).add(1, "day").toDate();
    }
  } else {
    // Generate times for non-all-day events
    startTime = dayjs(
      faker.datatype.boolean() ? faker.date.past(1) : faker.date.soon(30),
    )
      .second(faker.helpers.arrayElement([0, 15, 30, 45])) // Add some variance
      .minute(faker.helpers.arrayElement([0, 15, 30, 45]))
      .hour(faker.datatype.number({ min: 8, max: 20 })) // Wider range for hours
      .toDate();
    const durationMinutes = faker.helpers.arrayElement([30, 60, 90, 120, 180]); // Longer durations possible
    endTime = dayjs(startTime).add(durationMinutes, "minute").toDate();
  }

  // Use only valid statuses from the schema
  const status = faker.helpers.arrayElement<ScheduledEventStatus>([
    ScheduledEventStatus.confirmed,
    ScheduledEventStatus.canceled,
  ]);
  const timeZone = faker.address.timeZone();

  const data: Prisma.ScheduledEventCreateInput = {
    title,
    description,
    start: startTime, // Use correct model field name 'start'
    end: endTime, // Use correct model field name 'end'
    timeZone,
    status, // Assign the randomly selected valid status
    user: { connect: { id: userId } }, // Connect to existing user
    allDay: isAllDay,
    location: faker.datatype.boolean()
      ? faker.address.streetAddress()
      : undefined,
    // Add invites (optional, example below)
    invites: {
      create: Array.from({ length: randInt(5, 0) }).map(() => ({
        inviteeEmail: faker.internet.email(),
        inviteeName: faker.name.fullName(),
        inviteeTimeZone: faker.address.timeZone(),
        status: faker.helpers.arrayElement<ScheduledEventInviteStatus>([
          "accepted",
          "declined",
          "tentative",
          "pending",
        ]),
      })),
    },
  };

  await prisma.scheduledEvent.create({ data });
}

export async function seedScheduledEvents(userId: string) {
  console.info("Seeding scheduled events...");
  const eventPromises = Array.from({ length: 15 }).map((_, i) =>
    createScheduledEventForUser(userId),
  );

  await Promise.all(eventPromises);

  console.info(`✓ Seeded scheduled events for ${userId}`);
}