Merge pull request #4270 from Northeastern-Electric-Racing/feature/seed-data-revamp

Feature/seed data revamp merging first 3 commits

Author: wavehassman

package.json

@@ -71,6 +71,7 @@
     "@types/react-dom": "17.0.1"
   },
   "dependencies": {
+    "@faker-js/faker": "^10.4.0",
     "@microsoft/clarity": "^1.0.0",
     "@types/multer": "^1.4.12",
     "canvas-confetti": "^1.9.3",

src/backend/package.json

@@ -7,7 +7,8 @@
     "test": "vitest",
     "build": "NODE_OPTIONS='--max-old-space-size=8192' tsc --noEmit false",
     "start": "node -r dotenv/config dist/backend/index.js",
-    "prisma:manual": "tsx --import dotenv/config ./src/prisma/manual.ts"
+    "prisma:manual": "tsx --import dotenv/config ./src/prisma/manual.ts",
+    "prisma:dev-seed": "tsx --import dotenv/config ./src/prisma/dev-seed.ts"
   },
   "dependencies": {
     "@prisma/client": "^6.2.1",

src/backend/src/prisma/context.ts

@@ -0,0 +1,76 @@
+import {
+  Account_Code,
+  Description_Bullet_Type,
+  Index_Code,
+  Link_Type,
+  Manufacturer,
+  Material_Type,
+  Organization,
+  Prisma,
+  Reimbursement_Product_Other_Reason,
+  Role,
+  Team_Type,
+  Unit,
+  Vendor
+} from '@prisma/client';
+import { RoleEnum } from 'shared';
+import { getUserQueryArgs } from '../prisma-query-args/user.query-args.js';
+
+// Organization Context
+export type OrganizationContext = {
+  organization: Organization;
+  bootstrapUserId: string;
+};
+
+// User Context
+export type FullUser = Prisma.UserGetPayload<ReturnType<typeof getUserQueryArgs>>;
+
+export type UsersContext = {
+  appAdmins: FullUser[];
+  admins: FullUser[];
+  heads: FullUser[];
+  leadership: FullUser[];
+  members: FullUser[];
+  guests: FullUser[];
+  all: FullUser[];
+};
+
+// Role Context
+export type RoleContext = {
+  roles: Role[];
+  rolesByType: Record<RoleEnum, Role[]>;
+};
+
+// Config Data Context
+export type ConfigDataContext = {
+  teamTypes: Team_Type[];
+  linkTypes: Link_Type[];
+  descriptionBulletTypes: Description_Bullet_Type[];
+  materialTypes: Material_Type[];
+  manufacturers: Manufacturer[];
+  units: Unit[];
+  accountCodes: Account_Code[];
+  indexCodes: Index_Code[];
+  vendors: Vendor[];
+  reimbursementProductOtherReasons: Reimbursement_Product_Other_Reason[];
+};
+
+// Main Seed Context
+export type SeedContext = OrganizationContext & UsersContext & RoleContext & CarOutput & ConfigDataContext;
+
+// Car Context
+export type DateRange = {
+  start: Date;
+  end: Date;
+};
+
+export type CarContext = {
+  car: Prisma.CarGetPayload<{ include: { wbsElement: true } }>;
+  year: number;
+  dateRange: DateRange;
+};
+
+export type CarOutput = {
+  cars: CarContext[];
+  currentYearCar: CarContext;
+};

src/backend/src/prisma/dates.ts

@@ -0,0 +1,16 @@
+import { Faker } from '@faker-js/faker';
+/*
+  https://fakerjs.dev/api/date.html
+*/
+
+interface WithFaker {
+  faker: Faker;
+}
+
+export function generateRandomDate({ faker }: WithFaker, from?: Date, to?: Date) {
+  return faker.date.between({ from: from ?? '2000-01-01', to: to ?? Date.now() });
+}
+
+export function generateRandomDateAround({ faker }: WithFaker, date: Date) {
+  return faker.date.recent({ days: 5, refDate: date });
+}

src/backend/src/prisma/dev-seed.ts

@@ -0,0 +1,48 @@
+import { Prisma } from '@prisma/client';
+import { base, en, Faker } from '@faker-js/faker';
+import { DateRange } from '../context.js';
+
+const CURRENT_YEAR = new Date().getFullYear();
+const CAR_COUNT = 5;
+
+const FROM_MONTH = 3; // April
+const FROM_DAY = 1;
+const TO_MONTH = 8; // September
+const TO_DAY = 30;
+
+export const CAR_CONFIGS = Array.from({ length: CAR_COUNT }, (_, i) => {
+  const carYear = CURRENT_YEAR - (CAR_COUNT - 1) + i;
+  const shortYear = String(carYear).slice(2);
+
+  const carFaker = new Faker({ locale: [en, base] });
+  carFaker.seed(carYear);
+
+  const start = carFaker.date.between({
+    from: new Date(carYear - 1, FROM_MONTH, FROM_DAY),
+    to: new Date(carYear - 1, TO_MONTH, TO_DAY)
+  });
+
+  const end = carFaker.date.between({
+    from: new Date(carYear, FROM_MONTH, FROM_DAY),
+    to: new Date(carYear, TO_MONTH, TO_DAY)
+  });
+
+  return {
+    name: `NER-${shortYear}`,
+    carNumber: carYear - 2000,
+    year: carYear,
+    dateRange: { start, end } as DateRange
+  };
+});
+
+export const carCreateInput = (name: string, carNumber: number, organizationId: string): Prisma.CarCreateInput => ({
+  wbsElement: {
+    create: {
+      name,
+      carNumber,
+      projectNumber: 0,
+      workPackageNumber: 0,
+      organizationId
+    }
+  }
+});