mirror of
https://github.com/discordjs/discord.js.git
synced 2026-03-09 16:13:31 +01:00
708 lines
16 KiB
JavaScript
708 lines
16 KiB
JavaScript
var request = require( "superagent" );
|
|
var Endpoints = require( "./lib/endpoints.js" );
|
|
var Server = require( "./lib/server.js" ).Server;
|
|
var Message = require( "./lib/message.js" ).Message;
|
|
var User = require( "./lib/user.js" ).User;
|
|
var Channel = require( "./lib/channel.js" ).Channel;
|
|
var List = require( "./lib/list.js" ).List;
|
|
var Invite = require( "./lib/invite.js" ).Invite;
|
|
var PMChannel = require( "./lib/PMChannel.js" ).PMChannel;
|
|
var WebSocket = require( 'ws' );
|
|
var Internal = require( "./lib/internal.js" ).Internal;
|
|
|
|
exports.Endpoints = Endpoints;
|
|
exports.Server = Server;
|
|
exports.Message = Message;
|
|
exports.User = User;
|
|
exports.Channel = Channel;
|
|
exports.List = List;
|
|
exports.Invite = Invite;
|
|
exports.PMChannel = PMChannel;
|
|
|
|
exports.Client = function( options ) {
|
|
|
|
this.options = options || {};
|
|
this.options.maxmessage = 5000;
|
|
this.token = "";
|
|
this.loggedIn = false;
|
|
this.websocket = null;
|
|
this.events = {};
|
|
this.user = null;
|
|
this.ready = false;
|
|
this.serverList = new List( "id" );
|
|
this.PMList = new List( "id" );
|
|
|
|
}
|
|
|
|
exports.Client.prototype.getServers = function() {
|
|
return this.serverList;
|
|
}
|
|
|
|
exports.Client.prototype.getChannels = function() {
|
|
return this.serverList.concatSublists( "channels", "id" );
|
|
}
|
|
|
|
exports.Client.prototype.getServer = function( id ) {
|
|
return this.getServers().filter( "id", id, true );
|
|
}
|
|
|
|
exports.Client.prototype.getChannel = function( id ) {
|
|
return this.getChannels().filter( "id", id, true );
|
|
}
|
|
|
|
exports.Client.prototype.triggerEvent = function( event, args ) {
|
|
|
|
if ( !this.ready && event !== "raw" && event !== "disconnected" && event !== "debug" ) { //if we're not even loaded yet, don't try doing anything because it always ends badly!
|
|
return;
|
|
}
|
|
|
|
if ( this.events[ event ] ) {
|
|
this.events[ event ].apply( this, args );
|
|
} else {
|
|
return false;
|
|
}
|
|
|
|
}
|
|
|
|
exports.Client.prototype.on = function( name, fn ) {
|
|
this.events[ name ] = fn;
|
|
}
|
|
|
|
exports.Client.prototype.off = function( name ) {
|
|
this.events[ name ] = function() {};
|
|
}
|
|
|
|
exports.Client.prototype.cacheServer = function( id, cb, members ) {
|
|
var self = this;
|
|
var serverInput = {};
|
|
|
|
if ( typeof id === 'string' || id instanceof String ) {
|
|
//actually an ID
|
|
|
|
if ( this.serverList.filter( "id", id ).length > 0 ) {
|
|
return;
|
|
}
|
|
|
|
Internal.XHR.getServer( self.token, id, function( err, data ) {
|
|
if ( !err ) {
|
|
makeServer( data );
|
|
}
|
|
} )
|
|
|
|
} else {
|
|
// got objects because SPEEEDDDD
|
|
|
|
if ( this.serverList.filter( "id", id.id ).length > 0 ) {
|
|
return;
|
|
}
|
|
serverInput = id;
|
|
id = id.id;
|
|
makeServer( serverInput );
|
|
|
|
}
|
|
|
|
function channelsFromHTTP() {
|
|
Internal.XHR.getChannel( self.token, id, function( err ) {
|
|
if ( !err )
|
|
cacheChannels( res.body );
|
|
} )
|
|
}
|
|
|
|
var server;
|
|
|
|
function makeServer( dat ) {
|
|
server = new Server( dat.region, dat.owner_id, dat.name, id, serverInput.members || dat.members, dat.icon, dat.afk_timeout, dat.afk_channel_id );
|
|
if ( dat.channels )
|
|
cacheChannels( dat.channels );
|
|
else
|
|
channelsFromHTTP();
|
|
}
|
|
|
|
function cacheChannels( dat ) {
|
|
|
|
var channelList = dat;
|
|
for ( channel of channelList ) {
|
|
server.channels.add( new Channel( channel, server ) );
|
|
}
|
|
|
|
self.serverList.add( server );
|
|
|
|
cb( server );
|
|
}
|
|
|
|
}
|
|
|
|
exports.Client.prototype.login = function( email, password ) {
|
|
|
|
var self = this;
|
|
|
|
self.connectWebsocket();
|
|
|
|
var time = Date.now();
|
|
Internal.XHR.login( email, password, function( err, token ) {
|
|
console.log(Date.now() - time);
|
|
if ( err ) {
|
|
self.triggerEvent( "disconnected", [ {
|
|
reason: "failed to log in",
|
|
error: err
|
|
} ] );
|
|
self.websocket.close();
|
|
} else {
|
|
self.token = token;
|
|
self.websocket.sendData();
|
|
self.loggedIn = true;
|
|
}
|
|
|
|
} );
|
|
}
|
|
|
|
exports.Client.prototype.reply = function( destination, toSend, callback, options ) {
|
|
|
|
if ( toSend instanceof Array ) {
|
|
toSend = toSend.join( "\n" );
|
|
}
|
|
|
|
toSend = destination.author.mention() + ", " + toSend;
|
|
|
|
this.sendMessage( destination, toSend, callback, options );
|
|
|
|
}
|
|
|
|
exports.Client.prototype.connectWebsocket = function( cb ) {
|
|
|
|
var self = this;
|
|
|
|
var sentInitData = false;
|
|
|
|
this.websocket = new WebSocket( Endpoints.WEBSOCKET_HUB );
|
|
this.websocket.onclose = function( e ) {
|
|
self.triggerEvent( "disconnected", [ {
|
|
reason: "websocket disconnected",
|
|
error: e
|
|
} ] );
|
|
};
|
|
this.websocket.onmessage = function( e ) {
|
|
|
|
self.triggerEvent( "raw", [ e ] );
|
|
|
|
var dat = JSON.parse( e.data );
|
|
var webself = this;
|
|
|
|
switch ( dat.op ) {
|
|
|
|
case 0:
|
|
if ( dat.t === "READY" ) {
|
|
|
|
var data = dat.d;
|
|
|
|
setInterval( function() {
|
|
webself.keepAlive.apply( webself );
|
|
}, data.heartbeat_interval );
|
|
|
|
self.user = new User( data.user );
|
|
|
|
var _servers = data.guilds,
|
|
servers = [];
|
|
|
|
var cached = 0,
|
|
toCache = _servers.length;
|
|
|
|
for ( x in data.private_channels ) {
|
|
self.PMList.add( new PMChannel( data.private_channels[ x ].recipient, data.private_channels[ x ].id ) );
|
|
}
|
|
|
|
for ( x in _servers ) {
|
|
_server = _servers[ x ];
|
|
|
|
self.cacheServer( _server, function( server ) {
|
|
cached++;
|
|
if ( cached === toCache ) {
|
|
self.ready = true;
|
|
self.triggerEvent( "ready" );
|
|
}
|
|
} );
|
|
}
|
|
|
|
} else if ( dat.t === "MESSAGE_CREATE" ) {
|
|
var data = dat.d;
|
|
|
|
var channel = self.getChannel( data.channel_id );
|
|
var message = new Message( data, channel );
|
|
self.triggerEvent( "message", [ message ] );
|
|
if ( channel.messages )
|
|
channel.messages.add( message );
|
|
|
|
} else if ( dat.t === "MESSAGE_DELETE" ) {
|
|
var data = dat.d;
|
|
|
|
var channel = self.getChannel( data.channel_id );
|
|
|
|
if ( !channel.messages )
|
|
return;
|
|
|
|
var _msg = channel.messages.filter( "id", data.id, true );
|
|
|
|
if ( _msg ) {
|
|
self.triggerEvent( "messageDelete", [ _msg ] );
|
|
channel.messages.removeElement( _msg );
|
|
}
|
|
|
|
} else if ( dat.t === "MESSAGE_UPDATE" ) {
|
|
|
|
var data = dat.d;
|
|
if ( !self.ready )
|
|
return;
|
|
|
|
var formerMessage = false;
|
|
|
|
var channel = self.getChannel( data.channel_id );
|
|
|
|
if ( channel ) {
|
|
|
|
formerMessage = channel.messages.filter( "id", data.id, true );
|
|
newMessage = new Message( data, channel );
|
|
self.triggerEvent( "messageUpdate", [ formerMessage, newMessage ] );
|
|
|
|
if ( formerMessage )
|
|
channel.messages.updateElement( formerMessage, newMessage );
|
|
else
|
|
channel.messages.add( newMessage );
|
|
|
|
}
|
|
|
|
} else if ( dat.t === "PRESENCE_UPDATE" ) {
|
|
|
|
var data = dat.d;
|
|
|
|
self.triggerEvent( "presence", [ new User( data.user ), data.status, self.serverList.filter( "id", data.guild_id, true ) ] );
|
|
|
|
} else if ( dat.t === "GUILD_DELETE" ) {
|
|
|
|
var deletedServer = self.serverList.filter( "id", dat.d.id, true );
|
|
|
|
if ( deletedServer ) {
|
|
self.triggerEvent( "serverDelete", [ deletedServer ] );
|
|
}
|
|
|
|
} else if ( dat.t === "CHANNEL_DELETE" ) {
|
|
|
|
var delServer = self.serverList.filter( "id", dat.d.guild_id, true );
|
|
|
|
if ( delServer ) {
|
|
var channel = delServer.channels.filter( "id", dat.d.id, true );
|
|
|
|
if ( channel ) {
|
|
self.triggerEvent( "channelDelete", [ channel ] );
|
|
}
|
|
}
|
|
|
|
} else if ( dat.t === "GUILD_CREATE" ) {
|
|
|
|
if ( !self.serverList.filter( "id", dat.d.id, true ) ) {
|
|
self.cacheServer( dat.d, function( server ) {
|
|
self.triggerEvent( "serverJoin", [ server ] );
|
|
} );
|
|
}
|
|
|
|
} else if ( dat.t === "CHANNEL_CREATE" ) {
|
|
|
|
var srv = self.serverList.filter( "id", dat.d.guild_id, true );
|
|
|
|
if ( srv ) {
|
|
|
|
if ( !srv.channels.filter( "id", dat.d.d, true ) ) {
|
|
|
|
var chann = new Channel( dat.d, srv );
|
|
|
|
srv.channels.add( new Channel( dat.d, srv ) );
|
|
self.triggerEvent( "channelCreate", [ chann ] );
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
break;
|
|
|
|
}
|
|
|
|
};
|
|
this.websocket.sendPacket = function( p ) {
|
|
this.send( JSON.stringify( p ) );
|
|
}
|
|
this.websocket.keepAlive = function() {
|
|
|
|
if ( this.readyState !== 1 )
|
|
return false;
|
|
|
|
this.sendPacket( {
|
|
op: 1,
|
|
d: Date.now()
|
|
} );
|
|
|
|
}
|
|
this.websocket.onopen = function() {
|
|
|
|
this.sendData("onopen");
|
|
|
|
}
|
|
this.websocket.sendData = function(why){
|
|
if(this.readyState == 1 && !sentInitData && self.token){
|
|
sentInitData = true;
|
|
var connDat = {
|
|
op: 2,
|
|
d: {
|
|
token: self.token,
|
|
v: 2
|
|
}
|
|
};
|
|
|
|
connDat.d.properties = Internal.WebSocket.properties;
|
|
this.sendPacket( connDat );
|
|
}
|
|
}
|
|
}
|
|
|
|
exports.Client.prototype.logout = function( callback ) {
|
|
|
|
callback = callback || function(){};
|
|
|
|
var self = this;
|
|
|
|
Internal.XHR.logout( self.token, function( err ) {
|
|
callback( err ); //if there isn't an error it'll be null anyway so...
|
|
self.loggedIn = Boolean( err );
|
|
} );
|
|
|
|
}
|
|
|
|
exports.Client.prototype.createServer = function( name, region, cb ) {
|
|
|
|
var self = this;
|
|
|
|
Internal.XHR.createServer( self.token, name, region, function( err, data ) {
|
|
|
|
if ( err ) {
|
|
cb( err );
|
|
} else {
|
|
|
|
self.cacheServer( data, function( server ) {
|
|
cb( null, server );
|
|
} );
|
|
|
|
}
|
|
|
|
} );
|
|
|
|
}
|
|
|
|
exports.Client.prototype.leaveServer = function( server, callback ) {
|
|
|
|
var self = this;
|
|
|
|
// callback is not necessary for this function
|
|
callback = callback || function() {};
|
|
|
|
Internal.XHR.leaveServer( self.token, server.id, function( err ) {
|
|
|
|
if ( err ) {
|
|
callback( err );
|
|
} else {
|
|
self.serverList.removeElement( server );
|
|
callback( null );
|
|
}
|
|
|
|
} );
|
|
|
|
}
|
|
|
|
exports.Client.prototype.createInvite = function( channel, options, callback ) {
|
|
|
|
var self = this;
|
|
var options = options || {};
|
|
|
|
// callback is not necessary for this function
|
|
callback = callback || function() {};
|
|
|
|
if ( channel instanceof Server ) {
|
|
channel = channel.getDefaultChannel();
|
|
}
|
|
|
|
options.max_age = options.max_age || 0;
|
|
options.max_uses = options.max_uses || 0;
|
|
options.temporary = options.temporary || false;
|
|
options.xkcdpass = options.xkcd || false;
|
|
|
|
Internal.XHR.createInvite( self.token, channel.id, options, function( err, data ) {
|
|
|
|
if ( err ) {
|
|
callback( err );
|
|
} else {
|
|
callback( null, new Invite( data ) );
|
|
}
|
|
|
|
} );
|
|
|
|
}
|
|
|
|
exports.Client.prototype.startPM = function( user, callback ) {
|
|
|
|
var self = this;
|
|
|
|
callback = callback || function() {};
|
|
|
|
Internal.XHR.startPM( self.token, self.user.id, user.id, function( err, data ) {
|
|
|
|
if ( err ) {
|
|
callback( err );
|
|
} else {
|
|
var channel = new PMChannel( data.recipient, data.id );
|
|
self.PMList.add( channel );
|
|
callback( null, channel );
|
|
}
|
|
|
|
} );
|
|
|
|
}
|
|
|
|
exports.Client.prototype.sendMessage = function( destination, toSend, callback, options ) {
|
|
|
|
options = options || {};
|
|
callback = callback || function() {};
|
|
|
|
var channel_id, message, mentions, self = this;
|
|
|
|
channel_id = resolveChannel( destination, self );
|
|
message = resolveMessage( toSend );
|
|
mentions = resolveMentions( message, options.mention );
|
|
|
|
if ( channel_id ) {
|
|
send();
|
|
} else {
|
|
//a channel is being sorted
|
|
}
|
|
|
|
function send() {
|
|
|
|
Internal.XHR.sendMessage( self.token, channel_id, {
|
|
content: message,
|
|
mentions: mentions
|
|
}, function( err, data ) {
|
|
if ( err ) {
|
|
callback( err );
|
|
} else {
|
|
var msg = new Message( data, self.getChannel( data.channel_id ) );
|
|
if ( options.selfDestruct ) {
|
|
setTimeout( function() {
|
|
self.deleteMessage( msg );
|
|
}, options.selfDestruct );
|
|
}
|
|
callback( null, msg );
|
|
}
|
|
} );
|
|
}
|
|
|
|
function setChannId( id ) {
|
|
channel_id = id;
|
|
}
|
|
|
|
function resolveChannel( destination, self ) {
|
|
var channel_id = false;
|
|
if ( destination instanceof Server ) {
|
|
channel_id = destination.getDefaultChannel().id;
|
|
} else if ( destination instanceof Channel ) {
|
|
channel_id = destination.id;
|
|
} else if ( destination instanceof PMChannel ) {
|
|
channel_id = destination.id;
|
|
} else if ( destination instanceof Message ) {
|
|
channel_id = destination.channel.id;
|
|
} else if ( destination instanceof User ) {
|
|
var destId = self.PMList.deepFilter( [ "user", "id" ], destination.id, true );
|
|
|
|
if ( destId ) {
|
|
channel_id = destId.id;
|
|
} else {
|
|
//start a PM and then get that use that
|
|
self.startPM( destination, function( err, channel ) {
|
|
if ( err ) {
|
|
callback( err );
|
|
} else {
|
|
self.PMList.add( new PMChannel( channel.recipient, channel.id ) );
|
|
setChannId( channel.id );
|
|
send();
|
|
}
|
|
} );
|
|
return false;
|
|
}
|
|
} else {
|
|
channel_id = destination;
|
|
}
|
|
return channel_id;
|
|
}
|
|
|
|
function resolveMessage( toSend ) {
|
|
var message;
|
|
if ( typeof toSend === "string" || toSend instanceof String )
|
|
message = toSend;
|
|
else if ( toSend instanceof Array )
|
|
message = toSend.join( "\n" );
|
|
else if ( toSend instanceof Message )
|
|
message = toSend.content;
|
|
else
|
|
message = toSend;
|
|
return message.substring( 0, 2000 );
|
|
}
|
|
|
|
function resolveMentions( message, mentionsOpt ) {
|
|
var mentions = [];
|
|
if ( mentionsOpt === false ) {} else if ( mentionsOpt || mentionsOpt === "auto" || mentionsOpt == null || mentionsOpt == undefined ) {
|
|
for ( mention of( message.match( /<@[^>]*>/g ) || [] ) ) {
|
|
mentions.push( mention.substring( 2, mention.length - 1 ) );
|
|
}
|
|
} else if ( mentionsOpt instanceof Array ) {
|
|
for ( mention of mentionsOpt ) {
|
|
if ( mention instanceof User ) {
|
|
mentions.push( mention.id );
|
|
} else {
|
|
mentions.push( mention );
|
|
}
|
|
}
|
|
}
|
|
return mentions;
|
|
}
|
|
}
|
|
|
|
exports.Client.prototype.deleteMessage = function( message, callback ) {
|
|
callback = callback || function() {};
|
|
|
|
var self = this;
|
|
|
|
Internal.XHR.deleteMessage( self.token, message.channel.id, message.id, callback );
|
|
}
|
|
|
|
exports.Client.prototype.updateMessage = function(oldMessage, newContent, callback, options){
|
|
|
|
var self = this;
|
|
var channel = oldMessage.channel;
|
|
options = options || {};
|
|
|
|
Internal.XHR.updateMessage(self.token, channel.id, oldMessage.id, {
|
|
content : newContent,
|
|
mentions : []
|
|
}, function(err, data){
|
|
if(err){
|
|
callback(err);
|
|
return;
|
|
}
|
|
var msg = new Message( data, self.getChannel( data.channel_id ) );
|
|
if ( options.selfDestruct ) {
|
|
setTimeout( function() {
|
|
self.deleteMessage( msg );
|
|
}, options.selfDestruct );
|
|
}
|
|
callback( null, msg );
|
|
});
|
|
|
|
}
|
|
|
|
exports.Client.prototype.getChannelLogs = function( channel, amount, callback ) {
|
|
var self = this;
|
|
|
|
Internal.XHR.getChannelLogs( self.token, channel.id, ( amount || 50 ), function( err, data ) {
|
|
|
|
if ( err ) {
|
|
callback( err );
|
|
return;
|
|
}
|
|
|
|
var logs = new List( "id" );
|
|
for ( message of data ) {
|
|
logs.add( new Message( message, channel ) );
|
|
}
|
|
callback( null, logs );
|
|
|
|
} );
|
|
}
|
|
|
|
exports.Client.prototype.createChannel = function( server, name, type, callback ) {
|
|
|
|
var self = this;
|
|
|
|
Internal.XHR.createChannel( self.token, server.id, name, type, function( err, data ) {
|
|
|
|
if ( err ) {
|
|
callback( err );
|
|
} else {
|
|
var chann = new Channel( data, server );
|
|
server.channels.add( chann );
|
|
callback( null, chann );
|
|
}
|
|
|
|
} );
|
|
}
|
|
|
|
exports.Client.prototype.deleteChannel = function( channel, callback ) {
|
|
var self = this;
|
|
|
|
Internal.XHR.deleteChannel( self.token, channel.id, function( err ) {
|
|
|
|
channel.server.channels.removeElement( channel );
|
|
self.triggerEvent( "channelDelete", [ channel ] );
|
|
callback( null );
|
|
|
|
} );
|
|
}
|
|
|
|
exports.Client.prototype.deleteServer = function( server, callback ) {
|
|
|
|
var self = this;
|
|
|
|
Internal.XHR.deleteServer( self.token, server.id, function( err ) {
|
|
|
|
self.serverList.removeElement( server );
|
|
self.triggerEvent( "serverDelete", [ server ] );
|
|
callback( null );
|
|
|
|
} );
|
|
|
|
}
|
|
|
|
/**
|
|
UTILS
|
|
UTILS
|
|
UTILS
|
|
did I mention UTILS?
|
|
*/
|
|
|
|
exports.Client.prototype.getServers = function() {
|
|
return this.serverList;
|
|
}
|
|
|
|
exports.Client.prototype.getChannels = function() {
|
|
return this.serverList.concatSublists( "channels", "id" );
|
|
}
|
|
|
|
exports.Client.prototype.getUsers = function() {
|
|
return this.getServers().concatSublists( "members", "id" );
|
|
}
|
|
|
|
exports.Client.prototype.getServer = function( id ) {
|
|
return this.getServers().filter( "id", id, true );
|
|
}
|
|
|
|
exports.Client.prototype.getChannel = function( id ) {
|
|
return this.getChannels().filter( "id", id, true );
|
|
}
|
|
|
|
exports.Client.prototype.getUser = function( id ) {
|
|
return this.getUsers().filter( "id", id, true );
|
|
}
|
|
|
|
exports.isUserID = function( id ) {
|
|
return ( ( id + "" ).length === 17 && !isNaN( id ) );
|
|
}
|
|
|
|
exports.Client.prototype.addPM = function( pm ) {
|
|
this.PMList.add( pm );
|
|
}
|