Various fixes

Hopefully prevent danging child processes
Reduce unnecessary db calls in moderation manager

.gitea/workflows/Tests.yaml

@@ -0,0 +1,26 @@
+on: 
+  push:
+    branches:
+      - '*'
+      - '!master'
+
+jobs:
+  tests:
+    name: Tests
+    runs-on: ubuntu-latest
+    steps:
+      - uses: actions/checkout@v4
+      - name: Install node
+        uses: actions/setup-node@v4
+        with:
+          node-version: latest
+          registry-url: https://registry.corgi.wtf
+          cache: yarn
+      - name: Install deps
+        run: yarn install
+      - name: Lint project
+        run: yarn run eslint src/
+      - name: Build project
+        run: yarn build
+        env:
+          CI: false

src/client/DiscordClient.ts

@@ -210,14 +210,6 @@ class DiscordClient extends Client
             this.emit('invalidRequestWarning', ...args);
         });
 
-        // this.once('ready', () => {
-        //     this._setActivity();
-
-        //     setInterval(() => {
-        //         this._setActivity();
-        //     }, 1800000); // I think this is 30 minutes. I could be wrong.
-        // });
-
         this.#loadEevents();
 
         // process.on('uncaughtException', (err) => {
@@ -230,7 +222,13 @@ class DiscordClient extends Client
         });
 
         process.on('message', this.#handleMessage.bind(this));
-        process.on('SIGINT', () => this.shutdown());
+        process.on('SIGINT', () => this.logger.info('Received SIGINT'));
+        process.on('SIGTERM', () => this.logger.info('Received SIGTERM'));
+        process.on('disconnect', () =>
+        {
+            this.logger.info('Process disconnected from parent for some reason, exiting');
+            this.shutdown(1);
+        });
     }
 
     async build ()
@@ -271,7 +269,7 @@ class DiscordClient extends Client
             this.#activityInterval = setInterval(() =>
             {
                 this.setActivity();
-            }, Util.random(5, 10) * 60 * 60 * 1000);
+            }, Util.random(5, 10) * 60 * 1000);
         }
 
         this.#built = true;
@@ -296,6 +294,8 @@ class DiscordClient extends Client
         // Handle misc. messages.
         if (message._mEvalResult)
             this.evalResult(message);
+        if (message._shutdown)
+            this.shutdown();
     }
 
     // eslint-disable-next-line @typescript-eslint/ban-types
@@ -397,6 +397,10 @@ class DiscordClient extends Client
                 this.user?.setActivity(`${userCount} users`, { type: ActivityType.Listening });
             },
             async () =>
+            {
+                this.user?.setActivity('out for troublemakers', { type: ActivityType.Watching });
+            },
+            async () =>
             {
                 this.user?.setActivity('for /help', { type: ActivityType.Listening });
             }

src/client/components/commands/developer/Eval.ts

@@ -1,7 +1,5 @@
 import { userInfo } from 'os';
-// eslint-disable-next-line @typescript-eslint/ban-ts-comment
-// @ts-ignore
-import { Util, FilterUtil } from '../../../../utilities/index.js';
+import { Util as U, FilterUtil as FU } from '../../../../utilities/index.js';
 import { inspect } from 'util';
 import { CommandOptionType, CommandParams } from '../../../../../@types/Client.js';
 import DiscordClient from '../../../DiscordClient.js';
