zappyzep/backend/src/data/GuildPersistedData.ts

57 lines
1.3 KiB
TypeScript
Raw Normal View History

import { PersistedData } from "./entities/PersistedData";
import { BaseGuildRepository } from "./BaseGuildRepository";
import { getRepository, Repository } from "typeorm";
2018-07-30 23:35:44 +03:00
export interface IPartialPersistData {
roles?: string[];
nickname?: string;
}
export class GuildPersistedData extends BaseGuildRepository {
private persistedData: Repository<PersistedData>;
2018-07-30 23:35:44 +03:00
constructor(guildId) {
super(guildId);
this.persistedData = getRepository(PersistedData);
2018-07-30 23:35:44 +03:00
}
async find(userId: string) {
return this.persistedData.findOne({
where: {
guild_id: this.guildId,
user_id: userId,
},
});
2018-07-30 23:35:44 +03:00
}
async set(userId: string, data: IPartialPersistData = {}) {
const finalData: any = {};
if (data.roles) finalData.roles = data.roles.join(",");
if (data.nickname) finalData.nickname = data.nickname;
const existing = await this.find(userId);
if (existing) {
await this.persistedData.update(
{
guild_id: this.guildId,
user_id: userId,
},
finalData,
);
2018-07-30 23:35:44 +03:00
} else {
await this.persistedData.insert({
2018-07-30 23:35:44 +03:00
...finalData,
guild_id: this.guildId,
user_id: userId,
2018-07-30 23:35:44 +03:00
});
}
}
async clear(userId: string) {
await this.persistedData.delete({
guild_id: this.guildId,
user_id: userId,
});
2018-07-30 23:35:44 +03:00
}
}