Fix chat creation

Author: hussein-m-kandil

src/api/v1/chats/chats.service.ts

@@ -10,7 +10,7 @@ import db from '@/lib/db';
 
 const createPaginationArgs = (
   filters: Types.BasePaginationFilters & { orderBy: 'createdAt' | 'updatedAt' },
-  limit = 10
+  limit = 10,
 ) => {
   return {
     ...(filters.cursor ? { cursor: { id: filters.cursor }, skip: 1 } : {}),
@@ -36,7 +36,7 @@ const createChatAggregation = () => {
 
 const createCurrentProfileChatArgs = (
   currentProfile: Prisma.ProfileGetPayload<{ include: { user: true } }>,
-  now: Date
+  now: Date,
 ) => {
   return {
     profileName: currentProfile.user.username,
@@ -48,7 +48,7 @@ const createCurrentProfileChatArgs = (
 
 const prepareChat = (
   chat: Prisma.ChatGetPayload<{ include: ReturnType<typeof createChatAggregation> }>,
-  currentProfile: Profile
+  currentProfile: Profile,
 ) => {
   if (currentProfile.tangible) {
     chat.profiles = chat.profiles.map((p) => {
@@ -59,7 +59,7 @@ const prepareChat = (
     });
   } else {
     chat.profiles = chat.profiles.map((p) =>
-      p.profileId === currentProfile.id ? p : { ...p, lastSeenAt: null }
+      p.profileId === currentProfile.id ? p : { ...p, lastSeenAt: null },
     );
   }
   return chat;
@@ -69,23 +69,22 @@ export const createChat = async (
   user: Types.PublicUser,
   data: Schema.ValidChat,
   imageData?: Types.ImageFullData,
-  uploadedImage?: Storage.UploadedImageData
+  uploadedImage?: Storage.UploadedImageData,
 ) => {
   return await Utils.handleDBKnownErrors(
     db.$transaction(async (tx) => {
-      // Get current user's profile
-      const currentProfile = await tx.profile.findUnique({
-        where: { userId: user.id },
+      const allProfiles = await tx.profile.findMany({
+        where: { OR: [{ id: { in: data.profiles } }, { userId: user.id }] },
         include: { user: true },
+        distinct: 'id',
       });
+      const otherProfiles: typeof allProfiles = [];
+      const currentProfile = allProfiles.reduce<(typeof allProfiles)[number] | null>(
+        (acc, curr) => (curr.user.id === user.id ? curr : (otherProfiles.push(curr), acc)),
+        null,
+      );
       if (!currentProfile) throw new AppNotFoundError('Profile not found');
       const currentProfileChatArgs = createCurrentProfileChatArgs(currentProfile, new Date());
-      // Get all the other profiles
-      const otherProfiles = await tx.profile.findMany({
-        where: { id: { in: data.profiles } },
-        include: { user: true },
-        distinct: 'id',
-      });
       const chatAggregation = createChatAggregation();
       const messageArgs = {
         profileName: currentProfile.user.username,
@@ -101,19 +100,15 @@ export const createChat = async (
         profileId: currentProfile.id,
       };
       // Find chats with the same owner and same group of profiles (there should be at most one)
-      const existentChats = await tx.chat.findMany({
-        where: {
-          managers: { some: { profileId: currentProfile.id, role: 'OWNER' } },
-          profiles: {
-            every: {
-              profileName: {
-                in: [currentProfile.user.username, ...otherProfiles.map((p) => p.user.username)],
-              },
-            },
+      const existentChats = (
+        await tx.chat.findMany({
+          where: {
+            managers: { some: { profileId: currentProfile.id, role: 'OWNER' } },
+            profiles: { every: { profileName: { in: allProfiles.map((p) => p.user.username) } } },
           },
-        },
-        include: chatAggregation,
-      });
+          include: { ...chatAggregation, _count: { select: { profiles: true } } },
+        })
+      ).filter((c) => c._count.profiles === allProfiles.length); // Filter out the profiles with a mismatched count.
       // Upsert a chat
       let chat: Prisma.ChatGetPayload<{ include: typeof chatAggregation }>;
       if (existentChats.length > 0) {
@@ -163,7 +158,7 @@ export const createChat = async (
         });
       }
       return prepareChat(chat, currentProfile);
-    })
+    }),
   );
 };
 
@@ -187,13 +182,13 @@ export const deleteChat = async (userId: User['id'], chatId: Chat['id']) => {
           }
         }
       }
-    })
+    }),
   );
 };
 
 export const getUserChats = async (
   userId: User['id'],
-  filters: Types.BasePaginationFilters = {}
+  filters: Types.BasePaginationFilters = {},
 ) => {
   return await Utils.handleDBKnownErrors(
     db.$transaction(async (tx) => {
@@ -214,7 +209,7 @@ export const getUserChats = async (
         include: createChatAggregation(),
       });
       return chats.map((chat) => prepareChat(chat, currentProfile));
-    })
+    }),
   );
 };
 
@@ -242,13 +237,13 @@ export const getUserChatById = async (userId: User['id'], chatId: Chat['id']) =>
         throw new AppNotFoundError('Chat not found');
       }
       return prepareChat(chat, currentProfile);