@@ -12,6 +10,8 @@ import { ReplyOptions } from '../../../../../@types/Wrappers.js';
 const { username } = userInfo();
 class EvalCommand extends Command
 {
+    static Util = U;
+    static FilterUtil = FU;
     constructor (client: DiscordClient)
     {
         super(client, {
@@ -32,6 +32,10 @@ class EvalCommand extends Command
         if (async?.asBool)
             params = `(async () => {${params}})()`;
 
+        // eslint-disable-next-line @typescript-eslint/ban-ts-comment
+        // @ts-ignore // declared for ease of use
+        const { Util, FilterUtil } = EvalCommand;
+
         // eslint-disable-next-line @typescript-eslint/ban-ts-comment
         // @ts-ignore // declared for ease of use
         const { guild, author, member, client } = invoker;

src/client/components/commands/utility/Poll.ts

@@ -56,91 +56,76 @@ class PollCommand extends SlashCommand
 
     async execute (invoker: InvokerWrapper<true>, { choices, channel, duration, multichoice, message }: CommandParams)
     {
-        const { subcommand, author } = invoker;
-        const guild = invoker.guild!;
-        const member = invoker.member!;
+        const { subcommand } = invoker;
 
         if (subcommand!.name === 'create')
-        {
-            const targetChannel = (channel?.asChannel || invoker.channel) as GuildTextBasedChannel;
-            if (!targetChannel?.isTextBased())
-                throw new CommandError(invoker, { index: 'ERR_INVALID_CHANNEL_TYPE' });
-            const botMissing = targetChannel.permissionsFor(this.client.user!)?.missing([ 'SendMessages', 'EmbedLinks' ]);
-            const userMissing = targetChannel.permissionsFor(member).missing([ 'SendMessages' ]);
-            if (!botMissing || botMissing.length)
-                return invoker.editReply({ index: 'COMMAND_POLL_BOT_PERMS', params: { missing: botMissing?.join(', ') ?? 'ALL', channel: targetChannel.id } });
-            if (userMissing.length)
-                return invoker.editReply({ index: 'COMMAND_POLL_USER_PERMS', params: { missing: userMissing.join(', '), channel: targetChannel.id } });
-
-            const questions = await this.#queryQuestions(invoker, choices?.asNumber ?? 1, targetChannel);
-            if (!questions)
-                return;
-
-            await invoker.editReply({ index: 'COMMAND_POLL_STARTING' });
-
-            const embed = {
-                title: guild.format('COMMAND_POLL_TITLE', { user: author.tag }),
-                description: guild.format('COMMAND_POLL_DESCRIPTION', {
-                    multi: guild.format('COMMAND_POLL_MULTI', { multichoice: multichoice?.asBool && questions.length > 1 || false }, { code: true }),
-                    duration: duration ? guild.format('COMMAND_POLL_DURATION', { time: Util.humanise(duration.asNumber), duration: Math.floor(Date.now() / 1000) + duration.asNumber }) : guild.format('COMMAND_POLL_INFINITE')
-                }),
-                fields: questions,
-                color: EmbedDefaultColor,
-                timestamp: new Date().toISOString()
-            };
-
-            let pollMsg = null;
-            if (questions.length === 1)
-            {
-                questions[0].name = guild.format('COMMAND_POLL_FIELD_QUESTION');
-                pollMsg = await targetChannel.send({ embeds: [ embed ] });
-                await pollMsg.react('👍');
-                await pollMsg.react('👎');
-            }
-            else
-            {
-                pollMsg = await targetChannel.send({ embeds: [ embed ] });
-                for (const question of questions)
-                    await pollMsg.react(PollReactions.Multi[question.index + 1]);
-            }
-
-            const poll: PollData = {
-                questions: questions.map(q => q.value),
-                duration: (duration?.asNumber ?? 0),
-                multiChoice: multichoice?.asBool && questions.length > 1 || false,
-                user: author.id,
-                channel: targetChannel.id,
-                startedIn: invoker.channel!.id,
-                message: pollMsg.id
-            };
-
-            if (duration)
-                await this.client.polls.create(poll, guild.id);
-            await invoker.editReply({ emoji: 'success', index: 'COMMAND_POLL_START', params: { channel: targetChannel.id } });
-        }
+            return this.#createPoll(invoker, { choices, channel, duration, multichoice });
         else if (subcommand!.name === 'delete')
-        {
-            const poll = await this.client.polls.delete(message!.asString);
-            if (!poll)
-                return { index: 'COMMAND_POLL_404', emoji: 'failure' };
-            const pollChannel = await guild.resolveChannel<TextChannel>(poll.payload.channel);
-            if (!pollChannel)
-                return { index: 'COMMAND_POLL_MISSING_CHANNEL', emoji: 'failure' };
-            const msg = await pollChannel.messages.fetch(poll.payload.message).catch(() => null);
-            if (msg)
-                await msg.delete();
-            return { index: 'COMMAND_POLL_DELETED', emoji: 'success' };
-        }
+            return this.#deletePoll(invoker, { message });
         else if (subcommand!.name === 'end')
-        {
-            const poll = await this.client.polls.find(message!.asString);
-            if (!poll)
-                return { index: 'COMMAND_POLL_404', emoji: 'failure' };
-            await this.client.polls.end(poll.payload);
-            // await guild._poll(poll.data as PollData & CallbackData);
-            return { index: 'COMMAND_POLL_ENDED', emoji: 'success' };
-        }
+            return this.#endPoll(invoker, { message });
     }
+
+    async #createPoll (invoker: InvokerWrapper<true>, { choices, channel, duration, multichoice }: CommandParams)
+    {
+        const { guild, member, author } = invoker;
+        const targetChannel = (channel?.asChannel || invoker.channel) as GuildTextBasedChannel;
+        if (!targetChannel?.isTextBased())
+            throw new CommandError(invoker, { index: 'ERR_INVALID_CHANNEL_TYPE' });
+        const botMissing = targetChannel.permissionsFor(this.client.user!)?.missing([ 'SendMessages', 'EmbedLinks' ]);
+        const userMissing = targetChannel.permissionsFor(member).missing([ 'SendMessages' ]);
+        if (!botMissing || botMissing.length)
+            return invoker.editReply({ index: 'COMMAND_POLL_BOT_PERMS', params: { missing: botMissing?.join(', ') ?? 'ALL', channel: targetChannel.id } });
+        if (userMissing.length)
+            return invoker.editReply({ index: 'COMMAND_POLL_USER_PERMS', params: { missing: userMissing.join(', '), channel: targetChannel.id } });
+
+        const questions = await this.#queryQuestions(invoker, choices?.asNumber ?? 1, targetChannel);
+        if (!questions)
+            return;
+
+        await invoker.editReply({ index: 'COMMAND_POLL_STARTING' });
+
+        const embed = {
+            title: guild.format('COMMAND_POLL_TITLE', { user: author.tag }),
+            description: guild.format('COMMAND_POLL_DESCRIPTION', {
+                multi: guild.format('COMMAND_POLL_MULTI', { multichoice: multichoice?.asBool && questions.length > 1 || false }, { code: true }),
+                duration: duration ? guild.format('COMMAND_POLL_DURATION', { time: Util.humanise(duration.asNumber), duration: Math.floor(Date.now() / 1000) + duration.asNumber }) : guild.format('COMMAND_POLL_INFINITE')
+            }),
+            fields: questions,
+            color: EmbedDefaultColor,
+            timestamp: new Date().toISOString()
+        };
+
+        let pollMsg = null;
+        if (questions.length === 1)
+        {
+            questions[0].name = guild.format('COMMAND_POLL_FIELD_QUESTION');
+            pollMsg = await targetChannel.send({ embeds: [ embed ] });
+            await pollMsg.react('👍');
+            await pollMsg.react('👎');
+        }
+        else
+        {
+            pollMsg = await targetChannel.send({ embeds: [ embed ] });
+            for (const question of questions)
+                await pollMsg.react(PollReactions.Multi[question.index + 1]);
+        }
+
+        const poll: PollData = {
+            questions: questions.map(q => q.value),
+            duration: (duration?.asNumber ?? 0),
+            multiChoice: multichoice?.asBool && questions.length > 1 || false,
+            user: author.id,
+            channel: targetChannel.id,
+            startedIn: invoker.channel!.id,
+            message: pollMsg.id
+        };
+
+        if (duration)
+            await this.client.polls.create(poll, guild.id);
+        await invoker.editReply({ emoji: 'success', index: 'COMMAND_POLL_START', params: { channel: targetChannel.id } });
+    }
+
     async #queryQuestions (invoker: InvokerWrapper<true>, choices: number, targetchannel: GuildTextBasedChannel)
     {
         const { guild, member } = invoker;
@@ -166,6 +151,31 @@ class PollCommand extends SlashCommand
         }
         return questions;
     }
+
+    async #deletePoll (invoker: InvokerWrapper<true>, { message }: CommandParams)
+    {
+        const { guild } = invoker;
+        const poll = await this.client.polls.delete(message!.asString);
+        if (!poll)
+            return { index: 'COMMAND_POLL_404', emoji: 'failure' };
+        const pollChannel = await guild.resolveChannel<TextChannel>(poll.payload.channel);
+        if (!pollChannel)
+            return { index: 'COMMAND_POLL_MISSING_CHANNEL', emoji: 'failure' };
+        const msg = await pollChannel.messages.fetch(poll.payload.message).catch(() => null);
+        if (msg)
+            await msg.delete();
+        return { index: 'COMMAND_POLL_DELETED', emoji: 'success' };
+    }
+
+    async #endPoll (_invoker: InvokerWrapper<true>, { message }: CommandParams)
+    {
+        const poll = await this.client.polls.find(message!.asString);
+        if (!poll)
+            return { index: 'COMMAND_POLL_404', emoji: 'failure' };
+        await this.client.polls.end(poll._id, poll.payload);
+        // await guild._poll(poll.data as PollData & CallbackData);
+        return { index: 'COMMAND_POLL_ENDED', emoji: 'success' };
+    }
 }
 
 export default PollCommand;

src/client/components/managers/CallbackManager.ts

@@ -185,9 +185,9 @@ class CallbackManager implements Initialisable
      *
      * @async
      * @param {string} id
-     * @returns {*}
+     * @returns {Promise<void>}
      */
-    async removeCallback (id: string)
+    async removeCallback (id: string): Promise<void>
     {
         await this.storage.deleteOne({ _id: id });
         const timeout = this.#timeouts.get(id);

src/client/components/managers/ModerationManager.ts

@@ -591,9 +591,9 @@ class ModerationManager implements Initialisable, CallbackClient
         return responses;
     }
 
-    async handleCallback (id: string)
+    async handleCallback (_id: string, infraction: InfractionJSON)
     {
-        const infraction = await this.#client.mongodb.infractions.findOne({ id });
+        // const infraction = await this.#client.mongodb.infractions.findOne({ id });
         if (!infraction)
             return;
         this.#logger.debug(`Infraction callback: ${infraction.id} (${infraction.type})`);
@@ -667,7 +667,7 @@ class ModerationManager implements Initialisable, CallbackClient
         if (expiresAt - currentDate <= 0)
         {
             this.#logger.debug(`Expired infraction:\n${inspect(infraction)}`);
-            return this.handleCallback(infraction.id);
+            return this.handleCallback(infraction.id, infraction);
         }
 
         this.#logger.debug(`Creating infraction callback for ${infraction.id} (${infraction.type}), expiring in ${Util.humanise(duration / 1000)}`);

src/client/components/managers/PollManager.ts

@@ -47,12 +47,12 @@ class PollManager implements CallbackClient
         return poll;
     }
 
-    async handleCallback (_id: string, payload: PollData): Promise<void>
+    async handleCallback (id: string, payload: PollData): Promise<void>
     {
-        await this.end(payload);
+        await this.end(id, payload);
     }
 
-    async end ({ user, channel, startedIn, message, multiChoice }: PollData)
+    async end (id: string, { user, channel, startedIn, message, multiChoice }: PollData)
     {
         const startChannel = await this.#client.resolveChannel(startedIn);
         const pollChannel = await this.#client.resolveChannel(channel);
@@ -89,6 +89,7 @@ class PollManager implements CallbackClient
             if (startChannel && startChannel.isTextBased())
                 await startChannel.send(guild.format('COMMAND_POLL_NOTIFY_STARTER', { user, channel }));
         }
+        await this.callbacks.removeCallback(id);
     }
 }
 

