104 lines
3.8 KiB
JavaScript
104 lines
3.8 KiB
JavaScript
const logger = require('winston');
|
|
const fs = require('fs');
|
|
const publishController = require('../controllers/publishController.js');
|
|
const publishHelpers = require('../helpers/libraries/publishHelpers.js');
|
|
const errorHandlers = require('../helpers/libraries/errorHandlers.js');
|
|
const { postToStats } = require('../controllers/statsController.js');
|
|
|
|
module.exports = (app, siofu, hostedContentPath) => {
|
|
const http = require('http');
|
|
const server = http.Server(app);
|
|
const io = require('socket.io')(server);
|
|
|
|
io.on('connection', socket => {
|
|
logger.silly('a user connected via sockets');
|
|
// attach upload listeners
|
|
const uploader = new siofu();
|
|
uploader.dir = hostedContentPath;
|
|
uploader.listen(socket);
|
|
// listener for when file upload starts
|
|
uploader.on('start', ({ file }) => {
|
|
// log
|
|
logger.info('client started an upload:', file.name);
|
|
// server side test to make sure file is not a bad file type
|
|
if (/\.exe$/.test(file.name)) {
|
|
uploader.abort(file.id, socket);
|
|
}
|
|
});
|
|
// listener for when file upload encounters an error
|
|
uploader.on('error', ({ error }) => {
|
|
logger.error('an error occured while uploading', error);
|
|
postToStats('publish', '/', null, error);
|
|
socket.emit('publish-status', error);
|
|
});
|
|
// listener for when file has been uploaded
|
|
uploader.on('saved', ({ file }) => {
|
|
if (file.success) {
|
|
logger.debug(`Client successfully uploaded ${file.name}`);
|
|
socket.emit('publish-status', 'file upload successfully completed. Your image is being published (this might take a second)...');
|
|
// prepare the publish parameters
|
|
const publishParams = publishHelpers.createPublishParams(file.meta.name, file.pathName, file.meta.license, file.meta.nsfw);
|
|
// publish the file
|
|
publishController
|
|
.publish(publishParams, file.name, file.meta.type)
|
|
.then(result => {
|
|
postToStats('publish', '/', null, 'success');
|
|
socket.emit('publish-complete', { name: publishParams.name, result });
|
|
})
|
|
.catch(error => {
|
|
error = errorHandlers.handlePublishError(error);
|
|
postToStats('publish', '/', null, error);
|
|
socket.emit('publish-failure', error);
|
|
});
|
|
} else {
|
|
logger.error(`An error occurred in uploading the client's file`);
|
|
socket.emit('publish-failure', 'file uploaded, but with errors');
|
|
postToStats('publish', '/', null, 'file uploaded, but with errors');
|
|
// to-do: remove the file if not done automatically
|
|
}
|
|
});
|
|
// handle asset requests
|
|
socket.on('asset-request', filePath => {
|
|
logger.debug('received a request for the following file', filePath);
|
|
fs.readFile(filePath, (err, assetBuffer) => {
|
|
if (err) {
|
|
logger.error('fs.readFile error', err);
|
|
return;
|
|
};
|
|
// set the data
|
|
const fileExtension = filePath.substring(filePath.lastIndexOf('.'));
|
|
console.log(fileExtension);
|
|
let data = {
|
|
type : null,
|
|
buffer: assetBuffer.toString('base64'),
|
|
};
|
|
switch (fileExtension) {
|
|
case '.jpeg' || '.jpg':
|
|
data['type'] = 'image/jpeg';
|
|
break;
|
|
case '.gif':
|
|
data['type'] = 'image/gif';
|
|
break;
|
|
case '.png':
|
|
data['type'] = 'image/png';
|
|
break;
|
|
case '.mp4':
|
|
data['type'] = 'video/mp4';
|
|
break;
|
|
default:
|
|
data['type'] = 'image/jpeg';
|
|
logger.warn('showing file with unknown type as jpeg');
|
|
break;
|
|
}
|
|
// send the asset
|
|
socket.emit('asset-transfer', data);
|
|
});
|
|
});
|
|
// handle disconnect
|
|
socket.on('disconnect', () => {
|
|
logger.silly('a user disconnected via sockets');
|
|
});
|
|
});
|
|
|
|
return server;
|
|
};
|