cubetiq/code-server
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Expose websocket server to plugins

Browse Source 
Same reasoning used when exposing Express.
This commit is contained in:
Asher 2021-01-28 12:48:47 -06:00
parent b13db3124b
commit 5505959f7e
No known key found for this signature in database
GPG Key ID: D63C1EF81242354A
3 changed files with 9 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
src/node/plugin.ts
View File

@ -7,7 +7,7 @@ import * as pluginapi from "../../typings/pluginapi"
import { version } from "./constants"
import { proxy } from "./proxy"
import * as util from "./util"
import { Router as WsRouter, WebsocketRouter } from "./wsRouter"
import { Router as WsRouter, WebsocketRouter, wss } from "./wsRouter"
const fsp = fs.promises
/**
@ -24,6 +24,7 @@ require("module")._load = function (request: string, parent: object, isMain: boo
Level,
proxy,
WsRouter,
wss,
}
}
return originalLoad.apply(this, [request, parent, isMain])

5
test/test-plugin/src/index.ts
View File

@ -1,8 +1,5 @@
import * as cs from "code-server"
import * as fspath from "path"
import Websocket from "ws"
const wss = new Websocket.Server({ noServer: true })
export const plugin: cs.Plugin = {
displayName: "Test Plugin",
@ -28,7 +25,7 @@ export const plugin: cs.Plugin = {
wsRouter() {
const wr = cs.WsRouter()
wr.ws("/test-app", (req) => {
wss.handleUpgrade(req, req.socket, req.head, (ws) => {
cs.wss.handleUpgrade(req, req.socket, req.head, (ws) => {
ws.send("hello")
})
})

7
typings/pluginapi.d.ts vendored
View File

@ -6,6 +6,7 @@ import * as express from "express"
import * as expressCore from "express-serve-static-core"
import ProxyServer from "http-proxy"
import * as net from "net"
import Websocket from "ws"
/**
* Overlay
@ -102,6 +103,11 @@ export interface WebsocketRouter {
*/
export function WsRouter(): WebsocketRouter
/**
* The websocket server used by code-server.
*/
export const wss: Websocket.Server
/**
* The Express import used by code-server.
*
@ -110,7 +116,6 @@ export function WsRouter(): WebsocketRouter
* instances.
*/
export { express }
/**
* Use to add a field to a log.
*