feat: able to transfer vehicle to business

SnailyCAD · Oct 6, 2023 · 8462a8f · 8462a8f
1 parent 13e8271
commit 8462a8f
Show file tree

Hide file tree

Showing 2 changed files with 44 additions and 16 deletions.
diff --git a/apps/api/src/controllers/citizen/vehicles-controller.ts b/apps/api/src/controllers/citizen/vehicles-controller.ts
@@ -39,6 +39,7 @@ import { getLastOfArray, manyToManyHelper } from "lib/data/many-to-many";
 import { AllowedFileExtension, allowedFileExtensions } from "@snailycad/config";
 import { getImageWebPPath } from "~/lib/images/get-image-webp-path";
 import fs from "node:fs/promises";
+import { ExtendedNotFound } from "~/exceptions/extended-not-found";
 
 @Controller("/vehicles")
 @UseBeforeEach(IsAuth)
@@ -484,25 +485,51 @@ export class VehiclesController {
       throw new NotFound("vehicleNotFound");
     }
 
-    const newOwner = await prisma.citizen.findFirst({
-      where: {
-        AND: [{ id: data.ownerId }, { NOT: { id: String(vehicle.citizenId) } }],
-      },
-    });
+    if (data.businessId) {
+      const business = await prisma.business.findUnique({
+        where: { id: data.businessId },
+      });
 
-    if (!newOwner) {
-      throw new NotFound("newOwnerNotFound");
+      if (!business) {
+        throw new ExtendedNotFound({ business: "businessNotFound" });
+      }
+
+      const updatedVehicle = await prisma.registeredVehicle.update({
+        where: { id: vehicle.id },
+        data: {
+          Business: { connect: { id: data.businessId } },
+        },
+      });
+
+      return updatedVehicle;
     }
 
-    const updatedVehicle = await prisma.registeredVehicle.update({
-      where: { id: vehicle.id },
-      data: {
-        citizenId: newOwner.id,
-        userId: newOwner.userId,
-      },
-    });
+    if (data.ownerId) {
+      const newOwner = await prisma.citizen.findFirst({
+        where: {
+          AND: [{ id: data.ownerId }, { NOT: { id: String(vehicle.citizenId) } }],
+        },
+      });
 
-    return updatedVehicle;
+      if (!newOwner) {
+        throw new NotFound("newOwnerNotFound");
+      }
+
+      const updatedVehicle = await prisma.registeredVehicle.update({
+        where: { id: vehicle.id },
+        data: {
+          citizenId: newOwner.id,
+          userId: newOwner.userId,
+        },
+      });
+
+      return updatedVehicle;
+    }
+
+    throw new ExtendedBadRequest({
+      ownerId: "ownerIdOrBusinessIdRequired",
+      businessId: "ownerIdOrBusinessIdRequired",
+    });
   }
 
   @Delete("/:id")

diff --git a/packages/schemas/src/citizen.ts b/packages/schemas/src/citizen.ts
@@ -63,7 +63,8 @@ export const VEHICLE_SCHEMA = z.object({
 });
 
 export const TRANSFER_VEHICLE_SCHEMA = z.object({
-  ownerId: z.string().min(2).max(255),
+  ownerId: z.string().max(255).nullish(),
+  businessId: z.string().max(255).nullish(),
 });
 
 export const DELETE_VEHICLE_SCHEMA = z.object({