src/middleware/Controller.ts

@@ -1,27 +1,47 @@
-import { EventEmitter } from 'node:events';
-import { inspect } from 'node:util';
+import {
+    EventEmitter
+} from 'node:events';
+
+import {
+    inspect
+} from 'node:util';
+
+import {
+    existsSync
+} from 'node:fs';
+
 import path from 'node:path';
 
-import { CommandsDef, IPCMessage } from '../../@types/Shared.js';
-import { BroadcastEvalOptions, ShardMethod, ShardingOptions } from '../../@types/Shard.js';
-import { ControllerOptions } from '../../@types/Controller.js';
+import {
+    CommandsDef,
+    IPCMessage
+} from '../../@types/Shared.js';
 
-import { MasterLogger } from '@navy.gif/logger';
-import { Collection } from 'discord.js';
+import {
+    BroadcastEvalOptions,
+    ShardMethod,
+    ShardingOptions
+} from '../../@types/Shard.js';
+
+import {
+    ControllerOptions
+} from '../../@types/Controller.js';
+
+import {
+    MasterLogger
+} from '@navy.gif/logger';
+
+import {
+    Collection
+} from 'discord.js';
 
 // Available for evals
 import ClientUtils from './ClientUtils.js';
 import Metrics from './Metrics.js';
 // import ApiClientUtil from './ApiClientUtil.js';
