This commit is contained in:
Travis CI 2016-12-31 02:05:40 +00:00
parent 1eee36ef2f
commit 5381c3214e
2 changed files with 5 additions and 4 deletions

View file

@ -11946,17 +11946,18 @@ const botGateway = __webpack_require__(0).Endpoints.botGateway;
/** /**
* Gets the recommended shard count from Discord * Gets the recommended shard count from Discord
* @param {number} token Discord auth token * @param {string} token Discord auth token
* @param {number} [guildsPerShard=1000] Number of guilds per shard
* @returns {Promise<number>} the recommended number of shards * @returns {Promise<number>} the recommended number of shards
*/ */
module.exports = function fetchRecommendedShards(token) { module.exports = function fetchRecommendedShards(token, guildsPerShard = 1000) {
return new Promise((resolve, reject) => { return new Promise((resolve, reject) => {
if (!token) throw new Error('A token must be provided.'); if (!token) throw new Error('A token must be provided.');
superagent.get(botGateway) superagent.get(botGateway)
.set('Authorization', `Bot ${token.replace(/^Bot\s*/i, '')}`) .set('Authorization', `Bot ${token.replace(/^Bot\s*/i, '')}`)
.end((err, res) => { .end((err, res) => {
if (err) reject(err); if (err) reject(err);
resolve(res.body.shards); resolve(res.body.shards * (1000 / guildsPerShard));
}); });
}); });
}; };

File diff suppressed because one or more lines are too long