Browse Source

Allowing the bot to upload an avatar

develop
Drew Short 5 years ago
parent
commit
00d3d05883
  1. 38
      bot/bot.js

38
bot/bot.js

@ -1,3 +1,4 @@
let fs = require('fs');
let sdk = require('matrix-js-sdk'); let sdk = require('matrix-js-sdk');
let message = require('./message'); let message = require('./message');
let utility = require('./utility'); let utility = require('./utility');
@ -25,6 +26,7 @@ class Bot {
case "PREPARED": case "PREPARED":
this.connected = true; this.connected = true;
await this.sendStatusStartup(); await this.sendStatusStartup();
await this.updateAvatar(process.env.NODE_PATH + '/assets/avatar.jpg');
this.client.getJoinedRooms() this.client.getJoinedRooms()
.done((rooms) => { .done((rooms) => {
logger.info("Connected to: %o", rooms) logger.info("Connected to: %o", rooms)
@ -94,6 +96,42 @@ class Bot {
botClient.loginWithToken(this.config.accessToken, connectWithToken); botClient.loginWithToken(this.config.accessToken, connectWithToken);
} }
updateAvatar(avatarFile, overwrite=false) {
let matrixClient = this.client;
let botConfig = this.config;
let promises = [Promise.resolve(true)];
if (this.connected) {
promises.push(this.client.getProfileInfo(this.config.userId, "avatar_url")
.then((existingAvatarUrl) => {
logger.info("Recieved avatar_url: %o", existingAvatarUrl);
if (typeof existingAvatarUrl !== 'undefined' && typeof existingAvatarUrl == 'object'
&& existingAvatarUrl.constructor === Object && Object.keys(existingAvatarUrl).length !== 0
&& !overwrite) {
logger.info("Avatar already set");
} else {
logger.info("Setting avatar content from %s", avatarFile);
let avatarFileBuffer = fs.readFileSync(avatarFile);
logger.debug("Avatar Image Data %o", avatarFileBuffer);
matrixClient.uploadContent(avatarFileBuffer, {
name: botConfig.userId + " avatar",
type: "image/jpeg",
rawResponse: false
}).then((uploadedAvatar) => {
logger.info("Uploaded avatar %o", uploadedAvatar);
matrixClient.setAvatarUrl(uploadedAvatar.content_uri)
.then(() => {
logger.info("Updated %s avatar to %s", botConfig.userId, uploadedAvatar.content_uri);
return true;
});
});
}
}));
} else {
logger.warn("Attempting to update avatar while disconnected");
}
return Promise.all(promises);
}
sendStatusStartup() { sendStatusStartup() {
let promises = [Promise.resolve(true)] let promises = [Promise.resolve(true)]
if (this.connected) { if (this.connected) {

Loading…
Cancel
Save