SnowCatQQ/packages/snowcat-core/helper/qq-basic-info.ts

import fs from 'node:fs';
import { systemPlatform } from 'snowcat-common/src/system';
import { getDefaultQQVersionConfigInfo, getQQPackageInfoPath, getQQVersionConfigPath, parseAppidFromMajor, parseAppidFromMajorV2 } from 'snowcat-common/src/helper';
import AppidTable from '@/snowcat-core/external/appid.json';
import { LogWrapper } from './log';
import { getMajorPath } from '@/snowcat-core/index';
import { QQAppidTableType, QQPackageInfoType, QQVersionConfigType } from 'snowcat-common/src/types';
import path from 'node:path';

export class QQBasicInfoWrapper {
  QQMainPath: string | undefined;
  QQPackageInfoPath: string | undefined;
  QQVersionConfigPath: string | undefined;
  isQuickUpdate: boolean | undefined;
  QQVersionConfig: QQVersionConfigType | undefined;
  QQPackageInfo: QQPackageInfoType | undefined;
  QQVersionAppid: string | undefined;
  QQVersionQua: string | undefined;
  context: { logger: LogWrapper; };

  constructor (context: { logger: LogWrapper; }) {
    // 基础目录获取
    this.context = context;
    this.QQMainPath = process.execPath;
    if (process.platform === 'darwin' && path.basename(this.QQMainPath) === 'QQ Helper') {
      // 实用进程特殊处理 实用进程目录和QQ差远了
      this.QQMainPath = path.resolve(path.dirname(this.QQMainPath), '../../../../', 'MacOS', 'QQ');
    }
    this.QQVersionConfigPath = getQQVersionConfigPath(this.QQMainPath);

    // 基础信息获取 无快更则启用默认模板填充
    this.isQuickUpdate = !!this.QQVersionConfigPath;
    this.QQVersionConfig = this.isQuickUpdate
      ? JSON.parse(fs.readFileSync(this.QQVersionConfigPath!).toString())
      : getDefaultQQVersionConfigInfo();

    this.QQPackageInfoPath = getQQPackageInfoPath(this.QQMainPath, this.QQVersionConfig?.curVersion);
    this.QQPackageInfo = JSON.parse(fs.readFileSync(this.QQPackageInfoPath).toString());
    const { appid: IQQVersionAppid, qua: IQQVersionQua } = this.getAppidV2();
    this.QQVersionAppid = IQQVersionAppid;
    this.QQVersionQua = IQQVersionQua;
  }

  // 基础函数
  getQQBuildStr () {
    return this.QQVersionConfig?.curVersion.split('-')[1] ?? this.QQPackageInfo?.buildVersion;
  }

  getFullQQVersion () {
    const version = this.isQuickUpdate ? this.QQVersionConfig?.curVersion : this.QQPackageInfo?.version;
    if (!version) throw new Error('QQ版本获取失败');
    return version;
  }

  requireMinNTQQBuild (buildStr: string) {
    const currentBuild = +(this.getQQBuildStr() ?? '0');
    if (currentBuild === 0) throw new Error('QQBuildStr获取失败');
    return currentBuild >= parseInt(buildStr);
  }

  // 此方法不要直接使用
  getQUAFallback () {
    const platformMapping: Partial<Record<NodeJS.Platform, string>> = {
      win32: `V1_WIN_${this.getFullQQVersion()}_${this.getQQBuildStr()}_GW_B`,
      darwin: `V1_MAC_${this.getFullQQVersion()}_${this.getQQBuildStr()}_GW_B`,
      linux: `V1_LNX_${this.getFullQQVersion()}_${this.getQQBuildStr()}_GW_B`,
    };
    return platformMapping[systemPlatform] ?? (platformMapping.win32)!;
  }

  getAppIdFallback () {
    const platformMapping: Partial<Record<NodeJS.Platform, string>> = {
      win32: '537246092',
      darwin: '537246140',
      linux: '537246140',
    };
    return platformMapping[systemPlatform] ?? '537246092';
  }

  getAppidV2 (): { appid: string; qua: string; } {
    // 通过已有表 性能好
    const appidTable = AppidTable as unknown as QQAppidTableType;
    const fullVersion = this.getFullQQVersion();
    if (fullVersion) {
      // 同一 Linux 版本的 x64 和 arm64 使用不同的 AppID，优先匹配架构。
      const data = appidTable[`${fullVersion}-${process.arch}`] ?? appidTable[fullVersion];
      if (data) {
        return data;
      }
    }
    // 通过Major拉取 性能差
    try {
      const majorAppid = this.getAppidV2ByMajor(fullVersion);
      if (majorAppid) {
        this.context.logger.log('[QQ版本兼容性检测] 当前版本Appid未内置 通过Major获取 为了更好的性能请尝试更新NapCat');
        return { appid: majorAppid, qua: this.getQUAFallback() };
      }
    } catch {
      this.context.logger.log('[QQ版本兼容性检测] 通过Major 获取Appid异常 请检测NapCat/QQNT是否正常');
    }
    // 最终兜底为老版本
    this.context.logger.log('[QQ版本兼容性检测] 获取Appid异常 请检测NapCat/QQNT是否正常');
    this.context.logger.log(`[QQ版本兼容性检测] ${fullVersion} 版本兼容性不佳，可能会导致一些功能无法正常使用`);
    return { appid: this.getAppIdFallback(), qua: this.getQUAFallback() };
  }

  getAppidV2ByMajor (QQVersion: string) {
    if (!this.QQMainPath) {
      throw new Error('QQMainPath未定义 无法通过Major获取Appid');
    }
    const majorPath = getMajorPath(this.QQMainPath, QQVersion);
    // 优先通过 QQAppId/ 标记搜索
    const appidV2 = parseAppidFromMajorV2(majorPath);
    if (appidV2) {
      return appidV2;
    }
    // 回落到旧方式
    const appid = parseAppidFromMajor(majorPath);
    return appid;
  }
}