mirror of https://github.com/Nofated095/Q2TG.git
221 lines
7.8 KiB
TypeScript
221 lines
7.8 KiB
TypeScript
import Telegram from '../client/Telegram';
|
||
import { Friend, FriendInfo, Group, GroupInfo } from 'oicq';
|
||
import { Button } from 'telegram/tl/custom/button';
|
||
import { getLogger, Logger } from 'log4js';
|
||
import { getAvatar } from '../utils/urls';
|
||
import { CustomFile } from 'telegram/client/uploads';
|
||
import db from '../models/db';
|
||
import { Api, utils } from 'telegram';
|
||
import OicqClient from '../client/OicqClient';
|
||
import { md5 } from '../utils/hashing';
|
||
import TelegramChat from '../client/TelegramChat';
|
||
import Instance from '../models/Instance';
|
||
import getAboutText from '../utils/getAboutText';
|
||
import random from '../utils/random';
|
||
|
||
const DEFAULT_FILTER_ID = 114; // 514
|
||
|
||
export default class ConfigService {
|
||
private owner: Promise<TelegramChat>;
|
||
private readonly log: Logger;
|
||
|
||
constructor(private readonly instance: Instance,
|
||
private readonly tgBot: Telegram,
|
||
private readonly oicq: OicqClient) {
|
||
this.log = getLogger(`ConfigService - ${instance.id}`);
|
||
this.owner = tgBot.getChat(this.instance.owner);
|
||
}
|
||
|
||
private getAssociateLink(roomId: number) {
|
||
return `https://t.me/${this.tgBot.me.username}?startgroup=${roomId}`;
|
||
}
|
||
|
||
// region 打开添加关联的菜单
|
||
|
||
// 开始添加转发群组流程
|
||
public async addGroup() {
|
||
const qGroups = Array.from(this.oicq.gl).map(e => e[1])
|
||
.filter(it => !this.instance.forwardPairs.find(-it.group_id));
|
||
const buttons = qGroups.map(e =>
|
||
this.instance.workMode === 'personal' ?
|
||
[Button.inline(
|
||
`${e.group_name} (${e.group_id})`,
|
||
this.tgBot.registerCallback(() => this.onSelectChatPersonal(e)),
|
||
)] :
|
||
[Button.url(
|
||
`${e.group_name} (${e.group_id})`,
|
||
this.getAssociateLink(-e.group_id),
|
||
)]);
|
||
await (await this.owner).createPaginatedInlineSelector(
|
||
'选择 QQ 群组' + (this.instance.workMode === 'group' ? '\n然后选择在 TG 中的群组' : ''), buttons);
|
||
}
|
||
|
||
// 只可能是 personal 运行模式
|
||
public async addFriend() {
|
||
const classes = Array.from(this.oicq.classes);
|
||
const friends = Array.from(this.oicq.fl).map(e => e[1]);
|
||
classes.sort((a, b) => {
|
||
if (a[1] < b[1]) {
|
||
return -1;
|
||
}
|
||
else if (a[1] == b[1]) {
|
||
return 0;
|
||
}
|
||
else {
|
||
return 1;
|
||
}
|
||
});
|
||
await (await this.owner).createPaginatedInlineSelector('选择分组', classes.map(e => [
|
||
Button.inline(e[1], this.tgBot.registerCallback(
|
||
() => this.openFriendSelection(friends.filter(f => f.class_id === e[0]), e[1]),
|
||
)),
|
||
]));
|
||
}
|
||
|
||
private async openFriendSelection(clazz: FriendInfo[], name: string) {
|
||
clazz = clazz.filter(them => !this.instance.forwardPairs.find(them.user_id));
|
||
await (await this.owner).createPaginatedInlineSelector(`选择 QQ 好友\n分组：${name}`, clazz.map(e => [
|
||
Button.inline(`${e.remark || e.nickname} (${e.user_id})`, this.tgBot.registerCallback(
|
||
() => this.onSelectChatPersonal(e),
|
||
)),
|
||
]));
|
||
}
|
||
|
||
private async onSelectChatPersonal(info: FriendInfo | GroupInfo) {
|
||
const roomId = 'user_id' in info ? info.user_id : -info.group_id;
|
||
const name = 'user_id' in info ? info.remark || info.nickname : info.group_name;
|
||
const entity = this.oicq.getChat(roomId);
|
||
const avatar = await getAvatar(roomId);
|
||
const message = await (await this.owner).sendMessage({
|
||
message: await getAboutText(entity, true),
|
||
buttons: [
|
||
[Button.inline('自动创建群组', this.tgBot.registerCallback(
|
||
async () => {
|
||
await message.delete({ revoke: true });
|
||
this.createGroupAndLink(roomId, name);
|
||
}))],
|
||
[Button.url('手动选择现有群组', this.getAssociateLink(roomId))],
|
||
],
|
||
file: new CustomFile('avatar.png', avatar.length, '', avatar),
|
||
});
|
||
}
|
||
|
||
public async addExact(gin: number) {
|
||
const group = this.oicq.gl.get(gin);
|
||
let avatar: Buffer;
|
||
try {
|
||
avatar = await getAvatar(-group.group_id);
|
||
}
|
||
catch (e) {
|
||
avatar = null;
|
||
this.log.error(`加载 ${group.group_name} (${gin}) 的头像失败`, e);
|
||
}
|
||
const message = `${group.group_name}\n${group.group_id}\n${group.member_count} 名成员`;
|
||
await (await this.owner).sendMessage({
|
||
message,
|
||
file: avatar ? new CustomFile('avatar.png', avatar.length, '', avatar) : undefined,
|
||
buttons: Button.url('关联 Telegram 群组', this.getAssociateLink(-group.group_id)),
|
||
});
|
||
}
|
||
|
||
// endregion
|
||
|
||
/**
|
||
*
|
||
* @param room
|
||
* @param title
|
||
* @param status 传入 false 的话就不显示状态信息，可以传入一条已有消息覆盖
|
||
* @param chat
|
||
*/
|
||
public async createGroupAndLink(room: number | Friend | Group, title?: string, status: boolean | Api.Message = true, chat?: TelegramChat) {
|
||
this.log.info(`创建群组并关联：${room}`);
|
||
if (typeof room === 'number') {
|
||
room = this.oicq.getChat(room);
|
||
}
|
||
if (!title) {
|
||
// TS 这边不太智能
|
||
if (room instanceof Friend) {
|
||
title = room.remark || room.nickname;
|
||
}
|
||
else {
|
||
title = room.name;
|
||
}
|
||
}
|
||
let isFinish = false;
|
||
try {
|
||
let errorMessage = '';
|
||
// 状态信息
|
||
if (status === true) {
|
||
const avatar = await getAvatar(room);
|
||
const statusReceiver = chat ? await this.tgBot.getChat(chat.id) : await this.owner;
|
||
status = await statusReceiver.sendMessage({
|
||
message: '正在创建 Telegram 群…',
|
||
file: new CustomFile('avatar.png', avatar.length, '', avatar),
|
||
});
|
||
}
|
||
else if (status instanceof Api.Message) {
|
||
await status.edit({ text: '正在创建 Telegram 群…', buttons: Button.clear() });
|
||
}
|
||
|
||
// 关闭【添加成员】快捷条
|
||
try {
|
||
status && await status.edit({ text: '正在关闭【添加成员】快捷条…' });
|
||
await chat.hidePeerSettingsBar();
|
||
}
|
||
catch (e) {
|
||
errorMessage += `\n关闭【添加成员】快捷条失败：${e.message}`;
|
||
}
|
||
|
||
// 关联写入数据库
|
||
const chatForBot = await this.tgBot.getChat(chat.id);
|
||
status && await status.edit({ text: '正在写数据库…' });
|
||
const dbPair = await this.instance.forwardPairs.add(room, chatForBot);
|
||
isFinish = true;
|
||
|
||
// 更新头像
|
||
try {
|
||
status && await status.edit({ text: '正在更新头像…' });
|
||
const avatar = await getAvatar(room);
|
||
const avatarHash = md5(avatar);
|
||
await chatForBot.setProfilePhoto(avatar);
|
||
await db.avatarCache.create({
|
||
data: { forwardPairId: dbPair.id, hash: avatarHash },
|
||
});
|
||
}
|
||
catch (e) {
|
||
errorMessage += `\n更新头像失败：${e.message}`;
|
||
}
|
||
|
||
// 完成
|
||
if (status) {
|
||
await status.edit({ text: '正在获取链接…' });
|
||
const { link } = await chat.getInviteLink() as Api.ChatInviteExported;
|
||
await status.edit({
|
||
text: '创建完成！' + (errorMessage ? '但发生以下错误' + errorMessage : ''),
|
||
buttons: Button.url('打开', link),
|
||
});
|
||
}
|
||
}
|
||
catch (e) {
|
||
this.log.error('创建群组并关联失败', e);
|
||
await (await this.owner).sendMessage(`创建群组并关联${isFinish ? '成功了但没完全成功' : '失败'}\n<code>${e}</code>`);
|
||
}
|
||
}
|
||
|
||
public async promptNewQqChat(chat: Group | Friend) {
|
||
const message = await (await this.owner).sendMessage({
|
||
message: '你' +
|
||
(chat instanceof Group ? '加入了一个新的群' : '增加了一' + random.pick('位', '个', '只', '头') + '好友') +
|
||
'：\n' +
|
||
await getAboutText(chat, true) + '\n' +
|
||
'要创建关联群吗',
|
||
buttons: Button.inline('创建', this.tgBot.registerCallback(async () => {
|
||
await message.delete({ revoke: true });
|
||
this.createGroupAndLink(chat, chat instanceof Group ? chat.name : chat.remark || chat.nickname);
|
||
})),
|
||
});
|
||
return message;
|
||
}
|
||
|
||
}
|