-    })
+    }),
   );
 };
 
 export const getUserChatsByMemberProfileId = async (
   userId: User['id'],
-  profileId: Profile['id']
+  profileId: Profile['id'],
 ) => {
   return await Utils.handleDBKnownErrors(
     db.$transaction(async (tx) => {
@@ -281,14 +276,14 @@ export const getUserChatsByMemberProfileId = async (
       });
       if (!currentProfileWithChats) throw new AppNotFoundError('Profile not found');
       return currentProfileWithChats.chats.map((c) => prepareChat(c.chat, currentProfileWithChats));
-    })
+    }),
   );
 };
 
 export const getUserChatMessages = async (
   userId: User['id'],
   chatId: Chat['id'],
-  filters: Types.BasePaginationFilters = {}
+  filters: Types.BasePaginationFilters = {},
 ) => {
   return await Utils.handleDBKnownErrors(
     db.$transaction(async (tx) => {
@@ -315,14 +310,14 @@ export const getUserChatMessages = async (
         where: { chat: { id: chatId, profiles: { some: { profileId } } } },
         include: createMessageAggregation(),
       });
-    })
+    }),
   );
 };
 
 export const getUserChatMessageById = async (
   userId: User['id'],
   chatId: Chat['id'],
-  msgId: Message['id']
+  msgId: Message['id'],
 ) => {
   return await Utils.handleDBKnownErrors(
     db.$transaction(async (tx) => {
@@ -343,7 +338,7 @@ export const getUserChatMessageById = async (
       });
       if (!message) throw new AppNotFoundError('Message not found');
       return message;
-    })
+    }),
   );
 };
 
@@ -352,7 +347,7 @@ export const createUserChatMessage = async (
   chatId: Chat['id'],
   { body }: Schema.ValidMessage,
   imageData?: Types.ImageFullData,
-  uploadedImage?: Storage.UploadedImageData
+  uploadedImage?: Storage.UploadedImageData,
 ) => {
   return await Utils.handleDBKnownErrors(
     db.$transaction(async (tx) => {
@@ -393,7 +388,7 @@ export const createUserChatMessage = async (
         },
         include: createMessageAggregation(),
       });
-    })
+    }),
   );
 };
 
@@ -411,7 +406,7 @@ export const updateProfileChatLastSeenDate = async (userId: User['id'], chatId:
         where: { profileName_chatId: { profileName, chatId } },
         data: { lastSeenAt: now },
       });
-    })
+    }),
   );
   return now;
 };

src/tests/api/v1/chats.int.test.ts

@@ -799,7 +799,8 @@ describe('Chats endpoints', async () => {
       const oldChat = await createChat('', chatMembers);
       await createMessage(oldChat.id, dbUserOne);
       const chatData = {
-        profiles: chatMembers.map((u) => u.profile!.id),
+        // Remove current user from profile ids
+        profiles: chatMembers.filter((cm) => cm.id !== dbUserOne.id).map((u) => u.profile!.id),
         message: { body: 'Whats up?' },
       };
       const { authorizedApi } = await prepForAuthorizedTest(userOneData);
@@ -824,7 +825,8 @@ describe('Chats endpoints', async () => {
       const oldChat = await createChat('', chatMembers);
       await createMessage(oldChat.id, dbUserOne);
       const chatData = {
-        profiles: chatMembers.map((u) => u.profile!.id),
+        // Remove current user from profile ids
+        profiles: chatMembers.filter((cm) => cm.id !== intangibleUser.id).map((u) => u.profile!.id),
         message: { body: 'Whats up?' },
       };
       const { authorizedApi } = await prepForAuthorizedTest(intangibleUserData);
@@ -866,6 +868,23 @@ describe('Chats endpoints', async () => {
       expect(dbMsgs[0].chatId).toBe(dbChats[0].id);
     });
 
+    it('should not use an already exist self-chat, and start new chat with the given member profile id', async () => {
+      const dbSelfChat = await createChat('Hello, Me!', [dbUserOne], false);
+      const chatData = { profiles: [dbUserTwo.profile!.id], message: { body: 'Hello!' } };
+      const { authorizedApi } = await prepForAuthorizedTest(userOneData);
+      const res = await authorizedApi.post(CHATS_URL).send(chatData);
+      const dbMsgs = await db.message.findMany({});
+      const dbChats = await db.chat.findMany({});
+      const chat = res.body as ChatFullData;
+      expect(res.statusCode).toBe(201);
+      expect(res.type).toMatch(/json/);
+      expect(dbChats).toHaveLength(2);
+      expect(dbMsgs).toHaveLength(2);
+      assertChat(chat, dbChats.find((c) => c.id !== dbSelfChat.id)!.id, 1, 2);
+      expect(dbMsgs.find((m) => m.chatId === dbChats[0].id)).toBeTruthy();
+      expect(dbMsgs.find((m) => m.chatId === dbChats[1].id)).toBeTruthy();
+    });
+
     it('should create new chat with a non-image message, and ignore `imagedata` field without an image file', async () => {
       const profileId = dbUserTwo.profile!.id;
       const chatData = { profiles: [profileId], message: { body: 'Hello!' }, imagedata };