DawnLauncher/electron/commons/utils.ts

import { resolve, dirname, parse, join } from "node:path";
import { isAbsolutePath } from "../../commons/utils/common";
import mime from "mime";
import { readFileSync } from "node:fs";
import icojs from "icojs";

// 图标格式
const iconExts = ["jpg", "jpeg", "png", "gif", "ico", "svg", "webp"];

/**
 * 随机user-agent
 */
function getRandomUserAgent() {
  const userAgents = [
    "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/91.0.4472.124 Safari/537.36",
    "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/91.0.4472.124 Safari/537.36",
    "Mozilla/5.0 (Windows NT 10.0; Win64; x64; rv:89.0) Gecko/20100101 Firefox/89.0",
    "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Firefox/89.0",
    "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Safari/537.36",
    "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Safari/537.36",
    "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/91.0.4472.124 Safari/537.36 Edg/91.0.864.59",
    "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/91.0.4472.124 Safari/537.36 Edg/91.0.864.59",
    "Mozilla/5.0 (Windows NT 10.0; Win64; x64; rv:89.0) Gecko/20100101 Firefox/89.0",
    "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Firefox/89.0",
    "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/91.0.4472.124 Safari/537.36 OPR/76.0.4017.123",
    "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/91.0.4472.124 Safari/537.36 OPR/76.0.4017.123",
    "Mozilla/5.0 (Windows NT 10.0; Win64; x64; rv:89.0) Gecko/20100101 Firefox/89.0",
    "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Firefox/89.0",
  ];
  const randomIndex = Math.floor(Math.random() * userAgents.length);
  return userAgents[randomIndex];
}

/**
 * 组装参数
 * @param paramsMap
 * @returns
 */
function getURLParams(paramsMap: Map<string, any>) {
  let res = "";
  if (paramsMap.size > 0) {
    res += "?";
    let params: string | null = null;
    paramsMap.forEach((value, key) => {
      if (params) {
        params += "&" + key + "=" + value;
      } else {
        params = key + "=" + value;
      }
    });
    res += params;
  }
  return res;
}

/**
 * 解析环境变量
 * @param path
 */
function parseEnvPath(path: string) {
  // 尝试解析路径中的环境变量
  let parsedPath = parse(path);
  // 路径数组
  let pathArr: Array<string> = [];
  // 判断是否是网络路径，以\\开头
  let isNetwork = false;
  if (path.indexOf("\\\\") === 0) {
    isNetwork = true;
  }
  // 是否是一级路径
  let isBase = false;
  if (!parsedPath.dir || parsedPath.dir.trim() === "") {
    // 如果为空代表路径只有一级，也就是当前文件本身，比如C:/1.txt
    pathArr = parsedPath.base.split("\\");
    isBase = true;
  } else {
    // 不为空代表有父级目录
    pathArr = parsedPath.dir.split("\\");
  }
  // 新路径
  let newPathArr: Array<string> = [];
  // 正则提取环境变量 %{path}% 提取中间的path名称
  const pattern = /^%.*%$/;
  for (let string of pathArr) {
    // 符合环境变量正则
    if (pattern.test(string)) {
      // 尝试获取
      let env: string | null = global.addon.getEnvByName(
        string.substring(1, string.length - 1)
      );
      // 如果提取到环境变量了就使用环境变量路径，如果没有就使用原路径
      if (env && env.trim() !== "") {
        newPathArr.push(env);
      } else {
        newPathArr.push(string);
      }
    } else {
      // 没有匹配到正则，使用原路径
      newPathArr.push(string);
    }
  }
  // 根据上方解析，如果拥有dir的话，需要追加base变量
  if (!isBase) {
    newPathArr.push(parsedPath.base);
  }
  // 拼接
  let newPath = join(...newPathArr);
  // 如果是网络路径，在最前面加\\
  if (isNetwork) {
    newPath = "\\\\" + newPath;
  }
  return newPath;
}

/**
 * 解析路径
 * @returns
 */
function parsePath(path: string) {
  // 尝试解析环境变量
  path = parseEnvPath(path);
  // 是否是相对路径
  if (!isAbsolutePath(path)) {
    return resolve(
      process.env.NODE_ENV === "development"
        ? resolve(".")
        : dirname(process.execPath),
      path
    );
  }
  return path;
}

/**
 * 获取文件图标
 * @param filePath
 */
async function getFileIcon(filePath: string | null) {
  // 图标
  let icon: string | null = null;
  if (filePath) {
    // 获取后缀
    let ext = mime.getExtension(mime.getType(filePath));
    if (iconExts.includes(ext)) {
      // 读取文件
      let buffer = readFileSync(filePath);
      if (ext === "ico") {
        let images = await icojs.parseICO(buffer);
        const largestImage = images.reduce((max, current) => {
          if (current.width * current.height > max.width * max.height) {
            return current;
          }
          return max;
        });
        buffer = Buffer.from(largestImage.buffer);
      }
      icon =
        "data:" +
        mime.getType(filePath) +
        ";base64," +
        buffer.toString("base64");
    } else {
      // 获取图标
      icon = global.addon.getFileIcon(filePath);
    }
  }
  return icon;
}

export { getURLParams, parsePath, getFileIcon, iconExts, getRandomUserAgent };