galactic-bot/src/middleware/ApiClientUtil.ts

import { IPCMessage } from '../@types/Shared.js';
import BaseClient from './Controller.js';

class ApiClientUtil 
{
    #client: BaseClient;
    #cache: {};

    constructor (client: BaseClient) 
    {
        this.#client = client;
        this.#cache = {};
    }

    get apiManager () 
    {
        return this.#client.api;
    }

    async handleMessage (shard: APIShard, message: IPCMessage) 
    {
        const { type, id } = message;
        let response = null;
        
        try 
        {
            switch (type) 
            {
                case 'guild-live':
                    response = await this.getLiveGuild(message);
                    break;
                case 'settings-index':
                    response = await this.settingsIndex(message);
                    break;
                case 'user':
                    response = await this.user(message);
                    break;
                case 'settings':
                    response = await this.settings(message);
                    break;
                case 'restart':
                    response = await this.restartAPIShards(message);
                    break;
                default:
                    return shard.send({ id, failure: true, error: '[IPC] No such type' });
            }
        }
        catch (error) 
        {
            this.#client.logger.error(`[CliUtil] ${error.stack || error}`);
            return shard.send({ id, failure: true, error });
        }

        this.client.logger.debug(`Result: ${JSON.stringify(response)}`);
        shard.send({ id, data: response, failure: !response });
    }

    async settings ({ guildId }) 
    {
        const evalFunc = async (client, { guildId }) => 
        {
            const wrapper = client.getGuildWrapper(guildId);
            const settings = await wrapper.settings();
            return settings;
        };

        const results = await this.client.shardingManager.broadcastEval(evalFunc, { context: { guildId } });
        
        return results.find((result) => result !== null);
    }

    async user ({ userid }) 
    {
        const evalFunc = async (client, { user }) => 
        {
            const _user = await client.users.fetch(user);
            return _user.toJSON();
        };
        const _shard = this.client.shardingManager.shards.random();
        const user = await _shard.eval(evalFunc, { user: userid });
        return user;
    }

    async settingsIndex () 
    {
        if (this.cache['settings-index']?.last < Date.now() - 3600_000) 
            return this.cache['settings-index'].value;
        const evalFunc = (client) => 
        {
            const result = client.registry.settings.reduce((acc, setting) => 
            {
                if (!acc[setting.module.name]) 
                    acc[setting.module.name] = [];
                acc[setting.module.name].push({ name: setting.name, ...setting.definitions });
                return acc;
            }, {});
            return result;
        };
        const _shard = this.client.shardingManager.shards.random();
        const index = await _shard.eval(evalFunc);
        this.cache['settings-index'] = { value: index, last: Date.now() };
        return index;
    }

    async getLiveGuild (message) 
    { // eslint-disable-line no-unused-vars
        const { guildId } = message;
        const evalFunc = (client, { guildId }) => 
        {
            try 
            {
                const wrapper = client.getGuildWrapper(guildId);
                return wrapper.toJSON();
            }
            catch 
            {
                return null; 
            }
        };

        this.#client.logger.debug(`guild-live request - shard: ${message.shard}, message id ${message.id}`);
        const result = await this.client.shardingManager.broadcastEval(evalFunc, { context: { guildId } });
        const guild = result.find((elem) => elem !== null);
        return guild;
    }

    async restartAPIShards (message) 
    {
        const { shards } = message;
        await this.apiManager.restartShards(shards);
    }

}

export default ApiClientUtil;