将ipcMain的所有handle移至IpcMainHandler.ts
This commit is contained in:
Allen
parent
aad96a0866
commit
12377a42ed
101
src/IpcMainHandler.ts
Normal file
101
src/IpcMainHandler.ts
Normal file
@ -0,0 +1,101 @@
|
|||||||
|
import { ipcMain, screen, session } from "electron";
|
||||||
|
import http from 'http';
|
||||||
|
import https from 'https';
|
||||||
|
import { URL } from 'url';
|
||||||
|
|
||||||
|
export function initialize(): void {
|
||||||
|
// Get primary display（需要注意：size 是逻辑像素尺寸（logical screen size），已被 DPI 缩放影响，需要除以 DPI 缩放比例才是物理像素尺寸（physical screen size））
|
||||||
|
ipcMain.handle('get-primary-display', () => screen.getPrimaryDisplay());
|
||||||
|
|
||||||
|
// Check if the URL is available
|
||||||
|
ipcMain.handle('check-url-available', async (event, rawUrl: string) => {
|
||||||
|
try {
|
||||||
|
const url: URL = new URL(rawUrl);
|
||||||
|
const lib = url.protocol === 'https:' ? https : http;
|
||||||
|
|
||||||
|
return await new Promise((resolve) => {
|
||||||
|
// 先用HEAD请求，如果遇到403，则再用GET请求再试一次（部分服务器可能禁止HEAD请求）。
|
||||||
|
const requestOptions: http.RequestOptions | https.RequestOptions = {
|
||||||
|
hostname: url.hostname,
|
||||||
|
port: url.port || undefined,
|
||||||
|
path: url.pathname + url.search,
|
||||||
|
headers: {
|
||||||
|
'Accept': '*/*',
|
||||||
|
'Accept-Encoding': 'gzip, deflate, br',
|
||||||
|
'Connection': 'keep-alive',
|
||||||
|
'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/120.0.0.0 Safari/537.36'
|
||||||
|
}
|
||||||
|
};
|
||||||
|
const headReq = lib.request(
|
||||||
|
{
|
||||||
|
method: 'HEAD',
|
||||||
|
timeout: 3000,
|
||||||
|
...requestOptions
|
||||||
|
},
|
||||||
|
(res) => {
|
||||||
|
console.log('check-url-available HEAD', url.toString(), res.statusCode, res.statusMessage);
|
||||||
|
if (res.statusCode === 403) {
|
||||||
|
headReq.destroy()
|
||||||
|
const getReq = lib.get({
|
||||||
|
method: 'GET',
|
||||||
|
...requestOptions
|
||||||
|
}, (res) => {
|
||||||
|
console.log('check-url-available GET', url.toString(), res.statusCode, res.statusMessage);
|
||||||
|
resolve({ ok: true, status: res.statusCode, message: res.statusMessage });
|
||||||
|
getReq.destroy()
|
||||||
|
});
|
||||||
|
getReq.on('error', (err) => {
|
||||||
|
resolve({ ok: false, status: -1, message: err.message });
|
||||||
|
});
|
||||||
|
getReq.on('timeout', () => {
|
||||||
|
resolve({ ok: false, status: -1, message: 'GET Timeout' });
|
||||||
|
getReq.destroy();
|
||||||
|
});
|
||||||
|
} else {
|
||||||
|
resolve({ ok: true, status: res.statusCode, message: res.statusMessage });
|
||||||
|
headReq.destroy();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
);
|
||||||
|
|
||||||
|
headReq.on('error', (err) => {
|
||||||
|
resolve({ ok: false, status: -1, message: err.message });
|
||||||
|
});
|
||||||
|
|
||||||
|
headReq.on('timeout', () => {
|
||||||
|
resolve({ ok: false, status: -1, message: 'HEAD Timeout' });
|
||||||
|
headReq.destroy();
|
||||||
|
});
|
||||||
|
|
||||||
|
headReq.end();
|
||||||
|
})
|
||||||
|
} catch (e) {
|
||||||
|
return { ok: false, status: -1, message: 'Invalid URL' };
|
||||||
|
}
|
||||||
|
});
|
||||||
|
|
||||||
|
// Set webview‘s cookie
|
||||||
|
ipcMain.handle('set-webview-cookie', async (event, url: string, cookie: string) => {
|
||||||
|
try {
|
||||||
|
const parsedUrl = new URL(url);
|
||||||
|
const cookies: Array<{ Key: string, Value: string }> = JSON.parse(cookie);
|
||||||
|
|
||||||
|
for (const cookieItem of cookies) {
|
||||||
|
await session.defaultSession.cookies.set({
|
||||||
|
url: url,
|
||||||
|
name: cookieItem.Key.trim(),
|
||||||
|
value: cookieItem.Value.trim(),
|
||||||
|
domain: parsedUrl.hostname,
|
||||||
|
path: '/',
|
||||||
|
secure: parsedUrl.protocol === 'https:',
|
||||||
|
httpOnly: true
|
||||||
|
});
|
||||||
|
}
|
||||||
|
return true;
|
||||||
|
} catch (error) {
|
||||||
|
console.error('设置cookie失败:', error);
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
101
src/main.ts
101
src/main.ts
@ -1,8 +1,7 @@
|
|||||||
import { app, BrowserWindow, globalShortcut, ipcMain, Menu, session, screen, Tray } from 'electron';
|
import { app, BrowserWindow, globalShortcut, Menu, Tray } from 'electron';
|
||||||
import path from 'node:path';
|
import path from 'node:path';
|
||||||
import started from 'electron-squirrel-startup';
|
import started from 'electron-squirrel-startup';
|
||||||
import http from 'http';
|
import { initialize } from './IpcMainHandler';
|
||||||
import https from 'https';
|
|
||||||
|
|
||||||
const isDevelopment: boolean = process.env.NODE_ENV === 'development';
|
const isDevelopment: boolean = process.env.NODE_ENV === 'development';
|
||||||
|
|
||||||
@ -27,100 +26,6 @@ if (started) {
|
|||||||
app.exit(0); // 使用 exit 而不是 quit，确保立即退出
|
app.exit(0); // 使用 exit 而不是 quit，确保立即退出
|
||||||
}
|
}
|
||||||
|
|
||||||
// Get primary display（需要注意：size 是逻辑像素尺寸（logical screen size），已被 DPI 缩放影响，需要除以 DPI 缩放比例才是物理像素尺寸（physical screen size））
|
|
||||||
ipcMain.handle('get-primary-display', () => screen.getPrimaryDisplay());
|
|
||||||
|
|
||||||
// Check if the URL is available
|
|
||||||
ipcMain.handle('check-url-available', async (event, rawUrl: string) => {
|
|
||||||
try {
|
|
||||||
const url: URL = new URL(rawUrl);
|
|
||||||
const lib = url.protocol === 'https:' ? https : http;
|
|
||||||
|
|
||||||
return await new Promise((resolve) => {
|
|
||||||
// 先用HEAD请求，如果遇到403，则再用GET请求再试一次（部分服务器可能禁止HEAD请求）。
|
|
||||||
const requestOptions: http.RequestOptions | https.RequestOptions = {
|
|
||||||
hostname: url.hostname,
|
|
||||||
port: url.port || undefined,
|
|
||||||
path: url.pathname + url.search,
|
|
||||||
headers: {
|
|
||||||
'Accept': '*/*',
|
|
||||||
'Accept-Encoding': 'gzip, deflate, br',
|
|
||||||
'Connection': 'keep-alive',
|
|
||||||
'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/120.0.0.0 Safari/537.36'
|
|
||||||
}
|
|
||||||
};
|
|
||||||
const headReq = lib.request(
|
|
||||||
{
|
|
||||||
method: 'HEAD',
|
|
||||||
timeout: 3000,
|
|
||||||
...requestOptions
|
|
||||||
},
|
|
||||||
(res) => {
|
|
||||||
console.log('check-url-available HEAD', url.toString(), res.statusCode, res.statusMessage);
|
|
||||||
if (res.statusCode === 403) {
|
|
||||||
headReq.destroy()
|
|
||||||
const getReq = lib.get({
|
|
||||||
method: 'GET',
|
|
||||||
...requestOptions
|
|
||||||
}, (res) => {
|
|
||||||
console.log('check-url-available GET', url.toString(), res.statusCode, res.statusMessage);
|
|
||||||
resolve({ ok: true, status: res.statusCode, message: res.statusMessage });
|
|
||||||
getReq.destroy()
|
|
||||||
});
|
|
||||||
getReq.on('error', (err) => {
|
|
||||||
resolve({ ok: false, status: -1, message: err.message });
|
|
||||||
});
|
|
||||||
getReq.on('timeout', () => {
|
|
||||||
resolve({ ok: false, status: -1, message: 'GET Timeout' });
|
|
||||||
getReq.destroy();
|
|
||||||
});
|
|
||||||
} else {
|
|
||||||
resolve({ ok: true, status: res.statusCode, message: res.statusMessage });
|
|
||||||
headReq.destroy();
|
|
||||||
}
|
|
||||||
}
|
|
||||||
);
|
|
||||||
|
|
||||||
headReq.on('error', (err) => {
|
|
||||||
resolve({ ok: false, status: -1, message: err.message });
|
|
||||||
});
|
|
||||||
|
|
||||||
headReq.on('timeout', () => {
|
|
||||||
resolve({ ok: false, status: -1, message: 'HEAD Timeout' });
|
|
||||||
headReq.destroy();
|
|
||||||
});
|
|
||||||
|
|
||||||
headReq.end();
|
|
||||||
})
|
|
||||||
} catch (e) {
|
|
||||||
return { ok: false, status: -1, message: 'Invalid URL' };
|
|
||||||
}
|
|
||||||
});
|
|
||||||
|
|
||||||
// Set webview‘s cookie
|
|
||||||
ipcMain.handle('set-webview-cookie', async (event, url: string, cookie: string) => {
|
|
||||||
try {
|
|
||||||
const parsedUrl = new URL(url);
|
|
||||||
const cookies: Array<{ Key: string, Value: string }> = JSON.parse(cookie);
|
|
||||||
|
|
||||||
for (const cookieItem of cookies) {
|
|
||||||
await session.defaultSession.cookies.set({
|
|
||||||
url: url,
|
|
||||||
name: cookieItem.Key.trim(),
|
|
||||||
value: cookieItem.Value.trim(),
|
|
||||||
domain: parsedUrl.hostname,
|
|
||||||
path: '/',
|
|
||||||
secure: parsedUrl.protocol === 'https:',
|
|
||||||
httpOnly: true
|
|
||||||
});
|
|
||||||
}
|
|
||||||
return true;
|
|
||||||
} catch (error) {
|
|
||||||
console.error('设置cookie失败:', error);
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
});
|
|
||||||
|
|
||||||
const createWindow = () => {
|
const createWindow = () => {
|
||||||
// Create the browser window.
|
// Create the browser window.
|
||||||
const win = new BrowserWindow({
|
const win = new BrowserWindow({
|
||||||
@ -257,3 +162,5 @@ app.whenReady().then(() => {
|
|||||||
}
|
}
|
||||||
});
|
});
|
||||||
});
|
});
|
||||||
|
|
||||||
|
initialize();
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user