Merge pull request #10901 from overleaf/lg-openapi-chat

Migrate chat service to OpenAPI

GitOrigin-RevId: dc49c52e5e23aa6650e9a3fbe57ae6ac9be1c1da
This commit is contained in:
Lucie Germain
2022-12-22 09:04:40 +00:00
committed by Copybot
parent b7d8fa44b4
commit d07a7ce4a2
11 changed files with 802 additions and 150 deletions
@@ -3,24 +3,105 @@ import * as MessageManager from './MessageManager.js'
import * as MessageFormatter from './MessageFormatter.js'
import * as ThreadManager from '../Threads/ThreadManager.js'
import { ObjectId } from '../../mongodb.js'
import { expressify } from '../../util/promises.js'
const DEFAULT_MESSAGE_LIMIT = 50
const MAX_MESSAGE_LENGTH = 10 * 1024 // 10kb, about 1,500 words
export const getGlobalMessages = expressify(async (req, res) => {
async function readContext(context, req) {
req.body = context.requestBody
req.params = context.params.path
req.query = context.params.query
if (typeof req.params.projectId !== 'undefined') {
if (!ObjectId.isValid(req.params.projectId)) {
context.res.status(400).setBody('Invalid projectId')
}
}
if (typeof req.params.threadId !== 'undefined') {
if (!ObjectId.isValid(req.params.threadId)) {
context.res.status(400).setBody('Invalid threadId')
}
}
}
export async function callMessageHttpController(context, ControllerMethod) {
const req = {}
readContext(context, req)
if (context.res.statusCode !== 400) {
return await ControllerMethod(req, context.res)
} else {
return context.res.body
}
}
export async function getGlobalMessages(context) {
return await callMessageHttpController(context, _getGlobalMessages)
}
export async function sendGlobalMessage(context) {
return await callMessageHttpController(context, _sendGlobalMessage)
}
export async function sendMessage(context) {
return await callMessageHttpController(context, _sendThreadMessage)
}
export async function getThreads(context) {
return await callMessageHttpController(context, _getAllThreads)
}
export async function resolveThread(context) {
return await callMessageHttpController(context, _resolveThread)
}
export async function reopenThread(context) {
return await callMessageHttpController(context, _reopenThread)
}
export async function deleteThread(context) {
return await callMessageHttpController(context, _deleteThread)
}
export async function editMessage(context) {
return await callMessageHttpController(context, _editMessage)
}
export async function deleteMessage(context) {
return await callMessageHttpController(context, _deleteMessage)
}
export async function destroyProject(context) {
return await callMessageHttpController(context, _destroyProject)
}
export async function getStatus(context) {
const message = 'chat is alive'
context.res.status(200).setBody(message)
return message
}
const _getGlobalMessages = async (req, res) => {
await _getMessages(ThreadManager.GLOBAL_THREAD, req, res)
})
}
export const sendGlobalMessage = expressify(async (req, res) => {
await _sendMessage(ThreadManager.GLOBAL_THREAD, req, res)
})
async function _sendGlobalMessage(req, res) {
const { user_id: userId, content } = req.body
const { projectId } = req.params
return await _sendMessage(
userId,
projectId,
content,
ThreadManager.GLOBAL_THREAD,
res
)
}
export const sendThreadMessage = expressify(async (req, res) => {
await _sendMessage(req.params.threadId, req, res)
})
async function _sendThreadMessage(req, res) {
const { user_id: userId, content } = req.body
const { projectId, threadId } = req.params
return await _sendMessage(userId, projectId, content, threadId, res)
}
export const getAllThreads = expressify(async (req, res) => {
const _getAllThreads = async (req, res) => {
const { projectId } = req.params
logger.debug({ projectId }, 'getting all threads')
const rooms = await ThreadManager.findAllThreadRooms(projectId)
@@ -28,32 +109,32 @@ export const getAllThreads = expressify(async (req, res) => {
const messages = await MessageManager.findAllMessagesInRooms(roomIds)
const threads = MessageFormatter.groupMessagesByThreads(rooms, messages)
res.json(threads)
})
}
export const resolveThread = expressify(async (req, res) => {
const _resolveThread = async (req, res) => {
const { projectId, threadId } = req.params
const { user_id: userId } = req.body
logger.debug({ userId, projectId, threadId }, 'marking thread as resolved')
await ThreadManager.resolveThread(projectId, threadId, userId)
res.sendStatus(204)
})
res.status(204)
}
export const reopenThread = expressify(async (req, res) => {
const _reopenThread = async (req, res) => {
const { projectId, threadId } = req.params
logger.debug({ projectId, threadId }, 'reopening thread')
await ThreadManager.reopenThread(projectId, threadId)
res.sendStatus(204)
})
res.status(204)
}
export const deleteThread = expressify(async (req, res) => {
const _deleteThread = async (req, res) => {
const { projectId, threadId } = req.params
logger.debug({ projectId, threadId }, 'deleting thread')
const roomId = await ThreadManager.deleteThread(projectId, threadId)
await MessageManager.deleteAllMessagesInRoom(roomId)
res.sendStatus(204)
})
res.status(204)
}
export const editMessage = expressify(async (req, res) => {
const _editMessage = async (req, res) => {
const { content, userId } = req.body
const { projectId, threadId, messageId } = req.params
logger.debug({ projectId, threadId, messageId, content }, 'editing message')
@@ -66,20 +147,21 @@ export const editMessage = expressify(async (req, res) => {
Date.now()
)
if (!found) {
return res.sendStatus(404)
res.status(404)
return
}
res.sendStatus(204)
})
res.status(204)
}
export const deleteMessage = expressify(async (req, res) => {
const _deleteMessage = async (req, res) => {
const { projectId, threadId, messageId } = req.params
logger.debug({ projectId, threadId, messageId }, 'deleting message')
const room = await ThreadManager.findOrCreateThread(projectId, threadId)
await MessageManager.deleteMessage(room._id, messageId)
res.sendStatus(204)
})
res.status(204)
}
export const destroyProject = expressify(async (req, res) => {
const _destroyProject = async (req, res) => {
const { projectId } = req.params
logger.debug({ projectId }, 'destroying project')
const rooms = await ThreadManager.findAllThreadRoomsAndGlobalThread(projectId)
@@ -88,22 +170,24 @@ export const destroyProject = expressify(async (req, res) => {
await MessageManager.deleteAllMessagesInRooms(roomIds)
logger.debug({ projectId }, 'deleting all threads in project')
await ThreadManager.deleteAllThreadsInProject(projectId)
res.sendStatus(204)
})
res.status(204)
}
async function _sendMessage(clientThreadId, req, res) {
const { user_id: userId, content } = req.body
const { projectId } = req.params
async function _sendMessage(userId, projectId, content, clientThreadId, res) {
if (!ObjectId.isValid(userId)) {
return res.status(400).send('Invalid userId')
const message = 'Invalid userId'
res.status(400).setBody(message)
return message
}
if (!content) {
return res.status(400).send('No content provided')
const message = 'No content provided'
res.status(400).setBody(message)
return message
}
if (content.length > MAX_MESSAGE_LENGTH) {
return res
.status(400)
.send(`Content too long (> ${MAX_MESSAGE_LENGTH} bytes)`)
const message = `Content too long (> ${MAX_MESSAGE_LENGTH} bytes)`
res.status(400).setBody(message)
return message
}
logger.debug(
{ clientThreadId, projectId, userId, content },
@@ -121,7 +205,7 @@ async function _sendMessage(clientThreadId, req, res) {
)
message = MessageFormatter.formatMessageForClientSide(message)
message.room_id = projectId
res.status(201).send(message)
res.status(201).setBody(message)
}
async function _getMessages(clientThreadId, req, res) {
@@ -153,5 +237,5 @@ async function _getMessages(clientThreadId, req, res) {
let messages = await MessageManager.getMessages(threadObjectId, limit, before)
messages = MessageFormatter.formatMessagesForClientSide(messages)
logger.debug({ projectId, messages }, 'got messages')
res.status(200).send(messages)
res.status(200).setBody(messages)
}