Add Folder Id when Uploadin a File

Author: ThomasCode92

src/app/api/uploadthing/core.ts

@@ -1,26 +1,35 @@
+/* eslint-disable @typescript-eslint/only-throw-error */
+
 import { auth } from "@clerk/nextjs/server";
 import { createUploadthing, type FileRouter } from "uploadthing/next";
 import { UploadThingError } from "uploadthing/server";
+import { z } from "zod";
 
 import * as mutations from "~/server/db/mutations";
+import * as queries from "~/server/db/queries";
 
 const f = createUploadthing();
 
 export const ourFileRouter = {
   imageUploader: f({ image: { maxFileSize: "4MB", maxFileCount: 1 } })
-    .middleware(async () => {
+    .input(z.object({ folderId: z.number() }))
+    .middleware(async ({ input }) => {
       const user = await auth();
-
-      // eslint-disable-next-line @typescript-eslint/only-throw-error
       if (!user.userId) throw new UploadThingError("Unauthorized");
 
-      return { userId: user.userId };
+      const folder = await queries.getFolderById(input.folderId);
+      if (!folder) throw new UploadThingError("Folder not found");
+
+      if (folder.ownerId !== user.userId)
+        throw new UploadThingError("Unauthorized");
+
+      return { userId: user.userId, parent: folder.id };
     })
     .onUploadComplete(async ({ metadata, file }) => {
       console.log("Upload complete for userId:", metadata.userId);
       console.log("file url", file.ufsUrl);
 
-      const uploadedFile = { ...file, parent: 0 };
+      const uploadedFile = { ...file, parent: metadata.parent };
       await mutations.createFile(uploadedFile, metadata.userId);
 
       return { uploadedBy: metadata.userId };

src/server/db/mutations.ts

@@ -1,6 +1,9 @@
 import { db } from "~/server/db";
 import { type File, files_table as filesSchema } from "~/server/db/schema";
 
-export function createFile(file: Omit<File, "id" | "parent">, _userId: string) {
-  return db.insert(filesSchema).values({ ...file, parent: 1 });
+export function createFile(
+  file: Pick<File, "name" | "size" | "url" | "parent">,
+  userId: string,
+) {
+  return db.insert(filesSchema).values({ ...file, ownerId: userId });
 }

src/server/db/queries.ts

@@ -33,6 +33,14 @@ export function getAllFolders(folderId: number) {
     .where(eq(folderSchema.parent, folderId));
 }
 
+export async function getFolderById(folderId: number) {
+  const folders = await db
+    .select()
+    .from(folderSchema)
+    .where(eq(folderSchema.id, folderId));
+  return folders[0];
+}
+
 export function getAllFiles(folderId: number) {
   return db.select().from(fileSchema).where(eq(fileSchema.parent, folderId));
 }