mirror of https://github.com/nocodb/nocodb
Pranav C
2 years 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 { Test, TestingModule } from '@nestjs/testing'; |
||||||
import { ClientService } from './client.service'; |
import { SocketService } from './socket.service'; |
||||||
|
|
||||||
describe('ClientService', () => { |
describe('ClientService', () => { |
||||||
let service: ClientService; |
let service: SocketService; |
||||||
|
|
||||||
beforeEach(async () => { |
beforeEach(async () => { |
||||||
const module: TestingModule = await Test.createTestingModule({ |
const module: TestingModule = await Test.createTestingModule({ |
||||||
providers: [ClientService], |
providers: [SocketService], |
||||||
}).compile(); |
}).compile(); |
||||||
|
|
||||||
service = module.get<ClientService>(ClientService); |
service = module.get<SocketService>(SocketService); |
||||||
}); |
}); |
||||||
|
|
||||||
it('should be defined', () => { |
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