@studious-lms/server 1.20.0 → 1.21.0

package/src/routers/assignment.ts

@@ -180,6 +180,30 @@ const confirmAnnotationUploadSchema = z.object({
   errorMessage: z.string().optional(),
 });
 
+const listFileAnnotationsSchema = z.object({
+  classId: z.string(),
+  submissionId: z.string(),
+  fileId: z.string(),
+});
+
+const upsertFileAnnotationSchema = z.object({
+  classId: z.string(),
+  submissionId: z.string(),
+  fileId: z.string(),
+  id: z.string().optional(),
+  page: z.number().int().min(1),
+  type: z.enum(["HIGHLIGHT", "INK", "COMMENT", "TEXT"]),
+  geometry: z.record(z.string(), z.unknown()),
+  text: z.string().optional(),
+  color: z.string().optional(),
+});
+
+const deleteFileAnnotationSchema = z.object({
+  classId: z.string(),
+  submissionId: z.string(),
+  id: z.string(),
+});
+
 const updateUploadProgressSchema = z.object({
   fileId: z.string(),
   progress: z.number().min(0).max(100),
@@ -966,6 +990,186 @@ export const assignmentRouter = createTRPCRouter({
       };
     }),
 
+  listFileAnnotations: protectedClassMemberProcedure
+    .input(listFileAnnotationsSchema)
+    .query(async ({ ctx, input }) => {
+      const { classId, submissionId, fileId } = input;
+
+      const submission = await prisma.submission.findFirst({
+        where: {
+          id: submissionId,
+          assignment: { classId },
+        },
+        select: {
+          id: true,
+          studentId: true,
+          returned: true,
+          assignment: {
+            select: {
+              class: {
+                select: {
+                  teachers: { where: { id: ctx.user!.id }, select: { id: true } },
+                },
+              },
+            },
+          },
+        },
+      });
+
+      if (!submission) {
+        throw new TRPCError({
+          code: "NOT_FOUND",
+          message: "Submission not found",
+        });
+      }
+
+      const isTeacher = submission.assignment.class.teachers.length > 0;
+      const isOwningStudent = submission.studentId === ctx.user!.id;
+
+      if (!isTeacher && !(isOwningStudent && submission.returned)) {
+        throw new TRPCError({
+          code: "FORBIDDEN",
+          message: "You don't have access to these annotations",
+        });
+      }
+
+      const annotations = await prisma.fileAnnotation.findMany({
+        where: { submissionId, fileId },
+        orderBy: { createdAt: "asc" },
+        include: {
+          author: {
+            select: {
+              id: true,
+              username: true,
+              profile: { select: { displayName: true } },
+            },
+          },
+        },
+      });
+
+      return { annotations };
+    }),
+
+  upsertFileAnnotation: protectedTeacherProcedure
+    .input(upsertFileAnnotationSchema)
+    .mutation(async ({ ctx, input }) => {
+      const { classId, submissionId, fileId, id, page, type, geometry, text, color } = input;
+
+      const submission = await prisma.submission.findFirst({
+        where: {
+          id: submissionId,
+          assignment: {
+            classId,
+            class: {
+              teachers: {
+                some: { id: ctx.user!.id },
+              },
+            },
+          },
+        },
+        select: { id: true },
+      });
+
+      if (!submission) {
+        throw new TRPCError({
+          code: "NOT_FOUND",
+          message: "Submission not found or you are not a teacher of this class",
+        });
+      }
+
+      const file = await prisma.file.findFirst({
+        where: {
+          id: fileId,
+          OR: [{ submissionId }, { annotationId: submissionId }],
+        },
+        select: { id: true },
+      });
+
+      if (!file) {
+        throw new TRPCError({
+          code: "NOT_FOUND",
+          message: "File does not belong to this submission",
+        });
+      }
+
+      if (id) {
+        const existing = await prisma.fileAnnotation.findFirst({
+          where: { id, submissionId, fileId },
+          select: { id: true },
+        });
+
+        if (!existing) {
+          throw new TRPCError({
+            code: "NOT_FOUND",
+            message: "Annotation not found",
+          });
+        }
+
+        const annotation = await prisma.fileAnnotation.update({
+          where: { id },
+          data: {
+            page,
+            type,
+            geometry: geometry as object,
+            text: text ?? null,
+            color: color ?? null,
+          },
+        });
+
+        return { annotation };
+      }
+
+      const annotation = await prisma.fileAnnotation.create({
+        data: {
+          fileId,
+          submissionId,
+          authorId: ctx.user!.id,
+          page,
+          type,
+          geometry: geometry as object,
+          text: text ?? null,
+          color: color ?? null,
+        },
+      });
+
+      return { annotation };
+    }),
+
+  deleteFileAnnotation: protectedTeacherProcedure
+    .input(deleteFileAnnotationSchema)
+    .mutation(async ({ ctx, input }) => {
+      const { classId, submissionId, id } = input;
+
+      const annotation = await prisma.fileAnnotation.findFirst({
+        where: {
+          id,
+          submissionId,
+          submission: {
+            assignment: {
+              classId,
+              class: {
+                teachers: {
+                  some: { id: ctx.user!.id },
+                },
+              },
+            },
+          },
+        },
+        select: { id: true },
+      });
+
+      if (!annotation) {
+        throw new TRPCError({
+          code: "NOT_FOUND",
+          message: "Annotation not found or you are not a teacher of this class",
+        });
+      }
+
+      await prisma.fileAnnotation.delete({ where: { id } });
+
+      return { success: true };
+    }),
+
   updateUploadProgress: protectedProcedure
     .input(updateUploadProgressSchema)
     .mutation(async ({ ctx, input }) => {