import { OB11EmitEventContent, OB11NetworkReloadType } from './index'; import express, { Express, NextFunction, Request, Response } from 'express'; import http from 'http'; import { NapCatCore } from '@/core'; import { OB11Response } from '@/onebot/action/OneBotAction'; import { ActionMap } from '@/onebot/action'; import cors from 'cors'; import { HttpServerConfig } from '@/onebot/config/config'; import { NapCatOneBot11Adapter } from '@/onebot'; import { IOB11NetworkAdapter } from '@/onebot/network/adapter'; import json5 from 'json5'; export class OB11HttpServerAdapter extends IOB11NetworkAdapter { private app: Express | undefined; private server: http.Server | undefined; constructor(name: string, config: HttpServerConfig, core: NapCatCore, obContext: NapCatOneBot11Adapter, actions: ActionMap) { super(name, config, core, obContext, actions); } // eslint-disable-next-line @typescript-eslint/no-unused-vars override onEvent(_event: T) { // http server is passive, no need to emit event } open() { try { if (this.isEnable) { this.core.context.logger.logError('Cannot open a closed HTTP server'); return; } if (!this.isEnable) { this.initializeServer(); this.isEnable = true; } } catch (e) { this.core.context.logger.logError(`[OneBot] [HTTP Server Adapter] Boot Error: ${e}`); } } async close() { this.isEnable = false; this.server?.close(); this.app = undefined; } private isFinished(req: Request): boolean { return req.complete; } private hasbody(req: Request): boolean { return req.headers['content-length'] !== undefined && req.headers['content-length'] !== '0'; } private initializeServer() { this.app = express(); this.server = http.createServer(this.app); this.app.use(cors()); this.app.use(express.urlencoded({ extended: true, limit: '5000mb' })); this.app.use((req, res, next) => { if (this.isFinished(req)) { next(); return; } if (!this.hasbody(req)) { next(); return; } // 兼容处理没有带content-type的请求 req.headers['content-type'] = 'application/json'; let rawData = ''; req.on('data', (chunk) => { rawData += chunk; }); req.on('end', () => { try { req.body = { ...json5.parse(rawData || '{}'), ...req.body }; next(); } catch { return res.status(400).send('Invalid JSON'); } return; }); req.on('error', () => { return res.status(400).send('Invalid JSON'); }); }); //@ts-expect-error authorize this.app.use((req, res, next) => this.authorize(this.config.token, req, res, next)); this.app.use(async (req, res) => { await this.handleRequest(req, res); }); this.server.listen(this.config.port, () => { this.core.context.logger.log(`[OneBot] [HTTP Server Adapter] Start On Port ${this.config.port}`); }); } private authorize(token: string | undefined, req: Request, res: Response, next: NextFunction) { if (!token || token.length == 0) return next();//客户端未设置密钥 const HeaderClientToken = req.headers.authorization?.split('Bearer ').pop() || ''; const QueryClientToken = req.query['access_token']; const ClientToken = typeof (QueryClientToken) === 'string' && QueryClientToken !== '' ? QueryClientToken : HeaderClientToken; if (ClientToken === token) { return next(); } else { return res.status(403).send(JSON.stringify({ message: 'token verify failed!' })); } } async httpApiRequest(req: Request, res: Response) { let payload = req.body; if (req.method == 'get') { payload = req.query; } else if (req.query) { payload = { ...req.body, ...req.query }; } if (req.path === '' || req.path === '/') { const hello = OB11Response.ok({}); hello.message = 'NapCat4 Is Running'; return res.json(hello); } const actionName = req.path.split('/')[1]; // eslint-disable-next-line @typescript-eslint/no-explicit-any const action = this.actions.get(actionName as any); if (action) { try { const result = await action.handle(payload, this.name, this.config); return res.json(result); } catch (error: unknown) { return res.json(OB11Response.error((error as Error)?.stack?.toString() || (error as Error)?.message || 'Error Handle', 200)); } } else { return res.json(OB11Response.error('不支持的Api ' + actionName, 200)); } } async handleRequest(req: Request, res: Response) { if (!this.isEnable) { this.core.context.logger.log('[OneBot] [HTTP Server Adapter] Server is closed'); res.json(OB11Response.error('Server is closed', 200)); return; } this.httpApiRequest(req, res); return; } async reload(newConfig: HttpServerConfig) { const wasEnabled = this.isEnable; const oldPort = this.config.port; this.config = newConfig; if (newConfig.enable && !wasEnabled) { this.open(); return OB11NetworkReloadType.NetWorkOpen; } else if (!newConfig.enable && wasEnabled) { this.close(); return OB11NetworkReloadType.NetWorkClose; } if (oldPort !== newConfig.port) { this.close(); if (newConfig.enable) { this.open(); } return OB11NetworkReloadType.NetWorkReload; } return OB11NetworkReloadType.Normal; } }