discord.js/lib/Client/Resolver/Resolver.js

"use strict";
/* global Buffer */

/**
 * Resolves supplied data type to a Channel. If a String, it should be a Channel ID.
 * @typedef {(Channel|Server|Message|User|String)} ChannelResolvable
*/
/**
 * Resolves supplied data type to a TextChannel or PMChannel. If a String, it should be a Channel ID.
 * @typedef {(TextChannel|PMChannel|Server|Message|User|String)} TextChannelResolvable
*/
/**
 * If given an array, turns it into a newline-separated string.
 * @typedef {(String|Array)} StringResolvable
*/
/**
 * Resolves supplied data type to a Message. If a channel, it is the latest message from that channel.
 * @typedef {(Message|TextChannel|PMChannel)} MessageResolvable
*/
/**
 * Resolves supplied data type to a Server. If a String, it should be the server's ID.
 * @typedef {(Server|ServerChannel|Message|String)} ServerResolvable
 */
/**
 * Resolves supplied data type to something that can be attached to a message. If a String, it can be an URL or a path to a local file.
 * @typedef {(String|ReadableStream|Buffer)} FileResolvable
 */
/**
 * Resolves supplied data type to an invite ID. If a String, it should be an ID or a direct URL to the invite.
 * @typedef {(Invite|String)} InviteIDResolvable
 */

exports.__esModule = true;

function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { "default": obj }; }

function _classCallCheck(instance, Constructor) { if (!(instance instanceof Constructor)) { throw new TypeError("Cannot call a class as a function"); } }

var _fs = require("fs");

var _fs2 = _interopRequireDefault(_fs);

var _superagent = require("superagent");

var _superagent2 = _interopRequireDefault(_superagent);

var _StructuresUser = require("../../Structures/User");

var _StructuresUser2 = _interopRequireDefault(_StructuresUser);

var _StructuresChannel = require("../../Structures/Channel");

var _StructuresChannel2 = _interopRequireDefault(_StructuresChannel);

var _StructuresTextChannel = require("../../Structures/TextChannel");

var _StructuresTextChannel2 = _interopRequireDefault(_StructuresTextChannel);

var _StructuresVoiceChannel = require("../../Structures/VoiceChannel");

var _StructuresVoiceChannel2 = _interopRequireDefault(_StructuresVoiceChannel);

var _StructuresServerChannel = require("../../Structures/ServerChannel");

var _StructuresServerChannel2 = _interopRequireDefault(_StructuresServerChannel);

var _StructuresPMChannel = require("../../Structures/PMChannel");

var _StructuresPMChannel2 = _interopRequireDefault(_StructuresPMChannel);

var _StructuresRole = require("../../Structures/Role");

var _StructuresRole2 = _interopRequireDefault(_StructuresRole);

var _StructuresServer = require("../../Structures/Server");

var _StructuresServer2 = _interopRequireDefault(_StructuresServer);

var _StructuresMessage = require("../../Structures/Message");

var _StructuresMessage2 = _interopRequireDefault(_StructuresMessage);

var _StructuresInvite = require("../../Structures/Invite");

var _StructuresInvite2 = _interopRequireDefault(_StructuresInvite);

var _StructuresWebhook = require("../../Structures/Webhook");

var _StructuresWebhook2 = _interopRequireDefault(_StructuresWebhook);

var _VoiceVoiceConnection = require("../../Voice/VoiceConnection");

var _VoiceVoiceConnection2 = _interopRequireDefault(_VoiceVoiceConnection);

