From e4636243b222166d27e2c5d361f1821d2a6ea143 Mon Sep 17 00:00:00 2001 From: Schuyler Cebulskie Date: Fri, 14 Oct 2016 02:11:31 -0400 Subject: [PATCH] Improve token "Bot" removal --- src/client/rest/RESTMethods.js | 2 +- src/sharding/ShardingManager.js | 2 +- src/util/GetRecommendedShards.js | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) diff --git a/src/client/rest/RESTMethods.js b/src/client/rest/RESTMethods.js index 8cbd40b2b..85c4c698d 100644 --- a/src/client/rest/RESTMethods.js +++ b/src/client/rest/RESTMethods.js @@ -15,7 +15,7 @@ class RESTMethods { } loginToken(token = this.rest.client.token) { - token = token.replace('Bot ', ''); + token = token.replace(/^Bot /, ''); return new Promise((resolve, reject) => { this.rest.client.manager.connectToWebSocket(token, resolve, reject); }); diff --git a/src/sharding/ShardingManager.js b/src/sharding/ShardingManager.js index 2d770472d..640822455 100644 --- a/src/sharding/ShardingManager.js +++ b/src/sharding/ShardingManager.js @@ -72,7 +72,7 @@ class ShardingManager extends EventEmitter { * Token to use for obtaining the automatic shard count, and passing to shards * @type {?string} */ - this.token = options.token; + this.token = options.token ? options.token.replace(/^Bot /, '') : null; /** * A collection of shards that this manager has spawned diff --git a/src/util/GetRecommendedShards.js b/src/util/GetRecommendedShards.js index 050740218..abce3fb39 100644 --- a/src/util/GetRecommendedShards.js +++ b/src/util/GetRecommendedShards.js @@ -10,7 +10,7 @@ module.exports = function getRecommendedShards(token) { return new Promise((resolve, reject) => { if (!token) throw new Error('A token must be provided.'); superagent.get(botGateway) - .set('Authorization', `Bot ${token.replace('Bot ', '')}`) + .set('Authorization', `Bot ${token.replace(/^Bot /, '')}`) .end((err, res) => { if (err) reject(err); resolve(res.body.shards);