Migrated Persist to new Plugin structure

backend/src/plugins/Persist/events/LoadDataEvt.ts

@@ -0,0 +1,52 @@
+import { persistEvent } from "../types";
+import { MemberOptions } from "eris";
+import intersection from "lodash.intersection";
+import { LogType } from "src/data/LogType";
+import { stripObjectToScalars } from "src/utils";
+
+export const LoadDataEvt = persistEvent({
+  event: "guildMemberAdd",
+
+  async listener(meta) {
+    const member = meta.args.member;
+    const pluginData = meta.pluginData;
+
+    const memberRolesLock = await pluginData.locks.acquire(`member-roles-${member.id}`);
+
+    const persistedData = await pluginData.state.persistedData.find(member.id);
+    if (!persistedData) {
+      memberRolesLock.unlock();
+      return;
+    }
+
+    const toRestore: MemberOptions = {};
+    const config = pluginData.config.getForMember(member);
+    const restoredData = [];
+
+    const persistedRoles = config.persisted_roles;
+    if (persistedRoles.length) {
+      const rolesToRestore = intersection(persistedRoles, persistedData.roles);
+      if (rolesToRestore.length) {
+        restoredData.push("roles");
+        toRestore.roles = Array.from(new Set([...rolesToRestore, ...member.roles]));
+      }
+    }
+
+    if (config.persist_nicknames && persistedData.nickname) {
+      restoredData.push("nickname");
+      toRestore.nick = persistedData.nickname;
+    }
+
+    if (restoredData.length) {
+      await member.edit(toRestore, "Restored upon rejoin");
+      await pluginData.state.persistedData.clear(member.id);
+
+      pluginData.state.logs.log(LogType.MEMBER_RESTORE, {
+        member: stripObjectToScalars(member, ["user", "roles"]),
+        restoredData: restoredData.join(", "),
+      });
+    }
+
+    memberRolesLock.unlock();
+  },
+});

backend/src/plugins/Persist/events/StoreDataEvt.ts

@@ -0,0 +1,35 @@
+import { persistEvent } from "../types";
+import { IPartialPersistData } from "src/data/GuildPersistedData";
+import { Member } from "eris";
+import intersection from "lodash.intersection";
+
+export const StoreDataEvt = persistEvent({
+  event: "guildMemberRemove",
+
+  async listener(meta) {
+    const member = meta.args.member as Member;
+    const pluginData = meta.pluginData;
+
+    let persist = false;
+    const persistData: IPartialPersistData = {};
+    const config = pluginData.config.getForUser(member.user);
+
+    const persistedRoles = config.persisted_roles;
+    if (persistedRoles.length && member.roles) {
+      const rolesToPersist = intersection(persistedRoles, member.roles);
+      if (rolesToPersist.length) {
+        persist = true;
+        persistData.roles = rolesToPersist;
+      }
+    }
+
+    if (config.persist_nicknames && member.nick) {
+      persist = true;
+      persistData.nickname = member.nick;
+    }
+
+    if (persist) {
+      pluginData.state.persistedData.set(member.id, persistData);
+    }
+  },
+});