58 lines
1.4 KiB
TypeScript
58 lines
1.4 KiB
TypeScript
import { Repository } from "typeorm";
|
|
import { BaseGuildRepository } from "./BaseGuildRepository";
|
|
import { dataSource } from "./dataSource";
|
|
import { AutoReaction } from "./entities/AutoReaction";
|
|
|
|
export class GuildAutoReactions extends BaseGuildRepository {
|
|
private autoReactions: Repository<AutoReaction>;
|
|
|
|
constructor(guildId) {
|
|
super(guildId);
|
|
this.autoReactions = dataSource.getRepository(AutoReaction);
|
|
}
|
|
|
|
async all(): Promise<AutoReaction[]> {
|
|
return this.autoReactions.find({
|
|
where: {
|
|
guild_id: this.guildId,
|
|
},
|
|
});
|
|
}
|
|
|
|
async getForChannel(channelId: string): Promise<AutoReaction | null> {
|
|
return this.autoReactions.findOne({
|
|
where: {
|
|
guild_id: this.guildId,
|
|
channel_id: channelId,
|
|
},
|
|
});
|
|
}
|
|
|
|
async removeFromChannel(channelId: string) {
|
|
await this.autoReactions.delete({
|
|
guild_id: this.guildId,
|
|
channel_id: channelId,
|
|
});
|
|
}
|
|
|
|
async set(channelId: string, reactions: string[]) {
|
|
const existingRecord = await this.getForChannel(channelId);
|
|
if (existingRecord) {
|
|
this.autoReactions.update(
|
|
{
|
|
guild_id: this.guildId,
|
|
channel_id: channelId,
|
|
},
|
|
{
|
|
reactions,
|
|
},
|
|
);
|
|
} else {
|
|
await this.autoReactions.insert({
|
|
guild_id: this.guildId,
|
|
channel_id: channelId,
|
|
reactions,
|
|
});
|
|
}
|
|
}
|
|
}
|