feat: Custom status limit and multiple mention support

backend/src/data/AFK.ts

@@ -28,4 +28,9 @@ export class AFK extends BaseRepository {
 
     return await this.afk.delete({ user_id });
   }
+
+  async isAfk(user_id: string) {
+    const afk = await this.afk.findOne({ user_id });
+    return !!afk?.status;
+  }
 }

backend/src/plugins/AFK/AFKPlugin.ts

@@ -5,12 +5,15 @@ import { AFK } from "src/data/AFK";
 
 import { AfkSetCmd } from "./commands/AFKCmd";
 import { AFKNotificationEvt } from "./events/AFKNotificationEvt";
+import { ConfigPreprocessorFn } from "knub/dist/config/configTypes";
+import { StrictValidationError } from "../../validatorUtils";
 
 const defaultOptions: PluginOptions<AFKPluginType> = {
     config: {
         can_afk: false,
         allow_links: false,
         allow_invites: false,
+        max_status_limit: 12
     },
     overrides: [
         {
@@ -19,11 +22,22 @@ const defaultOptions: PluginOptions<AFKPluginType> = {
                 can_afk: true,
                 allow_links: true,
                 allow_invites: true,
+                max_status_limit: 12,
             }
         }
     ]
 }
 
+const configPreprocessor: ConfigPreprocessorFn<AFKPluginType> = options => {
+  if (options.config.max_status_limit) {
+    const max_limit = options.config.max_status_limit;
+    if (max_limit > 24) throw new StrictValidationError([
+      `max_status_limit needs to be under 24 characters.`
+    ]);
+  }
+  return options;
+}
+
 export const AFKPlugin = zeppelinGuildPlugin<AFKPluginType>()("afk", {
     showInDocs: true,
     info: {
@@ -33,6 +47,7 @@ export const AFKPlugin = zeppelinGuildPlugin<AFKPluginType>()("afk", {
 
     configSchema: ConfigSchema,
     defaultOptions,
+    configPreprocessor,
 
     commands: [AfkSetCmd],
     events: [AFKNotificationEvt],

backend/src/plugins/AFK/commands/AFKCmd.ts

@@ -19,8 +19,9 @@ export const AfkSetCmd = afkCmd({
         const status = args.status.join(" ");
 
         // Check status length
-        if (status.length > 124) {
+        const maxStatusLength = pluginData.config.getForMember(msg.member).max_status_limit ?? 12;
-            sendErrorMessage(pluginData, msg.channel, "Status length is above **124** characters.");
+        if (status.length > maxStatusLength) {
+            sendErrorMessage(pluginData, msg.channel, `Status length is above **${maxStatusLength}** characters.`);
             return;
         }
 

backend/src/plugins/AFK/events/AFKNotificationEvt.ts

@@ -7,22 +7,34 @@ export const AFKNotificationEvt = afkEvt({
     listener: async ({ pluginData, args: { message } }) => {
         // Mention Check (if someone mentions the AFK user)
         if (message.mentions.length) {
-            const afk = await pluginData.state.afkUsers.getUserAFKStatus(message.mentions[0].id);
+            const mentionedMembers: Array<{ id: string, status: string }> = [];
-            if (!afk) return;
+            for (const user of message.mentions) {
+              const isAfk = await pluginData.state.afkUsers.isAfk(user.id);
+              if (isAfk) {
+                const afk = (await pluginData.state.afkUsers.getUserAFKStatus(user.id))!;
+                mentionedMembers.push({
+                  id: afk.user_id,
+                  status: afk.status,
+                });
+              }
+            }
 
-            sendUserMentionMessage(message, afk.status);
+            const user = mentionedMembers.length > 1
+              ? mentionedMembers.map((u) => `<@!${u.id}>: **${u.status}**`)
+              : mentionedMembers[0];
+            await sendUserMentionMessage(message, user);
 
             return;
         }
 
         // Self AFK Check (if user is the one that's AFK)
-        const afk = await pluginData.state.afkUsers.getUserAFKStatus(message.author.id);
+        const user = await pluginData.state.afkUsers.getUserAFKStatus(message.author.id);
-        if (!afk) return;
+        if (!user) return;
 
         try {
             await pluginData.state.afkUsers.clearAFKStatus(message.author.id);
         } catch (err) {}
 
-        sendWelcomeBackMessage(message);
+        await sendWelcomeBackMessage(message);
     }
 });

backend/src/plugins/AFK/functions/buildAFKMessages.ts

@@ -1,14 +1,33 @@
 import { Message } from "eris";
 
-export function sendUserMentionMessage(message: Message, status: string) {
+export function sendUserMentionMessage(message: Message, mentionedMembers: string[] | { id: string, status: string }) {
+    if (!(mentionedMembers instanceof Array)) {
+      const member = mentionedMembers as { id: string, status: string };
       return message.channel.createMessage({
         allowedMentions: {
           users: [message.author.id],
           everyone: false,
           roles: false,
         },
-        content: `<@!${message.author.id}>, the user mentioned is currently AFK: **${status}**`,
+        content: `<@!${message.author.id}>, <@!${member.id}> is currently AFK: **${member.status}**`,
       });
+    } else {
+      const deliveredLength = mentionedMembers.length;
+      const arr = mentionedMembers.length > 3 ? mentionedMembers.splice(3) : mentionedMembers;
+
+      return message.channel.createMessage({
+        allowedMentions: {
+          users: [message.author.id],
+          everyone: false,
+          roles: false,
+        },
+        content: `<@!${message.author.id}>, the following users mentioned are AFK:\n\n${mentionedMembers.join('\n')}${
+          deliveredLength > 3
+            ? `\n*+${arr.length} more...*` 
+            : ''
+        }`,
+      });
+    }
 }
 
 export function sendWelcomeBackMessage(message: Message) {

backend/src/plugins/AFK/types.ts

@@ -6,6 +6,7 @@ export const ConfigSchema = t.type({
     can_afk: t.boolean,
     allow_links: t.boolean,
     allow_invites: t.boolean,
+    max_status_limit: t.number,
 });
 export type TConfigSchema = t.TypeOf<typeof ConfigSchema>;