var Resolver = (function () {
	function Resolver(internal) {
		_classCallCheck(this, Resolver);

		this.internal = internal;
	}

	Resolver.prototype.resolveToBase64 = function resolveToBase64(resource) {
		if (resource instanceof Buffer) {
			resource = resource.toString("base64");
			resource = "data:image/jpg;base64," + resource;
		}
		return resource;
	};

	Resolver.prototype.resolveInviteID = function resolveInviteID(resource) {
		if (resource instanceof _StructuresInvite2["default"]) {
			return resource.id;
		}
		if (typeof resource === "string" || resource instanceof String) {
			if (resource.indexOf("http") === 0) {
				var split = resource.split("/");
				return split.pop();
			}
			return resource;
		}
		return null;
	};

	Resolver.prototype.resolveServer = function resolveServer(resource) {
		if (resource instanceof _StructuresServer2["default"]) {
			return resource;
		}
		if (resource instanceof _StructuresServerChannel2["default"]) {
			return resource.server;
		}
		if (resource instanceof String || typeof resource === "string") {
			return this.internal.servers.get("id", resource);
		}
		if (resource instanceof _StructuresMessage2["default"]) {
			if (resource.channel instanceof _StructuresTextChannel2["default"]) {
				return resource.channel.server;
			}
		}
		return null;
	};

	Resolver.prototype.resolveRole = function resolveRole(resource) {
		if (resource instanceof _StructuresRole2["default"]) {
			return resource;
		}
		if (resource instanceof String || typeof resource === "string") {
			var role = null;
			for (var _iterator = this.internal.servers, _isArray = Array.isArray(_iterator), _i = 0, _iterator = _isArray ? _iterator : _iterator[Symbol.iterator]();;) {
				var _ref;

				if (_isArray) {
					if (_i >= _iterator.length) break;
					_ref = _iterator[_i++];
				} else {
					_i = _iterator.next();
					if (_i.done) break;
					_ref = _i.value;
				}

				var server = _ref;

				if (role = server.roles.get("id", resource)) {
					return role;
				}
			}
		}
		return null;
	};

	Resolver.prototype.resolveFile = function resolveFile(resource) {
		if (typeof resource === "string" || resource instanceof String) {
			if (/^https?:\/\//.test(resource)) {
				return new Promise(function (resolve, reject) {
					_superagent2["default"].get(resource).buffer().parse(function (res, cb) {
						res.setEncoding("binary");
						res.data = "";
						res.on("data", function (chunk) {
							res.data += chunk;
						});
						res.on("end", function () {
							cb(null, new Buffer(res.data, "binary"));
						});
					}).end(function (err, res) {
						if (err) {
							return reject(err);
						}
						return resolve(res.body);
					});
				});
			} else {
				return Promise.resolve(resource);
			}
		}
		return Promise.resolve(resource);
	};

	Resolver.prototype.resolveMentions = function resolveMentions(resource, channel) {
		var _mentions = [];
		var changed = resource;
		for (var _iterator2 = resource.match(/<@\!?[0-9]+>/g) || [], _isArray2 = Array.isArray(_iterator2), _i2 = 0, _iterator2 = _isArray2 ? _iterator2 : _iterator2[Symbol.iterator]();;) {
			var _ref2;

			if (_isArray2) {
				if (_i2 >= _iterator2.length) break;
				_ref2 = _iterator2[_i2++];
			} else {
				_i2 = _iterator2.next();
				if (_i2.done) break;
				_ref2 = _i2.value;
			}

			var mention = _ref2;
			// username mention
			if (mention[2] === '!') {
				var user = this.internal.users.get("id", mention.substring(3, mention.length - 1));
				if (user) {
					_mentions.push(user);
					var details = channel.server && channel.server.detailsOf(user);
					if (details) {
						changed = changed.replace(new RegExp(mention, "g"), "@" + (details.nick || user.username + "#" + user.discriminator));
					}
				}
			} else {
				var user = this.internal.users.get("id", mention.substring(2, mention.length - 1));
				if (user) {
					_mentions.push(user);
					changed = changed.replace(new RegExp(mention, "g"), "@" + (user.username + "#" + user.discriminator));
				}
			}
		}
		if (channel && channel.server && channel.server.roles) {
			for (var _iterator3 = resource.match(/<@&[0-9]+>/g) || [], _isArray3 = Array.isArray(_iterator3), _i3 = 0, _iterator3 = _isArray3 ? _iterator3 : _iterator3[Symbol.iterator]();;) {
				var _ref3;

				if (_isArray3) {
					if (_i3 >= _iterator3.length) break;
					_ref3 = _iterator3[_i3++];
				} else {
					_i3 = _iterator3.next();
					if (_i3.done) break;
					_ref3 = _i3.value;
				}

				var mention = _ref3;
				// role mention
				var role = channel.server.roles.get("id", mention.substring(3, mention.length - 1));
				if (role) {
					changed = changed.replace(new RegExp(mention, "g"), "@" + role.name);
				}
			}
		}
		for (var _iterator4 = resource.match(/<#[0-9]+>/g) || [], _isArray4 = Array.isArray(_iterator4), _i4 = 0, _iterator4 = _isArray4 ? _iterator4 : _iterator4[Symbol.iterator]();;) {
			var _ref4;

			if (_isArray4) {
				if (_i4 >= _iterator4.length) break;
				_ref4 = _iterator4[_i4++];
			} else {
				_i4 = _iterator4.next();
				if (_i4.done) break;
				_ref4 = _i4.value;
			}

			var mention = _ref4;
			// channel mention
			var channel = this.internal.channels.get("id", mention.substring(2, mention.length - 1));
			if (channel) {
				changed = changed.replace(new RegExp(mention, "g"), "#" + channel.name);
			}
		}
		return [_mentions, changed];
	};

	Resolver.prototype.resolveString = function resolveString(resource) {

		// accepts Array, Channel, Server, User, Message, String and anything
		// toString()-able

		if (resource instanceof Array) {
			resource = resource.join("\n");
		}

		return resource.toString();
	};

	Resolver.prototype.resolveUser = function resolveUser(resource) {
		/*
  	accepts a Message, Channel, Server, String ID, User, PMChannel
  */
		if (resource instanceof _StructuresUser2["default"]) {
			return resource;
		}
		if (resource instanceof _StructuresMessage2["default"]) {
			return resource.author;
		}
		if (resource instanceof _StructuresTextChannel2["default"]) {
			var lmsg = resource.lastMessage;
			if (lmsg) {
				return lmsg.author;
			}
		}
		if (resource instanceof _StructuresServer2["default"]) {
			return resource.owner;
		}
		if (resource instanceof _StructuresPMChannel2["default"]) {
			return resource.recipient;
		}
		if (resource instanceof String || typeof resource === "string") {
			return this.internal.users.get("id", resource);
		}

		return null;
	};

	Resolver.prototype.resolveWebhook = function resolveWebhook(resource) {
		/*
   accepts a Webhook
   */
		if (resource instanceof _StructuresWebhook2["default"]) {
			return Promise.resolve(resource);
		}
		if (resource instanceof String || typeof resource === "string") {
			var server = this.internal.servers.find(function (s) {
				return s.webhooks.has("id", resource);
			});
			if (server) {
				return Promise.resolve(server.webhooks.get("id", resource));
			}
		}

		var error = new Error("Could not resolve webhook");
		error.resource = resource;
		return Promise.reject(error);
	};

	Resolver.prototype.resolveMessage = function resolveMessage(resource) {
		// accepts a Message, PMChannel & TextChannel

		if (resource instanceof _StructuresTextChannel2["default"] || resource instanceof _StructuresPMChannel2["default"]) {
			return resource.lastMessage;
		}
		if (resource instanceof _StructuresMessage2["default"]) {
			return resource;
		}

		return null;
	};

	Resolver.prototype.resolveChannel = function resolveChannel(resource) {
		/*
  	accepts a Message, Channel, VoiceConnection, Server, String ID, User
  */

		if (resource instanceof _StructuresMessage2["default"]) {
			return Promise.resolve(resource.channel);
		}
		if (resource instanceof _StructuresChannel2["default"]) {
			return Promise.resolve(resource);
		}
		if (resource instanceof _VoiceVoiceConnection2["default"]) {
			return Promise.resolve(resource.voiceChannel);
		}
		if (resource instanceof _StructuresServer2["default"]) {
			return Promise.resolve(resource.defaultChannel);
		}
		if (resource instanceof String || typeof resource === "string") {
			var user = this.internal.users.get("id", resource);
			if (user) {
				resource = user;
			} else {
				return Promise.resolve(this.internal.channels.get("id", resource) || this.internal.private_channels.get("id", resource));
			}
		}
		if (resource instanceof _StructuresUser2["default"]) {
			// see if a PM exists
			for (var _iterator5 = this.internal.private_channels, _isArray5 = Array.isArray(_iterator5), _i5 = 0, _iterator5 = _isArray5 ? _iterator5 : _iterator5[Symbol.iterator]();;) {
				var _ref5;

				if (_isArray5) {
					if (_i5 >= _iterator5.length) break;
					_ref5 = _iterator5[_i5++];
				} else {
					_i5 = _iterator5.next();
					if (_i5.done) break;
					_ref5 = _i5.value;
				}

				var pmchat = _ref5;

				if (pmchat.recipients.length === 1 && pmchat.recipient && pmchat.recipient.equals(resource)) {
					return Promise.resolve(pmchat);
				}
			}

			// PM does not exist :\
			return this.internal.startPM(resource);
		}
		var error = new Error("Could not resolve channel");
		error.resource = resource;
		return Promise.reject(error);
	};

	return Resolver;
})();

exports["default"] = Resolver;
module.exports = exports["default"];