SquadJS/plugins/discord-server-status/index.js

import { COPYRIGHT_MESSAGE } from 'core/config';

import { SERVER_A2S_UPDATED } from 'squad-server/events/server';

function makeEmbed(server, options) {
  let players = `${server.a2sPlayerCount}`;
  if (server.publicQueue + server.reserveQueue > 0)
    players += ` (+${server.publicQueue + server.reserveQueue})`;
  players += ` / ${server.publicSlots}`;
  if (server.reserveSlots > 0) players += ` (+${server.reserveSlots})`;

  return {
    embed: {
      title: server.serverName,
      color: options.color,
      fields: [
        {
          name: 'Players',
          value: `\`\`\`${players}\`\`\``
        },
        {
          name: 'Current Layer',
          value: `\`\`\`${server.currentLayer}\`\`\``,
          inline: true
        },
        {
          name: 'Next Layer',
          value: `\`\`\`${server.nextLayer || 'Unknown'}\`\`\``,
          inline: true
        }
      ],
      timestamp: new Date().toISOString(),
      footer: {
        text: `Server Status by ${COPYRIGHT_MESSAGE}`
      }
    }
  };
}

export default async function plugin(server, discordClient, options = {}) {
  if (!server)
    throw new Error(
      'DiscordDebug must be provided with a reference to the server.'
    );

  if (!discordClient)
    throw new Error('DiscordDebug must be provided with a Discord.js client.');

  options = {
    color: 16761867,
    command: '!server',
    ...options
  };

  discordClient.on('message', async message => {
    if (message.content !== options.command) return;

    const serverStatus = await message.channel.send(makeEmbed(server, options));

    await serverStatus.react('🔄');
  });

  discordClient.on('messageReactionAdd', async reaction => {
    // confirm it's a status message
    if (
      reaction.message.embeds.length !== 1 ||
      reaction.message.embeds[0].footer.text !==
        `Server Status by ${COPYRIGHT_MESSAGE}`
    )
      return;

    // ignore bots reacting
    if (reaction.count === 1) return;

    // remove reaction and readd it
    await reaction.remove();
    await reaction.message.react('🔄');

    // update the message
    await reaction.message.edit(makeEmbed(server, options));
  });

  server.on(SERVER_A2S_UPDATED, () => {
    discordClient.user.setActivity(`${server.a2sPlayerCount} / ${server.publicSlots} on ${server.currentLayer}`, { type: 'WATCHING' });
  });
}