From 91bc3ab5257e848c8f73b2b121c36f4e95b9136e Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E6=89=8B=E7=93=9C=E4=B8=80=E5=8D=81=E9=9B=AA?= Date: Sat, 10 Aug 2024 13:52:06 +0800 Subject: [PATCH] =?UTF-8?q?=F0=9F=98=8A=E5=8F=B2=E5=B1=B1=E7=A7=BB?= =?UTF-8?q?=E5=87=BA=E5=8E=86=E5=8F=B2=E9=95=BF=E6=B2=B3?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/common/server/http.ts | 133 --------------------------------- src/common/server/websocket.ts | 129 -------------------------------- 2 files changed, 262 deletions(-) delete mode 100644 src/common/server/http.ts delete mode 100644 src/common/server/websocket.ts diff --git a/src/common/server/http.ts b/src/common/server/http.ts deleted file mode 100644 index 1a57d55e..00000000 --- a/src/common/server/http.ts +++ /dev/null @@ -1,133 +0,0 @@ -import express, { Express, Request, Response } from 'express'; -import cors from 'cors'; -import http from 'http'; -import { LogWrapper } from '../utils/log'; -type RegisterHandler = (res: Response, payload: any) => Promise - -export abstract class HttpServerBase { - name: string = 'NapCatQQ'; - private readonly expressAPP: Express; - private _server: http.Server | null = null; - logger: LogWrapper; - serverToken: string | undefined = undefined; - - public get server(): http.Server | null { - return this._server; - } - - private set server(value: http.Server | null) { - this._server = value; - } - - constructor(logger: LogWrapper, token: string) { - this.serverToken = token; - this.logger = logger; - this.expressAPP = express(); - this.expressAPP.use(cors()); - this.expressAPP.use(express.urlencoded({ extended: true, limit: '5000mb' })); - this.expressAPP.use((req, res, next) => { - // 兼容处理没有带content-type的请求 - // log("req.headers['content-type']", req.headers['content-type']) - req.headers['content-type'] = 'application/json'; - const originalJson = express.json({ limit: '5000mb' }); - // 调用原始的express.json()处理器 - originalJson(req, res, (err) => { - if (err) { - this.logger.logError('Error parsing JSON:', err); - return res.status(400).send('Invalid JSON'); - } - next(); - }); - }); - } - - authorize(req: Request, res: Response, next: () => void) { - let clientToken = ''; - const authHeader = req.get('authorization'); - if (authHeader) { - clientToken = authHeader.split('Bearer ').pop() || ''; - //logDebug('receive http header token', clientToken); - } else if (req.query.access_token) { - if (Array.isArray(req.query.access_token)) { - clientToken = req.query.access_token[0].toString(); - } else { - clientToken = req.query.access_token.toString(); - } - //logDebug('receive http url token', clientToken); - } - - if (this.serverToken && clientToken != this.serverToken) { - return res.status(403).send(JSON.stringify({ message: 'token verify failed!' })); - } - next(); - } - - start(port: number, host: string) { - try { - this.expressAPP.get('/', (req: Request, res: Response) => { - res.send(`${this.name}已启动`); - }); - this.listen(port, host); - } catch (e: any) { - this.logger.logError('HTTP服务启动失败', e.toString()); - // httpServerError = "HTTP服务启动失败, " + e.toString() - } - } - - stop() { - // httpServerError = "" - if (this.server) { - this.server.close(); - this.server = null; - } - } - - restart(port: number, host: string) { - this.stop(); - this.start(port, host); - } - - abstract handleFailed(res: Response, payload: any, err: Error): void - - registerRouter(method: 'post' | 'get' | string, url: string, handler: RegisterHandler) { - if (!url.startsWith('/')) { - url = '/' + url; - } - - // @ts-expect-error wait fix - if (!this.expressAPP[method]) { - const err = `${this.name} register router failed，${method} not exist`; - this.logger.logError(err); - throw err; - } - // @ts-expect-error wait fix - this.expressAPP[method](url, this.authorize, async (req: Request, res: Response) => { - let payload = req.body; - if (method == 'get') { - payload = req.query; - } else if (req.query) { - payload = { ...req.query, ...req.body }; - } - this.logger.logDebug('收到http请求', url, payload); - try { - res.send(await handler(res, payload)); - } catch (e: any) { - this.handleFailed(res, payload, e); - } - }); - } - - protected listen(port: number, host: string = '0.0.0.0') { - host = host || '0.0.0.0'; - try { - this.server = this.expressAPP.listen(port, host, () => { - const info = `${this.name} started ${host}:${port}`; - this.logger.log(info); - }).on('error', (err) => { - this.logger.logError('HTTP服务启动失败', err.toString()); - }); - } catch (e: any) { - this.logger.logError('HTTP服务启动失败, 请检查监听的ip地址和端口', e.stack.toString()); - } - } -} diff --git a/src/common/server/websocket.ts b/src/common/server/websocket.ts deleted file mode 100644 index 161ad09a..00000000 --- a/src/common/server/websocket.ts +++ /dev/null @@ -1,129 +0,0 @@ -import { WebSocket, WebSocketServer } from 'ws'; -import http from 'http'; -import urlParse from 'url'; -import { IncomingMessage } from 'node:http'; -import { LogWrapper } from '../utils/log'; - -class WebsocketClientBase { - private wsClient: WebSocket | undefined; - - constructor() { - } - - send(msg: string) { - if (this.wsClient && this.wsClient.readyState == WebSocket.OPEN) { - this.wsClient.send(msg); - } - } - - onMessage(msg: string) { - - } -} - -export class WebsocketServerBase { - private ws: WebSocketServer | null = null; - public token: string = ''; - logger: LogWrapper; - - constructor(logger:LogWrapper) { - this.logger = logger; - } - - start(port: number | http.Server, host: string = '') { - if (port instanceof http.Server) { - try { - const wss = new WebSocketServer({ - noServer: true, - maxPayload: 1024 * 1024 * 1024 - }).on('error', () => { - }); - this.ws = wss; - port.on('upgrade', function upgrade(request, socket, head) { - wss.handleUpgrade(request, socket, head, function done(ws) { - wss.emit('connection', ws, request); - }); - }); - this.logger.log('ws服务启动成功, 绑定到HTTP服务'); - } catch (e: any) { - throw Error('ws服务启动失败, 可能是绑定的HTTP服务异常' + e.toString()); - } - } else { - try { - this.ws = new WebSocketServer({ - port, - host: '', - maxPayload: 1024 * 1024 * 1024 - }).on('error', () => { - }); - this.logger.log(`ws服务启动成功, ${host}:${port}`); - } catch (e: any) { - throw Error('ws服务启动失败, 请检查监听的ip和端口' + e.toString()); - } - } - this.ws.on('connection', (wsClient, req) => { - const url: string = req.url!.split('?').shift() || '/'; - this.authorize(wsClient, req); - this.onConnect(wsClient, url, req); - wsClient.on('message', async (msg) => { - this.onMessage(wsClient, url, msg.toString()); - }); - }); - } - - stop() { - if (this.ws) { - this.ws.close((err) => { - if (err) this.logger.log('ws server close failed!', err); - }); - this.ws = null; - } - } - - restart(port: number) { - this.stop(); - this.start(port); - } - - authorize(wsClient: WebSocket, req: IncomingMessage) { - const url = req.url!.split('?').shift(); - this.logger.log('ws connect', url); - let clientToken: string = ''; - const authHeader = req.headers['authorization']; - if (authHeader) { - clientToken = authHeader.split('Bearer ').pop() || ''; - this.logger.log('receive ws header token', clientToken); - } else { - const parsedUrl = urlParse.parse(req.url || '/', true); - const urlToken = parsedUrl.query.access_token; - if (urlToken) { - if (Array.isArray(urlToken)) { - clientToken = urlToken[0]; - } else { - clientToken = urlToken; - } - this.logger.log('receive ws url token', clientToken); - } - } - if (this.token && clientToken != this.token) { - this.authorizeFailed(wsClient); - return wsClient.close(); - } - } - - authorizeFailed(wsClient: WebSocket) { - - } - - onConnect(wsClient: WebSocket, url: string, req: IncomingMessage) { - - } - - onMessage(wsClient: WebSocket, url: string, msg: string) { - - } - - sendHeart() { - - } -}