102 lines
3.6 KiB
TypeScript
102 lines
3.6 KiB
TypeScript
|
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;
|
|||
|
|
}
|
|||
|
|
});
|
|||
|
|
}
|
|||
|
|
|