The communications platform that puts data protection first.
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 
 
 
 
Rocket.Chat/packages/rocketchat-lib/server/functions/setUserAvatar.js

58 lines
2.2 KiB

RocketChat.setUserAvatar = function(user, dataURI, contentType, service) {
let encoding;
let image;
if (service === 'initials') {
return RocketChat.models.Users.setAvatarOrigin(user._id, service);
} else if (service === 'url') {
let result = null;
try {
result = HTTP.get(dataURI, { npmRequestOptions: { encoding: 'binary' } });
} catch (error) {
if (!error.response || error.response.statusCode !== 404) {
console.log(`Error while handling the setting of the avatar from a url (${ dataURI }) for ${ user.username }:`, error);
throw new Meteor.Error('error-avatar-url-handling', `Error while handling avatar setting from a URL (${ dataURI }) for ${ user.username }`, { function: 'RocketChat.setUserAvatar', url: dataURI, username: user.username });
}
}
if (result.statusCode !== 200) {
console.log(`Not a valid response, ${ result.statusCode }, from the avatar url: ${ dataURI }`);
throw new Meteor.Error('error-avatar-invalid-url', `Invalid avatar URL: ${ dataURI }`, { function: 'RocketChat.setUserAvatar', url: dataURI });
}
if (!/image\/.+/.test(result.headers['content-type'])) {
console.log(`Not a valid content-type from the provided url, ${ result.headers['content-type'] }, from the avatar url: ${ dataURI }`);
throw new Meteor.Error('error-avatar-invalid-url', `Invalid avatar URL: ${ dataURI }`, { function: 'RocketChat.setUserAvatar', url: dataURI });
}
encoding = 'binary';
image = result.content;
contentType = result.headers['content-type'];
} else if (service === 'rest') {
encoding = 'binary';
image = dataURI;
} else {
const fileData = RocketChatFile.dataURIParse(dataURI);
encoding = 'base64';
image = fileData.image;
contentType = fileData.contentType;
}
const buffer = new Buffer(image, encoding);
const fileStore = FileUpload.getStore('Avatars');
fileStore.deleteByName(user.username);
const file = {
userId: user._id,
type: contentType,
size: buffer.length,
};
fileStore.insert(file, buffer, () => {
Meteor.setTimeout(function() {
RocketChat.models.Users.setAvatarOrigin(user._id, service);
RocketChat.Notifications.notifyLogged('updateAvatar', { username: user.username });
}, 500);
});
};