Add persist plugin

src/plugins/Logs.ts

@@ -141,7 +141,7 @@ export class LogsPlugin extends Plugin {
     if (member.nick !== oldMember.nick) {
       this.serverLogs.log(LogType.MEMBER_NICK_CHANGE, {
         member,
-        oldNick: oldMember.nick,
+        oldNick: oldMember.nick || "<none>",
         newNick: member.nick
       });
     }

src/plugins/Persist.ts

@@ -0,0 +1,91 @@
+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"])
+      });
+    }
+  }
+}