79 lines
2.8 KiB
JavaScript
79 lines
2.8 KiB
JavaScript
'use strict';
|
|
|
|
const { Collection } = require('@discordjs/collection');
|
|
const { makeURLSearchParams } = require('@discordjs/rest');
|
|
const { ReactionType, Routes } = require('discord-api-types/v10');
|
|
const CachedManager = require('./CachedManager');
|
|
const { DiscordjsError, ErrorCodes } = require('../errors');
|
|
const User = require('../structures/User');
|
|
|
|
/**
|
|
* Manages API methods for users who reacted to a reaction and stores their cache.
|
|
* @extends {CachedManager}
|
|
*/
|
|
class ReactionUserManager extends CachedManager {
|
|
constructor(reaction, iterable) {
|
|
super(reaction.client, User, iterable);
|
|
|
|
/**
|
|
* The reaction that this manager belongs to
|
|
* @type {MessageReaction}
|
|
*/
|
|
this.reaction = reaction;
|
|
}
|
|
|
|
/**
|
|
* The cache of this manager
|
|
* @type {Collection<Snowflake, User>}
|
|
* @name ReactionUserManager#cache
|
|
*/
|
|
|
|
/**
|
|
* Options used to fetch users who gave a reaction.
|
|
* @typedef {Object} FetchReactionUsersOptions
|
|
* @property {ReactionType} [type=ReactionType.Normal] The reaction type to fetch
|
|
* @property {number} [limit=100] The maximum amount of users to fetch, defaults to `100`
|
|
* @property {Snowflake} [after] Limit fetching users to those with an id greater than the supplied id
|
|
*/
|
|
|
|
/**
|
|
* Fetches all the users that gave this reaction. Resolves with a collection of users, mapped by their ids.
|
|
* @param {FetchReactionUsersOptions} [options] Options for fetching the users
|
|
* @returns {Promise<Collection<Snowflake, User>>}
|
|
*/
|
|
async fetch({ type = ReactionType.Normal, limit = 100, after } = {}) {
|
|
const message = this.reaction.message;
|
|
const query = makeURLSearchParams({ limit, after, type });
|
|
const data = await this.client.rest.get(
|
|
Routes.channelMessageReaction(message.channelId, message.id, this.reaction.emoji.identifier),
|
|
{ query },
|
|
);
|
|
const users = new Collection();
|
|
for (const rawUser of data) {
|
|
const user = this.client.users._add(rawUser);
|
|
this.cache.set(user.id, user);
|
|
users.set(user.id, user);
|
|
}
|
|
return users;
|
|
}
|
|
|
|
/**
|
|
* Removes a user from this reaction.
|
|
* @param {UserResolvable} [user=this.client.user] The user to remove the reaction of
|
|
* @returns {Promise<MessageReaction>}
|
|
*/
|
|
async remove(user = this.client.user) {
|
|
const userId = this.client.users.resolveId(user);
|
|
if (!userId) throw new DiscordjsError(ErrorCodes.ReactionResolveUser);
|
|
const message = this.reaction.message;
|
|
const route =
|
|
userId === this.client.user.id
|
|
? Routes.channelMessageOwnReaction(message.channelId, message.id, this.reaction.emoji.identifier)
|
|
: Routes.channelMessageUserReaction(message.channelId, message.id, this.reaction.emoji.identifier, userId);
|
|
await this.client.rest.delete(route);
|
|
return this.reaction;
|
|
}
|
|
}
|
|
|
|
module.exports = ReactionUserManager;
|