whistle.pastekitlab 1.2.5 → 1.3.0

package/index.js

@@ -1,8 +1,187 @@
-console.log('>>> testplugin loaded!');
-module.exports = (server, options) => {
-    console.log('>>> plugin 333 loaded');
-    server.on('request', (req, res, next) => {
-        console.log('>>> [GLOBAL]', req.fullUrl);
-        if (next) next();
+/**
+ * Whistle Plugin for PasteKit Lab（最终版 - 只监听）
+ */
+
+const WebSocket = require('ws');
+const fs = require('fs');
+const path = require('path');
+
+// ==================== 配置 ====================
+const CONFIG = {
+    wsPort: 8889,
+    logEnabled: true,
+    logFile: path.join(__dirname, 'plugin.log')
+};
+
+// ==================== 全局状态 ====================
+let wss = null;
+let clients = new Map();
+let clientIdCounter = 0;
+let domainConfig = [];
+
+// ==================== 日志 ====================
+function log(message) {
+    if (!CONFIG.logEnabled) return;
+
+    const timestamp = new Date().toISOString();
+    const logMessage = `[${timestamp}] ${message}\n`;
+
+    console.log('[pastekitlab]', message);
+
+    try {
+        fs.appendFileSync(CONFIG.logFile, logMessage, 'utf8');
+    } catch (e) {}
+}
+
+// ==================== 工具 ====================
+function bufferToString(buffer) {
+    try {
+        if (!buffer) return null;
+        const result = buffer.toString('utf8');
+        if (/[^\x20-\x7E\r\n\t]/.test(result)) return null;
+        return result;
+    } catch {
+        return null;
+    }
+}
+
+function bufferToBase64(buffer) {
+    try {
+        return buffer ? buffer.toString('base64') : null;
+    } catch {
+        return null;
+    }
+}
+
+function generateId() {
+    return 'req_' + Date.now() + '_' + Math.random().toString(36).slice(2);
+}
+
+// ==================== 域名过滤 ====================
+function shouldIntercept(url) {
+    if (!domainConfig.length) return true;
+
+    try {
+        const { hostname } = new URL(url);
+
+        return domainConfig.some(domain => {
+            if (hostname === domain) return true;
+            if (domain.startsWith('*.')) {
+                const base = domain.slice(2);
+                return hostname.endsWith('.' + base) || hostname === base;
+            }
+            return false;
+        });
+    } catch {
+        return false;
+    }
+}
+
+// ==================== WebSocket ====================
+function startWebSocketServer() {
+    if (wss) return;
+
+    wss = new WebSocket.Server({
+        port: CONFIG.wsPort,
+        path: '/ws'
+    });
+
+    wss.on('connection', (ws, req) => {
+        const id = ++clientIdCounter;
+
+        clients.set(id, ws);
+        log(`客户端连接: ${id}`);
+
+        ws.send(JSON.stringify({
+            msgType: 'connected',
+            clientId: id
+        }));
+
+        ws.on('message', (data) => {
+            try {
+                const msg = JSON.parse(data.toString());
+
+                if (msg.msgType === 'domain') {
+                    domainConfig = msg.domain || [];
+                    log(`更新域名配置: ${domainConfig.join(',')}`);
+                }
+            } catch {}
+        });
+
+        ws.on('close', () => {
+            clients.delete(id);
+            log(`客户端断开: ${id}`);
+        });
+    });
+
+    log(`WebSocket 启动: ws://127.0.0.1:${CONFIG.wsPort}/ws`);
+}
+
+function broadcast(type, data) {
+    if (!clients.size) return;
+
+    const msg = JSON.stringify({
+        msgType: type,
+        ...data,
+        timestamp: Date.now()
+    });
+
+    clients.forEach(ws => {
+        if (ws.readyState === WebSocket.OPEN) {
+            ws.send(msg);
+        }
+    });
+}
+
+// ==================== Whistle 插件 ====================
+module.exports = (server) => {
+    log('插件已启动（server.on 模式）');
+    startWebSocketServer();
+
+    // ==================== 请求监听 ====================
+    server.on('request', (req, res) => {
+        try {
+            const url = req.fullUrl || req.url;
+            if (!shouldIntercept(url)) return;
+
+            const data = {
+                eventId: generateId(),
+                url,
+                method: req.method,
+                headers: req.headers,
+                body: bufferToString(req._reqBody),
+                bodyBase64: bufferToBase64(req._reqBody)
+            };
+
+            log(`REQ ${req.method} ${url}`);
+            broadcast('REQUEST', data);
+
+        } catch (e) {
+            log('request error: ' + e.message);
+        }
+    });
+
+    // ==================== 响应监听 ====================
+    server.on('response', (req, res) => {
+        try {
+            const url = req.fullUrl || req.url;
+            if (!shouldIntercept(url)) return;
+
+            const data = {
+                eventId: generateId(),
+                url,
+                method: req.method,
+                statusCode: res.statusCode,
+                headers: res.headers,
+                body: bufferToString(res._resBody),
+                bodyBase64: bufferToBase64(res._resBody)
+            };
+
+            log(`RES ${res.statusCode} ${url}`);
+            broadcast('RESPONSE', data);
+
+        } catch (e) {
+            log('response error: ' + e.message);
+        }
     });
 };

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "whistle.pastekitlab",
-  "version": "1.2.5",
+  "version": "1.3.0",
   "description": "Whistle plugin for PasteKit Lab - Intercepts requests and sends them to requestlistviewer via WebSocket",
   "main": "index.js",
   "scripts": {