增加系统托盘

This commit is contained in:
刘嘉伟 2023-12-01 15:00:11 +08:00
parent c9acbfc5f5
commit b957414e48

View File

@ -1,11 +1,11 @@
import { app, BrowserWindow, ipcMain, shell } from "electron"; import {app, BrowserWindow, ipcMain, Menu, MenuItem, MenuItemConstructorOptions, shell, Tray} from "electron";
import { release } from "node:os"; import {release} from "node:os";
import { join } from "node:path"; import node_path, {join} from "node:path";
import { initGitHubApi } from "../api/github"; import {initGitHubApi} from "../api/github";
import { initConfigApi } from "../api/config"; import {initConfigApi} from "../api/config";
import { initProxyApi } from "../api/proxy"; import {initProxyApi} from "../api/proxy";
import { initFrpcApi } from "../api/frpc"; import {initFrpcApi} from "../api/frpc";
import { initLoggerApi } from "../api/logger"; import {initLoggerApi} from "../api/logger";
import {initFileApi} from "../api/file"; import {initFileApi} from "../api/file";
// The built directory structure // The built directory structure
// //
@ -20,8 +20,8 @@ import {initFileApi} from "../api/file";
process.env.DIST_ELECTRON = join(__dirname, ".."); process.env.DIST_ELECTRON = join(__dirname, "..");
process.env.DIST = join(process.env.DIST_ELECTRON, "../dist"); process.env.DIST = join(process.env.DIST_ELECTRON, "../dist");
process.env.VITE_PUBLIC = process.env.VITE_DEV_SERVER_URL process.env.VITE_PUBLIC = process.env.VITE_DEV_SERVER_URL
? join(process.env.DIST_ELECTRON, "../public") ? join(process.env.DIST_ELECTRON, "../public")
: process.env.DIST; : process.env.DIST;
// Disable GPU Acceleration for Windows 7 // Disable GPU Acceleration for Windows 7
if (release().startsWith("6.1")) app.disableHardwareAcceleration(); if (release().startsWith("6.1")) app.disableHardwareAcceleration();
@ -30,8 +30,8 @@ if (release().startsWith("6.1")) app.disableHardwareAcceleration();
if (process.platform === "win32") app.setAppUserModelId(app.getName()); if (process.platform === "win32") app.setAppUserModelId(app.getName());
if (!app.requestSingleInstanceLock()) { if (!app.requestSingleInstanceLock()) {
app.quit(); app.quit();
process.exit(0); process.exit(0);
} }
// Remove electron security warnings // Remove electron security warnings
@ -40,98 +40,148 @@ if (!app.requestSingleInstanceLock()) {
// process.env['ELECTRON_DISABLE_SECURITY_WARNINGS'] = 'true' // process.env['ELECTRON_DISABLE_SECURITY_WARNINGS'] = 'true'
let win: BrowserWindow | null = null; let win: BrowserWindow | null = null;
let tray = null;
// Here, you can also use other preload // Here, you can also use other preload
const preload = join(__dirname, "../preload/index.js"); const preload = join(__dirname, "../preload/index.js");
const url = process.env.VITE_DEV_SERVER_URL; const url = process.env.VITE_DEV_SERVER_URL;
const indexHtml = join(process.env.DIST, "index.html"); const indexHtml = join(process.env.DIST, "index.html");
let isQuiting;
async function createWindow() { async function createWindow() {
win = new BrowserWindow({ win = new BrowserWindow({
title: "Main window", title: "Frpc Desktop",
icon: join(process.env.VITE_PUBLIC, "favicon.ico"), icon: join(process.env.VITE_PUBLIC, "logo/16x16.png"),
webPreferences: { webPreferences: {
preload, preload,
// Warning: Enable nodeIntegration and disable contextIsolation is not secure in production // Warning: Enable nodeIntegration and disable contextIsolation is not secure in production
// Consider using contextBridge.exposeInMainWorld // Consider using contextBridge.exposeInMainWorld
// Read more on https://www.electronjs.org/docs/latest/tutorial/context-isolation // Read more on https://www.electronjs.org/docs/latest/tutorial/context-isolation
nodeIntegration: true, nodeIntegration: true,
contextIsolation: false contextIsolation: false
}
});
if (process.env.VITE_DEV_SERVER_URL) {
// electron-vite-vue#298
win.loadURL(url);
// Open devTool if the app is not packaged
win.webContents.openDevTools();
} else {
win.loadFile(indexHtml);
} }
});
if (process.env.VITE_DEV_SERVER_URL) { // Test actively push message to the Electron-Renderer
// electron-vite-vue#298 win.webContents.on("did-finish-load", () => {
win.loadURL(url); win?.webContents.send("main-process-message", new Date().toLocaleString());
// Open devTool if the app is not packaged });
win.webContents.openDevTools();
} else {
win.loadFile(indexHtml);
}
// Test actively push message to the Electron-Renderer // Make all links open with the browser, not with the application
win.webContents.on("did-finish-load", () => { win.webContents.setWindowOpenHandler(({url}) => {
win?.webContents.send("main-process-message", new Date().toLocaleString()); if (url.startsWith("https:")) shell.openExternal(url);
}); return {action: "deny"};
});
// Make all links open with the browser, not with the application // 隐藏菜单栏
win.webContents.setWindowOpenHandler(({ url }) => { const {Menu} = require("electron");
if (url.startsWith("https:")) shell.openExternal(url); Menu.setApplicationMenu(null);
return { action: "deny" }; // hide menu for Mac
}); if (process.platform !== "darwin") {
app.dock.hide();
}
win.on('minimize', function (event) {
event.preventDefault();
win.hide();
});
win.on('close', function (event) {
if (!isQuiting) {
event.preventDefault();
win.hide();
if (process.platform === "darwin") {
app.dock.hide();
}
}
return false;
});
// 隐藏菜单栏
const { Menu } = require("electron");
Menu.setApplicationMenu(null);
// hide menu for Mac
if (process.platform !== "darwin") {
app.dock.hide();
}
// win.webContents.on('will-navigate', (event, url) => { }) #344
} }
app.whenReady().then(createWindow); export const createTray = () => {
let menu: Array<(MenuItemConstructorOptions) | (MenuItem)> = [
{
label: '显示主窗口', click: function () {
win.show();
if (process.platform === "darwin") {
app.dock.show();
}
}
},
{
label: '退出',
click: () => {
isQuiting = true;
app.quit();
}
}
];
tray = new Tray(node_path.join(process.env.VITE_PUBLIC, "logo/16x16.png"))
tray.setToolTip('Frpc Desktop')
const contextMenu = Menu.buildFromTemplate(menu)
tray.setContextMenu(contextMenu)
}
app.whenReady().then(() => {
createWindow().then(r => {
createTray()
})
});
app.on("window-all-closed", () => { app.on("window-all-closed", () => {
win = null; win = null;
if (process.platform !== "darwin") app.quit(); if (process.platform !== "darwin") app.quit();
}); });
app.on("second-instance", () => { app.on("second-instance", () => {
if (win) { if (win) {
// Focus on the main window if the user tried to open another // Focus on the main window if the user tried to open another
if (win.isMinimized()) win.restore(); if (win.isMinimized()) win.restore();
win.focus(); win.focus();
} }
}); });
app.on("activate", () => { app.on("activate", () => {
const allWindows = BrowserWindow.getAllWindows(); const allWindows = BrowserWindow.getAllWindows();
if (allWindows.length) { if (allWindows.length) {
allWindows[0].focus(); allWindows[0].focus();
} else { } else {
createWindow(); createWindow();
} }
}); });
app.on('before-quit', () => {
isQuiting = true;
})
// New window example arg: new windows url // New window example arg: new windows url
ipcMain.handle("open-win", (_, arg) => { ipcMain.handle("open-win", (_, arg) => {
const childWindow = new BrowserWindow({ const childWindow = new BrowserWindow({
webPreferences: { webPreferences: {
preload, preload,
nodeIntegration: true, nodeIntegration: true,
contextIsolation: false contextIsolation: false
} }
}); });
if (process.env.VITE_DEV_SERVER_URL) { if (process.env.VITE_DEV_SERVER_URL) {
childWindow.loadURL(`${url}#${arg}`); childWindow.loadURL(`${url}#${arg}`);
} else { } else {
childWindow.loadFile(indexHtml, { hash: arg }); childWindow.loadFile(indexHtml, {hash: arg});
} }
}); });
ipcMain.on('open-url', (event, url) => { ipcMain.on('open-url', (event, url) => {
shell.openExternal(url).then(r => {}); shell.openExternal(url).then(r => {
});
}); });
initGitHubApi(); initGitHubApi();