mirror of
https://github.com/discordjs/discord.js.git
synced 2026-03-12 01:23:31 +01:00
Added sendFile queue
This commit is contained in:
103
lib/Client.js
103
lib/Client.js
@@ -62,7 +62,7 @@ var Client = (function () {
|
||||
this.pmChannelCache = [];
|
||||
this.readyTime = null;
|
||||
this.checkingQueue = {};
|
||||
this.messageQueue = {};
|
||||
this.queue = {};
|
||||
}
|
||||
|
||||
_createClass(Client, [{
|
||||
@@ -339,7 +339,7 @@ var Client = (function () {
|
||||
|
||||
var self = this;
|
||||
|
||||
return new Promise(function (resolve, reject) {
|
||||
var prom = new Promise(function (resolve, reject) {
|
||||
if (timeout) {
|
||||
setTimeout(remove, timeout);
|
||||
} else {
|
||||
@@ -358,6 +358,8 @@ var Client = (function () {
|
||||
});
|
||||
}
|
||||
});
|
||||
|
||||
return prom;
|
||||
}
|
||||
}, {
|
||||
key: "updateMessage",
|
||||
@@ -563,23 +565,33 @@ var Client = (function () {
|
||||
self.resolveDestination(destination).then(send)["catch"](error);
|
||||
|
||||
function send(destination) {
|
||||
request.post(Endpoints.CHANNELS + "/" + destination + "/messages").set("authorization", self.token).attach("file", fstream, fileName).end(function (err, res) {
|
||||
|
||||
if (err) {
|
||||
error(err);
|
||||
} else {
|
||||
|
||||
var chann = self.getChannel("id", destination);
|
||||
if (chann) {
|
||||
var msg = chann.addMessage(new Message(res.body, chann, [], self.user));
|
||||
callback(null, msg);
|
||||
resolve(msg);
|
||||
}
|
||||
if (~self.options.queue.indexOf("send")) {
|
||||
//queue send file too
|
||||
if (!self.queue[destination]) {
|
||||
self.queue[destination] = [];
|
||||
}
|
||||
});
|
||||
|
||||
self.queue[destination].push({
|
||||
action: "sendFile",
|
||||
attachment: fstream,
|
||||
attachmentName: fileName,
|
||||
then: good,
|
||||
error: bad
|
||||
});
|
||||
|
||||
self.checkQueue(destination);
|
||||
} else {
|
||||
//not queue
|
||||
self._sendFile(destination, fstream, fileName).then(good)["catch"](bad);
|
||||
}
|
||||
}
|
||||
|
||||
function error(err) {
|
||||
function good(msg) {
|
||||
callback(null, msg);
|
||||
resolve(msg);
|
||||
}
|
||||
|
||||
function bad(err) {
|
||||
callback(err);
|
||||
reject(err);
|
||||
}
|
||||
@@ -605,18 +617,18 @@ var Client = (function () {
|
||||
}
|
||||
|
||||
function send(destination) {
|
||||
if (~self.options.queue.indexOf("sendMessage")) {
|
||||
if (~self.options.queue.indexOf("send")) {
|
||||
//we're QUEUEING messages, so sending them sequentially based on servers.
|
||||
if (!self.messageQueue[destination]) {
|
||||
self.messageQueue[destination] = [];
|
||||
if (!self.queue[destination]) {
|
||||
self.queue[destination] = [];
|
||||
}
|
||||
|
||||
self.messageQueue[destination].push({
|
||||
self.queue[destination].push({
|
||||
action: "sendMessage",
|
||||
content: message,
|
||||
mentions: mentions,
|
||||
then: [mgood],
|
||||
error: [mbad]
|
||||
then: mgood,
|
||||
error: mbad
|
||||
});
|
||||
|
||||
self.checkQueue(destination);
|
||||
@@ -1395,6 +1407,29 @@ var Client = (function () {
|
||||
});
|
||||
});
|
||||
}
|
||||
}, {
|
||||
key: "_sendFile",
|
||||
value: function _sendFile(destination, attachment) {
|
||||
var attachmentName = arguments.length <= 2 || arguments[2] === undefined ? "DEFAULT BECAUSE YOU DIDN'T SPECIFY WHY.png" : arguments[2];
|
||||
|
||||
var self = this;
|
||||
|
||||
return new Promise(function (resolve, reject) {
|
||||
request.post(Endpoints.CHANNELS + "/" + destination + "/messages").set("authorization", self.token).attach("file", attachment, attachmentName).end(function (err, res) {
|
||||
|
||||
if (err) {
|
||||
reject(err);
|
||||
} else {
|
||||
|
||||
var chann = self.getChannel("id", destination);
|
||||
if (chann) {
|
||||
var msg = chann.addMessage(new Message(res.body, chann, [], self.user));
|
||||
resolve(msg);
|
||||
}
|
||||
}
|
||||
});
|
||||
});
|
||||
}
|
||||
}, {
|
||||
key: "checkQueue",
|
||||
value: function checkQueue(channelID) {
|
||||
@@ -1405,21 +1440,33 @@ var Client = (function () {
|
||||
if (!this.checkingQueue[channelID]) {
|
||||
(function () {
|
||||
var doNext = function doNext() {
|
||||
if (self.messageQueue[channelID].length === 0) {
|
||||
if (self.queue[channelID].length === 0) {
|
||||
done();
|
||||
return;
|
||||
}
|
||||
var queuedEvent = self.messageQueue[channelID][0];
|
||||
var queuedEvent = self.queue[channelID][0];
|
||||
switch (queuedEvent.action) {
|
||||
case "sendMessage":
|
||||
var msgToSend = queuedEvent;
|
||||
self._sendMessage(channelID, msgToSend.content, msgToSend.mentions).then(function (msg) {
|
||||
msgToSend.then[0](msg);
|
||||
self.messageQueue[channelID].shift();
|
||||
msgToSend.then(msg);
|
||||
self.queue[channelID].shift();
|
||||
doNext();
|
||||
})["catch"](function (err) {
|
||||
msgToSend["catch"][0](err);
|
||||
self.messageQueue[channelID].shift();
|
||||
msgToSend["catch"](err);
|
||||
self.queue[channelID].shift();
|
||||
doNext();
|
||||
});
|
||||
break;
|
||||
case "sendFile":
|
||||
var fileToSend = queuedEvent;
|
||||
self._sendFile(channelID, fileToSend.attachment, fileToSend.attachmentName).then(function (msg) {
|
||||
fileToSend.then(msg);
|
||||
self.queue[channelID].shift();
|
||||
doNext();
|
||||
})["catch"](function (err) {
|
||||
fileToSend["catch"](err);
|
||||
self.queue[channelID].shift();
|
||||
doNext();
|
||||
});
|
||||
break;
|
||||
|
||||
Reference in New Issue
Block a user