res-downloader/electron/main/ipc.ts

import {ipcMain, dialog, BrowserWindow, app, shell} from 'electron'
import {startServer} from './proxyServer'
import {installCert, checkCertInstalled} from './cert'
import {decodeWxFile, typeSuffix, getCurrentDateTimeFormatted} from './utils'
// @ts-ignore
import {hexMD5} from '../../src/common/md5'
import {Aria2RPC} from './aria2Rpc'
import fs from "fs"
import urlTool from "url";
import {closeProxy, setProxy} from "./setProxy";
import path from 'path'

let win: BrowserWindow
let previewWin: BrowserWindow

const aria2RpcClient = new Aria2RPC()

export default function initIPC() {
    ipcMain.handle('invoke_app_is_init', async (event, arg) => {
        // 初始化应用 安装证书相关
        return checkCertInstalled()
    })

    ipcMain.handle('invoke_init_app', (event, arg) => {
        // 开始 初始化应用 安装证书相关
        installCert(false)
    })

    ipcMain.handle('invoke_set_config', (event, data) => {
        const filePath = path.join(app.getPath('userData'), 'resd_config.json');
        fs.writeFile(filePath, JSON.stringify(data), ()=>{})
        global.resdConfig = Object.assign({}, global.resdConfig, data)
        global.resdConfig.port = parseInt(global.resdConfig.port)
        return true
    })

    ipcMain.handle('invoke_set_proxy', async (event, arg) => {
        // 启动代理服务
        if (!global.isStartProxy) {
            dialog.showMessageBoxSync({
                type: "error",
                message: "代理未启动",
            });
            return false
        }
        try {
            if (arg.proxy) {
                await setProxy('127.0.0.1', global.resdConfig.port)
            }else{
                await closeProxy('127.0.0.1', global.resdConfig.port)
            }
            return true
        } catch (err) {
            console.error(err);
            dialog.showMessageBoxSync({
                type: "error",
                message: err.toString(),
            });
            return false
        }
        // let upstream_proxy = ""
        // if (arg.upstream_proxy && !arg.upstream_proxy.includes(':8899')) {
        //     upstream_proxy = arg.upstream_proxy
        // }
        //
        // global.isStartProxy = true
        // return startServer({
        //     win: win,
        //     upstreamProxy: upstream_proxy,
        //     setProxyErrorCallback: err => {
        //         console.log('setProxyErrorCallback', err)
        //     },
        // })
    })

    ipcMain.handle('invoke_select_down_dir', async (event, arg) => {
        // 选择下载位置
        const result = dialog.showOpenDialogSync({title: '保存', properties: ['openDirectory']})
        if (!result?.[0]) {
            return false
        }

        return result?.[0]
    })

    ipcMain.handle('invoke_select_wx_file', async (event, {index, data}) => {
        // 选择下载位置
        const result = dialog.showOpenDialogSync({title: '保存', properties: ['openFile']})
        if (!result?.[0]) {
            return false
        }
        return decodeWxFile(result?.[0], data.decode_key, result?.[0].replace(".mp4", "_解密.mp4"))
    })

    ipcMain.handle('invoke_file_exists', async (event, {save_path, url, description}) => {
        let fileName = description ? description.replace(/[^a-zA-Z\u4e00-\u9fa5]/g, '') : hexMD5(url);
        let res = fs.existsSync(`${save_path}/${fileName}.mp4`)
        return {is_file: res, fileName: `${save_path}/${fileName}.mp4`}
    })

    ipcMain.handle('invoke_down_file', async (event, {data, save_path, quality}) => {
        let down_url = data.url
        if (!down_url) {
            return new Promise((resolve, reject) => {
                resolve(false);
            });
        }
        if (quality === "0" && data.decode_key) {
            const urlInfo = urlTool.parse(down_url, true);
            console.log('urlInfo', urlInfo)
            if (urlInfo.query["token"] && urlInfo.query["encfilekey"]) {
                down_url = urlInfo.protocol + "//" + urlInfo.hostname + urlInfo.pathname.replace("251/20302", "251/20304") +
                    "?encfilekey=" + urlInfo.query["encfilekey"] +
                    "&token=" + urlInfo.query["token"]
                console.log("down_url:", down_url)
            }
        } else if (quality !== "-1" && data.decode_key && data.file_format) {
            const format = data.file_format.split('#');
            const qualityMap = [
                format[0],
                format[Math.floor(format.length / 2)],
                format[format.length - 1]
            ];
            down_url += "&X-snsvideoflag=" + qualityMap[quality];
        }
        let fileName = data?.description ? data.description.replace(/[^a-zA-Z0-9\u4e00-\u9fa5]/g, '') : hexMD5(down_url);
        fileName = fileName + "_" + getCurrentDateTimeFormatted() + typeSuffix(data.type)[1]
        let save_path_file = `${save_path}/${fileName}`
        if (process.platform === 'win32') {
            save_path_file = `${save_path}\\${fileName}`
        }

        let headers = {
            'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/100.0.4896.127 Safari/537.36',
        }

        return new Promise((resolve, reject) => {

            if (data?.referer) {
                headers['Referer'] = data?.referer
            }

            aria2RpcClient.addUri([down_url], save_path, fileName, headers).then((response) => {
                let currentGid = response.result
                let progressIntervalId = null
                // // 开始定时查询下载进度
                progressIntervalId = setInterval(() => {
                    aria2RpcClient.tellStatus(currentGid).then((status) => {
                        if (status.result.status !== "complete") {
                            const progress = aria2RpcClient.calculateDownloadProgress(status.result.bitfield)
                            win?.webContents.send('on_down_file_schedule', {schedule: `已下载${progress}%`})
                        } else {
                            clearInterval(progressIntervalId);
                            if (data.decode_key) {
                                win?.webContents.send('on_down_file_schedule', {schedule: `开始解密`})
                                decodeWxFile(save_path_file, data.decode_key, save_path_file.replace(".mp4", "_wx.mp4")).then((res) => {
                                    fs.unlink(save_path_file, (err) => {
                                    })
                                    resolve(res)
                                }).catch((error) => {
                                    console.log("err:", error)
                                    resolve(false);
                                })
                            } else {
                                resolve({
                                    fullFileName: save_path_file,
                                })
                            }
                        }
                    }).catch((error) => {
                        console.error(error)
                        clearInterval(progressIntervalId)
                        resolve(false)
                    });
                }, 1000)
            }).catch((error) => {
                console.log("err:", error)
                resolve(false)
            });
        });
    });

    ipcMain.handle('invoke_resources_preview', async (event, {url}) => {
        if (!url) {
            return
        }

        previewWin.loadURL(url).then(r => {
            return
        }).catch(res => {
        })
        previewWin.show()
        return
    })

    ipcMain.handle('invoke_open_default_browser', (event, {url}) => {
        shell.openExternal(url).then(r => {
        })
    })

    ipcMain.handle('invoke_open_file_dir', (event, {save_path}) => {
        shell.showItemInFolder(save_path)
    })

    ipcMain.handle('invoke_file_del', (event, {url_sign}) => {
        if (url_sign === "all") {
            global.videoList = {}
            return
        }
        if (url_sign) {
            delete global.videoList[url_sign]
            return
        }
    })

    ipcMain.handle('invoke_window_restart', (event) => {
        app.relaunch()
        app.exit()
    })
}

export function setWin(w, p) {
    win = w
    previewWin = p
}