frpc-desktop/electron/api/frpc.ts

import {app, ipcMain, Notification} from "electron";
import {Config, getConfig} from "../storage/config";
import {listProxy} from "../storage/proxy";
import {getVersionById} from "../storage/version";
import treeKill from "tree-kill";

const fs = require("fs");
const path = require("path");
const {exec, spawn} = require("child_process");
export let frpcProcess = null;
const runningCmd = {
    commandPath: null,
    configPath: null
};
let frpcStatusListener = null;

/**
 * 获取选择版本的工作目录
 * @param versionId 版本ID
 * @param callback
 */
const getFrpcVersionWorkerPath = (
    versionId: string,
    callback: (workerPath: string) => void
) => {
    getVersionById(versionId, (err2, version) => {
        if (!err2) {
            callback(version["frpcVersionPath"]);
        }
    });
};

/**
 * 生成配置文件
 */
export const generateConfig = (
    config: Config,
    callback: (configPath: string) => void
) => {
    listProxy((err3, proxys) => {
        if (!err3) {
            const proxyToml = proxys.map(m => {
                let toml = `
[[proxies]]
name = "${m.name}"
type = "${m.type}"
localIP = "${m.localIp}"
localPort = ${m.localPort}
`;
                switch (m.type) {
                    case "tcp":
                        toml += `remotePort = ${m.remotePort}`;
                        break;
                    case "http":
                    case "https":
                        toml += `customDomains=[${m.customDomains.map(m => `"${m}"`)}]`;
                        break;
                    default:
                        break;
                }

                return toml;
            });
            let toml = `
serverAddr = "${config.serverAddr}"
serverPort = ${config.serverPort}
auth.method = "${config.authMethod}"
auth.token = "${config.authToken}"
log.to = "frpc.log"
log.level = "${config.logLevel}"
log.maxDays = ${config.logMaxDays}
webServer.addr = "127.0.0.1"
webServer.port = 57400
transport.tls.enable = ${config.tlsConfigEnable}
${config.tlsConfigEnable ? `
transport.tls.certFile = "${config.tlsConfigCertFile}"
transport.tls.keyFile = "${config.tlsConfigKeyFile}"
transport.tls.trustedCaFile = "${config.tlsConfigTrustedCaFile}"
transport.tls.serverName = "${config.tlsConfigServerName}"
` : ""}


${proxyToml.join("")}
      `;

            // const configPath = path.join("frp.toml");
            const filename = "frp.toml";
            const configPath = path.join(app.getPath("userData"), filename)
            console.debug("生成配置成功", configPath)
            fs.writeFile(
                configPath, // 配置文件目录
                toml, // 配置文件内容
                {flag: "w"},
                err => {
                    if (!err) {
                        callback(filename);
                    }
                }
            );
        }
    });
};

/**
 * 启动frpc子进程
 * @param cwd
 * @param commandPath
 * @param configPath
 */
const startFrpcProcess = (commandPath: string, configPath: string) => {
    const command = `${commandPath} -c ${configPath}`;
    console.info("启动", command)
    frpcProcess = spawn(command, {
        cwd: app.getPath("userData"),
        shell: true
    });
    runningCmd.commandPath = commandPath;
    runningCmd.configPath = configPath;
    frpcProcess.stdout.on("data", data => {
        console.debug(`命令输出: ${data}`);
    });
    frpcProcess.stdout.on("error", data => {
        console.log("启动错误", data)
        stopFrpcProcess()
    });
    frpcStatusListener = setInterval(() => {
        const status = frpcProcessStatus()
        if (!status) {
            console.log("连接已断开")
            new Notification({
                title: "Frpc Desktop",
                body: "连接已断开，请前往日志查看原因"
            }).show()
            clearInterval(frpcStatusListener)
        }
    }, 3000)
};

/**
 *  重载frpc配置
 */
export const reloadFrpcProcess = () => {
    if (frpcProcess && !frpcProcess.killed) {
        getConfig((err1, config) => {
            if (!err1) {
                if (config) {
                    generateConfig(config, configPath => {
                        const command = `${runningCmd.commandPath} reload -c ${configPath}`;
                        console.info("重启", command);
                        exec(command, {
                            cwd: app.getPath("userData"),
                            shell: true
                        });
                    });
                }
            }
        });
    }
};

/**
 * 停止frpc子进程
 */
export const stopFrpcProcess = (callback?:() => void) => {
    if (frpcProcess) {
        treeKill(frpcProcess.pid, (error: Error) => {
            if (error) {
                console.log("关闭失败", frpcProcess.pid, error)
            } else {
                console.log('关闭成功')
                frpcProcess = null
                clearInterval(frpcStatusListener)
            }
            callback()
        })
    }
}

/**
 * 获取frpc子进程状态
 */
export const frpcProcessStatus = () => {
    if (!frpcProcess) {
        return false;
    }
    try {
        // 发送信号给进程，如果进程存在，会正常返回
        process.kill(frpcProcess.pid, 0);
        return true;
    } catch (error) {
        // 进程不存在，抛出异常
        return false;
    }
}


export const initFrpcApi = () => {
    ipcMain.handle("frpc.running", async (event, args) => {
        return frpcProcessStatus()
    });

    ipcMain.on("frpc.start", async (event, args) => {
        getConfig((err1, config) => {
            if (!err1) {
                if (config) {
                    getFrpcVersionWorkerPath(
                        config.currentVersion,
                        (frpcVersionPath: string) => {
                            generateConfig(config, configPath => {
                                const platform = process.platform;
                                if (platform === 'win32') {
                                    startFrpcProcess(
                                        path.join(frpcVersionPath, "frpc.exe"),
                                        configPath
                                    );
                                } else {
                                    startFrpcProcess(
                                        path.join(frpcVersionPath, "frpc"),
                                        configPath
                                    );
                                }
                            });
                        }
                    );
                } else {
                    event.reply(
                        "Home.frpc.start.error.hook",
                        "请先前往设置页面，修改配置后再启动"
                    );
                }
            }
        });
    });

    ipcMain.on("frpc.stop", () => {
        if (frpcProcess && !frpcProcess.killed) {
            stopFrpcProcess()
        }
    });
};