NapCatQQ/packages/napcat-common/src/qq-basic-info.ts

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

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;
    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 appidTbale = AppidTable as unknown as QQAppidTableType;
    const fullVersion = this.getFullQQVersion();
    if (fullVersion) {
      const data = appidTbale[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) {
    const majorPath = getMajorPath(QQVersion);
    const appid = parseAppidFromMajor(majorPath);
    return appid;
  }
}