4c4b80ba9b
* fix: Collaboration debounce shared between docs * Rename, Tracing -> Metrics * Add tracing * tsc * fix: Lock document row when loading document in collaboration service incase state needs writing * fix: Incorrect service name regression
44 lines
1.3 KiB
TypeScript
44 lines
1.3 KiB
TypeScript
import http from "http";
|
|
import url from "url";
|
|
import { Server } from "@hocuspocus/server";
|
|
import invariant from "invariant";
|
|
import Koa from "koa";
|
|
import WebSocket from "ws";
|
|
import AuthenticationExtension from "../collaboration/AuthenticationExtension";
|
|
import LoggerExtension from "../collaboration/LoggerExtension";
|
|
import MetricsExtension from "../collaboration/MetricsExtension";
|
|
import PersistenceExtension from "../collaboration/PersistenceExtension";
|
|
|
|
export default function init(app: Koa, server: http.Server) {
|
|
const path = "/collaboration";
|
|
const wss = new WebSocket.Server({
|
|
noServer: true,
|
|
});
|
|
|
|
const hocuspocus = Server.configure({
|
|
debounce: 3000,
|
|
maxDebounce: 10000,
|
|
extensions: [
|
|
new AuthenticationExtension(),
|
|
new PersistenceExtension(),
|
|
new LoggerExtension(),
|
|
new MetricsExtension(),
|
|
],
|
|
});
|
|
|
|
server.on("upgrade", function (req, socket, head) {
|
|
if (req.url && req.url.indexOf(path) > -1) {
|
|
const documentName = url.parse(req.url).pathname?.split("/").pop();
|
|
invariant(documentName, "Document name must be provided");
|
|
|
|
wss.handleUpgrade(req, socket, head, (client) => {
|
|
hocuspocus.handleConnection(client, req, documentName);
|
|
});
|
|
}
|
|
});
|
|
|
|
server.on("shutdown", () => {
|
|
return hocuspocus.destroy();
|
|
});
|
|
}
|