Skip to content

Commit

Permalink
sorry gus i borked
Browse files Browse the repository at this point in the history
  • Loading branch information
amishshah authored and Ratismal committed Jul 5, 2017
1 parent 4f93a06 commit 78d05b1
Show file tree
Hide file tree
Showing 4 changed files with 71 additions and 33 deletions.
39 changes: 39 additions & 0 deletions src/client/rest/DiscordAPIError.js
Original file line number Diff line number Diff line change
@@ -0,0 +1,39 @@
/**
* Represents an error from the Discord API
*/
class DiscordAPIError extends Error {
constructor(error) {
super();
const flattened = error.errors ? `\n${this.constructor.flattenErrors(error.errors).join('\n')}` : '';
this.name = 'DiscordAPIError';
this.message = `${error.message}${flattened}`;

/**
* HTTP error code returned by Discord
* @type {number}
*/
this.code = error.code;
}

/**
* Flattens an errors object returned from the API into an array
* @param {Object} obj Discord errors object
* @param {string} [key] idklol
* @returns {string[]}
*/
static flattenErrors(obj, key = '') {
let messages = [];
for (const k of Object.keys(obj)) {
const newKey = key ? isNaN(k) ? `${key}.${k}` : `${key}[${k}]` : k;
if (obj[k]._errors) {
messages.push(`${newKey}: ${obj[k]._errors.map(e => e.message).join(' ')}`);
} else {
messages = messages.concat(this.flattenErrors(obj[k], newKey));
}
}

return messages;
}
}

module.exports = DiscordAPIError;
3 changes: 2 additions & 1 deletion src/client/rest/RequestHandlers/Burst.js
Original file line number Diff line number Diff line change
@@ -1,4 +1,5 @@
const RequestHandler = require('./RequestHandler');
const DiscordAPIError = require('../DiscordAPIError');

class BurstRequestHandler extends RequestHandler {
constructor(restManager, endpoint) {
Expand Down Expand Up @@ -40,7 +41,7 @@ class BurstRequestHandler extends RequestHandler {
this.resetTimeout = null;
}, Number(res.headers['retry-after']) + this.client.options.restTimeOffset);
} else {
item.reject(err);
item.reject(err.status === 400 ? new DiscordAPIError(res.body) : err);
this.handle();
}
} else {
Expand Down
60 changes: 29 additions & 31 deletions src/client/rest/RequestHandlers/Sequential.js
Original file line number Diff line number Diff line change
@@ -1,4 +1,5 @@
const RequestHandler = require('./RequestHandler');
const DiscordAPIError = require('../DiscordAPIError');

/**
* Handles API Requests sequentially, i.e. we wait until the current request is finished before moving onto
Expand Down Expand Up @@ -48,42 +49,39 @@ class SequentialRequestHandler extends RequestHandler {
execute(item) {
this.busy = true;
return new Promise(resolve => {
item.request
.gen()
.on('error', e => item.reject(e))
.end((err, res) => {
if (res && res.headers) {
this.requestLimit = Number(res.headers['x-ratelimit-limit']);
this.requestResetTime = Number(res.headers['x-ratelimit-reset']) * 1000;
this.requestRemaining = Number(res.headers['x-ratelimit-remaining']);
this.timeDifference = Date.now() - new Date(res.headers.date).getTime();
}
if (err) {
if (err.status === 429) {
this.queue.unshift(item);
this.restManager.client.setTimeout(() => {
this.globalLimit = false;
resolve();
}, Number(res.headers['retry-after']) + this.restManager.client.options.restTimeOffset);
if (res.headers['x-ratelimit-global']) this.globalLimit = true;
} else {
item.reject(err);
resolve(err);
}
item.request.gen().end((err, res) => {
if (res && res.headers) {
this.requestLimit = Number(res.headers['x-ratelimit-limit']);
this.requestResetTime = Number(res.headers['x-ratelimit-reset']) * 1000;
this.requestRemaining = Number(res.headers['x-ratelimit-remaining']);
this.timeDifference = Date.now() - new Date(res.headers.date).getTime();
}
if (err) {
if (err.status === 429) {
this.queue.unshift(item);
this.restManager.client.setTimeout(() => {
this.globalLimit = false;
resolve();
}, Number(res.headers['retry-after']) + this.restManager.client.options.restTimeOffset);
if (res.headers['x-ratelimit-global']) this.globalLimit = true;
} else {
this.globalLimit = false;
const data = res && res.body ? res.body : {};
item.resolve(data);
if (this.requestRemaining === 0) {
this.restManager.client.setTimeout(
item.reject(err.status === 400 ? new DiscordAPIError(res.body) : err);
resolve(err);
}
} else {
this.globalLimit = false;
const data = res && res.body ? res.body : {};
item.resolve(data);
if (this.requestRemaining === 0) {
this.restManager.client.setTimeout(
() => resolve(data),
this.requestResetTime - Date.now() + this.timeDifference + this.restManager.client.options.restTimeOffset
);
} else {
resolve(data);
}
} else {
resolve(data);
}
});
}
});
});
}

Expand Down
2 changes: 1 addition & 1 deletion src/util/Constants.js
Original file line number Diff line number Diff line change
Expand Up @@ -64,7 +64,7 @@ exports.DefaultOptions = {
version: 6,
},
http: {
version: 6,
version: 7,
host: 'https://discordapp.com',
cdn: 'https://cdn.discordapp.com',
},
Expand Down

0 comments on commit 78d05b1

Please sign in to comment.