hyperzlib
/
nodebb
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

feat: #7743

Browse Source 
user/password
user/picture
v1.18.x
Barış Soner Uşaklı 6 years ago
parent 0a690c5710
commit cd80c2638c
4 changed files with 132 additions and 170 deletions
Show Stats Download Patch File Download Diff File

3
src/password.js
Unescape Escape View File

@ -69,3 +69,6 @@ function done(err, result) {
process.send(err ? { err: err.message } : { result: result }); process.send(err ? { err: err.message } : { result: result });
process.disconnect(); process.disconnect();
} }
require('./promisify')(exports);

45
src/user/password.js
Unescape Escape View File

@ -1,52 +1,39 @@
'use strict'; 'use strict';
var async = require('async');
var nconf = require('nconf');
var db = require('../database'); const nconf = require('nconf');
var Password = require('../password');
const db = require('../database');
const Password = require('../password');
module.exports = function (User) { module.exports = function (User) {
User.hashPassword = function (password, callback) { User.hashPassword = async function (password) {
if (!password) { if (!password) {
return callback(null, password); return password;
} }
Password.hash(nconf.get('bcrypt_rounds') || 12, password, callback); return await Password.hash(nconf.get('bcrypt_rounds') || 12, password);
}; };
User.isPasswordCorrect = function (uid, password, ip, callback) { User.isPasswordCorrect = async function (uid, password, ip) {
password = password || ''; password = password || '';
var hashedPassword; var hashedPassword = await db.getObjectField('user:' + uid, 'password');
async.waterfall([
function (next) {
db.getObjectField('user:' + uid, 'password', next);
},
function (_hashedPassword, next) {
hashedPassword = _hashedPassword;
if (!hashedPassword) { if (!hashedPassword) {
// Non-existant user, submit fake hash for comparison // Non-existant user, submit fake hash for comparison
hashedPassword = ''; hashedPassword = '';
} }
User.isPasswordValid(password, 0, next); await User.isPasswordValid(password, 0);
}, await User.auth.logAttempt(uid, ip);
async.apply(User.auth.logAttempt, uid, ip), const ok = await Password.compare(password, hashedPassword);
function (next) {
Password.compare(password, hashedPassword, next);
},
function (ok, next) {
if (ok) { if (ok) {
User.auth.clearLoginAttempts(uid); User.auth.clearLoginAttempts(uid);
} }
next(null, ok); return ok;
},
], callback);
}; };
User.hasPassword = function (uid, callback) { User.hasPassword = async function (uid) {
db.getObjectField('user:' + uid, 'password', function (err, hashedPassword) { const hashedPassword = await db.getObjectField('user:' + uid, 'password');
callback(err, !!hashedPassword); return !!hashedPassword;
});
}; };
}; };

168
src/user/picture.js
Unescape Escape View File

