mirror of
https://github.com/yaklang/yaklang-chrome-extension.git
synced 2026-09-22 03:10:43 +08:00
552 lines
19 KiB
JavaScript
552 lines
19 KiB
JavaScript
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';
|
|
import {proxyStore} from './db/proxy-store.js';
|
|
|
|
// 修改代理状态获取函数为 Promise 形式
|
|
function getProxySettings() {
|
|
return new Promise((resolve) => {
|
|
chrome.proxy.settings.get({}, resolve);
|
|
});
|
|
}
|
|
|
|
async function handleSetProxyConfig(config, sendResponse) {
|
|
try {
|
|
// 处理代理服务器的情况
|
|
if (config.proxyType === 'fixed_servers') {
|
|
// 固定代理服务器模式需要验证 host 和 port
|
|
if (!config || !config.host || !config.port) {
|
|
sendResponse({
|
|
success: false,
|
|
error: '无效的代理配置：缺少主机或端口'
|
|
});
|
|
return;
|
|
}
|
|
|
|
const proxyConfig = {
|
|
mode: "fixed_servers",
|
|
rules: {
|
|
singleProxy: {
|
|
scheme: config.scheme || 'http',
|
|
host: config.host,
|
|
port: parseInt(config.port)
|
|
},
|
|
bypassList: config.bypassList || []
|
|
}
|
|
};
|
|
|
|
await new Promise((resolve) => {
|
|
chrome.proxy.settings.set({
|
|
value: proxyConfig,
|
|
scope: 'regular'
|
|
}, resolve);
|
|
});
|
|
|
|
const settings = await getProxySettings();
|
|
const isSuccess = settings.value.mode === "fixed_servers" &&
|
|
settings.value.rules.singleProxy.host === config.host &&
|
|
settings.value.rules.singleProxy.port === parseInt(config.port);
|
|
|
|
if (isSuccess) {
|
|
await proxyStore.setCurrentProxy({
|
|
...config,
|
|
timestamp: Date.now()
|
|
});
|
|
|
|
const configs = await proxyStore.getProxyConfigs();
|
|
const updatedConfigs = configs.map(c => ({
|
|
...c,
|
|
enabled: c.id === config.id
|
|
}));
|
|
await proxyStore.saveProxyConfigs(updatedConfigs);
|
|
|
|
console.log('Proxy successfully set:', settings.value);
|
|
sendResponse({success: true});
|
|
|
|
// 通知所有 content scripts 更新
|
|
await notifyProxyStatusChanged();
|
|
} else {
|
|
console.error('Proxy settings verification failed');
|
|
sendResponse({
|
|
success: false,
|
|
error: '代理设置验证失败'
|
|
});
|
|
}
|
|
return;
|
|
} else if (config.proxyType === 'pac_script') {
|
|
// PAC 脚本模式需要验证 pacScript
|
|
if (!config || !config.pacScript || !config.pacScript.data) {
|
|
sendResponse({
|
|
success: false,
|
|
error: '无效的 PAC 脚本配置'
|
|
});
|
|
return;
|
|
}
|
|
|
|
const proxyConfig = {
|
|
mode: "pac_script",
|
|
pacScript: config.pacScript
|
|
};
|
|
|
|
await new Promise((resolve) => {
|
|
chrome.proxy.settings.set({
|
|
value: proxyConfig,
|
|
scope: 'regular'
|
|
}, resolve);
|
|
});
|
|
|
|
const settings = await getProxySettings();
|
|
const isSuccess = settings.value.mode === "pac_script" &&
|
|
settings.value.pacScript &&
|
|
settings.value.pacScript.data;
|
|
|
|
if (isSuccess) {
|
|
await proxyStore.setCurrentProxy({
|
|
...config,
|
|
timestamp: Date.now()
|
|
});
|
|
|
|
const configs = await proxyStore.getProxyConfigs();
|
|
const updatedConfigs = configs.map(c => ({
|
|
...c,
|
|
enabled: c.id === config.id
|
|
}));
|
|
await proxyStore.saveProxyConfigs(updatedConfigs);
|
|
|
|
console.log('PAC script proxy successfully set:', settings.value);
|
|
sendResponse({success: true});
|
|
|
|
// 通知所有 content scripts 更新
|
|
await notifyProxyStatusChanged();
|
|
} else {
|
|
console.error('PAC script settings verification failed', {
|
|
expected: config,
|
|
actual: settings.value
|
|
});
|
|
sendResponse({
|
|
success: false,
|
|
error: '代理设置验证失败'
|
|
});
|
|
}
|
|
return;
|
|
} else if (config.proxyType === 'direct' || config.proxyType === 'system') {
|
|
// 直接连接或系统代理模式
|
|
const proxyConfig = {
|
|
mode: config.proxyType
|
|
};
|
|
|
|
await new Promise((resolve) => {
|
|
chrome.proxy.settings.set({
|
|
value: proxyConfig,
|
|
scope: 'regular'
|
|
}, resolve);
|
|
});
|
|
|
|
const settings = await getProxySettings();
|
|
const isSuccess = settings.value.mode === config.proxyType;
|
|
|
|
if (isSuccess) {
|
|
await proxyStore.setCurrentProxy({
|
|
...config,
|
|
timestamp: Date.now()
|
|
});
|
|
|
|
const configs = await proxyStore.getProxyConfigs();
|
|
const updatedConfigs = configs.map(c => ({
|
|
...c,
|
|
enabled: c.id === config.id
|
|
}));
|
|
await proxyStore.saveProxyConfigs(updatedConfigs);
|
|
|
|
console.log('Proxy successfully set:', settings.value);
|
|
sendResponse({success: true});
|
|
|
|
// 通知所有 content scripts 更新
|
|
await notifyProxyStatusChanged();
|
|
} else {
|
|
console.error('Proxy settings verification failed');
|
|
sendResponse({
|
|
success: false,
|
|
error: '代理设置验证失败'
|
|
});
|
|
}
|
|
return;
|
|
} else {
|
|
sendResponse({
|
|
success: false,
|
|
error: '不支持的代理类型'
|
|
});
|
|
return;
|
|
}
|
|
} catch (error) {
|
|
console.error('Error setting proxy:', error);
|
|
sendResponse({
|
|
success: false,
|
|
error: error.message || '设置代理时发生错误'
|
|
});
|
|
}
|
|
}
|
|
|
|
async function handleClearProxyConfig(sendResponse) {
|
|
try {
|
|
await chrome.proxy.settings.clear({
|
|
scope: 'regular'
|
|
});
|
|
// 获取所有配置并禁用
|
|
const configs = await proxyStore.getProxyConfigs();
|
|
const updatedConfigs = configs.map(config => ({
|
|
...config,
|
|
enabled: false
|
|
}));
|
|
await proxyStore.saveProxyConfigs(updatedConfigs);
|
|
|
|
sendResponse({ success: true });
|
|
|
|
// 通知所有 content scripts 更新
|
|
await notifyProxyStatusChanged();
|
|
} catch (error) {
|
|
console.error('Error clearing proxy config:', error);
|
|
sendResponse({ success: false, error: error.message });
|
|
}
|
|
}
|
|
|
|
async function handleGetProxyStatus(sendResponse) {
|
|
try {
|
|
const settings = await getProxySettings();
|
|
const currentProxy = await proxyStore.getCurrentProxy();
|
|
|
|
const status = {
|
|
enabled: settings.value.mode === "fixed_servers",
|
|
config: currentProxy || null,
|
|
mode: settings.value.mode
|
|
};
|
|
|
|
console.log('Current proxy status:', status);
|
|
sendResponse({
|
|
success: true,
|
|
data: status
|
|
});
|
|
} catch (error) {
|
|
console.error('Error getting proxy status:', error);
|
|
sendResponse({
|
|
success: false,
|
|
error: error.message || '获取代理状态时发生错误'
|
|
});
|
|
}
|
|
}
|
|
|
|
// 添加代理请求监听器
|
|
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) {
|
|
try {
|
|
// 检查代理状态
|
|
const settings = await getProxySettings();
|
|
if (settings.value.mode !== "fixed_servers") {
|
|
return;
|
|
}
|
|
|
|
// 获取当前代理配置
|
|
const currentProxy = await proxyStore.getCurrentProxy();
|
|
if (!currentProxy) {
|
|
return;
|
|
}
|
|
|
|
// 记录日志
|
|
await proxyLogs.logRequest(details, currentProxy, error);
|
|
} catch (error) {
|
|
console.error('Error in queueProxyLog:', error);
|
|
}
|
|
}
|
|
|
|
// 添加检查和设置初始代理的函数
|
|
async function checkAndSetInitialProxy() {
|
|
try {
|
|
// 确保默认配置存在
|
|
await ProxySettings.setDefaultConfigs();
|
|
|
|
// 获取当前的代理设置
|
|
const settings = await getProxySettings();
|
|
console.log('Current proxy settings:', settings);
|
|
|
|
// 检查是否存在固定代理服务器设置
|
|
if (settings.value.mode === "fixed_servers" &&
|
|
settings.value.rules &&
|
|
settings.value.rules.singleProxy) {
|
|
|
|
const proxy = settings.value.rules.singleProxy;
|
|
|
|
// 获取现有配置
|
|
const existingConfigs = await proxyStore.getProxyConfigs();
|
|
|
|
// 检查是否已存在相同的 MITM 配置
|
|
const existingMitm = existingConfigs.find(config =>
|
|
config.host === proxy.host &&
|
|
config.port === proxy.port &&
|
|
config.scheme === proxy.scheme
|
|
);
|
|
|
|
if (!existingMitm) {
|
|
// 创建新的 MITM 配置
|
|
const newConfig = {
|
|
id: Date.now().toString(),
|
|
name: "Yakit MITM",
|
|
proxyType: 'fixed_servers',
|
|
scheme: proxy.scheme || 'http',
|
|
host: proxy.host,
|
|
port: proxy.port,
|
|
enabled: true,
|
|
// https://bugs.chromium.org/p/chromium/issues/detail?id=899126#c17
|
|
bypassList: ["<-loopback>"],
|
|
matchList: []
|
|
};
|
|
|
|
// 添加到现有配置中
|
|
const updatedConfigs = [...existingConfigs, newConfig];
|
|
await proxyStore.saveProxyConfigs(updatedConfigs);
|
|
|
|
// 启用新配置
|
|
await handleSetProxyConfig(newConfig, () => {});
|
|
|
|
console.log('Added and enabled Yakit MITM config from existing proxy settings');
|
|
return;
|
|
}
|
|
}
|
|
|
|
// 如果没有检测到代理或已存在配置，则设置为系统代理
|
|
await handleSetProxyConfig({
|
|
id: 'system',
|
|
name: '[系统代理]',
|
|
proxyType: 'system',
|
|
enabled: true
|
|
}, () => {});
|
|
|
|
// 设置认证监听
|
|
await ProxyAuth.setupAuthListener();
|
|
|
|
} catch (error) {
|
|
console.error('Error during initialization:', error);
|
|
}
|
|
}
|
|
|
|
// 修改 setupProxyHandlers 函数
|
|
export function setupProxyHandlers() {
|
|
// 设置代理错误处理和认证
|
|
ProxyAuth.setupErrorHandler();
|
|
ProxyAuth.setupAuthListener();
|
|
|
|
// 设置代理请求监听器
|
|
setupProxyRequestListener();
|
|
|
|
// 消息监听器
|
|
chrome.runtime.onMessage.addListener((msg, sender, sendResponse) => {
|
|
console.log("Proxy message:", msg);
|
|
|
|
switch (msg.action) {
|
|
case ProxyActionType.SET_PROXY_CONFIG:
|
|
handleSetProxyConfig(msg.config, sendResponse);
|
|
return true;
|
|
|
|
case ProxyActionType.CLEAR_PROXY_CONFIG:
|
|
(async () => {
|
|
await handleClearProxyConfig(sendResponse);
|
|
})();
|
|
return true;
|
|
|
|
case ProxyActionType.GET_PROXY_STATUS:
|
|
handleGetProxyStatus(sendResponse);
|
|
return true;
|
|
|
|
case ProxyActionType.GET_PROXY_LOGS:
|
|
proxyLogs.getLogs().then(logs => {
|
|
sendResponse({
|
|
success: true,
|
|
data: logs
|
|
});
|
|
}).catch(error => {
|
|
sendResponse({
|
|
success: false,
|
|
error: error.message
|
|
});
|
|
});
|
|
return true;
|
|
|
|
case ProxyActionType.CLEAR_PROXY_LOGS:
|
|
proxyLogs.clearLogs().then(() => {
|
|
sendResponse({success: true});
|
|
}).catch(error => {
|
|
sendResponse({
|
|
success: false,
|
|
error: error.message
|
|
});
|
|
});
|
|
return true;
|
|
|
|
case ProxyActionType.GET_PROXY_CONFIGS:
|
|
(async () => {
|
|
try {
|
|
const configs = await proxyStore.getProxyConfigs();
|
|
sendResponse({ success: true, data: configs });
|
|
} catch (error) {
|
|
console.error('Error getting proxy configs:', error);
|
|
sendResponse({ success: false, error: error.message });
|
|
}
|
|
})();
|
|
return true;
|
|
|
|
case ProxyActionType.ADD_PROXY_CONFIG:
|
|
proxyStore.getProxyConfigs().then(async configs => {
|
|
const newConfigs = [...configs, msg.config];
|
|
ProxyActionType
|
|
proxyStore.saveProxyConfigs(newConfigs);
|
|
sendResponse({success: true});
|
|
}).catch(error => {
|
|
sendResponse({
|
|
success: false,
|
|
error: error.message
|
|
});
|
|
});
|
|
return true;
|
|
|
|
case ProxyActionType.UPDATE_PROXY_CONFIG:
|
|
(async () => {
|
|
try {
|
|
if (!msg.configs || !Array.isArray(msg.configs)) {
|
|
throw new Error('无效的配置数据');
|
|
}
|
|
|
|
console.log('Updating proxy configs:', msg.configs);
|
|
await proxyStore.saveProxyConfigs(msg.configs);
|
|
|
|
// 获取最新的配置
|
|
const updatedConfigs = await proxyStore.getProxyConfigs();
|
|
console.log('Configs updated successfully:', updatedConfigs);
|
|
|
|
// 发送响应
|
|
sendResponse({
|
|
success: true,
|
|
data: updatedConfigs
|
|
});
|
|
|
|
// 通知所有 content scripts 更新
|
|
await notifyProxyStatusChanged();
|
|
} catch (error) {
|
|
console.error('Error updating proxy configs:', error);
|
|
sendResponse({
|
|
success: false,
|
|
error: error.message || '更新代理配置失败'
|
|
});
|
|
}
|
|
})();
|
|
return true;
|
|
|
|
case ProxyActionType.OPEN_OPTIONS_PAGE:
|
|
// 打开选项页
|
|
chrome.tabs.create({
|
|
url: chrome.runtime.getURL('/proxy/options.html')
|
|
}).then(tab => {
|
|
if (msg.triggerAdd) {
|
|
// 如果需要触发添加代理，等待页面加载完成
|
|
const listener = (tabId, changeInfo) => {
|
|
if (tabId === tab.id && changeInfo.status === 'complete') {
|
|
chrome.tabs.onUpdated.removeListener(listener);
|
|
// 向选项页发送消息触发添加代理
|
|
chrome.tabs.sendMessage(tab.id, {
|
|
action: 'TRIGGER_ADD_PROXY'
|
|
});
|
|
}
|
|
};
|
|
chrome.tabs.onUpdated.addListener(listener);
|
|
}
|
|
});
|
|
sendResponse({ success: true });
|
|
return true;
|
|
}
|
|
});
|
|
|
|
// 在扩展启动时初始化
|
|
chrome.runtime.onInstalled.addListener(async () => {
|
|
await checkAndSetInitialProxy();
|
|
});
|
|
|
|
// 浏览器启动时初始化
|
|
chrome.runtime.onStartup.addListener(async () => {
|
|
await checkAndSetInitialProxy();
|
|
});
|
|
}
|
|
|
|
// 当代理状态改变时通知所有内容脚本
|
|
async function notifyProxyStatusChanged() {
|
|
const tabs = await chrome.tabs.query({});
|
|
for (const tab of tabs) {
|
|
try {
|
|
chrome.tabs.sendMessage(tab.id, { action: 'PROXY_STATUS_CHANGED' });
|
|
} catch (error) {
|
|
// 忽略不支持的标签页
|
|
}
|
|
}
|
|
}
|
|
|
|
async function setProxyConfig(config) {
|
|
try {
|
|
let chromeProxyConfig;
|
|
|
|
if (config.proxyType === 'pac_script') {
|
|
chromeProxyConfig = {
|
|
mode: "pac_script",
|
|
pacScript: config.pacScript
|
|
};
|
|
} else if (config.proxyType === 'fixed_servers') {
|
|
chromeProxyConfig = {
|
|
mode: "fixed_servers",
|
|
rules: {
|
|
singleProxy: {
|
|
scheme: config.scheme,
|
|
host: config.host,
|
|
port: config.port
|
|
},
|
|
bypassList: config.bypassList || []
|
|
}
|
|
};
|
|
} else {
|
|
chromeProxyConfig = {
|
|
mode: config.proxyType // direct, system, auto_detect
|
|
};
|
|
}
|
|
|
|
await chrome.proxy.settings.set({
|
|
value: chromeProxyConfig,
|
|
scope: 'regular'
|
|
});
|
|
|
|
return { success: true };
|
|
} catch (error) {
|
|
console.error('Failed to set proxy config:', error);
|
|
return { success: false, error: error.message };
|
|
}
|
|
} |