vk-ssl-auto-deploy 0.6.7 → 0.7.0

package/utils/websocket.js

@@ -0,0 +1,116 @@
+const WebSocket = require('ws');
+
+// WebSocket 客户端集合
+const wsClients = new Set();
+
+// 清理ANSI颜色代码
+function stripAnsiCodes(str) {
+	// 移除所有ANSI转义序列
+	return str.replace(/\x1b\[[0-9;]*m/g, '');
+}
+
+// 广播日志到所有WebSocket客户端
+function broadcastLog(message, type = 'info') {
+	// 清理ANSI颜色代码
+	const cleanMessage = stripAnsiCodes(message);
+	const data = JSON.stringify({ type, message: cleanMessage });
+	wsClients.forEach(client => {
+		if (client.readyState === WebSocket.OPEN) {
+			try {
+				client.send(data);
+			} catch (error) {
+				// 使用原始console.error避免递归
+				originalConsoleError('[WebSocket] 发送失败:', error.message);
+			}
+		}
+	});
+}
+
+// 保存原始的 console 方法
+const originalConsoleLog = console.log;
+const originalConsoleError = console.error;
+const originalConsoleWarn = console.warn;
+
+// 拦截 console 方法，将日志广播到 WebSocket
+function interceptConsole() {
+	console.log = function(...args) {
+		originalConsoleLog.apply(console, args);
+		const message = args.map(arg => 
+			typeof arg === 'object' ? JSON.stringify(arg) : String(arg)
+		).join(' ');
+		
+		// 根据消息内容判断日志类型
+		let type = 'info';
+		// 优先判断错误类型
+		if ((message.includes('失败') || message.includes('错误') || message.includes('Error') || message.includes('error') || message.includes('[失败]')) && !message.includes('✓') && !message.includes('成功') && !message.includes('统计') ) {
+			type = 'error';
+		} else if (message.includes('[警告]') || message.includes('[需要更新]') || message.includes('[需要下载]')) {
+			type = 'warning';
+		} else if (message.includes('[成功]') || message.includes('✓')) {
+			type = 'success';
+		} else if (message.includes('[步骤') || message.includes('[任务')) {
+			type = 'info';
+		} else {
+			type = 'dim';
+		}
+		
+		broadcastLog(message, type);
+	};
+
+	console.error = function(...args) {
+		originalConsoleError.apply(console, args);
+		const message = args.map(arg => 
+			typeof arg === 'object' ? JSON.stringify(arg) : String(arg)
+		).join(' ');
+		broadcastLog(message, 'error');
+	};
+
+	console.warn = function(...args) {
+		originalConsoleWarn.apply(console, args);
+		const message = args.map(arg => 
+			typeof arg === 'object' ? JSON.stringify(arg) : String(arg)
+		).join(' ');
+		broadcastLog(message, 'warning');
+	};
+}
+
+// 设置 WebSocket 服务器
+function setupWebSocket(server) {
+	// 创建 WebSocket 服务器
+	const wss = new WebSocket.Server({ 
+		server,
+		path: '/ws'
+	});
+
+	// WebSocket 连接处理
+	wss.on('connection', (ws) => {
+		console.log('[WebSocket] 新客户端连接');
+		wsClients.add(ws);
+		
+		// 发送欢迎消息
+		ws.send(JSON.stringify({
+			type: 'success',
+			message: '服务器 连接成功'
+		}));
+		
+		ws.on('close', () => {
+			console.log('[WebSocket] 客户端断开连接');
+			wsClients.delete(ws);
+		});
+		
+		ws.on('error', (error) => {
+			console.error('[WebSocket] 错误:', error.message);
+			wsClients.delete(ws);
+		});
+	});
+
+	// 拦截 console 方法
+	interceptConsole();
+
+	return wss;
+}
+
+module.exports = {
+	setupWebSocket,
+	broadcastLog
+};