☰

[MWI] WebSocket 调试工具

显示 WebSocket 收发的 JSON 数据，并支持手动发包和屏蔽指定类型

// ==UserScript==
// @name         [MWI] WebSocket 调试工具
// @namespace    http://tampermonkey.net/
// @version      1.0.1
// @description  显示 WebSocket 收发的 JSON 数据，并支持手动发包和屏蔽指定类型
// @author       XIxixi297
// @license      MIT
// @match        https://www.milkywayidle.com/*
// @match        https://test.milkywayidle.com/*
// @grant        none
// @run-at       document-start
// ==/UserScript==

(function () {
    'use strict';

    if (window.__wsLoggerInstalled) return;
    window.__wsLoggerInstalled = true;

    // 全局状态
    const wsLogger = {
        instances: new Set(),
        currentWS: null,
        blockedTypes: new Set(),
        STORAGE_KEY: 'ws_blocked_types'
    };

    // 初始化屏蔽列表
    try {
        const stored = localStorage.getItem(wsLogger.STORAGE_KEY);
        if (stored) wsLogger.blockedTypes = new Set(JSON.parse(stored));
    } catch (e) {}

    function saveBlockedTypes() {
        try {
            localStorage.setItem(wsLogger.STORAGE_KEY, JSON.stringify([...wsLogger.blockedTypes]));
        } catch (e) {}
    }

    // 消息日志记录
    function logMessage(data, direction) {
        try {
            const parsed = typeof data === 'string' ? JSON.parse(data) : data;
            const msgType = parsed.type || '未知类型';

            if (!wsLogger.blockedTypes.has(msgType)) {
                const color = direction === 'send' ? '#03A9F4' : '#4CAF50';
                const arrow = direction === 'send' ? '→' : '←';
                console.groupCollapsed(`%c${arrow} ${msgType}`, `color: ${color}; font-weight: bold;`);
                console.log(parsed);
                console.groupEnd();
            }
        } catch (e) {
            const color = direction === 'send' ? '#03A9F4' : '#4CAF50';
            const arrow = direction === 'send' ? '→' : '←';
            console.log(`%c${arrow} 非JSON:`, `color: ${color};`, data);
        }
    }

    // WebSocket劫持
    const OriginalWebSocket = window.WebSocket;
    window.WebSocket = function(url, protocols) {
        const ws = new OriginalWebSocket(url, protocols);

        // 添加到实例集合
        wsLogger.instances.add(ws);
        wsLogger.currentWS = ws;

        // Hook send 方法
        const originalSend = ws.send;
        ws.send = function(data) {
            logMessage(data, 'send');
            return originalSend.apply(this, arguments);
        };

        // 监听消息
        ws.addEventListener('message', (event) => {
            logMessage(event.data, 'receive');
        });

        ws.addEventListener('open', () => {
            console.info('%cWebSocket 已连接: ' + url, 'color: gray;');
        });

        ws.addEventListener('close', () => {
            console.warn('%cWebSocket 已断开', 'color: orange;');
            wsLogger.instances.delete(ws);
            if (wsLogger.currentWS === ws) {
                const remaining = [...wsLogger.instances];
                wsLogger.currentWS = remaining[remaining.length - 1] || null;
            }
        });

        return ws;
    };

    // 保持原型链
    Object.setPrototypeOf(window.WebSocket, OriginalWebSocket);
    window.WebSocket.prototype = OriginalWebSocket.prototype;

    // 工具函数
    window.sendWS = function(data) {
        if (!wsLogger.currentWS || wsLogger.currentWS.readyState !== WebSocket.OPEN) {
            console.error('没有可用的WebSocket连接');
            return false;
        }
        const jsonData = typeof data === 'string' ? data : JSON.stringify(data);
        wsLogger.currentWS.send(jsonData);
        console.log('%c✅ 手动发送:', 'color: #FF9800; font-weight: bold;', data);
        return true;
    };

    window.listWS = function() {
        console.log('%cWebSocket 连接列表:', 'color: #9C27B0; font-weight: bold;');
        [...wsLogger.instances].forEach((ws, index) => {
            const status = ws.readyState === WebSocket.OPEN ? '✅' : '❌';
            console.log(`[${index}] ${ws.url} ${status}`);
        });
    };

    window.blockType = function(type) {
        if (typeof type === 'string') {
            wsLogger.blockedTypes.add(type);
        } else if (Array.isArray(type)) {
            type.forEach(t => wsLogger.blockedTypes.add(t));
        }
        saveBlockedTypes();
        console.log('%c🚫 已屏蔽:', 'color: #F44336; font-weight: bold;', type);
    };

    window.unblockType = function(type) {
        if (typeof type === 'string') {
            wsLogger.blockedTypes.delete(type);
        } else if (Array.isArray(type)) {
            type.forEach(t => wsLogger.blockedTypes.delete(t));
        }
        saveBlockedTypes();
        console.log('%c✅ 已取消屏蔽:', 'color: #4CAF50; font-weight: bold;', type);
    };

    window.listBlocked = function() {
        if (wsLogger.blockedTypes.size === 0) {
            console.log('%c无屏蔽类型', 'color: #607D8B;');
        } else {
            console.log('%c屏蔽列表:', 'color: #F44336; font-weight: bold;', [...wsLogger.blockedTypes]);
        }
    };

    window.clearBlocked = function() {
        const count = wsLogger.blockedTypes.size;
        wsLogger.blockedTypes.clear();
        saveBlockedTypes();
        console.log(`%c✅ 已清空 ${count} 个屏蔽类型`, 'color: #4CAF50; font-weight: bold;');
    };

    // 启动提示
    console.info('%c[MWI] WebSocket监听器已启用', 'color: purple; font-weight: bold;');

    if (wsLogger.blockedTypes.size > 0) {
        console.info(`%c已加载 ${wsLogger.blockedTypes.size} 个屏蔽类型:`, 'color: #FF9800; font-weight: bold;', [...wsLogger.blockedTypes]);
    }

    console.info('%c使用方法:', 'color: #2196F3; font-weight: bold;');
    console.info('%c  sendWS(data) - 发送消息到当前WebSocket', 'color: #2196F3;');
    console.info('%c  blockType(type) - 屏蔽指定类型消息 (支持字符串或数组)', 'color: #F44336;');
    console.info('%c  unblockType(type) - 取消屏蔽指定类型', 'color: #4CAF50;');
    console.info('%c  listBlocked() - 查看当前屏蔽的消息类型', 'color: #607D8B;');
    console.info('%c  clearBlocked() - 清空所有屏蔽类型', 'color: #607D8B;');
    console.info('%c示例: blockType(["chat_message_received", "ping"]) - 屏蔽多个类型', 'color: #9E9E9E;');

})();