@studious-lms/server 1.2.53 → 1.4.0

package/src/routers/file.ts

@@ -1,378 +1,55 @@
 import { z } from "zod";
-import { createTRPCRouter, protectedProcedure, protectedTeacherProcedure } from "../trpc.js";
-import { TRPCError } from "@trpc/server";
-import { getSignedUrl, deleteFile } from "../lib/googleCloudStorage.js";
-import type { User } from "@prisma/client";
-import { prisma } from "../lib/prisma.js";
-import { logger } from "../utils/logger.js";
+import {
+  createTRPCRouter,
+  protectedProcedure,
+  protectedTeacherProcedure,
+} from "../trpc.js";
+import {
+  getFileSignedUrl,
+  moveFile,
+  renameFile,
+  deleteFileRecord,
+} from "../services/file.js";
 
 export const fileRouter = createTRPCRouter({
   getSignedUrl: protectedProcedure
-    .input(z.object({
-      fileId: z.string(),
-    }))
-    .mutation(async ({ ctx, input }) => {
-      const { fileId } = input;
-      const userId = ctx.user?.id;
-
-      if (!userId) {
-        throw new TRPCError({
-          code: "UNAUTHORIZED",
-          message: "You must be logged in to access files",
-        });
-      }
-
-      // Get file metadata from database
-      const file = await prisma.file.findUnique({
-        where: { id: fileId },
-        include: {
-          assignment: {
-            include: {
-              class: {
-                include: {
-                  students: true,
-                  teachers: true
-                }
-              }
-            }
-          },
-          submission: {
-            include: {
-              student: true,
-              assignment: {
-                include: {
-                  class: {
-                    include: {
-                      teachers: true
-                    }
-                  }
-                }
-              }
-            }
-          },
-          annotations: {
-            include: {
-              student: true,
-              assignment: {
-                include: {
-                  class: {
-                    include: {
-                      teachers: true,
-                    }
-                  }
-                }
-              }
-            }
-          },
-          folder: {
-            include: {
-              class: {
-                include: {
-                  students: true,
-                  teachers: true
-                }
-              }
-            }
-          },
-          announcement: {
-            include: {
-              class: {
-                include: {
-                  students: true,
-                  teachers: true
-                }
-              }
-            }
-          }
-        }
-      });
-
-      if (!file) {
-        throw new TRPCError({
-          code: "NOT_FOUND",
-          message: "File does not exist",
-        });
-      }
-
-      // Check if user has access to this file
-      let hasAccess = false;
-
-      let classId: string | null = null;
-
-      // Check if user is a teacher of the class
-      if (file.assignment?.class) {
-        classId = file.assignment.class.id;
-        const isTeacher = file.assignment.class.teachers.some(teacher => teacher.id === userId);
-        const isStudent = file.assignment.class.students.some(student => student.id === userId);
-        logger.info(`Assignment file access check - userId: ${userId}, isTeacher: ${isTeacher}, isStudent: ${isStudent}`);
-        hasAccess = isTeacher || isStudent;
-      }
-
-      // Check if user has access to announcement files (teachers or students in the class)
-      if ((file as any).announcement?.class) {
-        classId = (file as any).announcement.class.id;
-        const isTeacher = (file as any).announcement.class.teachers.some((teacher: any) => teacher.id === userId);
-        const isStudent = (file as any).announcement.class.students.some((student: any) => student.id === userId);
-        logger.info(`Announcement file access check - userId: ${userId}, isTeacher: ${isTeacher}, isStudent: ${isStudent}`);
-        hasAccess = hasAccess || isTeacher || isStudent;
-      }
-
-      if (file.submission?.assignment?.classId) {
-        classId = file.submission.assignment.classId;
-        hasAccess = hasAccess || file.submission?.studentId === userId || false;
-        if (!hasAccess) hasAccess = file.submission.assignment.class.teachers.some(teacher => teacher.id === userId) || false;
-      }
-
-      if (file.annotations?.assignment?.classId) {
-        classId = file.annotations?.assignment.classId;
-        hasAccess = hasAccess || file.annotations?.studentId === userId || false;
-        if (!hasAccess) hasAccess = file.annotations.assignment.class.teachers.some(teacher => teacher.id === userId) || false;
-      }
-
-      // Check if user is the file owner
-      if (file.userId === userId) {
-        hasAccess = true;
-      }
-
-      // Check if file is in a folder and user has access to the class
-      if (file.folder?.class) {
-        const isTeacher = file.folder.class.teachers.some(teacher => teacher.id === userId);
-        const isStudent = file.folder.class.students.some(student => student.id === userId);
-        hasAccess = hasAccess || isTeacher;
-        hasAccess = hasAccess || isStudent;
-      }
-
-      if (!hasAccess) {
-        throw new TRPCError({
-          code: "FORBIDDEN",
-          message: "You do not have access to this file",
-        });
-      }
-
-      try {
-        const signedUrl = await getSignedUrl(file.path);
-        return { url: signedUrl };
-      } catch (error) {
-        logger.error('Error generating signed URL:', error as Record<string, any>);
-        throw new TRPCError({
-          code: "INTERNAL_SERVER_ERROR",
-          message: "Failed to generate download URL",
-        });
-      }
-    }),
+    .input(z.object({ fileId: z.string() }))
+    .mutation(({ ctx, input }) =>
+      getFileSignedUrl(ctx.user!.id, input.fileId)
+    ),
 
   move: protectedTeacherProcedure
-    .input(z.object({
-      fileId: z.string(),
-      targetFolderId: z.string(),
-      classId: z.string(),
-    }))
-    .mutation(async ({ ctx, input }) => {
-      const { fileId, targetFolderId } = input;
-
-      // Get the file
-      const file = await prisma.file.findUnique({
-        where: { id: fileId },
-        include: {
-          folder: {
-            include: {
-              class: true,
-            },
-          },
-        },
-      });
-
-      if (!file) {
-        throw new TRPCError({
-          code: "NOT_FOUND",
-          message: "File not found",
-        });
-      }
-
-      // Get the target folder
-      const targetFolder = await prisma.folder.findUnique({
-        where: { id: targetFolderId },
-        include: {
-          class: true,
-        },
-      });
-
-      if (!targetFolder) {
-        throw new TRPCError({
-          code: "NOT_FOUND",
-          message: "Target folder not found",
-        });
-      }
-
-      // Move the file
-      const updatedFile = await prisma.file.update({
-        where: { id: fileId },
-        data: {
-          folderId: targetFolderId,
-        },
-        include: {
-          user: {
-            select: {
-              id: true,
-              username: true,
-            },
-          },
-        },
-      });
-
-      return updatedFile;
-    }),
+    .input(
+      z.object({
+        fileId: z.string(),
+        targetFolderId: z.string(),
+        classId: z.string(),
+      })
+    )
+    .mutation(({ input }) =>
+      moveFile(input.fileId, input.targetFolderId)
+    ),
 
   rename: protectedTeacherProcedure
-    .input(z.object({
-      fileId: z.string(),
-      newName: z.string(),
-      classId: z.string(),
-    }))
-    .mutation(async ({ ctx, input }) => {
-      const { fileId, newName, classId } = input;
-
-      // Verify user is a teacher of the class
-      const classData = await prisma.class.findFirst({
-        where: {
-          id: classId,
-          teachers: {
-            some: {
-              id: ctx.user!.id,
-            },
-          },
-        },
-      });
-
-      if (!classData) {
-        throw new TRPCError({
-          code: "FORBIDDEN",
-          message: "You must be a teacher of this class to rename files",
-        });
-      }
-
-      // Get the file
-      const file = await prisma.file.findUnique({
-        where: { id: fileId },
-        include: {
-          folder: {
-            include: {
-              class: true,
-            },
-          },
-        },
-      });
-
-      if (!file) {
-        throw new TRPCError({
-          code: "NOT_FOUND",
-          message: "File not found",
-        });
-      }
-
-      // Validate new name
-      if (!newName.trim()) {
-        throw new TRPCError({
-          code: "BAD_REQUEST",
-          message: "File name cannot be empty",
-        });
-      }
-
-      // Rename the file
-      const updatedFile = await prisma.file.update({
-        where: { id: fileId },
-        data: {
-          name: newName.trim(),
-        },
-        include: {
-          user: {
-            select: {
-              id: true,
-              username: true,
-            },
-          },
-        },
-      });
-
-      return updatedFile;
-    }),
+    .input(
+      z.object({
+        fileId: z.string(),
+        newName: z.string(),
+        classId: z.string(),
+      })
+    )
+    .mutation(({ input }) =>
+      renameFile(input.fileId, input.newName)
+    ),
 
   delete: protectedTeacherProcedure
-    .input(z.object({
-      fileId: z.string(),
-      classId: z.string(),
-    }))
-    .mutation(async ({ ctx, input }) => {
-      const { fileId, classId } = input;
-
-      // Verify user is a teacher of the class
-      const classData = await prisma.class.findFirst({
-        where: {
-          id: classId,
-          teachers: {
-            some: {
-              id: ctx.user!.id,
-            },
-          },
-        },
-      });
-
-      if (!classData) {
-        throw new TRPCError({
-          code: "FORBIDDEN",
-          message: "You must be a teacher of this class to delete files",
-        });
-      }
-
-      // Get the file
-      const file = await prisma.file.findUnique({
-        where: { id: fileId },
-        include: {
-          folder: {
-            include: {
-              class: true,
-            },
-          },
-          thumbnail: true,
-        },
-      });
-
-      if (!file) {
-        throw new TRPCError({
-          code: "NOT_FOUND",
-          message: "File not found",
-        });
-      }
-
-      // Verify the file belongs to this class
-      if (file.folder?.classId !== classId) {
-        throw new TRPCError({
-          code: "FORBIDDEN",
-          message: "File does not belong to this class",
-        });
-      }
-
-      // Delete files from storage (only if they were actually uploaded)
-      try {
-        // Only delete from GCS if the file was successfully uploaded
-        if (file.uploadStatus === 'COMPLETED') {
-          // Delete the main file
-          await deleteFile(file.path);
-          
-          // Delete thumbnail if it exists
-          if (file.thumbnail) {
-            await deleteFile(file.thumbnail.path);
-          }
-        }
-      } catch (error) {
-        logger.warn(`Failed to delete file ${file.path}:`, error as Record<string, any>);
-      }
-
-      // Delete the file record from database
-      await prisma.file.delete({
-        where: { id: fileId },
-      });
-
-      return { success: true };
-    }),
-}); 
+    .input(
+      z.object({
+        fileId: z.string(),
+        classId: z.string(),
+      })
+    )
+    .mutation(({ input }) =>
+      deleteFileRecord(input.fileId, input.classId)
+    ),
+});