mirror of https://github.com/nocodb/nocodb
mertmit
2 years ago
5 changed files with 2 additions and 88 deletions
@ -1,19 +0,0 @@
|
||||
import { Test } from '@nestjs/testing'; |
||||
import { SocketGateway } from './socket.gateway'; |
||||
import type { TestingModule } from '@nestjs/testing'; |
||||
|
||||
describe('SocketGateway', () => { |
||||
let gateway: SocketGateway; |
||||
|
||||
beforeEach(async () => { |
||||
const module: TestingModule = await Test.createTestingModule({ |
||||
providers: [SocketGateway], |
||||
}).compile(); |
||||
|
||||
gateway = module.get<SocketGateway>(SocketGateway); |
||||
}); |
||||
|
||||
it('should be defined', () => { |
||||
expect(gateway).toBeDefined(); |
||||
}); |
||||
}); |
@ -1,67 +0,0 @@
|
||||
import crypto from 'crypto'; |
||||
import { WebSocketGateway, WebSocketServer } from '@nestjs/websockets'; |
||||
import { Inject, Injectable } from '@nestjs/common'; |
||||
import { HttpAdapterHost } from '@nestjs/core'; |
||||
import { T } from 'nc-help'; |
||||
import { Server } from 'socket.io'; |
||||
import { AuthGuard } from '@nestjs/passport'; |
||||
import { ExecutionContextHost } from '@nestjs/core/helpers/execution-context-host'; |
||||
import { JwtStrategy } from '../strategies/jwt.strategy'; |
||||
import type { OnModuleInit } from '@nestjs/common'; |
||||
import type { Socket } from 'socket.io'; |
||||
|
||||
function getHash(str) { |
||||
return crypto.createHash('md5').update(str).digest('hex'); |
||||
} |
||||
|
||||
@WebSocketGateway({ |
||||
cors: { |
||||
origin: '*', |
||||
allowedHeaders: ['xc-auth'], |
||||
credentials: true, |
||||
}, |
||||
}) |
||||
@Injectable() |
||||
export class SocketGateway implements OnModuleInit { |
||||
// private server: HttpServer;
|
||||
private clients: { [id: string]: Socket } = {}; |
||||
|
||||
constructor( |
||||
private jwtStrategy: JwtStrategy, |
||||
@Inject(HttpAdapterHost) private httpAdapterHost: HttpAdapterHost, |
||||
) {} |
||||
|
||||
@WebSocketServer() |
||||
server: Server; |
||||
|
||||
async onModuleInit() { |
||||
this.server |
||||
.use(async (socket, next) => { |
||||
try { |
||||
const context = new ExecutionContextHost([socket.handshake as any]); |
||||
const guard = new (AuthGuard('jwt'))(context); |
||||
await guard.canActivate(context); |
||||
} catch {} |
||||
|
||||
next(); |
||||
}) |
||||
.on('connection', (socket) => { |
||||
this.clients[socket.id] = socket; |
||||
const id = getHash( |
||||
(process.env.NC_SERVER_UUID || T.id) + |
||||
(socket?.handshake as any)?.user?.id, |
||||
); |
||||
|
||||
socket.on('page', (args) => { |
||||
T.page({ ...args, id }); |
||||
}); |
||||
socket.on('event', (args) => { |
||||
T.event({ ...args, id }); |
||||
}); |
||||
}); |
||||
} |
||||
|
||||
public get io() { |
||||
return this.server; |
||||
} |
||||
} |
Loading…
Reference in new issue