mirror of
https://github.com/ZeppelinBot/Zeppelin.git
synced 2025-03-16 14:11:50 +00:00
92 lines
2.6 KiB
TypeScript
92 lines
2.6 KiB
TypeScript
![]() |
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 = new GuildPersistedData(this.guildId);
|
||
|
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")) {
|
||
|
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"])
|
||
|
});
|
||
|
}
|
||
|
}
|
||
|
}
|