first commit

services/chat/app/js/Features/Threads/ThreadManager.js

@@ -0,0 +1,157 @@
+import { db, ObjectId } from '../../mongodb.js'
+
+export class MissingThreadError extends Error {}
+
+export const GLOBAL_THREAD = 'GLOBAL'
+
+export async function findOrCreateThread(projectId, threadId) {
+  let query, update
+  projectId = new ObjectId(projectId.toString())
+  if (threadId !== GLOBAL_THREAD) {
+    threadId = new ObjectId(threadId.toString())
+  }
+
+  if (threadId === GLOBAL_THREAD) {
+    query = {
+      project_id: projectId,
+      thread_id: { $exists: false },
+    }
+    update = {
+      project_id: projectId,
+    }
+  } else {
+    query = {
+      project_id: projectId,
+      thread_id: threadId,
+    }
+    update = {
+      project_id: projectId,
+      thread_id: threadId,
+    }
+  }
+
+  const result = await db.rooms.findOneAndUpdate(
+    query,
+    { $set: update },
+    { upsert: true, returnDocument: 'after' }
+  )
+  return result
+}
+
+export async function findAllThreadRooms(projectId) {
+  return await db.rooms
+    .find(
+      {
+        project_id: new ObjectId(projectId.toString()),
+        thread_id: { $exists: true },
+      },
+      {
+        thread_id: 1,
+        resolved: 1,
+      }
+    )
+    .toArray()
+}
+
+export async function findAllThreadRoomsAndGlobalThread(projectId) {
+  return await db.rooms
+    .find(
+      {
+        project_id: new ObjectId(projectId.toString()),
+      },
+      {
+        thread_id: 1,
+        resolved: 1,
+      }
+    )
+    .toArray()
+}
+
+export async function resolveThread(projectId, threadId, userId) {
+  await db.rooms.updateOne(
+    {
+      project_id: new ObjectId(projectId.toString()),
+      thread_id: new ObjectId(threadId.toString()),
+    },
+    {
+      $set: {
+        resolved: {
+          user_id: userId,
+          ts: new Date(),
+        },
+      },
+    }
+  )
+}
+
+export async function reopenThread(projectId, threadId) {
+  await db.rooms.updateOne(
+    {
+      project_id: new ObjectId(projectId.toString()),
+      thread_id: new ObjectId(threadId.toString()),
+    },
+    {
+      $unset: {
+        resolved: true,
+      },
+    }
+  )
+}
+
+export async function deleteThread(projectId, threadId) {
+  const room = await findOrCreateThread(projectId, threadId)
+  await db.rooms.deleteOne({
+    _id: room._id,
+  })
+  return room._id
+}
+
+export async function deleteAllThreadsInProject(projectId) {
+  await db.rooms.deleteMany({
+    project_id: new ObjectId(projectId.toString()),
+  })
+}
+
+export async function getResolvedThreadIds(projectId) {
+  const resolvedThreadIds = await db.rooms
+    .find(
+      {
+        project_id: new ObjectId(projectId),
+        thread_id: { $exists: true },
+        resolved: { $exists: true },
+      },
+      { projection: { thread_id: 1 } }
+    )
+    .map(record => record.thread_id.toString())
+    .toArray()
+  return resolvedThreadIds
+}
+
+export async function duplicateThread(projectId, threadId) {
+  const room = await db.rooms.findOne({
+    project_id: new ObjectId(projectId),
+    thread_id: new ObjectId(threadId),
+  })
+  if (!room) {
+    throw new MissingThreadError('Trying to duplicate a non-existent thread')
+  }
+  const newRoom = {
+    project_id: room.project_id,
+    thread_id: new ObjectId(),
+  }
+  if (room.resolved) {
+    newRoom.resolved = room.resolved
+  }
+  const confirmation = await db.rooms.insertOne(newRoom)
+  newRoom._id = confirmation.insertedId
+  return { oldRoom: room, newRoom }
+}
+
+export async function findThreadsById(projectId, threadIds) {
+  return await db.rooms
+    .find({
+      project_id: new ObjectId(projectId),
+      thread_id: { $in: threadIds.map(id => new ObjectId(id)) },
+    })
+    .toArray()
+}