factorio-clusterio-plugin-Cluster-Chat-Sync/controller.ts

import type {ControllerPluginContext} from '@clusterio/controller';
import {ChatEvent} from './index.js';
import * as Discord from 'discord.js';

const MAX_DISCORD_MESSAGE_LENGTH = 1950;
const ACTION_VERBS: Record<string, string> = {
    JOIN: 'joined the game',
    LEAVE: 'left the game',
    KICK: 'was kicked from the game',
    BAN: 'was banned from the game',
};

export default async function(context: ControllerPluginContext) {
    const {controller, logger, plugin} = context;
    let client: Discord.Client | null = null;

    const clientDestroy = async () => {
        if (client) {
            await client.destroy();
        }

        client = null;
    };

    const connect = async () => {
        await clientDestroy();
        const token = controller.config.get('chat_relay.discord_bot_token') as string | undefined;

        if (!token) {
            logger.error('[Chat Relay] Discord bot token not configured.');
            return;
        }

        client = new Discord.Client({intents: [
            Discord.GatewayIntentBits.Guilds,
            Discord.GatewayIntentBits.GuildMessages,
            Discord.GatewayIntentBits.MessageContent,
        ]});

        logger.info('[Chat Relay] Logging into Discord.');

        try {
            await client.login(token);
        } catch (err) {
            logger.error(`[Chat Relay] Discord login error:\n${(err as Error).stack}`);
            await clientDestroy();
            return;
        }

        logger.info('[Chat Relay] Logged in Discord successfully.');
    };

    const sendMessage = async (instanceName: string, message: string) => {
        if (!client) return;

        const mapping = controller.config.get('chat_relay.discord_channel_mapping') as Record<string, string>;
        const channelId = mapping?.[instanceName];
        if (!channelId) return;

        let channel: Discord.Channel | null;

        try {
            channel = await client.channels.fetch(channelId);
        } catch (err) {
            if ((err as {code?: number}).code !== 10003) {
                logger.error(`[Chat Relay] Discord channel fetch error:\n${(err as Error).stack}`);
            }
            return;
        }

        if (channel === null || !channel.isSendable()) {
            logger.error(`[Chat Relay] Discord Channel ID ${channelId} is not a usable channel.`);
            return;
        }

        let nrcMsg = message;

        if (controller.config.get('chat_relay.datetime_on_message')) {
            const now = new Date();
            const p = (n: number) => String(n).padStart(2, '0');
            const stamp = `${now.getFullYear()}${p(now.getMonth() + 1)}${p(now.getDate())} ${p(now.getHours())}${p(now.getMinutes())}${p(now.getSeconds())}`;
            nrcMsg = `${stamp} ${nrcMsg}`;
        }

        const send = (content: string) => channel.send({content, allowedMentions: {parse: []}});

        if (nrcMsg.length <= MAX_DISCORD_MESSAGE_LENGTH) {
            await send(nrcMsg);
            return;
        }

        let remaining = nrcMsg;

        while (remaining.length > 0) {
            let chunk = remaining.slice(0, MAX_DISCORD_MESSAGE_LENGTH);
            const lastSpace = chunk.lastIndexOf(' ');

            if (lastSpace > 0) {
                chunk = chunk.slice(0, lastSpace);
                remaining = remaining.slice(lastSpace).trim();
            } else {
                remaining = remaining.slice(MAX_DISCORD_MESSAGE_LENGTH).trim();
            }

            await send(chunk);
        }
    };

    controller.handle(ChatEvent, async (request: ChatEvent) => {
        const {instanceName, action, content} = request;

        if (action === 'CHAT' || action === 'SHOUT') {
            const nrc = content.replace(/\[special-item=.*?\]/g, '<blueprint>').replace(/<@/g, '<@\u200c>');
            const index = nrc.indexOf(':');
            const username = index === -1 ? nrc.trim() : nrc.slice(0, index);
            const msg = index === -1 ? '' : nrc.slice(index + 1).trim();
            await sendMessage(instanceName, `**\`${username}\`**: ${msg}`);
            return;
        }

        const verb = ACTION_VERBS[action];
        const text = content.trim();
    
        if (!verb) {
            await sendMessage(instanceName, `unknown action ${action} **\`${text}\`**`);
            return;
        }
    
        const name = text.toLowerCase().endsWith(verb)
            ? text.slice(0, -verb.length).trim()
            : text;

        const id = [...controller.instances.values()].find(i => i.instance_name === instanceName)?.id;
        const connection = await controller.instances.get(id);

        if (!connection) {
            await sendMessage(instanceName, `**\`${name}\`** ${verb}`);
            return;
        }
        
        let pc = '';
        try {
            pc = String(await connection.sendRcon('/_rcon return #game.connected_players', false, instanceName)).trim();
        } catch {
        }
        
        await sendMessage(instanceName, `**\`${name}\`** ${verb}${pc ? ` (${pc})` : ''}`);
    });

    controller.config.on('fieldChanged', (field) => {
        if (field === 'chat_relay.discord_bot_token') {
            connect().catch(err => {
                logger.error(`[Chat Relay] Discord bot token:\n${(err as Error).stack}`);
            });
        }
    });

    controller.hooks.shutdown.attach(plugin.name, async () => {
        await clientDestroy();
    });

    await connect();
}