mirror of
https://github.com/lensapp/lens.git
synced 2025-05-20 05:10:56 +00:00
58 lines
2.4 KiB
TypeScript
58 lines
2.4 KiB
TypeScript
/**
|
|
* Copyright (c) 2021 OpenLens Authors
|
|
*
|
|
* Permission is hereby granted, free of charge, to any person obtaining a copy of
|
|
* this software and associated documentation files (the "Software"), to deal in
|
|
* the Software without restriction, including without limitation the rights to
|
|
* use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of
|
|
* the Software, and to permit persons to whom the Software is furnished to do so,
|
|
* subject to the following conditions:
|
|
*
|
|
* The above copyright notice and this permission notice shall be included in all
|
|
* copies or substantial portions of the Software.
|
|
*
|
|
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
|
* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS
|
|
* FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR
|
|
* COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER
|
|
* IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
|
|
* CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
|
|
*/
|
|
|
|
/**
|
|
* This file is here so that the "../shell-session" import can be injected into
|
|
* LensProxy at creation time. So that the `pty.node` extension isn't loaded
|
|
* into Lens Extension webpack bundle.
|
|
*/
|
|
|
|
import * as WebSocket from "ws";
|
|
import type http from "http";
|
|
import type net from "net";
|
|
import url from "url";
|
|
import { NodeShellSession, LocalShellSession } from "../shell-session";
|
|
import { ClusterManager } from "../cluster-manager";
|
|
import logger from "../logger";
|
|
|
|
function createWsListener(): WebSocket.Server {
|
|
const ws = new WebSocket.Server({ noServer: true });
|
|
|
|
return ws.on("connection", ((socket: WebSocket, req: http.IncomingMessage) => {
|
|
const cluster = ClusterManager.getInstance().getClusterForRequest(req);
|
|
const nodeParam = url.parse(req.url, true).query["node"]?.toString();
|
|
const shell = nodeParam
|
|
? new NodeShellSession(socket, cluster, nodeParam)
|
|
: new LocalShellSession(socket, cluster);
|
|
|
|
shell.open()
|
|
.catch(error => logger.error(`[SHELL-SESSION]: failed to open: ${error}`, { error }));
|
|
}));
|
|
}
|
|
|
|
export async function handleWsUpgrade(req: http.IncomingMessage, socket: net.Socket, head: Buffer) {
|
|
const wsServer = createWsListener();
|
|
|
|
wsServer.handleUpgrade(req, socket, head, (con) => {
|
|
wsServer.emit("connection", con, req);
|
|
});
|
|
}
|