WYF-koubo/electron/lib/env.ts

import {execSync} from "child_process";
import {resolve} from "node:path";
import fs from "node:fs";
import os from "os";
import {Log} from "../mapi/log";
import FileIndex from "../mapi/file";

export const isPackaged = ["true"].includes(process.env.IS_PACKAGED);

export const isDev = !isPackaged;

// 🆕 调试模式开关（支持环境变量和特殊标志）
export const isDebugMode =
    process.env.ELECTRON_DEBUG_MODE === 'true' ||
    process.env.ELECTRON_FORCE_DEVTOOLS === '1';

// 🆕 是否应该启用DevTools（开发环境或调试模式）
export const shouldEnableDevTools = isDev || isDebugMode;

export const isWin = process.platform === "win32";

export const isMac = process.platform === "darwin";

export const isLinux = process.platform === "linux";

export const isMain = process.type === "browser";

export const isRender = process.type === "renderer";

export const platformName = (): "win" | "osx" | "linux" | null => {
    if (isWin) return "win";
    if (isMac) return "osx";
    if (isLinux) return "linux";
    return null;
};

export const memoryInfo = () => {
    return {
        total: os.totalmem(),
        free: os.freemem(),
    };
};

const tryFirst = (functionList: (() => any)[]) => {
    for (const fun of functionList) {
        try {
            return fun();
        } catch (e) {
        }
    }
    return null;
};

let platformVersionCache: string | null = null;
export const platformVersion = () => {
    if (null === platformVersionCache) {
        const functionList: any[] = [];
        if (isWin) {
            functionList.push(() => execSync("wmic os get Version").toString().split("\n")[1].trim());
            functionList.push(() =>
                execSync(
                    "powershell -command \"(Get-ItemProperty 'HKLM:\\SOFTWARE\\Microsoft\\Windows NT\\CurrentVersion').ReleaseId\""
                )
                    .toString()
                    .trim()
            );
        } else if (isMac) {
            functionList.push(() => execSync("sw_vers -productVersion").toString().trim());
        } else if (isLinux) {
            functionList.push(() =>
                execSync("cat /etc/os-release | grep VERSION_ID").toString().split("=")[1].trim().replace(/"/g, "")
            );
        }
        platformVersionCache = tryFirst(functionList);
        if (!platformVersionCache) {
            Log.error("env.platformVersion.error");
            platformVersionCache = "0.0.0";
        }
    }
    return platformVersionCache;
};

export const platformArch = (): "x86" | "arm64" | null => {
    switch (os.arch()) {
        case "x64":
            return "x86";
        case "arm64":
            return "arm64";
    }
    return null;
};

let platformUUIDCache: string | null = null;
export const platformUUID = () => {
    if (null === platformUUIDCache) {
        const functionList: any[] = [];
        if (isWin) {
            functionList.push(() => execSync("wmic csproduct get UUID").toString().split("\n")[1].trim());
            functionList.push(() =>
                execSync('powershell -command "(Get-WmiObject Win32_ComputerSystemProduct).UUID"').toString().trim()
            );
        } else if (isMac) {
            functionList.push(() =>
                execSync("system_profiler SPHardwareDataType | grep UUID").toString().split(": ")[1].trim()
            );
        } else if (isLinux) {
            functionList.push(() => execSync("cat /var/lib/dbus/machine-id").toString().trim().toUpperCase());
        }
        platformUUIDCache = tryFirst(functionList);
        if (!platformUUIDCache) {
            Log.error("env.platformUUID.error");
            platformUUIDCache = "000000";
        }
    }
    return platformUUIDCache;
};

export const buildResolve = (value: string): string => {
    const basePath = isPackaged ? process.resourcesPath : "electron/resources";
    return resolve(basePath, "build", value);
};

export const binResolve = (value: string): string => {
    return resolve(process.resourcesPath, "bin", value);
};

export const extraResolve = (filePath: string): string => {
    const basePath = isPackaged ? process.resourcesPath : "electron/resources";
    return resolve(basePath, "extra", filePath);
};

export const extraResolveBin = (filePath: string): string => {
    const originalFilePath = filePath;
    if (isWin) {
        if (!filePath.endsWith(".exe")) {
            filePath += ".exe";
        }
    }
    const dir = [platformName(), platformArch()].join("-");
    const p = [dir, filePath].join("/");
    const binaryPath = extraResolve(p);
    if (!fs.existsSync(binaryPath)) {
        // 对于 ffmpeg 和 ffprobe，回退到系统PATH
        if (originalFilePath === 'ffmpeg' || originalFilePath === 'ffprobe') {
            Log.info(`本地${originalFilePath}未找到，尝试使用系统PATH中的${originalFilePath}`);
            return originalFilePath;
        }
        throw new Error(`错误：未找到${originalFilePath}，请先安装${originalFilePath}并添加到系统环境变量`);
    }
    return binaryPath;
};