2023-06-30 11:26:45 +00:00
|
|
|
|
import { Friend, Group } from 'icqq';
|
2022-02-23 12:25:26 +00:00
|
|
|
|
import TelegramChat from '../client/TelegramChat';
|
|
|
|
|
import OicqClient from '../client/OicqClient';
|
|
|
|
|
import Telegram from '../client/Telegram';
|
|
|
|
|
import db from './db';
|
2022-02-26 10:15:40 +00:00
|
|
|
|
import { Entity } from 'telegram/define';
|
2022-03-01 08:14:04 +00:00
|
|
|
|
import { BigInteger } from 'big-integer';
|
2022-03-07 10:05:14 +00:00
|
|
|
|
import { Pair } from './Pair';
|
2022-03-27 11:35:57 +00:00
|
|
|
|
import { getLogger, Logger } from 'log4js';
|
2023-02-23 06:08:53 +00:00
|
|
|
|
import Instance from './Instance';
|
2022-02-23 12:25:26 +00:00
|
|
|
|
|
2022-03-07 10:05:14 +00:00
|
|
|
|
export default class ForwardPairs {
|
|
|
|
|
private pairs: Pair[] = [];
|
2022-03-27 11:35:57 +00:00
|
|
|
|
private readonly log: Logger;
|
2022-03-01 08:14:04 +00:00
|
|
|
|
|
2022-03-07 10:05:14 +00:00
|
|
|
|
private constructor(private readonly instanceId: number) {
|
2022-03-27 11:35:57 +00:00
|
|
|
|
this.log = getLogger(`ForwardPairs - ${instanceId}`);
|
2022-03-01 08:14:04 +00:00
|
|
|
|
}
|
|
|
|
|
|
2022-02-23 12:25:26 +00:00
|
|
|
|
// 在 forwardController 创建时初始化
|
2024-02-09 07:57:58 +00:00
|
|
|
|
private async init(oicq: OicqClient, tgBot: Telegram, tgUser: Telegram) {
|
2022-03-07 10:05:14 +00:00
|
|
|
|
const dbValues = await db.forwardPair.findMany({
|
|
|
|
|
where: { instanceId: this.instanceId },
|
|
|
|
|
});
|
2022-02-23 12:25:26 +00:00
|
|
|
|
for (const i of dbValues) {
|
2022-03-27 11:35:57 +00:00
|
|
|
|
try {
|
|
|
|
|
const qq = oicq.getChat(Number(i.qqRoomId));
|
|
|
|
|
const tg = await tgBot.getChat(Number(i.tgChatId));
|
2024-02-09 07:57:58 +00:00
|
|
|
|
const tgUserChat = await tgUser.getChat(Number(i.tgChatId));
|
|
|
|
|
if (qq && tg && tgUserChat) {
|
|
|
|
|
this.pairs.push(new Pair(qq, tg, tgUserChat, i.id, i.flags));
|
2022-03-27 11:35:57 +00:00
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
catch (e) {
|
|
|
|
|
this.log.warn(`初始化遇到问题,QQ: ${i.qqRoomId} TG: ${i.tgChatId}`);
|
|
|
|
|
}
|
2022-02-23 12:25:26 +00:00
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
2024-02-09 07:57:58 +00:00
|
|
|
|
public static async load(instanceId: number, oicq: OicqClient, tgBot: Telegram, tgUser: Telegram) {
|
2022-03-07 10:05:14 +00:00
|
|
|
|
const instance = new this(instanceId);
|
2024-02-09 07:57:58 +00:00
|
|
|
|
await instance.init(oicq, tgBot, tgUser);
|
2022-03-07 10:05:14 +00:00
|
|
|
|
return instance;
|
|
|
|
|
}
|
|
|
|
|
|
2024-02-09 07:57:58 +00:00
|
|
|
|
public async add(qq: Friend | Group, tg: TelegramChat, tgUser: TelegramChat) {
|
2022-03-01 08:14:04 +00:00
|
|
|
|
const dbEntry = await db.forwardPair.create({
|
2022-02-23 12:25:26 +00:00
|
|
|
|
data: {
|
|
|
|
|
qqRoomId: qq instanceof Friend ? qq.user_id : -qq.group_id,
|
|
|
|
|
tgChatId: Number(tg.id),
|
2022-03-07 10:05:14 +00:00
|
|
|
|
instanceId: this.instanceId,
|
2022-02-23 12:25:26 +00:00
|
|
|
|
},
|
|
|
|
|
});
|
2024-02-09 07:57:58 +00:00
|
|
|
|
this.pairs.push(new Pair(qq, tg, tgUser, dbEntry.id, dbEntry.flags));
|
2022-03-01 08:14:04 +00:00
|
|
|
|
return dbEntry;
|
2022-02-23 12:25:26 +00:00
|
|
|
|
}
|
|
|
|
|
|
2022-03-08 06:38:40 +00:00
|
|
|
|
public async remove(pair: Pair) {
|
|
|
|
|
this.pairs.splice(this.pairs.indexOf(pair), 1);
|
|
|
|
|
await db.forwardPair.delete({
|
|
|
|
|
where: { id: pair.dbId },
|
|
|
|
|
});
|
|
|
|
|
}
|
|
|
|
|
|
2022-03-01 08:14:04 +00:00
|
|
|
|
public find(target: Friend | Group | TelegramChat | Entity | number | BigInteger) {
|
2022-03-06 15:32:31 +00:00
|
|
|
|
if (!target) return null;
|
2022-02-23 12:25:26 +00:00
|
|
|
|
if (target instanceof Friend) {
|
|
|
|
|
return this.pairs.find(e => e.qq instanceof Friend && e.qq.user_id === target.user_id);
|
|
|
|
|
}
|
|
|
|
|
else if (target instanceof Group) {
|
|
|
|
|
return this.pairs.find(e => e.qq instanceof Group && e.qq.group_id === target.group_id);
|
|
|
|
|
}
|
2022-03-01 08:14:04 +00:00
|
|
|
|
else if (typeof target === 'number' || 'eq' in target) {
|
|
|
|
|
return this.pairs.find(e => e.qqRoomId === target || e.tg.id.eq(target));
|
2022-02-24 10:27:06 +00:00
|
|
|
|
}
|
2022-02-23 12:25:26 +00:00
|
|
|
|
else {
|
|
|
|
|
return this.pairs.find(e => e.tg.id.eq(target.id));
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|