overleaf/services/real-time/app/js/HttpApiController.js

/* eslint-disable
    camelcase,
    no-unused-vars,
*/
// TODO: This file was created by bulk-decaffeinate.
// Fix any style issues and re-enable lint.
/*
 * decaffeinate suggestions:
 * DS101: Remove unnecessary use of Array.from
 * DS102: Remove unnecessary code created because of implicit returns
 * Full docs: https://github.com/decaffeinate/decaffeinate/blob/master/docs/suggestions.md
 */
let HttpApiController;
const WebsocketLoadBalancer = require("./WebsocketLoadBalancer");
const DrainManager = require("./DrainManager");
const logger = require("logger-sharelatex");

module.exports = (HttpApiController = {
	sendMessage(req, res, next) {
		logger.log({message: req.params.message}, "sending message");
		if (Array.isArray(req.body)) {
			for (const payload of Array.from(req.body)) {
				WebsocketLoadBalancer.emitToRoom(req.params.project_id, req.params.message, payload);
			}
		} else {
			WebsocketLoadBalancer.emitToRoom(req.params.project_id, req.params.message, req.body);
		}
		return res.send(204);
	}, // No content
	
	startDrain(req, res, next) {
		const io = req.app.get("io");
		let rate = req.query.rate || "4";
		rate = parseFloat(rate) || 0;
		logger.log({rate}, "setting client drain rate");
		DrainManager.startDrain(io, rate);
		return res.send(204);
	},

	disconnectClient(req, res, next) {
		const io = req.app.get("io");
		const {
            client_id
        } = req.params;
		const client = io.sockets.sockets[client_id];

		if (!client) {
			logger.info({client_id}, "api: client already disconnected");
			res.sendStatus(404);
			return;
		}
		logger.warn({client_id}, "api: requesting client disconnect");
		client.on("disconnect", () => res.sendStatus(204));
		return client.disconnect();
	}
});