FilesExpand file tree

main

/

wsRouter.ts

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

54 lines (46 loc) · 1.57 KB

main

/

wsRouter.ts

Copy path

Top

File metadata and controls

• Code
• Blame

54 lines (46 loc) · 1.57 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

39

40

41

42

43

44

45

46

47

48

49

50

51

52

53

54

import * as express from "express"

import * as expressCore from "express-serve-static-core"

import * as http from "http"

import Websocket from "ws"

import * as pluginapi from "../../typings/pluginapi"

export const handleUpgrade = (app: express.Express, server: http.Server): void => {

server.on("upgrade", (req, socket, head) => {

socket.pause()

req.ws = socket

req.head = head

req._ws_handled = false

// Send the request off to be handled by Express.

;(app as any).handle(req, new http.ServerResponse(req), () => {

if (!req._ws_handled) {

socket.end("HTTP/1.1 404 Not Found\r\n\r\n")

}

})

})

}

interface InternalWebsocketRequest extends pluginapi.WebsocketRequest {

_ws_handled: boolean

}

export class WebsocketRouter {

public readonly router = express.Router()

/**

* Handle a websocket at this route. Note that websockets are immediately

* paused when they come in.

*/

public ws(route: expressCore.PathParams, ...handlers: pluginapi.WebSocketHandler[]): void {

this.router.get(

route,

...handlers.map((handler) => {

const wrapped: express.Handler = (req, res, next) => {

;(req as InternalWebsocketRequest)._ws_handled = true

return handler(req as pluginapi.WebsocketRequest, res, next)

}

return wrapped

}),

)

}

}

export function Router(): WebsocketRouter {

return new WebsocketRouter()

}

// eslint-disable-next-line import/no-named-as-default-member -- the typings are not updated correctly

export const wss = new Websocket.Server({ noServer: true })