app.js

/*!
 * AriaNg GUI
 * 
 * Copyright (c) 2018-2020 Xmader
 * Released under the MIT license
 * 
 * Source Code: https://github.com/Xmader/aria-ng-gui
 * 
*/

// @ts-check

const os = require("os")
const path = require("path")
const fs = require("fs")
const { app, BrowserWindow, Menu, ipcMain, shell } = require("electron")

const edit_conf = require("./edit_conf.js")
const { buildMenu } = require("./menu.js")
const { displayTray, destroyTray } = require("./tray.js")
const { syncConfigFiles, confPath } = require("./conf.js")

/** @type {Electron.BrowserWindow} */
let mainWindow = null

const icon = path.join(__dirname, "assets", "AriaNg.png")
const trayIcon = path.join(__dirname, "assets", "tray-icon.png")

/**
 * @param {string} p 
 */
const findFirstSubdirName = (p) => {
    return fs.readdirSync(p, { withFileTypes: true })
        .find((x) => x.isDirectory())
        .name
}

app.commandLine.appendSwitch("ignore-certificate-errors") // 忽略证书相关错误, 适用于使用自签名证书将Aria2的RPC配置成HTTPS协议的情况

app.on("window-all-closed", () => {
    app.quit()
})

app.on("quit", () => {
    syncConfigFiles()
})

app.on("ready", () => {

    // 只允许运行单一实例 (进程)
    const gotTheLock = app.requestSingleInstanceLock()
    if (!gotTheLock) {
        return app.quit()
    }

    mainWindow = new BrowserWindow({
        title: "AriaNg",
        width: 1000,
        height: 600,
        minWidth: 400,
        minHeight: 400,
        icon,
        show: false,
        webPreferences: {
            nodeIntegration: false,
            preload: path.join(__dirname, "pre.js")
        }
    })

    const platform = os.platform()
    const arch = os.arch()

    /**
     * @param {NodeJS.Platform} _platform
     * @param {"arm"| "arm64"| "ia32"| "mips"| "mipsel"| "ppc"| "ppc64"| "s390"| "s390x"| "x32"| "x64" | string} _arch
     * @returns {string}
     */
    const getAria2cPath = (_platform, _arch) => {
        if (_arch == "x32") {
            _arch = "ia32"
        }

        const aria2Dir = path.join(__dirname, "aria2")
        const p = path.join(aria2Dir, _platform, _arch, aria2c_bin)

        if (!fs.existsSync(p)) {
            // find existed aria2c
            const _platform = findFirstSubdirName(aria2Dir)
            const platformDir = path.join(aria2Dir, _platform)
            const _arch = findFirstSubdirName(platformDir)
            return path.join(platformDir, _arch, aria2c_bin)
        } else {
            return p
        }
    }

    const aria2c_bin = (platform == "linux" || platform == "darwin") ? "aria2c" : "aria2c.exe"
    const aria2c_path = getAria2cPath(platform, arch)

    // 将配置文件保存在两处，方便升级和在电脑之间迁移，实际只使用保存在 userData (AppData) 下的配置文件
    // 同步 在程序目录 和 在 userData (AppData) 下的 配置文件
    syncConfigFiles()

    edit_conf(confPath) // 根据用户的操作系统动态编辑aria2的配置文件

    //打开主程序
    fs.chmodSync(aria2c_path, 0o777)

    /** @type {import("child_process").ChildProcessWithoutNullStreams} */
    let aria2c = null

    function runAria2() {
        killAria2()

        aria2c = require("child_process").spawn(aria2c_path, [`--conf-path=${confPath}`], {
            stdio: "pipe"
        })
        aria2c.stdout.pipe(process.stdout, { end: false })
        aria2c.stderr.pipe(process.stderr, { end: false })

        aria2c.on("error", runAria2)
        aria2c.on("exit", runAria2)
    }

    function killAria2() {
        if (aria2c) {
            aria2c.removeAllListeners("error")
            aria2c.removeAllListeners("exit")
            aria2c.kill("SIGINT")
            aria2c = null
        }
    }

    runAria2()

    // 打开窗口的调试工具
    //mainWindow.webContents.openDevTools()

    const locale = app.getLocale().includes("zh") ? "zh-CN" : "en-US"
    const { contextMenu, appMenu } = buildMenu(locale)

    if (platform == "darwin") {
        Menu.setApplicationMenu(appMenu)
    } else {
        Menu.setApplicationMenu(null)
        mainWindow.setMenu(null)
    }

    mainWindow.loadURL(`file://${__dirname}/pages/index.html`)

    mainWindow.once("ready-to-show", () => {
        mainWindow.show()
    })

    mainWindow.on("close", (e) => {
        e.preventDefault()
        mainWindow.hide()
        displayTray(icon, trayIcon)
    })

    const onClosed = () => {
        killAria2()
        mainWindow = null
    }

    mainWindow.on("closed", onClosed)
    process.on("SIGINT", onClosed)
    process.on("SIGTERM", onClosed)

    ipcMain.on("right_btn", () => {
        contextMenu.popup({ window: mainWindow })
    })
})

app.on("second-instance", () => {
    destroyTray()
    if (mainWindow) {
        mainWindow.focus()
        shell.beep()
    }
})

ipcMain.on("show_progress_bar", (event, n) => {
    if (mainWindow && mainWindow.setProgressBar) {
        mainWindow.setProgressBar(n ? n : -1)
    }
})