Added ClientUser modification support

This commit is contained in:
hydrabolt
2016-05-01 21:12:42 +01:00
parent f95c588d87
commit ebfc2169d5
10 changed files with 143 additions and 19 deletions

View File

@@ -33,6 +33,10 @@ class Client extends EventEmitter{
}
}
get user() {
return this.store.user;
}
}
module.exports = Client;

View File

@@ -54,6 +54,13 @@ class ClientDataResolver {
return guild.store.get('members', user.id);
}
ResolveBase64(data) {
if (data instanceof Buffer) {
return 'data:image/jpg;base64,' + data.toString('base64');
}
return data;
}
}
module.exports = ClientDataResolver;

View File

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

View File

@@ -0,0 +1,44 @@
'use strict';
const Action = require('./Action');
const Constants = require('../../util/Constants');
const CloneObject = require('../../util/CloneObject');
const Message = require('../../structures/Message');
class UserUpdateAction extends Action {
constructor(client) {
super(client);
}
handle(data) {
let client = this.client;
if (client.store.user) {
if (client.store.user.equals(data)) {
return {
old: client.store.user,
updated: client.store.user,
};
}
let oldUser = CloneObject(client.store.user);
client.store.user.setup(data);
client.emit(Constants.Events.USER_UPDATE, oldUser, client.store.user);
return {
old: oldUser,
updated: client.store.user,
};
}
return {
old: null,
updated: null,
};
}
};
module.exports = UserUpdateAction;

View File

@@ -12,7 +12,8 @@ class RESTMethods{
LoginEmailPassword(email, password) {
return new Promise((resolve, reject) => {
this.rest.client.store.email = email;
this.rest.client.store.password = password;
this.rest.makeRequest('post', Constants.Endpoints.LOGIN, false, { email, password })
.then(data => {
this.rest.client.manager.connectToWebSocket(data.token, resolve, reject);
@@ -132,6 +133,25 @@ class RESTMethods{
.catch(reject);
});
}
UpdateCurrentUser(_data) {
return new Promise((resolve, reject) => {
let user = this.rest.client.store.user;
let data = {};
data.username = _data.username || user.username;
data.avatar = this.rest.client.resolver.ResolveBase64(_data.avatar) || user.avatar;
if (!user.bot) {
data.password = this.rest.client.store.password;
data.email = _data.email || this.rest.client.store.email;
data.new_password = _data.newPassword;
}
this.rest.makeRequest('patch', Constants.Endpoints.ME, true, data)
.then(data => resolve(this.rest.client.actions.UserUpdate.handle(data).updated))
.catch(reject);
});
}
}
module.exports = RESTMethods;

View File

@@ -3,6 +3,7 @@
const AbstractHandler = require('./AbstractHandler');
const Structure = name => require(`../../../../structures/${name}`);
const CloneObject = name => require(`../../../../util/CloneObject`);
const Constants = require(`../../../../util/Constants`);
const ClientUser = Structure('ClientUser');
const Guild = Structure('Guild');
@@ -18,22 +19,7 @@ class UserUpdateHandler extends AbstractHandler {
let data = packet.d;
let client = this.packetManager.client;
let user = client.store.user;
if (!user) {
return;
}
let oldUser = CloneObject(user);
user.username = data.username || user.username;
user.id = data.id || user.id;
user.avatar = data.avatar || user.avatar;
user.discriminator = data.discriminator || user.discriminator;
user.email = data.email || user.email;
user.verified = data.verified || user.verified;
client.emit(Constants.Events.USER_UPDATE, oldUser, user);
let response = client.actions.UserUpdate.handle(data);
}

View File

@@ -12,6 +12,26 @@ class ClientUser extends User {
this.verified = data.verified;
this.email = data.email;
}
setUsername(username) {
return this.client.rest.methods.UpdateCurrentUser({ username, });
}
setEmail(email) {
return this.client.rest.methods.UpdateCurrentUser({ email, });
}
setPassword(password) {
return this.client.rest.methods.UpdateCurrentUser({ password, });
}
setAvatar(avatar) {
return this.client.rest.methods.UpdateCurrentUser({ avatar, });
}
edit(data) {
return this.client.rest.methods.UpdateCurrentUser(data);
}
}
module.exports = ClientUser;

View File

@@ -14,13 +14,35 @@ class User {
this.discriminator = data.discriminator;
this.avatar = data.avatar;
this.bot = Boolean(data.bot);
this.status = data.status || 'offline';
this.game = data.game;
this.status = data.status || this.status || 'offline';
this.game = data.game || this.game;
}
toString() {
return `<@${this.id}>`;
}
equals(user) {
let base = (
this.username === user.username &&
this.id === user.id &&
this.discriminator === user.discriminator &&
this.avatar === user.avatar &&
this.bot === Boolean(user.bot)
);
if (base) {
if (user.status) {
base = this.status === user.status;
}
if (user.game) {
base = this.game === user.game;
}
}
return base;
}
}
module.exports = User;

View File

@@ -18,6 +18,8 @@ class ClientDataStore extends AbstractDataStore{
this.token = null;
this.session = null;
this.user = null;
this.email = null;
this.password = null;
this.register('users');
this.register('guilds');

View File

@@ -1,6 +1,7 @@
'use strict';
const Discord = require('../');
const request = require('superagent');
let client = new Discord.Client();
@@ -84,12 +85,29 @@ client.on('message', message => {
message.channel.setName(message.content.substr(8)).then(chanLoop).catch(console.log);
}
if (message.content.startsWith('botname')) {
client.user.setUsername(message.content.substr(8)).then(nameLoop).catch(console.log);
}
if (message.content.startsWith('botavatar')) {
request
.get('url')
.end((err, res) => {
client.user.setAvatar(res.body).catch(console.log)
.then(user => message.channel.sendMessage('Done!'));
});
}
if (message.content === 'leave') {
message.guild.leave().then(guild => console.log('left guild', guild.name)).catch(console.log);
}
}
});
function nameLoop(user) {
user.setUsername(user.username + 'a').then(nameLoop).catch(console.log);
}
function chanLoop(channel) {
channel.setName(channel.name + 'a').then(chanLoop).catch(console.log);
}