NapCatQQ/src/index.ts

import { napCatCore } from '@/core';
import { program } from 'commander';
import qrcode from 'qrcode-terminal';
import fs from 'fs/promises';
import fsSync from 'fs';
import path from 'node:path';
import { checkVersion } from '@/common/utils/version';
import { log, logDebug, logError, LogLevel, logWarn, setLogLevel } from '@/common/utils/log';
import { NapCatOnebot11 } from '@/onebot11/main';
import { InitWebUi } from './webui/index';
import { WebUiDataRuntime } from './webui/src/helper/Data';
import { deleteOldFiles, UpdateConfig } from './common/utils/helper';
import { dirname } from 'node:path';
import { fileURLToPath } from 'node:url';
import chalk from 'chalk';
import { randomInt } from 'crypto';
import { MessageUnique } from './common/utils/MessageUnique';


const __filename = fileURLToPath(import.meta.url);
const __dirname = dirname(__filename);
const tagColor = chalk.cyan;
program
  .option('-q, --qq [type]', 'QQ号')
  .parse(process.argv);

//deleteOldFiles(path.join(__dirname, 'logs'), 3).then().catch();
// UpdateConfig().catch(logError); 移除支持
// 启动WebUi
InitWebUi();
const cmdOptions = program.opts();
// console.log(process.argv);
checkVersion().then(async (remoteVersion: string) => {
  const localVersion = JSON.parse(fsSync.readFileSync(path.join(__dirname, 'package.json')).toString()).version;
  const localVersionList = localVersion.split('.');
  const remoteVersionList = remoteVersion.split('.');
  log(tagColor('[NapCat]'), '当前版本:', localVersion);
  for (const k of [0, 1, 2]) {
    if (parseInt(remoteVersionList[k]) > parseInt(localVersionList[k])) {
      logWarn(tagColor('[NapCat]'), `检测到更新,请前往 https://github.com/NapNeko/NapCatQQ 下载 NapCatQQ V ${remoteVersion}`);
      return;
    } else if (parseInt(remoteVersionList[k]) < parseInt(localVersionList[k])) {
      break;
    }
  }
  logDebug(tagColor('[NapCat]'), '当前已是最新版本');
  return;
}).catch((e) => {
  logError(tagColor('[NapCat]'), '检测更新失败', e);
});
// 不是很好待优化
const NapCat_OneBot11 = new NapCatOnebot11();

WebUiDataRuntime.setOB11ConfigCall(NapCat_OneBot11.SetConfig);

napCatCore.onLoginSuccess((uin, uid) => {
  log('登录成功!');
  WebUiDataRuntime.setQQLoginStatus(true);
  WebUiDataRuntime.setQQLoginUin(uin.toString());
});
const showQRCode = async (url: string, base64: string, buffer: Buffer) => {
  await WebUiDataRuntime.setQQLoginQrcodeURL(url);
  logWarn('请扫描下面的二维码，然后在手Q上授权登录：');
  const qrcodePath = path.join(__dirname, 'qrcode.png');
  qrcode.generate(url, { small: true }, (res) => {
    logWarn(`\n${res}\n二维码解码URL: ${url}\n如果控制台二维码无法扫码，可以复制解码url到二维码生成网站生成二维码再扫码，也可以打开下方的二维码路径图片进行扫码`);
    fs.writeFile(qrcodePath, buffer).then(() => {
      logWarn('二维码已保存到', qrcodePath);
    });
  });
};

let quickLoginQQ = cmdOptions.qq;
const QuickLoginList = await napCatCore.getQuickLoginList();
if (quickLoginQQ == true) {
  if (QuickLoginList.LocalLoginInfoList.length > 0) {
    quickLoginQQ = QuickLoginList.LocalLoginInfoList[0].uin;
    log('-q 指令指定使用最近的QQ进行快速登录');
  } else {
    quickLoginQQ = '';
  }
}

// napCatCore.on('system.login.error', (result) => {
//   console.error('登录失败', result);
//   napCatCore.qrLogin().then().catch(console.error);
// });
napCatCore.getQuickLoginList().then((res) => {
  // 遍历 res.LocalLoginInfoList[x].isQuickLogin是否可以 res.LocalLoginInfoList[x].uin 转为string 加入string[] 最后遍历完成调用WebUiDataRuntime.setQQQuickLoginList
  WebUiDataRuntime.setQQQuickLoginList(res.LocalLoginInfoList.filter((item) => item.isQuickLogin).map((item) => item.uin.toString()));
});

WebUiDataRuntime.setQQQuickLoginCall(async (uin: string) => {
  const QuickLogin: Promise<{ result: boolean, message: string }> = new Promise((resolve, reject) => {
    if (quickLoginQQ) {
      log('正在快速登录 ', quickLoginQQ);
      napCatCore.quickLogin(quickLoginQQ).then(res => {
        if (res.loginErrorInfo.errMsg) {
          resolve({ result: false, message: res.loginErrorInfo.errMsg });
        }
        resolve({ result: true, message: '' });
      }).catch((e) => {
        logError(e);
        resolve({ result: false, message: '快速登录发生错误' });
      });
    } else {
      resolve({ result: false, message: '快速登录失败' });
    }
  });
  const result = await QuickLogin;
  return result;
});

if (quickLoginQQ) {
  log('正在快速登录 ', quickLoginQQ);
  napCatCore.quickLogin(quickLoginQQ).then(res => {
    if (res.loginErrorInfo.errMsg) {
      logError('快速登录错误:', res.loginErrorInfo.errMsg);
    }
  }).catch((e) => {
    logError('快速登录错误:', e);
    napCatCore.qrLogin(showQRCode);
  });
} else {
  log('没有 -q 指令指定快速登录，将使用二维码登录方式');
  if (QuickLoginList.LocalLoginInfoList.length > 0) {
    log(`可用于快速登录的QQ：${QuickLoginList.LocalLoginInfoList.map((u, index) => `\n${index}: ${u.uin} ${u.nickName}`)}`);
  }
  napCatCore.qrLogin(showQRCode);
}

// napCatCore.login.service.getLoginList().then((res) => {
//     const quickLoginUinList = res.LocalLoginInfoList.filter((item) => item.isQuickLogin).map((item) => item.uin);
//     if (quickLoginUinList.length !== 0) {
//       const askQuickLoginUin = readline.createInterface({
//         input: process.stdin,
//         output: process.stdout
//       });
//       const prompt = `选择快速登录的账号\n\n ${quickLoginUinList.map((u, index) => `${index}: ${u}\n`)}\n输入对应序号按回车确定: `;
//       askQuickLoginUin.question(prompt, (uinIndex) => {
//         console.log('你选择的是:', uinIndex);
//         const uin = quickLoginUinList[parseInt(uinIndex)];
//         if (!uin) {
//           console.error('请输入正确的序号');
//           return;
//         }
//         console.log('开始登录', uin);
//         napCatCore.login.quick(uin).then().catch((e) => {
//           console.error(e);
//         });
//       });
//     }
//   }
// );
//napCatCore.passwordLogin("", "").then(console.log).catch((e) => {
//   console.log(e)
//})