messages.js

import { getMentionType, MENTION, MENTION_TYPES } from '@aquarius-bot/regex';
import Discord from 'discord.js';

/**
 * Check whether a message is a one-to-one direct message or not.
 * @param {Discord.Message} message - The message to check
 * @returns {boolean} whether the message is a direct message
 */
export function isDirectMessage(message) {
  return message.channel instanceof Discord.DMChannel;
}

/**
 * Generates a URL linking to a Discord Message
 * @param {Discord.Message} message - Message to get a link to
 * @returns {string} A URL linking to the message
 */
export function getLink(message) {
  const { guild, channel } = message;

  return `https://discordapp.com/channels/${guild.id}/${channel.id}/${message.id}`;
}

/**
 * Unfortunately discord.js doesn't order mentions by order as they appear in
 * the message but by Snowflake ID instead. Sometimes it's important to get
 * the mentions in order - this helper function will do that.
 * @param {Discord.Message} message
 * @param {boolean} [repeats=true] - Include repeated mentions
 * @returns {Discord.MessageMentions[]} Ordered list of Mentions
 */
export async function getOrderedMentions(message, repeats = true) {
  const mentions = await Promise.all(
    Array.from(message.content.matchAll(new RegExp(MENTION, 'g'))).map(
      (mention) => {
        switch (getMentionType(mention[0])) {
          case MENTION_TYPES.USER:
            return message.guild.members.fetch(mention.groups.id);
          case MENTION_TYPES.CHANNEL:
            return message.guild.channels.cache.get(mention.groups.id);
          case MENTION_TYPES.ROLE:
            return message.guild.roles.fetch(mention.groups.id);
          default:
            return null;
        }
      }
    )
  ).then((values) => values.filter(Boolean));

  if (repeats) {
    return mentions;
  }

  return Array.from(new Set(mentions));
}