pyers/prsm
1
0
Fork 0
mirror of https://github.com/nvms/prsm.git synced 2025-12-16 16:10:54 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

allow typing on the callback return value for registerCommand

Browse Source
This commit is contained in:
nvms 2024-09-04 10:49:32 -04:00
parent 8ef12272be
commit 8dbc214e6f
2 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

BIN
packages/express-keepalive-ws/bun.lockb
View File

Binary file not shown.

8
packages/keepalive-ws/src/server/index.ts
View File

@ -130,7 +130,9 @@ export type KeepAliveServerOptions = ServerOptions & {
export class KeepAliveServer extends WebSocketServer { export class KeepAliveServer extends WebSocketServer {
connections: { [id: string]: Connection } = {}; connections: { [id: string]: Connection } = {};
remoteAddressToConnections: { [address: string]: Connection[] } = {}; remoteAddressToConnections: { [address: string]: Connection[] } = {};
commands: { [command: string]: (context: WSContext<any>) => Promise<void> } = {}; commands: {
[command: string]: (context: WSContext<any>) => Promise<any> | any;
} = {};
globalMiddlewares: SocketMiddleware[] = []; globalMiddlewares: SocketMiddleware[] = [];
middlewares: { [key: string]: SocketMiddleware[] } = {}; middlewares: { [key: string]: SocketMiddleware[] } = {};
rooms: { [roomName: string]: Set<string> } = {}; rooms: { [roomName: string]: Set<string> } = {};
@ -330,9 +332,9 @@ export class KeepAliveServer extends WebSocketServer {
this.rooms[roomName] = new Set(); this.rooms[roomName] = new Set();
} }
registerCommand( registerCommand<T>(
command: string, command: string,
callback: SocketMiddleware, callback: (context: WSContext<any>) => Promise<T> | T,
middlewares: SocketMiddleware[] = [], middlewares: SocketMiddleware[] = [],
) { ) {
this.commands[command] = callback; this.commands[command] = callback;