frpc-desktop/electron/api/local.ts

import {ipcMain} from "electron";
import log from "electron-log";

const {exec, spawn} = require("child_process");

type LocalPort = {
    protocol: string;
    ip: string;
    port: number;
}

export const initLocalApi = () => {
    const command = process.platform === 'win32'
        ? 'netstat -a -n'
        : 'netstat -an | grep LISTEN\n';

    ipcMain.on("local.getLocalPorts", async (event, args) => {
        log.info("开始获取本地端口")
        // 执行命令
        exec(command, (error, stdout, stderr) => {
            if (error) {
                log.error(`getLocalPorts - error ${error.message}`)
                return;
            }
            if (stderr) {
                log.error(`getLocalPorts - stderr ${stderr}`)
                return;
            }

            log.debug(`sc ${stdout}`)
            let ports = [];
            if (stdout) {
                if (process.platform === 'win32') {
                    // window
                    ports = stdout.split('\r\n')
                        .filter(f => f.indexOf('TCP') > 0 || f.indexOf('UDP') > 0)
                        .map(m => {
                            const cols = m.split(' ')
                                .filter(f => f != '')
                            const local = cols[1]
                            const s = local.lastIndexOf(":")
                            let localIP = local.slice(0, s);
                            let localPort = local.slice(s - local.length + 1);
                            console.log(1)
                            // if (local.indexOf('[') == -1) {
                            //     // ipv4
                            //     const tmp = cols[1].split(":")
                            //     localIP = tmp[0]
                            //     localPort = tmp[1]
                            // } else {
                            //     // ipv6
                            //     console.log(1)
                            // }

                            const singe: LocalPort = {
                                protocol: cols[0],
                                ip: localIP,
                                port: localPort
                            }

                            return singe;
                        })
                } else if (process.platform === 'darwin') {
                    // mac
                    ports = stdout.split('\n')
                        .filter(m => {
                            const cols = m.split(' ')
                                .filter(f => f != '')
                            const local = cols[3]
                            return local
                        })
                        .map(m => {
                            const cols = m.split(' ')
                                .filter(f => f != '')
                            const local = cols[3]
                            console.log(local, '1')
                            const s = local.lastIndexOf(".")
                            let localIP = local.slice(0, s);
                            let localPort = local.slice(s - local.length + 1);
                            const singe: LocalPort = {
                                protocol: cols[0],
                                ip: localIP,
                                port: localPort
                            }
                            return singe;
                        })
                    // .filter(f => f.indexOf('TCP') > 0 || f.indexOf('UDP') > 0)

                }

            }

            ports.sort((a, b) => a.port - b.port);

            event.reply("local.getLocalPorts.hook", {
                data: ports
            });
        });
    });
}