SquadJS/plugins/mapvote/mapvote-did-you-mean.js

import { COPYRIGHT_MESSAGE } from 'core/config';
import { LOG_PARSER_NEW_GAME } from 'squad-server/events/log-parser';
import { RCON_CHAT_MESSAGE } from 'squad-server/events/rcon';

import MapVote from './mapvote.js';

export default function(server, squadLayerFilter, options = {}) {
  options = {
    alwaysOn: true,
    minPlayerCount: null,
    minVoteCount: null,
    ...options
  };

  let mapvote;
  let manuallyCreated;

  async function newMapvote(manuallyCreatedOption = true) {
    mapvote = new MapVote(server, squadLayerFilter, {
      minVoteCount: options.minVoteCount
    });

    manuallyCreated = manuallyCreatedOption;

    mapvote.on('NEW_WINNER', async results => {
      await server.rcon.broadcast(
        `New Map Vote Winner: ${results[0].layer.layer}. Participate in the map vote by typing "!mapvote help" in chat.`
      );
    });

    if (manuallyCreated)
      await server.rcon.broadcast(
        `A new map vote has started. Participate in the map vote by typing "!mapvote help" in chat.`
      );
  }

  if (options.alwaysOn) newMapvote(false);

  server.on(LOG_PARSER_NEW_GAME, () => {
    if (options.alwaysOn) {
      newMapvote(false);
    } else {
      mapvote = null;
    }
  });

  server.on(RCON_CHAT_MESSAGE, async info => {
    const match = info.message.match(/^!mapvote ?(.*)/);
    if (!match) return;

    if (match[1] === 'help') {
      await server.rcon.warn(
        info.steamID,
        'You may use any of the following commands in chat:'
      );
      await server.rcon.warn(
        info.steamID,
        '!mapvote results - View the current vote counts.'
      );
      await server.rcon.warn(
        info.steamID,
        '!mapvote <layer name> - Vote for the specified layer.'
      );
      await server.rcon.warn(
        info.steamID,
        'When inputting a layer name, we autocorrect any miss spelling.'
      );

      if (options.minVoteCount !== null)
        await server.rcon.warn(
          info.steamID,
          `${options.minVoteCount} votes need to be made for a winner to be selected.`
        );

      return;
    }

    if (match[1] === 'start') {
      if (info.chat !== 'ChatAdmin') return;

      if (mapvote) {
        await server.rcon.warn(info.steamID, 'A mapvote has already begun.');
      } else {
        await newMapvote();
      }
      return;
    }

    if (!mapvote) {
      await server.rcon.warn(info.steamID, 'A map vote has not begun.');
      return;
    }

    if (match[1] === 'restart') {
      if (info.chat !== 'ChatAdmin') return;
      await newMapvote();
      return;
    }

    if (match[1] === 'end') {
      if (info.chat !== 'ChatAdmin') return;

      const results = mapvote.getResults(true);

      if (results.length === 0)
        await server.rcon.broadcast(`No layer gained enough votes to win.`);
      else
        await server.rcon.broadcast(
          `${mapvote.getResults()[0].layer.layer} won the mapvote!`
        );

      mapvote = null;
      return;
    }

    if (match[1] === 'destroy') {
      if (info.chat !== 'ChatAdmin') return;
      mapvote = null;
      return;
    }

    if (match[1] === 'results') {
      const results = mapvote.getResults();

      if (results.length === 0) {
        await server.rcon.warn(info.steamID, 'No one has voted yet.');
      } else {
        await server.rcon.warn(
          info.steamID,
          'The current vote counts are as follows:'
        );
        for (const result of results) {
          await server.rcon.warn(
            info.steamID,
            `${result.layer.layer} - ${result.votes} vote${
              result.votes > 1 ? 's' : ''
            }`
          );
        }
        return;
      }
    }

    if (!manuallyCreated && server.players.length < options.minPlayerCount) {
      await server.rcon.warn(
        info.steamID,
        'Not enough players online to vote.'
      );
      return;
    }

    try {
      const layerName = await mapvote.makeVoteByDidYouMean(
        info.steamID,
        match[1]
      );
      await server.rcon.warn(info.steamID, `You voted for ${layerName}.`);
    } catch (err) {
      await server.rcon.warn(info.steamID, err.message);
    }
    await server.rcon.warn(info.steamID, `Powered by: ${COPYRIGHT_MESSAGE}`);
  });
}