mirror of
https://github.com/discordjs/discord.js.git
synced 2026-03-09 16:13:31 +01:00
313 lines
8.4 KiB
JavaScript
313 lines
8.4 KiB
JavaScript
"use strict";
|
|
|
|
function _classCallCheck(instance, Constructor) { if (!(instance instanceof Constructor)) { throw new TypeError("Cannot call a class as a function"); } }
|
|
|
|
var EventEmitter = require("events");
|
|
var request = require("superagent");
|
|
var WebSocket = require("ws");
|
|
var ConnectionState = require("./ConnectionState.js");
|
|
|
|
var Constants = require("../Constants.js"),
|
|
Endpoints = Constants.Endpoints,
|
|
PacketType = Constants.PacketType;
|
|
|
|
var Cache = require("../Util/Cache.js");
|
|
var Resolver = require("./Resolver/Resolver.js");
|
|
|
|
var User = require("../Structures/User.js"),
|
|
Channel = require("../Structures/Channel.js"),
|
|
TextChannel = require("../Structures/TextChannel.js"),
|
|
VoiceChannel = require("../Structures/VoiceChannel.js"),
|
|
PMChannel = require("../Structures/PMChannel.js"),
|
|
Server = require("../Structures/Server.js"),
|
|
Message = require("../Structures/Message.js");
|
|
|
|
var zlib;
|
|
|
|
var InternalClient = (function () {
|
|
function InternalClient(discordClient) {
|
|
_classCallCheck(this, InternalClient);
|
|
|
|
this.client = discordClient;
|
|
this.state = ConnectionState.IDLE;
|
|
this.websocket = null;
|
|
|
|
if (this.client.options.compress) {
|
|
zlib = require("zlib");
|
|
}
|
|
|
|
// creates 4 caches with discriminators based on ID
|
|
this.users = new Cache();
|
|
this.channels = new Cache();
|
|
this.servers = new Cache();
|
|
this.private_channels = new Cache();
|
|
this.resolver = new Resolver(this);
|
|
}
|
|
|
|
// def login
|
|
|
|
InternalClient.prototype.login = function login(email, password) {
|
|
var self = this;
|
|
var client = self.client;
|
|
return new Promise(function (resolve, reject) {
|
|
if (self.state === ConnectionState.DISCONNECTED || self.state === ConnectionState.IDLE) {
|
|
|
|
self.state = ConnectionState.LOGGING_IN;
|
|
|
|
request.post(Endpoints.LOGIN).send({ email: email, password: password }).end(function (err, res) {
|
|
|
|
if (err) {
|
|
self.state = ConnectionState.DISCONNECTED;
|
|
self.websocket = null;
|
|
client.emit("disconnected");
|
|
reject(new Error(err.response.text));
|
|
} else {
|
|
var token = res.body.token;
|
|
self.state = ConnectionState.LOGGED_IN;
|
|
self.token = token;
|
|
self.email = email;
|
|
self.password = password;
|
|
|
|
self.getGateway().then(function (url) {
|
|
|
|
self.createWS(url);
|
|
resolve(token);
|
|
})["catch"](function (e) {
|
|
self.state = ConnectionState.DISCONNECTED;
|
|
client.emit("disconnected");
|
|
reject(new Error(err.response.text));
|
|
});
|
|
}
|
|
});
|
|
} else {
|
|
reject(new Error("already logging in/logged in/ready!"));
|
|
}
|
|
});
|
|
};
|
|
|
|
// def logout
|
|
|
|
InternalClient.prototype.logout = function logout() {
|
|
var _this = this;
|
|
|
|
var self = this;
|
|
return new Promise(function (resolve, reject) {
|
|
|
|
if (self.state === ConnectionState.DISCONNECTED || self.state === ConnectionState.IDLE) {
|
|
reject(new Error("Client is not logged in!"));
|
|
return;
|
|
}
|
|
|
|
request.post(Endpoints.LOGOUT).set("authorization", self.token).end(function (err, res) {
|
|
if (err) {
|
|
reject(new Error(err.response.text));
|
|
} else {
|
|
if (_this.websocket) {
|
|
_this.websocket.close();
|
|
_this.websocket = null;
|
|
}
|
|
self.token = null;
|
|
self.email = null;
|
|
self.password = null;
|
|
self.state = ConnectionState.DISCONNECTED;
|
|
resolve();
|
|
}
|
|
});
|
|
});
|
|
};
|
|
|
|
// def startPM
|
|
|
|
InternalClient.prototype.startPM = function startPM(resUser) {
|
|
var self = this;
|
|
return new Promise(function (resolve, reject) {
|
|
var user = self.resolver.resolveUser(resUser);
|
|
|
|
if (user) {
|
|
|
|
// start the PM
|
|
request.post("" + Endpoints.USER_CHANNELS(user.id)).set("authorization", self.token).end(function (err, res) {
|
|
if (err) {
|
|
reject(new Error(err.response.text));
|
|
} else {
|
|
resolve(self.private_channels.add(new PMChannel(res.body, self.client)));
|
|
}
|
|
});
|
|
} else {
|
|
reject(new Error("Unable to resolve resUser to a User"));
|
|
}
|
|
});
|
|
};
|
|
|
|
// def getGateway
|
|
|
|
InternalClient.prototype.getGateway = function getGateway() {
|
|
var self = this;
|
|
return new Promise(function (resolve, reject) {
|
|
|
|
request.get(Endpoints.GATEWAY).set("authorization", self.token).end(function (err, res) {
|
|
if (err) reject(err);else resolve(res.body.url);
|
|
});
|
|
});
|
|
};
|
|
|
|
// def sendMessage
|
|
|
|
InternalClient.prototype.sendMessage = function sendMessage(where, _content) {
|
|
var options = arguments.length <= 2 || arguments[2] === undefined ? {} : arguments[2];
|
|
|
|
var self = this;
|
|
return new Promise(function (resolve, reject) {
|
|
|
|
self.resolver.resolveChannel(where).then(next)["catch"](function (e) {
|
|
return reject(new Error("Error resolving destination"));
|
|
});
|
|
|
|
function next(destination) {
|
|
//var destination;
|
|
var content = self.resolver.resolveString(_content);
|
|
var mentions = self.resolver.resolveMentions(content);
|
|
|
|
request.post(Endpoints.CHANNEL_MESSAGES(destination.id)).set("authorization", self.token).send({
|
|
content: content,
|
|
mentions: mentions,
|
|
tts: options.tts
|
|
}).end(function (err, res) {
|
|
if (err) {
|
|
reject(new Error(err.response.text));
|
|
} else {
|
|
|
|
resolve(destination.messages.add(new Message(res.body, destination, self.client)));
|
|
}
|
|
});
|
|
}
|
|
});
|
|
};
|
|
|
|
InternalClient.prototype.deleteMessage = function deleteMessage(_message) {
|
|
var options = arguments.length <= 1 || arguments[1] === undefined ? {} : arguments[1];
|
|
|
|
var self = this;
|
|
return new Promise(function (resolve, reject) {
|
|
|
|
var message = self.resolver.resolveMessage(_message);
|
|
|
|
if (message) {
|
|
var deleteMsg = function deleteMsg() {
|
|
request.del(Endpoints.CHANNEL_MESSAGE(message.channel.id, message.id)).set("authorization", self.token).end(function (err, res) {
|
|
if (err) {
|
|
reject(new Error(err.response.text));
|
|
} else {
|
|
message.channel.messages.remove(message);
|
|
resolve();
|
|
}
|
|
});
|
|
};
|
|
|
|
if (options.wait) {
|
|
setTimeout(deleteMsg, options.wait);
|
|
} else {
|
|
deleteMsg();
|
|
}
|
|
} else {
|
|
reject(new Error("Supplied message did not resolve to a message!"));
|
|
}
|
|
});
|
|
};
|
|
|
|
InternalClient.prototype.sendWS = function sendWS(object) {
|
|
this.websocket.send(JSON.stringify(object));
|
|
};
|
|
|
|
InternalClient.prototype.createWS = function createWS(url) {
|
|
var self = this;
|
|
var client = self.client;
|
|
|
|
if (this.websocket) return false;
|
|
|
|
this.websocket = new WebSocket(url);
|
|
|
|
this.websocket.onopen = function () {
|
|
|
|
self.sendWS({
|
|
op: 2,
|
|
d: {
|
|
token: self.token,
|
|
v: 3,
|
|
compress: self.client.options.compress,
|
|
properties: {
|
|
"$os": "discord.js",
|
|
"$browser": "discord.js",
|
|
"$device": "discord.js",
|
|
"$referrer": "discord.js",
|
|
"$referring_domain": "discord.js"
|
|
}
|
|
}
|
|
});
|
|
};
|
|
|
|
this.websocket.onclose = function () {
|
|
self.websocket = null;
|
|
self.state = ConnectionState.DISCONNECTED;
|
|
client.emit("disconnected");
|
|
};
|
|
|
|
this.websocket.onmessage = function (e) {
|
|
|
|
if (e.type === "Binary") {
|
|
if (!zlib) zlib = require("zlib");
|
|
e.data = zlib.inflateSync(e.data).toString();
|
|
}
|
|
|
|
var packet, data;
|
|
try {
|
|
packet = JSON.parse(e.data);
|
|
data = packet.d;
|
|
} catch (e) {
|
|
client.emit("error", e);
|
|
return;
|
|
}
|
|
|
|
client.emit("raw", packet);
|
|
|
|
switch (packet.t) {
|
|
|
|
case PacketType.READY:
|
|
var startTime = Date.now();
|
|
self.users.add(new User(data.user, client));
|
|
data.guilds.forEach(function (server) {
|
|
self.servers.add(new Server(server, client));
|
|
});
|
|
data.private_channels.forEach(function (pm) {
|
|
self.private_channels.add(new PMChannel(pm, client));
|
|
});
|
|
self.state = ConnectionState.READY;
|
|
|
|
setInterval(function () {
|
|
return self.sendWS({ op: 1, d: Date.now() });
|
|
}, data.heartbeat_interval);
|
|
|
|
client.emit("ready");
|
|
client.emit("debug", "ready packet took " + (Date.now() - startTime) + "ms to process");
|
|
client.emit("debug", "ready with " + self.servers.length + " servers, " + self.channels.length + " channels and " + self.users.length + " users cached.");
|
|
break;
|
|
|
|
case PacketType.MESSAGE_CREATE:
|
|
// format: https://discordapi.readthedocs.org/en/latest/reference/channels/messages.html#message-format
|
|
var channel = self.channels.get("id", data.channel_id);
|
|
if (channel) {
|
|
var msg = channel.messages.add(new Message(data, channel, client));
|
|
client.emit("message", msg);
|
|
} else {
|
|
client.emit("warn", "message created but channel is not cached");
|
|
}
|
|
break;
|
|
|
|
}
|
|
};
|
|
};
|
|
|
|
return InternalClient;
|
|
})();
|
|
|
|
module.exports = InternalClient; |