Merge pull request #4295 from Northeastern-Electric-Racing/#4240-shop-machinery-data

#4240 Generate Shop and Machinery

Author: wavehassman

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

@@ -11,6 +11,7 @@ import { UsersProcess } from './seed/user.process.js';
 import { OrganizationProcess } from './seed/organization.process.js';
 import { CarProcess } from './seed/car.process.js';
 import { ConfigDataProcess } from './seed/config-data.process.js';
+import { ShopProcess } from './seed/shop.process.js';
 import { TeamProcess } from './seed/team.process.js';
 import { SchedulingProcess } from './seed/scheduling.process.js';
 
@@ -25,7 +26,8 @@ await new SeedRunner()
     new UsersProcess(),
     new ConfigDataProcess(),
     new SchedulingProcess(),
-    new TeamProcess()
+    new TeamProcess(),
+    new ShopProcess()
   )
   .run();
 

src/backend/src/prisma/factories/config-data.factory.ts

@@ -1,15 +1,8 @@
 import { Prisma } from '@prisma/client';
+import { connectOrganization, connectUser } from '../utils/common.factory.js';
 
 const SEED_CREATED_AT = new Date('2024-01-01T00:00:00.000Z');
 
-const connectOrganization = (organizationId: string) => ({
-  connect: { organizationId }
-});
-
-const connectUser = (userId: string) => ({
-  connect: { userId }
-});
-
 export const teamTypeCreateInputs = (organizationId: string): Prisma.Team_TypeCreateInput[] => [
   {
     name: 'Mechanical',

src/backend/src/prisma/factories/shop.factory.ts

@@ -0,0 +1,91 @@
+import { Prisma } from '@prisma/client';
+import { connectUser, connectOrganization } from '../utils/common.factory.js';
+
+type ShopConfig = {
+  name: string;
+  description: string;
+};
+
+type MachineryConfig = {
+  name: string;
+  shopName: string;
+  quantity: number;
+};
+
+export const shopConfigs: ShopConfig[] = [
+  { name: 'Forsyth Machine Shop', description: 'MIE Club Machine Shop' },
+  { name: 'Richards 054', description: 'Main Richards Makerspace' },
+  { name: 'Richards 055', description: 'Richards Basement Shop' },
+  { name: 'Richards 050', description: 'THE BAYYYYYY WOOOO' }, // Fire description
+  { name: 'EXP Machine Shop', description: 'Basement of EXP' }
+];
+
+export const machineryConfigs: MachineryConfig[] = [
+  // Forsyth Machine Shop
+  { name: 'Front Tormach 1100MX', shopName: 'Forsyth Machine Shop', quantity: 1 },
+  { name: 'Middle Tormach 1100MX', shopName: 'Forsyth Machine Shop', quantity: 1 },
+  { name: 'Rear Tormach 1100MX', shopName: 'Forsyth Machine Shop', quantity: 1 },
+  { name: 'Right Tormach 1100MX', shopName: 'Forsyth Machine Shop', quantity: 1 },
+  { name: 'Front Manual Mill', shopName: 'Forsyth Machine Shop', quantity: 1 },
+  { name: 'Rear Manual Mill', shopName: 'Forsyth Machine Shop', quantity: 1 },
+  { name: 'Front Manual Lathe', shopName: 'Forsyth Machine Shop', quantity: 1 },
+  { name: 'ProtoTRAK CNC Lathe', shopName: 'Forsyth Machine Shop', quantity: 1 },
+  { name: 'Tormach 24R CNC Router', shopName: 'Forsyth Machine Shop', quantity: 1 },
+  { name: 'Haas VF2 YT', shopName: 'Forsyth Machine Shop', quantity: 1 },
+  { name: 'Cold Cut Saw', shopName: 'Forsyth Machine Shop', quantity: 1 },
+  { name: 'Rear Manual Lathe', shopName: 'Forsyth Machine Shop', quantity: 1 },
+  { name: 'Vertical Bandsaw', shopName: 'Forsyth Machine Shop', quantity: 1 },
+  { name: 'Horizontal Bandsaw', shopName: 'Forsyth Machine Shop', quantity: 1 },
+
+  // Richards 054
+  { name: 'Hydraulic Press', shopName: 'Richards 054', quantity: 1 },
+  { name: 'Arbor Press (Baja)', shopName: 'Richards 054', quantity: 1 },
+  { name: 'Sheet Metal Brake', shopName: 'Richards 054', quantity: 1 },
+
+  // Richards 055
+  { name: 'Protomax Waterjet', shopName: 'Richards 055', quantity: 1 },
+  { name: 'Drill Press', shopName: 'Richards 055', quantity: 1 },
+  { name: 'Belt/Rotary Sander', shopName: 'Richards 055', quantity: 1 },
+  { name: 'Bench Grinder', shopName: 'Richards 055', quantity: 2 },
+  { name: 'Vertical Bandsaw', shopName: 'Richards 055', quantity: 1 },
+  { name: 'Horizontal Bandsaw', shopName: 'Richards 055', quantity: 1 },
+
+  // Richards 050
+  { name: 'Bambu Lab P1P', shopName: 'Richards 050', quantity: 1 },
+  { name: 'Bambu Lab H2S', shopName: 'Richards 050', quantity: 1 },
+
+  // EXP Machine Shop
+  { name: 'Tormach 1100MX', shopName: 'EXP Machine Shop', quantity: 1 },
+  { name: 'Omax Waterjet', shopName: 'EXP Machine Shop', quantity: 1 }
+];
+
+export const shopCreateInput = (
+  userCreatedId: string,
+  organizationId: string,
+  config: ShopConfig
+): Prisma.ShopCreateInput => ({
+  name: config.name,
+  description: config.description,
+  userCreated: connectUser(userCreatedId),
+  organization: connectOrganization(organizationId)
+});
+
+export const machineryCreateInput = (
+  userCreatedId: string,
+  organizationId: string,
+  config: MachineryConfig
+): Prisma.MachineryCreateInput => ({
+  name: config.name,
+  userCreated: connectUser(userCreatedId),
+  organization: connectOrganization(organizationId)
+});
+
+export const shopMachineryCreateInput = (
+  shopId: string,
+  machineryId: string,
+  quantity: number
+): Prisma.Shop_MachineryCreateInput => ({
+  shop: { connect: { shopId } },
+  machinery: { connect: { machineryId } },
+  quantity
+});

src/backend/src/prisma/seed/shop.process.ts

@@ -0,0 +1,65 @@
+import { Machinery, Shop, Shop_Machinery } from '@prisma/client';
+import { OrganizationOutput, OrganizationProcess } from './organization.process.js';
+import { UsersOutput, UsersProcess } from './user.process.js';
+import {
+  machineryConfigs,
+  machineryCreateInput,
+  shopConfigs,
+  shopCreateInput,
+  shopMachineryCreateInput
+} from '../factories/shop.factory.js';
+import { SeedProcess } from '../processes/seed-process.js';
+
+type ShopInput = OrganizationOutput & UsersOutput;
+
+export type ShopOutput = {
+  shops: Shop[];
+  machinery: Machinery[];
+  shopMachinery: Shop_Machinery[];
+};
+
+export class ShopProcess extends SeedProcess<ShopInput, ShopOutput> {
+  dependencies() {
+    return [OrganizationProcess, UsersProcess];
+  }
+
+  async run({ organization, appAdmins }: ShopInput): Promise<ShopOutput> {
+    const { organizationId } = organization;
+    const [creator] = appAdmins;
+
+    if (!creator) throw new Error('ShopProcess requires at least one app admin user.');
+
+    const shops = await Promise.all(
+      shopConfigs.map((config) =>
+        this.prisma.shop.create({
+          data: shopCreateInput(creator.userId, organizationId, config)
+        })
+      )
+    );
+
+    const shopIdsByName = shops.reduce<Record<string, string>>((acc, shop) => {
+      acc[shop.name] = shop.shopId;
+      return acc;
+    }, {});
+
+    const machinery = await Promise.all(
+      machineryConfigs.map((config) =>
+        this.prisma.machinery.create({
+          data: machineryCreateInput(creator.userId, organizationId, config)
+        })
+      )
+    );
+
+    const shopMachinery = await Promise.all(
+      machineryConfigs.map((config, i) => {
+        const shopId = shopIdsByName[config.shopName];
+        if (!shopId) throw new Error(`Missing shop for machinery: ${config.shopName}`);
+        return this.prisma.shop_Machinery.create({
+          data: shopMachineryCreateInput(shopId, machinery[i].machineryId, config.quantity)
+        });
+      })
+    );
+
+    return { shops, machinery, shopMachinery };
+  }
+}

src/backend/src/prisma/utils/common.factory.ts

@@ -0,0 +1,2 @@
+export const connectUser = (userId: string) => ({ connect: { userId } });
+export const connectOrganization = (organizationId: string) => ({ connect: { organizationId } });