add channel.sendMessage and channel.sendTTSMessage

This commit is contained in:
hydrabolt
2016-04-21 17:42:52 +01:00
parent b8283a8f29
commit f3e1760538
7 changed files with 61 additions and 11 deletions

View File

@@ -1,6 +1,9 @@
'use strict'; 'use strict';
const Constants = require('../../util/Constants'); const Constants = require('../../util/Constants');
const Structure = name => require('../../structures/' + name);
const Message = Structure('Message');
class RESTMethods{ class RESTMethods{
constructor(restManager) { constructor(restManager) {
@@ -32,6 +35,21 @@ class RESTMethods{
.catch(reject); .catch(reject);
}); });
} }
SendMessage(channel, content, tts, nonce) {
return new Promise((resolve, reject) => {
this.rest.makeRequest('post', Constants.Endpoints.CHANNEL_MESSAGES(channel.id), true, {
content, tts, nonce,
})
.then(data => {
let message = new Message(channel, data, this.rest.client);
channel._cacheMessage(message);
resolve(message);
this.rest.client.emit(Constants.Events.MESSAGE_CREATE, message);
})
.catch(reject);
});
}
} }
module.exports = RESTMethods; module.exports = RESTMethods;

View File

@@ -19,7 +19,7 @@ class MessageCreateHandler extends AbstractHandler {
let channel = client.store.get('channels', data.channel_id); let channel = client.store.get('channels', data.channel_id);
if (channel) { if (channel) {
let message = new Message(channel, data); let message = new Message(channel, data, client);
channel._cacheMessage(message); channel._cacheMessage(message);
client.emit(Constants.Events.MESSAGE_CREATE, message); client.emit(Constants.Events.MESSAGE_CREATE, message);
} }

View File

@@ -1,6 +1,7 @@
'use strict'; 'use strict';
const Channel = require('./Channel'); const Channel = require('./Channel');
const TextBasedChannel = require('./interface/TextBasedChannel');
const User = require('./User'); const User = require('./User');
const TextChannelDataStore = require('./datastore/TextChannelDataStore'); const TextChannelDataStore = require('./datastore/TextChannelDataStore');
@@ -26,6 +27,7 @@ class DMChannel extends Channel{
} }
setup(data) { setup(data) {
super.setup(data);
this.recipient = this.client.store.add('users', new User(this.client, data.recipient)); this.recipient = this.client.store.add('users', new User(this.client, data.recipient));
this.lastMessageID = data.last_message_id; this.lastMessageID = data.last_message_id;
} }
@@ -35,4 +37,6 @@ class DMChannel extends Channel{
} }
} }
TextBasedChannel.applyToClass(DMChannel);
module.exports = DMChannel; module.exports = DMChannel;

View File

@@ -1,16 +1,21 @@
'use strict'; 'use strict';
class Message { class Message {
constructor(serverChannel, data) { constructor(channel, data, client) {
this.channel = serverChannel; this.channel = channel;
this.guild = serverChannel.guild;
if (channel.guild) {
this.guild = channel.guild;
}
this.client = client;
if (data) { if (data) {
this.setup(data); this.setup(data);
} }
} }
setup(data) { setup(data) {
this.author = this.guild.client.store.NewUser(data.author); this.author = this.client.store.NewUser(data.author);
this.content = data.content; this.content = data.content;
this.timestamp = new Date(data.timestamp); this.timestamp = new Date(data.timestamp);
this.editedTimestamp = data.edited_timestamp ? new Date(data.edited_timestamp) : null; this.editedTimestamp = data.edited_timestamp ? new Date(data.edited_timestamp) : null;
@@ -22,11 +27,11 @@ class Message {
this.mentions = []; this.mentions = [];
this.id = data.id; this.id = data.id;
for (let mention of data.mentions) { for (let mention of data.mentions) {
let user = this.guild.client.store.get('users', mention.id); let user = this.client.store.get('users', mention.id);
if (user) { if (user) {
this.mentions.push(user); this.mentions.push(user);
} else { } else {
user = this.guild.client.store.NewUser(mention); user = this.client.store.NewUser(mention);
this.mentions.push(user); this.mentions.push(user);
} }
} }
@@ -34,7 +39,7 @@ class Message {
patch(data) { patch(data) {
if (data.author) if (data.author)
this.author = this.guild.client.store.get('users', data.author.id); this.author = this.client.store.get('users', data.author.id);
if (data.content) if (data.content)
this.content = data.content; this.content = data.content;
if (data.timestamp) if (data.timestamp)
@@ -53,11 +58,11 @@ class Message {
this.attachments = data.attachments; this.attachments = data.attachments;
if (data.mentions) { if (data.mentions) {
for (let mention of data.mentions) { for (let mention of data.mentions) {
let user = this.guild.client.store.get('users', mention.id); let user = this.client.store.get('users', mention.id);
if (user) { if (user) {
this.mentions.push(user); this.mentions.push(user);
} else { } else {
user = this.guild.client.store.NewUser(mention); user = this.client.store.NewUser(mention);
this.mentions.push(user); this.mentions.push(user);
} }
} }

View File

@@ -2,6 +2,7 @@
const ServerChannel = require('./ServerChannel'); const ServerChannel = require('./ServerChannel');
const TextChannelDataStore = require('./datastore/TextChannelDataStore'); const TextChannelDataStore = require('./datastore/TextChannelDataStore');
const TextBasedChannel = require('./interface/TextBasedChannel');
class TextChannel extends ServerChannel { class TextChannel extends ServerChannel {
@@ -26,4 +27,6 @@ class TextChannel extends ServerChannel {
} }
} }
TextBasedChannel.applyToClass(TextChannel);
module.exports = TextChannel; module.exports = TextChannel;

View File

@@ -0,0 +1,20 @@
'use strict';
function sendMessage(content, options) {
options = options || {};
return this.client.rest.methods.SendMessage(this, content, options.tts);
}
function sendTTSMessage(content, options) {
options = options || {};
return this.client.rest.methods.SendMessage(this, content, true);
}
exports.applyToClass = structure => {
if (structure.name !== 'TextChannel' && structure.name !== 'DMChannel') {
throw new Error(structure + ' cannot implement TextBasedChannel');
}
structure.prototype.sendMessage = sendMessage;
};

View File

@@ -70,7 +70,7 @@ client.on('typingStop.', (channel, user, data) => {
client.on('message', message => { client.on('message', message => {
if (message.author.username === 'hydrabolt') if (message.author.username === 'hydrabolt')
console.log(message.author.username, 'said', message.content, 'in', message.channel.name); message.channel.sendMessage('hydrabolt said: ' + message.content).then(console.log).catch(console.log);
}); });
client.on('messageDelete', message => { client.on('messageDelete', message => {