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

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()
}