Files
discord.js/lib/Client/InternalClient.js
2015-11-22 15:12:39 +00:00

1740 lines
47 KiB
JavaScript

"use strict";
var _createClass = (function () { function defineProperties(target, props) { for (var i = 0; i < props.length; i++) { var descriptor = props[i]; descriptor.enumerable = descriptor.enumerable || false; descriptor.configurable = true; if ("value" in descriptor) descriptor.writable = true; Object.defineProperty(target, descriptor.key, descriptor); } } return function (Constructor, protoProps, staticProps) { if (protoProps) defineProperties(Constructor.prototype, protoProps); if (staticProps) defineProperties(Constructor, staticProps); return Constructor; }; })();
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"),
Role = require("../Structures/Role.js"),
Invite = require("../Structures/Invite.js"),
VoiceConnection = require("../Voice/VoiceConnection.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.typingIntervals = [];
this.voiceConnection = null;
this.resolver = new Resolver(this);
this.readyTime = null;
this.messageAwaits = {};
}
_createClass(InternalClient, [{
key: "leaveVoiceChannel",
//def leaveVoiceChannel
value: function leaveVoiceChannel() {
var self = this;
return new Promise(function (resolve, reject) {
if (self.voiceConnection) {
self.voiceConnection.destroy();
self.voiceConnection = null;
resolve();
} else {
resolve();
}
});
}
//def awaitResponse
}, {
key: "awaitResponse",
value: function awaitResponse(msg) {
var _this = this;
return new Promise(function (resolve, reject) {
msg = _this.resolver.resolveMessage(msg);
if (!msg) {
reject(new Error("message undefined"));
return;
}
var awaitID = msg.channel.id + msg.author.id;
if (!_this.messageAwaits[awaitID]) {
_this.messageAwaits[awaitID] = [];
}
_this.messageAwaits[awaitID].push(resolve);
});
}
//def joinVoiceChannel
}, {
key: "joinVoiceChannel",
value: function joinVoiceChannel(chann) {
var self = this;
return new Promise(function (resolve, reject) {
var channel = self.resolver.resolveVoiceChannel(chann);
if (channel) {
var next = function next() {
var session,
token,
server = channel.server,
endpoint;
var check = function check(m) {
var data = JSON.parse(m);
if (data.t === "VOICE_STATE_UPDATE") {
session = data.d.session_id;
} else if (data.t === "VOICE_SERVER_UPDATE") {
token = data.d.token;
endpoint = data.d.endpoint;
var chan = self.voiceConnection = new VoiceConnection(channel, self.client, session, token, server, endpoint);
chan.on("ready", function () {
return resolve(chan);
});
chan.on("error", reject);
self.client.emit("debug", "removed temporary voice websocket listeners");
self.websocket.removeListener("message", check);
}
};
self.websocket.on("message", check);
self.sendWS({
op: 4,
d: {
"guild_id": server.id,
"channel_id": channel.id,
"self_mute": false,
"self_deaf": false
}
});
};
self.leaveVoiceChannel().then(next);
} else {
reject(new Error("voice channel does not exist"));
}
});
}
// def createServer
}, {
key: "createServer",
value: function createServer(name) {
var region = arguments.length <= 1 || arguments[1] === undefined ? "london" : arguments[1];
var self = this;
return new Promise(function (resolve, reject) {
name = self.resolver.resolveString(name);
request.post(Endpoints.SERVERS).set("authorization", self.token).send({ name: name, region: region }).end(function (err, res) {
if (err) {
reject(new Error(err));
} else {
// valid server, wait until it is cached
var inter = setInterval(function () {
if (self.servers.get("id", res.body.id)) {
clearInterval(inter);
resolve(self.servers.get("id", res.body.id));
}
}, 20);
}
});
});
}
//def joinServer
}, {
key: "joinServer",
value: function joinServer(invite) {
var self = this;
return new Promise(function (resolve, reject) {
invite = self.resolver.resolveInviteID(invite);
if (invite) {
request.post(Endpoints.INVITE(invite)).set("authorization", self.token).end(function (err, res) {
if (err) {
reject(err);
} else {
// valid server, wait until it is received via ws and cached
var inter = setInterval(function () {
if (self.servers.get("id", res.body.guild.id)) {
clearInterval(inter);
resolve(self.servers.get("id", res.body.guild.id));
}
}, 20);
}
});
} else {
reject(new Error("Not a valid invite"));
}
});
}
//def leaveServer
}, {
key: "leaveServer",
value: function leaveServer(srv) {
var self = this;
return new Promise(function (resolve, reject) {
var server = self.resolver.resolveServer(srv);
if (server) {
request.del(Endpoints.SERVER(server.id)).set("authorization", self.token).end(function (err, res) {
if (err) {
reject(new Error(err));
} else {
// remove channels of server then the server
var _iteratorNormalCompletion = true;
var _didIteratorError = false;
var _iteratorError = undefined;
try {
for (var _iterator = server.channels[Symbol.iterator](), _step; !(_iteratorNormalCompletion = (_step = _iterator.next()).done); _iteratorNormalCompletion = true) {
var chan = _step.value;
self.channels.remove(chan);
}
// remove server
} catch (err) {
_didIteratorError = true;
_iteratorError = err;
} finally {
try {
if (!_iteratorNormalCompletion && _iterator.return) {
_iterator.return();
}
} finally {
if (_didIteratorError) {
throw _iteratorError;
}
}
}
self.servers.remove(server);
resolve();
}
});
} else {
reject(new Error("server did not resolve"));
}
});
}
// def login
}, {
key: "login",
value: 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));
} 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));
});
}
});
} else {
reject(new Error("already logging in/logged in/ready!"));
}
});
}
// def logout
}, {
key: "logout",
value: function logout() {
var _this2 = 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));
} else {
if (_this2.websocket) {
_this2.websocket.close();
_this2.websocket = null;
}
self.token = null;
self.email = null;
self.password = null;
self.state = ConnectionState.DISCONNECTED;
resolve();
}
});
});
}
// def startPM
}, {
key: "startPM",
value: 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).send({
recipient_id: user.id
}).end(function (err, res) {
if (err) {
reject(new Error(err));
} 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
}, {
key: "getGateway",
value: 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
}, {
key: "sendMessage",
value: 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 - " + e));
});
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));
} else {
resolve(destination.messages.add(new Message(res.body, destination, self.client)));
}
});
}
});
}
// def deleteMessage
}, {
key: "deleteMessage",
value: 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));
} 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!"));
}
});
}
// def updateMessage
}, {
key: "updateMessage",
value: function updateMessage(msg, _content) {
var options = arguments.length <= 2 || arguments[2] === undefined ? {} : arguments[2];
var self = this;
return new Promise(function (resolve, reject) {
var message = self.resolver.resolveMessage(msg);
if (message) {
var content = self.resolver.resolveString(_content);
var mentions = self.resolver.resolveMentions(content);
request.patch(Endpoints.CHANNEL_MESSAGE(message.channel.id, message.id)).set("authorization", self.token).send({
content: content,
tts: options.tts,
mentions: mentions
}).end(function (err, res) {
if (err) {
reject(new Error(err));
} else {
resolve(message.channel.messages.update(message, new Message(res.body, message.channel, self.client)));
}
});
} else {
reject(new Error("Supplied message did not resolve to a message!"));
}
});
}
// def sendFile
}, {
key: "sendFile",
value: function sendFile(where, _file) {
var name = arguments.length <= 2 || arguments[2] === undefined ? "image.png" : arguments[2];
var self = this;
return new Promise(function (resolve, reject) {
self.resolver.resolveChannel(where).then(next).catch(function (e) {
return reject(new Error("couldn't resolve to channel - " + e));
});
function next(channel) {
var file = self.resolver.resolveFile(_file);
request.post(Endpoints.CHANNEL_MESSAGES(channel.id)).set("authorization", self.token).attach("file", file, name).end(function (err, res) {
if (err) {
reject(new Error(err));
} else {
resolve(channel.messages.add(new Message(res.body, channel, self.client)));
}
});
}
});
}
// def getChannelLogs
}, {
key: "getChannelLogs",
value: function getChannelLogs(_channel) {
var limit = arguments.length <= 1 || arguments[1] === undefined ? 500 : arguments[1];
var options = arguments.length <= 2 || arguments[2] === undefined ? {} : arguments[2];
var self = this;
return new Promise(function (resolve, reject) {
self.resolver.resolveChannel(_channel).then(next).catch(function (e) {
return reject(new Error("couldn't resolve to channel - " + e));
});
function next(channel) {
if (options.before) options.before = self.resolver.resolveMessage(options.before);
if (options.after) options.after = self.resolver.resolveMessage(options.after);
var params = [];
if (options.before) params.push("before=" + options.before.id);
if (options.after) params.push("after=" + options.after.id);
var joinedParams = params.join();
if (joinedParams !== "") joinedParams = "&" + params.join();
request.get(Endpoints.CHANNEL_MESSAGES(channel.id) + "?limit=" + limit + joinedParams).set("authorization", self.token).end(function (err, res) {
if (err) {
reject(new Error(err));
} else {
var logs = [];
res.body.forEach(function (msg) {
logs.push(channel.messages.add(new Message(msg, channel, self.client)));
});
resolve(logs);
}
});
}
});
}
// def getBans
}, {
key: "getBans",
value: function getBans(server) {
var self = this;
return new Promise(function (resolve, reject) {
server = self.resolver.resolveServer(server);
request.get("" + Endpoints.SERVER_BANS(server.id)).set("authorization", self.token).end(function (err, res) {
if (err) {
reject(new Error(err));
} else {
var bans = [];
res.body.forEach(function (ban) {
bans.push(self.users.add(new User(ban.user, self.client)));
});
resolve(bans);
}
});
});
}
// def createChannel
}, {
key: "createChannel",
value: function createChannel(server, name) {
var type = arguments.length <= 2 || arguments[2] === undefined ? "text" : arguments[2];
var self = this;
return new Promise(function (resolve, reject) {
server = self.resolver.resolveServer(server);
request.post(Endpoints.SERVER_CHANNELS(server.id)).set("authorization", self.token).send({
name: name, type: type
}).end(function (err, res) {
if (err) {
reject(err);
} else {
var channel;
if (res.body.type === "text") {
channel = new TextChannel(res.body, self.client, server);
} else {
channel = new VoiceChannel(res.body, self.client, server);
}
resolve(server.channels.add(self.channels.add(channel)));
}
});
});
}
// def deleteChannel
}, {
key: "deleteChannel",
value: function deleteChannel(_channel) {
var self = this;
return new Promise(function (resolve, reject) {
self.resolver.resolveChannel(_channel).then(next).catch(reject);
function next(channel) {
request.del(Endpoints.CHANNEL(channel.id)).set("authorization", self.token).end(function (err, res) {
if (err) {
reject(err);
} else {
channel.server.channels.remove(channel);
self.channels.remove(channel);
resolve();
}
});
}
});
}
// def banMember
}, {
key: "banMember",
value: function banMember(user, server) {
var length = arguments.length <= 2 || arguments[2] === undefined ? 1 : arguments[2];
var self = this;
return new Promise(function (resolve, reject) {
user = self.resolver.resolveUser(user);
server = self.resolver.resolveServer(server);
request.put(Endpoints.SERVER_BANS(server.id) + "/" + user.id + "/?delete-message-days=" + length).set("authorization", self.token).end(function (err, res) {
if (err) {
reject(err);
} else {
resolve();
}
});
});
}
// def unbanMember
}, {
key: "unbanMember",
value: function unbanMember(user, server) {
var self = this;
return new Promise(function (resolve, reject) {
server = self.resolver.resolveServer(server);
user = self.resolver.resolveUser(user);
request.del(Endpoints.SERVER_BANS(server.id) + "/" + user.id).set("authorization", self.token).end(function (err, res) {
if (err) {
reject(err);
} else {
resolve();
}
});
});
}
// def kickMember
}, {
key: "kickMember",
value: function kickMember(user, server) {
var self = this;
return new Promise(function (resolve, reject) {
user = self.resolver.resolveUser(user);
server = self.resolver.resolveServer(server);
request.del(Endpoints.SERVER_MEMBERS(server.id) + "/" + user.id).set("authorization", self.token).end(function (err, res) {
if (err) {
reject(err);
} else {
resolve();
}
});
});
}
// def createRole
}, {
key: "createRole",
value: function createRole(server, data) {
var self = this;
return new Promise(function (resolve, reject) {
server = self.resolver.resolveServer(server);
request.post(Endpoints.SERVER_ROLES(server.id)).set("authorization", self.token).end(function (err, res) {
if (err) {
reject(err);
} else {
var role = server.roles.add(new Role(res.body, server, self.client));
if (data) {
self.updateRole(role, data).then(resolve).catch(reject);
} else {
resolve(role);
}
}
});
});
}
// def updateRole
}, {
key: "updateRole",
value: function updateRole(role, data) {
var self = this;
data = data || {};
data.permissions = data.permissions || [];
return new Promise(function (resolve, reject) {
var server = self.resolver.resolveServer(role.server);
var permissions = 0;
var _iteratorNormalCompletion2 = true;
var _didIteratorError2 = false;
var _iteratorError2 = undefined;
try {
for (var _iterator2 = data.permissions[Symbol.iterator](), _step2; !(_iteratorNormalCompletion2 = (_step2 = _iterator2.next()).done); _iteratorNormalCompletion2 = true) {
var perm = _step2.value;
if (perm instanceof String || typeof perm === "string") {
permissions |= Constants.Permissions[perm] || 0;
} else {
permissions |= perm;
}
}
} catch (err) {
_didIteratorError2 = true;
_iteratorError2 = err;
} finally {
try {
if (!_iteratorNormalCompletion2 && _iterator2.return) {
_iterator2.return();
}
} finally {
if (_didIteratorError2) {
throw _iteratorError2;
}
}
}
data.color = data.color || 0;
request.patch(Endpoints.SERVER_ROLES(server.id) + "/" + role.id).set("authorization", self.token).send({
color: data.color || role.color,
hoist: data.hoist || role.hoist,
name: data.name || role.name,
permissions: permissions
}).end(function (err, res) {
if (err) {
reject(err);
} else {
var nrole = new Role(res.body, server, self.client);
resolve(server.roles.update(role, nrole));
}
});
});
}
// def deleteRole
}, {
key: "deleteRole",
value: function deleteRole(role) {
var self = this;
return new Promise(function (resolve, reject) {
request.del(Endpoints.SERVER_ROLES(role.server.id) + "/" + role.id).set("authorization", self.token).end(function (err, res) {
if (err) {
reject(err);
} else {
resolve();
// the ws cache will handle it
// role.server.roles.remove(role);
}
});
});
}
//def addMemberToRole
}, {
key: "addMemberToRole",
value: function addMemberToRole(member, role) {
var self = this;
return new Promise(function (resolve, reject) {
member = self.resolver.resolveUser(member);
if (!member || !role) {
reject(new Error("member/role not in server"));
return;
}
if (role.server.memberMap[member.id]) {
var roleIDS = role.server.memberMap[member.id].roles.map(function (r) {
return r.id;
}).concat(role.id);
request.patch(Endpoints.SERVER_MEMBERS(role.server.id) + "/" + member.id).set("authorization", self.token).send({
roles: roleIDS
}).end(function (err) {
if (err) {
reject(err);
} else {
resolve();
}
});
} else {
reject(new Error("member not in server"));
}
});
}
//def removeMemberFromRole
}, {
key: "removeMemberFromRole",
value: function removeMemberFromRole(member, role) {
var self = this;
return new Promise(function (resolve, reject) {
member = self.resolver.resolveUser(member);
if (!member || !role) {
reject(new Error("member/role not in server"));
return;
}
if (role.server.memberMap[member.id]) {
var roleIDS = role.server.memberMap[member.id].roles.map(function (r) {
return r.id;
});
for (var item in roleIDS) {
if (roleIDS[item] === role.id) {
roleIDS.splice(item, 1);
}
}
request.patch(Endpoints.SERVER_MEMBERS(role.server.id) + "/" + member.id).set("authorization", self.token).send({
roles: roleIDS
}).end(function (err) {
if (err) {
reject(err);
} else {
resolve();
}
});
} else {
reject(new Error("member not in server"));
}
});
}
// def createInvite
}, {
key: "createInvite",
value: function createInvite(chanServ, options) {
var self = this;
return new Promise(function (resolve, reject) {
if (chanServ instanceof Channel) {
// do something
} else if (chanServ instanceof Server) {
// do something
} else {
chanServ = self.resolver.resolveServer(chanServ) || self.resolver.resolveChannel(chanServ);
}
if (!chanServ) {
reject(new Error("couldn't resolve where"));
return;
}
if (!options) {
options = { validate: null };
} else {
options.max_age = options.maxAge || 0;
options.max_uses = options.maxUses || 0;
options.temporary = options.temporary || false;
options.xkcdpass = options.xkcd || false;
}
var epoint;
if (chanServ instanceof Channel) {
epoint = Endpoints.CHANNEL_INVITES(chanServ.id);
} else {
epoint = Endpoints.SERVER_INVITES(chanServ.id);
}
request.post(epoint).set("authorization", self.token).send(options).end(function (err, res) {
if (err) {
reject(err);
} else {
resolve(new Invite(res.body, self.channels.get("id", res.body.channel.id), self.client));
}
});
});
}
//def deleteInvite
}, {
key: "deleteInvite",
value: function deleteInvite(invite) {
var self = this;
return new Promise(function (resolve, reject) {
invite = self.resolver.resolveInviteID(invite);
if (invite) {
request.del(Endpoints.INVITE(invite)).set("authorization", self.token).end(function (err, res) {
if (err) {
reject(err);
} else {
resolve();
}
});
} else {
reject(new Error("Not a valid invite"));
}
});
}
//def overwritePermissions
}, {
key: "overwritePermissions",
value: function overwritePermissions(channel, role, updated) {
var self = this;
return new Promise(function (resolve, reject) {
channel = self.resolver.resolveChannel(channel).catch(reject).then(next);
function next(channel) {
var user;
if (role instanceof User) {
user = role;
}
var data = {};
data.allow = 0;
data.deny = 0;
updated.allow = updated.allow || [];
updated.deny = updated.deny || [];
if (role instanceof Role) {
data.id = role.id;
data.type = "role";
} else if (user) {
data.id = user.id;
data.type = "member";
} else {
reject(new Error("role incorrect"));
return;
}
for (var perm in updated) {
if (updated[perm]) {
if (perm instanceof String || typeof perm === "string") {
data.allow |= Constants.Permissions[perm] || 0;
} else {
data.allow |= perm;
}
} else {
if (perm instanceof String || typeof perm === "string") {
data.deny |= Constants.Permissions[perm] || 0;
} else {
data.deny |= perm;
}
}
}
request.put(Endpoints.CHANNEL_PERMISSIONS(channel.id) + "/" + data.id).set("authorization", self.token).send(data).end(function (err) {
if (err) {
reject(err);
} else {
resolve();
}
});
}
});
}
//def setStatus
}, {
key: "setStatus",
value: function setStatus(idleStatus, gameID) {
var self = this;
self.idleStatus = idleStatus || self.idleStatus || null;
if (idleStatus) {
if (idleStatus == "online" || idleStatus == "here" || idleStatus == "available") {
self.idleStatus = null;
}
}
self.gameID = self.resolver.resolveGameID(gameID) || self.gameID || null;
return new Promise(function (resolve, reject) {
var packet = {
op: 3,
d: {
idle_since: self.idleStatus,
game_id: self.gameID
}
};
if (self.idleStatus == "idle" || self.idleStatus == "away") {
packet.d.idle_since = Date.now();
}
self.sendWS(packet);
resolve();
});
}
//def sendTyping
}, {
key: "sendTyping",
value: function sendTyping(channel) {
var self = this;
return new Promise(function (resolve, reject) {
self.resolver.resolveChannel(channel).then(next).catch(reject);
function next(channel) {
request.post(Endpoints.CHANNEL(channel.id) + "/typing").set("authorization", self.token).end(function (err, res) {
if (err) {
reject(err);
} else {
resolve();
}
});
}
});
}
//def startTyping
}, {
key: "startTyping",
value: function startTyping(channel) {
var self = this;
return new Promise(function (resolve, reject) {
self.resolver.resolveChannel(channel).then(next).catch(reject);
function next(channel) {
if (self.typingIntervals[channel.id]) {
// typing interval already exists, leave it alone
reject(new Error("Already typing in that channel"));
return;
}
self.sendTyping(channel);
self.typingIntervals[channel.id] = setInterval(function () {
return self.sendTyping(channel);
}, 4000);
}
});
}
//def stopTyping
}, {
key: "stopTyping",
value: function stopTyping(channel) {
var self = this;
return new Promise(function (resolve, reject) {
self.resolver.resolveChannel(channel).then(next).catch(reject);
function next(channel) {
if (!self.typingIntervals[channel.id]) {
// typing interval doesn't exist
reject(new Error("Not typing in that channel"));
return;
}
clearInterval(self.typingIntervals[channel.id]);
self.typingIntervals[channel.id] = false;
}
});
}
//def updateDetails
}, {
key: "updateDetails",
value: function updateDetails(data) {
var self = this;
return new Promise(function (resolve, reject) {
request.patch(Endpoints.ME).set("authorization", self.token).send({
avatar: self.resolver.resolveToBase64(data.avatar) || self.user.avatar,
email: data.email || self.email,
new_password: data.newPassword || null,
password: data.password || self.password,
username: data.username || self.user.username
}).end(function (err) {
if (err) {
reject(err);
} else {
resolve();
}
});
});
}
//def setAvatar
}, {
key: "setAvatar",
value: function setAvatar(avatar) {
return this.updateDetails({ avatar: avatar });
}
//def setUsername
}, {
key: "setUsername",
value: function setUsername(username) {
return this.updateDetails({ username: username });
}
//def setTopic
}, {
key: "setTopic",
value: function setTopic(chann) {
var topic = arguments.length <= 1 || arguments[1] === undefined ? "" : arguments[1];
var self = this;
return new Promise(function (resolve, reject) {
self.resolver.resolveChannel(chann).then(next).catch(reject);
function next(channel) {
request.patch(Endpoints.CHANNEL(channel.id)).set("authorization", self.token).send({
name: channel.name,
position: channel.position,
topic: topic
}).end(function (err, res) {
if (err) {
reject(err);
} else {
channel.topic = res.body.topic;
resolve();
}
});
}
});
}
//def setChannelName
}, {
key: "setChannelName",
value: function setChannelName(chann) {
var name = arguments.length <= 1 || arguments[1] === undefined ? "discordjs_is_the_best" : arguments[1];
var self = this;
return new Promise(function (resolve, reject) {
self.resolver.resolveChannel(chann).then(next).catch(reject);
function next(channel) {
request.patch(Endpoints.CHANNEL(channel.id)).set("authorization", self.token).send({
name: name,
position: channel.position,
topic: channel.topic
}).end(function (err, res) {
if (err) {
reject(err);
} else {
channel.name = res.body.name;
resolve();
}
});
}
});
}
//def setChannelNameAndTopic
}, {
key: "setChannelNameAndTopic",
value: function setChannelNameAndTopic(chann) {
var name = arguments.length <= 1 || arguments[1] === undefined ? "discordjs_is_the_best" : arguments[1];
var topic = arguments.length <= 2 || arguments[2] === undefined ? "" : arguments[2];
var self = this;
return new Promise(function (resolve, reject) {
self.resolver.resolveChannel(chann).then(next).catch(reject);
function next(channel) {
request.patch(Endpoints.CHANNEL(channel.id)).set("authorization", self.token).send({
name: name,
position: channel.position,
topic: topic
}).end(function (err, res) {
if (err) {
reject(err);
} else {
channel.name = res.body.name;
channel.topic = res.body.topic;
resolve();
}
});
}
});
}
//def updateChannel
}, {
key: "updateChannel",
value: function updateChannel(chann, data) {
return this.setChannelNameAndTopic(chann, data.name, data.topic);
}
//def ack
}, {
key: "ack",
value: function ack(msg) {
var self = this;
return new Promise(function (resolve, reject) {
msg = self.resolver.resolveMessage(msg);
if (msg) {
request.post(Endpoints.CHANNEL_MESSAGE(msg.channel.id, msg.id) + "/ack").set("authorization", self.token).end(function (err) {
if (err) {
reject(err);
} else {
resolve();
}
});
} else {
reject(new Error("Message does not exist"));
}
});
}
}, {
key: "sendWS",
value: function sendWS(object) {
if (this.websocket) this.websocket.send(JSON.stringify(object));
}
}, {
key: "createWS",
value: 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.onerror = function (e) {
client.emit("error", e);
};
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.user = 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.");
self.readyTime = Date.now();
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) || self.private_channels.get("id", data.channel_id);
if (channel) {
var msg = channel.messages.add(new Message(data, channel, client));
if (self.messageAwaits[channel.id + msg.author.id]) {
self.messageAwaits[channel.id + msg.author.id].map(function (fn) {
return fn(msg);
});
self.messageAwaits[channel.id + msg.author.id] = null;
client.emit("message", msg, true); //2nd param is isAwaitedMessage
} else {
client.emit("message", msg);
}
self.ack(msg);
} else {
client.emit("warn", "message created but channel is not cached");
}
break;
case PacketType.MESSAGE_DELETE:
// format https://discordapi.readthedocs.org/en/latest/reference/channels/messages.html#message-delete
var channel = self.channels.get("id", data.channel_id) || self.private_channels.get("id", data.channel_id);
if (channel) {
// potentially blank
var msg = channel.messages.get("id", data.id);
client.emit("messageDeleted", msg);
if (msg) {
channel.messages.remove(msg);
}
} else {
client.emit("warn", "message was deleted but channel is not cached");
}
break;
case PacketType.MESSAGE_UPDATE:
// format https://discordapi.readthedocs.org/en/latest/reference/channels/messages.html#message-format
var channel = self.channels.get("id", data.channel_id) || self.private_channels.get("id", data.channel_id);
if (channel) {
// potentially blank
var msg = channel.messages.get("id", data.id);
if (msg) {
// old message exists
data.nonce = data.nonce || msg.nonce;
data.attachments = data.attachments || msg.attachments;
data.tts = data.tts || msg.tts;
data.embeds = data.embeds || msg.embeds;
data.timestamp = data.timestamp || msg.timestamp;
data.mention_everyone = data.mention_everyone || msg.everyoneMentioned;
data.content = data.content || msg.content;
data.mentions = data.mentions || msg.mentions;
data.author = data.author || msg.author;
var nmsg = channel.messages.update(msg, new Message(data, channel, client));
client.emit("messageUpdated", nmsg, msg);
}
} else {
client.emit("warn", "message was updated but channel is not cached");
}
break;
case PacketType.SERVER_CREATE:
var server = self.servers.get("id", data.id);
if (!server) {
self.servers.add(new Server(data, client));
client.emit("serverCreated", server);
}
break;
case PacketType.SERVER_DELETE:
var server = self.servers.get("id", data.id);
if (server) {
var _iteratorNormalCompletion3 = true;
var _didIteratorError3 = false;
var _iteratorError3 = undefined;
try {
for (var _iterator3 = server.channels[Symbol.iterator](), _step3; !(_iteratorNormalCompletion3 = (_step3 = _iterator3.next()).done); _iteratorNormalCompletion3 = true) {
var channel = _step3.value;
self.channels.remove(channel);
}
} catch (err) {
_didIteratorError3 = true;
_iteratorError3 = err;
} finally {
try {
if (!_iteratorNormalCompletion3 && _iterator3.return) {
_iterator3.return();
}
} finally {
if (_didIteratorError3) {
throw _iteratorError3;
}
}
}
self.servers.remove(server);
client.emit("serverDeleted", server);
} else {
client.emit("warn", "server was deleted but it was not in the cache");
}
break;
case PacketType.SERVER_UPDATE:
var server = self.servers.get("id", data.id);
if (server) {
// server exists
data.members = data.members || [];
data.channels = data.channels || [];
var newserver = new Server(data, self);
newserver.members = server.members;
newserver.memberMap = server.memberMap;
newserver.channels = server.channels;
if (newserver.equalsStrict(server)) {
// already the same don't do anything
client.emit("debug", "received server update but server already updated");
} else {
self.servers.update(server, newserver);
client.emit("serverUpdated", server, newserver);
}
} else if (!server) {
client.emit("warn", "server was updated but it was not in the cache");
self.servers.add(new Server(data, self));
client.emit("serverCreated", server);
}
break;
case PacketType.CHANNEL_CREATE:
var channel = self.channels.get("id", data.id);
if (!channel) {
var server = self.servers.get("id", data.guild_id);
if (server) {
if (data.is_private) {
client.emit("channelCreated", self.private_channels.add(new PMChannel(data, client)));
} else {
var chan = null;
if (data.type === "text") {
chan = self.channels.add(new TextChannel(data, client, server));
} else {
chan = self.channels.add(new VoiceChannel(data, client, server));
}
client.emit("channelCreated", server.channels.add(chan));
}
} else {
client.emit("warn", "channel created but server does not exist");
}
} else {
client.emit("warn", "channel created but already in cache");
}
break;
case PacketType.CHANNEL_DELETE:
var channel = self.channels.get("id", data.id);
if (channel) {
if (channel.server) // accounts for PMs
channel.server.channels.remove(channel);
self.channels.remove(channel);
client.emit("channelDeleted", channel);
} else {
client.emit("warn", "channel deleted but already out of cache?");
}
break;
case PacketType.CHANNEL_UPDATE:
var channel = self.channels.get("id", data.id) || self.private_channels.get("id", data.id);
if (channel) {
if (channel instanceof PMChannel) {
//PM CHANNEL
client.emit("channelUpdated", channel, self.private_channels.update(channel, new PMChannel(data, client)));
} else {
if (channel.server) {
if (channel.type === "text") {
//TEXT CHANNEL
var chan = new TextChannel(data, client, channel.server);
chan.messages = channel.messages;
channel.server.channels.update(channel, chan);
self.channels.update(channel, chan);
client.emit("channelUpdated", channel, chan);
} else {
//VOICE CHANNEL
var chan = new VoiceChannel(data, client, channel.server);
channel.server.channels.update(channel, chan);
self.channels.update(channel, chan);
client.emit("channelUpdated", channel, chan);
}
} else {
client.emit("warn", "channel updated but server non-existant");
}
}
} else {
client.emit("warn", "channel updated but not in cache");
}
break;
case PacketType.SERVER_ROLE_CREATE:
var server = self.servers.get("id", data.guild_id);
if (server) {
client.emit("serverRoleCreated", server.roles.add(new Role(data.role, server, client)), server);
} else {
client.emit("warn", "server role made but server not in cache");
}
break;
case PacketType.SERVER_ROLE_DELETE:
var server = self.servers.get("id", data.guild_id);
if (server) {
var role = server.roles.get("id", data.role_id);
if (role) {
server.roles.remove(role);
client.emit("serverRoleDeleted", role);
} else {
client.emit("warn", "server role deleted but role not in cache");
}
} else {
client.emit("warn", "server role deleted but server not in cache");
}
break;
case PacketType.SERVER_ROLE_UPDATE:
var server = self.servers.get("id", data.guild_id);
if (server) {
var role = server.roles.get("id", data.role.id);
if (role) {
var newRole = new Role(data.role, server, client);
server.roles.update(role, newRole);
client.emit("serverRoleUpdated", role, newRole);
} else {
client.emit("warn", "server role updated but role not in cache");
}
} else {
client.emit("warn", "server role updated but server not in cache");
}
break;
case PacketType.SERVER_MEMBER_ADD:
var server = self.servers.get("id", data.guild_id);
if (server) {
server.memberMap[data.user.id] = {
roles: data.roles.map(function (pid) {
return server.roles.get("id", pid);
}),
mute: false,
deaf: false,
joinedAt: Date.parse(data.joined_at)
};
client.emit("serverNewMember", server, server.members.add(self.users.add(new User(data.user, client))));
} else {
client.emit("warn", "server member added but server doesn't exist in cache");
}
break;
case PacketType.SERVER_MEMBER_REMOVE:
var server = self.servers.get("id", data.guild_id);
if (server) {
var user = self.users.get("id", data.user.id);
if (user) {
server.memberMap[data.user.id] = null;
server.members.remove(user);
client.emit("serverMemberRemoved", server, user);
} else {
client.emit("warn", "server member removed but user doesn't exist in cache");
}
} else {
client.emit("warn", "server member removed but server doesn't exist in cache");
}
break;
case PacketType.SERVER_MEMBER_UPDATE:
var server = self.servers.get("id", data.guild_id);
if (server) {
var user = self.users.get("id", data.user.id);
if (user) {
server.memberMap[data.user.id].roles = data.roles.map(function (pid) {
return server.roles.get("id", pid);
});
server.memberMap[data.user.id].mute = data.mute;
server.memberMap[data.user.id].deaf = data.deaf;
client.emit("serverMemberUpdated", server, user);
} else {
client.emit("warn", "server member removed but user doesn't exist in cache");
}
} else {
client.emit("warn", "server member updated but server doesn't exist in cache");
}
break;
case PacketType.PRESENCE_UPDATE:
var user = self.users.get("id", data.user.id);
if (user) {
data.user.username = data.user.username || user.username;
data.user.id = data.user.id || user.id;
data.user.avatar = data.user.avatar || user.avatar;
data.user.discriminator = data.user.discriminator || user.discriminator;
var presenceUser = new User(data.user, client);
if (presenceUser.equalsStrict(user)) {
// a real presence update
client.emit("presence", user, data.status, data.game_id);
user.status = data.status;
user.gameID = data.game_id;
} else {
// a name change or avatar change
client.emit("userUpdate", user, presenceUser);
self.users.update(user, presenceUser);
}
} else {
client.emit("warn", "presence update but user not in cache");
}
break;
case PacketType.TYPING:
var user = self.users.get("id", data.user_id);
var channel = self.channels.get("id", data.channel_id) || self.private_channels.get("id", data.channel_id);
if (user && channel) {
if (user.typing.since) {
user.typing.since = Date.now();
user.typing.channel = channel;
} else {
user.typing.since = Date.now();
user.typing.channel = channel;
client.emit("userTypingStart", user, channel);
}
setTimeout(function () {
if (Date.now() - user.typing.since > 5500) {
// they haven't typed since
user.typing.since = null;
user.typing.channel = null;
client.emit("userTypingStop", user, channel);
}
}, 6000);
} else {
client.emit("warn", "user typing but user or channel not existant in cache");
}
break;
case PacketType.SERVER_BAN_ADD:
var user = self.users.get("id", data.user.id);
var server = self.servers.get("id", data.guild_id);
if (user && server) {
client.emit("userBanned", user, server);
} else {
client.emit("warn", "user banned but user/server not in cache.");
}
break;
case PacketType.SERVER_BAN_REMOVE:
var user = self.users.get("id", data.user.id);
var server = self.servers.get("id", data.guild_id);
if (user && server) {
client.emit("userUnbanned", user, server);
} else {
client.emit("warn", "user unbanned but user/server not in cache.");
}
break;
}
};
}
}, {
key: "uptime",
get: function get() {
return this.readyTime ? Date.now() - this.readyTime : null;
}
}]);
return InternalClient;
})();
module.exports = InternalClient;