import { IPCMessage } from "../@types/Shared"; import BaseClient from "./Controller"; 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;