feat: 优化主入口代码 添加歌曲下载功能

This commit is contained in:
alger
2025-01-02 00:14:05 +08:00
parent 38a9d6ed31
commit 018218a5bf
12 changed files with 441 additions and 172 deletions
+42
View File
@@ -0,0 +1,42 @@
import { app, ipcMain } from 'electron';
import Store from 'electron-store';
import set from '../set.json';
interface StoreType {
set: {
isProxy: boolean;
noAnimate: boolean;
animationSpeed: number;
author: string;
authorUrl: string;
musicApiPort: number;
};
}
let store: Store<StoreType>;
/**
* 初始化配置管理
*/
export function initializeConfig() {
store = new Store<StoreType>({
name: 'config',
defaults: {
set: set
}
});
store.get('set.downloadPath') || store.set('set.downloadPath', app.getPath('downloads'));
// 定义ipcRenderer监听事件
ipcMain.on('set-store-value', (_, key, value) => {
store.set(key, value);
});
ipcMain.on('get-store-value', (_, key) => {
const value = store.get(key);
_.returnValue = value || '';
});
return store;
}
+69
View File
@@ -0,0 +1,69 @@
import { app, dialog, shell, ipcMain } from 'electron';
import Store from 'electron-store';
import * as fs from 'fs';
import * as path from 'path';
import axios from 'axios';
/**
* 初始化文件管理相关的IPC监听
*/
export function initializeFileManager() {
// 通用的选择目录处理
ipcMain.handle('select-directory', async () => {
const result = await dialog.showOpenDialog({
properties: ['openDirectory'],
title: '选择目录'
});
return result;
});
// 通用的打开目录处理
ipcMain.on('open-directory', (_, path) => {
shell.openPath(path);
});
// 下载音乐处理
ipcMain.on('download-music', downloadMusic);
}
/**
* 下载音乐功能
*/
async function downloadMusic(event: Electron.IpcMainEvent, { url, filename }: { url: string; filename: string }) {
try {
const store = new Store();
const downloadPath = store.get('set.downloadPath') as string || app.getPath('downloads');
// 直接使用配置的下载路径
const filePath = path.join(downloadPath, `${filename}.mp3`);
// 检查文件是否已存在,如果存在则添加序号
let finalFilePath = filePath;
let counter = 1;
while (fs.existsSync(finalFilePath)) {
const ext = path.extname(filePath);
const nameWithoutExt = filePath.slice(0, -ext.length);
finalFilePath = `${nameWithoutExt} (${counter})${ext}`;
counter++;
}
const response = await axios({
url,
method: 'GET',
responseType: 'stream'
});
const writer = fs.createWriteStream(finalFilePath);
response.data.pipe(writer);
writer.on('finish', () => {
event.reply('music-download-complete', { success: true, path: finalFilePath });
});
writer.on('error', (err) => {
event.reply('music-download-complete', { success: false, error: err.message });
});
} catch (error: any) {
event.reply('music-download-complete', { success: false, error: error.message });
}
}
+43
View File
@@ -0,0 +1,43 @@
import { app, Menu, nativeImage, Tray, BrowserWindow } from 'electron';
import { join } from 'path';
let tray: Tray | null = null;
/**
* 初始化系统托盘
*/
export function initializeTray(iconPath: string, mainWindow: BrowserWindow) {
const trayIcon = nativeImage.createFromPath(join(iconPath, 'icon_16x16.png')).resize({ width: 16, height: 16 });
tray = new Tray(trayIcon);
// 创建一个上下文菜单
const contextMenu = Menu.buildFromTemplate([
{
label: '显示',
click: () => {
mainWindow.show();
},
},
{
label: '退出',
click: () => {
mainWindow.destroy();
app.quit();
},
},
]);
// 设置系统托盘图标的上下文菜单
tray.setContextMenu(contextMenu);
// 当系统托盘图标被点击时,切换窗口的显示/隐藏
tray.on('click', () => {
if (mainWindow.isVisible()) {
mainWindow.hide();
} else {
mainWindow.show();
}
});
return tray;
}
+81
View File
@@ -0,0 +1,81 @@
import { BrowserWindow, shell, ipcMain } from 'electron';
import { is } from '@electron-toolkit/utils';
import { join } from 'path';
/**
* 初始化窗口管理相关的IPC监听
*/
export function initializeWindowManager(mainWindow: BrowserWindow) {
ipcMain.on('minimize-window', (event) => {
const win = BrowserWindow.fromWebContents(event.sender);
if (win) {
win.minimize();
}
});
ipcMain.on('maximize-window', (event) => {
const win = BrowserWindow.fromWebContents(event.sender);
if (win) {
if (win.isMaximized()) {
win.unmaximize();
} else {
win.maximize();
}
}
});
ipcMain.on('close-window', (event) => {
const win = BrowserWindow.fromWebContents(event.sender);
if (win) {
win.destroy();
}
});
ipcMain.on('mini-tray', (event) => {
const win = BrowserWindow.fromWebContents(event.sender);
if (win) {
win.hide();
}
});
}
/**
* 创建主窗口
*/
export function createMainWindow(icon: Electron.NativeImage): BrowserWindow {
const mainWindow = new BrowserWindow({
width: 1200,
height: 780,
show: false,
frame: false,
autoHideMenuBar: true,
icon,
webPreferences: {
preload: join(__dirname, '../preload/index.js'),
sandbox: false,
contextIsolation: true
}
});
mainWindow.setMinimumSize(1200, 780);
mainWindow.on('ready-to-show', () => {
mainWindow.show();
});
mainWindow.webContents.setWindowOpenHandler((details) => {
shell.openExternal(details.url);
return { action: 'deny' };
});
// HMR for renderer base on electron-vite cli.
// Load the remote URL for development or the local html file for production.
if (is.dev && process.env.ELECTRON_RENDERER_URL) {
mainWindow.webContents.openDevTools({ mode: 'detach' });
mainWindow.loadURL(process.env.ELECTRON_RENDERER_URL);
} else {
mainWindow.loadFile(join(__dirname, '../renderer/index.html'));
}
return mainWindow;
}