-import SlashCommandManager from './rest/SlashCommandManager.js';
-import { Shard } from './shard/index.js';
-import { existsSync } from 'node:fs';
+import SlashCommandManager from './SlashCommandManager.js';
 import Util from '../utilities/Util.js';
-
-// Placeholder
-type GalacticAPI = {
-    init: () => Promise<void>
-}
+import Shard from './Shard.js';
 
 class Controller extends EventEmitter
 {
@@ -39,7 +59,6 @@ class Controller extends EventEmitter
     #readyAt: number | null;
     #built: boolean;
 
-    #api?: GalacticAPI;
     clientUtils: typeof ClientUtils;
 
     constructor (options: ControllerOptions, version: string)
@@ -92,28 +111,7 @@ class Controller extends EventEmitter
     async build ()
     {
         const start = Date.now();
-        // const API = this._options.api.load ? await import('/Documents/My programs/GBot/api/index.js')
-        //     .catch(() => this.logger.warn(`Error importing API files, continuing without`)) : null;
-
-        // let API = null;
-        // if (this.#options.api.load)
-        //     API = await import('../../api/index.js').catch(() => this.#logger.warn(`Error importing API files, continuing without`));
-        // if (API) {
-        //     this.#logger.info('Booting up API');
-        //     const { default: APIManager } = API;
-        //     this.#api = new APIManager(this, this.#options.api) as GalacticAPI;
-        //     await this.#api.init();
-        //     const now = Date.now();
-        //     this.#logger.info(`API ready. Took ${now - start} ms`);
-        //     start = now;
-        // }
-
         this.#logger.status('Starting bot shards');
-        // await this.shardingManager.spawn().catch((error) => {
-        //     this.#logger.error(`Fatal error during shard spawning:\n${error.stack || inspect(error)}`);
-        //     // eslint-disable-next-line no-process-exit
-        //     process.exit(); // Prevent a boot loop when shards die due to an error in the client
-        // });
 
         const { totalShards, token } = this.#shardingOptions;
         let shardCount = 0;
@@ -133,7 +131,6 @@ class Controller extends EventEmitter
                 throw new TypeError('Amount of shards must be an integer.');
         }
 
-        // const promises = [];
         const retry: Shard[] = [];
         for (let i = 0; i < shardCount; i++)
         {
@@ -148,9 +145,9 @@ class Controller extends EventEmitter
                 this.logger.info('Retrying shard', i);
                 retry.push(shard);
             }
-            // promises.push();
         }
 