@ -1,6 +1,5 @@
'use strict'; 'use strict';
var async = require('async');
var winston = require('winston'); var winston = require('winston');
var file = require('../file'); var file = require('../file');
@ -9,156 +8,129 @@ var meta = require('../meta');
var db = require('../database'); var db = require('../database');
module.exports = function (User) { module.exports = function (User) {
User.updateCoverPosition = function (uid, position, callback) { User.updateCoverPosition = async function (uid, position) {
// Reject anything that isn't two percentages // Reject anything that isn't two percentages
if (!/^[\d.]+%\s[\d.]+%$/.test(position)) { if (!/^[\d.]+%\s[\d.]+%$/.test(position)) {
winston.warn('[user/updateCoverPosition] Invalid position received: ' + position); winston.warn('[user/updateCoverPosition] Invalid position received: ' + position);
return callback(new Error('[[error:invalid-data]]')); throw new Error('[[error:invalid-data]]');
} }
User.setUserField(uid, 'cover:position', position, callback); await User.setUserField(uid, 'cover:position', position);
}; };
User.updateCoverPicture = function (data, callback) { User.updateCoverPicture = async function (data) {
var url; const picture = {
var picture = {
name: 'profileCover', name: 'profileCover',
uid: data.uid, uid: data.uid,
}; };
try {
if (!data.imageData && data.position) { if (!data.imageData && data.position) {
return User.updateCoverPosition(data.uid, data.position, callback); return await User.updateCoverPosition(data.uid, data.position);
} }
if (!data.imageData && !data.file) { if (!data.imageData && !data.file) {
return callback(new Error('[[error:invalid-data]]')); throw new Error('[[error:invalid-data]]');
} }
const size = data.file ? data.file.size : image.sizeFromBase64(data.imageData);
async.waterfall([
function (next) {
var size = data.file ? data.file.size : image.sizeFromBase64(data.imageData);
if (size > meta.config.maximumCoverImageSize * 1024) { if (size > meta.config.maximumCoverImageSize * 1024) {
return next(new Error('[[error:file-too-big, ' + meta.config.maximumCoverImageSize + ']]')); throw new Error('[[error:file-too-big, ' + meta.config.maximumCoverImageSize + ']]');
} }
if (data.file) { if (data.file) {
return setImmediate(next, null, data.file.path); picture.path = data.file.path;
} else {
picture.path = await image.writeImageDataToTempFile(data.imageData);
} }
image.writeImageDataToTempFile(data.imageData, next); const type = data.file ? data.file.type : image.mimeFromBase64(data.imageData);
},
function (path, next) {
picture.path = path;
var type = data.file ? data.file.type : image.mimeFromBase64(data.imageData);
if (!type || !type.match(/^image./)) { if (!type || !type.match(/^image./)) {
return next(new Error('[[error:invalid-image]]')); throw new Error('[[error:invalid-image]]');
} }
var extension = file.typeToExtension(type); const extension = file.typeToExtension(type);
var filename = generateProfileImageFilename(data.uid, 'profilecover', extension); const filename = generateProfileImageFilename(data.uid, 'profilecover', extension);
image.uploadImage(filename, 'profile', picture, next); const uploadData = await image.uploadImage(filename, 'profile', picture);
},
function (uploadData, next) { await User.setUserField(data.uid, 'cover:url', uploadData.url);
url = uploadData.url;
User.setUserField(data.uid, 'cover:url', uploadData.url, next);
},
function (next) {
if (data.position) { if (data.position) {
User.updateCoverPosition(data.uid, data.position, next); await User.updateCoverPosition(data.uid, data.position);
} else {
setImmediate(next);
} }
},
], function (err) { return {
file.delete(picture.path); url: uploadData.url,
callback(err, { };
url: url, } finally {
}); file.delete(picture.path || (data.file && data.file.path));
}); }
};
User.uploadCroppedPicture = async function (data) {
const picture = {
name: 'profileAvatar',
uid: data.uid,
}; };
User.uploadCroppedPicture = function (data, callback) { try {
if (!meta.config.allowProfileImageUploads) { if (!meta.config.allowProfileImageUploads) {
return callback(new Error('[[error:profile-image-uploads-disabled]]')); throw new Error('[[error:profile-image-uploads-disabled]]');
} }
if (!data.imageData && !data.file) { if (!data.imageData && !data.file) {
return callback(new Error('[[error:invalid-data]]')); throw new Error('[[error:invalid-data]]');
} }
var size = data.file ? data.file.size : image.sizeFromBase64(data.imageData); const size = data.file ? data.file.size : image.sizeFromBase64(data.imageData);
var uploadSize = meta.config.maximumProfileImageSize; const uploadSize = meta.config.maximumProfileImageSize;
if (size > uploadSize * 1024) { if (size > uploadSize * 1024) {
return callback(new Error('[[error:file-too-big, ' + uploadSize + ']]')); throw new Error('[[error:file-too-big, ' + uploadSize + ']]');
} }
var type = data.file ? data.file.type : image.mimeFromBase64(data.imageData); const type = data.file ? data.file.type : image.mimeFromBase64(data.imageData);
if (!type || !type.match(/^image./)) { if (!type || !type.match(/^image./)) {
return callback(new Error('[[error:invalid-image]]')); throw new Error('[[error:invalid-image]]');
} }
var extension = file.typeToExtension(type); const extension = file.typeToExtension(type);
if (!extension) { if (!extension) {
return callback(new Error('[[error:invalid-image-extension]]')); throw new Error('[[error:invalid-image-extension]]');
} }
var uploadedImage; if (data.file) {
picture.path = data.file.path;
} else {
picture.path = await image.writeImageDataToTempFile(data.imageData);
}
var picture = { picture.path = await convertToPNG(picture.path, extension);
name: 'profileAvatar',
uid: data.uid,
};
async.waterfall([ await image.resizeImage({
function (next) {
if (data.file) {
return setImmediate(next, null, data.file.path);
}
image.writeImageDataToTempFile(data.imageData, next);
},
function (path, next) {
convertToPNG(path, extension, next);
},
function (path, next) {
picture.path = path;
image.resizeImage({
path: picture.path, path: picture.path,
width: meta.config.profileImageDimension, width: meta.config.profileImageDimension,
height: meta.config.profileImageDimension, height: meta.config.profileImageDimension,
}, next); });
},
function (next) { const filename = generateProfileImageFilename(data.uid, 'profileavatar', extension);
var filename = generateProfileImageFilename(data.uid, 'profileavatar', extension); const uploadedImage = await image.uploadImage(filename, 'profile', picture);
image.uploadImage(filename, 'profile', picture, next);
}, await User.setUserFields(data.uid, {
function (_uploadedImage, next) {
uploadedImage = _uploadedImage;
User.setUserFields(data.uid, {
uploadedpicture: uploadedImage.url, uploadedpicture: uploadedImage.url,
picture: uploadedImage.url, picture: uploadedImage.url,
}, next);
},
], function (err) {
file.delete(picture.path);
callback(err, uploadedImage);
}); });
return uploadedImage;
} finally {
file.delete(picture.path || (data.file && data.file.path));
}
}; };
function convertToPNG(path, extension, callback) { async function convertToPNG(path, extension) {
var convertToPNG = meta.config['profile:convertProfileImageToPNG'] === 1; var convertToPNG = meta.config['profile:convertProfileImageToPNG'] === 1;
if (!convertToPNG) { if (!convertToPNG) {
return setImmediate(callback, null, path); return path;
} }
async.waterfall([ const newPath = await image.normalise(path, extension);
function (next) {
image.normalise(path, extension, next);
},
function (newPath, next) {
file.delete(path); file.delete(path);
next(null, newPath); return newPath;
},
], callback);
} }
function generateProfileImageFilename(uid, type, extension) { function generateProfileImageFilename(uid, type, extension) {
@ -167,7 +139,7 @@ module.exports = function (User) {
return uid + '-' + type + (keepAllVersions ? '-' + Date.now() : '') + (convertToPNG ? '.png' : extension); return uid + '-' + type + (keepAllVersions ? '-' + Date.now() : '') + (convertToPNG ? '.png' : extension);
} }
User.removeCoverPicture = function (data, callback) { User.removeCoverPicture = async function (data) {
db.deleteObjectFields('user:' + data.uid, ['cover:url', 'cover:position'], callback); await db.deleteObjectFields('user:' + data.uid, ['cover:url', 'cover:position']);
}; };
}; };

6
test/user.js
Unescape Escape View File

@ -899,7 +899,7 @@ describe('User', function () {
it('should return error if profile image uploads disabled', function (done) { it('should return error if profile image uploads disabled', function (done) {
meta.config.allowProfileImageUploads = 0; meta.config.allowProfileImageUploads = 0;
var picture = { var picture = {
path: path.join(nconf.get('base_dir'), 'test/files/test.png'), path: path.join(nconf.get('base_dir'), 'test/files/test_copy.png'),
size: 7189, size: 7189,
name: 'test.png', name: 'test.png',
type: 'image/png', type: 'image/png',
@ -916,7 +916,7 @@ describe('User', function () {
it('should return error if profile image is too big', function (done) { it('should return error if profile image is too big', function (done) {
meta.config.allowProfileImageUploads = 1; meta.config.allowProfileImageUploads = 1;
var picture = { var picture = {
path: path.join(nconf.get('base_dir'), 'test/files/test.png'), path: path.join(nconf.get('base_dir'), 'test/files/test_copy.png'),
size: 265000, size: 265000,
name: 'test.png', name: 'test.png',
type: 'image/png', type: 'image/png',
@ -933,7 +933,7 @@ describe('User', function () {
it('should return error if profile image has no mime type', function (done) { it('should return error if profile image has no mime type', function (done) {
var picture = { var picture = {
path: path.join(nconf.get('base_dir'), 'test/files/test.png'), path: path.join(nconf.get('base_dir'), 'test/files/test_copy.png'),
size: 7189, size: 7189,
name: 'test', name: 'test',
}; };

Write Preview
Loading…
Cancel
Save