closes #3005
parent
663ad38727
commit
5501c0c759
@ -0,0 +1,131 @@
|
||||
'use strict';
|
||||
|
||||
var async = require('async'),
|
||||
path = require('path'),
|
||||
fs = require('fs'),
|
||||
nconf = require('nconf'),
|
||||
winston = require('winston'),
|
||||
request = require('request'),
|
||||
mime = require('mime'),
|
||||
|
||||
plugins = require('../plugins'),
|
||||
file = require('../file'),
|
||||
image = require('../image'),
|
||||
meta = require('../meta');
|
||||
|
||||
module.exports = function(User) {
|
||||
|
||||
User.uploadPicture = function (uid, picture, callback) {
|
||||
|
||||
var uploadSize = parseInt(meta.config.maximumProfileImageSize, 10) || 256;
|
||||
var extension = path.extname(picture.name);
|
||||
var updateUid = uid;
|
||||
var imageDimension = parseInt(meta.config.profileImageDimension, 10) || 128;
|
||||
var convertToPNG = parseInt(meta.config['profile:convertProfileImageToPNG'], 10) === 1;
|
||||
|
||||
async.waterfall([
|
||||
function(next) {
|
||||
next(picture.size > uploadSize * 1024 ? new Error('[[error:file-too-big, ' + uploadSize + ']]') : null);
|
||||
},
|
||||
function(next) {
|
||||
next(!extension ? new Error('[[error:invalid-image-extension]]') : null);
|
||||
},
|
||||
function(next) {
|
||||
file.isFileTypeAllowed(picture.path, ['png', 'jpeg', 'jpg', 'gif'], next);
|
||||
},
|
||||
function(next) {
|
||||
image.resizeImage(picture.path, extension, imageDimension, imageDimension, next);
|
||||
},
|
||||
function(next) {
|
||||
if (convertToPNG) {
|
||||
image.convertImageToPng(picture.path, extension, next);
|
||||
} else {
|
||||
next();
|
||||
}
|
||||
}
|
||||
], function(err, result) {
|
||||
|
||||
function done(err, image) {
|
||||
if (err) {
|
||||
return callback(err);
|
||||
}
|
||||
|
||||
User.setUserFields(updateUid, {uploadedpicture: image.url, picture: image.url});
|
||||
|
||||
callback(null, image);
|
||||
}
|
||||
|
||||
if (err) {
|
||||
return callback(err);
|
||||
}
|
||||
|
||||
if (plugins.hasListeners('filter:uploadImage')) {
|
||||
return plugins.fireHook('filter:uploadImage', {image: picture, uid: updateUid}, done);
|
||||
}
|
||||
|
||||
var filename = updateUid + '-profileimg' + (convertToPNG ? '.png' : extension);
|
||||
|
||||
User.getUserField(updateUid, 'uploadedpicture', function (err, oldpicture) {
|
||||
if (err) {
|
||||
return callback(err);
|
||||
}
|
||||
|
||||
if (!oldpicture) {
|
||||
return file.saveFileToLocal(filename, 'profile', picture.path, done);
|
||||
}
|
||||
|
||||
var absolutePath = path.join(nconf.get('base_dir'), nconf.get('upload_path'), 'profile', path.basename(oldpicture));
|
||||
|
||||
fs.unlink(absolutePath, function (err) {
|
||||
if (err) {
|
||||
winston.error(err);
|
||||
}
|
||||
|
||||
file.saveFileToLocal(filename, 'profile', picture.path, done);
|
||||
});
|
||||
});
|
||||
});
|
||||
};
|
||||
|
||||
User.uploadFromUrl = function(uid, url, callback) {
|
||||
var filename = 'uid:' + uid + ':tmp-image';
|
||||
downloadFromUrl(url, filename, function(err, downloadedImage) {
|
||||
if (err) {
|
||||
return callback(err);
|
||||
}
|
||||
|
||||
User.uploadPicture(uid, downloadedImage, function(err, image) {
|
||||
fs.unlink(filename);
|
||||
callback(err, image);
|
||||
});
|
||||
});
|
||||
};
|
||||
|
||||
function downloadFromUrl(url, filename, callback) {
|
||||
request.head(url, function(err, res, body) {
|
||||
if (err) {
|
||||
return callback(err);
|
||||
}
|
||||
var uploadSize = parseInt(meta.config.maximumProfileImageSize, 10) || 256;
|
||||
var size = res.headers['content-length'];
|
||||
var type = res.headers['content-type'];
|
||||
var extension = mime.extension(type);
|
||||
|
||||
if (size > uploadSize * 1024) {
|
||||
return callback(new Error('[[error:file-too-big, ' + uploadSize + ']]'));
|
||||
}
|
||||
|
||||
request.get(url)
|
||||
.on('error', function(err) {
|
||||
winston.error(err);
|
||||
})
|
||||
.pipe(fs.createWriteStream(filename))
|
||||
.on('close', function(err) {
|
||||
if (err) {
|
||||
return callback(err);
|
||||
}
|
||||
callback(null, {path: filename, size: size, type: type, name: filename + '.' + extension});
|
||||
});
|
||||
});
|
||||
}
|
||||
};
|
Loading…
Reference in New Issue