add proxy switch demo

public/proxy.js

@@ -2,27 +2,82 @@ import { ProxyManager } from './proxy/proxy-manager.js';
 import { ProxySettings } from './proxy/proxy-settings.js';
 import { ProxyAuth } from './proxy/proxy-auth.js';
 import { ProxyActionType } from './types/action.js';
+import { ProxyLogs } from './proxy/proxy-logs.js';
 
 // 记录代理日志
 async function logProxyRequest(details, proxyConfig, error = null) {
-    const log = {
-        id: Date.now().toString(),
-        timestamp: Date.now(),
-        url: details.url,
-        proxyId: proxyConfig.id,
-        proxyName: proxyConfig.name,
-        status: error ? 'error' : 'success',
-        errorMessage: error?.message
-    };
+    try {
+        // 只记录主文档和 XHR 请求
+        if (!['main_frame', 'xmlhttprequest'].includes(details.type)) {
+            return;
+        }
 
-    const result = await chrome.storage.local.get('proxyLogs');
-    const logs = result.proxyLogs || [];
-    const updatedLogs = [log, ...logs].slice(0, 1000);
-    await chrome.storage.local.set({ proxyLogs: updatedLogs });
+        const log = {
+            id: Date.now().toString(),
+            timestamp: Date.now(),
+            url: details.url,
+            proxyId: proxyConfig.id,
+            proxyName: proxyConfig.name,
+            type: details.type,
+            status: error ? 'error' : 'success',
+            errorMessage: error?.message
+        };
+
+        // 使用 storage 存储日志
+        const result = await chrome.storage.local.get('proxyLogs');
+        const logs = result.proxyLogs || [];
+        const updatedLogs = [log, ...logs].slice(0, 100); // 只保留最新的 100 条
+        await chrome.storage.local.set({ proxyLogs: updatedLogs });
+
+    } catch (error) {
+        console.error('Error logging proxy request:', error);
+    }
 }
 
 async function handleSetProxyConfig(config, sendResponse) {
     try {
+        // 处理直接连接的情况
+        if (config.proxyType === 'direct') {
+            await chrome.proxy.settings.set({
+                value: { mode: "direct" },
+                scope: 'regular'
+            });
+
+            const settings = await chrome.proxy.settings.get({});
+            const isSuccess = settings.value.mode === "direct";
+
+            if (isSuccess) {
+                // 更新存储
+                await chrome.storage.local.set({
+                    currentProxy: {
+                        ...config,
+                        timestamp: Date.now()
+                    }
+                });
+
+                // 更新代理列表状态
+                const result = await chrome.storage.local.get('proxyConfigs');
+                if (result.proxyConfigs) {
+                    const updatedConfigs = result.proxyConfigs.map(c => ({
+                        ...c,
+                        enabled: c.id === config.id
+                    }));
+                    await chrome.storage.local.set({ proxyConfigs: updatedConfigs });
+                }
+
+                console.log('Direct connection set successfully');
+                sendResponse({ success: true });
+            } else {
+                console.error('Failed to set direct connection');
+                sendResponse({ 
+                    success: false, 
+                    error: '无法设置直接连接' 
+                });
+            }
+            return;
+        }
+
+        // 处理代理服务器的情况
         if (!config || !config.host || !config.port) {
             sendResponse({ 
                 success: false, 
@@ -162,12 +217,64 @@ async function handleGetProxyStatus(sendResponse) {
     }
 }
 
+// 添加代理请求监听器
+function setupProxyRequestListener() {
+    // 监听请求发送
+    chrome.webRequest.onBeforeRequest.addListener(
+        (details) => {
+            // 使用非阻塞方式处理请求
+            queueProxyLog(details).catch(error => {
+                console.error('Error in proxy request listener:', error);
+            });
+            // 不需要返回值
+        },
+        { urls: ["<all_urls>"] }
+    );
+
+    // 监听请求错误
+    chrome.webRequest.onErrorOccurred.addListener(
+        (details) => {
+            // 使用非阻塞方式处理错误
+            queueProxyLog(details, new Error(details.error)).catch(error => {
+                console.error('Error in proxy error listener:', error);
+            });
+        },
+        { urls: ["<all_urls>"] }
+    );
+}
+
+// 使用队列处理日志
+async function queueProxyLog(details, error = null) {
+    // 检查是否是扩展自身的请求
+    if (details.url.startsWith('chrome-extension://')) {
+        return;
+    }
+
+    // 检查代理状态
+    const settings = await chrome.proxy.settings.get({});
+    if (settings.value.mode !== "fixed_servers") {
+        return;
+    }
+
+    // 获取当前代理配置
+    const { currentProxy } = await chrome.storage.local.get('currentProxy');
+    if (!currentProxy) {
+        return;
+    }
+
+    // 记录日志
+    await logProxyRequest(details, currentProxy, error);
+}
+
 // 导出代理处理器设置函数
 export function setupProxyHandlers() {
     // 设置代理错误处理和认证
     ProxyAuth.setupErrorHandler();
     ProxyAuth.setupAuthListener();
 
+    // 设置代理请求监听器
+    setupProxyRequestListener();
+
     // 消息监听器
     chrome.runtime.onMessage.addListener((msg, sender, sendResponse) => {
         console.log("Proxy message:", msg);
@@ -184,6 +291,19 @@ export function setupProxyHandlers() {
             case ProxyActionType.GET_PROXY_STATUS:
                 handleGetProxyStatus(sendResponse);
                 return true;
+
+            case ProxyActionType.CLEAR_PROXY_LOGS:
+                chrome.storage.local.set({ proxyLogs: [] }, () => {
+                    if (chrome.runtime.lastError) {
+                        sendResponse({ 
+                            success: false, 
+                            error: chrome.runtime.lastError.message 
+                        });
+                    } else {
+                        sendResponse({ success: true });
+                    }
+                });
+                return true;
         }
     });