/[sudobot]/trunk/src/services/AFKEngine.ts

Diff of /trunk/src/services/AFKEngine.ts

Parent Directory | Revision Log | View Patch Patch

-revision 226 by rakin,
Mon Jul 29 17:29:06 2024 UTC
+revision 393 by rakin,
Mon Jul 29 17:29:59 2024 UTC
 Line 1
- import { CommandInteraction, GuildMember, Message, User } from "discord.js";
+ /**
+ * This file is part of SudoBot.
+ *
+ * Copyright (C) 2021-2022 OSN Inc.
+ *
+ * SudoBot is free software; you can redistribute it and/or modify it
+ * under the terms of the GNU Affero General Public License as published by
+ * the Free Software Foundation, either version 3 of the License, or
+ * (at your option) any later version.
+ *
+ * SudoBot is distributed in the hope that it will be useful, but
+ * WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+ * GNU Affero General Public License for more details.
+ *
+ * You should have received a copy of the GNU Affero General Public License
+ * along with SudoBot. If not, see <https://www.gnu.org/licenses/>.
+ */
+ import { formatDistanceToNowStrict } from "date-fns";
+ import { CommandInteraction, GuildMember, Message, Util } from "discord.js";
  import DiscordClient from "../client/Client";
  import MessageEmbed from "../client/MessageEmbed";
- import { notAFK } from "../commands/utils/AFKCommand";
+ import AFK, { IAFK } from "../models/AFK";
  import Service from "../utils/structures/Service";
+ export interface MentionSchema {
+     date: number;
+     user: string;
+ }
  export default class AFKEngine extends Service {
-     mention(msg: Message, user: GuildMember, cb: (data: any) => void, msg1?: any) {
+     list: IAFK[] = [];
-         this.client.db.get('SELECT * FROM afk WHERE user_id = ?', [user.id], (err: any, data: any) => {
-             if (data) {
+     constructor(client: DiscordClient) {
-                 if (msg1 === undefined) {
+         super(client);
-                     msg.channel!.send({
+         AFK.find().then(models => this.list = models).catch(console.error);
-                         embeds: [
+     }
-                             new MessageEmbed()
-                             .setDescription(`**${user.user.tag}** is AFK right now${data.reason.trim() == '' ? '.' : (' for reason: **' + data.reason.replace(/\*/g, '\\*') + '**')}`)
+     findUsers(ids: string[], guild: string) {
-                         ]
+         return this.list.filter(afk => ids.includes(afk.user) && afk.guild_id === guild);
-                     });
+     }
+     async removeUser(id: string, guild: string) {
+         let index = 0;
+         for await (const afk of this.list) {
+             if (afk.user === id && afk.guild_id === guild) {
+                 await afk.delete();
+                 this.list.splice(index, 1);
+             }
+             index++;
+         }
+     }
+     async toggle(message: Message | CommandInteraction, enable: boolean = false, status?: string) {
+         const afk = this.findUsers([message.member!.user.id], message.guild!.id);
+         if (afk.length > 0) {
+             const mentions = afk[0].get("mentions")! as Array<MentionSchema>;
+             let count = 0, text = '';
+             for await (const m of mentions) {
+                 if (count >= 3) {
+                     break;
                  }
-                 this.client.db.get('UPDATE afk SET mentions = ? WHERE id = ?', [parseInt(data.mentions) + 1, data.id], (err: any) => {});
+                 let member: GuildMember | undefined;
-                 cb(data);
+                 try {
+                     member = await message.guild!.members.fetch(m.user);
+                     if (!member) {
+                         throw new Error("user not found");
+                     }
+                 }
+                 catch (e) {
+                     console.log(e);
+                     continue;
+                 }
+                 text += `\nFrom ${member.toString()}, ${formatDistanceToNowStrict(m.date, { addSuffix: true })}`;
+                 count++;
              }
-         });
+             await this.client.afkEngine.removeUser(message.member!.user.id, message.guild!.id);
+             await message.reply({
+                 embeds: [
+                     new MessageEmbed({
+                         description: `You're no longer AFK. You had ${mentions.length ?? 0} mentions in the server(s) where SudoBot is in.${mentions.length > 0 ? `\n\n**Mentions**:${text}` : ''}`,
+                     })
+                 ]
+             });
+         }
+         else if (enable) {
+             this.client.afkEngine.list.push(await AFK.create({
+                 user: message.member!.user.id,
+                 guild_id: message.guild!.id,
+                 mentions: [],
+                 reason: status ?? undefined,
+                 createdAt: new Date()
+             }));
+             await message.reply({
+                 embeds: [
+                     new MessageEmbed({
+                         description: `You're AFK now${status ? `, for reason: **${Util.escapeMarkdown(status)}**` : ''}.`
+                     })
+                 ]
+             });
+         }
      }
-     start(msg: Message) {
+     async start(msg: Message) {
          if (msg.author.bot)
              return;
+         const selfAFK = this.findUsers([msg.author.id], msg.guild!.id);
+         if (selfAFK.length > 0) {
+             this.toggle(msg, false);
+         }
-         const mention = msg.mentions.members!.first();
+         const mention = msg.mentions.members?.first();
          if (mention) {
-             this.mention(msg, msg.member!, data => {
+             const afkRecords: Array<IAFK> = this.findUsers([...msg.mentions.members!.keys()].slice(0, 3), msg.guild!.id).filter(afk => afk.user !== msg.author.id);
-                 if (msg.author.id === data.user_id) {
-                     notAFK(this.client, msg, data);
+             if (!afkRecords || afkRecords.length < 1) {
-                 }
+                 return;
-             }, true);
+             }
-             msg.mentions.members!.forEach((member) => {
+             for (const record of afkRecords) {
-                 this.mention(msg, member, data => {
+                 const mentions = record.mentions as MentionSchema[];
-                     if (msg.author.id === data.user_id) {
-                         notAFK(this.client, msg, data);
+                 mentions.push({
-                     }
+                     date: Date.now(),
+                     user: msg.author.id
                  });
-             });
-         }
+                 record.set("mentions", mentions).save();
-         else {
+             }
-             this.client.db.get('SELECT * FROM afk WHERE user_id = ?', [msg.author.id], (err: any, data: any) => {
-                 if (data) {
+             let text = `The following users are AFK right now:`;
-                     notAFK(this.client, msg, data);
+             if (afkRecords.length > 1) {
+                 for await (const afkRecord of afkRecords) {
+                     text += `\n**${msg.mentions.members!.get(afkRecord.get("user") as string)!.user.tag}**${afkRecord.get("reason") as (null | string) ? `\n**Reason**: ${Util.escapeMarkdown(afkRecord.get("reason") as string)}` : ""}`;
                  }
+             }
+             else {
+                 text = `${msg.mentions.members!.get(afkRecords[0].get("user") as string)!.user.tag} is AFK right now${afkRecords[0].get("reason") as (null | string) ? `, for reason **${Util.escapeMarkdown(afkRecords[0].get("reason") as string)}**` : ""}.`;
+             }
+             await msg.reply({
+                 embeds: [
+                     new MessageEmbed({
+                         description: text
+                     })
+                 ]
              });
          }
      }

 Legend:



Removed from v.226
 


changed lines


 
Added in v.393
 Legend:



Removed from v.226
 


changed lines


 
Added in v.393
-Removed from v.226
+Added in v.393

[email protected]	ViewVC Help
Powered by ViewVC 1.1.26