frpc-desktop/electron/api/frpc.ts

import {app, ipcMain, Notification} from "electron";
import {Config, getConfig} from "../storage/config";
import {listProxy, Proxy} 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");
const log = require('electron-log');
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) {
            if (version) {
                callback(version["frpcVersionPath"]);
            }
        }
    });
};

/**
 * 生成toml配置文件
 * @param config
 * @param proxys
 */
const genTomlConfig = (config: Config, proxys: Proxy[]) => {
    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;
    });
    const toml = `
serverAddr = "${config.serverAddr}"
serverPort = ${config.serverPort}
${config.authMethod === 'token' ? `
auth.method = "token"
auth.token = "${config.authToken}"
` : ""}
${config.authMethod === 'multiuser' ? `
user = ${config.user}
metadatas.token = ${config.metaToken}
` : ""}


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}"
` : ""}
${config.proxyConfigEnable ? `
transport.proxyURL = "${config.proxyConfigProxyUrl}"
` : ""}

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


/**
 * 生成ini配置
 * @param config
 * @param proxys
 */
const genIniConfig = (config: Config, proxys: Proxy[]) => {
    const proxyIni = proxys.map(m => {
        let ini = `
[${m.name}]
type = "${m.type}"
local_ip = "${m.localIp}"
local_port = ${m.localPort}
`;
        switch (m.type) {
            case "tcp":
                ini += `remote_port = ${m.remotePort}`;
                break;
            case "http":
            case "https":
                ini += `custom_domains=[${m.customDomains.map(m => `"${m}"`)}]`;
                break;
            default:
                break;
        }

        return ini;
    });
    const ini = `
[common]
server_addr = ${config.serverAddr}
server_port = ${config.serverPort}
${config.authMethod === 'token' ? `
authentication_method = ${config.authMethod}
token = ${config.authToken}
` : ""}
${config.authMethod === 'multiuser' ? `
user = ${config.user}
meta_token = ${config.metaToken}
` : ""}
log_file = "frpc.log"
log_level = ${config.logLevel}
log_max_days = ${config.logMaxDays}
admin_addr = 127.0.0.1
admin_port = 57400
tls_enable = ${config.tlsConfigEnable}
${config.tlsConfigEnable ? `
tls_cert_file = ${config.tlsConfigCertFile}
tls_key_file = ${config.tlsConfigKeyFile}
tls_trusted_ca_file = ${config.tlsConfigTrustedCaFile}
tls_server_name = ${config.tlsConfigServerName}
` : ""}
${config.proxyConfigEnable ? `
http_proxy = "${config.proxyConfigProxyUrl}"
` : ""}

${proxyIni.join("")}
    `
    return ini;
}

/**
 * 生成配置文件
 */
export const generateConfig = (
    config: Config,
    callback: (configPath: string) => void
) => {
    listProxy((err3, proxys) => {
        if (!err3) {
            console.log(config, 'config')
            const {currentVersion} = config;
            let filename = null;
            let configContent = "";
            console.log(currentVersion, "currentVersion")
            if (currentVersion < 124395282) {
                // 版本小于v0.52.0
                filename = "frp.ini";
                configContent = genIniConfig(config, proxys)
            } else {
                filename = "frp.toml";
                configContent = genTomlConfig(config, proxys)
            }
            const configPath = path.join(app.getPath("userData"), filename)
            console.debug("生成配置成功", configPath)
            fs.writeFile(
                configPath, // 配置文件目录
                configContent, // 配置文件内容
                {flag: "w"},
                err => {
                    if (!err) {
                        callback(filename);
                    }
                }
            );
        }
    });
};

/**
 * 启动frpc子进程
 * @param cwd
 * @param commandPath
 * @param configPath
 */
const startFrpcProcess = (commandPath: string, configPath: string) => {
    log.info(`启动frpc 目录：${app.getPath("userData")} 命令：${commandPath}`,)
    const command = `${commandPath} -c ${configPath}`;
    frpcProcess = spawn(command, {
        cwd: app.getPath("userData"),
        shell: true
    });
    runningCmd.commandPath = commandPath;
    runningCmd.configPath = configPath;
    frpcProcess.stdout.on("data", data => {
        log.debug(`启动输出：${data}`)
    });
    frpcProcess.stdout.on("error", data => {
        log.error(`启动错误：${data}`)
        stopFrpcProcess(() => {
        })
    });
    frpcStatusListener = setInterval(() => {
        const status = frpcProcessStatus()
        if (!status) {
            log.info("连接已断开")
            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}`;
                        log.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;
    }
}

/**
 * 启动frpc流程
 * @param config
 */
export const startFrpWorkerProcess = async (config: Config) => {
    getFrpcVersionWorkerPath(
        config.currentVersion,
        (frpcVersionPath: string) => {
            console.log(1, '1')
            if (frpcVersionPath) {
                generateConfig(config, configPath => {
                    const platform = process.platform;
                    if (platform === 'win32') {
                        startFrpcProcess(
                            path.join(frpcVersionPath, "frpc.exe"),
                            configPath
                        );
                    } else {
                        startFrpcProcess(
                            path.join(frpcVersionPath, "frpc"),
                            configPath
                        );
                    }
                });
            }
        }
    );
}


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) {
                    startFrpWorkerProcess(config)
                } else {
                    event.reply(
                        "Home.frpc.start.error.hook", "请先前往设置页面，修改配置后再启动"
                    );
                }
            }
        });

    });

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