mirror of
https://github.com/NapNeko/NapCatQQ.git
synced 2024-11-21 09:36:35 +00:00
141 lines
5.5 KiB
TypeScript
141 lines
5.5 KiB
TypeScript
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, setLogLevel } from '@/common/utils/log';
|
||
import { NapCatOnebot11 } from '@/onebot11/main';
|
||
import { InitWebUi } from './webui/index';
|
||
import { WebUiDataRuntime } from './webui/src/helper/Data';
|
||
import { UpdateConfig } from './common/utils/helper';
|
||
import { dirname } from 'node:path';
|
||
import { fileURLToPath } from 'node:url';
|
||
|
||
|
||
const __filename = fileURLToPath(import.meta.url);
|
||
const __dirname = dirname(__filename);
|
||
|
||
program
|
||
.option('-q, --qq <type>', 'QQ号')
|
||
.parse(process.argv);
|
||
|
||
// 无缝升级旧的配置到新的 应在1.3.5发布 1.5.5 左右移除兼容支持
|
||
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('[NapCat] 当前版本:', localVersion);
|
||
for (const k of [0, 1, 2]) {
|
||
if (parseInt(remoteVersionList[k]) > parseInt(localVersionList[k])) {
|
||
console.log(`[NapCat] 检测到更新,请前往 https://github.com/NapNeko/NapCatQQ 下载 NapCatQQ V ${remoteVersion}`);
|
||
return;
|
||
} else if (parseInt(remoteVersionList[k]) < parseInt(localVersionList[k])) {
|
||
break;
|
||
}
|
||
}
|
||
logDebug('[NapCat] 当前已是最新版本');
|
||
return;
|
||
}).catch((e) => {
|
||
logError('[NapCat] 检测更新失败', e);
|
||
});
|
||
// 不是很好待优化
|
||
const NapCat_OneBot11 = new NapCatOnebot11();
|
||
|
||
WebUiDataRuntime.setOB11ConfigCall(NapCat_OneBot11.SetConfig);
|
||
|
||
napCatCore.onLoginSuccess((uin, uid) => {
|
||
console.log('登录成功!');
|
||
WebUiDataRuntime.setQQLoginStatus(true);
|
||
WebUiDataRuntime.setQQLoginUin(uin.toString());
|
||
});
|
||
const showQRCode = async (url: string, base64: string, buffer: Buffer) => {
|
||
await WebUiDataRuntime.setQQLoginQrcodeURL(url);
|
||
console.log('请扫描下面的二维码,然后在手Q上授权登录:');
|
||
const qrcodePath = path.join(__dirname, 'qrcode.png');
|
||
qrcode.generate(url, { small: true }, (res) => {
|
||
console.log(`${res}\n二维码解码URL: ${url}\n如果控制台二维码无法扫码,可以复制解码url到二维码生成网站生成二维码再扫码,也可以打开下方的二维码路径图片进行扫码`);
|
||
fs.writeFile(qrcodePath, buffer).then(() => {
|
||
console.log('二维码已保存到', qrcodePath);
|
||
});
|
||
});
|
||
};
|
||
const quickLoginQQ = cmdOptions.qq;
|
||
// 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) => {
|
||
console.error(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) => {
|
||
console.error(e);
|
||
napCatCore.qrLogin(showQRCode);
|
||
});
|
||
} else {
|
||
log('没有 -q 参数指定快速登录的QQ,将使用二维码登录方式');
|
||
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)
|
||
//})
|