mirror of
https://github.com/ZeppelinBot/Zeppelin.git
synced 2025-03-16 06:11:49 +00:00
95 lines
2.7 KiB
TypeScript
95 lines
2.7 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 {
|
|
public static pluginName = "persist";
|
|
|
|
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);
|
|
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 && member.roles) {
|
|
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 && member.voiceState.mute) {
|
|
persist = true;
|
|
persistData.is_voice_muted = true;
|
|
}
|
|
|
|
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");
|
|
await this.persistedData.clear(member.id);
|
|
|
|
this.logs.log(LogType.MEMBER_RESTORE, {
|
|
member: stripObjectToScalars(member, ["user"]),
|
|
});
|
|
}
|
|
}
|
|
}
|