From 85e5d9257e959bc6170e8ed62edf0ea0db5b4a7d Mon Sep 17 00:00:00 2001 From: Bastien OLLIER Date: Mon, 3 Jun 2024 15:03:31 +0200 Subject: [PATCH] Implement private user rooms (#4) MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Co-authored-by: bastien ollier Reviewed-on: https://codefirst.iut.uca.fr/git/sandkasten/labyrinth/pulls/4 Reviewed-by: Clément FRÉVILLE Co-authored-by: Bastien OLLIER Co-committed-by: Bastien OLLIER --- package.json | 1 + src/database.ts | 33 ++++++----- src/runner.ts | 4 +- src/server.ts | 148 ++++++++++++++++++++++++++++++++---------------- 4 files changed, 119 insertions(+), 67 deletions(-) diff --git a/package.json b/package.json index 44ed1b8..d2e192b 100644 --- a/package.json +++ b/package.json @@ -18,6 +18,7 @@ "@fastify/type-provider-typebox": "^4.0.0", "@fastify/websocket": "^10.0.1", "@sinclair/typebox": "^0.32.9", + "dprint": "^0.46.1", "fastify": "^4.27.0", "nanoid": "^5.0.4", "sqlite3": "^5.1.7", diff --git a/src/database.ts b/src/database.ts index c8b3d48..3aa61be 100644 --- a/src/database.ts +++ b/src/database.ts @@ -10,7 +10,7 @@ const dbFilePath = `${dbDirectory}/database.db`; export function runDB( db: sqlite3.Database, query: string, - params: any[] + params: any[], ): Promise { return new Promise((resolve, reject) => { db.run(query, params, (err) => { @@ -26,7 +26,7 @@ export function runDB( /* Fonction pour récupérer plusieurs lignes de la base de données */ export function allDB( db: sqlite3.Database, - query: string + query: string, ): Promise { return new Promise((resolve, reject) => { db.all(query, (err, rows) => { @@ -43,7 +43,7 @@ export function allDB( export function getDB( db: sqlite3.Database, query: string, - params: any[] + params: any[], ): Promise { return new Promise((resolve, reject) => { db.get(query, params, (err, row: any) => { @@ -73,7 +73,7 @@ export function openDatabase() { sqlite3.OPEN_READWRITE | sqlite3.OPEN_CREATE, (err: Error | null) => { if (err) console.error(err.message); - } + }, ); } @@ -105,7 +105,8 @@ export function createTables(db: sqlite3.Database) { /* Créer la table registered_user dans la base de données */ export function createRegisteredUserTable(db: sqlite3.Database): Promise { - const tableRegisteredUser = `CREATE TABLE IF NOT EXISTS registered_user (id_user INTEGER PRIMARY KEY AUTOINCREMENT, login TEXT NOT NULL, password TEXT NOT NULL, permissions INTEGER NOT NULL, UNIQUE (login))`; + const tableRegisteredUser = + `CREATE TABLE IF NOT EXISTS registered_user (id_user INTEGER PRIMARY KEY AUTOINCREMENT, login TEXT NOT NULL, password TEXT NOT NULL, permissions INTEGER NOT NULL, UNIQUE (login))`; return runDB(db, tableRegisteredUser, []); } @@ -115,7 +116,7 @@ export function insertUser( db: sqlite3.Database, login: string, password: string, - permissions: number + permissions: number, ) { const insertUserQuery = `INSERT INTO registered_user (login, password, permissions) VALUES (?, ?, ?)`; @@ -126,7 +127,7 @@ export function insertUser( export function updateUserLogin( db: sqlite3.Database, id: number, - newLogin: string + newLogin: string, ) { const updateUserLoginQuery = `UPDATE registered_user SET login = ? WHERE id_user = ?`; @@ -137,7 +138,7 @@ export function updateUserLogin( export function updateUserPassword( db: sqlite3.Database, id: number, - newPassword: string + newPassword: string, ) { const updateUserPasswordQuery = `UPDATE registered_user SET password = ? WHERE id_user = ?`; @@ -148,7 +149,7 @@ export function updateUserPassword( export function updateUserPermissions( db: sqlite3.Database, id: number, - newPermissions: number + newPermissions: number, ) { const updateUserPermissionsQuery = `UPDATE registered_user SET permissions = ? WHERE id_user = ?`; @@ -206,7 +207,8 @@ export function selectUserById(db: sqlite3.Database, id: number) { /* Créer la table language dans la base de données */ export function createLanguageTable(db: sqlite3.Database): Promise { - const tableLanguage = `CREATE TABLE IF NOT EXISTS language (id_language INTEGER PRIMARY KEY AUTOINCREMENT, designation TEXT NOT NULL, version INTEGER NOT NULL)`; + const tableLanguage = + `CREATE TABLE IF NOT EXISTS language (id_language INTEGER PRIMARY KEY AUTOINCREMENT, designation TEXT NOT NULL, version INTEGER NOT NULL)`; return runDB(db, tableLanguage, []); } @@ -214,7 +216,7 @@ export function createLanguageTable(db: sqlite3.Database): Promise { export function insertLanguage( db: sqlite3.Database, designation: string, - version: number + version: number, ) { const insertLanguageQuery = `INSERT INTO language (designation, version) VALUES (?, ?)`; @@ -225,7 +227,7 @@ export function insertLanguage( export function updateLanguageDesignation( db: sqlite3.Database, id: number, - newDesignation: string + newDesignation: string, ) { const updateLanguageDesignationQuery = `UPDATE language SET designation = ? WHERE id_language = ?`; @@ -236,7 +238,7 @@ export function updateLanguageDesignation( export function updateLanguageVersion( db: sqlite3.Database, id: number, - newVersion: number + newVersion: number, ) { const updateLanguageVersionQuery = `UPDATE language SET version = ? WHERE id_language = ?`; @@ -282,7 +284,8 @@ export function selectLanguageById(db: sqlite3.Database, id: number) { /* Créer la table work dans la base de données */ export function createWorkTable(db: sqlite3.Database): Promise { - const tableWork = `CREATE TABLE IF NOT EXISTS work (id_work INTEGER PRIMARY KEY AUTOINCREMENT, link CHAR(36) NOT NULL, user_id INTEGER REFERENCES registered_user(id_user), language_id INTEGER NOT NULL REFERENCES language(id_language), content TEXT NOT NULL)`; + const tableWork = + `CREATE TABLE IF NOT EXISTS work (id_work INTEGER PRIMARY KEY AUTOINCREMENT, link CHAR(36) NOT NULL, user_id INTEGER REFERENCES registered_user(id_user), language_id INTEGER NOT NULL REFERENCES language(id_language), content TEXT NOT NULL)`; return runDB(db, tableWork, []); } @@ -292,7 +295,7 @@ export function insertWork( link: string, user_id: number, language_id: number, - content: string + content: string, ) { const insertWorkQuery = `INSERT INTO work (link, user_id, language_id, content) VALUES (?, ?, ?, ?)`; diff --git a/src/runner.ts b/src/runner.ts index 290e8bb..1a8567b 100644 --- a/src/runner.ts +++ b/src/runner.ts @@ -19,11 +19,11 @@ export const IMAGES = { export function allocateBuffer( jobId: string, code: string, - image: string + image: string, ): Buffer { let cur = 0; const buffer = Buffer.allocUnsafe( - jobId.length + image.length + code.length + 9 + jobId.length + image.length + code.length + 9, ); cur = buffer.writeUInt8(0, cur); cur += buffer.write(jobId, cur); diff --git a/src/server.ts b/src/server.ts index 8690712..51d8de6 100644 --- a/src/server.ts +++ b/src/server.ts @@ -1,12 +1,12 @@ +import { rebaseUpdates, Update } from "@codemirror/collab"; +import { ChangeSet, Text } from "@codemirror/state"; import cors from "@fastify/cors"; -import websocket, { WebSocket } from '@fastify/websocket'; import { Type, TypeBoxTypeProvider } from "@fastify/type-provider-typebox"; +import websocket, { WebSocket } from "@fastify/websocket"; import Fastify, { FastifyReply } from "fastify"; import { nanoid } from "nanoid"; import { allocateBuffer, getRunner } from "runner"; import { Pull, Push } from "zeromq"; -import { ChangeSet, Text } from "@codemirror/state"; -import { Update, rebaseUpdates } from "@codemirror/collab"; import * as db from "./database"; const sender = new Push(); @@ -14,17 +14,25 @@ await sender.bind(`tcp://127.0.0.1:5557`); const receiver = new Pull(); await receiver.bind(`tcp://127.0.0.1:5558`); + const clients: Record = {}; const generateId = () => nanoid(32); -let updates: Update[] = []; -let doc = Text.of(['']); -const liveClients: WebSocket[] = []; +//let updates: Update[] = []; +//let doc = Text.of(["foo"]); + +type Room = { + sockets: WebSocket[]; + updates: Update[]; + doc: Text; +} + +const rooms: Record = {}; function send(socket: WebSocket, requestId: number, payload: unknown) { const response = { _request: requestId, - payload + payload, }; socket.send(JSON.stringify(response)); } @@ -32,39 +40,79 @@ function send(socket: WebSocket, requestId: number, payload: unknown) { const fastify = Fastify({ logger: true, }).withTypeProvider(); +type Fastify = typeof fastify; await fastify.register(cors, { - origin: process.env.ALLOW_ORIGIN || '*', + origin: process.env.ALLOW_ORIGIN || "*", }); fastify.register(websocket); -fastify.get("/live", { websocket: true }, (socket, req) => { - liveClients.push(socket); - socket.on("message", message => { - const data = JSON.parse(message.toString()); - const requestId = data._request; - if (data.type === "pullUpdates") { - send(socket, requestId, updates.slice(data.version)) - } else if (data.type === "pushUpdates") { - let received = data.updates.map((json: any) => ({ - clientID: json.clientID, - changes: ChangeSet.fromJSON(json.changes) - })) - if (data.version != updates.length) { - received = rebaseUpdates(received, updates.slice(data.version)) - } - for (let update of received) { - updates.push(update) - doc = update.changes.apply(doc) + + +fastify.register(async function (fastify: Fastify) { + fastify.get( + "/live/:roomId", + { + schema: { + params: Type.Object({ + roomId: Type.String(), + }) + }, + websocket: true + }, + (socket, request) => { + const { roomId } = request.params; + let room = rooms[roomId]; + if(!room){ + room = { + sockets: [], + updates: [], + doc: Text.of(['']) + }; + rooms[roomId] = room; } - send(socket, requestId, received.map((update: any) => ({ - clientID: update.clientID, - changes: update.changes.toJSON() - }))); - } else if (data.type == "getDocument") { - send(socket, requestId, {version: updates.length, doc: doc.toString()}) - } - }) + room.sockets.push(socket); + socket.on("message", message => { + const data = JSON.parse(message.toString()); + const requestId = data._request; + + if (data.type === "pullUpdates") { + send(socket, requestId, room.updates.slice(data.version)); + } else if (data.type === "pushUpdates") { + let received = data.updates.map((json: any) => ({ + clientID: json.clientID, + changes: ChangeSet.fromJSON(json.changes), + })); + + if (data.version != room.updates.length) { + received = rebaseUpdates(received, room.updates.slice(data.version)); + } + + for (let update of received) { + room.updates.push(update); + room.doc = update.changes.apply(room.doc); + } + send( + socket, + requestId, + received.map((update: any) => ({ + clientID: update.clientID, + changes: update.changes.toJSON(), + })), + ); + } else if (data.type === "getDocument") { + send(socket, requestId, { version: room.updates.length, doc: room.doc.toString() }); + } + }); + }); }) +/* Route pour créer une room */ +fastify.post( + "/live", + async (request, reply) => { + return generateId(); + }, +); + fastify.post("/run", { schema: { body: Type.Object({ @@ -122,7 +170,7 @@ fastify.post( const { login, password, permissions } = request.body; db.insertUser(database, login, password, permissions); reply.send({ success: true }); - } + }, ); /* Route pour mettre à jour le login d'un utilisateur */ @@ -145,7 +193,7 @@ fastify.put( const { newLogin } = request.body; db.updateUserLogin(database, id, newLogin); reply.send({ success: true }); - } + }, ); /* Route pour mettre à jour le mot de passe d'un utilisateur */ @@ -168,7 +216,7 @@ fastify.put( const { newPassword } = request.body; db.updateUserPassword(database, id, newPassword); reply.send({ success: true }); - } + }, ); /* Route pour mettre à jour les permissions d'un utilisateur */ @@ -191,7 +239,7 @@ fastify.put( const { newPermissions } = request.body; await db.updateUserPermissions(database, id, newPermissions); reply.send({ success: true }); - } + }, ); /* Route pour supprimer un utilisateur par son ID */ @@ -210,7 +258,7 @@ fastify.delete( const { id } = request.params; await db.deleteUserById(database, id); reply.send({ success: true }); - } + }, ); /* Route pour supprimer un utilisateur par son login */ @@ -227,7 +275,7 @@ fastify.delete( const { login } = request.params; await db.deleteUserByLogin(database, login); reply.send({ success: true }); - } + }, ); /* Route pour supprimer tous les utilisateurs */ @@ -258,7 +306,7 @@ fastify.get( const { id } = request.params; const user = await db.selectUserById(database, id); reply.send(user); - } + }, ); /* Route pour récupérer un utilisateur par son login */ @@ -275,7 +323,7 @@ fastify.get( const { login } = request.params; const user = await db.selectUserByLogin(database, login); reply.send(user); - } + }, ); /* Route pour créer un language */ @@ -293,7 +341,7 @@ fastify.post( const { designation, version } = request.body; db.insertLanguage(database, designation, version); reply.send({ success: true }); - } + }, ); /* Route pour mettre à jour la désignation d'un language */ @@ -316,7 +364,7 @@ fastify.put( const { newDesignation } = request.body; db.updateLanguageDesignation(database, id, newDesignation); reply.send({ success: true }); - } + }, ); /* Route pour mettre à jour la version d'un language */ @@ -339,7 +387,7 @@ fastify.put( const { newVersion } = request.body; db.updateLanguageVersion(database, id, newVersion); reply.send({ success: true }); - } + }, ); /* Route pour supprimer un language */ @@ -358,7 +406,7 @@ fastify.delete( const { id } = request.params; db.deleteLanguage(database, id); reply.send({ success: true }); - } + }, ); /* Route pour supprimer tous les languages */ @@ -383,7 +431,7 @@ fastify.get( const { id } = request.params; const language = await db.selectLanguageById(database, id); reply.send(language); - } + }, ); /* Route pour récupérer tous les languages */ @@ -409,7 +457,7 @@ fastify.post( const { id_user, link, id_language, code } = request.body; db.insertWork(database, link, id_user, id_language, code); reply.send({ success: true }); - } + }, ); /* Route pour récupérer tous les works */ @@ -440,7 +488,7 @@ fastify.delete( const { id } = request.params; db.deleteWork(database, id); reply.send({ success: true }); - } + }, ); /* Route pour récupérer un work par son ID */ @@ -459,7 +507,7 @@ fastify.get( const { id } = request.params; const work = await db.selectWorkById(database, id); reply.send(work); - } + }, ); /* Forward output est une fonction asynchrone qui permet de récupérer les messages envoyés par le container et de les renvoyer au client */