Ryan/docmost
1
0
Fork 0
mirror of https://github.com/docmost/docmost.git synced 2025-11-15 13:01:13 +10:00
Code Issues Packages Projects Releases Wiki Activity

feat: websocket rooms (#515)

Browse Source
This commit is contained in:
Philip Okugbe
2024-11-28 18:53:29 +00:00
committed by GitHub
parent d97baf5824
commit a16d5d1bf4
5 changed files with 79 additions and 33 deletions
Download Patch File Download Diff File Expand all files Collapse all files

43
apps/server/src/ws/ws.gateway.ts
View File

@ -9,6 +9,7 @@ import { Server, Socket } from 'socket.io';
import { TokenService } from '../core/auth/services/token.service';
import { JwtType } from '../core/auth/dto/jwt-payload';
import { OnModuleDestroy } from '@nestjs/common';
import { SpaceMemberRepo } from '@docmost/db/repos/space/space-member.repo';
@WebSocketGateway({
cors: { origin: '*' },
@ -17,7 +18,10 @@ import { OnModuleDestroy } from '@nestjs/common';
export class WsGateway implements OnGatewayConnection, OnModuleDestroy {
@WebSocketServer()
server: Server;
constructor(private tokenService: TokenService) {}
constructor(
private tokenService: TokenService,
private spaceMemberRepo: SpaceMemberRepo,
) {}
async handleConnection(client: Socket, ...args: any[]): Promise<void> {
try {
@ -27,24 +31,43 @@ export class WsGateway implements OnGatewayConnection, OnModuleDestroy {
if (token.type !== JwtType.ACCESS) {
client.disconnect();
}
const userId = token.sub;
const workspaceId = token.workspaceId;
const userSpaceIds = await this.spaceMemberRepo.getUserSpaceIds(userId);
const workspaceRoom = `workspace-${workspaceId}`;
const spaceRooms = userSpaceIds.map((id) => this.getSpaceRoomName(id));
client.join([workspaceRoom, ...spaceRooms]);
} catch (err) {
client.disconnect();
}
}
@SubscribeMessage('message')
handleMessage(client: Socket, data: string): void {
client.broadcast.emit('message', data);
}
handleMessage(client: Socket, data: any): void {
const spaceEvents = [
'updateOne',
'addTreeNode',
'moveTreeNode',
'deleteTreeNode',
];
@SubscribeMessage('messageToRoom')
handleSendMessageToRoom(@MessageBody() message: any) {
this.server.to(message?.roomId).emit('messageToRoom', message);
if (spaceEvents.includes(data?.operation) && data?.spaceId) {
const room = this.getSpaceRoomName(data.spaceId);
client.broadcast.to(room).emit('message', data);
return;
}
client.broadcast.emit('message', data);
}
@SubscribeMessage('join-room')
handleJoinRoom(client: Socket, @MessageBody() roomName: string): void {
client.join(roomName);
// if room is a space, check if user has permissions
//client.join(roomName);
}
@SubscribeMessage('leave-room')
@ -57,4 +80,8 @@ export class WsGateway implements OnGatewayConnection, OnModuleDestroy {
this.server.close();
}
}
getSpaceRoomName(spaceId: string): string {
return `space-${spaceId}`;
}
}