3
0
Fork 0
mirror of https://github.com/ZeppelinBot/Zeppelin.git synced 2025-03-16 14:11:50 +00:00
zeppelin/src/plugins/Persist.ts

92 lines
2.6 KiB
TypeScript
Raw Normal View History

2018-07-30 23:35:44 +03:00
import { Plugin, decorators as d } from "knub";
import { GuildPersistedData, IPartialPersistData } from "../data/GuildPersistedData";
import intersection from "lodash.intersection";
import { Member, MemberOptions } from "eris";
import { GuildLogs } from "../data/GuildLogs";
import { LogType } from "../data/LogType";
import { stripObjectToScalars } from "../utils";
export class PersistPlugin extends Plugin {
protected persistedData: GuildPersistedData;
protected logs: GuildLogs;
getDefaultOptions() {
return {
config: {
persisted_roles: [],
persist_nicknames: false,
persist_voice_mutes: false
}
};
}
onLoad() {
this.persistedData = GuildPersistedData.getInstance(this.guildId);
2018-07-30 23:35:44 +03:00
this.logs = new GuildLogs(this.guildId);
}
@d.event("guildMemberRemove")
onGuildMemberRemove(_, member: Member) {
let persist = false;
const persistData: IPartialPersistData = {};
const persistedRoles = this.configValue("persisted_roles");
if (persistedRoles.length) {
const rolesToPersist = intersection(persistedRoles, member.roles);
if (rolesToPersist.length) {
persist = true;
persistData.roles = rolesToPersist;
}
}
if (this.configValue("persist_nicknames") && member.nick) {
persist = true;
persistData.nickname = member.nick;
}
if (this.configValue("persist_voice_mutes") && member.voiceState.mute != null) {
2018-07-30 23:35:44 +03:00
persist = true;
persistData.is_voice_muted = member.voiceState.mute;
}
if (persist) {
this.persistedData.set(member.id, persistData);
}
}
@d.event("guildMemberAdd")
async onGuildMemberAdd(_, member: Member) {
const persistedData = await this.persistedData.find(member.id);
if (!persistedData) return;
let restore = false;
const toRestore: MemberOptions = {};
const persistedRoles = this.configValue("persisted_roles");
if (persistedRoles.length) {
const rolesToRestore = intersection(persistedRoles, persistedData.roles);
if (rolesToRestore.length) {
restore = true;
toRestore.roles = rolesToRestore;
}
}
if (this.configValue("persist_nicknames") && persistedData.nickname) {
restore = true;
toRestore.nick = persistedData.nickname;
}
if (this.configValue("persist_voice_mutes") && persistedData.is_voice_muted) {
restore = true;
toRestore.mute = true;
}
if (restore) {
await member.edit(toRestore, "Restored upon rejoin");
this.logs.log(LogType.MEMBER_RESTORE, {
member: stripObjectToScalars(member, ["user"])
});
}
}
}