Added Guild Deletion

This commit is contained in:
hydrabolt
2016-05-01 14:51:46 +01:00
parent 3a0426482e
commit f95c588d87
12 changed files with 129 additions and 26 deletions

View File

@@ -12,6 +12,7 @@ class ActionsManager {
this.register('ChannelCreate');
this.register('ChannelDelete');
this.register('ChannelUpdate');
this.register('GuildDelete');
}
register(name) {

View File

@@ -8,6 +8,8 @@ class ChannelDeleteAction extends Action {
constructor(client) {
super(client);
this.timeouts = [];
this.deleted = {};
}
handle(data) {
@@ -15,13 +17,28 @@ class ChannelDeleteAction extends Action {
let channel = client.store.get('channels', data.id);
if (channel) {
client.store.KillChannel(channel);
this.deleted[channel.id] = channel;
this.scheduleForDeletion(channel.id);
} else if (this.deleted[data.id]) {
channel = this.deleted[data.id];
}
return {
channel,
};
}
scheduleForDeletion(id) {
this.timeouts.push(
setTimeout(() => delete this.deleted[id],
this.client.options.rest_ws_bridge_timeout)
);
}
};
module.exports = ChannelDeleteAction;

View File

@@ -0,0 +1,54 @@
'use strict';
const Action = require('./Action');
const Constants = require('../../util/Constants');
const Message = require('../../structures/Message');
class GuildDeleteAction extends Action {
constructor(client) {
super(client);
this.deleted = {};
this.timeouts = [];
}
handle(data) {
let client = this.client;
let guild = client.store.get('guilds', data.id);
if (guild) {
if (guild.available && data.unavailable) {
// guild is unavailable
guild.available = false;
client.emit(Constants.Events.GUILD_UNAVAILABLE, guild);
// stops the GuildDelete packet thinking a guild was actually deleted,
// handles emitting of event itself
return {
guild: null,
};
} else {
// delete guild
client.store.remove('guilds', guild);
this.deleted[guild.id] = guild;
this.scheduleForDeletion(guild.id);
}
} else if (this.deleted[data.id]) {
guild = this.deleted[data.id];
}
return {
guild,
};
}
scheduleForDeletion(id) {
this.timeouts.push(
setTimeout(() => delete this.deleted[id],
this.client.options.rest_ws_bridge_timeout)
);
}
};
module.exports = GuildDeleteAction;

View File

@@ -9,6 +9,7 @@ class MessageDeleteAction extends Action {
constructor(client) {
super(client);
this.timeouts = [];
this.deleted = {};
}
handle(data) {
@@ -16,9 +17,17 @@ class MessageDeleteAction extends Action {
let channel = client.store.get('channels', data.channel_id);
if (channel) {
let message = channel.store.get('messages', data.id);
if (message && !message._deleted) {
message._deleted = true;
this.scheduleForDeletion(channel, message.id);
if (message) {
channel.store.remove('messages', message.id);
this.deleted[channel.id + message.id] = message;
this.scheduleForDeletion(channel.id, message.id);
} else if (this.deleted[channel.id + data.id]) {
message = this.deleted[channel.id + data.id];
}
return {
@@ -31,9 +40,9 @@ class MessageDeleteAction extends Action {
};
}
scheduleForDeletion(channel, id) {
scheduleForDeletion(channelID, messageID) {
this.timeouts.push(
setTimeout(() => channel.store.remove('messages', id),
setTimeout(() => delete this.deleted[channelID + messageID],
this.client.options.rest_ws_bridge_timeout)
);
}

View File

@@ -111,6 +111,27 @@ class RESTMethods{
.catch(reject);
});
}
LeaveGuild(guild) {
return new Promise((resolve, reject) => {
this.rest.makeRequest('del', Constants.Endpoints.ME_GUILD(guild.id), true)
.then(() => {
resolve(this.rest.client.actions.GuildDelete.handle({ id:guild.id }).guild);
})
.catch(reject);
});
}
// untested but probably will work
DeleteGuild(guild) {
return new Promise((resolve, reject) => {
this.rest.makeRequest('del', Constants.Endpoints.GUILD(guild.id), true)
.then(() => {
resolve(this.rest.client.actions.GuildDelete.handle({ id:guild.id }).guild);
})
.catch(reject);
});
}
}
module.exports = RESTMethods;

View File

@@ -19,7 +19,7 @@ class ChannelDeleteHandler extends AbstractHandler {
let data = packet.d;
let client = this.packetManager.client;
let response = client.actions.ChannelCreate.handle(data);
let response = client.actions.ChannelDelete.handle(data);
if (response.channel) {
client.emit(Constants.Events.CHANNEL_DELETE, response.channel);

View File

@@ -18,21 +18,10 @@ class GuildDeleteHandler extends AbstractHandler {
let data = packet.d;
let client = this.packetManager.client;
let guild = client.store.get('guilds', data.id);
let response = client.actions.GuildDelete.handle(data);
if (guild) {
if (guild.available && data.unavailable) {
// guild is unavailable
guild.available = false;
client.emit(Constants.Events.GUILD_UNAVAILABLE, guild);
} else {
// delete guild
client.store.KillGuild(guild);
this.packetManager.ws.checkIfReady();
}
} else {
// it's not there! :(
client.emit('warn', 'guild deleted but not cached in first place. missed packet?');
if (response.guild) {
client.emit(Constants.Events.GUILD_DELETE, response.guild);
}
}