Added support for Reaction:getUsers query

SinisterRectus · Oct 23, 2017 · b9b181d · b9b181d
1 parent a090a7e
commit b9b181d
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 4 deletions.
diff --git a/libs/client/API.lua b/libs/client/API.lua
@@ -293,9 +293,9 @@ function API:deleteUserReaction(channel_id, message_id, emoji, user_id) -- Messa
 	return self:request("DELETE", endpoint)
 end
 
-function API:getReactions(channel_id, message_id, emoji) -- Reaction:getUsers
+function API:getReactions(channel_id, message_id, emoji, query) -- Reaction:getUsers
 	local endpoint = f(endpoints.CHANNEL_MESSAGE_REACTION, channel_id, message_id, emoji)
-	return self:request("GET", endpoint)
+	return self:request("GET", endpoint, nil, query)
 end
 
 function API:deleteAllReactions(channel_id, message_id) -- Message:clearReactions

diff --git a/libs/containers/Reaction.lua b/libs/containers/Reaction.lua
@@ -18,18 +18,32 @@ function Reaction:__hash()
 	return self._emoji_id or self._emoji_name
 end
 
-function Reaction:getUsers()
+local function getUsers(self, query)
 	local emoji = Resolver.emoji(self)
 	local message = self._parent
 	local channel = message._parent
-	local data, err = self.client._api:getReactions(channel._id, message._id, emoji)
+	local data, err = self.client._api:getReactions(channel._id, message._id, emoji, query)
 	if data then
 		return SecondaryCache(data, self.client._users)
 	else
 		return nil, err
 	end
 end
 
+function Reaction:getUsers(limit)
+	return getUsers(self, limit and {limit = limit})
+end
+
+function Reaction:getUsersBefore(id, limit)
+	id = Resolver.userId(id)
+	return getUsers(self, {before = id, limit = limit})
+end
+
+function Reaction:getUsersAfter(id, limit)
+	id = Resolver.userId(id)
+	return getUsers(self, {after = id, limit = limit})
+end
+
 function Reaction:delete(id)
 	return self._parent:removeReaction(self, id)
 end