@studious-lms/server 1.4.2 → 1.4.4

package/src/services/message.ts

@@ -2,9 +2,14 @@
  * Message service – send, update, delete messages; mark as read; track mentions.
  * Broadcasts real-time updates via Pusher.
  */
+import type { Prisma } from "@prisma/client";
 import { TRPCError } from "@trpc/server";
 import { prisma } from "../lib/prisma.js";
 import { chatChannel, pusher } from "../lib/pusher.js";
+import {
+  confirmDirectUpload,
+  createDirectUploadFiles,
+} from "../lib/fileUpload.js";
 import { logger } from "../utils/logger.js";
 import {
   findConversationMembership,
@@ -16,6 +21,126 @@ import {
   countUnreadMentions,
 } from "../models/message.js";
 
+export function ensureMessageHasContentOrAttachments(
+  content: string,
+  fileIds: string[] = []
+) {
+  if (!content.trim() && fileIds.length === 0) {
+    throw new TRPCError({
+      code: "BAD_REQUEST",
+      message: "Message must include content or attachments",
+    });
+  }
+}
+
+export async function claimMessageAttachments(
+  tx: Prisma.TransactionClient,
+  userId: string,
+  conversationId: string,
+  messageId: string,
+  fileIds: string[] = []
+) {
+  const uniqueFileIds = [...new Set(fileIds)];
+
+  if (uniqueFileIds.length === 0) {
+    return;
+  }
+
+  const { count } = await tx.file.updateMany({
+    where: {
+      id: { in: uniqueFileIds },
+      userId,
+      conversationId,
+      uploadStatus: "COMPLETED",
+      messageId: null,
+    },
+    data: {
+      messageId,
+    },
+  });
+
+  if (count !== uniqueFileIds.length) {
+    throw new TRPCError({
+      code: "BAD_REQUEST",
+      message: "Some attachments are missing, unfinished, or unavailable",
+    });
+  }
+}
+
+export async function getMessageUploadUrls(
+  userId: string,
+  input: {
+    conversationId: string;
+    files: { name: string; type: string; size: number }[];
+  }
+) {
+  const membership = await findConversationMembership(input.conversationId, userId);
+  if (!membership) {
+    throw new TRPCError({
+      code: "FORBIDDEN",
+      message: "Not a member of this conversation",
+    });
+  }
+
+  return createDirectUploadFiles(
+    input.files,
+    userId,
+    undefined,
+    undefined,
+    undefined,
+    undefined,
+    input.conversationId
+  );
+}
+
+export async function confirmMessageUpload(
+  userId: string,
+  input: {
+    conversationId: string;
+    fileId: string;
+    uploadSuccess: boolean;
+    errorMessage?: string;
+  }
+) {
+  const membership = await findConversationMembership(input.conversationId, userId);
+  if (!membership) {
+    throw new TRPCError({
+      code: "FORBIDDEN",
+      message: "Not a member of this conversation",
+    });
+  }
+
+  const file = await prisma.file.findFirst({
+    where: {
+      id: input.fileId,
+      userId,
+      conversationId: input.conversationId,
+      messageId: null,
+    },
+    select: { id: true },
+  });
+
+  if (!file) {
+    throw new TRPCError({
+      code: "NOT_FOUND",
+      message: "File not found or you do not have permission",
+    });
+  }
+
+  await confirmDirectUpload(
+    input.fileId,
+    input.uploadSuccess,
+    input.errorMessage
+  );
+
+  return {
+    success: true,
+    message: input.uploadSuccess
+      ? "Upload confirmed successfully"
+      : "Upload failed",
+  };
+}
+
 /** List messages in a conversation with cursor-based pagination. */
 export async function listMessages(
   userId: string,
@@ -74,9 +199,18 @@ export async function sendMessage(
     conversationId: string;
     content: string;
     mentionedUserIds?: string[];
+    fileIds?: string[];
   }
 ) {
-  const { conversationId, content, mentionedUserIds = [] } = input;
+  const {
+    conversationId,
+    content,
+    mentionedUserIds = [],
+    fileIds = [],
+  } = input;
+
+  ensureMessageHasContentOrAttachments(content, fileIds);
+  const normalizedContent = content.trim();
 
   const membership = await findConversationMembership(conversationId, userId);
   if (!membership) {
@@ -102,10 +236,36 @@ export async function sendMessage(
   const result = await prisma.$transaction(async (tx) => {
     const message = await tx.message.create({
       data: {
-        content,
+        content: normalizedContent,
         senderId: userId,
         conversationId,
       },
+    });
+
+    await claimMessageAttachments(
+      tx,
+      userId,
+      conversationId,
+      message.id,
+      fileIds
+    );
+
+    if (mentionedUserIds.length > 0) {
+      await tx.mention.createMany({
+        data: mentionedUserIds.map((mentionedUserId) => ({
+          messageId: message.id,
+          userId: mentionedUserId,
+        })),
+      });
+    }
+
+    await tx.conversation.update({
+      where: { id: conversationId },
+      data: { updatedAt: new Date() },
+    });
+
+    return tx.message.findUniqueOrThrow({
+      where: { id: message.id },
       include: {
         sender: {
           select: {
@@ -124,22 +284,6 @@ export async function sendMessage(
         },
       },
     });
-
-    if (mentionedUserIds.length > 0) {
-      await tx.mention.createMany({
-        data: mentionedUserIds.map((mentionedUserId) => ({
-          messageId: message.id,
-          userId: mentionedUserId,
-        })),
-      });
-    }
-
-    await tx.conversation.update({
-      where: { id: conversationId },
-      data: { updatedAt: new Date() },
-    });
-
-    return message;
   });
 
   try {
@@ -173,6 +317,7 @@ export async function sendMessage(
     conversationId: result.conversationId,
     createdAt: result.createdAt,
     sender: result.sender,
+    attachments: result.attachments ?? [],
     mentionedUserIds,
   };
 }

package/src/services/newtonChat.ts

@@ -7,6 +7,10 @@ import { logger } from "../utils/logger.js";
 import { chatChannel, pusher } from "../lib/pusher.js";
 import { isAIUser } from "../utils/aiUser.js";
 import { generateAndSendNewtonIntroduction, generateAndSendNewtonResponse } from "../pipelines/aiNewtonChat.js";
+import {
+  claimMessageAttachments,
+  ensureMessageHasContentOrAttachments,
+} from "./message.js";
 import {
   findClassWithStudent,
   findSubmissionByAssignmentAndStudent,
@@ -118,7 +122,8 @@ export async function postToNewtonChat(
   userId: string,
   newtonChatId: string,
   content: string,
-  mentionedUserIds: string[] = []
+  mentionedUserIds: string[] = [],
+  fileIds: string[] = []
 ) {
   const newtonChat = await findNewtonChatByIdAndMember(newtonChatId, userId);
   if (!newtonChat) {
@@ -141,29 +146,33 @@ export async function postToNewtonChat(
     }
   }
 
+  ensureMessageHasContentOrAttachments(content, fileIds);
+  const normalizedContent = content.trim();
+
+  if (!normalizedContent) {
+    throw new TRPCError({
+      code: "BAD_REQUEST",
+      message: "Newton chat messages must include text content",
+    });
+  }
+
   const result = await prisma.$transaction(async (tx) => {
     const message = await tx.message.create({
       data: {
-        content,
+        content: normalizedContent,
         senderId: userId,
         conversationId: newtonChat.conversationId,
       },
-      include: {
-        sender: {
-          select: {
-            id: true,
-            username: true,
-            profile: {
-              select: {
-                displayName: true,
-                profilePicture: true,
-              },
-            },
-          },
-        },
-      },
     });
 
+    await claimMessageAttachments(
+      tx,
+      userId,
+      newtonChat.conversationId,
+      message.id,
+      fileIds
+    );
+
     if (mentionedUserIds.length > 0) {
       await tx.mention.createMany({
         data: mentionedUserIds.map((mentionedUserId) => ({
@@ -178,7 +187,26 @@ export async function postToNewtonChat(
       data: { updatedAt: new Date() },
     });
 
-    return message;
+    return tx.message.findUniqueOrThrow({
+      where: { id: message.id },
+      include: {
+        sender: {
+          select: {
+            id: true,
+            username: true,
+            profile: {
+              select: {
+                displayName: true,
+                profilePicture: true,
+              },
+            },
+          },
+        },
+        attachments: {
+          select: { id: true, name: true, type: true },
+        },
+      },
+    });
   });
 
   try {
@@ -192,7 +220,9 @@ export async function postToNewtonChat(
         conversationId: result.conversationId,
         createdAt: result.createdAt,
         sender: result.sender,
+        attachments: result.attachments ?? [],
         mentionedUserIds,
+        meta: {},
       }
     );
   } catch (error) {
@@ -202,7 +232,7 @@ export async function postToNewtonChat(
   if (!isAIUser(userId)) {
     generateAndSendNewtonResponse(
       newtonChatId,
-      content,
+      normalizedContent,
       newtonChat.conversationId,
       newtonChat.submission
     ).catch((error) => {
@@ -217,6 +247,7 @@ export async function postToNewtonChat(
     conversationId: result.conversationId,
     createdAt: result.createdAt,
     sender: result.sender,
+    attachments: result.attachments ?? [],
     mentionedUserIds,
   };
 }

package/src/services/notification.ts

@@ -1,12 +1,13 @@
 /**
  * Notification service – list, send, and mark notifications as read.
  */
+import { TRPCError } from "@trpc/server";
 import {
   findNotificationsByReceiverId,
-  findNotificationById,
+  findNotificationByIdForReceiver,
   createNotification,
   createManyNotifications,
-  markNotificationAsRead,
+  markNotificationAsReadForReceiver,
 } from "../models/notification.js";
 
 /** List notifications for a user. */
@@ -14,9 +15,18 @@ export async function listNotifications(userId: string) {
   return findNotificationsByReceiverId(userId);
 }
 
-/** Get a single notification by ID. */
-export async function getNotification(id: string) {
-  return findNotificationById(id);
+/** Get a single notification by ID for its receiver. */
+export async function getNotification(userId: string, id: string) {
+  const notification = await findNotificationByIdForReceiver(id, userId);
+
+  if (!notification) {
+    throw new TRPCError({
+      code: "NOT_FOUND",
+      message: "Notification not found",
+    });
+  }
+
+  return notification;
 }
 
 /** Send a notification to one user. */
@@ -44,7 +54,16 @@ export async function sendToMultiple(input: {
   return result;
 }
 
-/** Mark a notification as read. */
-export async function markAsRead(id: string) {
-  return markNotificationAsRead(id);
+/** Mark a notification as read for its receiver. */
+export async function markAsRead(userId: string, id: string) {
+  const notification = await markNotificationAsReadForReceiver(id, userId);
+
+  if (!notification) {
+    throw new TRPCError({
+      code: "NOT_FOUND",
+      message: "Notification not found",
+    });
+  }
+
+  return notification;
 }

package/src/services/studentProgress.ts

@@ -635,7 +635,10 @@ export async function chatAboutStudentProgress(
   ];
 
   try {
-    const response = await inference<string>(messages);
+    const response = await inference<string>(messages, undefined, {
+      reason: "student_progress_chat",
+      metadata: { viewerId, classId: input.classId, studentId: input.studentId },
+    });
     if (typeof response !== "string" || response.trim().length === 0) {
       throw new Error("Student progress chat returned an empty response");
     }