mirror of https://github.com/Nofated095/Q2TG.git
113 lines
4.0 KiB
TypeScript
113 lines
4.0 KiB
TypeScript
import Telegram from '../client/Telegram';
|
||
import { getLogger, Logger } from 'log4js';
|
||
import { BigInteger } from 'big-integer';
|
||
import { Platform } from 'icqq';
|
||
import { MarkupLike } from 'telegram/define';
|
||
import OicqClient from '../client/OicqClient';
|
||
import { Button } from 'telegram/tl/custom/button';
|
||
import { CustomFile } from 'telegram/client/uploads';
|
||
import { WorkMode } from '../types/definitions';
|
||
import TelegramChat from '../client/TelegramChat';
|
||
import Instance from '../models/Instance';
|
||
import db from '../models/db';
|
||
|
||
export default class SetupService {
|
||
private owner: TelegramChat;
|
||
private readonly log: Logger;
|
||
|
||
constructor(private readonly instance: Instance,
|
||
private readonly tgBot: Telegram) {
|
||
this.log = getLogger(`SetupService - ${instance.id}`);
|
||
}
|
||
|
||
public setWorkMode(mode: WorkMode) {
|
||
this.instance.workMode = mode;
|
||
}
|
||
|
||
/**
|
||
* 在设置阶段，第一个 start bot 的用户成为 bot 主人
|
||
* @param userId 申请成为主人的用户 ID
|
||
* @return {boolean} 是否成功，false 的话就是被占用了
|
||
*/
|
||
public async claimOwner(userId: number | BigInteger) {
|
||
userId = Number(userId);
|
||
if (!this.owner) {
|
||
this.instance.owner = userId;
|
||
await this.setupOwner();
|
||
this.log.info(`用户 ID: ${userId} 成为了 Bot 主人`);
|
||
return true;
|
||
}
|
||
return false;
|
||
}
|
||
|
||
private async setupOwner() {
|
||
if (!this.owner && this.instance.owner) {
|
||
this.owner = await this.tgBot.getChat(this.instance.owner);
|
||
}
|
||
}
|
||
|
||
public async informOwner(message: string, buttons?: MarkupLike) {
|
||
if (!this.owner) {
|
||
throw new Error('应该不会运行到这里');
|
||
}
|
||
return await this.owner.sendMessage({ message, buttons: buttons || Button.clear(), linkPreview: false });
|
||
}
|
||
|
||
public async waitForOwnerInput(message?: string, buttons?: MarkupLike, remove = false) {
|
||
if (!this.owner) {
|
||
throw new Error('应该不会运行到这里');
|
||
}
|
||
message && await this.informOwner(message, buttons);
|
||
const reply = await this.owner.waitForInput();
|
||
remove && await reply.delete({ revoke: true });
|
||
return reply.message;
|
||
}
|
||
|
||
public async createUserBot(phoneNumber: string) {
|
||
if (!this.owner) {
|
||
throw new Error('应该不会运行到这里');
|
||
}
|
||
return await Telegram.create({
|
||
phoneNumber,
|
||
password: async (hint?: string) => {
|
||
return await this.waitForOwnerInput(
|
||
`请输入你的二步验证密码${hint ? '\n密码提示：' + hint : ''}`, undefined, true);
|
||
},
|
||
phoneCode: async (isCodeViaApp?: boolean) => {
|
||
await this.informOwner(`请输入你${isCodeViaApp ? ' Telegram APP 中' : '手机上'}收到的验证码\n` +
|
||
'👇请使用下面的按钮输入，不要在文本框输入，<b>否则验证码会发不出去并立即失效</b>',
|
||
Button.text('👆请使用上面的按钮输入', true, true));
|
||
return await this.owner.inlineDigitInput(5);
|
||
},
|
||
onError: (err) => this.log.error(err),
|
||
});
|
||
}
|
||
|
||
public async createOicq(uin: number, password: string, platform: Platform) {
|
||
const dbQQBot = await db.qqBot.create({ data: { uin, password, platform } });
|
||
return await OicqClient.create({
|
||
id: dbQQBot.id,
|
||
uin, password, platform,
|
||
onQrCode: async (file) => {
|
||
await this.owner.sendMessage({
|
||
message: '请使用已登录这个账号的手机 QQ 扫描这个二维码授权',
|
||
file: new CustomFile('qrcode.png', file.length, '', file),
|
||
buttons: Button.text('我已扫码', true, true),
|
||
});
|
||
await this.waitForOwnerInput();
|
||
},
|
||
onVerifyDevice: async (phone) => {
|
||
return await this.waitForOwnerInput(`请输入手机 ${phone} 收到的验证码`);
|
||
},
|
||
onVerifySlider: async (url) => {
|
||
return await this.waitForOwnerInput(`收到滑块验证码 <code>${url}</code>\n` +
|
||
'请使用<a href="https://github.com/mzdluo123/TxCaptchaHelper/releases">此软件</a>验证并输入 Ticket');
|
||
},
|
||
});
|
||
}
|
||
|
||
public async finishConfig() {
|
||
this.instance.isSetup = true;
|
||
}
|
||
}
|