mirror of https://github.com/nocodb/nocodb
Pranav C
1 year ago
15 changed files with 194 additions and 307 deletions
@ -1,83 +0,0 @@
|
||||
import crypto from 'crypto'; |
||||
import { Inject, Injectable, UnauthorizedException } from '@nestjs/common'; |
||||
import { HttpAdapterHost } from '@nestjs/core'; |
||||
import { T } from 'nc-help'; |
||||
import { Server } from 'socket.io'; |
||||
import { AuthGuard } from '@nestjs/passport'; |
||||
import { JwtStrategy } from '../../strategies/jwt.strategy'; |
||||
import type { OnModuleInit } from '@nestjs/common'; |
||||
import type { Socket } from 'socket.io'; |
||||
import { ExecutionContextHost } from '@nestjs/core/helpers/execution-context-host'; |
||||
|
||||
function getHash(str) { |
||||
return crypto.createHash('md5').update(str).digest('hex'); |
||||
} |
||||
|
||||
@Injectable() |
||||
export class ClientService implements OnModuleInit { |
||||
// private server: HttpServer;
|
||||
private clients: { [id: string]: Socket } = {}; |
||||
private jobs: { [id: string]: { last_message: any } } = {}; |
||||
|
||||
constructor( |
||||
private jwtStrategy: JwtStrategy, |
||||
@Inject(HttpAdapterHost) private httpAdapterHost: HttpAdapterHost, |
||||
) { |
||||
} |
||||
|
||||
async onModuleInit() { |
||||
const io = new Server(this.httpAdapterHost.httpAdapter.getHttpServer(), { |
||||
cors: { |
||||
origin: '*', |
||||
allowedHeaders: ['xc-auth'], |
||||
credentials: true, |
||||
}, |
||||
}); |
||||
io.use(async (socket, next) => { |
||||
// const authGuard = new (AuthGuard('jwt'))();
|
||||
// const result = await authGuard.canActivate(socket.handshake as any);
|
||||
// if (!result) {
|
||||
// throw new UnauthorizedException();
|
||||
// }
|
||||
// return new Promise((resolve, reject) => {
|
||||
// this.jwtStrategy.authenticate(
|
||||
// socket.handshake as any,
|
||||
// (error, user) => {
|
||||
// if (error) {
|
||||
// reject(new UnauthorizedException(error.message));
|
||||
// } else {
|
||||
// resolve(user);
|
||||
// }
|
||||
// },
|
||||
// );
|
||||
// });
|
||||
try { |
||||
const context = new ExecutionContextHost([socket.handshake as any]); |
||||
const guard = new (AuthGuard('jwt'))(context); |
||||
const canActivate = 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 }); |
||||
}); |
||||
socket.on('subscribe', (room) => { |
||||
if (room in this.jobs) { |
||||
socket.join(room); |
||||
socket.emit('job'); |
||||
socket.emit('progress', this.jobs[room].last_message); |
||||
} |
||||
}); |
||||
}); |
||||
} |
||||
} |
||||
@ -1,15 +1,15 @@
|
||||
import { Test, TestingModule } from '@nestjs/testing'; |
||||
import { ClientService } from './client.service'; |
||||
import { SocketService } from './socket.service'; |
||||
|
||||
describe('ClientService', () => { |
||||
let service: ClientService; |
||||
let service: SocketService; |
||||
|
||||
beforeEach(async () => { |
||||
const module: TestingModule = await Test.createTestingModule({ |
||||
providers: [ClientService], |
||||
providers: [SocketService], |
||||
}).compile(); |
||||
|
||||
service = module.get<ClientService>(ClientService); |
||||
service = module.get<SocketService>(SocketService); |
||||
}); |
||||
|
||||
it('should be defined', () => { |
||||
@ -0,0 +1,81 @@
|
||||
import crypto from 'crypto'; |
||||
import { Inject, Injectable, UnauthorizedException } 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 Noco from '../../Noco'; |
||||
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'); |
||||
} |
||||
|
||||
@Injectable() |
||||
export class SocketService implements OnModuleInit { |
||||
// private server: HttpServer;
|
||||
private clients: { [id: string]: Socket } = {}; |
||||
private _jobs: { [id: string]: { last_message: any } } = {}; |
||||
private _io: Server; |
||||
|
||||
constructor( |
||||
private jwtStrategy: JwtStrategy, |
||||
@Inject(HttpAdapterHost) private httpAdapterHost: HttpAdapterHost, |
||||
) {} |
||||
|
||||
async onModuleInit() { |
||||
console.log(Noco.httpServer) |
||||
this._io = new Server( |
||||
Noco.httpServer ?? this.httpAdapterHost.httpAdapter.getHttpServer(), |
||||
{ |
||||
cors: { |
||||
origin: '*', |
||||
allowedHeaders: ['xc-auth'], |
||||
credentials: true, |
||||
}, |
||||
}, |
||||
); |
||||
this.io |
||||
.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 }); |
||||
}); |
||||
socket.on('subscribe', (room) => { |
||||
if (room in this.jobs) { |
||||
socket.join(room); |
||||
socket.emit('job'); |
||||
socket.emit('progress', this.jobs[room].last_message); |
||||
} |
||||
}); |
||||
}); |
||||
} |
||||
|
||||
public get io() { |
||||
return this._io; |
||||
} |
||||
|
||||
public get jobs() { |
||||
return this._jobs; |
||||
} |
||||
} |
||||
Loading…
Reference in new issue