chore: 整理 third-party Dark Reader 目录并移除本地说明文件跟踪

third-party/darkreader/tasks/reload.js

@@ -0,0 +1,130 @@
+// @ts-check
+import process from 'node:process';
+
+import {WebSocketServer} from 'ws';
+
+import {log} from './utils.js';
+
+export const PORT = 8890;
+const WAIT_FOR_CONNECTION = 2000;
+
+/** @type {import('ws').WebSocketServer | null} */
+let server = null;
+
+/** @type {Set<import('ws').WebSocket>} */
+const sockets = new Set();
+/** @type {WeakMap<import('ws').WebSocket, number>} */
+const times = new WeakMap();
+/** @type {WeakMap<import('ws').WebSocket, string>} */
+const userAgents = new WeakMap();
+
+/**
+ * @returns {Promise<import('ws').WebSocketServer>}
+ */
+function createServer() {
+    return new Promise((resolve) => {
+        const server = new WebSocketServer({port: PORT});
+        server.on('listening', () => {
+            log.ok('Auto-reloader started');
+            resolve(server);
+        });
+        server.on('connection', (ws, request) => {
+            const userAgent = request.headers['user-agent'];
+            log.ok(`Extension connected: ${userAgent}`);
+
+            sockets.add(ws);
+            times.set(ws, Date.now());
+            userAgent && userAgents.set(ws, userAgent);
+
+            ws.on('message', (data) => {
+                const message = JSON.parse(data.toString());
+                if (message.type === 'reloading') {
+                    log.ok('Extension reloading...');
+                }
+            });
+            ws.on('close', () => {
+                const userAgent = userAgents.get(ws);
+                log.warn(`Extension disconnected: ${userAgent}`);
+                sockets.delete(ws);
+            });
+            if (connectionAwaiter !== null) {
+                connectionAwaiter();
+            }
+        });
+    });
+}
+
+function closeServer() {
+    server && server.close(() => log.ok('Auto-reloader exit'));
+    sockets.forEach((ws) => ws.close());
+    sockets.clear();
+    server = null;
+}
+
+process.on('exit', closeServer);
+process.on('SIGINT', closeServer);
+
+/** @type {(() => void) | null} */
+let connectionAwaiter = null;
+
+function waitForConnection() {
+    return new Promise((resolve) => {
+        connectionAwaiter = () => {
+            connectionAwaiter = null;
+            clearTimeout(timeoutId);
+            setTimeout(resolve, WAIT_FOR_CONNECTION);
+        };
+        const timeoutId = setTimeout(() => {
+            log.warn('Auto-reloader did not connect');
+            connectionAwaiter = null;
+            resolve(true);
+        }, WAIT_FOR_CONNECTION);
+    });
+}
+
+/**
+ * @param {import('ws').WebSocket} ws
+ * @param {any} message
+ */
+function send(ws, message) {
+    ws.send(JSON.stringify(message));
+}
+
+/**
+ * @param {Object} options
+ * @param {string} options.type
+ */
+export async function reload({type}) {
+    if (!server) {
+        server = await createServer();
+    }
+    if (sockets.size === 0) {
+        await waitForConnection();
+    }
+    const now = Date.now();
+    Array.from(sockets.values())
+        .filter((ws) => {
+            const created = times.get(ws);
+            return created && created < now;
+        })
+        .forEach((ws) => send(ws, {type}));
+}
+
+export function getConnectedBrowsers() {
+    /** @type {Set<string>} */
+    const browsers = new Set();
+    sockets.forEach((ws) => {
+        const userAgent = userAgents.get(ws);
+        if (userAgent?.includes('Chrome') || userAgent?.includes('Chromium')) {
+            browsers.add('chrome');
+        }
+        if (userAgent?.includes('Firefox')) {
+            browsers.add('firefox');
+        }
+    });
+    return Array.from(browsers);
+}
+
+export const CSS = 'reload:css';
+export const FULL = 'reload:full';
+export const UI = 'reload:ui';