+        // Retry failed spawns
         for (const shard of retry)
         {
             try
@@ -163,8 +160,6 @@ class Controller extends EventEmitter
             }
         }
 
-        // await Promise.all(promises);
-
         this.#logger.status(`Shards spawned, spawned ${this.#shards.size} shards. Took ${Date.now() - start} ms`);
 
         this.#built = true;
@@ -173,12 +168,11 @@ class Controller extends EventEmitter
 
     async shutdown ()
     {
-        this.logger.info('Received SIGINT, shutting down');
+        this.logger.info('Received SIGINT or SIGTERM, shutting down');
         setTimeout(process.exit, 90_000);
         const promises = this.shards
             .filter(shard => shard.ready)
-            .map(shard => shard.awaitShutdown()
-                .then(() => shard.removeAllListeners()));
+            .map(shard => shard.kill());
         if (promises.length)
             await Promise.all(promises);
         this.logger.status('Shutdown complete, goodbye');
@@ -406,11 +400,6 @@ class Controller extends EventEmitter
         return this.#logger;
     }
 
-    get api ()
-    {
-        return this.#api;
-    }
-
     get shards ()
     {
         return this.#shards.clone();

src/middleware/Metrics.ts

@@ -1,6 +1,6 @@
 import BaseClient from './Controller.js';
 import os from 'node:os';
-import Shard from './shard/Shard.js';
+import Shard from './Shard.js';
 import { IPCMessage } from '../../@types/Shared.js';
 import djs, { GuildMessageManager } from 'discord.js';
 import DiscordClient from '../client/DiscordClient.js';

src/middleware/Shard.ts

@@ -1,13 +1,14 @@
 import EventEmitter from 'node:events';
 import path from 'node:path';
-import { makePlainError, makeError, MakeErrorOptions } from 'discord.js';
-import { Util } from '../../utilities/index.js';
 import childProcess, { ChildProcess } from 'node:child_process';
+import { makePlainError, makeError, MakeErrorOptions } from 'discord.js';
 
-import { EnvObject, IPCMessage } from '../../../@types/Shared.js';
-import Controller from '../Controller.js';
-import { ShardOptions } from '../../../@types/Shard.js';
-import { ClientOptions } from '../../../@types/Client.js';
+import Controller from './Controller.js';
+import { Util } from '../utilities/index.js';
+
+import { ShardOptions } from '../../@types/Shard.js';
+import { ClientOptions } from '../../@types/Client.js';
+import { EnvObject, IPCMessage } from '../../@types/Shared.js';
 
 const KillTO = 90 * 1000;
 
@@ -16,8 +17,8 @@ class Shard extends EventEmitter
     [key: string]: unknown;
 
     #id: number;
-    #manager: Controller;
-    #env: EnvObject; // { [key: string]: string | boolean | number };
+    #controller: Controller;
+    #env: EnvObject;
     #ready: boolean;
     #clientOptions: ClientOptions;
 
@@ -38,10 +39,10 @@ class Shard extends EventEmitter
     #crashes: number[];
     #fatal: boolean;
 
-    constructor (manager: Controller, id: number, options: ShardOptions)
+    constructor (controller: Controller, id: number, options: ShardOptions)
     {
         super();
-        this.#manager = manager;
+        this.#controller = controller;
         this.#id = id;
 
         this.#args = options.args ?? [];
@@ -370,7 +371,7 @@ class Shard extends EventEmitter
             if (message._sFetchProp)
             {
                 const resp = { _sFetchProp: message._sFetchProp, _sFetchPropShard: message._sFetchPropShard };
-                this.#manager.fetchClientValues(message._sFetchProp, message._sFetchPropShard).then(
+                this.#controller.fetchClientValues(message._sFetchProp, message._sFetchPropShard).then(
                     (results: unknown) => this.send({ ...resp, _result: results }),
                     (error: Error) => this.send({ ...resp, _error: makePlainError(error) })
                 );
@@ -380,7 +381,7 @@ class Shard extends EventEmitter
             if (message._sEval)
             {
                 const resp = { _sEval: message._sEval, _sEvalShard: message._sEvalShard };
-                this.#manager._performOnShards('eval', [ message._sEval ], message._sEvalShard).then(
+                this.#controller._performOnShards('eval', [ message._sEval ], message._sEvalShard).then(
                     (results: unknown) => this.send({ ...resp, _result: results }),
                     (error: Error) => this.send({ ...resp, _error: makePlainError(error) })
                 );
@@ -390,7 +391,7 @@ class Shard extends EventEmitter
             if (message._sRespawnAll)
             {
                 const { shardDelay, respawnDelay, timeout } = message._sRespawnAll;
-                this.#manager.respawnAll({ shardDelay, respawnDelay, timeout }).catch(() =>
+                this.#controller.respawnAll({ shardDelay, respawnDelay, timeout }).catch(() =>
                 {
                     //
                 });

src/middleware/SlashCommandManager.ts

@@ -5,8 +5,8 @@ import fs from 'node:fs';
 import path from 'node:path';
 import { inspect } from 'node:util';
 
-import BaseClient from '../Controller.js';
-import { Command, CommandOption, CommandsDef } from '../../../@types/Shared.js';
+import BaseClient from './Controller.js';
+import { Command, CommandOption, CommandsDef } from '../../@types/Shared.js';
 
 type CommandHashTable = {
     global: string | null,

src/middleware/shard/index.ts

@@ -1,6 +0,0 @@
-import Shard from './Shard.js';
-// import ShardingManager from './ShardingManager.js';
-export {
-    Shard,
-    // ShardingManager
-};